iotjava/iotruntime/cpp/iotslave/IoTSlave.hpp

   1 #ifndef _IOTSLAVE_HPP__
   2 #define _IOTSLAVE_HPP__
   3
   4 #include <iostream>
   5 #include <fstream>
   6 #include <vector>
   7 #include <thread>
   8 #include <cstdlib>
   9
  10 #include <dlfcn.h>              // For dlopen, dlsym, etc.
  11
  12 #include "IoTSet.hpp"
  13 #include "IoTDeviceAddress.hpp"
  14 #include "IoTRelation.hpp"
  15 #include "Socket.cpp"
  16
  17 /** Class IoTSlave is a communication class
  18  *  that interacts with IoTSlave.java to set up C++
  19  *  objects in Sentinel.
  20  *
  21  * @author      Rahmadi Trimananda <rtrimana @ uci.edu>
  22  * @version     1.0
  23  * @since       2017-01-12
  24  */
  25 // Enumeration of master-slave communication codes
  26 enum IoTCommCode {
  27
  28         ACKNOWLEDGED,
  29         CREATE_DRIVER_OBJECT,
  30         CREATE_OBJECT,
  31         CREATE_MAIN_OBJECT,
  32         CREATE_NEW_IOTSET,
  33         CREATE_NEW_IOTRELATION,
  34         END_TRANSFER,
  35         END_SESSION,
  36         GET_ADD_IOTSET_OBJECT,
  37         GET_DEVICE_IOTSET_OBJECT,
  38         GET_IOTSET_OBJECT,
  39         GET_IOTRELATION_FIRST_OBJECT,
  40         GET_IOTRELATION_SECOND_OBJECT,
  41         GET_ZB_DEV_IOTSET_OBJECT,
  42         INVOKE_INIT_METHOD,
  43         REINITIALIZE_IOTSET_FIELD,
  44         REINITIALIZE_IOTRELATION_FIELD,
  45         TRANSFER_FILE,
  46
  47 };
  48
  49
  50 // Defining generic function pointers for
  51 // create, destroy, and init functions of each class object
  52 typedef void* create_t(void**);
  53 typedef void destroy_t(void*);
  54 typedef void init_t(void*);
  55
  56
  57 class IoTSlave {
  58
  59         private:
  60                 // Constants
  61                 const static int RCVBUFSIZE = 1024;                     // Size of receive buffer
  62                 const static int SKELPARAMSIZE = 3;                     // Number of params for skeleton
  63                 const static int STUBPARAMSIZE = 5;                     // Number of params for stub
  64                 const static string FILEPATH;                           // File path
  65                 const static string FILEEXT;                            // File extension
  66                 const static string SOEXT;                              // Shared object (.so) extension
  67                 const static string STRINGCLASS;                        // String class
  68                 const static string INTCLASS;                           // Int class
  69                 const static string CREATEFUNCTION;                     // The create function in class
  70                 const static string DESTROYFUNCTION;            // The destroy function in class
  71                 const static string INITFUNCTION;                       // The init function in class
  72                 const static string LOCALHOST;                          // String "localhost"
  73                 const static string SHELL;                                      // String ".sh"
  74
  75                 // Class properties
  76                 string serverAddress;
  77                 int serverPort;
  78                 string hostAddress;
  79                 string mainObjectName;
  80                 string objectName;
  81                 string objectClassName;
  82                 string objectInterfaceName;
  83                 string objectSkelClass;         // Need to send from Java IoTSlave: sMessage.getObjectInterfaceName() + SKEL_CLASS_SUFFIX
  84                 string objectStubClass;         // Need to send from Java IoTSlave: sMessage.getObjectStubInterfaceName() + STUB_CLASS_SUFFIX
  85                 int objectRegPort;
  86                 int objectStubPort;
  87                 string objectFieldName;                                         // Field name that is going to be initialized with IoTSet or IoTRelation
  88                 unordered_set<void*>* isetObject;                       // Set of object
  89                 IoTSet<void*>* iotsetObject;                            // IoTSet of object
  90                 vector<IoTSet<void*>*> vecIoTSet;                       // IoTSet of object
  91                 void* irelFirstObject;                                                  // First object of IoTRelation
  92                 void* irelSecondObject;                                                 // Second object of IoTRelation
  93                 unordered_multimap<void*,void*>* irelObject;    // Relation of object
  94                 IoTRelation<void*,void*>* iotrelObject;                 // IoTRelation of objects
  95                 vector<IoTRelation<void*,void*>*> vecIoTRel;    // IoTRelation of object
  96
  97                 TCPSocket* socket;
  98                 ofstream log;                                           // Log the messages
  99                 vector<string> args;                            // Hold the arguments for constructor (in string format)
 100                 vector<string> argClasses;                      // Hold the argument classes
 101                 //bool isDriverObject;                          // Set to true if this is IoTSlave instance for a driver object
 102                 void* objMainCls;                                       // Main class handler, i.e. driver or controller object
 103                 void* objSkelCls;                                       // Skeleton handler
 104                 void* objStubCls;                                       // Stub handler
 105                 unordered_map<string, void*> mapObjNameStub;    // Mapping between object name and stub
 106                 // Object handlers
 107                 create_t* create_object;
 108                 destroy_t* destroy_object;
 109                 init_t* init_object;
 110
 111         public:
 112                 // Constructors
 113                 IoTSlave(string _serverAddress, int _serverPort, string _objectName);
 114                 ~IoTSlave();
 115                 // Class methods
 116                 string getServerAddress();
 117                 int getServerPort();
 118                 string getObjectName();
 119                 void sendInteger(int intSend);
 120                 int recvInteger();
 121                 void sendString(string strSend);
 122                 string recvString();
 123                 string recvFile();
 124                 void unzipFile(string fileName);
 125                 // Main loop
 126                 void sendAck();
 127                 bool recvEndTransfer();
 128                 void commIoTMaster();
 129                 void createObject();            // Create driver object
 130                 void createMainObject();        // Create main object
 131                 void createNewIoTSet();
 132                 void createNewIoTRelation();
 133                 void getDeviceIoTSetObject();
 134                 void getIoTRelationFirstObject();
 135                 void getIoTRelationSecondObject();
 136                 void reinitializeIoTSetField();
 137                 void reinitializeIoTRelationField();
 138                 void getIoTSetObject();
 139                 void invokeInitMethod();
 140                 void createDriverObject();
 141                 void transferFile();
 142
 143         private:
 144                 // Private helper functions
 145                 int* byteToInt(int* result, char* bytes);
 146                 char* intToByteArray(int i, char* bytes);
 147                 char* recvIter(char* recvBuffer, int recvLen);
 148                 char* recvFileIter(char* recvBuffer, int recvLen);
 149                 void* getObjectConverted(void* retObj, string object, string objectClass);
 150                 void openFile(string fileName);
 151                 void writeToFile(string logMsg);
 152                 void closeFile();
 153                 void getObjectHandler(string objectClassName);
 154                 void instantiateMainObject();
 155                 void instantiateDriverObject();
 156                 void instantiateSkelObject();
 157                 void instantiateStubObject();
 158                 void runInitObject(IoTSlave* iotslave);
 159                 void getIoTSetRelationObject();
 160                 void createStub();
 161 };
 162
 163 // Constant initialization
 164 const string IoTSlave::FILEPATH = "./";
 165 const string IoTSlave::FILEEXT = "_cpp.log";
 166 const string IoTSlave::SOEXT = ".so";
 167 const string IoTSlave::STRINGCLASS = "string";
 168 const string IoTSlave::INTCLASS = "int";
 169 const string IoTSlave::CREATEFUNCTION = "create";
 170 const string IoTSlave::DESTROYFUNCTION = "destroy";
 171 const string IoTSlave::INITFUNCTION = "init";
 172 const string IoTSlave::LOCALHOST = "localhost";
 173 const string IoTSlave::SHELL = ".sh";
 174
 175 #endif