java http代理

初學java就寫了個http代理練練手，把以前C語言寫的移植了下，不得不說Java寫起來是要比C語言簡單的多。

• LocalServer類 創(chuàng)建本地監(jiān)聽 接收客戶端請求

package cn.lbxx;

import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;

public class LocalServer{
    private static int   LPort   ;  // 監(jiān)聽端口
    private ServerSocket loSocket;  // 本地套接字
    public LocalServer() throws IOException {
        loSocket = new ServerSocket(LPort);
    }
    public void start()
    {
        try {
            System.out.println("run on " + LPort);
            while(true)
            {
                Socket ct = loSocket.accept();
                new HandleRequest(ct).start(); // 新建線程處理客戶端請求
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
    public static void main(String[] args)
    {
        LPort = 8853;
        try {
            new LocalServer().start();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

• HandleRequest類 處理客戶端請求

package cn.lbxx;

import java.net.ConnectException;
import java.net.Socket;
import java.net.SocketException;
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;

public class HandleRequest extends Thread {
    private static String KHost = "Host:";
    
    private Socket       clSocket   ;
    private Socket       seSocket   ;
    private Scanner      cdata      ; 
    private String       remotehost ;
    private int          remoteport ;
    private boolean      https      ;
    private List<String> bufflist   ;
    
    public HandleRequest(Socket c){
        this.clSocket = c;
        this.bufflist = new ArrayList<String>();
    }
    
    public void run()
    {
        try {
            cdata = new Scanner(clSocket.getInputStream());
            int beginIndex = KHost.length() + 1;
            String line;
            // 讀取客戶端的請求頭
            while(cdata.hasNextLine() && (line = cdata.nextLine()).length() != 0)
            {
                if(line.length() > 5)
                {
                    if(line.substring(0, KHost.length()).equals(KHost))
                    {
                        int hend;
                        if((hend = line.indexOf(':', beginIndex)) != -1)
                        {
                            remotehost = line.substring(beginIndex, hend);
                            remoteport = Integer.parseInt(line.substring(hend + 1));
                        } else {
                            remotehost = line.substring(beginIndex);
                            remoteport = 80;
                        }
                    }
                    // 判斷是否為https請求
                    if(line.substring(0, line.indexOf(' ')).equals("CONNECT")){
                        https = true;
                    }
                    
                    if(line.matches("Proxy-Connection(.*)")){
                        line = "Connection: keep-alive";
                    }
                }
                bufflist.add(line);
            }
            
            if(remotehost != null)
            {
                System.out.println(remotehost + " -> " + remoteport + "  " + https);
                seSocket = new Socket(remotehost, remoteport); // 連接到遠程主機
                if (https) { // 客戶端與遠程主機間數(shù)據(jù)轉發(fā)
                    List<String> list = new ArrayList<>();
                    list.add("HTTP/1.1 200 Connection Established");
                    
                    new ForwardData(list, seSocket, clSocket).start();
                    new ForwardData(null, clSocket, seSocket).start();
                } else {
                    toUri(bufflist);
                    
                    new ForwardData(bufflist, clSocket, seSocket).start();
                    new ForwardData(null, seSocket, clSocket).start();
                }
            }
        } catch (ConnectException c) {
            System.err.println("鏈接超時");
        } catch (SocketException se) {
            System.err.println("無法連接-> " + remotehost + ":" + remoteport);
        } catch (Exception e) {
            System.err.println("發(fā)生錯誤" + e);
        }
    }
    
    // 將header轉換為path形式
    private void toUri(List<String> buff)
    {
        for (int i = 0; i < buff.size(); i++)
        {
            String line = buff.get(i);
            String head = line.substring(0, line.indexOf(' '));
            int    hlen = head.length() + 1;
            if(line.substring(hlen, hlen + 7).equals("http://"))
            {
                String uri = line.substring(line.indexOf('/', hlen + 7));
                buff.set(i, head + " " + uri);
                break;
            }
        }
    }
}

• ForwardData類 數(shù)據(jù)轉發(fā)

package cn.lbxx

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.Socket;
import java.util.List;

public class ForwardData extends Thread {
    private List<String> buff        ;
    private Socket       source      ;
    private Socket       destination ;
    
    public ForwardData(List<String> buff, Socket sou, Socket des) {
        this.buff        = buff;
        this.source      = sou ;
        this.destination = des ;
    }
    
    @Override
    public void run() {
        try {
            OutputStream outStream = destination.getOutputStream();
            InputStream  inStream  = source.getInputStream();

            if(buff != null && buff.size() > 0){
                for(String str : buff)
                {
                    outStream.write((str + "\r\n").getBytes());
                    outStream.flush();
                }
                outStream.write("\r\n".getBytes());
                outStream.flush();
            }

            byte[] bs = new byte[4096];
            int len;
            while ((len = inStream.read(bs)) != -1) {
                outStream.write(bs, 0, len);
                outStream.flush();
            }
            
            outStream.close();
            inStream.close();
        } catch (IOException ie) {
            System.err.println("http 數(shù)據(jù)轉發(fā)異常 " + ie);
        }
    }
}

• 測試

smallproxy.png

很簡單的代理例子，問題有不少經過測試多數(shù)網站都能代理，性能速度方面就沒去測試，多線程應該還不錯吧，有空繼續(xù)優(yōu)化吧。

源碼地址: https://github.com/lkl0304/SmallProxy