1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
module U_Result = struct
let map fn r = match r with Ok v -> Ok (fn v) | Error e -> Error e
let flat_map fn r = match r with Ok v -> fn v | Error e -> Error e
let map_error fn r = match r with Ok v -> Ok v | Error e -> Error (fn e)
let to_opt = function Ok v -> Some v | Error _ -> None
let return v = Ok v
let get_exn = function Ok v -> v | Error _ -> raise Not_found
let both a b =
match (a, b) with
| Ok a, Ok b -> Ok (a, b)
| Error e, _ -> Error e
| _, Error e -> Error e
let bind v f = match v with Ok v -> f v | Error _ as e -> e
let all8 a b c d e f g h =
match (a, b, c, d, e, f, g, h) with
| Ok a, Ok b, Ok c, Ok d, Ok e, Ok f, Ok g, Ok h ->
Ok (a, b, c, d, e, f, g, h)
| _ -> Error (`Msg "all 8 was not Ok")
module Infix = struct
let ( >>= ) = bind
let ( >|= ) r fn = match r with Ok v -> Ok (fn v) | Error e -> Error e
end
end
module U_Opt = struct
let flatten o = match o with Some v -> v | None -> None
let map fn o = match o with Some v -> Some (fn v) | None -> None
let get_with_default ~default o = match o with Some v -> v | None -> default
let return x = Some x
end
module U_List = struct
let filter_map f =
let rec aux accu = function
| [] -> List.rev accu
| x :: l -> (
match f x with None -> aux accu l | Some v -> aux (v :: accu) l)
in
aux []
let rec find_opt p = function
| [] -> None
| x :: l -> if p x then Some x else find_opt p l
end
module U_String = struct
let rev s =
let len = Astring.String.length s in
Astring.String.mapi (fun i _ -> s.[len - (i + 1)]) s
let pad ~c length s =
let len = Astring.String.length s in
if len >= length then s
else
let diff = length - len in
Astring.String.v ~len:length (fun i ->
if i < diff then c else s.[i - diff])
let trim_leading_null s =
Astring.String.trim ~drop:(function '\000' -> true | _ -> false) s
end
module U_Base64 = struct
let url_encode_string ?(pad = false) payload =
Base64.encode_string ~pad ~alphabet:Base64.uri_safe_alphabet payload
let url_encode ?(pad = false) ?off ?len payload =
Base64.encode ~pad ~alphabet:Base64.uri_safe_alphabet ?off ?len payload
let url_decode ?(pad = false) ?off ?len payload =
Base64.decode ~pad ~alphabet:Base64.uri_safe_alphabet ?off ?len payload
end
module RJson = struct
let to_json_string_opt key value =
match value with Some s -> Some (key, `String s) | None -> None
end
module Pkcs7 = struct
let pad data block_size =
let pad_size = block_size - (Cstruct.length data mod block_size) in
if pad_size = 0 then data
else
let pad = Cstruct.create pad_size in
Cstruct.memset pad pad_size;
Cstruct.append data pad
let unpad cs =
let cs_len = Cstruct.length cs in
let pad_len = Cstruct.get_uint8 cs (cs_len - 1) in
let data, padding = Cstruct.split cs (cs_len - pad_len) in
let rec check idx =
if idx >= pad_len then true
else Cstruct.get_uint8 padding idx = pad_len && check (idx + 1)
in
if check 0 then Ok data else Error (`Msg "bad padding")
end