File: receive_notify.py

package info (click to toggle)
dnspython 2.7.0-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 2,448 kB
  • sloc: python: 34,885; sh: 7; makefile: 4
file content (38 lines) | stat: -rw-r--r-- 1,021 bytes parent folder | download
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
27
28
29
30
31
32
33
34
35
36
37
38
#!/usr/bin/env python3

# This is just a toy, real code would check that the received message
# really was a NOTIFY, and otherwise handle errors.

import socket
from typing import cast

import dns.flags
import dns.message
import dns.name
import dns.rdataclass
import dns.rdatatype

address = "127.0.0.1"
port = 53535

s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
s.bind((address, port))
while True:
    (wire, address) = s.recvfrom(512)
    notify = dns.message.from_wire(wire)

    try:
        soa = notify.find_rrset(
            notify.answer, notify.question[0].name, dns.rdataclass.IN, dns.rdatatype.SOA
        )

        # Do something with the SOA RR here
        print("The serial number for", soa.name, "is", soa[0].serial)
    except KeyError:
        # No SOA RR in the answer section.
        pass

    response = dns.message.make_response(notify)  # type: dns.message.Message
    response.flags |= dns.flags.AA
    wire = response.to_wire(cast(dns.name.Name, response))
    s.sendto(wire, address)