1 package be
.nikiroo
.utils
.serial
.server
;
3 import java
.io
.IOException
;
4 import java
.lang
.reflect
.Array
;
5 import java
.net
.Socket
;
7 import be
.nikiroo
.utils
.StringUtils
;
8 import be
.nikiroo
.utils
.TraceHandler
;
9 import be
.nikiroo
.utils
.Version
;
10 import be
.nikiroo
.utils
.serial
.Importer
;
13 * This class implements a simple server that can bridge two other
16 * It can, of course, inspect the data that goes through it (by default, it
17 * prints traces of the data).
19 * Note: this {@link ServerBridge} has to be discarded after use (cannot be
24 public class ServerBridge
extends Server
{
25 private final String forwardToHost
;
26 private final int forwardToPort
;
27 private final boolean forwardToSsl
;
30 * Create a new server that will start listening on the network when
31 * {@link ServerBridge#start()} is called.
34 * the port to listen on, or 0 to assign any unallocated port
35 * found (which can later on be queried via
36 * {@link ServerBridge#getPort()}
38 * use an SSL connection (or not)
39 * @param forwardToHost
40 * the host server to forward the calls to
41 * @param forwardToPort
42 * the host port to forward the calls to
44 * use an SSL connection for the forward server or not
47 * in case of I/O error
49 public ServerBridge(int port
, boolean ssl
, String forwardToHost
,
50 int forwardToPort
, boolean forwardToSsl
) throws IOException
{
52 this.forwardToHost
= forwardToHost
;
53 this.forwardToPort
= forwardToPort
;
54 this.forwardToSsl
= forwardToSsl
;
58 * Create a new server that will start listening on the network when
59 * {@link ServerBridge#start()} is called.
62 * the server name (only used for debug info and traces)
64 * the port to listen on
66 * use an SSL connection (or not)
67 * @param forwardToHost
68 * the host server to forward the calls to
69 * @param forwardToPort
70 * the host port to forward the calls to
72 * use an SSL connection for the forward server or not
75 * in case of I/O error
77 public ServerBridge(String name
, int port
, boolean ssl
,
78 String forwardToHost
, int forwardToPort
, boolean forwardToSsl
)
80 super(name
, port
, ssl
);
81 this.forwardToHost
= forwardToHost
;
82 this.forwardToPort
= forwardToPort
;
83 this.forwardToSsl
= forwardToSsl
;
87 * The traces handler for this {@link Server}.
89 * The trace levels are handled as follow:
91 * <li>1: it will only print basic IN/OUT messages with length</li>
92 * <li>2: it will try to interpret it as an object (SLOW) and print the
93 * object class if possible</li>
94 * <li>3: it will try to print the {@link Object#toString()} value, or the
95 * data if it is not an object</li>
96 * <li>4: it will also print the unzipped serialised value if it is an
101 * the new traces handler
104 public void setTraceHandler(TraceHandler tracer
) {
105 super.setTraceHandler(tracer
);
109 protected ConnectActionServer
createConnectActionServer(Socket s
) {
110 return new ConnectActionServerString(s
) {
112 public void action(final Version clientVersion
) throws Exception
{
113 onClientContact(clientVersion
);
114 final ConnectActionServerString bridge
= this;
116 new ConnectActionClientString(forwardToHost
, forwardToPort
,
117 forwardToSsl
, clientVersion
) {
119 public void action(final Version serverVersion
)
121 onServerContact(serverVersion
);
123 for (String fromClient
= bridge
.rec(); fromClient
!= null; fromClient
= bridge
125 onRec(clientVersion
, fromClient
);
126 String fromServer
= send(fromClient
);
127 onSend(serverVersion
, fromServer
);
128 bridge
.send(fromServer
);
137 * This is the method that is called each time a client contact us.
139 * @param clientVersion
142 protected void onClientContact(Version clientVersion
) {
143 getTraceHandler().trace("<<< CLIENT " + clientVersion
);
147 * This is the method that is called each time a client contact us.
149 * @param serverVersion
152 protected void onServerContact(Version serverVersion
) {
153 getTraceHandler().trace(">>> SERVER " + serverVersion
);
154 getTraceHandler().trace("");
158 * This is the method that is called each time a client contact us.
160 * @param clientVersion
163 * the data sent by the client
165 protected void onRec(Version clientVersion
, String data
) {
166 trace("<<< CLIENT (" + clientVersion
+ ")", data
);
170 * This is the method that is called each time the forwarded server contact
173 * @param serverVersion
176 * the data sent by the client
178 protected void onSend(Version serverVersion
, String data
) {
179 trace(">>> SERVER (" + serverVersion
+ ")", data
);
183 * Trace the data with the given prefix.
186 * the prefix (client, server, version...)
190 private void trace(String prefix
, String data
) {
191 int size
= data
.length();
192 String ssize
= size
+ " byte";
194 ssize
= size
+ " bytes";
197 ssize
= size
+ " kb";
200 ssize
= size
+ " MB";
205 getTraceHandler().trace(prefix
+ ": " + ssize
, 1);
207 if (getTraceHandler().getTraceLevel() >= 2) {
209 if (data
.startsWith("ZIP:")) {
210 data
= StringUtils
.unzip64(data
.substring(4));
213 Object obj
= new Importer().read(data
).getValue();
215 getTraceHandler().trace("NULL", 2);
216 getTraceHandler().trace("NULL", 3);
217 getTraceHandler().trace("NULL", 4);
219 if (obj
.getClass().isArray()) {
220 getTraceHandler().trace(
221 "(" + obj
.getClass() + ") with "
222 + Array
.getLength(obj
) + "element(s)",
225 getTraceHandler().trace("(" + obj
.getClass() + ")", 2);
227 getTraceHandler().trace("" + obj
.toString(), 3);
228 getTraceHandler().trace(data
, 4);
230 } catch (NoSuchMethodException e
) {
231 getTraceHandler().trace("(not an object)", 2);
232 getTraceHandler().trace(data
, 3);
233 getTraceHandler().trace("", 4);
234 } catch (NoSuchFieldException e
) {
235 getTraceHandler().trace(
236 "(incompatible: " + e
.getMessage() + ")", 2);
237 getTraceHandler().trace(data
, 3);
238 getTraceHandler().trace("", 4);
239 } catch (ClassNotFoundException e
) {
240 getTraceHandler().trace(
241 "(unknown object: " + e
.getMessage() + ")", 2);
242 getTraceHandler().trace(data
, 3);
243 getTraceHandler().trace("", 4);
244 } catch (Exception e
) {
245 getTraceHandler().trace(
246 "(decode error: " + e
.getMessage() + ")", 2);
247 getTraceHandler().trace(data
, 3);
248 getTraceHandler().trace("", 4);
251 getTraceHandler().trace("", 2);
256 * Start a bridge between 2 servers.
259 * an array containing:
261 * <li>The bridge name</li>
262 * <li>The bridge port</li>
263 * <li>TRUE for an SSL bridge, FALSE for plain text</li>
264 * <li>The forward server host</li>
265 * <li>The forward server port</li>
266 * <li>TRUE for an SSL forward server, FALSE for plain text</li>
267 * <li>(optional) a trace level</li>
268 * <li>(optional) a truncate size for data</li>
271 public static void main(String
[] args
) {
272 final TraceHandler tracer
= new TraceHandler(true, false, 0);
274 if (args
.length
< 6) {
275 tracer
.error("Invalid syntax.\n"
276 + "Syntax: [name] [port] [ssl] [fhost] [fport] [fssl] ([trace level]) ([max])\n"
277 + "\tname: the bridge name\n"
278 + "\tport: the bridge port\n"
279 + "\tssl: TRUE for an SSL bridge, FALSE for plain text\n"
280 + "\tfhost: the forward server host\n"
281 + "\tfport: the forward server port\n"
282 + "\tfssl: TRUE for an SSL forward server, FALSE for plain text\n"
283 + "\ttrace level: the optional trace level (default is 1)\n"
284 + "\tmax: the maximum size after which to truncate data\n");
289 String name
= args
[i
++];
290 int port
= Integer
.parseInt(args
[i
++]);
291 boolean ssl
= Boolean
.parseBoolean(args
[i
++]);
292 String fhost
= args
[i
++];
293 int fport
= Integer
.parseInt(args
[i
++]);
294 boolean fssl
= Boolean
.parseBoolean(args
[i
++]);
297 if (args
.length
> 6) {
298 traceLevel
= Integer
.parseInt(args
[i
++]);
300 int maxPrintSize
= 1;
301 if (args
.length
> 7) {
302 maxPrintSize
= Integer
.parseInt(args
[i
++]);
305 ServerBridge bridge
= new ServerBridge(name
, port
, ssl
, fhost
,
307 bridge
.setTraceHandler(new TraceHandler(true, true, traceLevel
,
310 } catch (Exception e
) {