Merge: serialization: redef inspect for an output useful to humans
[nit.git] / lib / core / bytes.nit
1 # This file is part of NIT ( http://www.nitlanguage.org ).
2 #
3 # Licensed under the Apache License, Version 2.0 (the "License");
4 # you may not use this file except in compliance with the License.
5 # You may obtain a copy of the License at
6 #
7 # http://www.apache.org/licenses/LICENSE-2.0
8 #
9 # Unless required by applicable law or agreed to in writing, software
10 # distributed under the License is distributed on an "AS IS" BASIS,
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
14
15 # Services for byte streams and arrays
16 module bytes
17
18 import kernel
19 import collection::array
20 intrude import text::flat
21
22 # Any kind of entity which can be searched for in a Sequence of Byte
23 interface BytePattern
24 # Return the first occurence of `self` in `b`, or -1 if not found
25 fun first_index_in(b: SequenceRead[Byte]): Int do return first_index_in_from(b, 0)
26
27 # Return the first occurence of `self` in `b` starting at `from`, or -1 if not found
28 fun first_index_in_from(b: SequenceRead[Byte], from: Int): Int is abstract
29
30 # Return the last occurence of `self` in `b`, or -1 if not found
31 fun last_index_in(b: SequenceRead[Byte]): Int do return last_index_in_from(b, b.length - 1)
32
33 # Return the last occurence of `self` in `b`, or -1 if not found
34 fun last_index_in_from(b: SequenceRead[Byte], from: Int): Int is abstract
35
36 # Returns the indexes of all the occurences of `self` in `b`
37 fun search_all_in(b: SequenceRead[Byte]): SequenceRead[Int] is abstract
38
39 # Length of the pattern
40 fun pattern_length: Int is abstract
41
42 # Appends `self` to `b`
43 fun append_to(b: Sequence[Byte]) is abstract
44
45 # Is `self` a prefix for `b` ?
46 fun is_prefix(b: SequenceRead[Byte]): Bool is abstract
47
48 # Is `self` a suffix for `b` ?
49 fun is_suffix(b: SequenceRead[Byte]): Bool is abstract
50 end
51
52 redef class Byte
53 super BytePattern
54
55 # Write self as a string into `ns` at position `pos`
56 private fun add_digest_at(ns: CString, pos: Int) do
57 var tmp = (0xF0u8 & self) >> 4
58 ns[pos] = if tmp >= 0x0Au8 then tmp + 0x37u8 else tmp + 0x30u8
59 tmp = 0x0Fu8 & self
60 ns[pos + 1] = if tmp >= 0x0Au8 then tmp + 0x37u8 else tmp + 0x30u8
61 end
62
63 # Is `self` a valid hexadecimal digit (in ASCII)
64 #
65 # ~~~nit
66 # intrude import core::bytes
67 # assert not '/'.ascii.is_valid_hexdigit
68 # assert '0'.ascii.is_valid_hexdigit
69 # assert '9'.ascii.is_valid_hexdigit
70 # assert not ':'.ascii.is_valid_hexdigit
71 # assert not '@'.ascii.is_valid_hexdigit
72 # assert 'A'.ascii.is_valid_hexdigit
73 # assert 'F'.ascii.is_valid_hexdigit
74 # assert not 'G'.ascii.is_valid_hexdigit
75 # assert not '`'.ascii.is_valid_hexdigit
76 # assert 'a'.ascii.is_valid_hexdigit
77 # assert 'f'.ascii.is_valid_hexdigit
78 # assert not 'g'.ascii.is_valid_hexdigit
79 # ~~~
80 private fun is_valid_hexdigit: Bool do
81 return (self >= 0x30u8 and self <= 0x39u8) or
82 (self >= 0x41u8 and self <= 0x46u8) or
83 (self >= 0x61u8 and self <= 0x66u8)
84 end
85
86 # `self` as a hexdigit to its byte value
87 #
88 # ~~~nit
89 # intrude import core::bytes
90 # assert 0x39u8.hexdigit_to_byteval == 0x09u8
91 # assert 0x43u8.hexdigit_to_byteval == 0x0Cu8
92 # ~~~
93 #
94 # REQUIRE: `self.is_valid_hexdigit`
95 private fun hexdigit_to_byteval: Byte do
96 if self >= 0x30u8 and self <= 0x39u8 then
97 return self - 0x30u8
98 else if self >= 0x41u8 and self <= 0x46u8 then
99 return self - 0x37u8
100 else if self >= 0x61u8 and self <= 0x66u8 then
101 return self - 0x57u8
102 end
103 # Happens only if the requirement is not met.
104 # i.e. this abort is here to please the compiler
105 abort
106 end
107
108 redef fun first_index_in_from(b, from) do
109 for i in [from .. b.length[ do if b[i] == self then return i
110 return -1
111 end
112
113 redef fun last_index_in_from(b, from) do
114 for i in [0 .. from].step(-1) do if b[i] == self then return i
115 return -1
116 end
117
118 redef fun search_all_in(b) do
119 var ret = new Array[Int]
120 var pos = 0
121 loop
122 pos = first_index_in_from(b, pos)
123 if pos == -1 then return ret
124 ret.add pos
125 pos += 1
126 end
127 end
128
129 redef fun pattern_length do return 1
130
131 redef fun append_to(b) do b.push self
132
133 # assert 'b'.ascii.is_suffix("baqsdb".to_bytes)
134 # assert not 'b'.ascii.is_suffix("baqsd".to_bytes)
135 redef fun is_suffix(b) do return b.length != 0 and b.last == self
136
137 # assert 'b'.ascii.is_prefix("baqsdb".to_bytes)
138 # assert not 'b'.ascii.is_prefix("aqsdb".to_bytes)
139 redef fun is_prefix(b) do return b.length != 0 and b.first == self
140 end
141
142 # A buffer containing Byte-manipulation facilities
143 #
144 # Uses Copy-On-Write when persisted
145 class Bytes
146 super AbstractArray[Byte]
147 super BytePattern
148
149 # A CString being a char*, it can be used as underlying representation here.
150 var items: CString
151
152 # Number of bytes in the array
153 redef var length
154
155 # Capacity of the array
156 private var capacity: Int
157
158 # Has this buffer been persisted (to_s'd)?
159 #
160 # Used for Copy-On-Write
161 private var persisted = false
162
163 # var b = new Bytes.empty
164 # assert b.to_s == ""
165 init empty do
166 var ns = new CString(0)
167 init(ns, 0, 0)
168 end
169
170 # Init a `Bytes` with capacity `cap`
171 init with_capacity(cap: Int) do
172 var ns = new CString(cap)
173 init(ns, 0, cap)
174 end
175
176 redef fun pattern_length do return length
177
178 redef fun is_empty do return length == 0
179
180 # var b = new Bytes.empty
181 # b.add 101u8
182 # assert b[0] == 101u8
183 redef fun [](i) do
184 assert i >= 0
185 assert i < length
186 return items[i]
187 end
188
189 # Returns a copy of `self`
190 fun clone: Bytes do
191 var b = new Bytes.with_capacity(length)
192 b.append(self)
193 return b
194 end
195
196 # Trims off the whitespaces at the beginning and the end of `self`
197 #
198 # var b = "102041426E6F1020" .hexdigest_to_bytes
199 # assert b.trim.hexdigest == "41426E6F"
200 #
201 # NOTE: A whitespace is defined here as a byte whose value is <= 0x20
202 fun trim: Bytes do
203 var st = 0
204 while st < length do
205 if self[st] > 0x20u8 then break
206 st += 1
207 end
208 if st >= length then return new Bytes.empty
209 var ed = length - 1
210 while ed > 0 do
211 if self[ed] > 0x20u8 then break
212 ed -= 1
213 end
214 return slice(st, ed - st + 1)
215 end
216
217 # Copy a subset of `self` starting at `from` and of `count` bytes
218 #
219 # var b = "abcd".to_bytes
220 # assert b.slice(1, 2).hexdigest == "6263"
221 # assert b.slice(-1, 2).hexdigest == "61"
222 # assert b.slice(1, 0).hexdigest == ""
223 # assert b.slice(2, 5).hexdigest == "6364"
224 fun slice(from, count: Int): Bytes do
225 if count <= 0 then return new Bytes.empty
226
227 if from < 0 then
228 count += from
229 if count < 0 then count = 0
230 from = 0
231 end
232
233 if (count + from) > length then count = length - from
234 if count <= 0 then return new Bytes.empty
235
236 var ret = new Bytes.with_capacity(count)
237
238 ret.append_ns(items.fast_cstring(from), count)
239 return ret
240 end
241
242 # Copy of `self` starting at `from`
243 #
244 # var b = "abcd".to_bytes
245 # assert b.slice_from(1).hexdigest == "626364"
246 # assert b.slice_from(-1).hexdigest == "61626364"
247 # assert b.slice_from(2).hexdigest == "6364"
248 fun slice_from(from: Int): Bytes do
249 if from >= length then return new Bytes.empty
250 if from < 0 then from = 0
251 return slice(from, length)
252 end
253
254 # Returns self as an hexadecimal digest.
255 #
256 # Also known as plain hexdump or postscript hexdump.
257 #
258 # ~~~
259 # var b = "abcd".to_bytes
260 # assert b.hexdigest == "61626364"
261 # assert b.hexdigest.hexdigest_to_bytes == b
262 # ~~~
263 fun hexdigest: String do
264 var elen = length * 2
265 var ns = new CString(elen)
266 var i = 0
267 var oi = 0
268 while i < length do
269 self[i].add_digest_at(ns, oi)
270 i += 1
271 oi += 2
272 end
273 return new FlatString.full(ns, elen, 0, elen)
274 end
275
276 # Return self as a C hexadecimal digest where bytes are prefixed by `\x`
277 #
278 # The output is compatible with literal stream of bytes for most languages
279 # including C and Nit.
280 #
281 # ~~~
282 # var b = "abcd".to_bytes
283 # assert b.chexdigest == "\\x61\\x62\\x63\\x64"
284 # assert b.chexdigest.unescape_to_bytes == b
285 # ~~~
286 fun chexdigest: String do
287 var elen = length * 4
288 var ns = new CString(elen)
289 var i = 0
290 var oi = 0
291 while i < length do
292 ns[oi] = 0x5Cu8 # b'\\'
293 ns[oi+1] = 0x78u8 # b'x'
294 self[i].add_digest_at(ns, oi+2)
295 i += 1
296 oi += 4
297 end
298 return new FlatString.full(ns, elen, 0, elen)
299 end
300
301
302 # Returns self as a stream of bits (0 and 1)
303 #
304 # ~~~
305 # var b = "abcd".to_bytes
306 # assert b.binarydigest == "01100001011000100110001101100100"
307 # assert b.binarydigest.binarydigest_to_bytes == b
308 # ~~~
309 fun binarydigest: String do
310 var elen = length * 8
311 var ns = new CString(elen)
312 var i = 0
313 var oi = 0
314 while i < length do
315 var c = self[i]
316 var b = 128u8
317 while b > 0u8 do
318 if c & b == 0u8 then
319 ns[oi] = 0x30u8 # b'0'
320 else
321 ns[oi] = 0x31u8 # b'1'
322 end
323 oi += 1
324 b = b >> 1
325 end
326 i += 1
327 end
328 return new FlatString.full(ns, elen, 0, elen)
329 end
330
331 # Interprets `self` as a big-endian positive integer.
332 #
333 # ~~~
334 # var b = "0102".hexdigest_to_bytes
335 # assert b.to_i == 258
336 # ~~~
337 #
338 # Nul bytes on the left are trimmed.
339 # 0 is returned for an empty Bytes object.
340 #
341 # ~~~
342 # assert "01".hexdigest_to_bytes.to_i == 1
343 # assert "0001".hexdigest_to_bytes.to_i == 1
344 #
345 # assert "0000".hexdigest_to_bytes.to_i == 0
346 # assert "00".hexdigest_to_bytes.to_i == 0
347 # assert "".hexdigest_to_bytes.to_i == 0
348 # ~~~
349 #
350 # `Int::to_bytes` is a loosely reverse method.
351 #
352 # ~~~
353 # assert b.to_i.to_bytes == b
354 # assert (b.to_i + 1).to_bytes.hexdigest == "0103"
355 # assert "0001".hexdigest_to_bytes.to_i.to_bytes.hexdigest == "01"
356 # ~~~
357 #
358 # Warning: `Int` might overflow for bytes with more than 60 bits.
359 fun to_i: Int do
360 var res = 0
361 var i = 0
362 while i < length do
363 res *= 256
364 res += self[i].to_i
365 i += 1
366 end
367 return res
368 end
369
370 # var b = new Bytes.with_capacity(1)
371 # b[0] = 101u8
372 # assert b.to_s == "e"
373 redef fun []=(i, v) do
374 if persisted then regen
375 assert i >= 0
376 assert i <= length
377 if i == length then add(v)
378 items[i] = v
379 end
380
381 # var b = new Bytes.empty
382 # b.add 101u8
383 # assert b.to_s == "e"
384 redef fun add(c) do
385 if persisted then regen
386 if length >= capacity then
387 enlarge(length)
388 end
389 items[length] = c
390 length += 1
391 end
392
393 # Adds the UTF-8 representation of `c` to `self`
394 #
395 # var b = new Bytes.empty
396 # b.add_char('A')
397 # b.add_char('キ')
398 # assert b.hexdigest == "41E382AD"
399 fun add_char(c: Char) do
400 if persisted then regen
401 var cln = c.u8char_len
402 var ln = length
403 enlarge(ln + cln)
404 items.set_char_at(length, c)
405 length += cln
406 end
407
408 # var b = new Bytes.empty
409 # b.append([104u8, 101u8, 108u8, 108u8, 111u8])
410 # assert b.to_s == "hello"
411 redef fun append(arr) do
412 if arr isa Bytes then
413 append_ns(arr.items, arr.length)
414 else
415 for i in arr do add i
416 end
417 end
418
419 # var b = new Bytes.empty
420 # b.append([0x41u8, 0x41u8, 0x18u8])
421 # b.pop
422 # assert b.to_s == "AA"
423 redef fun pop do
424 assert length >= 1
425 length -= 1
426 return items[length]
427 end
428
429 redef fun clear do length = 0
430
431 # Regenerates the buffer, necessary when it was persisted
432 private fun regen do
433 var nns = new CString(capacity)
434 items.copy_to(nns, length, 0, 0)
435 persisted = false
436 end
437
438 # Appends the `ln` first bytes of `ns` to self
439 fun append_ns(ns: CString, ln: Int) do
440 if persisted then regen
441 var nlen = length + ln
442 if nlen > capacity then enlarge(nlen)
443 ns.copy_to(items, ln, 0, length)
444 length += ln
445 end
446
447 # Appends `ln` bytes from `ns` starting at index `from` to self
448 fun append_ns_from(ns: CString, ln, from: Int) do
449 if persisted then regen
450 var nlen = length + ln
451 if nlen > capacity then enlarge(nlen)
452 ns.copy_to(items, ln, from, length)
453 length += ln
454 end
455
456 # Appends the bytes of `str` to `self`
457 fun append_text(str: Text) do str.append_to_bytes self
458
459 redef fun append_to(b) do b.append self
460
461 redef fun enlarge(sz) do
462 if capacity >= sz then return
463 persisted = false
464 if capacity < 16 then capacity = 16
465 while capacity < sz do capacity = capacity * 2 + 2
466 var ns = new CString(capacity)
467 items.copy_to(ns, length, 0, 0)
468 items = ns
469 end
470
471 redef fun to_s do
472 persisted = true
473 var b = self
474 var r = b.items.to_s_unsafe(length, copy=false)
475 if r != items then persisted = false
476 return r
477 end
478
479 redef fun iterator do return new BytesIterator.with_buffer(self)
480
481 redef fun first_index_in_from(b, from) do
482 if is_empty then return -1
483 var fst = self[0]
484 var bpos = fst.first_index_in_from(self, from)
485 for i in [0 .. length[ do
486 if self[i] != b[bpos] then return first_index_in_from(b, bpos + 1)
487 bpos += 1
488 end
489 return bpos
490 end
491
492 redef fun last_index_in_from(b, from) do
493 if is_empty then return -1
494 var lst = self[length - 1]
495 var bpos = lst.last_index_in_from(b, from)
496 for i in [0 .. length[.step(-1) do
497 if self[i] != b[bpos] then return last_index_in_from(b, bpos - 1)
498 bpos -= 1
499 end
500 return bpos
501 end
502
503 redef fun search_all_in(b) do
504 var ret = new Array[Int]
505 var pos = first_index_in_from(b, 0)
506 if pos == -1 then return ret
507 pos = pos + 1
508 ret.add pos
509 loop
510 pos = first_index_in_from(b, pos)
511 if pos == -1 then return ret
512 ret.add pos
513 pos += length
514 end
515 end
516
517 # Splits the content on self when encountering `b`
518 #
519 # var a = "String is string".to_bytes.split_with('s'.ascii)
520 # assert a.length == 3
521 # assert a[0].hexdigest == "537472696E672069"
522 # assert a[1].hexdigest == "20"
523 # assert a[2].hexdigest == "7472696E67"
524 fun split_with(b: BytePattern): Array[Bytes] do
525 var fst = b.search_all_in(self)
526 if fst.is_empty then return [clone]
527 var retarr = new Array[Bytes]
528 var prev = 0
529 for i in fst do
530 retarr.add(slice(prev, i - prev))
531 prev = i + b.pattern_length
532 end
533 retarr.add slice_from(prev)
534 return retarr
535 end
536
537 # Splits `self` in two parts at the first occurence of `b`
538 #
539 # var a = "String is string".to_bytes.split_once_on('s'.ascii)
540 # assert a[0].hexdigest == "537472696E672069"
541 # assert a[1].hexdigest == "20737472696E67"
542 fun split_once_on(b: BytePattern): Array[Bytes] do
543 var spl = b.first_index_in(self)
544 if spl == -1 then return [clone]
545 var ret = new Array[Bytes].with_capacity(2)
546 ret.add(slice(0, spl))
547 ret.add(slice_from(spl + b.pattern_length))
548 return ret
549 end
550
551 # Replaces all the occurences of `this` in `self` by `by`
552 #
553 # var b = "String is string".to_bytes.replace(0x20u8, 0x41u8)
554 # assert b.hexdigest == "537472696E6741697341737472696E67"
555 fun replace(pattern: BytePattern, bytes: BytePattern): Bytes do
556 if is_empty then return new Bytes.empty
557 var pos = pattern.search_all_in(self)
558 if pos.is_empty then return clone
559 var ret = new Bytes.with_capacity(length)
560 var prev = 0
561 for i in pos do
562 ret.append_ns(items.fast_cstring(prev), i - prev)
563 bytes.append_to ret
564 prev = i + pattern.pattern_length
565 end
566 ret.append(slice_from(pos.last + pattern.pattern_length))
567 return ret
568 end
569
570 # Decode `self` from percent (or URL) encoding to a clear string
571 #
572 # Invalid '%' are not decoded.
573 #
574 # assert "aBc09-._~".to_bytes.from_percent_encoding == "aBc09-._~".to_bytes
575 # assert "%25%28%29%3c%20%3e".to_bytes.from_percent_encoding == "%()< >".to_bytes
576 # assert ".com%2fpost%3fe%3dasdf%26f%3d123".to_bytes.from_percent_encoding == ".com/post?e=asdf&f=123".to_bytes
577 # assert "%25%28%29%3C%20%3E".to_bytes.from_percent_encoding == "%()< >".to_bytes
578 # assert "incomplete %".to_bytes.from_percent_encoding == "incomplete %".to_bytes
579 # assert "invalid % usage".to_bytes.from_percent_encoding == "invalid % usage".to_bytes
580 # assert "%c3%a9%e3%81%82%e3%81%84%e3%81%86".to_bytes.from_percent_encoding == "éあいう".to_bytes
581 # assert "%1 %A %C3%A9A9".to_bytes.from_percent_encoding == "%1 %A éA9".to_bytes
582 fun from_percent_encoding: Bytes do
583 var tmp = new Bytes.with_capacity(length)
584 var pos = 0
585 while pos < length do
586 var b = self[pos]
587 if b != '%'.ascii then
588 tmp.add b
589 pos += 1
590 continue
591 end
592 if length - pos < 2 then
593 tmp.add '%'.ascii
594 pos += 1
595 continue
596 end
597 var bn = self[pos + 1]
598 var bnn = self[pos + 2]
599 if not bn.is_valid_hexdigit or not bnn.is_valid_hexdigit then
600 tmp.add '%'.ascii
601 pos += 1
602 continue
603 end
604 tmp.add((bn.hexdigit_to_byteval << 4) + bnn.hexdigit_to_byteval)
605 pos += 3
606 end
607 return tmp
608 end
609
610 # Is `b` a prefix of `self` ?
611 fun has_prefix(b: BytePattern): Bool do return b.is_prefix(self)
612
613 # Is `b` a suffix of `self` ?
614 fun has_suffix(b: BytePattern): Bool do return b.is_suffix(self)
615
616 redef fun is_suffix(b) do
617 if length > b.length then return false
618 var j = b.length - 1
619 var i = length - 1
620 while i > 0 do
621 if self[i] != b[j] then return false
622 i -= 1
623 j -= 1
624 end
625 return true
626 end
627
628 redef fun is_prefix(b) do
629 if length > b.length then return false
630 for i in [0 .. length[ do if self[i] != b[i] then return false
631 return true
632 end
633 end
634
635 private class BytesIterator
636 super IndexedIterator[Byte]
637
638 var tgt: CString
639
640 redef var index
641
642 var max: Int
643
644 init with_buffer(b: Bytes) do init(b.items, 0, b.length)
645
646 redef fun is_ok do return index < max
647
648 redef fun next do index += 1
649
650 redef fun item do return tgt[index]
651 end
652
653 redef class Int
654 # A big-endian representation of self.
655 #
656 # ~~~
657 # assert 1.to_bytes.hexdigest == "01"
658 # assert 255.to_bytes.hexdigest == "FF"
659 # assert 256.to_bytes.hexdigest == "0100"
660 # assert 65535.to_bytes.hexdigest == "FFFF"
661 # assert 65536.to_bytes.hexdigest == "010000"
662 # ~~~
663 #
664 # For 0, a Bytes object with single nul byte is returned (instead of an empty Bytes object).
665 #
666 # ~~~
667 # assert 0.to_bytes.hexdigest == "00"
668 # ~~~
669 #
670 # `Bytes::to_i` can be used to do the reverse operation.
671 #
672 # ~~~
673 # assert 1234.to_bytes.to_i == 1234
674 # ~~~
675 #
676 # Require self >= 0
677 fun to_bytes: Bytes do
678 if self == 0 then return "\0".to_bytes
679 assert self > 0
680
681 # Compute the len (log256)
682 var len = 1
683 var max = 256
684 while self >= max do
685 len += 1
686 max *= 256
687 end
688
689 # Allocate the buffer
690 var res = new Bytes.with_capacity(len)
691 for i in [0..len[ do res[i] = 0u8
692
693 # Fill it starting with the end
694 var i = len
695 var sum = self
696 while i > 0 do
697 i -= 1
698 res[i] = (sum % 256).to_b
699 sum /= 256
700 end
701 return res
702 end
703 end
704
705 redef class Text
706 # Returns a mutable copy of `self`'s bytes
707 #
708 # ~~~nit
709 # assert "String".to_bytes isa Bytes
710 # assert "String".to_bytes == [83u8, 116u8, 114u8, 105u8, 110u8, 103u8]
711 # ~~~
712 fun to_bytes: Bytes do
713 var b = new Bytes.with_capacity(byte_length)
714 append_to_bytes b
715 return b
716 end
717
718 # Is `self` a valid hexdigest ?
719 #
720 # assert "0B1d3F".is_valid_hexdigest
721 # assert not "5G".is_valid_hexdigest
722 fun is_valid_hexdigest: Bool do
723 for i in bytes do if not i.is_valid_hexdigit then return false
724 return true
725 end
726
727 # Appends `self.bytes` to `b`
728 fun append_to_bytes(b: Bytes) do
729 for s in substrings do
730 var from = if s isa FlatString then s.first_byte else 0
731 b.append_ns_from(s.items, s.byte_length, from)
732 end
733 end
734
735 # Returns a new `Bytes` instance with the digest as content
736 #
737 # assert "0B1F4D".hexdigest_to_bytes == [0x0Bu8, 0x1Fu8, 0x4Du8]
738 # assert "0B1F4D".hexdigest_to_bytes.hexdigest == "0B1F4D"
739 #
740 # Characters that are not hexadecimal digits are ignored.
741 #
742 # assert "z0B1 F4\nD".hexdigest_to_bytes.hexdigest == "0B1F4D"
743 # assert "\\x0b1 \\xf4d".hexdigest_to_bytes.hexdigest == "0B1F4D"
744 #
745 # When the number of hexadecimal digit is not even, then a leading 0 is
746 # implicitly considered to fill the left byte (the most significant one).
747 #
748 # assert "1".hexdigest_to_bytes.hexdigest == "01"
749 # assert "FFF".hexdigest_to_bytes.hexdigest == "0FFF"
750 #
751 # `Bytes::hexdigest` is a loosely reverse method since its
752 # results contain only pairs of uppercase hexadecimal digits.
753 #
754 # assert "ABCD".hexdigest_to_bytes.hexdigest == "ABCD"
755 # assert "a b c".hexdigest_to_bytes.hexdigest == "0ABC"
756 fun hexdigest_to_bytes: Bytes do
757 var b = bytes
758 var max = byte_length
759
760 var dlength = 0 # Number of hex digits
761 var pos = 0
762 while pos < max do
763 var c = b[pos]
764 if c.is_valid_hexdigit then dlength += 1
765 pos += 1
766 end
767
768 # Allocate the result buffer
769 var ret = new Bytes.with_capacity((dlength+1) / 2)
770
771 var i = (dlength+1) % 2 # current hex digit (1=high, 0=low)
772 var byte = 0u8 # current accumulated byte value
773
774 pos = 0
775 while pos < max do
776 var c = b[pos]
777 if c.is_valid_hexdigit then
778 byte = byte << 4 | c.hexdigit_to_byteval
779 i -= 1
780 if i < 0 then
781 # Last digit known: store and restart
782 ret.add byte
783 i = 1
784 byte = 0u8
785 end
786 end
787 pos += 1
788 end
789 return ret
790 end
791
792 # Gets the hexdigest of the bytes of `self`
793 #
794 # assert "&lt;STRING&#47;&rt;".hexdigest == "266C743B535452494E47262334373B2672743B"
795 fun hexdigest: String do
796 var ln = byte_length
797 var outns = new CString(ln * 2)
798 var oi = 0
799 for i in [0 .. ln[ do
800 bytes[i].add_digest_at(outns, oi)
801 oi += 2
802 end
803 return new FlatString.with_infos(outns, ln * 2, 0)
804 end
805
806 # Return a `Bytes` instance where Nit escape sequences are transformed.
807 #
808 # assert "B\\n\\x41\\u0103D3".unescape_to_bytes.hexdigest == "420A41F0908F93"
809 #
810 # `Bytes::chexdigest` is a loosely reverse methods since its result is only made
811 # of `"\x??"` escape sequences.
812 #
813 # assert "\\x41\\x42\\x43".unescape_to_bytes.chexdigest == "\\x41\\x42\\x43"
814 # assert "B\\n\\x41\\u0103D3".unescape_to_bytes.chexdigest == "\\x42\\x0A\\x41\\xF0\\x90\\x8F\\x93"
815 fun unescape_to_bytes: Bytes do
816 var res = new Bytes.with_capacity(self.byte_length)
817 var was_slash = false
818 var i = 0
819 while i < length do
820 var c = self[i]
821 if not was_slash then
822 if c == '\\' then
823 was_slash = true
824 else
825 res.add_char(c)
826 end
827 i += 1
828 continue
829 end
830 was_slash = false
831 if c == 'n' then
832 res.add_char('\n')
833 else if c == 'r' then
834 res.add_char('\r')
835 else if c == 't' then
836 res.add_char('\t')
837 else if c == '0' then
838 res.add_char('\0')
839 else if c == 'x' or c == 'X' then
840 var hx = substring(i + 1, 2)
841 if hx.is_hex then
842 res.add(hx.to_hex.to_b)
843 else
844 res.add_char(c)
845 end
846 i += 2
847 else if c == 'u' or c == 'U' then
848 var hx = substring(i + 1, 6)
849 if hx.is_hex then
850 res.add_char(hx.to_hex.code_point)
851 else
852 res.add_char(c)
853 end
854 i += 6
855 else
856 res.add_char(c)
857 end
858 i += 1
859 end
860 return res
861 end
862
863 # Return a `Bytes` by reading 0 and 1.
864 #
865 # assert "1010101100001101".binarydigest_to_bytes.hexdigest == "AB0D"
866 #
867 # Note that characters that are neither 0 or 1 are just ignored.
868 #
869 # assert "a1B01 010\n1100あ001101".binarydigest_to_bytes.hexdigest == "AB0D"
870 # assert "hello".binarydigest_to_bytes.is_empty
871 #
872 # When the number of bits is not divisible by 8, then leading 0 are
873 # implicitly considered to fill the left byte (the most significant one).
874 #
875 # assert "1".binarydigest_to_bytes.hexdigest == "01"
876 # assert "1111111".binarydigest_to_bytes.hexdigest == "7F"
877 # assert "1000110100".binarydigest_to_bytes.hexdigest == "0234"
878 #
879 # `Bytes::binarydigest` is a loosely reverse method since its
880 # results contain only 1 and 0 by blocks of 8.
881 #
882 # assert "1010101100001101".binarydigest_to_bytes.binarydigest == "1010101100001101"
883 # assert "1".binarydigest_to_bytes.binarydigest == "00000001"
884 fun binarydigest_to_bytes: Bytes
885 do
886 var b = bytes
887 var max = byte_length
888
889 # Count bits
890 var bitlen = 0
891 var pos = 0
892 while pos < max do
893 var c = b[pos]
894 pos += 1
895 if c == 0x30u8 or c == 0x31u8 then bitlen += 1 # b'0' or b'1'
896 end
897
898 # Allocate (and take care of the padding)
899 var ret = new Bytes.with_capacity((bitlen+7) / 8)
900
901 var i = (bitlen+7) % 8 # current bit (7th=128, 0th=1)
902 var byte = 0u8 # current accumulated byte value
903
904 pos = 0
905 while pos < max do
906 var c = b[pos]
907 pos += 1
908 if c == 0x30u8 then # b'0'
909 byte = byte << 1
910 else if c == 0x31u8 then # b'1'
911 byte = byte << 1 | 1u8
912 else
913 continue
914 end
915
916 i -= 1
917 if i < 0 then
918 # Last bit known: store and restart
919 ret.add byte
920 i = 7
921 byte = 0u8
922 end
923 end
924 return ret
925 end
926 end
927
928 redef class FlatText
929 redef fun append_to_bytes(b) do
930 var from = if self isa FlatString then first_byte else 0
931 if isset _items then b.append_ns_from(items, byte_length, from)
932 end
933 end
934
935 redef class CString
936 # Creates a new `Bytes` object from `self` with `len` as length
937 #
938 # If `len` is null, strlen will determine the length of the Bytes
939 fun to_bytes(len: nullable Int): Bytes do
940 if len == null then len = cstring_length
941 return new Bytes(self, len, len)
942 end
943
944 # Creates a new `Bytes` object from a copy of `self` with `len` as length
945 #
946 # If `len` is null, strlen will determine the length of the Bytes
947 fun to_bytes_with_copy(len: nullable Int): Bytes do
948 if len == null then len = cstring_length
949 var nns = new CString(len)
950 copy_to(nns, len, 0, 0)
951 return new Bytes(nns, len, len)
952 end
953 end
954
955 # Joins an array of bytes `arr` separated by `sep`
956 #
957 # assert join_bytes(["String".to_bytes, "is".to_bytes, "string".to_bytes], ' '.ascii).hexdigest == "537472696E6720697320737472696E67"
958 fun join_bytes(arr: Array[Bytes], sep: nullable BytePattern): Bytes do
959 if arr.is_empty then return new Bytes.empty
960 sep = sep or else new Bytes.empty
961 var endln = sep.pattern_length * (arr.length - 1)
962 for i in arr do endln += i.length
963 var ret = new Bytes.with_capacity(endln)
964 ret.append(arr.first)
965 for i in [1 .. arr.length[ do
966 sep.append_to(ret)
967 ret.append arr[i]
968 end
969 return ret
970 end