FlowContext.java

package org.littleshoot.proxy;

import java.net.InetSocketAddress;

import javax.net.ssl.SSLEngine;
import javax.net.ssl.SSLSession;

import org.littleshoot.proxy.impl.ClientToProxyConnection;

import io.netty.channel.ChannelHandlerContext;

/**
 * <p>
 * Encapsulates contextual information for flow information that's being
 * reported to a {@link ActivityTracker}.
 * </p>
 */
public class FlowContext {
    private final InetSocketAddress clientAddress;
    private final SSLSession clientSslSession;
    private final ChannelHandlerContext ctx;

    public FlowContext(ClientToProxyConnection clientConnection) {
        super();
        this.ctx = clientConnection.getContext();
        this.clientAddress = clientConnection.getClientAddress();
        SSLEngine sslEngine = clientConnection.getSslEngine();
        this.clientSslSession = sslEngine != null ? sslEngine.getSession()
                : null;
    }

    /**
     * The client to proxy channel context.
     *
     * @return
     */
    public ChannelHandlerContext getClientToProxyContext() {
        return ctx;
    }

    /**
     * The address of the client.
     * 
     * @return
     */
    public InetSocketAddress getClientAddress() {
        return clientAddress;
    }

    /**
     * If using SSL, this returns the {@link SSLSession} on the client
     * connection.
     * 
     * @return
     */
    public SSLSession getClientSslSession() {
        return clientSslSession;
    }

}