nil0x42/phpsploit

import metadict
from core import session, tunnel, plugins
from datatypes import Path


class Payload(metadict.MetaDict):
    # the phpsploit env vars to auto add
    # to $PHPSPLOIT array on php side
    _unherited_env_vars = ["PATH_SEP"]

    _php_vars_template = 'global $PHPSPLOIT;$PHPSPLOIT=%s;\n'

    def __init__(self, filename, **kwargs):
        self.response = None
        for key in self._unherited_env_vars:
            self[key] = session.Env[key]
        for key, value in kwargs.items():
            self[key] = value
        plugin_path = plugins.current_plugin.path
        self.payload = Path(plugin_path, filename, mode='fr').phpcode()

    def send(self, **kwargs):
        var_list = dict(self)
        for key, value in kwargs.items():
            var_list[key] = value
        php_vars = self._php_vars_template % tunnel.payload.py2php(var_list)

        result = tunnel.send(php_vars + self.payload)
        if result.response_error:
            raise PayloadError(result.response_error)
        return result.response


class PayloadError(Exception):
    """Exception raised when a send payload returned an __ERROR__ obj"""