Source file `util.ml`

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
open Core_kernel

(* utility function for bin-io'ing out of a Bigstring.t *)
let bin_read_from_bigstring (bin_reader_t : _ Bin_prot.Type_class.reader)
      ?add_len buf ~pos_ref ~(len : Nat0.t) ~location =
  try
    let init_pos = !pos_ref in
    let data = bin_reader_t.read buf ~pos_ref in
    let add_len =
      match add_len with
      | None -> 0
      | Some add_len -> add_len data
    in
    if !pos_ref - init_pos + add_len <> (len :> int) then
      failwithf "message length (%d) did not match expected length (%d)"
        (!pos_ref - init_pos) (len : Nat0.t :> int) ();
    Ok data
  with
  | e -> Rpc_result.bin_io_exn ~location e