add support for subdomains

vandot · Dec 5, 2023 · 23c7905 · 23c7905
1 parent dbcd173
commit 23c7905
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lodns.nimble b/lodns.nimble
@@ -1,6 +1,6 @@
 # Package
 
-version       = "0.1.8"
+version       = "0.1.9"
 author        = "vandot"
 description   = "Simple DNS server for local development"
 license       = "MIT"

diff --git a/src/lodnspkg/server.nim b/src/lodnspkg/server.nim
@@ -27,7 +27,7 @@ proc serve*(ip: string, port: int, tld: string) =
         else:
             log("error", "type=", message.questions[0].qtype, " domain=", domain, " error=unsupported-type")
             rr = initResourceRecord(domain, Type.NULL, Class.IN, 0'i32, 0'u16, RDataNULL())
-        if rsplit(domain, ".", maxsplit=2)[1] != tld:
+        if not endsWith(domain, tld & "."):
             log("error", "type=", message.questions[0].qtype, " domain=", domain, " error=unsupported-domain")
             rr = initResourceRecord(domain, Type.NULL, Class.IN, 0'i32, 0'u16, RDataNULL())
         let header = initHeader(message.header.id, QR.Response)