package com.subgraph.orchid.socks;
import java.io.IOException;
import java.net.Socket;
import java.util.concurrent.TimeoutException;
import java.util.logging.Level;
import java.util.logging.Logger;
import com.subgraph.orchid.CircuitManager;
import com.subgraph.orchid.OpenFailedException;
import com.subgraph.orchid.Stream;
import com.subgraph.orchid.TorConfig;
import com.subgraph.orchid.TorException;
public class SocksClientTask implements Runnable {
private final static Logger logger = Logger.getLogger(SocksClientTask.class.getName());
private final TorConfig config;
private final Socket socket;
private final CircuitManager circuitManager;
SocksClientTask(TorConfig config, Socket socket, CircuitManager circuitManager) {
this.config = config;
this.socket = socket;
this.circuitManager = circuitManager;
}
public void run() {
final int version = readByte();
dispatchRequest(version);
closeSocket();
}
private int readByte() {
try {
return socket.getInputStream().read();
} catch (IOException e) {
logger.warning("IO error reading version byte: "+ e.getMessage());
return -1;
}
}
private void dispatchRequest(int versionByte) {
switch(versionByte) {
case 'H':
case 'G':
case 'P':
sendHttpPage();
break;
case 4:
processRequest(new Socks4Request(config, socket));
break;
case 5:
processRequest(new Socks5Request(config, socket));
break;
default:
// fall through, do nothing
break;
}
}
private void processRequest(SocksRequest request) {
try {
request.readRequest();
if(!request.isConnectRequest()) {
logger.warning("Non connect command ("+ request.getCommandCode() + ")");
request.sendError(true);
return;
}
try {
final Stream stream = openConnectStream(request);
logger.fine("SOCKS CONNECT to "+ request.getTarget()+ " completed");
request.sendSuccess();
runOpenConnection(stream);
} catch (InterruptedException e) {
logger.info("SOCKS CONNECT to "+ request.getTarget() + " was thread interrupted");
Thread.currentThread().interrupt();
request.sendError(false);
} catch (TimeoutException e) {
logger.info("SOCKS CONNECT to "+ request.getTarget() + " timed out");
request.sendError(false);
} catch (OpenFailedException e) {
logger.info("SOCKS CONNECT to "+ request.getTarget() + " failed: "+ e.getMessage());
request.sendConnectionRefused();
}
} catch (SocksRequestException e) {
logger.log(Level.WARNING, "Failure reading SOCKS request: "+ e.getMessage());
try {
request.sendError(false);
socket.close();
} catch (Exception ignore) { }
}
}
private void runOpenConnection(Stream stream) {
SocksStreamConnection.runConnection(socket, stream);
}
private Stream openConnectStream(SocksRequest request) throws InterruptedException, TimeoutException, OpenFailedException {
if(request.hasHostname()) {
logger.fine("SOCKS CONNECT request to "+ request.getHostname() +":"+ request.getPort());
return circuitManager.openExitStreamTo(request.getHostname(), request.getPort());
} else {
logger.fine("SOCKS CONNECT request to "+ request.getAddress() +":"+ request.getPort());
return circuitManager.openExitStreamTo(request.getAddress(), request.getPort());
}
}
private void sendHttpPage() {
throw new TorException("Returning HTTP page not implemented");
}
private void closeSocket() {
try {
socket.close();
} catch (IOException e) {
logger.warning("Error closing SOCKS socket: "+ e.getMessage());
}
}
}