source: trunk/loci/jvmlink/cpp/JVMLinkClient.cpp @ 3834

Revision 3834, 11.0 KB checked in by curtis, 12 years ago (diff)

Add support for nulls.

Line 
1//
2// JVMLinkClient.cpp
3//
4
5/*
6JVMLink client/server architecture for communicating between Java and
7non-Java programs using sockets.
8Copyright (c) 2008 Hidayath Ansari and Curtis Rueden. All rights reserved.
9
10Redistribution and use in source and binary forms, with or without
11modification, are permitted provided that the following conditions are met:
12  * Redistributions of source code must retain the above copyright
13    notice, this list of conditions and the following disclaimer.
14  * Redistributions in binary form must reproduce the above copyright
15    notice, this list of conditions and the following disclaimer in the
16    documentation and/or other materials provided with the distribution.
17  * Neither the name of the UW-Madison LOCI nor the names of its
18    contributors may be used to endorse or promote products derived from
19    this software without specific prior written permission.
20
21THIS SOFTWARE IS PROVIDED BY THE UW-MADISON LOCI ``AS IS'' AND ANY
22EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
23WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
24DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDERS BE LIABLE FOR ANY
25DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
26(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
27LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
28ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
29(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
30SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
31*/
32
33#include "stdafx.h"
34#include "JVMLinkClient.h"
35#include <Windows.h>
36
37#define DEFAULT_PORT 20345
38
39//TODO: clear memory at appropriate points.
40
41JVMLinkClient::JVMLinkClient(void)
42{
43}
44
45JVMLinkClient::~JVMLinkClient(void)
46{
47}
48
49// -- Public API methods --
50
51void JVMLinkClient::startJava(int arg_port, CString classpath) {
52        port = arg_port == NULL ? DEFAULT_PORT : arg_port;
53        CString command;
54        // NB: Toggle comments to control debugging output for the server.
55        command.Format("-cp %s loci.jvmlink.JVMLinkServer %d", classpath, port);
56        //command.Format("-cp %s loci.jvmlink.JVMLinkServer -debug %d", classpath, port);
57        debug("java " << command);
58        ShellExecute(NULL, "open", "javaw.exe" , command, "", SW_SHOW);
59        //ShellExecute(NULL, "open", "java.exe" , command, "", SW_SHOW);
60}
61
62void JVMLinkClient::shutJava() {
63        debug("Terminating JVMLink server");
64        sendInt(EXIT_CMD);
65}
66
67JVMLinkClient::ConnectionCode JVMLinkClient::establishConnection() {
68        WSADATA wsaData;
69        struct hostent *hp;
70        unsigned int addr;
71        struct sockaddr_in server;
72        CString servername = "127.0.0.1"; 
73
74        int wsaret=WSAStartup(0x101,&wsaData);
75        if (wsaret) return WINSOCK_ERR;
76        debug("Initialized WinSock");
77
78        conn=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
79        if (conn==INVALID_SOCKET) return SOCKET_ERR;
80        debug("Socket created");
81
82        if (inet_addr(servername)==INADDR_NONE) {
83                hp=gethostbyname(servername);
84        }
85        else {
86                addr=inet_addr(servername);
87                hp=gethostbyaddr((char*)&addr,sizeof(addr),AF_INET);
88        }
89        if (hp == NULL) {
90                closesocket(conn);
91                debug("Could not resolve network address: " << servername);
92                return RESOLVE_ERR;
93        }
94        debug("Network address resolved");
95
96        server.sin_addr.s_addr=*((unsigned long*)hp->h_addr);
97        server.sin_family=AF_INET;
98        server.sin_port=htons(port);
99        if (connect(conn,(struct sockaddr*)&server,sizeof(server))) {
100                closesocket(conn);
101                debug("No server response on port " << port);
102                return RESPONSE_ERR;   
103        }
104        debug("Connected to server: " << servername);
105        return CONNECTION_SUCCESS;
106}
107
108int JVMLinkClient::closeConnection() {
109        debug("Closing connection");
110        shutdown(conn, SD_SEND);
111        closesocket(conn);
112        debug("Socket closed");
113        WSACleanup();
114        debug("De-initialized WinSock");
115        return CONNECTION_SUCCESS;
116}
117
118JVMLinkObject* JVMLinkClient::getVar(CString name) {
119        debug("getVar: requesting " << name);
120        JVMLinkObject* obj = new JVMLinkObject(name);
121        sendInt(GETVAR_CMD);
122        sendMessage(name);
123        obj->type = (Type) readInt();
124        if (obj->type == ARRAY_TYPE) {
125                obj->insideType = (Type) readInt();
126                obj->length = readInt();
127                if (obj->insideType == STRING_TYPE) {
128                        CString* s = new CString[obj->length];
129                        for (int i=0; i<obj->length; i++) s[i] = *readString();
130                        obj->data = s;
131                }
132                else {
133                        obj->size = readInt();
134                        obj->data = readMessage(obj->size * obj->length);
135                }
136                debug("getVar: got array: length=" << obj->length << ", type=" << obj->insideType);
137        }
138        else if (obj->type == STRING_TYPE) {   
139                obj->data = readString();
140                obj->size = 0;
141                debug("getVar: got string: length=" << len << ", value=" << buf);
142        }
143        else if (obj->type == NULL_TYPE) {
144                obj->data = NULL;
145                obj->size = 0;
146                debug("getVar: got NULL value");
147        }
148        else {
149                int size = readInt();
150                obj->data = readMessage(size);
151                obj->size = size;
152                obj->insideType = NULL_TYPE;
153                debug("getVar: got object: type=" << obj->type << ", size=" << obj->size);
154        }
155        return obj;
156}
157
158void JVMLinkClient::exec(CString command) {
159        debug("exec: " << command);
160        sendInt(EXEC_CMD);
161        sendMessage(command);
162}
163
164void JVMLinkClient::setVar(JVMLinkObject* obj) {
165        sendInt(SETVAR_CMD);
166        sendMessage(obj->name);
167        sendInt((int) obj->type);
168        if (obj->type == ARRAY_TYPE) {
169                sendInt((int) obj->insideType);
170                sendInt(obj->length);
171                if (obj->insideType == STRING_TYPE) {
172                        CString* s = (CString*) obj->data;
173                        for (int i=0; i<obj->length; i++) {
174                                sendMessage(s[i]);
175                        }
176                }
177                else {
178                        int sent = 0;
179                        int total = obj->size * obj->length;
180                        char* buf = (char*) obj->data;
181                        while (sent < total) {
182                                sent += send(conn, buf + sent, total - sent, 0);
183                        }
184                }
185        }
186        else {
187                if (obj->type == STRING_TYPE) sendMessage(*(CString*) obj->data);
188                else send(conn, (char*) obj->data, obj->size, 0);
189        }
190}
191
192void JVMLinkClient::setVar(CString argname, int obj) {
193        debug("setVar: " << argname << " = " << obj << " (int)");
194        JVMLinkObject* jvmObj = new JVMLinkObject(argname, INT_TYPE, &obj);
195        setVar(jvmObj);
196        delete jvmObj;
197}
198
199void JVMLinkClient::setVar(CString argname, int* obj, int length) {
200        debug("setVar: " << argname << " (int array)");
201        JVMLinkObject* jvmObj = new JVMLinkObject(argname, INT_TYPE, length, obj);
202        setVar(jvmObj);
203        delete jvmObj;
204}
205
206void JVMLinkClient::setVar(CString argname, CString* obj) {
207        debug("setVar: " << argname << " = " << obj << " (string)");
208        JVMLinkObject* jvmObj = new JVMLinkObject(argname, STRING_TYPE, obj);
209        setVar(jvmObj);
210        delete jvmObj;
211}
212
213void JVMLinkClient::setVar(CString argname, CString* obj, int length) {
214        debug("setVar: " << argname << " (string array)");
215        JVMLinkObject* jvmObj = new JVMLinkObject(argname, STRING_TYPE, length, obj);
216        setVar(jvmObj);
217        delete jvmObj;
218}
219
220void JVMLinkClient::setVar(CString argname, char obj) {
221        debug("setVar: " << argname << " = " << obj << " (char)");
222        JVMLinkObject* jvmObj = new JVMLinkObject(argname, CHAR_TYPE, &obj);
223        setVar(jvmObj);
224        delete jvmObj;
225}
226
227void JVMLinkClient::setVar(CString argname, char* obj, int length) {
228        debug("setVar: " << argname << " (char array)");
229        JVMLinkObject* jvmObj = new JVMLinkObject(argname, CHAR_TYPE, length, obj);
230        setVar(jvmObj);
231        delete jvmObj;
232}
233
234void JVMLinkClient::setVar(CString argname, Byte obj) {
235        debug("setVar: " << argname << " = " << obj.data << " (byte)");
236        JVMLinkObject* jvmObj = new JVMLinkObject(argname, BYTE_TYPE, &obj);
237        setVar(jvmObj);
238        delete jvmObj;
239}
240
241void JVMLinkClient::setVar(CString argname, Byte* obj, int length) {
242        debug("setVar: " << argname << " (byte array)");
243        JVMLinkObject* jvmObj = new JVMLinkObject(argname, BYTE_TYPE, length, obj);
244        setVar(jvmObj);
245        delete jvmObj;
246}
247
248void JVMLinkClient::setVar(CString argname, float obj) {
249        debug("setVar: " << argname << " = " << obj << " (float)");
250        JVMLinkObject* jvmObj = new JVMLinkObject(argname, FLOAT_TYPE, &obj);
251        setVar(jvmObj);
252        delete jvmObj;
253}
254
255void JVMLinkClient::setVar(CString argname, float* obj, int length) {
256        debug("setVar: " << argname << " (float array)");
257        JVMLinkObject* jvmObj = new JVMLinkObject(argname, FLOAT_TYPE, length, obj);
258        setVar(jvmObj);
259        delete jvmObj;
260}
261
262void JVMLinkClient::setVar(CString argname, bool obj) {
263        debug("setVar: " << argname << " = " << obj << " (bool)");
264        JVMLinkObject* jvmObj = new JVMLinkObject(argname, BOOL_TYPE, &obj);
265        setVar(jvmObj);
266        delete jvmObj;
267}
268
269void JVMLinkClient::setVar(CString argname, bool* obj, int length) {
270        debug("setVar: " << argname << " (bool array)");
271        JVMLinkObject* jvmObj = new JVMLinkObject(argname, BOOL_TYPE, length, obj);
272        setVar(jvmObj);
273        delete jvmObj;
274}
275
276void JVMLinkClient::setVar(CString argname, double obj) {
277        debug("setVar: " << argname << " = " << obj << " (double)");
278        JVMLinkObject* jvmObj = new JVMLinkObject(argname, DOUBLE_TYPE, &obj);
279        setVar(jvmObj);
280        delete jvmObj;
281}
282
283void JVMLinkClient::setVar(CString argname, double* obj, int length) {
284        debug("setVar: " << argname << " (double array)");
285        JVMLinkObject* jvmObj = new JVMLinkObject(argname, DOUBLE_TYPE, length, obj);
286        setVar(jvmObj);
287        delete jvmObj;
288}
289
290void JVMLinkClient::setVar(CString argname, long long obj) {
291        debug("setVar: " << argname << " = " << obj << " (long)");
292        JVMLinkObject* jvmObj = new JVMLinkObject(argname, LONG_TYPE, &obj);
293        setVar(jvmObj);
294        delete jvmObj;
295}
296
297void JVMLinkClient::setVar(CString argname, long long* obj, int length) {
298        debug("setVar: " << argname << " (long array)");
299        JVMLinkObject* jvmObj = new JVMLinkObject(argname, LONG_TYPE, length, obj);
300        setVar(jvmObj);
301        delete jvmObj;
302}
303
304void JVMLinkClient::setVar(CString argname, short obj) {
305        debug("setVar: " << argname << " = " << obj << " (short)");
306        JVMLinkObject* jvmObj = new JVMLinkObject(argname, SHORT_TYPE, &obj);
307        setVar(jvmObj);
308        delete jvmObj;
309}
310
311void JVMLinkClient::setVar(CString argname, short* obj, int length) {
312        debug("setVar: " << argname << " (short array)");
313        JVMLinkObject* jvmObj = new JVMLinkObject(argname, SHORT_TYPE, length, obj);
314        setVar(jvmObj);
315        delete jvmObj;
316}
317
318void JVMLinkClient::setVarNull(CString argname) {
319        debug("setVarNull: " << argname);
320        JVMLinkObject* jvmObj = new JVMLinkObject(argname, NULL_TYPE, NULL);
321        setVar(jvmObj);
322        delete jvmObj; 
323}
324
325// -- Private methods --
326
327void JVMLinkClient::sendMessage(CString message) {
328        int sent = 0;
329        char* buf = (char*) (LPCTSTR) message;
330        int total = message.GetLength();
331        sendInt(total);
332        while (sent < total) sent += send(conn, buf + sent, total - sent, 0);
333}
334
335void JVMLinkClient::sendInt(int value) {
336        char* buf = (char*) (&value);
337        send(conn, buf, 4, 0);
338}
339
340void* JVMLinkClient::readMessage(int size) {
341        int read = 0;
342        char* buf = (char*) malloc(size);
343        while (read < size) read += recv(conn, buf + read, size - read, 0);
344        return (void*) buf;
345}
346
347int JVMLinkClient::readInt() {
348        return *(int*) readMessage(4);
349}
350
351CString* JVMLinkClient::readString() {
352        int read = 0;
353        int total = readInt();
354        char* buf = new char[total + 1];
355        while (read < total) read += recv(conn, buf + read, total - read, 0);
356        buf[total] = '\0';
357        return new CString(buf);
358}
Note: See TracBrowser for help on using the repository browser.