bta-proxy/bta_proxy/dpi.py

from asyncio.queues import Queue
import time

from bta_proxy.datainputstream import AsyncDataInputStream
from bta_proxy.packets import *


async def inspect_client(queue: Queue, addr: tuple[str, int]):
    dis = AsyncDataInputStream(queue)
    last_time = time.time()
    while True:
        pkt = await Packet.read_packet(dis)
        now = time.time()
        delta = now - last_time
        last_time = now

        match pkt.packet_id:
            case Packet10Flying.packet_id:
                continue
            case Packet11PlayerPosition.packet_id:
                continue
            case Packet12PlayerLook.packet_id:
                continue
            case Packet13LookMove.packet_id:
                continue
            case Packet255KickDisconnect.packet_id:
                break
            case _:
                print(f"C {delta*1000:+8.1f}ms {pkt}")

async def inspect_server(queue: Queue, addr: tuple[str, int]):
    dis = AsyncDataInputStream(queue)
    last_time = time.time()
    while True:
        pkt = await Packet.read_packet(dis)
        now = time.time()
        delta = now - last_time
        last_time = now

        match pkt.packet_id:
            case Packet50PreChunk.packet_id:
                continue
            case Packet51MapChunk.packet_id:
                continue
            case Packet31RelEntityMove.packet_id:
                continue
            case Packet32EntityLook.packet_id:
                continue
            case Packet33RelEntityMoveLook.packet_id:
                continue
            case _:
                print(f"S {delta*1000:+8.1f}ms {pkt}")