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
|
const bun = @import("root").bun;
const JSC = bun.JSC;
const Encoder = JSC.WebCore.Encoder;
pub const BufferVectorized = struct {
pub fn fill(
str: *JSC.ZigString,
buf_ptr: [*]u8,
fill_length: usize,
encoding: JSC.Node.Encoding,
) callconv(.C) bool {
if (str.len == 0) return true;
var buf = buf_ptr[0..fill_length];
const written = switch (encoding) {
.utf8 => if (str.is16Bit())
Encoder.writeU16(str.utf16SliceAligned().ptr, str.utf16SliceAligned().len, buf.ptr, buf.len, .utf8, true)
else
Encoder.writeU8(str.slice().ptr, str.slice().len, buf.ptr, buf.len, .utf8),
.ascii => if (str.is16Bit())
Encoder.writeU16(str.utf16SliceAligned().ptr, str.utf16SliceAligned().len, buf.ptr, buf.len, .ascii, true)
else
Encoder.writeU8(str.slice().ptr, str.slice().len, buf.ptr, buf.len, .ascii),
.latin1 => if (str.is16Bit())
Encoder.writeU16(str.utf16SliceAligned().ptr, str.utf16SliceAligned().len, buf.ptr, buf.len, .latin1, true)
else
Encoder.writeU8(str.slice().ptr, str.slice().len, buf.ptr, buf.len, .latin1),
.buffer => if (str.is16Bit())
Encoder.writeU16(str.utf16SliceAligned().ptr, str.utf16SliceAligned().len, buf.ptr, buf.len, .buffer, true)
else
Encoder.writeU8(str.slice().ptr, str.slice().len, buf.ptr, buf.len, .buffer),
.utf16le, .ucs2 => if (str.is16Bit())
Encoder.writeU16(str.utf16SliceAligned().ptr, str.utf16SliceAligned().len, buf.ptr, buf.len, .utf16le, true)
else
Encoder.writeU8(str.slice().ptr, str.slice().len, buf.ptr, buf.len, .utf16le),
.base64 => if (str.is16Bit())
Encoder.writeU16(str.utf16SliceAligned().ptr, str.utf16SliceAligned().len, buf.ptr, buf.len, .base64, true)
else
Encoder.writeU8(str.slice().ptr, str.slice().len, buf.ptr, buf.len, .base64),
.base64url => if (str.is16Bit())
Encoder.writeU16(str.utf16SliceAligned().ptr, str.utf16SliceAligned().len, buf.ptr, buf.len, .base64url, true)
else
Encoder.writeU8(str.slice().ptr, str.slice().len, buf.ptr, buf.len, .base64url),
.hex => if (str.is16Bit())
Encoder.writeU16(str.utf16SliceAligned().ptr, str.utf16SliceAligned().len, buf.ptr, buf.len, .hex, true)
else
Encoder.writeU8(str.slice().ptr, str.slice().len, buf.ptr, buf.len, .hex),
} catch return false;
switch (written) {
0 => {},
1 => @memset(buf, buf[0]),
else => {
var contents = buf[0..written];
buf = buf[written..];
while (buf.len >= contents.len) {
bun.copy(u8, buf, contents);
buf = buf[contents.len..];
contents.len *= 2;
}
if (buf.len > 0) {
bun.copy(u8, buf, contents[0..buf.len]);
}
},
}
return true;
}
};
comptime {
if (!JSC.is_bindgen) {
@export(BufferVectorized.fill, .{ .name = "Bun__Buffer_fill" });
}
}
|
