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
);
157 * This is the method that is called each time a client contact us.
159 * @param clientVersion
162 * the data sent by the client
164 protected void onRec(Version clientVersion
, String data
) {
165 trace("<<< CLIENT (" + clientVersion
+ ")", data
);
169 * This is the method that is called each time the forwarded server contact
172 * @param serverVersion
175 * the data sent by the client
177 protected void onSend(Version serverVersion
, String data
) {
178 trace(">>> SERVER (" + serverVersion
+ ")", data
);
182 * Trace the data with the given prefix.
185 * the prefix (client, server, version...)
189 private void trace(String prefix
, String data
) {
190 int size
= data
.length();
191 String ssize
= size
+ " byte";
193 ssize
= size
+ " bytes";
196 ssize
= size
+ " kb";
199 ssize
= size
+ " MB";
204 getTraceHandler().trace(prefix
+ ": " + ssize
, 1);
206 if (getTraceHandler().getTraceLevel() >= 2) {
208 if (data
.startsWith("ZIP:")) {
209 data
= StringUtils
.unzip64(data
.substring(4));
212 Object obj
= new Importer().read(data
).getValue();
214 getTraceHandler().trace("NULL", 2);
215 getTraceHandler().trace("NULL", 3);
216 getTraceHandler().trace("NULL", 4);
218 if (obj
.getClass().isArray()) {
219 getTraceHandler().trace(
220 "(" + obj
.getClass() + ") with "
221 + Array
.getLength(obj
) + "element(s)",
224 getTraceHandler().trace("(" + obj
.getClass() + ")", 2);
226 getTraceHandler().trace("" + obj
.toString(), 3);
227 getTraceHandler().trace(data
, 4);
229 } catch (NoSuchMethodException e
) {
230 getTraceHandler().trace("(not an object)", 2);
231 getTraceHandler().trace(data
, 3);
232 getTraceHandler().trace("", 4);
233 } catch (NoSuchFieldException e
) {
234 getTraceHandler().trace(
235 "(object known but incompatible: " + e
.getMessage()
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
) {
246 .trace("(error when trying to decode: "
247 + e
.getMessage() + ")", 2);
248 getTraceHandler().trace(data
, 3);
249 getTraceHandler().trace("", 4);
252 getTraceHandler().trace("", 2);
257 * Start a bridge between 2 servers.
260 * an array containing:
262 * <li>The bridge name</li>
263 * <li>The bridge port</li>
264 * <li>TRUE for an SSL bridge, FALSE for plain text</li>
265 * <li>The forward server host</li>
266 * <li>The forward server port</li>
267 * <li>TRUE for an SSL forward server, FALSE for plain text</li>
268 * <li>(optional) a trace level</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])\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");
288 String name
= args
[i
++];
289 int port
= Integer
.parseInt(args
[i
++]);
290 boolean ssl
= Boolean
.parseBoolean(args
[i
++]);
291 String fhost
= args
[i
++];
292 int fport
= Integer
.parseInt(args
[i
++]);
293 boolean fssl
= Boolean
.parseBoolean(args
[i
++]);
296 if (args
.length
> 6) {
297 traceLevel
= Integer
.parseInt(args
[i
++]);
300 ServerBridge bridge
= new ServerBridge(name
, port
, ssl
, fhost
,
302 bridge
.setTraceHandler(new TraceHandler(true, true, traceLevel
));
304 } catch (Exception e
) {