add IPv6 support for queryAddr pack

This commit is contained in:
ghost 2023-12-24 09:07:51 +02:00
parent 2f9c232540
commit 5f07a83832

View File

@ -1,5 +1,6 @@
from time import time
from struct import pack
import ipaddress
class ServerEntry:
challenge2 = 0
@ -67,8 +68,7 @@ class ServerEntry:
self.addr = addr
# Shortcuts for generating query
self.queryAddr = b''
for i in addr[0].split('.'):
self.queryAddr += pack('!B', int(i))
self.queryAddr += ipaddress.ip_address(addr[0]).packed
self.queryAddr += pack('!H', int(addr[1]))
# Random number that server must return