GithubFollow/One-KVM
1
0
Fork 0
mirror of https://github.com/mofeng-git/One-KVM.git synced 2025-12-13 01:30:31 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

ipv6 stun

Browse Source
This commit is contained in:
Devaev Maxim 2021-05-25 23:54:20 +03:00
parent 6ce07208a1
commit 9b4cbfab7f
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
kvmd/apps/janus/stun.py
View File

@ -1,5 +1,6 @@
import asyncio
import socket
import ipaddress
import struct
import secrets
import dataclasses
@ -168,10 +169,9 @@ class Stun:
def __parse_address(self, data: bytes) -> StunAddress:
family = data[1]
if family == 1:
parts = struct.unpack(">HBBBB", data[2:8])
return StunAddress(
ip=".".join(map(str, parts[1:])),
port=parts[0],
)
raise RuntimeError(f"Only IPv4 supported; received: {family}")
port = struct.unpack(">H", data[2:4])[0]
if family == 0x01:
return StunAddress(str(ipaddress.IPv4Address(data[4:8])), port)
elif family == 0x02:
return StunAddress(str(ipaddress.IPv6Address(data[4:20])), port)
raise RuntimeError(f"Unknown family; received: {family}")