import java.io.IOException;
import java.lang.reflect.Array;
import java.net.Socket;
+import java.net.UnknownHostException;
import be.nikiroo.utils.StringUtils;
import be.nikiroo.utils.TraceHandler;
*
* @throws IOException
* in case of I/O error
+ * @throws UnknownHostException
+ * if the IP address of the host could not be determined
+ * @throws IllegalArgumentException
+ * if the port parameter is outside the specified range of valid
+ * port values, which is between 0 and 65535, inclusive
*/
public ServerBridge(int port, boolean ssl, String forwardToHost,
int forwardToPort, boolean forwardToSsl) throws IOException {
*
* @throws IOException
* in case of I/O error
+ * @throws UnknownHostException
+ * if the IP address of the host could not be determined
+ * @throws IllegalArgumentException
+ * if the port parameter is outside the specified range of valid
+ * port values, which is between 0 and 65535, inclusive
*/
public ServerBridge(String name, int port, boolean ssl,
String forwardToHost, int forwardToPort, boolean forwardToSsl)
onClientContact(clientVersion);
final ConnectActionServerString bridge = this;
- new ConnectActionClientString(forwardToHost, forwardToPort,
- forwardToSsl, clientVersion) {
- @Override
- public void action(final Version serverVersion)
- throws Exception {
- onServerContact(serverVersion);
+ try {
+ new ConnectActionClientString(forwardToHost, forwardToPort,
+ forwardToSsl, clientVersion) {
+ @Override
+ public void action(final Version serverVersion)
+ throws Exception {
+ onServerContact(serverVersion);
- for (String fromClient = bridge.rec(); fromClient != null; fromClient = bridge
- .rec()) {
- onRec(clientVersion, fromClient);
- String fromServer = send(fromClient);
- onSend(serverVersion, fromServer);
- bridge.send(fromServer);
+ for (String fromClient = bridge.rec(); fromClient != null; fromClient = bridge
+ .rec()) {
+ onRec(clientVersion, fromClient);
+ String fromServer = send(fromClient);
+ onSend(serverVersion, fromServer);
+ bridge.send(fromServer);
+ }
+
+ getTraceHandler().trace("=== DONE", 1);
+ getTraceHandler().trace("", 1);
}
- }
- @Override
- protected void onError(Exception e) {
- getTraceHandler().error(e);
- }
- }.connect();
+ @Override
+ protected void onError(Exception e) {
+ ServerBridge.this.onError(e);
+ }
+ }.connect();
+ } catch (Exception e) {
+ ServerBridge.this.onError(e);
+ }
}
};
}
* the client version
*/
protected void onClientContact(Version clientVersion) {
- getTraceHandler().trace("<<< CLIENT " + clientVersion);
+ getTraceHandler().trace(">>> CLIENT " + clientVersion);
}
/**
* the server version
*/
protected void onServerContact(Version serverVersion) {
- getTraceHandler().trace(">>> SERVER " + serverVersion);
+ getTraceHandler().trace("<<< SERVER " + serverVersion);
getTraceHandler().trace("");
}
* the data sent by the client
*/
protected void onRec(Version clientVersion, String data) {
- trace("<<< CLIENT (" + clientVersion + ")", data);
+ trace(">>> CLIENT (" + clientVersion + ")", data);
}
/**
* the data sent by the client
*/
protected void onSend(Version serverVersion, String data) {
- trace(">>> SERVER (" + serverVersion + ")", data);
+ trace("<<< SERVER (" + serverVersion + ")", data);
+ }
+
+ @Override
+ public void run() {
+ getTraceHandler().trace(
+ getName() + ": will forward to " + forwardToHost + ":"
+ + forwardToPort + " ("
+ + (forwardToSsl ? "SSL" : "plain text") + ")");
+ super.run();
}
/**
* the data to trace
*/
private void trace(String prefix, String data) {
- int size = data.length();
+ int size = data == null ? 0 : data.length();
String ssize = size + " byte";
if (size > 1) {
ssize = size + " bytes";
if (getTraceHandler().getTraceLevel() >= 2) {
try {
- if (data.startsWith("ZIP:")) {
- data = StringUtils.unzip64(data.substring(4));
+ while (data.startsWith("ZIP:") || data.startsWith("B64:")) {
+ if (data.startsWith("ZIP:")) {
+ data = StringUtils.unbase64s(data.substring(4), true);
+ } else if (data.startsWith("B64:")) {
+ data = StringUtils.unbase64s(data.substring(4), false);
+ }
}
Object obj = new Importer().read(data).getValue();
getTraceHandler().trace("", 2);
}
}
-
- /**
- * Start a bridge between 2 servers.
- *
- * @param args
- * an array containing:
- * <ul>
- * <li>The bridge name</li>
- * <li>The bridge port</li>
- * <li>TRUE for an SSL bridge, FALSE for plain text</li>
- * <li>The forward server host</li>
- * <li>The forward server port</li>
- * <li>TRUE for an SSL forward server, FALSE for plain text</li>
- * <li>(optional) a trace level</li>
- * <li>(optional) a truncate size for data</li>
- * </ul>
- */
- public static void main(String[] args) {
- final TraceHandler tracer = new TraceHandler(true, false, 0);
- try {
- if (args.length < 6) {
- tracer.error("Invalid syntax.\n"
- + "Syntax: [name] [port] [ssl] [fhost] [fport] [fssl] ([trace level]) ([max])\n"
- + "\tname: the bridge name\n"
- + "\tport: the bridge port\n"
- + "\tssl: TRUE for an SSL bridge, FALSE for plain text\n"
- + "\tfhost: the forward server host\n"
- + "\tfport: the forward server port\n"
- + "\tfssl: TRUE for an SSL forward server, FALSE for plain text\n"
- + "\ttrace level: the optional trace level (default is 1)\n"
- + "\tmax: the maximum size after which to truncate data\n");
- return;
- }
-
- int i = 0;
- String name = args[i++];
- int port = Integer.parseInt(args[i++]);
- boolean ssl = Boolean.parseBoolean(args[i++]);
- String fhost = args[i++];
- int fport = Integer.parseInt(args[i++]);
- boolean fssl = Boolean.parseBoolean(args[i++]);
-
- int traceLevel = 1;
- if (args.length > 6) {
- traceLevel = Integer.parseInt(args[i++]);
- }
- int maxPrintSize = 1;
- if (args.length > 7) {
- maxPrintSize = Integer.parseInt(args[i++]);
- }
-
- ServerBridge bridge = new ServerBridge(name, port, ssl, fhost,
- fport, fssl);
- bridge.setTraceHandler(new TraceHandler(true, true, traceLevel,
- maxPrintSize));
- bridge.run();
- } catch (Exception e) {
- tracer.error(e);
- }
- }
}
git://git.nikiroo.be / nikiroo-utils.git / blobdiff