VComLink: Async Communication Between EVE-OS and VMs

import socket
import messages_pb2
import http.client

VSOCK_CID_HOST = 2
VSOCK_PORT = 2000

class VsockHTTPConnection(http.client.HTTPConnection):
    def connect(self):
        self.sock = socket.socket(socket.AF_VSOCK, socket.SOCK_STREAM)
        self.sock.connect((VSOCK_CID_HOST, VSOCK_PORT))

def main():
    # Create the protobuf request
    request_proto = messages_pb2.TpmRequestGetPub()
    request_proto.index = 0x81010001  # Example TPM index
    request_bytes = request_proto.SerializeToString()

    # Set up connection to vsock host
    conn = VsockHTTPConnection("vsock", VSOCK_PORT)
    conn.request("POST", "/tpm/getpub", body=request_bytes, headers={
        "Content-Type": "application/octet-stream"
    })

    # Get the HTTP response
    resp = conn.getresponse()
    if resp.status != 200:
        raise Exception(f"HTTP error: {resp.status} {resp.reason}")

    # Read and parse the response protobuf
    response_bytes = resp.read()
    response_proto = messages_pb2.TpmResponseGetPub()
    response_proto.ParseFromString(response_bytes)

    # Print result
    print("TPM Public Key Retrieved:")
    print(f"- Public Key (hex): {response_proto.public.hex()}...")
    print(f"- Algorithm: {response_proto.algorithm}")
    print(f"- Attributes (bitmask): {response_proto.attributes:#010x}")

    conn.close()

if __name__ == "__main__":
    main()

TPM Public Key Retrieved:
- Public Key (hex): 0001000b000300b200208371976... (truncated)
- Algorithm: 1
- Attributes (bitmask): 0x000300b2