Java實現端口映射/轉發

20190630 14：57更新

添加了一個功能 檢查當前連接狀態 同時發現清除超時連接忘了加定時器 現在清除超時連接可以正常工作了

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.ServerSocket;
import java.net.Socket;
import java.text.SimpleDateFormat;
import java.util.*;

public class SimpleForward implements Runnable {
    // 服務器
    private ServerSocket server;
    // 監聽本地端口
    private int localPort = 1080;
    // 目標主機地址
    private String remoteHostAddr = "0.0.0.0";
    // 目標主機端口
    private int remoteHostPort = 8388;
    // 設置超時時間 30s
    private static int TIMEOUT = 30;
    // 客戶端列表 用於刪除失效連接和超時連接
    private static HashMap<Socket, Date> clientList = new HashMap<>();

    public static void main(String[] args) throws IOException {
//        new SimpleForward();
        new SimpleForward(1234, "127.0.0.1", 1080);
    }

    public SimpleForward() throws IOException {
        run();
    }

    public SimpleForward(int localPort, String remoteHostAddr, int remoteHostPort) throws IOException {
        this.localPort = localPort;
        this.remoteHostAddr = remoteHostAddr;
        this.remoteHostPort = remoteHostPort;
        run();
    }

    @Override
    public void run() {
        try {
            this.server = new ServerSocket(this.localPort);
            System.out.println("服務器開啟成功");
            System.out.println("監聽端口 : " + this.localPort);
        } catch (IOException e) {
            System.out.println("服務器開啟失敗");
            System.out.println(e.getMessage());
            System.out.println("退出運行");
            return;
        }
        // 自動清除失效連接和超時連接
        new Thread(new Terminal()).start();
        new Thread(new AutoDestroy()).start();
        while (true) {
            Socket socket = null;
            Socket remoteHost = null;
            try {
                socket = server.accept();
                // 接收到請求就把socket扔進map,value為刷新時間
                clientList.put(socket, new Date());
                String address = socket.getRemoteSocketAddress().toString();
                System.out.println("新連接 ： " + address);
                // 建立與目標主機的連接
                remoteHost = new Socket(this.remoteHostAddr, this.remoteHostPort);
                System.out.println("連接地址 : " + this.remoteHostAddr + ":" + this.remoteHostPort);
                // 端口轉發
                new Thread(new Switch(socket, remoteHost, remoteHost.getInputStream(), socket.getOutputStream())).start();
                new Thread(new Switch(socket, remoteHost, socket.getInputStream(), remoteHost.getOutputStream())).start();
            } catch (IOException e) {
                System.out.println("連接異常");
                System.out.println(e.getMessage());
                close(socket);
                close(remoteHost);
            }
        }
    }

    private void close(Socket socket) {
        try {
            if (socket != null) {
                socket.close();
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    // 用於端口轉發
    private class Switch implements Runnable {
        private Socket host;
        private Socket remoteHost;
        private InputStream in;
        private OutputStream out;

        Switch(Socket host, Socket remoteHost, InputStream in, OutputStream out) {
            this.host = host;
            this.remoteHost = remoteHost;
            this.in = in;
            this.out = out;
        }

        @Override
        public void run() {
            int length = 0;
            byte[] buffer = new byte[1024];
            try {
                while (!host.isClosed() && (length = in.read(buffer)) > -1) {
                    clientList.put(host, new Date());
                    out.write(buffer, 0, length);
                }
            } catch (IOException e) {
                System.out.println("連接關閉");
            } finally {
                close(host);
                close(remoteHost);
            }
        }
    }

    // 用於清除失效連接和超時連接
    private class AutoDestroy implements Runnable {

        @Override
        public void run() {
            Timer timer = new Timer();
            timer.schedule(new TimerTask() {
                @Override
                public void run() {
                    List<Socket> list = new LinkedList<>();
                    System.out.println("開始掃描失效與超時連接");
                    Date start = new Date();
                    for (Socket socket : clientList.keySet()) {
                        Date lastTime = clientList.get(socket);
                        long time = new Date().getTime() - lastTime.getTime();
                        if (socket.isClosed() || time / 1000 >= TIMEOUT) {
                            list.add(socket);
                        }
                    }
                    System.out.println("找到" + list.size() + "個,用時 : " + (new Date().getTime() - start.getTime()) + "毫秒");
                    System.out.println("開始清除失效與超時連接");
                    for (Socket socket : list) {
                        try {
                            clientList.remove(socket);
                            socket.close();
                        } catch (IOException e) {
                            e.printStackTrace();
                        }
                    }
                    System.out.println("當前連接數 : " + clientList.size());
                }
            }, 30 * 1000, 30 * 1000);
        }
    }

    private class Terminal implements Runnable {
        private ReaderUtil reader = new ReaderUtil();
        private String format = "yyyy-MM-dd HH:mm:ss";
        private SimpleDateFormat dateFormat = new SimpleDateFormat(format);

        @Override
        public void run() {
            while (!server.isClosed()) {
                System.out.print("請輸入命令 : ");
                String cmd = reader.readKB();
                handler(cmd);
            }
        }

        private void handler(String cmd) {
            switch (cmd) {
                case "status":
                    System.out.println("當前時間 : " + dateFormat.format(new Date()));
                    System.out.println("總連接數 : " + clientList.size());
                    for (Socket socket : clientList.keySet()) {
                        long time = new Date().getTime() - clientList.get(socket).getTime();
                        System.out.println("<" + socket.getRemoteSocketAddress().toString() + "> " + time / 1000);
                    }
                    break;
            }
        }
    }
}

 ReaderUtil這個東西是我自己寫的從鍵盤讀一行的工具類 沒啥東西 你們需要自己寫