-
Notifications
You must be signed in to change notification settings - Fork 125
/
buf.ml
171 lines (138 loc) · 5.08 KB
/
buf.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
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
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
(*
Copyright © 2011 MLstate
This file is part of OPA.
OPA is free software: you can redistribute it and/or modify it under the
terms of the GNU Affero General Public License, version 3, as published by
the Free Software Foundation.
OPA is distributed in the hope that it will be useful, but WITHOUT ANY
WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for
more details.
You should have received a copy of the GNU Affero General Public License
along with OPA. If not, see <http://www.gnu.org/licenses/>.
*)
(* Simple library, like Buffer but fixed size but can also look like String if required *)
type buf = { mutable str : string; mutable i : int }
type t = buf
type resize_mode =
| RM_stdout
| RM_stderr
| RM_custom of (string -> unit)
| RM_failwith
| RM_exit
| RM_noresize
let auto_resize = ref RM_stderr
let empty () = { str=""; i=0; }
let create size = { str=String.create size; i=0; }
let make size ch = { str=String.make size ch; i=size; }
(* More conservative than Buffer, we grow more slowly and
* we allow shrinkage by giving negative values to extra.
*)
let resize buf extra =
let strlen = String.length buf.str in
let target = max 0 (if extra >= 0 then buf.i + extra else strlen + extra) in
let newsize =
if extra >= 0
then
let newsize = ref (max strlen 2) in
while !newsize < target do
newsize := max (!newsize+1) ((!newsize + !newsize + !newsize) / 2)
done;
if !newsize > Sys.max_string_length
then
if target <= Sys.max_string_length
then Sys.max_string_length
else failwith "Buf.resize: cannot increase size of buffer"
else !newsize
else target
in
let str = String.create newsize in
let newlen = min buf.i newsize in
if buf.i > 0 then String.unsafe_blit buf.str 0 str 0 newlen;
buf.str <- str;
buf.i <- newlen;
if newsize > strlen
then
let msg = Printf.sprintf "Buf.resize called (now %d), please resize your buffers" newsize in
match !auto_resize with
| RM_stdout -> Printf.printf "%s\n%!" msg
| RM_stderr -> Printf.eprintf "%s\n%!" msg
| RM_custom f -> f msg
| RM_failwith -> failwith msg
| RM_exit -> exit 1
| RM_noresize -> ()
let autoresize buf extra msg =
if !auto_resize <> RM_noresize
then resize buf extra
else invalid_arg msg
let copy buf = { str=String.copy buf.str; i=buf.i }
let clear buf = buf.i <- 0
let reset buf = buf.str <- ""; buf.i <- 0
let length buf = buf.i
let real_length buf = String.length buf.str
let get buf i =
if i < 0 || i >= buf.i then invalid_arg (Printf.sprintf "Buf.get index out of bounds %d" i);
String.get buf.str i
let nth = get
let unsafe_get buf i = String.unsafe_get buf.str i
let set buf i ch =
if i < 0 || i >= buf.i then invalid_arg (Printf.sprintf "Buf.set index out of bounds %d" i);
String.set buf.str i ch
let unsafe_set buf i ch = String.unsafe_set buf.str i ch
let sub buf base len =
if base < 0 || base + len > buf.i then invalid_arg (Printf.sprintf "Buf.sub index out of bounds %d %d" base len);
String.sub buf.str base len
let add_char buf ch =
if String.length buf.str - buf.i < 1 then autoresize buf 1 (Printf.sprintf "Buf.add_char %c" ch);
buf.str.[buf.i] <- ch;
buf.i <- buf.i + 1
let add_substring buf str base len =
if String.length buf.str - buf.i < len then autoresize buf len (Printf.sprintf "Buf.add_substring %s %d %d" str base len);
String.unsafe_blit str base buf.str buf.i len;
buf.i <- buf.i + len
let append buf str len = add_substring buf str 0 len
let extend buf len =
if String.length buf.str - buf.i < len then autoresize buf len (Printf.sprintf "Buf.extend %d" len);
buf.i <- buf.i + len
let add_string buf str =
append buf str (String.length str)
let add_buf buf1 buf2 =
append buf1 buf2.str buf2.i
let of_string str = { str; i=String.length str; }
let to_string buf = String.sub buf.str 0 buf.i
let contents = to_string
let spare buf = String.length buf.str - buf.i
(* Test code *)
(*
let buf = of_string "abc";;
let () = set buf 1 'B';;
let ch = get buf 0;;
let ch = get buf 1;;
let ch = get buf 2;;
let str = try ignore (get buf 3); "NOT OK" with Invalid_argument str -> "OK: "^str;;
let str = try ignore (set buf 4 'x'); "NOT OK" with Invalid_argument str -> "OK: "^str;;
let buf = create 5;;
let () = add_char buf 'D';;
let () = add_string buf "ef";;
let len = length buf;;
let rlen = real_length buf;;
let str = to_string buf;;
let () = resize buf 8;;
let len = length buf;;
let rlen = real_length buf;;
let str = to_string buf;;
let () = resize buf (-2);;
let len = length buf;;
let rlen = real_length buf;;
let str = to_string buf;;
let str =
try ignore (add_char (empty()) 'x');
if !auto_resize <> RM_noresize then "OK: resized" else "NOT OK"
with Invalid_argument str ->
if !auto_resize <> RM_noresize then "NOT OK" else "OK: "^str;;
let () = resize buf 10;;
let () = add_string buf "fghi";;
let str = to_string buf;;
let str = sub buf 1 3;;
let str = try ignore (sub buf 100 100); "NOT OK" with Invalid_argument str -> "OK: "^str;;
*)