3 import java.io.IOException;
4 import java.nio.ByteBuffer;
5 import java.util.Arrays;
6 import java.util.ArrayList;
7 import java.util.HashMap;
11 import java.lang.reflect.*;
13 import java.util.concurrent.*;
14 import java.util.concurrent.atomic.AtomicBoolean;
17 /** Class IoTRMICommClient is a class that extends IoTRMIComm
19 * This is a version of IoTRMIComm that sits on the main stub.
21 * @author Rahmadi Trimananda <rtrimana @ uci.edu>
25 public class IoTRMICommClient extends IoTRMIComm {
30 private IoTSocketClient rmiClientSend;
31 private IoTSocketClient rmiClientRecv;
35 * Constructor (for stub) - send and recv from the perspective of RMI socket servers
37 public IoTRMICommClient(int _localPortSend, int _localPortRecv, int _portSend, int _portRecv, String _address, int _rev) throws
38 ClassNotFoundException, InstantiationException,
39 IllegalAccessException, IOException {
42 rmiClientRecv = new IoTSocketClient(_localPortSend, _portSend, _address, _rev);
43 rmiClientSend = new IoTSocketClient(_localPortRecv, _portRecv, _address, _rev);
44 waitForPacketsOnClient();
49 * Constructor (for stub) - only destination port numbers
51 public IoTRMICommClient(int _portSend, int _portRecv, String _address, int _rev) throws
52 ClassNotFoundException, InstantiationException,
53 IllegalAccessException, IOException {
56 rmiClientRecv = new IoTSocketClient(_portSend, _address, _rev);
57 rmiClientSend = new IoTSocketClient(_portRecv, _address, _rev);
58 waitForPacketsOnClient();
63 * waitForPacketsOnClient() starts a thread that waits for packet bytes on client side
65 public void waitForPacketsOnClient() {
67 Thread thread = new Thread() {
69 byte[] packetBytes = null;
72 packetBytes = rmiClientRecv.receiveBytes(packetBytes);
73 if (packetBytes != null) {
74 int packetType = IoTRMIComm.getPacketType(packetBytes);
75 if (packetType == IoTRMIUtil.METHOD_TYPE) {
76 System.out.println("Method packet: " + Arrays.toString(packetBytes));
77 methodQueue.offer(packetBytes);
78 } else if (packetType == IoTRMIUtil.RET_VAL_TYPE) {
79 System.out.println("Return value packet: " + Arrays.toString(packetBytes));
80 returnQueue.offer(packetBytes);
82 throw new Error("IoTRMICommClient: Packet type is unknown: " + packetType);
86 } catch (Exception ex) {
88 throw new Error("IoTRMICommClient: Error receiving return value bytes on client!");
98 * sendReturnObj() for non-struct objects (client side)
100 public synchronized void sendReturnObj(Object retObj, byte[] methodBytes) {
102 // Send back return value
103 byte[] retObjBytes = IoTRMIUtil.getObjectBytes(retObj);
104 // Send return value together with OBJECT_ID and METHOD_ID for arbitration
105 int objMethIdLen = IoTRMIUtil.OBJECT_ID_LEN + IoTRMIUtil.METHOD_ID_LEN;
106 int headerLen = objMethIdLen + IoTRMIUtil.PACKET_TYPE_LEN;
107 byte[] retAllBytes = new byte[headerLen + retObjBytes.length];
108 // Copy OBJECT_ID and METHOD_ID
109 System.arraycopy(methodBytes, 0, retAllBytes, 0, objMethIdLen);
110 int packetType = IoTRMIUtil.RET_VAL_TYPE; // This is a return value
111 byte[] packetTypeBytes = IoTRMIUtil.intToByteArray(packetType);
112 System.arraycopy(packetTypeBytes, 0, retAllBytes, objMethIdLen, IoTRMIUtil.PACKET_TYPE_LEN);
113 // Copy array of bytes (return object)
114 System.arraycopy(retObjBytes, 0, retAllBytes, headerLen, retObjBytes.length);
116 rmiClientSend.sendBytes(retAllBytes);
117 } catch (IOException ex) {
118 ex.printStackTrace();
119 throw new Error("IoTRMICommClient: Error sending bytes in sendReturnObj()!");
125 * sendReturnObj() overloaded to send multiple return objects for structs (client side)
127 public synchronized void sendReturnObj(Class<?>[] retCls, Object[] retObj, byte[] methodBytes) {
129 // Send back return value
130 byte[] retObjBytes = returnToBytes(retCls, retObj);
131 // Send return value together with OBJECT_ID and METHOD_ID for arbitration
132 int objMethIdLen = IoTRMIUtil.OBJECT_ID_LEN + IoTRMIUtil.METHOD_ID_LEN;
133 int headerLen = objMethIdLen + IoTRMIUtil.PACKET_TYPE_LEN;
134 byte[] retAllBytes = new byte[headerLen + retObjBytes.length];
135 // Copy OBJECT_ID and METHOD_ID
136 System.arraycopy(methodBytes, 0, retAllBytes, 0, objMethIdLen);
137 int packetType = IoTRMIUtil.RET_VAL_TYPE; // This is a return value
138 byte[] packetTypeBytes = IoTRMIUtil.intToByteArray(packetType);
139 System.arraycopy(packetTypeBytes, 0, retAllBytes, objMethIdLen, IoTRMIUtil.PACKET_TYPE_LEN);
140 // Copy array of bytes (return object)
142 rmiClientSend.sendBytes(retAllBytes);
143 } catch (IOException ex) {
144 ex.printStackTrace();
145 throw new Error("IoTRMICommClient: Error sending bytes in sendReturnObj()!");
151 * remoteCall() calls a method remotely by passing in parameters (client side)
153 public synchronized void remoteCall(int objectId, int methodId, Class<?>[] paramCls, Object[] paramObj) {
156 byte[] methodBytes = methodToBytes(objectId, methodId, paramCls, paramObj);
158 rmiClientSend.sendBytes(methodBytes);
159 } catch (IOException ex) {
160 ex.printStackTrace();
161 throw new Error("IoTRMICommClient: Error when sending bytes in remoteCall()!");