--- /dev/null
+package be.nikiroo.utils.serial.server;
+
+import java.io.IOException;
+import java.net.Socket;
+
+import be.nikiroo.utils.TraceHandler;
+import be.nikiroo.utils.Version;
+import be.nikiroo.utils.serial.Importer;
+
+/**
+ * This class implements a simple server that can bridge two other
+ * {@link Server}s.
+ * <p>
+ * It can, of course, inspect the data that goes through it (by default, it
+ * prints traces of the data).
+ * <p>
+ * Note: this {@link ServerBridge} has to be discarded after use (cannot be
+ * started twice).
+ *
+ * @author niki
+ */
+public class ServerBridge extends Server {
+ private final String forwardToHost;
+ private final int forwardToPort;
+ private final boolean forwardToSsl;
+
+ /**
+ * Create a new server that will start listening on the network when
+ * {@link ServerBridge#start()} is called.
+ *
+ * @param port
+ * the port to listen on, or 0 to assign any unallocated port
+ * found (which can later on be queried via
+ * {@link ServerBridge#getPort()}
+ * @param ssl
+ * use a SSL connection (or not)
+ *
+ * @throws IOException
+ * in case of I/O error
+ */
+ public ServerBridge(int port, boolean ssl, String forwardToHost,
+ int forwardToPort, boolean forwardToSsl) throws IOException {
+ super(port, ssl);
+ this.forwardToHost = forwardToHost;
+ this.forwardToPort = forwardToPort;
+ this.forwardToSsl = forwardToSsl;
+ }
+
+ /**
+ * Create a new server that will start listening on the network when
+ * {@link ServerBridge#start()} is called.
+ *
+ * @param name
+ * the server name (only used for debug info and traces)
+ * @param port
+ * the port to listen on
+ * @param ssl
+ * use a SSL connection (or not)
+ *
+ * @throws IOException
+ * in case of I/O error
+ */
+ public ServerBridge(String name, int port, boolean ssl,
+ String forwardToHost, int forwardToPort, boolean forwardToSsl)
+ throws IOException {
+ super(name, port, ssl);
+ this.forwardToHost = forwardToHost;
+ this.forwardToPort = forwardToPort;
+ this.forwardToSsl = forwardToSsl;
+ }
+
+ /**
+ * The traces handler for this {@link Server}.
+ * <p>
+ * <ul>
+ * <li>At level 1, it will only print basic IN/OUT messages with length.</li>
+ * <li>At level 2, it will also print the data as a String.</li>
+ * <li>At level 3, it will try to interpret it (SLOW) and print the object
+ * type if possible.</li>
+ * <li>At level 4, it will try to print the {@link Object#toString()} value.
+ * </li>
+ * </ul>
+ *
+ * @param tracer
+ * the new traces handler
+ */
+ @Override
+ public void setTraceHandler(TraceHandler tracer) {
+ super.setTraceHandler(tracer);
+ }
+
+ @Override
+ protected ConnectActionServer createConnectActionServer(Socket s) {
+ return new ConnectActionServerString(s) {
+ @Override
+ public void action(final Version clientVersion) throws Exception {
+ onClientContact(clientVersion);
+ final ConnectActionServerString bridge = this;
+
+ 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);
+ }
+ }
+ }.connect();
+ }
+ };
+ }
+
+ /**
+ * This is the method that is called each time a client contact us.
+ *
+ * @param clientVersion
+ * the client version
+ * @param data
+ * the data sent by the client
+ */
+ protected void onClientContact(Version clientVersion) {
+ getTraceHandler().trace("CLIENT " + clientVersion);
+ }
+
+ /**
+ * This is the method that is called each time a client contact us.
+ *
+ * @param serverVersion
+ * the server version
+ * @param data
+ * the data sent by the client
+ */
+ protected void onServerContact(Version serverVersion) {
+ getTraceHandler().trace("SERVER " + serverVersion);
+ }
+
+ /**
+ * This is the method that is called each time a client contact us.
+ *
+ * @param clientVersion
+ * the client version
+ * @param data
+ * the data sent by the client
+ */
+ protected void onRec(Version clientVersion, String data) {
+ trace("<<< CLIENT (" + clientVersion + ")", data);
+ }
+
+ /**
+ * This is the method that is called each time the forwarded server contact
+ * us.
+ *
+ * @param serverVersion
+ * the client version
+ * @param data
+ * the data sent by the client
+ */
+ protected void onSend(Version serverVersion, String data) {
+ trace(">>> SERVER (" + serverVersion + ")", data);
+ }
+
+ /**
+ * Trace the data with the given prefix.
+ *
+ * @param prefix
+ * the prefix (client, server, version...)
+ * @param data
+ * the data to trace
+ */
+ private void trace(String prefix, String data) {
+ getTraceHandler().trace(prefix + ": " + data.length() + " characters",
+ 1);
+
+ if (getTraceHandler().getTraceLevel() >= 2) {
+ try {
+ Object obj = new Importer().read(data).getValue();
+ if (obj == null) {
+ getTraceHandler().trace("NULL", 3);
+ getTraceHandler().trace("NULL", 4);
+ } else {
+ getTraceHandler().trace("(" + obj.getClass() + ")", 3);
+ getTraceHandler().trace("" + obj.toString(), 4);
+ }
+ } catch (Exception e) {
+ getTraceHandler().trace("(not an object)", 3);
+ getTraceHandler().trace(data, 4);
+ }
+
+ getTraceHandler().trace("", 4);
+ }
+ }
+
+ /**
+ * 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 a ssl bridge, FALSE for plain text</li>
+ * <li>The forward server host</li>
+ * <li>The forward server port</li>
+ * <li>TRUE for a ssl forward server, FALSE for plain text</li>
+ * <li>(optional) a trace level</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])\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");
+ 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++]);
+ }
+
+ ServerBridge bridge = new ServerBridge(name, port, ssl, fhost,
+ fport, fssl);
+ bridge.setTraceHandler(new TraceHandler(true, true, traceLevel));
+ bridge.run();
+ } catch (Exception e) {
+ tracer.error(e);
+ }
+ }
+}