Class JavaAgentHandler

hallucinate/java.py:17–65

class JavaAgentHandler(socketserver.BaseRequestHandler):

    def read_bytes(self, n):
        r = b''
        while len(r) < n:
            r += self.request.recv(n - len(r))
        return r

    def read_string(self):
        length, = struct.unpack(">H", self.read_bytes(2))
        return self.request.recv(length).decode('utf-8')

    def read_entry(self):
        return self.read_string(), self.read_string()

    def handle(self):
        key = self.read_string()

        if not compare_digest(key, self.server.key):
            logging.warning("Client provided wrong key")
            self.request.close()
            return

        while True:
            r = self.read_bytes(4)
            nentries, = struct.unpack('>I', r)
            rdata = dict()
            for i in range(nentries):
                k, v = self.read_entry()
                rdata[k] = v

            #  emulate conn entry, no nested array support
            rdata['conn'] = {
                'l': {'ip': rdata['localAddr'], 'port': int(rdata['localPort'])},
                'r': {'ip': rdata['remoteAddr'], 'port': int(rdata['remotePort'])}
            }
            logging.debug('Received request from Java agent: %s', rdata)

            def rhandle(data):
                logging.debug('Sending response to Java agent: %s', data)
                p = struct.pack('>I', len(data))
                for k, v in data.items():
                    ke = str(k).encode('utf-8')
                    ve = str(v).encode('utf-8')
                    p += struct.pack('>H', len(ke)) + ke
                    p += struct.pack('>H', len(ve)) + ve
                self.request.sendall(p)

            self.server.handler(rdata, rhandle)


class JavaAgentServer(socketserver.ThreadingMixIn, socketserver.TCPServer):