import net.lightbody.bmp.BrowserMobProxy; //导入依赖的package包/类
private ProxyDriverIntegrator getProxyDriverIntegrator(RequestFilter recordRequestFilter,
                                                       WebDriverSupplier webDriverSupplier,
                                                       DriverServiceSupplier driverServiceSupplier,
                                                       @Named(PATH_TO_DRIVER) String pathToDriverExecutable,
                                                       @Named(SCREEN) String screen,
                                                       @Named(TIMEOUT) int timeout,
                                                       ResponseFilter responseFilter) throws IOException {
    BrowserMobProxy proxy = createBrowserMobProxy(timeout, recordRequestFilter, responseFilter);
    proxy.start(0);
    logger.info("Proxy running on port " + proxy.getPort());
    Proxy seleniumProxy = createSeleniumProxy(proxy);
    DesiredCapabilities desiredCapabilities = createDesiredCapabilities(seleniumProxy);
    DriverService driverService = driverServiceSupplier.getDriverService(pathToDriverExecutable, screen);
    WebDriver driver = webDriverSupplier.get(driverService, desiredCapabilities);

    return new ProxyDriverIntegrator(driver, proxy, driverService);
}
 

import net.lightbody.bmp.BrowserMobProxy; //导入依赖的package包/类
public BrowserMobProxy createBrowserMobProxy(int timeout, RequestFilter requestFilter, ResponseFilter responseFilter) {
    BrowserMobProxyServer proxy = new BrowserMobProxyServer();
    proxy.enableHarCaptureTypes(CaptureType.REQUEST_CONTENT, CaptureType.RESPONSE_CONTENT);
    proxy.newHar("measurements");
    proxy.addRequestFilter(requestFilter);
    proxy.addResponseFilter(responseFilter);
    proxy.setIdleConnectionTimeout(timeout, TimeUnit.SECONDS);
    proxy.setRequestTimeout(timeout, TimeUnit.SECONDS);
    return proxy;
}
 

import net.lightbody.bmp.BrowserMobProxy; //导入依赖的package包/类
@Test
public void providesAccessToConnectedDriverProxyAndDriverService() {
    WebDriver driver = mock(WebDriver.class);
    BrowserMobProxy proxy = mock(BrowserMobProxy.class);
    DriverService driverService = mock(DriverService.class);

    ProxyDriverIntegrator proxyDriverIntegrator = new ProxyDriverIntegrator(driver, proxy, driverService);

    assertEquals(driver, proxyDriverIntegrator.getWebDriver());
    assertEquals(proxy, proxyDriverIntegrator.getProxy());
    assertEquals(driverService, proxyDriverIntegrator.getDriverService());
}
 

import net.lightbody.bmp.BrowserMobProxy; //导入依赖的package包/类
@Test
public void shouldCallFilterByRegistry() throws IOException {
  // given
  BrowserMobProxy proxy = proxyController.startProxyServer(InetAddress.getLocalHost());
  proxyController.startAnalysis();
  requestFilterRegistry.add(requestFilter);
  // when
  DesiredCapabilities capabilities = proxyCapabilities(proxy);
  visitSamplePage(capabilities);
  proxyController.stopAnalysis();
  proxyController.stopProxyServer();
  // then
  verify(requestFilter, atLeastOnce()).filterRequest(any(HttpRequest.class),
      any(HttpMessageContents.class), any(HttpMessageInfo.class));
}
 

import net.lightbody.bmp.BrowserMobProxy; //导入依赖的package包/类
@Test
public void shouldCallFilterByRegistry() throws IOException {
  // given
  BrowserMobProxy browserMobProxy = new BrowserMobProxyServer();
  startProxyServer(browserMobProxy);
  requestFilterRegistry.add(requestFilter);
  browserMobProxy.addRequestFilter(requestFilterRegistry);
  // when
  DesiredCapabilities capabilities = proxyCapabilities(browserMobProxy);
  visitSamplePage(capabilities);
  browserMobProxy.stop();
  // then
  verify(requestFilter, atLeastOnce()).filterRequest(any(HttpRequest.class),
      any(HttpMessageContents.class), any(HttpMessageInfo.class));
}
 

import net.lightbody.bmp.BrowserMobProxy; //导入依赖的package包/类
DesiredCapabilities proxyCapabilities(BrowserMobProxy browserMobProxy) {
  browserMobProxy.enableHarCaptureTypes(
      CaptureType.REQUEST_HEADERS,
      CaptureType.RESPONSE_HEADERS,
      CaptureType.REQUEST_CONTENT,
      CaptureType.RESPONSE_CONTENT);

  Proxy seleniumProxy = ClientUtil.createSeleniumProxy(browserMobProxy);
  DesiredCapabilities capabilities = new DesiredCapabilities();
  capabilities.setCapability(CapabilityType.PROXY, seleniumProxy);
  return capabilities;
}
 

import net.lightbody.bmp.BrowserMobProxy; //导入依赖的package包/类
public BrowserMobProxy startProxyServer(InetAddress proxyAddress) {
  if (!browserMobProxy.isStarted()) {
    try {
      browserMobProxy.start(port, proxyAddress);
      browserMobProxy.addRequestFilter(filterRegistry);
    } catch (Exception e) {
      LOG.error("Can't start proxy", e);
    }
  }
  return browserMobProxy;
}
 

import net.lightbody.bmp.BrowserMobProxy; //导入依赖的package包/类
private DesiredCapabilities enableProxy(Capabilities capabilities) {
  DesiredCapabilities caps = new DesiredCapabilities(capabilities);
  try {
    InetAddress proxyInetAddress = InetAddress.getByName(proxyIp);
    BrowserMobProxy browserMobProxy = proxyController.startProxyServer(proxyInetAddress);
    Proxy seleniumProxy = ClientUtil.createSeleniumProxy(browserMobProxy, proxyInetAddress);
    caps.setCapability(CapabilityType.PROXY, seleniumProxy);
  } catch (UnknownHostException e) {
    throw new IllegalStateException(e);
  }
  return caps;
}
 

import net.lightbody.bmp.BrowserMobProxy; //导入依赖的package包/类
/**
 * create BrowserMobProxy Server object
 * @param port number 
 * @return BrowserMobProxy
 * 
 */
public static BrowserMobProxy createProxy(int port) {
	BrowserMobProxyServer proxy = new BrowserMobProxyServer();
	proxy.setTrustAllServers(true);
	//System.setProperty("jsse.enableSNIExtension", "false");
	return proxy;
}
 

import net.lightbody.bmp.BrowserMobProxy; //导入依赖的package包/类
public static void setupBrowserMobProxy()
{
	if (Configuration.getBoolean(Parameter.BROWSERMOB_PROXY)) {
		BrowserMobProxy proxy = startProxy();
		Integer port = proxy.getPort();

		String currentIP = NetworkUtil.getIpAddress();
		LOGGER.debug("Set http proxy settings to use BrowserMobProxy host: " + currentIP + "; port: " + port);
		
		R.CONFIG.put("proxy_host", currentIP);
		R.CONFIG.put("proxy_port", port.toString());
		R.CONFIG.put("proxy_protocols", "http");
		
	}
}
 

import net.lightbody.bmp.BrowserMobProxy; //导入依赖的package包/类
/**
 * start BrowserMobProxy Server
 * 
 * @return BrowserMobProxy
 * 
 */
public static BrowserMobProxy startProxy() {
	if (!Configuration.getBoolean(Parameter.BROWSERMOB_PROXY)) {
		return null;
	}
	// integrate browserMob proxy if required here
	BrowserMobProxy proxy = null;
	long threadId = Thread.currentThread().getId();
	if (proxies.containsKey(threadId)) {
		proxy = proxies.get(threadId);
	} else {
		proxy = ProxyPool.createProxy(Configuration.getInt(Parameter.BROWSERMOB_PORT));
		proxies.put(Thread.currentThread().getId(), proxy);
	}
	
	if (!proxy.isStarted()) {
		LOGGER.info("Starting BrowserMob proxy...");
		proxy.start(Configuration.getInt(Parameter.BROWSERMOB_PORT));
	} else {
		LOGGER.info("BrowserMob proxy is already started on port " + proxy.getPort());
	}

	Integer port = proxy.getPort();

	String currentIP = NetworkUtil.getIpAddress();
	LOGGER.warn("Set http/https proxy settings ONLY to use with BrowserMobProxy host: " + currentIP + "; port: " + port);

	//TODO: double check mobile proxy support
	R.CONFIG.put("proxy_host", currentIP);
	R.CONFIG.put("proxy_port", port.toString());
	R.CONFIG.put("proxy_protocols", "http,https");

	return proxy;
}
 

import net.lightbody.bmp.BrowserMobProxy; //导入依赖的package包/类
/**
 * get registered BrowserMobProxy Server
 * 
 * @return BrowserMobProxy
 * 
 */
public static BrowserMobProxy getProxy() {
	BrowserMobProxy proxy = null;
	long threadId = Thread.currentThread().getId();
	if (proxies.containsKey(threadId)) {
		proxy = proxies.get(threadId);
	} else {
		Assert.fail("There is not registered BrowserMobProxy for thread: " + threadId);
	}
	return proxy;
}
 

import net.lightbody.bmp.BrowserMobProxy; //导入依赖的package包/类
/**
 * register custom BrowserMobProxy Server
 * 
 * @param proxy
 *            custom BrowserMobProxy
 * 
 */
public static void registerProxy(BrowserMobProxy proxy) {
	long threadId = Thread.currentThread().getId();
	if (proxies.containsKey(threadId)) {
		LOGGER.warn("Existing proxy is detected and will be overriten");
		// No sense to stop as it is not supported
		proxies.remove(threadId);
	}
	
	LOGGER.info("Register custom proxy with thread: " + threadId);
	proxies.put(threadId, proxy);
}
 

import net.lightbody.bmp.BrowserMobProxy; //导入依赖的package包/类
@BeforeClass(alwaysRun = true)
public void beforeClass() {
	// do nothing
	R.CONFIG.put("browsermob_proxy", "true");
	R.CONFIG.put("browsermob_port", "0");
	R.CONFIG.put("proxy_set_to_system", "true");
	
	ProxyPool.setupBrowserMobProxy();
	SystemProxy.setupProxy();

	BrowserMobProxy proxy = ProxyPool.getProxy();
	proxy.addHeader(header, headerValue);
}
 

import net.lightbody.bmp.BrowserMobProxy; //导入依赖的package包/类
public Proxy createSeleniumProxy(BrowserMobProxy proxy) {
    return ClientUtil.createSeleniumProxy(proxy);
}
 

import net.lightbody.bmp.BrowserMobProxy; //导入依赖的package包/类
public BrowserMobProxy getProxy() {
    return proxy;
}
 

import net.lightbody.bmp.BrowserMobProxy; //导入依赖的package包/类
void startProxyServer(BrowserMobProxy browserMobProxy) throws UnknownHostException {
  int port = browserMobProxy.getPort();
  browserMobProxy.start(port);
  LOG.debug("proxy server started on port: '{}'", port);
}
 

import net.lightbody.bmp.BrowserMobProxy; //导入依赖的package包/类
/**
 * Creates a new object that holds  a driver, a proxy and a driver service that are connected together
 * @param driver the driver
 * @param proxy the proxy that the driver uses
 * @param driverService the driver service that is used for connecting to the virtual screen
 */
public ProxyDriverIntegrator(WebDriver driver, BrowserMobProxy proxy, DriverService driverService) {
    this.driver = driver;
    this.proxy = proxy;
    this.driverService = driverService;
}