Exemplo n.º 1
0
  @Post
  public Reply<?> newProxy(Request<String> request) {
    String systemProxyHost = System.getProperty("http.proxyHost");
    String systemProxyPort = System.getProperty("http.proxyPort");
    String httpProxy = request.param("httpProxy");
    Hashtable<String, String> options = new Hashtable<String, String>();

    // If the upstream proxy is specified via query params that should override any default system
    // level proxy.
    if (httpProxy != null) {
      options.put("httpProxy", httpProxy);
    } else if ((systemProxyHost != null) && (systemProxyPort != null)) {
      options.put("httpProxy", String.format("%s:%s", systemProxyHost, systemProxyPort));
    }

    String paramBindAddr = request.param("bindAddress");
    Integer paramPort =
        request.param("port") == null ? null : Integer.parseInt(request.param("port"));
    String useEccString = request.param("useEcc");
    boolean useEcc = Boolean.parseBoolean(useEccString);
    LOG.debug("POST proxy instance on bindAddress `{}` & port `{}`", paramBindAddr, paramPort);
    LegacyProxyServer proxy;
    try {
      proxy = proxyManager.create(options, paramPort, paramBindAddr, useEcc);
    } catch (ProxyExistsException ex) {
      return Reply.with(new ProxyDescriptor(ex.getPort())).status(455).as(Json.class);
    } catch (ProxyPortsExhaustedException ex) {
      return Reply.saying().status(456);
    } catch (Exception ex) {
      StringWriter s = new StringWriter();
      ex.printStackTrace(new PrintWriter(s));
      return Reply.with(s).as(Text.class).status(550);
    }
    return Reply.with(new ProxyDescriptor(proxy.getPort())).as(Json.class);
  }
Exemplo n.º 2
0
  @Put
  @At("/:port/har")
  public Reply<?> newHar(@Named("port") int port, Request<String> request) {
    LegacyProxyServer proxy = proxyManager.get(port);
    if (proxy == null) {
      return Reply.saying().notFound();
    }

    String initialPageRef = request.param("initialPageRef");
    String initialPageTitle = request.param("initialPageTitle");
    Har oldHar = proxy.newHar(initialPageRef, initialPageTitle);

    String captureHeaders = request.param("captureHeaders");
    String captureContent = request.param("captureContent");
    String captureBinaryContent = request.param("captureBinaryContent");
    proxy.setCaptureHeaders(Boolean.parseBoolean(captureHeaders));
    proxy.setCaptureContent(Boolean.parseBoolean(captureContent));
    proxy.setCaptureBinaryContent(Boolean.parseBoolean(captureBinaryContent));

    if (oldHar != null) {
      return Reply.with(oldHar).as(Json.class);
    } else {
      return Reply.saying().noContent();
    }
  }
Exemplo n.º 3
0
 @Get
 @At("/:port/limit")
 public Reply<?> getLimits(@Named("port") int port, Request<String> request) {
   LegacyProxyServer proxy = proxyManager.get(port);
   if (proxy == null) {
     return Reply.saying().notFound();
   }
   return Reply.with(new BandwidthLimitDescriptor(proxy.getStreamManager())).as(Json.class);
 }
Exemplo n.º 4
0
  @Get
  @At("/:port/whitelist")
  public Reply<?> getWhitelist(@Named("port") int port, Request<String> request) {
    LegacyProxyServer proxy = proxyManager.get(port);
    if (proxy == null) {
      return Reply.saying().notFound();
    }

    return Reply.with(proxy.getWhitelistUrls()).as(Json.class);
  }
Exemplo n.º 5
0
  @Delete
  @At("/:port/dns/cache")
  public Reply<?> clearDnsCache(@Named("port") int port) {
    LegacyProxyServer proxy = proxyManager.get(port);
    if (proxy == null) {
      return Reply.saying().notFound();
    }

    proxy.clearDNSCache();
    return Reply.saying().ok();
  }
Exemplo n.º 6
0
  @Delete
  @At("/:port/rewrite")
  public Reply<?> clearRewriteRules(@Named("port") int port, Request<String> request) {
    LegacyProxyServer proxy = proxyManager.get(port);
    if (proxy == null) {
      return Reply.saying().notFound();
    }

    proxy.clearRewriteRules();
    return Reply.saying().ok();
  }
Exemplo n.º 7
0
  @Delete
  @At("/:port")
  public Reply<?> delete(@Named("port") int port) {
    LegacyProxyServer proxy = proxyManager.get(port);
    if (proxy == null) {
      return Reply.saying().notFound();
    }

    proxyManager.delete(port);
    return Reply.saying().ok();
  }
Exemplo n.º 8
0
  @Get
  @At("/:port/har")
  public Reply<?> getHar(@Named("port") int port) {
    LegacyProxyServer proxy = proxyManager.get(port);
    if (proxy == null) {
      return Reply.saying().notFound();
    }

    Har har = proxy.getHar();

    return Reply.with(har).as(Json.class);
  }
Exemplo n.º 9
0
  @Put
  @At("/:port/retry")
  public Reply<?> retryCount(@Named("port") int port, Request<String> request) {
    LegacyProxyServer proxy = proxyManager.get(port);
    if (proxy == null) {
      return Reply.saying().notFound();
    }

    String count = request.param("retrycount");
    proxy.setRetryCount(Integer.parseInt(count));
    return Reply.saying().ok();
  }
Exemplo n.º 10
0
  @Put
  @At("/:port/rewrite")
  public Reply<?> rewriteUrl(@Named("port") int port, Request<String> request) {
    LegacyProxyServer proxy = proxyManager.get(port);
    if (proxy == null) {
      return Reply.saying().notFound();
    }

    String match = request.param("matchRegex");
    String replace = request.param("replace");
    proxy.rewriteUrl(match, replace);
    return Reply.saying().ok();
  }
Exemplo n.º 11
0
  @Post
  @At("/:port/auth/basic/:domain")
  public Reply<?> autoBasicAuth(
      @Named("port") int port, @Named("domain") String domain, Request<String> request) {
    LegacyProxyServer proxy = proxyManager.get(port);
    if (proxy == null) {
      return Reply.saying().notFound();
    }

    Map<String, String> credentials = request.read(HashMap.class).as(Json.class);
    proxy.autoBasicAuthorization(domain, credentials.get("username"), credentials.get("password"));

    return Reply.saying().ok();
  }
Exemplo n.º 12
0
  @Put
  @At("/:port/whitelist")
  public Reply<?> whitelist(@Named("port") int port, Request<String> request) {
    LegacyProxyServer proxy = proxyManager.get(port);
    if (proxy == null) {
      return Reply.saying().notFound();
    }

    String regex = request.param("regex");
    int responseCode = parseResponseCode(request.param("status"));
    proxy.whitelistRequests(regex.split(","), responseCode);

    return Reply.saying().ok();
  }
Exemplo n.º 13
0
  @Put
  @At("/:port/har/pageRef")
  public Reply<?> setPage(@Named("port") int port, Request<String> request) {
    LegacyProxyServer proxy = proxyManager.get(port);
    if (proxy == null) {
      return Reply.saying().notFound();
    }

    String pageRef = request.param("pageRef");
    String pageTitle = request.param("pageTitle");
    proxy.newPage(pageRef, pageTitle);

    return Reply.saying().ok();
  }
Exemplo n.º 14
0
  @Post
  @RequiresLogin
  public Reply<?> postStatus() {
    newStatus.setAuthor(userSession.get());
    newStatus.setPostTime(clock.now());

    if (newStatus.getText().length() > 140) {
      return Reply.saying().redirect("/"); // TODO: Error message
    }

    statusService.postStatus(newStatus);

    return Reply.saying().redirect("/");
  }
Exemplo n.º 15
0
  @Put
  @At("/:port/wait")
  public Reply<?> wait(@Named("port") int port, Request<String> request) {
    LegacyProxyServer proxy = proxyManager.get(port);
    if (proxy == null) {
      return Reply.saying().notFound();
    }

    String quietPeriodInMs = request.param("quietPeriodInMs");
    String timeoutInMs = request.param("timeoutInMs");
    proxy.waitForNetworkTrafficToStop(
        Integer.parseInt(quietPeriodInMs), Integer.parseInt(timeoutInMs));
    return Reply.saying().ok();
  }
Exemplo n.º 16
0
  @Put
  @At("/:port/blacklist")
  public Reply<?> blacklist(@Named("port") int port, Request<String> request) {
    LegacyProxyServer proxy = proxyManager.get(port);
    if (proxy == null) {
      return Reply.saying().notFound();
    }

    String blacklist = request.param("regex");
    int responseCode = parseResponseCode(request.param("status"));
    String method = request.param("method");
    proxy.blacklistRequests(blacklist, responseCode, method);

    return Reply.saying().ok();
  }
Exemplo n.º 17
0
  @Post
  @At("/:port/headers")
  public Reply<?> updateHeaders(@Named("port") int port, Request<String> request) {
    LegacyProxyServer proxy = proxyManager.get(port);
    if (proxy == null) {
      return Reply.saying().notFound();
    }

    Map<String, String> headers = request.read(Map.class).as(Json.class);
    for (Map.Entry<String, String> entry : headers.entrySet()) {
      String key = entry.getKey();
      String value = entry.getValue();
      proxy.addHeader(key, value);
    }
    return Reply.saying().ok();
  }
Exemplo n.º 18
0
 @Get
 public Reply<?> getProxies() {
   Collection<ProxyDescriptor> proxyList = new ArrayList<ProxyDescriptor>();
   for (LegacyProxyServer proxy : proxyManager.get()) {
     proxyList.add(new ProxyDescriptor(proxy.getPort()));
   }
   return Reply.with(new ProxyListDescriptor(proxyList)).as(Json.class);
 }
Exemplo n.º 19
0
  @Post
  @At("/:port/hosts")
  public Reply<?> remapHosts(@Named("port") int port, Request<String> request) {
    LegacyProxyServer proxy = proxyManager.get(port);
    if (proxy == null) {
      return Reply.saying().notFound();
    }

    @SuppressWarnings("unchecked")
    Map<String, String> headers = request.read(Map.class).as(Json.class);

    for (Map.Entry<String, String> entry : headers.entrySet()) {
      String key = entry.getKey();
      String value = entry.getValue();
      proxy.remapHost(key, value);
      proxy.setDNSCacheTimeout(0);
      proxy.clearDNSCache();
    }

    return Reply.saying().ok();
  }
Exemplo n.º 20
0
  @Post
  @At("/:port/interceptor/request")
  public Reply<?> addRequestInterceptor(@Named("port") int port, Request<String> request)
      throws IOException, ScriptException {
    LegacyProxyServer proxy = proxyManager.get(port);
    if (proxy == null) {
      return Reply.saying().notFound();
    }

    if (!(proxy instanceof ProxyServer)) {
      return Reply.saying().badRequest();
    }

    ByteArrayOutputStream baos = new ByteArrayOutputStream();
    request.readTo(baos);

    ScriptEngineManager mgr = new ScriptEngineManager();
    final ScriptEngine engine = mgr.getEngineByName("JavaScript");
    Compilable compilable = (Compilable) engine;
    final CompiledScript script = compilable.compile(baos.toString());

    proxy.addRequestInterceptor(
        new RequestInterceptor() {
          @Override
          public void process(BrowserMobHttpRequest request, Har har) {
            Bindings bindings = engine.createBindings();
            bindings.put("request", request);
            bindings.put("har", har);
            bindings.put("log", LOG);
            try {
              script.eval(bindings);
            } catch (ScriptException e) {
              LOG.error("Could not execute JS-based response interceptor", e);
            }
          }
        });

    return Reply.saying().ok();
  }
Exemplo n.º 21
0
  @Put
  @At("/:port/timeout")
  public Reply<?> timeout(@Named("port") int port, Request<String> request) {
    LegacyProxyServer proxy = proxyManager.get(port);
    if (proxy == null) {
      return Reply.saying().notFound();
    }

    String requestTimeout = request.param("requestTimeout");
    if (requestTimeout != null) {
      try {
        proxy.setRequestTimeout(Integer.parseInt(requestTimeout));
      } catch (NumberFormatException e) {
      }
    }
    String readTimeout = request.param("readTimeout");
    if (readTimeout != null) {
      try {
        proxy.setSocketOperationTimeout(Integer.parseInt(readTimeout));
      } catch (NumberFormatException e) {
      }
    }
    String connectionTimeout = request.param("connectionTimeout");
    if (connectionTimeout != null) {
      try {
        proxy.setConnectionTimeout(Integer.parseInt(connectionTimeout));
      } catch (NumberFormatException e) {
      }
    }
    String dnsCacheTimeout = request.param("dnsCacheTimeout");
    if (dnsCacheTimeout != null) {
      try {
        proxy.setDNSCacheTimeout(Integer.parseInt(dnsCacheTimeout));
      } catch (NumberFormatException e) {
      }
    }
    return Reply.saying().ok();
  }
Exemplo n.º 22
0
  @Get
  public Reply<String> get() {
    String runId = request.getParameter("runId");
    if (StringUtils.isBlank(runId)) {
      return Reply.with(StringUtils.EMPTY);
    }

    LogBus logBus = LogBus.get(runId);
    synchronized (logBus.getLock()) {
      if (logBus.size() > 0) {
        log.debug("Found logs from cache, size={}", logBus.size());
        StringBuffer logs = new StringBuffer();
        for (String log : logBus) {
          logs.append(log).append("\n");
        }

        // Clean log cache after get all logs
        logBus.clear();
        return Reply.with(logs.toString());
      }
    }
    return Reply.with(StringUtils.EMPTY);
  }
Exemplo n.º 23
0
  @Post
  @At("/:port/filter/response")
  public Reply<?> addResponseFilter(@Named("port") int port, Request<String> request)
      throws IOException, ScriptException {
    LegacyProxyServer legacyProxy = proxyManager.get(port);
    if (legacyProxy == null) {
      return Reply.saying().notFound();
    }

    if (!(legacyProxy instanceof BrowserMobProxyServer)) {
      return Reply.saying().badRequest();
    }

    BrowserMobProxy proxy = (BrowserMobProxy) legacyProxy;

    JavascriptRequestResponseFilter requestResponseFilter = new JavascriptRequestResponseFilter();

    String script = getEntityBodyFromRequest(request);
    requestResponseFilter.setResponseFilterScript(script);

    proxy.addResponseFilter(requestResponseFilter);

    return Reply.saying().ok();
  }
 public Reply<?> reply() {
   FormResponse response = get();
   return Reply.with(response).as(Json.class);
 }
Exemplo n.º 25
0
  @Put
  @At("/:port/limit")
  public Reply<?> limit(@Named("port") int port, Request<String> request) {
    LegacyProxyServer proxy = proxyManager.get(port);
    if (proxy == null) {
      return Reply.saying().notFound();
    }

    StreamManager streamManager = proxy.getStreamManager();
    String upstreamKbps = request.param("upstreamKbps");
    if (upstreamKbps != null) {
      try {
        streamManager.setUpstreamKbps(Integer.parseInt(upstreamKbps));
        streamManager.enable();
      } catch (NumberFormatException e) {
      }
    }

    String upstreamBps = request.param("upstreamBps");
    if (upstreamBps != null) {
      try {
        ((BrowserMobProxy) proxy).setWriteBandwidthLimit(Integer.parseInt(upstreamBps));
      } catch (NumberFormatException e) {
      }
    }

    String downstreamKbps = request.param("downstreamKbps");
    if (downstreamKbps != null) {
      try {
        streamManager.setDownstreamKbps(Integer.parseInt(downstreamKbps));
        streamManager.enable();
      } catch (NumberFormatException e) {
      }
    }

    String downstreamBps = request.param("downstreamBps");
    if (downstreamBps != null) {
      try {
        ((BrowserMobProxy) proxy).setReadBandwidthLimit(Integer.parseInt(downstreamBps));
      } catch (NumberFormatException e) {
      }
    }

    String upstreamMaxKB = request.param("upstreamMaxKB");
    if (upstreamMaxKB != null) {
      try {
        streamManager.setUpstreamMaxKB(Integer.parseInt(upstreamMaxKB));
        streamManager.enable();
      } catch (NumberFormatException e) {
      }
    }
    String downstreamMaxKB = request.param("downstreamMaxKB");
    if (downstreamMaxKB != null) {
      try {
        streamManager.setDownstreamMaxKB(Integer.parseInt(downstreamMaxKB));
        streamManager.enable();
      } catch (NumberFormatException e) {
      }
    }
    String latency = request.param("latency");
    if (latency != null) {
      try {
        streamManager.setLatency(Integer.parseInt(latency));
        streamManager.enable();
      } catch (NumberFormatException e) {
      }
    }
    String payloadPercentage = request.param("payloadPercentage");
    if (payloadPercentage != null) {
      try {
        streamManager.setPayloadPercentage(Integer.parseInt(payloadPercentage));
      } catch (NumberFormatException e) {
      }
    }
    String maxBitsPerSecond = request.param("maxBitsPerSecond");
    if (maxBitsPerSecond != null) {
      try {
        streamManager.setMaxBitsPerSecondThreshold(Integer.parseInt(maxBitsPerSecond));
      } catch (NumberFormatException e) {
      }
    }
    String enable = request.param("enable");
    if (enable != null) {
      if (Boolean.parseBoolean(enable)) {
        streamManager.enable();
      } else {
        streamManager.disable();
      }
    }
    return Reply.saying().ok();
  }