|
const Uri = @This(); const std = @import("std.zig"); const testing = std.testing; scheme: []const u8, user: ?[]const u8, password: ?[]const u8, host: ?[]const u8, port: ?u16, path: []const u8, query: ?[]const u8, fragment: ?[]const u8, |
escapeString()Applies URI encoding and replaces all reserved characters with their respective %XX code. |
pub fn escapeString(allocator: std.mem.Allocator, input: []const u8) error{OutOfMemory}![]u8 { return escapeStringWithFn(allocator, input, isUnreserved); } |
escapePath() |
pub fn escapePath(allocator: std.mem.Allocator, input: []const u8) error{OutOfMemory}![]u8 { return escapeStringWithFn(allocator, input, isPathChar); } |
escapeQuery() |
pub fn escapeQuery(allocator: std.mem.Allocator, input: []const u8) error{OutOfMemory}![]u8 { return escapeStringWithFn(allocator, input, isQueryChar); } |
writeEscapedString() |
pub fn writeEscapedString(writer: anytype, input: []const u8) !void { return writeEscapedStringWithFn(writer, input, isUnreserved); } |
writeEscapedPath() |
pub fn writeEscapedPath(writer: anytype, input: []const u8) !void { return writeEscapedStringWithFn(writer, input, isPathChar); } |
writeEscapedQuery() |
pub fn writeEscapedQuery(writer: anytype, input: []const u8) !void { return writeEscapedStringWithFn(writer, input, isQueryChar); } |
escapeStringWithFn() |
pub fn escapeStringWithFn(allocator: std.mem.Allocator, input: []const u8, comptime keepUnescaped: fn (c: u8) bool) std.mem.Allocator.Error![]u8 { var outsize: usize = 0; for (input) |c| { outsize += if (keepUnescaped(c)) @as(usize, 1) else 3; } var output = try allocator.alloc(u8, outsize); var outptr: usize = 0; for (input) |c| { if (keepUnescaped(c)) { output[outptr] = c; outptr += 1; } else { var buf: [2]u8 = undefined; _ = std.fmt.bufPrint(&buf, "{X:0>2}", .{c}) catch unreachable; output[outptr + 0] = '%'; output[outptr + 1] = buf[0]; output[outptr + 2] = buf[1]; outptr += 3; } } return output; } |
writeEscapedStringWithFn() |
pub fn writeEscapedStringWithFn(writer: anytype, input: []const u8, comptime keepUnescaped: fn (c: u8) bool) @TypeOf(writer).Error!void { for (input) |c| { if (keepUnescaped(c)) { try writer.writeByte(c); } else { try writer.print("%{X:0>2}", .{c}); } } } |
unescapeString()Parses a URI string and unescapes all %XX where XX is a valid hex number. Otherwise, verbatim copies them to the output. |
pub fn unescapeString(allocator: std.mem.Allocator, input: []const u8) error{OutOfMemory}![]u8 { var outsize: usize = 0; var inptr: usize = 0; while (inptr < input.len) { if (input[inptr] == '%') { inptr += 1; if (inptr + 2 <= input.len) { _ = std.fmt.parseInt(u8, input[inptr..][0..2], 16) catch { outsize += 3; inptr += 2; continue; }; inptr += 2; outsize += 1; } } else { inptr += 1; outsize += 1; } } var output = try allocator.alloc(u8, outsize); var outptr: usize = 0; inptr = 0; while (inptr < input.len) { if (input[inptr] == '%') { inptr += 1; if (inptr + 2 <= input.len) { const value = std.fmt.parseInt(u8, input[inptr..][0..2], 16) catch { output[outptr + 0] = input[inptr + 0]; output[outptr + 1] = input[inptr + 1]; inptr += 2; outptr += 2; continue; }; output[outptr] = value; inptr += 2; outptr += 1; } } else { output[outptr] = input[inptr]; inptr += 1; outptr += 1; } } return output; } |
ParseError |
pub const ParseError = error{ UnexpectedCharacter, InvalidFormat, InvalidPort }; |
parseWithoutScheme() Parses the URI or returns an error. This function is not compliant, but is required to parse some forms of URIs in the wild, such as HTTP Location headers. The return value will contain unescaped strings pointing into the original |
pub fn parseWithoutScheme(text: []const u8) ParseError!Uri { var reader = SliceReader{ .slice = text }; var uri = Uri{ .scheme = "", .user = null, .password = null, .host = null, .port = null, .path = "", // path is always set, but empty by default. .query = null, .fragment = null, }; if (reader.peekPrefix("//")) a: { // authority part std.debug.assert(reader.get().? == '/'); std.debug.assert(reader.get().? == '/'); const authority = reader.readUntil(isAuthoritySeparator); if (authority.len == 0) { if (reader.peekPrefix("/")) break :a else return error.InvalidFormat; } var start_of_host: usize = 0; if (std.mem.indexOf(u8, authority, "@")) |index| { start_of_host = index + 1; const user_info = authority[0..index]; if (std.mem.indexOf(u8, user_info, ":")) |idx| { uri.user = user_info[0..idx]; if (idx < user_info.len - 1) { // empty password is also "no password" uri.password = user_info[idx + 1 ..]; } } else { uri.user = user_info; uri.password = null; } } // only possible if uri consists of only `userinfo@` if (start_of_host >= authority.len) break :a; var end_of_host: usize = authority.len; if (authority.len > start_of_host and authority[start_of_host] == '[') { // IPv6 end_of_host = std.mem.lastIndexOf(u8, authority, "]") orelse return error.InvalidFormat; end_of_host += 1; if (std.mem.lastIndexOf(u8, authority, ":")) |index| { if (index >= end_of_host) { // if not part of the V6 address field end_of_host = @min(end_of_host, index); uri.port = std.fmt.parseInt(u16, authority[index + 1 ..], 10) catch return error.InvalidPort; } } } else if (std.mem.lastIndexOf(u8, authority, ":")) |index| { if (index >= start_of_host) { // if not part of the userinfo field end_of_host = @min(end_of_host, index); uri.port = std.fmt.parseInt(u16, authority[index + 1 ..], 10) catch return error.InvalidPort; } } uri.host = authority[start_of_host..end_of_host]; } uri.path = reader.readUntil(isPathSeparator); if ((reader.peek() orelse 0) == '?') { // query part std.debug.assert(reader.get().? == '?'); uri.query = reader.readUntil(isQuerySeparator); } if ((reader.peek() orelse 0) == '#') { // fragment part std.debug.assert(reader.get().? == '#'); uri.fragment = reader.readUntilEof(); } return uri; } |
WriteToStreamOptions |
pub const WriteToStreamOptions = struct { scheme: bool = false, authentication: bool = false, authority: bool = false, path: bool = false, query: bool = false, fragment: bool = false, raw: bool = false, }; |
writeToStream() When true, include the scheme part of the URI. When true, include the user and password part of the URI. Ignored if |
pub fn writeToStream( uri: Uri, options: WriteToStreamOptions, writer: anytype, ) @TypeOf(writer).Error!void { if (options.scheme) { try writer.writeAll(uri.scheme); try writer.writeAll(":"); if (options.authority and uri.host != null) { try writer.writeAll("//"); } } if (options.authority) { if (options.authentication and uri.host != null) { if (uri.user) |user| { try writer.writeAll(user); if (uri.password) |password| { try writer.writeAll(":"); try writer.writeAll(password); } try writer.writeAll("@"); } } if (uri.host) |host| { try writer.writeAll(host); if (uri.port) |port| { try writer.writeAll(":"); try std.fmt.formatInt(port, 10, .lower, .{}, writer); } } } if (options.path) { if (uri.path.len == 0) { try writer.writeAll("/"); } else if (options.raw) { try writer.writeAll(uri.path); } else { try writeEscapedPath(writer, uri.path); } if (options.query) if (uri.query) |q| { try writer.writeAll("?"); if (options.raw) { try writer.writeAll(q); } else { try writeEscapedQuery(writer, q); } }; if (options.fragment) if (uri.fragment) |f| { try writer.writeAll("#"); if (options.raw) { try writer.writeAll(f); } else { try writeEscapedQuery(writer, f); } }; } } |
format() |
pub fn format( uri: Uri, comptime fmt: []const u8, options: std.fmt.FormatOptions, writer: anytype, ) @TypeOf(writer).Error!void { _ = options; const scheme = comptime std.mem.indexOf(u8, fmt, ":") != null or fmt.len == 0; const authentication = comptime std.mem.indexOf(u8, fmt, "@") != null or fmt.len == 0; const authority = comptime std.mem.indexOf(u8, fmt, "+") != null or fmt.len == 0; const path = comptime std.mem.indexOf(u8, fmt, "/") != null or fmt.len == 0; const query = comptime std.mem.indexOf(u8, fmt, "?") != null or fmt.len == 0; const fragment = comptime std.mem.indexOf(u8, fmt, "#") != null or fmt.len == 0; const raw = comptime std.mem.indexOf(u8, fmt, "r") != null or fmt.len == 0; return writeToStream(uri, .{ .scheme = scheme, .authentication = authentication, .authority = authority, .path = path, .query = query, .fragment = fragment, .raw = raw, }, writer); } |
parse() Parses the URI or returns an error. The return value will contain unescaped strings pointing into the original |
pub fn parse(text: []const u8) ParseError!Uri { var reader = SliceReader{ .slice = text }; const scheme = reader.readWhile(isSchemeChar); // after the scheme, a ':' must appear if (reader.get()) |c| { if (c != ':') return error.UnexpectedCharacter; } else { return error.InvalidFormat; } var uri = try parseWithoutScheme(reader.readUntilEof()); uri.scheme = scheme; return uri; } |
resolve()Resolves a URI against a base URI, conforming to RFC 3986, Section 5. arena owns any memory allocated by this function. |
pub fn resolve(Base: Uri, R: Uri, strict: bool, arena: std.mem.Allocator) !Uri { var T: Uri = undefined; if (R.scheme.len > 0 and !((!strict) and (std.mem.eql(u8, R.scheme, Base.scheme)))) { T.scheme = R.scheme; T.user = R.user; T.host = R.host; T.port = R.port; T.path = try std.fs.path.resolvePosix(arena, &.{ "/", R.path }); T.query = R.query; } else { if (R.host) |host| { T.user = R.user; T.host = host; T.port = R.port; T.path = R.path; T.path = try std.fs.path.resolvePosix(arena, &.{ "/", R.path }); T.query = R.query; } else { if (R.path.len == 0) { T.path = Base.path; if (R.query) |query| { T.query = query; } else { T.query = Base.query; } } else { if (R.path[0] == '/') { T.path = try std.fs.path.resolvePosix(arena, &.{ "/", R.path }); } else { T.path = try std.fs.path.resolvePosix(arena, &.{ "/", Base.path, R.path }); } T.query = R.query; } T.user = Base.user; T.host = Base.host; T.port = Base.port; } T.scheme = Base.scheme; } T.fragment = R.fragment; return T; } const SliceReader = struct { const Self = @This(); slice: []const u8, offset: usize = 0, fn get(self: *Self) ?u8 { if (self.offset >= self.slice.len) return null; const c = self.slice[self.offset]; self.offset += 1; return c; } fn peek(self: Self) ?u8 { if (self.offset >= self.slice.len) return null; return self.slice[self.offset]; } fn readWhile(self: *Self, comptime predicate: fn (u8) bool) []const u8 { const start = self.offset; var end = start; while (end < self.slice.len and predicate(self.slice[end])) { end += 1; } self.offset = end; return self.slice[start..end]; } fn readUntil(self: *Self, comptime predicate: fn (u8) bool) []const u8 { const start = self.offset; var end = start; while (end < self.slice.len and !predicate(self.slice[end])) { end += 1; } self.offset = end; return self.slice[start..end]; } fn readUntilEof(self: *Self) []const u8 { const start = self.offset; self.offset = self.slice.len; return self.slice[start..]; } fn peekPrefix(self: Self, prefix: []const u8) bool { if (self.offset + prefix.len > self.slice.len) return false; return std.mem.eql(u8, self.slice[self.offset..][0..prefix.len], prefix); } }; fn isSchemeChar(c: u8) bool { return switch (c) { 'A'...'Z', 'a'...'z', '0'...'9', '+', '-', '.' => true, else => false, }; } fn isAuthoritySeparator(c: u8) bool { return switch (c) { '/', '?', '#' => true, else => false, }; } fn isReserved(c: u8) bool { return isGenLimit(c) or isSubLimit(c); } fn isGenLimit(c: u8) bool { return switch (c) { ':', ',', '?', '#', '[', ']', '@' => true, else => false, }; } fn isSubLimit(c: u8) bool { return switch (c) { '!', '$', '&', '\'', '(', ')', '*', '+', ',', ';', '=' => true, else => false, }; } fn isUnreserved(c: u8) bool { return switch (c) { 'A'...'Z', 'a'...'z', '0'...'9', '-', '.', '_', '~' => true, else => false, }; } fn isPathSeparator(c: u8) bool { return switch (c) { '?', '#' => true, else => false, }; } fn isPathChar(c: u8) bool { return isUnreserved(c) or isSubLimit(c) or c == '/' or c == ':' or c == '@'; } fn isQueryChar(c: u8) bool { return isPathChar(c) or c == '?' or c == '%'; } fn isQuerySeparator(c: u8) bool { return switch (c) { '#' => true, else => false, }; } |
Test:basicscheme = ALPHA *( ALPHA / DIGIT / "+" / "-" / "." ) reserved = gen-delims / sub-delims gen-delims = ":" / "/" / "?" / "#" / "[" / "]" / "@" sub-delims = "!" / "$" / "&" / "'" / "(" / ")" / "*" / "+" / "," / ";" / "=" unreserved = ALPHA / DIGIT / "-" / "." / "_" / "~" |
test "basic" { const parsed = try parse("https://ziglang.org/download"); try testing.expectEqualStrings("https", parsed.scheme); try testing.expectEqualStrings("ziglang.org", parsed.host orelse return error.UnexpectedNull); try testing.expectEqualStrings("/download", parsed.path); try testing.expectEqual(@as(?u16, null), parsed.port); } |
Test:with port |
test "with port" { const parsed = try parse("http://example:1337/"); try testing.expectEqualStrings("http", parsed.scheme); try testing.expectEqualStrings("example", parsed.host orelse return error.UnexpectedNull); try testing.expectEqualStrings("/", parsed.path); try testing.expectEqual(@as(?u16, 1337), parsed.port); } |
Test:should fail gracefully |
test "should fail gracefully" { try std.testing.expectEqual(@as(ParseError!Uri, error.InvalidFormat), parse("foobar://")); } |
Test:file |
test "file" { const parsed = try parse("file:///"); try std.testing.expectEqualSlices(u8, "file", parsed.scheme); try std.testing.expectEqual(@as(?[]const u8, null), parsed.host); try std.testing.expectEqualSlices(u8, "/", parsed.path); const parsed2 = try parse("file:///an/absolute/path/to/something"); try std.testing.expectEqualSlices(u8, "file", parsed2.scheme); try std.testing.expectEqual(@as(?[]const u8, null), parsed2.host); try std.testing.expectEqualSlices(u8, "/an/absolute/path/to/something", parsed2.path); const parsed3 = try parse("file://localhost/an/absolute/path/to/another/thing/"); try std.testing.expectEqualSlices(u8, "file", parsed3.scheme); try std.testing.expectEqualSlices(u8, "localhost", parsed3.host.?); try std.testing.expectEqualSlices(u8, "/an/absolute/path/to/another/thing/", parsed3.path); } |
Test:scheme |
test "scheme" { try std.testing.expectEqualSlices(u8, "http", (try parse("http:_")).scheme); try std.testing.expectEqualSlices(u8, "scheme-mee", (try parse("scheme-mee:_")).scheme); try std.testing.expectEqualSlices(u8, "a.b.c", (try parse("a.b.c:_")).scheme); try std.testing.expectEqualSlices(u8, "ab+", (try parse("ab+:_")).scheme); try std.testing.expectEqualSlices(u8, "X+++", (try parse("X+++:_")).scheme); try std.testing.expectEqualSlices(u8, "Y+-.", (try parse("Y+-.:_")).scheme); } |
Test:authority |
test "authority" { try std.testing.expectEqualSlices(u8, "hostname", (try parse("scheme://hostname")).host.?); try std.testing.expectEqualSlices(u8, "hostname", (try parse("scheme://userinfo@hostname")).host.?); try std.testing.expectEqualSlices(u8, "userinfo", (try parse("scheme://userinfo@hostname")).user.?); try std.testing.expectEqual(@as(?[]const u8, null), (try parse("scheme://userinfo@hostname")).password); try std.testing.expectEqual(@as(?[]const u8, null), (try parse("scheme://userinfo@")).host); try std.testing.expectEqualSlices(u8, "hostname", (try parse("scheme://user:password@hostname")).host.?); try std.testing.expectEqualSlices(u8, "user", (try parse("scheme://user:password@hostname")).user.?); try std.testing.expectEqualSlices(u8, "password", (try parse("scheme://user:password@hostname")).password.?); try std.testing.expectEqualSlices(u8, "hostname", (try parse("scheme://hostname:0")).host.?); try std.testing.expectEqual(@as(u16, 1234), (try parse("scheme://hostname:1234")).port.?); try std.testing.expectEqualSlices(u8, "hostname", (try parse("scheme://userinfo@hostname:1234")).host.?); try std.testing.expectEqual(@as(u16, 1234), (try parse("scheme://userinfo@hostname:1234")).port.?); try std.testing.expectEqualSlices(u8, "userinfo", (try parse("scheme://userinfo@hostname:1234")).user.?); try std.testing.expectEqual(@as(?[]const u8, null), (try parse("scheme://userinfo@hostname:1234")).password); try std.testing.expectEqualSlices(u8, "hostname", (try parse("scheme://user:password@hostname:1234")).host.?); try std.testing.expectEqual(@as(u16, 1234), (try parse("scheme://user:password@hostname:1234")).port.?); try std.testing.expectEqualSlices(u8, "user", (try parse("scheme://user:password@hostname:1234")).user.?); try std.testing.expectEqualSlices(u8, "password", (try parse("scheme://user:password@hostname:1234")).password.?); } |
Test:authority.password |
test "authority.password" { try std.testing.expectEqualSlices(u8, "username", (try parse("scheme://username@a")).user.?); try std.testing.expectEqual(@as(?[]const u8, null), (try parse("scheme://username@a")).password); try std.testing.expectEqualSlices(u8, "username", (try parse("scheme://username:@a")).user.?); try std.testing.expectEqual(@as(?[]const u8, null), (try parse("scheme://username:@a")).password); try std.testing.expectEqualSlices(u8, "username", (try parse("scheme://username:password@a")).user.?); try std.testing.expectEqualSlices(u8, "password", (try parse("scheme://username:password@a")).password.?); try std.testing.expectEqualSlices(u8, "username", (try parse("scheme://username::@a")).user.?); try std.testing.expectEqualSlices(u8, ":", (try parse("scheme://username::@a")).password.?); } fn testAuthorityHost(comptime hostlist: anytype) !void { inline for (hostlist) |hostname| { try std.testing.expectEqualSlices(u8, hostname, (try parse("scheme://" ++ hostname)).host.?); } } |
Test:authority.dns-names |
test "authority.dns-names" { try testAuthorityHost(.{ "a", "a.b", "example.com", "www.example.com", "example.org.", "www.example.org.", "xn--nw2a.xn--j6w193g", // internationalized URI: 見.香港 "fe80--1ff-fe23-4567-890as3.ipv6-literal.net", }); } |
Test:authority.IPv4 |
test "authority.IPv4" { try testAuthorityHost(.{ "127.0.0.1", "255.255.255.255", "0.0.0.0", "8.8.8.8", "1.2.3.4", "192.168.0.1", "10.42.0.0", }); } |
Test:authority.IPv6 |
test "authority.IPv6" { try testAuthorityHost(.{ "[2001:db8:0:0:0:0:2:1]", "[2001:db8::2:1]", "[2001:db8:0000:1:1:1:1:1]", "[2001:db8:0:1:1:1:1:1]", "[0:0:0:0:0:0:0:0]", "[0:0:0:0:0:0:0:1]", "[::1]", "[::]", "[2001:db8:85a3:8d3:1319:8a2e:370:7348]", "[fe80::1ff:fe23:4567:890a%25eth2]", "[fe80::1ff:fe23:4567:890a]", "[fe80::1ff:fe23:4567:890a%253]", "[fe80:3::1ff:fe23:4567:890a]", }); } |
Test:RFC example 1 |
test "RFC example 1" { const uri = "foo://example.com:8042/over/there?name=ferret#nose"; try std.testing.expectEqual(Uri{ .scheme = uri[0..3], .user = null, .password = null, .host = uri[6..17], .port = 8042, .path = uri[22..33], .query = uri[34..45], .fragment = uri[46..50], }, try parse(uri)); } |
Test:RFC example 2 |
test "RFC example 2" { const uri = "urn:example:animal:ferret:nose"; try std.testing.expectEqual(Uri{ .scheme = uri[0..3], .user = null, .password = null, .host = null, .port = null, .path = uri[4..], .query = null, .fragment = null, }, try parse(uri)); } // source: // https://en.wikipedia.org/wiki/Uniform_Resource_Identifier#Examples |
Test:Examples from wikipedia |
test "Examples from wikipedia" { const list = [_][]const u8{ "https://john.doe@www.example.com:123/forum/questions/?tag=networking&order=newest#top", "ldap://[2001:db8::7]/c=GB?objectClass?one", "mailto:John.Doe@example.com", "news:comp.infosystems.www.servers.unix", "tel:+1-816-555-1212", "telnet://192.0.2.16:80/", "urn:oasis:names:specification:docbook:dtd:xml:4.1.2", "http://a/b/c/d;p?q", }; for (list) |uri| { _ = try parse(uri); } } // source: // https://tools.ietf.org/html/rfc3986#section-5.4.1 |
Test:Examples from RFC3986 |
test "Examples from RFC3986" { const list = [_][]const u8{ "http://a/b/c/g", "http://a/b/c/g", "http://a/b/c/g/", "http://a/g", "http://g", "http://a/b/c/d;p?y", "http://a/b/c/g?y", "http://a/b/c/d;p?q#s", "http://a/b/c/g#s", "http://a/b/c/g?y#s", "http://a/b/c/;x", "http://a/b/c/g;x", "http://a/b/c/g;x?y#s", "http://a/b/c/d;p?q", "http://a/b/c/", "http://a/b/c/", "http://a/b/", "http://a/b/", "http://a/b/g", "http://a/", "http://a/", "http://a/g", }; for (list) |uri| { _ = try parse(uri); } } |
Test:Special test |
test "Special test" { // This is for all of you code readers ♥ _ = try parse("https://www.youtube.com/watch?v=dQw4w9WgXcQ&feature=youtu.be&t=0"); } |
Test:URI escaping |
test "URI escaping" { const input = "\\ö/ äöß ~~.adas-https://canvas:123/#ads&&sad"; const expected = "%5C%C3%B6%2F%20%C3%A4%C3%B6%C3%9F%20~~.adas-https%3A%2F%2Fcanvas%3A123%2F%23ads%26%26sad"; const actual = try escapeString(std.testing.allocator, input); defer std.testing.allocator.free(actual); try std.testing.expectEqualSlices(u8, expected, actual); } |
Test:URI unescaping |
test "URI unescaping" { const input = "%5C%C3%B6%2F%20%C3%A4%C3%B6%C3%9F%20~~.adas-https%3A%2F%2Fcanvas%3A123%2F%23ads%26%26sad"; const expected = "\\ö/ äöß ~~.adas-https://canvas:123/#ads&&sad"; const actual = try unescapeString(std.testing.allocator, input); defer std.testing.allocator.free(actual); try std.testing.expectEqualSlices(u8, expected, actual); } |
Test:URI query escaping |
test "URI query escaping" { const address = "https://objects.githubusercontent.com/?response-content-type=application%2Foctet-stream"; const parsed = try Uri.parse(address); // format the URI to escape it const formatted_uri = try std.fmt.allocPrint(std.testing.allocator, "{/?}", .{parsed}); defer std.testing.allocator.free(formatted_uri); try std.testing.expectEqualStrings("/?response-content-type=application%2Foctet-stream", formatted_uri); } |
Test:format |
test "format" { const uri = Uri{ .scheme = "file", .user = null, .password = null, .host = null, .port = null, .path = "/foo/bar/baz", .query = null, .fragment = null, }; var buf = std.ArrayList(u8).init(std.testing.allocator); defer buf.deinit(); try uri.format(":/?#", .{}, buf.writer()); try std.testing.expectEqualSlices(u8, "file:/foo/bar/baz", buf.items); } |
Generated by zstd-browse2 on 2023-11-04 14:12:32 -0400. |