com.github.monkeywie.proxyee.util.ProtoUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of proxyee Show documentation
HTTP proxy server,support HTTPS and websocket.MITM impl,intercept and tamper HTTPS traffic.
There is a newer version: 1.7.6
package com.github.monkeywie.proxyee.util;

import io.netty.handler.codec.http.HttpHeaderNames;
import io.netty.handler.codec.http.HttpRequest;

import java.io.Serializable;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.Objects;

public class ProtoUtil {

    /*
    代理服务器需要处理两种握手类型，一种是非CONNECT的http报文代理，另外一种是CONNECT的TCP报文原始转发
    示例：
        GET http://www.google.com/ HTTP/1.1
        CONNECT www.google.com:443 HTTP/1.1
        CONNECT echo.websocket.org:443 HTTP/1.1
        CONNECT echo.websocket.org:80 HTTP/1.1
    当客户端请求协议为TLS(https、wss)、WebSocket(ws)的时候，都会发起CONNECT请求进行原始转发，
    所以在握手的时候是无法区分客户端原始请求是否为TLS。
     */
    public static RequestProto getRequestProto(HttpRequest httpRequest) {
        RequestProto requestProto = new RequestProto();
        String uri = httpRequest.uri().toLowerCase();

        if (!uri.startsWith("http://")) {
            uri = "http://" + uri;
        }
        URL url;
        try {
            url = new URL(uri);
        } catch (MalformedURLException e) {
            return null;
        }

        requestProto.setHost(url.getHost());
        requestProto.setPort(url.getPort() != -1 ? url.getPort() : url.getDefaultPort());
        requestProto.setProxy(httpRequest.headers().contains(HttpHeaderNames.PROXY_CONNECTION));
        return requestProto;
    }

    public static class RequestProto implements Serializable {

        private static final long serialVersionUID = -6471051659605127698L;
        /**
         * 请求是否来源于http代理，用于区分是通过代理服务访问的还是直接通过http访问的代理服务器
         */
        private boolean proxy;
        private String host;
        private int port;
        private boolean ssl;

        public RequestProto() {
        }

        public RequestProto(String host, int port, boolean ssl) {
            this.host = host;
            this.port = port;
            this.ssl = ssl;
        }

        public String getHost() {
            return host;
        }

        public void setHost(String host) {
            this.host = host;
        }

        public int getPort() {
            return port;
        }

        public void setPort(int port) {
            this.port = port;
        }

        public boolean getSsl() {
            return ssl;
        }

        public void setSsl(boolean ssl) {
            this.ssl = ssl;
        }

        public boolean getProxy() {
            return proxy;
        }

        public void setProxy(boolean proxy) {
            this.proxy = proxy;
        }

        @Override
        public boolean equals(Object o) {
            if (this == o) return true;
            if (o == null || getClass() != o.getClass()) return false;
            RequestProto that = (RequestProto) o;
            return port == that.port &&
                    ssl == that.ssl &&
                    host.equals(that.host);
        }

        @Override
        public int hashCode() {
            return Objects.hash(host, port, ssl);
        }

        public RequestProto copy() {
            RequestProto requestProto = new RequestProto();
            requestProto.setProxy(proxy);
            requestProto.setHost(host);
            requestProto.setPort(port);
            requestProto.setSsl(ssl);
            return requestProto;
        }
    }
}