コード例 #1
0
ファイル: proxy.py プロジェクト: jeffreywugz/antigfw
def http(req, sock_factory, timeout=None):
    t = time.time()
    tout = conn.set_timeout(timeout)
    hostname, port, uri = parse_target(req.url)
    reqx = copy.copy(req)
    reqx.uri = uri
    reqx.headers = [(h, v) for h, v in req.headers if not h.startswith('Proxy')]
    with sock_factory.socket() as sock:
        sock.connect((hostname, port))
        stream1 = sock.makefile()

        if VERBOSE: req.debug()
        tout(reqx.send_header)(stream1)
        streamcopy(reqx, req.stream, stream1, tout)
        stream1.flush()

        res = recv_msg(stream1, HttpResponse)
        if VERBOSE: res.debug()
        tout(res.send_header)(req.stream)
        hasbody = req.method.upper() != 'HEAD' and res.code not in CODE_NOBODY
        streamcopy(res, stream1, req.stream, tout, hasbody)
        req.stream.flush()
    res.connection = req.get_header('Proxy-Connection', '').lower() == 'keep-alive' and\
        res.get_header('Connection', 'close').lower() != 'close'
    logger.debug('%s with %d in %0.2f, %s' % (
            req.uri.split('?', 1)[0], res.code, time.time() - t,
            'keep' if res.connection else 'close'))
    return res
コード例 #2
0
def http(req, sock_factory, timeout=None):
    t = time.time()
    tout = conn.set_timeout(timeout)
    hostname, port, uri = parse_target(req.url)
    reqx = copy.copy(req)
    reqx.uri = uri
    reqx.headers = [(h, v) for h, v in req.headers
                    if not h.startswith('Proxy')]
    with sock_factory.socket() as sock:
        sock.connect((hostname, port))
        stream1 = sock.makefile()

        if VERBOSE: req.debug()
        tout(reqx.send_header)(stream1)
        streamcopy(reqx, req.stream, stream1, tout)
        stream1.flush()

        res = recv_msg(stream1, HttpResponse)
        if VERBOSE: res.debug()
        tout(res.send_header)(req.stream)
        hasbody = req.method.upper() != 'HEAD' and res.code not in CODE_NOBODY
        streamcopy(res, stream1, req.stream, tout, hasbody)
        req.stream.flush()
    res.connection = req.get_header('Proxy-Connection', '').lower() == 'keep-alive' and\
        res.get_header('Connection', 'close').lower() != 'close'
    logger.debug('%s with %d in %0.2f, %s' %
                 (req.uri.split('?', 1)[0], res.code, time.time() - t,
                  'keep' if res.connection else 'close'))
    return res
コード例 #3
0
ファイル: serve.py プロジェクト: jeffreywugz/antigfw
    def loadconfig(self):
        self.config = import_config(self.cfgs, self.env)
        self.proxy_auth = proxy.get_proxy_auth(self.config.get('users'))

        self.proxies = self.config.get('proxies', None)
        if self.proxies is None: self.proxies = []
        if self.config.get('max_conn', None):
            self.proxies.extend(map(self.ssh2proxy, self.config['sshs']))
        self.upstream = self.config.get('upstream')

        if self.dns is not None: self.dns.stop()
        self.dns = self.config.get('dnsserver')
        dnsport = self.config.get('dnsport', None)
        if dnsport: self.dns.runserver(dnsport)
        self.dofilter = self.config.get('dofilter')
        self.whitenf = self.config.get('whitenets')
        self.blacknf = self.config.get('blacknets')
        self.direct = conn.DirectManager(self.dns)

        self.func_connect = conn.set_timeout(self.config.get('conn_tout'))(proxy.connect)
        self.func_http = conn.set_timeout(self.config.get('http_tout'))(proxy.http)
コード例 #4
0
    def loadconfig(self):
        self.config = import_config(self.cfgs, self.env)
        self.proxy_auth = proxy.get_proxy_auth(self.config.get('users'))

        self.proxies = self.config.get('proxies', None)
        if self.proxies is None: self.proxies = []
        if self.config.get('max_conn', None):
            self.proxies.extend(map(self.ssh2proxy, self.config['sshs']))
        self.upstream = self.config.get('upstream')

        if self.dns is not None: self.dns.stop()
        self.dns = self.config.get('dnsserver')
        dnsport = self.config.get('dnsport', None)
        if dnsport: self.dns.runserver(dnsport)
        self.dofilter = self.config.get('dofilter')
        self.whitenf = self.config.get('whitenets')
        self.blacknf = self.config.get('blacknets')
        self.direct = conn.DirectManager(self.dns)

        self.func_connect = conn.set_timeout(self.config.get('conn_tout'))(
            proxy.connect)
        self.func_http = conn.set_timeout(self.config.get('http_tout'))(
            proxy.http)