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
|
bytes: []const u8,
index: usize,
is_trailer: bool,
pub fn init(bytes: []const u8) HeaderIterator {
return .{
.bytes = bytes,
.index = std.mem.indexOfPosLinear(u8, bytes, 0, "\r\n").? + 2,
.is_trailer = false,
};
}
pub fn next(it: *HeaderIterator) ?std.http.Header {
const end = std.mem.indexOfPosLinear(u8, it.bytes, it.index, "\r\n").?;
if (it.index == end) { // found the trailer boundary (\r\n\r\n)
if (it.is_trailer) return null;
const next_end = std.mem.indexOfPosLinear(u8, it.bytes, end + 2, "\r\n") orelse
return null;
var kv_it = std.mem.splitScalar(u8, it.bytes[end + 2 .. next_end], ':');
const name = kv_it.first();
const value = kv_it.rest();
it.is_trailer = true;
it.index = next_end + 2;
if (name.len == 0)
return null;
return .{
.name = name,
.value = std.mem.trim(u8, value, " \t"),
};
} else { // normal header
var kv_it = std.mem.splitScalar(u8, it.bytes[it.index..end], ':');
const name = kv_it.first();
const value = kv_it.rest();
it.index = end + 2;
if (name.len == 0)
return null;
return .{
.name = name,
.value = std.mem.trim(u8, value, " \t"),
};
}
}
test next {
var it = HeaderIterator.init("200 OK\r\na: b\r\nc: \r\nd:e\r\n\r\nf: g\r\n\r\n");
try std.testing.expect(!it.is_trailer);
{
const header = it.next().?;
try std.testing.expect(!it.is_trailer);
try std.testing.expectEqualStrings("a", header.name);
try std.testing.expectEqualStrings("b", header.value);
}
{
const header = it.next().?;
try std.testing.expect(!it.is_trailer);
try std.testing.expectEqualStrings("c", header.name);
try std.testing.expectEqualStrings("", header.value);
}
{
const header = it.next().?;
try std.testing.expect(!it.is_trailer);
try std.testing.expectEqualStrings("d", header.name);
try std.testing.expectEqualStrings("e", header.value);
}
{
const header = it.next().?;
try std.testing.expect(it.is_trailer);
try std.testing.expectEqualStrings("f", header.name);
try std.testing.expectEqualStrings("g", header.value);
}
try std.testing.expectEqual(null, it.next());
it = HeaderIterator.init("200 OK\r\n: ss\r\n\r\n");
try std.testing.expect(!it.is_trailer);
try std.testing.expectEqual(null, it.next());
it = HeaderIterator.init("200 OK\r\na:b\r\n\r\n: ss\r\n\r\n");
try std.testing.expect(!it.is_trailer);
{
const header = it.next().?;
try std.testing.expect(!it.is_trailer);
try std.testing.expectEqualStrings("a", header.name);
try std.testing.expectEqualStrings("b", header.value);
}
try std.testing.expectEqual(null, it.next());
try std.testing.expect(it.is_trailer);
}
const HeaderIterator = @This();
const std = @import("../std.zig");
const assert = std.debug.assert;
|
