From: Tomas Krizek <tomas.krizek@nic.cz>
Date: Thu, 5 Nov 2020 09:42:21 +0000 (+0100)
Subject: scripts: utility to get base64 url encoded query for DoH GET
X-Git-Tag: v5.2.0~2^2
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=077287d0c62076ce4c84803c30457becb8b90d46;p=thirdparty%2Fknot-resolver.git

scripts: utility to get base64 url encoded query for DoH GET
---

diff --git a/scripts/doh_b64encode_query.py b/scripts/doh_b64encode_query.py
new file mode 100755
index 000000000..59569b81a
--- /dev/null
+++ b/scripts/doh_b64encode_query.py
@@ -0,0 +1,26 @@
+#!/usr/bin/env python3
+import argparse
+import base64
+
+import dns
+import dns.message
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        description='Convert query name and type to base64 URL-encoded form')
+    parser.add_argument('qname', type=str, help='query name')
+    parser.add_argument('qtype', type=str, help='query type')
+    args = parser.parse_args()
+
+    msg = dns.message.make_query(args.qname, args.qtype, dns.rdataclass.IN)
+    msg.id = 0
+    wire = msg.to_wire()
+    encoded = base64.urlsafe_b64encode(wire)
+    printable = encoded.decode('utf-8')
+
+    print(printable)
+
+
+if __name__ == '__main__':
+    main()