Source file `inflate.ml`

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
(*
 * Copyright (c) 2013-2017 Thomas Gazagnaire <thomas@gazagnaire.org>
 * and Romain Calascibetta <romain.calascibetta@gmail.com>
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 *)

open Decompress
module Inflate = Zlib_inflate

type t = (Buffer.Bigstring.t, Buffer.Bigstring.t) Inflate.t
type error = [`Inflate of Inflate.error]
type window = (Buffer.Bigstring.t, Window.adler32) Window.t

let src = Logs.Src.create "decompress.inflate" ~doc:"logs inflate event"

module Log = (val Logs.src_log src : Logs.LOG)

let pp_error : error Fmt.t =
 fun ppf -> function
  | `Inflate err -> Fmt.pf ppf "(`Inflate %a)" (Fmt.hvbox Inflate.pp_error) err

let pp : t Fmt.t = Inflate.pp
let window_reset : window -> window = Window.reset
let window () = Window.create ~crc:Window.adler32 ~witness:Buffer.bigstring
let default : window -> t = fun w -> Inflate.default ~witness:Buffer.bigstring w

let eval ~src ~dst t :
    [`Await of t | `End of t | `Error of t * error | `Flush of t] =
  let src' = Cstruct.to_bigarray src in
  let dst' = Cstruct.to_bigarray dst in
  Inflate.eval src' dst' t
  |> function
  | `Error (t, err) -> `Error (t, `Inflate err)
  | (`Await _ | `Flush _ | `End _) as x -> x

let used_in : t -> int = Inflate.used_in
let used_out : t -> int = Inflate.used_out
let write : t -> int = Inflate.write
let flush : int -> int -> t -> t = Inflate.flush
let refill : int -> int -> t -> t = Inflate.refill