File: doh_b64encode_query.py

package info (click to toggle)
knot-resolver 6.0.17-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 16,376 kB
  • sloc: javascript: 42,732; ansic: 40,311; python: 12,580; cpp: 2,121; sh: 1,988; xml: 193; makefile: 181
file content (26 lines) | stat: -rwxr-xr-x 633 bytes parent folder | download | duplicates (4)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
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()