aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/bun.js/api/bun/dns_resolver.zig71
-rw-r--r--src/bun.js/bindings/bindings.zig30
-rw-r--r--test/bun.js/node-dns.test.js34
3 files changed, 75 insertions, 60 deletions
diff --git a/src/bun.js/api/bun/dns_resolver.zig b/src/bun.js/api/bun/dns_resolver.zig
index 08202a09e..ef1c6bc4a 100644
--- a/src/bun.js/api/bun/dns_resolver.zig
+++ b/src/bun.js/api/bun/dns_resolver.zig
@@ -629,11 +629,10 @@ pub fn ResolveInfoRequest(comptime cares_type: type, comptime type_name: []const
pub fn init(
cache: DNSResolver.LookupCacheHit(@This()),
resolver: ?*DNSResolver,
- name_str: *const JSC.ZigString.Slice,
+ name: []const u8,
globalThis: *JSC.JSGlobalObject,
comptime cache_field: []const u8,
) !*@This() {
- const name = name_str.slice();
var request = try globalThis.allocator().create(@This());
var hasher = std.hash.Wyhash.init(0);
hasher.update(name);
@@ -643,7 +642,7 @@ pub fn ResolveInfoRequest(comptime cares_type: type, comptime type_name: []const
request.* = .{
.resolver_for_caching = resolver,
.hash = hash,
- .head = .{ .poll_ref = poll_ref, .globalThis = globalThis, .promise = JSC.JSPromise.Strong.init(globalThis), .allocated = false, .name = name_str },
+ .head = .{ .poll_ref = poll_ref, .globalThis = globalThis, .promise = JSC.JSPromise.Strong.init(globalThis), .allocated = false, .name = name },
};
request.tail = &request.head;
if (cache == .new) {
@@ -943,9 +942,9 @@ pub fn CAresLookup(comptime cares_type: type, comptime type_name: []const u8) ty
poll_ref: JSC.PollRef,
allocated: bool = false,
next: ?*@This() = null,
- name: *const JSC.ZigString.Slice,
+ name: []const u8,
- pub fn init(globalThis: *JSC.JSGlobalObject, allocator: std.mem.Allocator, name: *const JSC.ZigString.Slice) !*@This() {
+ pub fn init(globalThis: *JSC.JSGlobalObject, allocator: std.mem.Allocator, name: []const u8) !*@This() {
var this = try allocator.create(@This());
var poll_ref = JSC.PollRef.init();
poll_ref.ref(globalThis.bunVM());
@@ -998,7 +997,7 @@ pub fn CAresLookup(comptime cares_type: type, comptime type_name: []const u8) ty
pub fn deinit(this: *@This()) void {
this.poll_ref.unrefOnNextTick(this.globalThis.bunVM());
- this.name.deinit();
+ bun.default_allocator.free(this.name);
if (this.allocated)
this.globalThis.allocator().destroy(this);
@@ -1529,7 +1528,7 @@ pub const DNSResolver = struct {
return .zero;
}
- const name = name_str.toSlice(globalThis, bun.default_allocator);
+ const name = name_str.toSliceClone(globalThis, bun.default_allocator);
var vm = globalThis.bunVM();
var resolver = vm.rareData().globalDNSResolver(vm);
@@ -1546,25 +1545,25 @@ pub const DNSResolver = struct {
return resolver.doLookup(name.slice(), 0, options, globalThis);
},
RecordType.CNAME => {
- return resolver.doResolveCAres(c_ares.struct_hostent, "cname", &name, globalThis);
+ return resolver.doResolveCAres(c_ares.struct_hostent, "cname", name.slice(), globalThis);
},
RecordType.MX => {
- return resolver.doResolveCAres(c_ares.struct_ares_mx_reply, "mx", &name, globalThis);
+ return resolver.doResolveCAres(c_ares.struct_ares_mx_reply, "mx", name.slice(), globalThis);
},
RecordType.NS => {
- return resolver.doResolveCAres(c_ares.struct_hostent, "ns", &name, globalThis);
+ return resolver.doResolveCAres(c_ares.struct_hostent, "ns", name.slice(), globalThis);
},
RecordType.PTR => {
- return resolver.doResolveCAres(c_ares.struct_hostent, "ptr", &name, globalThis);
+ return resolver.doResolveCAres(c_ares.struct_hostent, "ptr", name.slice(), globalThis);
},
RecordType.SOA => {
- return resolver.doResolveCAres(c_ares.struct_ares_soa_reply, "soa", &name, globalThis);
+ return resolver.doResolveCAres(c_ares.struct_ares_soa_reply, "soa", name.slice(), globalThis);
},
RecordType.SRV => {
- return resolver.doResolveCAres(c_ares.struct_ares_srv_reply, "srv", &name, globalThis);
+ return resolver.doResolveCAres(c_ares.struct_ares_srv_reply, "srv", name.slice(), globalThis);
},
RecordType.TXT => {
- return resolver.doResolveCAres(c_ares.struct_ares_txt_reply, "txt", &name, globalThis);
+ return resolver.doResolveCAres(c_ares.struct_ares_txt_reply, "txt", name.slice(), globalThis);
},
}
}
@@ -1663,12 +1662,12 @@ pub const DNSResolver = struct {
return .zero;
}
- const name = name_str.toSlice(globalThis, bun.default_allocator);
+ const name = name_str.toSliceClone(globalThis, bun.default_allocator);
var vm = globalThis.bunVM();
var resolver = vm.rareData().globalDNSResolver(vm);
- return resolver.doResolveCAres(c_ares.struct_ares_srv_reply, "srv", &name, globalThis);
+ return resolver.doResolveCAres(c_ares.struct_ares_srv_reply, "srv", name.slice(), globalThis);
}
pub fn resolveSoa(globalThis: *JSC.JSGlobalObject, callframe: *JSC.CallFrame) callconv(.C) JSC.JSValue {
@@ -1694,12 +1693,12 @@ pub const DNSResolver = struct {
return .zero;
}
- const name = name_str.toSlice(globalThis, bun.default_allocator);
+ const name = name_str.toSliceClone(globalThis, bun.default_allocator);
var vm = globalThis.bunVM();
var resolver = vm.rareData().globalDNSResolver(vm);
- return resolver.doResolveCAres(c_ares.struct_ares_soa_reply, "soa", &name, globalThis);
+ return resolver.doResolveCAres(c_ares.struct_ares_soa_reply, "soa", name.slice(), globalThis);
}
pub fn resolveCaa(globalThis: *JSC.JSGlobalObject, callframe: *JSC.CallFrame) callconv(.C) JSC.JSValue {
@@ -1725,12 +1724,12 @@ pub const DNSResolver = struct {
return .zero;
}
- const name = name_str.toSlice(globalThis, bun.default_allocator);
+ const name = name_str.toSliceClone(globalThis, bun.default_allocator);
var vm = globalThis.bunVM();
var resolver = vm.rareData().globalDNSResolver(vm);
- return resolver.doResolveCAres(c_ares.struct_ares_caa_reply, "caa", &name, globalThis);
+ return resolver.doResolveCAres(c_ares.struct_ares_caa_reply, "caa", name.slice(), globalThis);
}
pub fn resolveNs(globalThis: *JSC.JSGlobalObject, callframe: *JSC.CallFrame) callconv(.C) JSC.JSValue {
@@ -1756,12 +1755,12 @@ pub const DNSResolver = struct {
return .zero;
}
- const name = name_str.toSlice(globalThis, bun.default_allocator);
+ const name = name_str.toSliceClone(globalThis, bun.default_allocator);
var vm = globalThis.bunVM();
var resolver = vm.rareData().globalDNSResolver(vm);
- return resolver.doResolveCAres(c_ares.struct_hostent, "ns", &name, globalThis);
+ return resolver.doResolveCAres(c_ares.struct_hostent, "ns", name.slice(), globalThis);
}
pub fn resolvePtr(globalThis: *JSC.JSGlobalObject, callframe: *JSC.CallFrame) callconv(.C) JSC.JSValue {
@@ -1787,12 +1786,12 @@ pub const DNSResolver = struct {
return .zero;
}
- const name = name_str.toSlice(globalThis, bun.default_allocator);
+ const name = name_str.toSliceClone(globalThis, bun.default_allocator);
var vm = globalThis.bunVM();
var resolver = vm.rareData().globalDNSResolver(vm);
- return resolver.doResolveCAres(c_ares.struct_hostent, "ptr", &name, globalThis);
+ return resolver.doResolveCAres(c_ares.struct_hostent, "ptr", name.slice(), globalThis);
}
pub fn resolveCname(globalThis: *JSC.JSGlobalObject, callframe: *JSC.CallFrame) callconv(.C) JSC.JSValue {
@@ -1818,12 +1817,12 @@ pub const DNSResolver = struct {
return .zero;
}
- const name = name_str.toSlice(globalThis, bun.default_allocator);
+ const name = name_str.toSliceClone(globalThis, bun.default_allocator);
var vm = globalThis.bunVM();
var resolver = vm.rareData().globalDNSResolver(vm);
- return resolver.doResolveCAres(c_ares.struct_hostent, "cname", &name, globalThis);
+ return resolver.doResolveCAres(c_ares.struct_hostent, "cname", name.slice(), globalThis);
}
pub fn resolveMx(globalThis: *JSC.JSGlobalObject, callframe: *JSC.CallFrame) callconv(.C) JSC.JSValue {
@@ -1849,12 +1848,12 @@ pub const DNSResolver = struct {
return .zero;
}
- const name = name_str.toSlice(globalThis, bun.default_allocator);
+ const name = name_str.toSliceClone(globalThis, bun.default_allocator);
var vm = globalThis.bunVM();
var resolver = vm.rareData().globalDNSResolver(vm);
- return resolver.doResolveCAres(c_ares.struct_ares_mx_reply, "mx", &name, globalThis);
+ return resolver.doResolveCAres(c_ares.struct_ares_mx_reply, "mx", name.slice(), globalThis);
}
pub fn resolveNaptr(globalThis: *JSC.JSGlobalObject, callframe: *JSC.CallFrame) callconv(.C) JSC.JSValue {
@@ -1880,12 +1879,12 @@ pub const DNSResolver = struct {
return .zero;
}
- const name = name_str.toSlice(globalThis, bun.default_allocator);
+ const name = name_str.toSliceClone(globalThis, bun.default_allocator);
var vm = globalThis.bunVM();
var resolver = vm.rareData().globalDNSResolver(vm);
- return resolver.doResolveCAres(c_ares.struct_ares_naptr_reply, "naptr", &name, globalThis);
+ return resolver.doResolveCAres(c_ares.struct_ares_naptr_reply, "naptr", name.slice(), globalThis);
}
pub fn resolveTxt(globalThis: *JSC.JSGlobalObject, callframe: *JSC.CallFrame) callconv(.C) JSC.JSValue {
@@ -1911,17 +1910,15 @@ pub const DNSResolver = struct {
return .zero;
}
- const name = name_str.toSlice(globalThis, bun.default_allocator);
+ const name = name_str.toSliceClone(globalThis, bun.default_allocator);
var vm = globalThis.bunVM();
var resolver = vm.rareData().globalDNSResolver(vm);
- return resolver.doResolveCAres(c_ares.struct_ares_txt_reply, "txt", &name, globalThis);
+ return resolver.doResolveCAres(c_ares.struct_ares_txt_reply, "txt", name.slice(), globalThis);
}
- pub fn doResolveCAres(this: *DNSResolver, comptime cares_type: type, comptime type_name: []const u8, name_str: *const JSC.ZigString.Slice, globalThis: *JSC.JSGlobalObject) JSC.JSValue {
- var name = name_str.slice();
-
+ pub fn doResolveCAres(this: *DNSResolver, comptime cares_type: type, comptime type_name: []const u8, name: []const u8, globalThis: *JSC.JSGlobalObject) JSC.JSValue {
var channel: *c_ares.Channel = switch (this.getChannel()) {
.result => |res| res,
.err => |err| {
@@ -1943,7 +1940,7 @@ pub const DNSResolver = struct {
var cache = this.getOrPutIntoResolvePendingCache(ResolveInfoRequest(cares_type, type_name), key, cache_name);
if (cache == .inflight) {
// CAresLookup will have the name ownership
- var cares_lookup = CAresLookup(cares_type, type_name).init(globalThis, globalThis.allocator(), name_str) catch unreachable;
+ var cares_lookup = CAresLookup(cares_type, type_name).init(globalThis, globalThis.allocator(), name) catch unreachable;
cache.inflight.append(cares_lookup);
return cares_lookup.promise.value();
}
@@ -1951,7 +1948,7 @@ pub const DNSResolver = struct {
var request = ResolveInfoRequest(cares_type, type_name).init(
cache,
this,
- name_str, // CAresLookup will have the ownership
+ name, // CAresLookup will have the ownership
globalThis,
cache_name,
) catch unreachable;
diff --git a/src/bun.js/bindings/bindings.zig b/src/bun.js/bindings/bindings.zig
index 9a775990c..fa7efa728 100644
--- a/src/bun.js/bindings/bindings.zig
+++ b/src/bun.js/bindings/bindings.zig
@@ -368,7 +368,7 @@ pub const ZigString = extern struct {
}
pub fn sliceZ(this: Slice) [:0]const u8 {
- return this.ptr[0..this.len:0];
+ return this.ptr[0..this.len :0];
}
pub fn toSliceZ(this: Slice, buf: []u8) [:0]const u8 {
@@ -386,7 +386,7 @@ pub const ZigString = extern struct {
std.mem.copy(u8, buf[0..this.len], this.slice());
buf[this.len] = 0;
- return buf[0..this.len:0];
+ return buf[0..this.len :0];
}
pub fn mut(this: Slice) []u8 {
@@ -604,7 +604,7 @@ pub const ZigString = extern struct {
if (this.len == 0)
return Slice.empty;
if (is16Bit(&this)) {
- var buffer = this.toOwnedSlice(allocator) catch unreachable;
+ const buffer = this.toOwnedSlice(allocator) catch unreachable;
return Slice{
.allocator = NullableAllocator.init(allocator),
.ptr = buffer.ptr,
@@ -613,7 +613,7 @@ pub const ZigString = extern struct {
}
if (!this.isUTF8() and !strings.isAllASCII(untagged(this.ptr)[0..this.len])) {
- var buffer = this.toOwnedSlice(allocator) catch unreachable;
+ const buffer = this.toOwnedSlice(allocator) catch unreachable;
return Slice{
.allocator = NullableAllocator.init(allocator),
.ptr = buffer.ptr,
@@ -627,6 +627,17 @@ pub const ZigString = extern struct {
};
}
+ pub fn toSliceClone(this: ZigString, allocator: std.mem.Allocator) Slice {
+ if (this.len == 0)
+ return Slice.empty;
+ const buffer = this.toOwnedSlice(allocator) catch unreachable;
+ return Slice{
+ .allocator = NullableAllocator.init(allocator),
+ .ptr = buffer.ptr,
+ .len = @truncate(u32, buffer.len),
+ };
+ }
+
pub fn toSliceZ(this: ZigString, allocator: std.mem.Allocator) Slice {
if (this.len == 0)
return Slice.empty;
@@ -1305,6 +1316,7 @@ pub const JSString = extern struct {
return out;
}
+ // doesn't always allocate
pub fn toSlice(
this: *JSString,
global: *JSGlobalObject,
@@ -1315,6 +1327,16 @@ pub const JSString = extern struct {
return str.toSlice(allocator);
}
+ pub fn toSliceClone(
+ this: *JSString,
+ global: *JSGlobalObject,
+ allocator: std.mem.Allocator,
+ ) ZigString.Slice {
+ var str = ZigString.init("");
+ this.toZigString(global, &str);
+ return str.toSliceClone(allocator);
+ }
+
pub fn toSliceZ(
this: *JSString,
global: *JSGlobalObject,
diff --git a/test/bun.js/node-dns.test.js b/test/bun.js/node-dns.test.js
index b3ee09d48..b84b8c9ba 100644
--- a/test/bun.js/node-dns.test.js
+++ b/test/bun.js/node-dns.test.js
@@ -25,7 +25,9 @@ test("dns.resolveSrv (_test._tcp.test.socketify.dev)", (done) => {
dns.resolveSrv("_test._tcp.test.socketify.dev", (err, results) => {
expect(err).toBeNull();
expect(results instanceof Array).toBe(true);
- expect(results[0].name).toBe("_dc-srv.130c90ab9de1._test._tcp.test.socketify.dev");
+ expect(results[0].name).toBe(
+ "_dc-srv.130c90ab9de1._test._tcp.test.socketify.dev",
+ );
expect(results[0].priority).toBe(50);
expect(results[0].weight).toBe(50);
expect(results[0].port).toBe(80);
@@ -50,11 +52,9 @@ test("dns.resolveTxt (txt.socketify.dev)", (done) => {
});
});
-
test("dns.resolveSoa (bun.sh)", (done) => {
dns.resolveSoa("bun.sh", (err, result) => {
expect(err).toBeNull();
-
expect(result.serial).toBe(2295878541);
expect(result.refresh).toBe(10000);
expect(result.retry).toBe(2400);
@@ -62,7 +62,6 @@ test("dns.resolveSoa (bun.sh)", (done) => {
expect(result.minttl).toBe(3600);
expect(result.nsname).toBe("hans.ns.cloudflare.com");
expect(result.hostmaster).toBe("dns.cloudflare.com");
-
done(err);
});
});
@@ -71,10 +70,10 @@ test("dns.resolveNaptr (naptr.socketify.dev)", (done) => {
dns.resolveNaptr("naptr.socketify.dev", (err, results) => {
expect(err).toBeNull();
expect(results instanceof Array).toBe(true);
- expect(results[0].flags).toBe('S');
- expect(results[0].service).toBe('test');
- expect(results[0].regexp).toBe('');
- expect(results[0].replacement).toBe('');
+ expect(results[0].flags).toBe("S");
+ expect(results[0].service).toBe("test");
+ expect(results[0].regexp).toBe("");
+ expect(results[0].replacement).toBe("");
expect(results[0].order).toBe(1);
expect(results[0].preference).toBe(12);
done(err);
@@ -86,29 +85,27 @@ test("dns.resolveCaa (caa.socketify.dev)", (done) => {
expect(err).toBeNull();
expect(results instanceof Array).toBe(true);
expect(results[0].critical).toBe(0);
- expect(results[0].issue).toBe('bun.sh');
+ expect(results[0].issue).toBe("bun.sh");
done(err);
});
});
-
test("dns.resolveMx (bun.sh)", (done) => {
dns.resolveMx("bun.sh", (err, results) => {
expect(err).toBeNull();
expect(results instanceof Array).toBe(true);
- expect(results[0].priority).toBe(10);
- expect(results[0].exchange).toBe('eforward1.registrar-servers.com');
+ const priority = results[0].priority;
+ expect(priority >= 0 && priority < 65535).toBe(true);
+ expect(results[0].exchange.includes(".registrar-servers.com")).toBe(true);
done(err);
});
});
-
-
test("dns.resolveNs (bun.sh) ", (done) => {
dns.resolveNs("bun.sh", (err, results) => {
expect(err).toBeNull();
expect(results instanceof Array).toBe(true);
- expect(results[0]).toBe('hans.ns.cloudflare.com');
+ expect(results[0].includes(".ns.cloudflare.com")).toBe(true);
done(err);
});
});
@@ -117,7 +114,7 @@ test("dns.resolvePtr (ptr.socketify.dev)", (done) => {
dns.resolvePtr("ptr.socketify.dev", (err, results) => {
expect(err).toBeNull();
expect(results instanceof Array).toBe(true);
- expect(results[0]).toBe('bun.sh');
+ expect(results[0]).toBe("bun.sh");
done(err);
});
});
@@ -126,12 +123,11 @@ test("dns.resolveCname (cname.socketify.dev)", (done) => {
dns.resolveCname("cname.socketify.dev", (err, results) => {
expect(err).toBeNull();
expect(results instanceof Array).toBe(true);
- expect(results[0]).toBe('bun.sh');
+ expect(results[0]).toBe("bun.sh");
done(err);
});
});
-
test("dns.lookup (example.com)", (done) => {
dns.lookup("example.com", (err, address, family) => {
expect(err).toBeNull();
@@ -151,4 +147,4 @@ test("dns.lookup (localhost)", (done) => {
done(err);
});
-}); \ No newline at end of file
+});