Cleaning up drivers/Cpp, Cpp/Lifxtest, virtuals, and iotrmi/C++ (revisiting the C...

[iot2.git] / iotjava / iotrmi / C++ / IoTSocket.hpp

diff --git a/iotjava/iotrmi/C++/IoTSocket.hpp b/iotjava/iotrmi/C++/IoTSocket.hpp
index 3bbf74fa6f4ab2b963a8b8a661c8b212d5c56fdd..d967ded098860167ed9e97d94b93dd803efbc9f2 100644 (file)
--- a/iotjava/iotrmi/C++/IoTSocket.hpp
+++ b/iotjava/iotrmi/C++/IoTSocket.hpp
@@ -32,6 +32,7 @@ static const int MSG_LEN_SIZE = 4;
 #include <sys/socket.h>
 #include <sys/wait.h>
 #include <unistd.h>
+#include <mutex>
 
 #include "IoTRMIUtil.hpp"
 
@@ -40,25 +41,29 @@ static const int MSG_LEN_SIZE = 4;
 #define SD_SEND         0x01
 #define SD_BOTH         0x02
 
+mutex sendBytesMutex;
+mutex recvBytesMutex;
+mutex sendAckMutex;
+mutex recvAckMutex;
 
 class IoTSocket {
        public:
                IoTSocket(int iPort, bool* pResult);
                ~IoTSocket();
 
-               bool                            close();                                                                // Close the socket
-               bool                            sendBytes(char* pVals, int _iLen);              // Send a set of bytes
-               char*                           receiveBytes(char* pVals, int* len);    // Receive a set of bytes
+               bool                    close();                                // Close the socket
+               bool                    sendBytes(char* pVals, int _iLen);      // Send a set of bytes
+               char*                   receiveBytes(char* pVals, int* len);    // Receive a set of bytes
 
        protected:              
-               int                                     m_iPort;                                                        // Port I'm listening on
-               int                                     m_iSock;                                                        // Socket connection
-               struct sockaddr_in      m_addrRemote;                                           // Connector's address information
-               double*                         m_pBuffer;                                                      // Reuse the same memory for buffer
+               int                     m_iPort;                                // Port I'm listening on
+               int                     m_iSock;                                // Socket connection
+               struct sockaddr_in      m_addrRemote;                           // Connector's address information
+               double*                 m_pBuffer;                              // Reuse the same memory for buffer
 
        private:
-               bool                            receiveAck();
-               bool                            sendAck();
+               bool                    receiveAck();
+               bool                    sendAck();
 };
 
 
@@ -92,6 +97,9 @@ IoTSocket::~IoTSocket() {
 // Send bytes over the wire
 bool IoTSocket::sendBytes(char* pVals, int iLen) {
 
+       // Critical section that is used by different objects
+       lock_guard<mutex> guard(sendBytesMutex);
+
        int i = 0;
        char size[MSG_LEN_SIZE];
        // Convert int to byte array and fix endianness
@@ -106,6 +114,7 @@ bool IoTSocket::sendBytes(char* pVals, int iLen) {
                perror("IoTSocket: Send bytes error!");
                return false;
        }
+
 #ifdef DEBUG_ACK
        if (!receiveAck())
                return false;
@@ -121,12 +130,15 @@ bool IoTSocket::sendBytes(char* pVals, int iLen) {
 // Generate an array of char on the heap and return it
 char* IoTSocket::receiveBytes(char* pVals, int* len)
 {
-       int                     i                               = 0;
-       int                     j                               = 0;
-       char*           pTemp                   = NULL;
+       // Critical section that is used by different objects
+       lock_guard<mutex> guard(recvBytesMutex);
+
+       int                     i                       = 0;
+       int                     j                       = 0;
+       char*                   pTemp                   = NULL;
        int                     iTotalBytes             = 0;
        int                     iNumBytes               = 0;
-       bool            bEnd                    = false;
+       bool                    bEnd                    = false;
 
        int iTotal = 0;
        int iResult = 0;
@@ -136,6 +148,7 @@ char* IoTSocket::receiveBytes(char* pVals, int* len)
                iResult = recv(m_iSock, size, MSG_LEN_SIZE, 0);         
                iTotal += iResult;
        }
+
        if (iResult == -1) {
                perror("IoTSocket: Receive size error!");
                return NULL;
@@ -162,6 +175,7 @@ char* IoTSocket::receiveBytes(char* pVals, int* len)
                if (iTotalBytes == iLen)
                        bEnd = true;
        }
+
 #ifdef DEBUG_ACK
        if (!sendAck())
                return NULL;
@@ -187,6 +201,8 @@ bool IoTSocket::close()
 // Receive a short ack from the client 
 bool IoTSocket::receiveAck()
 {
+       // Critical section that is used by different objects
+       lock_guard<mutex> guard(recvAckMutex);
        char temp[1];
        int iTotal = 0;
        int iResult = 0;
@@ -208,6 +224,8 @@ bool IoTSocket::receiveAck()
 // Send a short ack to the client 
 bool IoTSocket::sendAck()
 {
+       // Critical section that is used by different objects
+       lock_guard<mutex> guard(sendAckMutex);
        char temp[1];
        temp[0] = 42;