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 Vigilia.
  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 final {
  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 LOGFILEPATH;                        // Log file path
  65                 const static string FILEPATH;                           // File path
  66                 const static string FILEEXT;                            // File extension
  67                 const static string SOEXT;                              // Shared object (.so) extension
  68                 const static string STRINGCLASS;                        // String class
  69                 const static string INTCLASS;                           // Int class
  70                 const static string CREATEFUNCTION;                     // The create function in class
  71                 const static string DESTROYFUNCTION;            // The destroy function in class
  72                 const static string INITFUNCTION;                       // The init function in class
  73                 const static string LOCALHOST;                          // String "localhost"
  74                 const static string SHELL;                                      // String ".sh"
  75
  76                 // Class properties
  77                 string serverAddress;
  78                 int serverPort;
  79                 string hostAddress;
  80                 string mainObjectName;
  81                 string objectName;
  82                 string objectClassName;
  83                 string objectInterfaceName;
  84                 string objectSkelClass;         // Need to send from Java IoTSlave: sMessage.getObjectInterfaceName() + SKEL_CLASS_SUFFIX
  85                 string objectStubClass;         // Need to send from Java IoTSlave: sMessage.getObjectStubInterfaceName() + STUB_CLASS_SUFFIX
  86                 int objectRegPort;
  87                 int objectStubPort;
  88                 string objectFieldName;                                         // Field name that is going to be initialized with IoTSet or IoTRelation
  89                 unordered_set<void*>* isetObject;                       // Set of object
  90                 IoTSet<void*>* iotsetObject;                            // IoTSet of object
  91                 vector<IoTSet<void*>*> vecIoTSet;                       // IoTSet of object
  92                 void* irelFirstObject;                                                  // First object of IoTRelation
  93                 void* irelSecondObject;                                                 // Second object of IoTRelation
  94                 unordered_multimap<void*,void*>* irelObject;    // Relation of object
  95                 IoTRelation<void*,void*>* iotrelObject;                 // IoTRelation of objects
  96                 vector<IoTRelation<void*,void*>*> vecIoTRel;    // IoTRelation of object
  97
  98                 TCPSocket* socket;
  99                 ofstream log;                                           // Log the messages
 100                 vector<string> args;                            // Hold the arguments for constructor (in string format)
 101                 vector<string> argClasses;                      // Hold the argument classes
 102                 //bool isDriverObject;                          // Set to true if this is IoTSlave instance for a driver object
 103                 void* objMainCls;                                       // Main class handler, i.e. driver or controller object
 104                 void* objSkelCls;                                       // Skeleton handler
 105                 void* objStubCls;                                       // Stub handler
 106                 unordered_map<string, void*> mapObjNameStub;    // Mapping between object name and stub
 107                 // Object handlers
 108                 create_t* create_object;
 109                 destroy_t* destroy_object;
 110                 init_t* init_object;
 111
 112         public:
 113                 // Constructors
 114                 IoTSlave(string _serverAddress, int _serverPort, string _objectName);
 115                 ~IoTSlave();
 116                 // Class methods
 117                 string getServerAddress();
 118                 int getServerPort();
 119                 string getObjectName();
 120                 void sendInteger(int intSend);
 121                 int recvInteger();
 122                 void sendString(string strSend);
 123                 string recvString();
 124                 string recvFile();
 125                 void unzipFile(string fileName);
 126                 // Main loop
 127                 void sendAck();
 128                 bool recvEndTransfer();
 129                 void commIoTMaster();
 130                 void createObject();            // Create driver object
 131                 void createMainObject();        // Create main object
 132                 void createNewIoTSet();
 133                 void createNewIoTRelation();
 134                 void getDeviceIoTSetObject();
 135                 void getIoTRelationFirstObject();
 136                 void getIoTRelationSecondObject();
 137                 void reinitializeIoTSetField();
 138                 void reinitializeIoTRelationField();
 139                 void getIoTSetObject();
 140                 void invokeInitMethod();
 141                 void createDriverObject();
 142                 void transferFile();
 143
 144         private:
 145                 // Private helper functions
 146                 int* byteToInt(int* result, char* bytes);
 147                 char* intToByteArray(int i, char* bytes);
 148                 char* recvIter(char* recvBuffer, int recvLen);
 149                 char* recvFileIter(char* recvBuffer, int recvLen);
 150                 void* getObjectConverted(void* retObj, string object, string objectClass);
 151                 void openFile(string fileName);
 152                 void writeToFile(string logMsg);
 153                 void closeFile();
 154                 void getObjectHandler(string objectClassName);
 155                 void instantiateMainObject();
 156                 void instantiateDriverObject();
 157                 void instantiateSkelObject();
 158                 void instantiateStubObject();
 159                 void runInitObject(IoTSlave* iotslave);
 160                 void getIoTSetRelationObject();
 161                 void createStub();
 162 };
 163
 164 // Constant initialization
 165 const string IoTSlave::LOGFILEPATH = "../log/";
 166 const string IoTSlave::FILEPATH = "./";
 167 const string IoTSlave::FILEEXT = "_cpp.log";
 168 const string IoTSlave::SOEXT = ".so";
 169 const string IoTSlave::STRINGCLASS = "string";
 170 const string IoTSlave::INTCLASS = "int";
 171 const string IoTSlave::CREATEFUNCTION = "create";
 172 const string IoTSlave::DESTROYFUNCTION = "destroy";
 173 const string IoTSlave::INITFUNCTION = "init";
 174 const string IoTSlave::LOCALHOST = "localhost";
 175 const string IoTSlave::SHELL = ".sh";
 176
 177 #endif