mirrors/zits
1
0
Fork 0
mirror of https://git.robbyzambito.me/zits synced 2026-02-04 19:54:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Subject validation

Browse Source
This commit is contained in:
Robby Zambito
2026-01-02 23:51:49 +00:00
parent a21dbfe3bb
commit 5a7d3caf9c
1 changed files with 43 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

59
src/server/message_parser.zig
View File

@@ -294,7 +294,7 @@ pub const Message = union(enum) {
@branchHint(.unlikely);
return error.InvalidStream;
}
const subject = try readSubject(alloc, in);
const subject = try readSubject(alloc, in, .sub);
errdefer alloc.free(subject);
const second = blk: {
// Drop whitespace
@@ -392,7 +392,7 @@ fn parsePub(alloc: std.mem.Allocator, in: *std.Io.Reader) !Message {
try in.discardAll(1); // throw away space
// Parse subject
const subject: []const u8 = try readSubject(alloc, in);
const subject: []const u8 = try readSubject(alloc, in, .@"pub");
errdefer alloc.free(subject);
const States = enum {
@@ -526,7 +526,7 @@ fn parseHPub(alloc: std.mem.Allocator, in: *std.Io.Reader) !Message {
try in.discardAll(1); // throw away space
// Parse subject
const subject: []const u8 = try readSubject(alloc, in);
const subject: []const u8 = try readSubject(alloc, in, .@"pub");
errdefer alloc.free(subject);
const States = enum {
@@ -699,30 +699,57 @@ test parseHPub {
}
}
fn readSubject(alloc: std.mem.Allocator, in: *std.Io.Reader) ![]const u8 {
fn readSubject(alloc: std.mem.Allocator, in: *std.Io.Reader, comptime pub_or_sub: enum { @"pub", sub }) ![]const u8 {
var subject_list: std.ArrayList(u8) = .empty;
errdefer subject_list.deinit(alloc);
// Handle the first character
{
const byte = try in.takeByte();
if (std.ascii.isWhitespace(byte) or byte == '.')
if (std.ascii.isWhitespace(byte) or byte == '.' or (pub_or_sub == .@"pub" and (byte == '*' or byte == '>')))
return error.InvalidStream;
try subject_list.append(alloc, byte);
}
while (in.takeByte()) |byte| {
if (std.ascii.isWhitespace(byte)) break;
if (std.ascii.isAscii(byte)) {
if (byte == '.') {
const next_byte = try in.peekByte();
if (next_byte == '.' or std.ascii.isWhitespace(next_byte))
return error.InvalidSubject;
}
try subject_list.append(alloc, byte);
}
} else |err| return err;
switch (pub_or_sub) {
.sub => {
while (in.takeByte()) |byte| {
if (std.ascii.isWhitespace(byte)) break;
if (std.ascii.isAscii(byte)) {
if (byte == '.') {
const next_byte = try in.peekByte();
if (next_byte == '.' or std.ascii.isWhitespace(next_byte))
return error.InvalidStream;
} else if (byte == '>') {
const next_byte = try in.takeByte();
if (!std.ascii.isWhitespace(next_byte))
return error.InvalidStream;
} else if (byte == '*') {
const next_byte = try in.peekByte();
if (next_byte != '.' and !std.ascii.isWhitespace(next_byte))
return error.InvalidStream;
}
try subject_list.append(alloc, byte);
}
} else |err| return err;
},
.@"pub" => {
while (in.takeByte()) |byte| {
if (std.ascii.isWhitespace(byte)) break;
if (std.ascii.isAscii(byte)) {
if (byte == '*' or byte == '>') return error.InvalidStream;
if (byte == '.') {
const next_byte = try in.peekByte();
if (next_byte == '.' or std.ascii.isWhitespace(next_byte))
return error.InvalidStream;
}
try subject_list.append(alloc, byte);
}
} else |err| return err;
},
}
return subject_list.toOwnedSlice(alloc);
}