mproxy-java.1.5.source-code.MitmproxyServer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mitmproxy-java Show documentation

A bridge between Python's mitmproxy and Java programs. Built on top of mitmproxy-node

There is a newer version: 2.0.2

import com.fasterxml.jackson.core.JsonProcessingException;
import org.java_websocket.WebSocket;
import org.java_websocket.handshake.ClientHandshake;
import org.java_websocket.server.WebSocketServer;

import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.util.function.Function;

public class MitmproxyServer extends WebSocketServer {

    private Function interceptor;

    public MitmproxyServer(InetSocketAddress address, Function interceptor) {
        super(address);
        this.interceptor = interceptor;
    }

    @Override
    public void onOpen(WebSocket conn, ClientHandshake handshake) {
        System.out.println("new connection to websocket server" + conn.getRemoteSocketAddress());
    }

    @Override
    public void onClose(WebSocket conn, int code, String reason, boolean remote) {
        System.out.println("closed " + conn.getRemoteSocketAddress() + " with exit code " + code + " additional info: " + reason);
    }

    @Override
    public void onMessage(WebSocket conn, String message) {
        System.out.println("received message from "	+ conn.getRemoteSocketAddress() + ": " + message);
    }

    @Override
    public void onMessage( WebSocket conn, ByteBuffer message ) {
        InterceptedMessage intercepted = null;
        InterceptedMessage modifiedMessage = null;

        try {
            intercepted = new InterceptedMessage(message);
        } catch (IOException e) {
            System.out.println("Could not parse message");
            e.printStackTrace();
        }

        modifiedMessage = interceptor.apply(intercepted);

        // if the supplied interceptor function does not return a message, assume no changes were intended and just
        // complete the request
        if (modifiedMessage == null) {
            modifiedMessage = intercepted;
        }

        try {
            conn.send(modifiedMessage.serializedResponseToMitmproxy());
        } catch (JsonProcessingException e) {
            System.out.println("Could not encode response to mitmproxy");
            e.printStackTrace();
        }
    }

    @Override
    public void onError(WebSocket conn, Exception ex) {
        System.err.println("an error occured on connection " + conn.getRemoteSocketAddress()  + ":" + ex);
    }

    @Override
    public void onStart() {
        System.out.println("websocket server started successfully");
    }
}