def setUp(self): # Start the proxy server create_temp_dir() self._proxy = InterceptProxy(self.IP, 0, ExtendedUrllib()) self._proxy.start() self._proxy.wait_for_start() port = self._proxy.get_port() # Build the proxy opener proxy_handler = urllib2.ProxyHandler( {'http': 'http://%s:%s' % (self.IP, port)}) self.proxy_opener = urllib2.build_opener(proxy_handler, urllib2.HTTPHandler)
def _start_proxy(self, ip=None, port=None, silent=False): """Starts the proxy.""" if not ip: ipport = self.pref.get_value('proxy', 'ipport') ip, port = ipport.split(":") self.w3af.mainwin.sb(_("Starting local proxy")) try: self.proxy = InterceptProxy(ip, int(port), self.w3af.uri_opener) except ProxyException as w3: if not silent: self.show_alert(_(str(w3))) raise w3 else: self.proxy.start()