60 lines
2.0 KiB
Java
60 lines
2.0 KiB
Java
/*
|
|
* To change this license header, choose License Headers in Project Properties.
|
|
* To change this template file, choose Tools | Templates
|
|
* and open the template in the editor.
|
|
*/
|
|
package org.redkale.net.socks;
|
|
|
|
import org.redkale.util.AnyValue;
|
|
import org.redkale.net.Context;
|
|
import org.redkale.util.Utility;
|
|
import org.redkale.util.AutoLoad;
|
|
import java.io.*;
|
|
import java.net.*;
|
|
import java.nio.*;
|
|
import java.util.logging.*;
|
|
|
|
/**
|
|
*
|
|
* @see http://www.redkale.org
|
|
* @author zhangjx
|
|
*/
|
|
@AutoLoad(false)
|
|
public class SocksConnectServlet extends SocksServlet {
|
|
|
|
private InetSocketAddress bindAddress;
|
|
|
|
private byte[] bindAddressBytes = new byte[0];
|
|
|
|
@Override
|
|
public void init(Context context, AnyValue config) {
|
|
if (config == null) {
|
|
this.bindAddress = new InetSocketAddress(Utility.localInetAddress(), context.getServerAddress().getPort());
|
|
} else {
|
|
this.bindAddress = new InetSocketAddress(config.getValue("bindaddr", Utility.localInetAddress().getHostAddress()), context.getServerAddress().getPort());
|
|
}
|
|
Logger logger = context.getLogger();
|
|
if (logger.isLoggable(Level.INFO)) logger.info("[" + Thread.currentThread().getName() + "] bindAddress = " + bindAddress);
|
|
ByteBuffer bb;
|
|
InetAddress addr = bindAddress.getAddress();
|
|
if (addr instanceof Inet6Address) {
|
|
bb = ByteBuffer.allocate(1 + 16 + 2);
|
|
bb.put((byte) 0x04);
|
|
} else {
|
|
bb = ByteBuffer.allocate(1 + 4 + 2);
|
|
bb.put((byte) 0x01);
|
|
}
|
|
bb.put(addr.getAddress());
|
|
bb.putChar((char) bindAddress.getPort());
|
|
bb.flip();
|
|
this.bindAddressBytes = bb.array();
|
|
}
|
|
|
|
@Override
|
|
public void execute(SocksRequest request, SocksResponse response) throws IOException {
|
|
response.getContext().submit(new SocksRunner((SocksContext) response.getContext(), response.removeChannel(), bindAddressBytes));
|
|
response.finish(true);
|
|
}
|
|
|
|
}
|