71 lines
1.7 KiB
Python
71 lines
1.7 KiB
Python
# Kyler Olsen
|
|
# CS 2450 Final Project
|
|
# Apr 2025
|
|
|
|
from __future__ import annotations
|
|
from typing import TYPE_CHECKING
|
|
import struct
|
|
|
|
if TYPE_CHECKING:
|
|
from socket import socket
|
|
|
|
def pack_varint(data: int) -> bytes:
|
|
ordinal = b''
|
|
while True:
|
|
byte = data & 0x7F
|
|
data >>= 7
|
|
ordinal += struct.pack('B', byte | (0x80 if data > 0 else 0))
|
|
if data == 0:
|
|
break
|
|
return ordinal
|
|
|
|
def unpack_varint(conn: socket) -> int:
|
|
data = 0
|
|
for i in range(5):
|
|
ordinal = conn.recv(1)
|
|
if len(ordinal) == 0:
|
|
break
|
|
byte = ord(ordinal)
|
|
data |= (byte & 0x7F) << 7*i
|
|
if not byte & 0x80:
|
|
break
|
|
return data
|
|
|
|
def pack_string(text: str) -> bytes:
|
|
utf = text.encode('utf-8')
|
|
data = pack_varint(len(utf))
|
|
data += utf
|
|
return data
|
|
|
|
def unpack_string(conn: socket) -> str:
|
|
length = unpack_varint(conn)
|
|
data = conn.recv(length)
|
|
text = data.decode('utf-8')
|
|
return text
|
|
|
|
def pack_varint_array(array: list[int]) -> bytes:
|
|
data = pack_varint(len(array))
|
|
for i in array:
|
|
data += pack_varint(i)
|
|
return data
|
|
|
|
def unpack_varint_array(conn: socket) -> list[int]:
|
|
length = unpack_varint(conn)
|
|
array: list[int] = []
|
|
for _ in range(length):
|
|
array.append(unpack_varint(conn))
|
|
return array
|
|
|
|
def pack_string_array(array: list[str]) -> bytes:
|
|
data = pack_varint(len(array))
|
|
for i in array:
|
|
data += pack_string(i)
|
|
return data
|
|
|
|
def unpack_string_array(conn: socket) -> list[str]:
|
|
length = unpack_varint(conn)
|
|
array: list[str] = []
|
|
for _ in range(length):
|
|
array.append(unpack_string(conn))
|
|
return array
|