org.littleshoot.proxy.HttpFiltersSource Java Examples

@Test
void testClearRequestFilters()
{
    configureProxy();
    BrowserUpProxyServer browserMobProxyServer = mock(BrowserUpProxyServer.class);
    ResponseFilter responseFilter = mock(ResponseFilter.class);
    RequestFilter requestFilter = mock(RequestFilter.class);
    ResponseFilterAdapter.FilterSource fsResponse = new ResponseFilterAdapter.FilterSource(responseFilter);
    RequestFilterAdapter.FilterSource fsRequest = new RequestFilterAdapter.FilterSource(requestFilter);
    when(proxyServerFactory.createProxyServer()).thenReturn(browserMobProxyServer);
    List<HttpFiltersSource> toRemove = new ArrayList<>();
    toRemove.add(fsResponse);
    toRemove.add(fsRequest);
    when(browserMobProxyServer.getFilterFactories()).thenReturn(toRemove).thenReturn(toRemove);
    proxy.start();
    proxy.clearRequestFilters();
    assertTrue(toRemove.size() == 1 && toRemove.contains(fsResponse));
}
 

public BrowserMobHttpFilterChain(BrowserMobProxyServer proxyServer, HttpRequest originalRequest, ChannelHandlerContext ctx) {
    super(originalRequest, ctx);

    this.proxyServer = proxyServer;

    if (proxyServer.getFilterFactories() != null) {
        filters = new ArrayList<>(proxyServer.getFilterFactories().size());

        // instantiate all HttpFilters using the proxy's filter factories
        for (HttpFiltersSource filterFactory : proxyServer.getFilterFactories()) {
            HttpFilters filter = filterFactory.filterRequest(originalRequest, ctx);
            // allow filter factories to avoid adding a filter on a per-request basis by returning a null
            // HttpFilters instance
            if (filter != null) {
                filters.add(filter);
            }
        }
    } else {
        filters = Collections.emptyList();
    }
}
 

public BrowserMobHttpFilterChain(BrowserMobProxyServer proxyServer, HttpRequest originalRequest, ChannelHandlerContext ctx) {
    super(originalRequest, ctx);

    this.proxyServer = proxyServer;

    if (proxyServer.getFilterFactories() != null) {
        filters = new ArrayList<>(proxyServer.getFilterFactories().size());

        // instantiate all HttpFilters using the proxy's filter factories
        for (HttpFiltersSource filterFactory : proxyServer.getFilterFactories()) {
            HttpFilters filter = filterFactory.filterRequest(originalRequest, ctx);
            // allow filter factories to avoid adding a filter on a per-request basis by returning a null
            // HttpFilters instance
            if (filter != null) {
                filters.add(filter);
            }
        }
    } else {
        filters = Collections.emptyList();
    }
}
 

public BrowserMobHttpFilterChain(BrowserMobProxyServer proxyServer, HttpRequest originalRequest, ChannelHandlerContext ctx) {
    super(originalRequest, ctx);

    this.proxyServer = proxyServer;

    if (proxyServer.getFilterFactories() != null) {
        filters = new ArrayList<>(proxyServer.getFilterFactories().size());

        // instantiate all HttpFilters using the proxy's filter factories
        for (HttpFiltersSource filterFactory : proxyServer.getFilterFactories()) {
            HttpFilters filter = filterFactory.filterRequest(originalRequest, ctx);
            // allow filter factories to avoid adding a filter on a per-request basis by returning a null
            // HttpFilters instance
            if (filter != null) {
                filters.add(filter);
            }
        }
    } else {
        filters = Collections.emptyList();
    }
}
 

@Override
public void clearRequestFilters()
{
    List<HttpFiltersSource> toRemove = new ArrayList<>();
    for (HttpFiltersSource source : ((BrowserUpProxyServer) proxyServer).getFilterFactories())
    {
        if (source instanceof FilterSource)
        {
            toRemove.add(source);
        }
    }
    ((BrowserUpProxyServer) proxyServer).getFilterFactories().removeAll(toRemove);
}
 

private int getMaximumRequestBufferSize() {
    int maxBufferSize = 0;
    for (HttpFiltersSource source : filterFactories) {
        int requestBufferSize = source.getMaximumRequestBufferSizeInBytes();
        if (requestBufferSize > maxBufferSize) {
            maxBufferSize = requestBufferSize;
        }
    }

    return maxBufferSize;
}
 

private int getMaximumResponseBufferSize() {
    int maxBufferSize = 0;
    for (HttpFiltersSource source : filterFactories) {
        int requestBufferSize = source.getMaximumResponseBufferSizeInBytes();
        if (requestBufferSize > maxBufferSize) {
            maxBufferSize = requestBufferSize;
        }
    }

    return maxBufferSize;
}
 

private int getMaximumResponseBufferSize() {
    int maxBufferSize = 0;
    for (HttpFiltersSource source : filterFactories) {
        int requestBufferSize = source.getMaximumResponseBufferSizeInBytes();
        if (requestBufferSize > maxBufferSize) {
            maxBufferSize = requestBufferSize;
        }
    }

    return maxBufferSize;
}
 

private int getMaximumRequestBufferSize() {
    int maxBufferSize = 0;
    for (HttpFiltersSource source : filterFactories) {
        int requestBufferSize = source.getMaximumRequestBufferSizeInBytes();
        if (requestBufferSize > maxBufferSize) {
            maxBufferSize = requestBufferSize;
        }
    }

    return maxBufferSize;
}
 

private int getMaximumResponseBufferSize() {
    int maxBufferSize = 0;
    for (HttpFiltersSource source : filterFactories) {
        int requestBufferSize = source.getMaximumResponseBufferSizeInBytes();
        if (requestBufferSize > maxBufferSize) {
            maxBufferSize = requestBufferSize;
        }
    }

    return maxBufferSize;
}
 

private int getMaximumRequestBufferSize() {
    int maxBufferSize = 0;
    for (HttpFiltersSource source : filterFactories) {
        int requestBufferSize = source.getMaximumRequestBufferSizeInBytes();
        if (requestBufferSize > maxBufferSize) {
            maxBufferSize = requestBufferSize;
        }
    }

    return maxBufferSize;
}
 

private int getMaximumResponseBufferSize() {
    int maxBufferSize = 0;
    for (HttpFiltersSource source : filterFactories) {
        int requestBufferSize = source.getMaximumResponseBufferSizeInBytes();
        if (requestBufferSize > maxBufferSize) {
            maxBufferSize = requestBufferSize;
        }
    }

    return maxBufferSize;
}
 

private int getMaximumRequestBufferSize() {
    int maxBufferSize = 0;
    for (HttpFiltersSource source : filterFactories) {
        int requestBufferSize = source.getMaximumRequestBufferSizeInBytes();
        if (requestBufferSize > maxBufferSize) {
            maxBufferSize = requestBufferSize;
        }
    }

    return maxBufferSize;
}
 

@Override
public HttpProxyServerBootstrap withFiltersSource(
        HttpFiltersSource filtersSource) {
    this.filtersSource = filtersSource;
    return this;
}
 

private DefaultHttpProxyServerBootstrap(
        ServerGroup serverGroup,
        TransportProtocol transportProtocol,
        InetSocketAddress requestedAddress,
        SslEngineSource sslEngineSource,
        boolean authenticateSslClients,
        ProxyAuthenticator proxyAuthenticator,
        ChainedProxyManager chainProxyManager,
        MitmManager mitmManager,
        HttpFiltersSource filtersSource,
        boolean transparent, int idleConnectionTimeout,
        Collection<ActivityTracker> activityTrackers,
        int connectTimeout, HostResolver serverResolver,
        long readThrottleBytesPerSecond,
        long writeThrottleBytesPerSecond,
        InetSocketAddress localAddress,
        String proxyAlias,
        int maxInitialLineLength,
        int maxHeaderSize,
        int maxChunkSize,
        boolean allowRequestToOriginServer) {
    this.serverGroup = serverGroup;
    this.transportProtocol = transportProtocol;
    this.requestedAddress = requestedAddress;
    this.port = requestedAddress.getPort();
    this.sslEngineSource = sslEngineSource;
    this.authenticateSslClients = authenticateSslClients;
    this.proxyAuthenticator = proxyAuthenticator;
    this.chainProxyManager = chainProxyManager;
    this.mitmManager = mitmManager;
    this.filtersSource = filtersSource;
    this.transparent = transparent;
    this.idleConnectionTimeout = idleConnectionTimeout;
    if (activityTrackers != null) {
        this.activityTrackers.addAll(activityTrackers);
    }
    this.connectTimeout = connectTimeout;
    this.serverResolver = serverResolver;
    this.readThrottleBytesPerSecond = readThrottleBytesPerSecond;
    this.writeThrottleBytesPerSecond = writeThrottleBytesPerSecond;
    this.localAddress = localAddress;
    this.proxyAlias = proxyAlias;
    this.maxInitialLineLength = maxInitialLineLength;
    this.maxHeaderSize = maxHeaderSize;
    this.maxChunkSize = maxChunkSize;
    this.allowRequestToOriginServer = allowRequestToOriginServer;
}
 

@Override
public void addFirstHttpFilterFactory(HttpFiltersSource filterFactory) {
    filterFactories.add(0, filterFactory);
}
 

@Override
public void addLastHttpFilterFactory(HttpFiltersSource filterFactory) {
    filterFactories.add(filterFactory);
}
 

public void addHttpFilterFactory(HttpFiltersSource filterFactory) {
    filterFactories.add(filterFactory);
}
 

public List<HttpFiltersSource> getFilterFactories() {
    return filterFactories;
}
 

@Override
public void addFirstHttpFilterFactory(HttpFiltersSource filterFactory) {
    filterFactories.add(0, filterFactory);
}
 

@Override
public void addFirstHttpFilterFactory(HttpFiltersSource filterFactory) {
    filterFactories.add(0, filterFactory);
}
 

@Override
public void addLastHttpFilterFactory(HttpFiltersSource filterFactory) {
    filterFactories.add(filterFactory);
}
 

public void addHttpFilterFactory(HttpFiltersSource filterFactory) {
    filterFactories.add(filterFactory);
}
 

public List<HttpFiltersSource> getFilterFactories() {
    return filterFactories;
}
 

public void addHttpFilterFactory(HttpFiltersSource filterFactory) {
    filterFactories.add(filterFactory);
}
 

public List<HttpFiltersSource> getFilterFactories() {
    return filterFactories;
}
 

@Override
public void addFirstHttpFilterFactory(HttpFiltersSource filterFactory) {
    filterFactories.add(0, filterFactory);
}
 

@Override
public void addLastHttpFilterFactory(HttpFiltersSource filterFactory) {
    filterFactories.add(filterFactory);
}
 

public void addHttpFilterFactory(HttpFiltersSource filterFactory) {
    filterFactories.add(filterFactory);
}
 

public List<HttpFiltersSource> getFilterFactories() {
    return filterFactories;
}