java 实现socks代理，包含sock4 sock5代理

leehunter

使用方法，直接编译后
USing port openSock4 openSock5 user pwd
端口号
是否开通sock4
是否开通sock5
user 不为空时表示 sock5需要使用账号登录 。

示例  java SocksServerOneThread 1080

[Java]代码

1. import java.io.ByteArrayOutputStream;
   
2. import java.io.Closeable;
   
3. import java.io.IOException;
   
4. import java.io.InputStream;
   
5. import java.io.OutputStream;
   
6. import java.net.InetAddress;
   
7. import java.net.ServerSocket;
   
8. import java.net.Socket;
   
9. import java.nio.ByteBuffer;
   
10. import java.util.Arrays;
    
11. import java.util.Date;
    
12. import java.util.concurrent.CountDownLatch;
    
13. 
    
14. /**
    
15. * 标准的socks代理服务器，支持sock4与sock4代理
    
16. *
    
17. * @author Administrator
    
18. *
    
19. */
    
20. public class SocksServerOneThread implements Runnable {
    
21. 
    
22.     /**
    
23.      * 来源的代理socket
    
24.      */
    
25.     private final Socket socket;
    
26.     /**
    
27.      * 是否开启socks4代理
    
28.      */
    
29.     private final boolean openSock4;
    
30.     /**
    
31.      * 是否开启socks5代理
    
32.      */
    
33.     private final boolean openSock5;
    
34.     /**
    
35.      * socks5代理的登录用户名，如果 不为空表示需要登录验证
    
36.      */
    
37.     private final String user;
    
38.     /**
    
39.      * socks5代理的登录密码，
    
40.      */
    
41.     private final String pwd;
    
42.     /**
    
43.      * socks是否需要进行登录验证
    
44.      */
    
45.     private final boolean socksNeekLogin;
    
46. 
    
47.     /**
    
48.      * @param socket
    
49.      *            来源的代理socket
    
50.      * @param openSock4
    
51.      *            是否开启socks4代理
    
52.      * @param openSock5
    
53.      *            是否开启socks5代理
    
54.      * @param user
    
55.      *            socks5代理的登录用户名，如果 不为空表示需要登录验证
    
56.      * @param pwd
    
57.      *            socks5代理的登录密码，
    
58.      */
    
59.     protected SocksServerOneThread(Socket socket, boolean openSock4, boolean openSock5, String user, String pwd) {
    
60.         this.socket = socket;
    
61.         this.openSock4 = openSock4;
    
62.         this.openSock5 = openSock5;
    
63.         this.user = user;
    
64.         this.pwd = pwd;
    
65.         this.socksNeekLogin = null != user;
    
66.     }
    
67. 
    
68.     public void run() {
    
69.         // 获取来源的地址用于日志打印使用
    
70.         String addr = socket.getRemoteSocketAddress().toString();
    
71.         log("process one socket : %s", addr);
    
72.         // 声明流
    
73.         InputStream a_in = null, b_in = null;
    
74.         OutputStream a_out = null, b_out = null;
    
75.         Socket proxy_socket = null;
    
76.         ByteArrayOutputStream cache = null;
    
77.         try {
    
78.             a_in = socket.getInputStream();
    
79.             a_out = socket.getOutputStream();
    
80. 
    
81.             // 获取协议头。取代理的类型，只有 4，5。
    
82.             byte[] tmp = new byte[1];
    
83.             int n = a_in.read(tmp);
    
84.             if (n == 1) {
    
85.                 byte protocol = tmp[0];
    
86.                 if ((openSock4 && 0x04 == protocol)) {// 如果开启代理4，并以socks4协议请求
    
87.                     proxy_socket = sock4_check(a_in, a_out);
    
88.                 } else if ((openSock5 && 0x05 == protocol)) {// 如果开启代理5，并以socks5协议请求
    
89.                     proxy_socket = sock5_check(a_in, a_out);
    
90.                 } else {// 非socks 4 ,5 协议的请求
    
91.                     log("not socks proxy : %s  openSock4[] openSock5[]", tmp[0], openSock4, openSock5);
    
92.                 }
    
93.                 if (null != proxy_socket) {
    
94.                     CountDownLatch latch = new CountDownLatch(1);
    
95.                     b_in = proxy_socket.getInputStream();
    
96.                     b_out = proxy_socket.getOutputStream();
    
97.                     // 交换流数据
    
98.                     if (80 == proxy_socket.getPort()) {
    
99.                         cache = new ByteArrayOutputStream();
    
100.                     }
     
101.                     transfer(latch, a_in, b_out, cache);
     
102.                     transfer(latch, b_in, a_out, cache);
     
103.                     try {
     
104.                         latch.await();
     
105.                     } catch (Exception e) {
     
106.                     }
     
107.                 }
     
108.             } else {
     
109.                 log("socks error : %s", Arrays.toString(tmp));
     
110.             }
     
111.         } catch (Exception e) {
     
112.             log("exception : %s %s", e.getClass(), e.getLocalizedMessage());
     
113.             e.printStackTrace();
     
114.         } finally {
     
115.             log("close socket, system cleanning ...  %s ", addr);
     
116.             closeIo(a_in);
     
117.             closeIo(b_in);
     
118.             closeIo(b_out);
     
119.             closeIo(a_out);
     
120.             closeIo(socket);
     
121.             closeIo(proxy_socket);
     
122.             if (null != cache) {
     
123.                 cache2Local(cache);
     
124.             }
     
125.         }
     
126.     }
     
127. 
     
128.     private void cache2Local(ByteArrayOutputStream cache) {
     
129.         // OutputStream result = null;
     
130.         // try {
     
131.         // String fileName = System.currentTimeMillis() + "_"
     
132.         // + Thread.currentThread().getId();
     
133.         // result = new FileOutputStream("E:/cache/" + fileName + ".info");
     
134.         // result.write(cache.toByteArray());
     
135.         // } catch (Exception e) {
     
136.         // e.printStackTrace();
     
137.         // } finally {
     
138.         // closeIo(result);
     
139.         // }
     
140.     }
     
141. 
     
142.     /**
     
143.      * sock5代理头处理
     
144.      *
     
145.      * @param in
     
146.      * @param out
     
147.      * @return
     
148.      * @throws IOException
     
149.      */
     
150.     private Socket sock5_check(InputStream in, OutputStream out) throws IOException {
     
151.         byte[] tmp = new byte[2];
     
152.         in.read(tmp);
     
153.         boolean isLogin = false;
     
154.         byte method = tmp[1];
     
155.         if (0x02 == tmp[0]) {
     
156.             method = 0x00;
     
157.             in.read();
     
158.         }
     
159.         if (socksNeekLogin) {
     
160.             method = 0x02;
     
161.         }
     
162.         tmp = new byte[] { 0x05, method };
     
163.         out.write(tmp);
     
164.         out.flush();
     
165.         // Socket result = null;
     
166.         Object resultTmp = null;
     
167.         if (0x02 == method) {// 处理登录.
     
168.             int b = in.read();
     
169.             String user = null;
     
170.             String pwd = null;
     
171.             if (0x01 == b) {
     
172.                 b = in.read();
     
173.                 tmp = new byte[b];
     
174.                 in.read(tmp);
     
175.                 user = new String(tmp);
     
176.                 b = in.read();
     
177.                 tmp = new byte[b];
     
178.                 in.read(tmp);
     
179.                 pwd = new String(tmp);
     
180.                 if (null != user && user.trim().equals(this.user) && null != pwd && pwd.trim().equals(this.pwd)) {// 权限过滤
     
181.                     isLogin = true;
     
182.                     tmp = new byte[] { 0x05, 0x00 };// 登录成功
     
183.                     out.write(tmp);
     
184.                     out.flush();
     
185.                     log("%s login success !", user);
     
186.                 } else {
     
187.                     log("%s login faild !", user);
     
188.                 }
     
189.             }
     
190.         }
     
191.         byte cmd = 0;
     
192.         if (!socksNeekLogin || isLogin) {// 验证是否需要登录
     
193.             tmp = new byte[4];
     
194.             in.read(tmp);
     
195.             log("proxy header >>  %s", Arrays.toString(tmp));
     
196.             cmd = tmp[1];
     
197.             String host = getHost(tmp[3], in);
     
198.             tmp = new byte[2];
     
199.             in.read(tmp);
     
200.             int port = ByteBuffer.wrap(tmp).asShortBuffer().get() & 0xFFFF;
     
201.             log("connect %s:%s", host, port);
     
202.             ByteBuffer rsv = ByteBuffer.allocate(10);
     
203.             rsv.put((byte) 0x05);
     
204.             try {
     
205.                 if (0x01 == cmd) {
     
206.                     resultTmp = new Socket(host, port);
     
207.                     rsv.put((byte) 0x00);
     
208.                 } else if (0x02 == cmd) {
     
209.                     resultTmp = new ServerSocket(port);
     
210.                     rsv.put((byte) 0x00);
     
211.                 } else {
     
212.                     rsv.put((byte) 0x05);
     
213.                     resultTmp = null;
     
214.                 }
     
215.             } catch (Exception e) {
     
216.                 rsv.put((byte) 0x05);
     
217.                 resultTmp = null;
     
218.             }
     
219.             rsv.put((byte) 0x00);
     
220.             rsv.put((byte) 0x01);
     
221.             rsv.put(socket.getLocalAddress().getAddress());
     
222.             Short localPort = (short) ((socket.getLocalPort()) & 0xFFFF);
     
223.             rsv.putShort(localPort);
     
224.             tmp = rsv.array();
     
225.         } else {
     
226.             tmp = new byte[] { 0x05, 0x01 };// 登录失败
     
227.             log("socks server need login,but no login info .");
     
228.         }
     
229.         out.write(tmp);
     
230.         out.flush();
     
231.         if (null != resultTmp && 0x02 == cmd) {
     
232.             ServerSocket ss = (ServerSocket) resultTmp;
     
233.             try {
     
234.                 resultTmp = ss.accept();
     
235.             } catch (Exception e) {
     
236.             } finally {
     
237.                 closeIo(ss);
     
238.             }
     
239.         }
     
240.         return (Socket) resultTmp;
     
241.     }
     
242. 
     
243.     /**
     
244.      * sock4代理的头处理
     
245.      *
     
246.      * @param in
     
247.      * @param out
     
248.      * @return
     
249.      * @throws IOException
     
250.      */
     
251.     private Socket sock4_check(InputStream in, OutputStream out) throws IOException {
     
252.         Socket proxy_socket = null;
     
253.         byte[] tmp = new byte[3];
     
254.         in.read(tmp);
     
255.         // 请求协议|VN1|CD1|DSTPORT2|DSTIP4|NULL1|
     
256.         int port = ByteBuffer.wrap(tmp, 1, 2).asShortBuffer().get() & 0xFFFF;
     
257.         String host = getHost((byte) 0x01, in);
     
258.         in.read();
     
259.         byte[] rsv = new byte[8];// 返回一个8位的响应协议
     
260.         // |VN1|CD1|DSTPORT2|DSTIP 4|
     
261.         try {
     
262.             proxy_socket = new Socket(host, port);
     
263.             log("connect [%s] %s:%s", tmp[1], host, port);
     
264.             rsv[1] = 90;// 代理成功
     
265.         } catch (Exception e) {
     
266.             log("connect exception  %s:%s", host, port);
     
267.             rsv[1] = 91;// 代理失败.
     
268.         }
     
269.         out.write(rsv);
     
270.         out.flush();
     
271.         return proxy_socket;
     
272.     }
     
273. 
     
274.     /**
     
275.      * 获取目标的服务器地址
     
276.      *
     
277.      * @createTime 2014年12月14日 下午8:32:15
     
278.      * @param type
     
279.      * @param in
     
280.      * @return
     
281.      * @throws IOException
     
282.      */
     
283.     private String getHost(byte type, InputStream in) throws IOException {
     
284.         String host = null;
     
285.         byte[] tmp = null;
     
286.         switch (type) {
     
287.         case 0x01:// IPV4协议
     
288.             tmp = new byte[4];
     
289.             in.read(tmp);
     
290.             host = InetAddress.getByAddress(tmp).getHostAddress();
     
291.             break;
     
292.         case 0x03:// 使用域名
     
293.             int l = in.read();
     
294.             tmp = new byte[l];
     
295.             in.read(tmp);
     
296.             host = new String(tmp);
     
297.             break;
     
298.         case 0x04:// 使用IPV6
     
299.             tmp = new byte[16];
     
300.             in.read(tmp);
     
301.             host = InetAddress.getByAddress(tmp).getHostAddress();
     
302.             break;
     
303.         default:
     
304.             break;
     
305.         }
     
306.         return host;
     
307.     }
     
308. 
     
309.     /**
     
310.      * IO操作**同的关闭方法
     
311.      *
     
312.      * @createTime 2014年12月14日 下午7:50:56
     
313.      * @param socket
     
314.      */
     
315.     protected static final void closeIo(Socket closeable) {
     
316.         if (null != closeable) {
     
317.             try {
     
318.                 closeable.close();
     
319.             } catch (IOException e) {
     
320.             }
     
321.         }
     
322.     }
     
323. 
     
324.     /**
     
325.      * IO操作**同的关闭方法
     
326.      *
     
327.      * @createTime 2014年12月14日 下午7:50:56
     
328.      * @param socket
     
329.      */
     
330.     protected static final void closeIo(Closeable closeable) {
     
331.         if (null != closeable) {
     
332.             try {
     
333.                 closeable.close();
     
334.             } catch (IOException e) {
     
335.             }
     
336.         }
     
337.     }
     
338. 
     
339.     /**
     
340.      * 数据交换.主要用于tcp协议的交换
     
341.      *
     
342.      * @createTime 2014年12月13日 下午11:06:47
     
343.      * @param lock
     
344.      *            锁
     
345.      * @param in
     
346.      *            输入流
     
347.      * @param out
     
348.      *            输出流
     
349.      */
     
350.     protected static final void transfer(final CountDownLatch latch, final InputStream in, final OutputStream out,
     
351.             final OutputStream cache) {
     
352.         new Thread() {
     
353.             public void run() {
     
354.                 byte[] bytes = new byte[1024];
     
355.                 int n = 0;
     
356.                 try {
     
357.                     while ((n = in.read(bytes)) > 0) {
     
358.                         out.write(bytes, 0, n);
     
359.                         out.flush();
     
360.                         if (null != cache) {
     
361.                             synchronized (cache) {
     
362.                                 cache.write(bytes, 0, n);
     
363.                             }
     
364.                         }
     
365.                     }
     
366.                 } catch (Exception e) {
     
367.                 }
     
368.                 if (null != latch) {
     
369.                     latch.countDown();
     
370.                 }
     
371.             };
     
372.         }.start();
     
373.     }
     
374. 
     
375.     private final static void log(String message, Object... args) {
     
376.         Date dat = new Date();
     
377.         String msg = String.format("%1$tF %1$tT %2$-5s %3$s%n", dat, Thread.currentThread().getId(),
     
378.                 String.format(message, args));
     
379.         System.out.print(msg);
     
380.     }
     
381. 
     
382.     public static void startServer(int port, boolean openSock4, boolean openSock5, String user, String pwd)
     
383.             throws IOException {
     
384.         log("config >> port[%s] openSock4[%s] openSock5[%s] user[%s] pwd[%s]", port, openSock4, openSock5, user, pwd);
     
385.         ServerSocket ss = new ServerSocket(port);
     
386.         Socket socket = null;
     
387.         log("Socks server port : %s listenning...", port);
     
388.         while (null != (socket = ss.accept())) {
     
389.             new Thread(new SocksServerOneThread(socket, openSock4, openSock5, user, pwd)).start();
     
390.         }
     
391.         ss.close();
     
392.     }
     
393. 
     
394.     public static void main(String[] args) throws IOException {
     
395.         java.security.Security.setProperty("networkaddress.cache.ttl", "86400");
     
396.         log("\n\tUSing port openSock4 openSock5 user pwd");
     
397.         int port = 1080;
     
398.         boolean openSock4 = true;
     
399.         boolean openSock5 = true;
     
400.         String user = null, pwd = null;
     
401.         // user = "user";
     
402.         pwd = "123456";
     
403.         int i = 0;
     
404.         if (args.length > i && null != args[i++]) {
     
405.             port = Integer.valueOf(args[i].trim());
     
406.         }
     
407.         if (args.length > i && null != args[i++]) {
     
408.             openSock4 = Boolean.valueOf(args[i].trim());
     
409.         }
     
410.         if (args.length > i && null != args[i++]) {
     
411.             openSock5 = Boolean.valueOf(args[i].trim());
     
412.         }
     
413.         if (args.length > i && null != args[i++]) {
     
414.             user = args[i].trim();
     
415.         }
     
416.         if (args.length > i && null != args[i++]) {
     
417.             pwd = args[i].trim();
     
418.         }
     
419.         SocksServerOneThread.startServer(port, openSock4, openSock5, user, pwd);
     
420.     }
     
421. }

复制代码