Bug fixes + tabbing
[iotcloud.git] / version2 / src / C / CloudComm.cc
index 06d6a89459624c18aded6691f349ad637cdfb064..c20488dd268d59632d09634068961b918a9a878c 100644 (file)
@@ -6,8 +6,16 @@
 #include "URL.h"
 #include "Mac.h"
 #include "Table.h"
+#include "Slot.h"
 #include "Crypto.h"
 #include "ByteBuffer.h"
+#include "aes.h"
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <arpa/inet.h>
+#include <netinet/tcp.h>
+#include <unistd.h>
+#include <netdb.h>
 
 /**
  * Empty Constructor needed for child class.
@@ -21,13 +29,14 @@ CloudComm::CloudComm() :
        salt(NULL),
        table(NULL),
        listeningPort(-1),
-       localServerThread(NULL),
        doEnd(false),
-       timer(TimingSingleton_getInstance())
+       timer(TimingSingleton_getInstance()),
+       getslot(new Array<char>("getslot", 7)),
+       putslot(new Array<char>("putslot", 7))
 {
 }
 
-void * threadWrapper(void * cloud) {
+void *threadWrapper(void *cloud) {
        CloudComm *c = (CloudComm *) cloud;
        c->localServerWorkerFunction();
        return NULL;
@@ -40,28 +49,35 @@ CloudComm::CloudComm(Table *_table,  IoTString *_baseurl, IoTString *_password,
        baseurl(_baseurl),
        key(NULL),
        mac(NULL),
-       password(_password),
+       password(new IoTString(_password)),
        random(new SecureRandom()),
        salt(NULL),
        table(_table),
        listeningPort(_listeningPort),
-       localServerThread(NULL),
        doEnd(false),
-       timer(TimingSingleton_getInstance()) {
+       timer(TimingSingleton_getInstance()),
+       getslot(new Array<char>("getslot", 7)),
+       putslot(new Array<char>("putslot", 7)) {
        if (listeningPort > 0) {
                pthread_create(&localServerThread, NULL, threadWrapper, this);
        }
 }
 
+CloudComm::~CloudComm() {
+       delete random;
+       delete getslot;
+       delete putslot;
+}
+
 /**
  * Generates Key from password.
  */
 AESKey *CloudComm::initKey() {
        try {
-               AESKey * key = new AESKey(password->internalBytes(),
-                                                                                                                       salt,
-                                                                                                                       65536,
-                                                                                                                       128);
+               AESKey *key = new AESKey(password->internalBytes(),
+                                                                                                                salt,
+                                                                                                                65536,
+                                                                                                                128);
                return key;
        } catch (Exception *e) {
                throw new Error("Failed generating key.");
@@ -91,8 +107,9 @@ void CloudComm::initCrypt() {
        }
        try {
                key = initKey();
+               delete password;
                password = NULL;// drop password
-               mac = Mac_getInstance("HmacSHA256");
+               mac = new Mac();
                mac->init(key);
        } catch (Exception *e) {
                throw new Error("Failed To Initialize Ciphers");
@@ -105,7 +122,7 @@ void CloudComm::initCrypt() {
 IoTString *CloudComm::buildRequest(bool isput, int64_t sequencenumber, int64_t maxentries) {
        const char *reqstring = isput ? "req=putslot" : "req=getslot";
        char *buffer = (char *) malloc(baseurl->length() + 200);
-       memcpy(buffer, baseurl->internalBytes(), baseurl->length());
+       memcpy(buffer, baseurl->internalBytes()->internalArray(), baseurl->length());
        int offset = baseurl->length();
        offset += sprintf(&buffer[offset], "?%s&seq=%" PRId64, reqstring, sequencenumber);
        if (maxentries != 0)
@@ -114,22 +131,257 @@ IoTString *CloudComm::buildRequest(bool isput, int64_t sequencenumber, int64_t m
        return urlstr;
 }
 
-int openURL(IoTString *url, bool isPost) {
-       return 0;
+void loopWrite(int fd, char *array, int bytestowrite) {
+       int byteswritten = 0;
+       while (bytestowrite) {
+               int bytes = write(fd, &array[byteswritten], bytestowrite);
+               if (bytes >= 0) {
+                       byteswritten += bytes;
+                       bytestowrite -= bytes;
+               } else {
+                       printf("Error in write\n");
+                       exit(-1);
+               }
+       }
 }
 
-void writeURLData(int fd, Array<char> *data) {
+void loopRead(int fd, char *array, int bytestoread) {
+       int bytesread = 0;
+       while (bytestoread) {
+               int bytes = read(fd, &array[bytesread], bytestoread);
+               if (bytes >= 0) {
+                       bytesread += bytes;
+                       bytestoread -= bytes;
+               } else {
+                       printf("Error in read\n");
+                       exit(-1);
+               }
+       }
 }
 
-void readURLData(int fd, Array<char> * output) {
+WebConnection openURL(IoTString *url) {
+       if (url->length() < 7 || memcmp(url->internalBytes()->internalArray(), "http://", 7)) {
+               printf("BOGUS URL\n");
+               exit(-1);
+       }
+       int i = 7;
+       for (; i < url->length(); i++)
+               if (url->get(i) == '/')
+                       break;
+
+       if ( i == url->length()) {
+               printf("ERROR in openURL\n");
+               exit(-1);
+       }
+
+       char *host = (char *) malloc(i - 6);
+       memcpy(host, &url->internalBytes()->internalArray()[7], i - 7);
+       host[i - 7] = 0;
+       printf("%s\n", host);
+
+       char *message = (char *)malloc(sizeof("POST  HTTP/1.1\r\n") + sizeof("Host: \r\n") + 2 * url->length());
+
+       /* fill in the parameters */
+       int post = sprintf(message,"POST ");
+       /* copy data */
+       memcpy(&message[post], &url->internalBytes()->internalArray()[i], url->length() - i);
+       int endpost = sprintf(&message[post + url->length() - i], " HTTP/1.1\r\n");
+
+       int hostlen = sprintf(&message[endpost + post + url->length() - i], "Host: ");
+       memcpy(&message[endpost + post + url->length() + hostlen - i], host, i - 7);
+       sprintf(&message[endpost + post + url->length() + hostlen - 7], "\r\n");
+
+       /* create the socket */
+       int sockfd = socket(AF_INET, SOCK_STREAM, 0);
+       if (sockfd < 0) {printf("ERROR opening socket\n"); exit(-1);}
+
+       /* lookup the ip address */
+       struct hostent *server = gethostbyname(host);
+       free(host);
+
+       if (server == NULL) {printf("ERROR, no such host"); exit(-1);}
+
+       /* fill in the structure */
+       struct sockaddr_in serv_addr;
+
+       memset(&serv_addr,0,sizeof(serv_addr));
+       serv_addr.sin_family = AF_INET;
+       serv_addr.sin_port = htons(80);
+       memcpy(&serv_addr.sin_addr.s_addr,server->h_addr,server->h_length);
+
+       /* connect the socket */
+       if (connect(sockfd,(struct sockaddr *)&serv_addr,sizeof(serv_addr)) < 0) {
+               printf("ERROR connecting");
+               exit(-1);
+       }
+
+       /* send the request */
+       int total = strlen(message);
+       loopWrite(sockfd, message, total);
+       return (WebConnection) {sockfd, -1};
 }
 
-int readURLInt(int fd) {
-       return 0;
+int createSocket(IoTString *name, int port) {
+       char *host = (char *) malloc(name->length() + 1);
+       memcpy(host, name->internalBytes()->internalArray(), name->length());
+       host[name->length()] = 0;
+       printf("%s\n", host);
+       /* How big is the message? */
+
+       /* create the socket */
+       int sockfd = socket(AF_INET, SOCK_STREAM, 0);
+       if (sockfd < 0) {printf("ERROR opening socket\n"); exit(-1);}
+
+       /* lookup the ip address */
+       struct hostent *server = gethostbyname(host);
+       free(host);
+
+       if (server == NULL) {printf("ERROR, no such host"); exit(-1);}
+
+       /* fill in the structure */
+       struct sockaddr_in serv_addr;
+
+       memset(&serv_addr,0,sizeof(serv_addr));
+       serv_addr.sin_family = AF_INET;
+       serv_addr.sin_port = htons(port);
+       memcpy(&serv_addr.sin_addr.s_addr,server->h_addr,server->h_length);
+
+       /* connect the socket */
+       if (connect(sockfd,(struct sockaddr *)&serv_addr,sizeof(serv_addr)) < 0) {
+               printf("ERROR connecting");
+               exit(-1);
+       }
+
+       return sockfd;
 }
 
-int getResponseCode(int fd) {
-       return 0;
+int createSocket(int port) {
+       int fd;
+       struct sockaddr_in sin;
+
+       bzero(&sin, sizeof(sin));
+       sin.sin_family = AF_INET;
+       sin.sin_port = htons(port);
+       sin.sin_addr.s_addr = htonl(INADDR_ANY);
+       fd = socket(AF_INET, SOCK_STREAM, 0);
+       int n = 1;
+       if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, (char *)&n, sizeof (n)) < 0) {
+               close(fd);
+               printf("Create Socket Error\n");
+               exit(-1);
+       }
+       if (bind(fd, (struct sockaddr *) &sin, sizeof(sin)) < 0) {
+               close(fd);
+               exit(-1);
+       }
+       if (listen(fd, 5) < 0) {
+               close(fd);
+               exit(-1);
+       }
+       return fd;
+}
+
+int acceptSocket(int socket) {
+       struct sockaddr_in sin;
+       unsigned int sinlen = sizeof(sin);
+       int newfd = accept(socket, (struct sockaddr *)&sin, &sinlen);
+       int flag = 1;
+       setsockopt(newfd, IPPROTO_TCP, TCP_NODELAY, (char *) &flag, sizeof(flag));
+       if (newfd < 0) {
+               printf("Accept Error\n");
+               exit(-1);
+       }
+       return newfd;
+}
+
+void writeSocketData(int fd, Array<char> *data) {
+       loopWrite(fd, data->internalArray(), data->length());
+}
+
+void writeSocketInt(int fd, int32_t value) {
+       char array[4];
+       array[0] = value >> 24;
+       array[1] = (value >> 16) & 0xff;
+       array[2] = (value >> 8) & 0xff;
+       array[3] = value & 0xff;
+       loopWrite(fd, array, 4);
+}
+
+int readSocketInt(int fd) {
+       char array[4];
+       loopRead(fd, array, 4);
+       return (((int32_t)(unsigned char) array[0]) << 24) |
+                                (((int32_t)(unsigned char) array[1]) << 16) |
+                                (((int32_t)(unsigned char) array[2]) << 8) |
+                                ((int32_t)(unsigned char) array[3]);
+}
+
+void readSocketData(int fd, Array<char> *data) {
+       loopRead(fd, data->internalArray(), data->length());
+}
+
+void writeURLDataAndClose(WebConnection *wc, Array<char> *data) {
+       dprintf(wc->fd, "Content-Length: %d\r\n\r\n", data->length());
+       loopWrite(wc->fd, data->internalArray(), data->length());
+}
+
+void closeURLReq(WebConnection *wc) {
+       dprintf(wc->fd, "\r\n");
+}
+
+void readURLData(WebConnection *wc, Array<char> *output) {
+       loopRead(wc->fd, output->internalArray(), output->length());
+}
+
+int readURLInt(WebConnection *wc) {
+       char array[4];
+       loopRead(wc->fd, array, 4);
+       return (((int32_t)(unsigned char) array[0]) << 24) |
+                                (((int32_t)(unsigned char) array[1]) << 16) |
+                                (((int32_t)(unsigned char) array[2]) << 8) |
+                                ((int32_t)(unsigned char) array[3]);
+}
+
+void readLine(WebConnection *wc, char *response, int numBytes) {
+       int offset = 0;
+       char newchar;
+       while (true) {
+               int bytes = read(wc->fd, &newchar, 1);
+               if (bytes <= 0)
+                       break;
+               if (offset == (numBytes - 1)) {
+                       printf("Response too long");
+                       exit(-1);
+               }
+               response[offset++] = newchar;
+               if (newchar == '\n')
+                       break;
+       }
+       response[offset] = 0;
+}
+
+int getResponseCode(WebConnection *wc) {
+       char response[600];
+       readLine(wc, response, sizeof(response));
+       int ver1 = 0, ver2 = 0, respcode = 0;
+       sscanf(response, "HTTP/%d.%d %d", &ver1, &ver2, &respcode);
+       printf("Response code %d\n", respcode);
+       return respcode;
+}
+
+void readHeaders(WebConnection *wc) {
+       char response[600];
+       int numBytes;
+
+       while (true) {
+               readLine(wc, response, sizeof(response));
+               if (response[0] == '\r')
+                       return;
+               else if (memcmp(response, "Content-Length:", sizeof("Content-Length:") - 1) == 0) {
+                       sscanf(response, "Content-Length: %d", &numBytes);
+                       wc->numBytes = numBytes;
+               }
+       }
 }
 
 void CloudComm::setSalt() {
@@ -138,26 +390,27 @@ void CloudComm::setSalt() {
                return;
        }
 
-       int fd = -1;
+       WebConnection wc = {-1, -1};
        try {
                Array<char> *saltTmp = new Array<char>(CloudComm_SALT_SIZE);
                random->nextBytes(saltTmp);
 
                char *buffer = (char *) malloc(baseurl->length() + 100);
-               memcpy(buffer, baseurl->internalBytes(), baseurl->length());
+               memcpy(buffer, baseurl->internalBytes()->internalArray(), baseurl->length());
                int offset = baseurl->length();
                offset += sprintf(&buffer[offset], "?req=setsalt");
                IoTString *urlstr = new IoTString(buffer);
                free(buffer);
 
                timer->startTime();
-               fd = openURL(urlstr, true);
-               writeURLData(fd, saltTmp);
+               wc = openURL(urlstr);
+               writeURLDataAndClose(&wc, saltTmp);
 
-               int responsecode = getResponseCode(fd);
+               int responsecode = getResponseCode(&wc);
                if (responsecode != HttpURLConnection_HTTP_OK) {
                        throw new Error("Invalid response");
                }
+               close(wc.fd);
 
                timer->endTime();
                salt = saltTmp;
@@ -168,12 +421,12 @@ void CloudComm::setSalt() {
 }
 
 bool CloudComm::getSalt() {
-       int fd = -1;
+       WebConnection wc = {-1, -1};
        IoTString *urlstr = NULL;
-       
+
        try {
                char *buffer = (char *) malloc(baseurl->length() + 100);
-               memcpy(buffer, baseurl->internalBytes(), baseurl->length());
+               memcpy(buffer, baseurl->internalBytes()->internalArray(), baseurl->length());
                int offset = baseurl->length();
                offset += sprintf(&buffer[offset], "?req=getsalt");
                urlstr = new IoTString(buffer);
@@ -183,7 +436,8 @@ bool CloudComm::getSalt() {
        }
        try {
                timer->startTime();
-               fd = openURL(urlstr, true);
+               wc = openURL(urlstr);
+               closeURLReq(&wc);
                timer->endTime();
        } catch (SocketTimeoutException *e) {
                timer->endTime();
@@ -194,21 +448,26 @@ bool CloudComm::getSalt() {
 
        try {
                timer->startTime();
-               int responsecode = getResponseCode(fd);
+               int responsecode = getResponseCode(&wc);
+               readHeaders(&wc);
                if (responsecode != HttpURLConnection_HTTP_OK) {
                        throw new Error("Invalid response");
                }
-               if (is->available() > 0) {
-                       int salt_length = readURLInt(fd);
-                       Array<char> *tmp = new Array<char>(salt_length);
-                       readURLData(fd, tmp);
-                       salt = tmp;
-                       timer->endTime();
-                       return true;
-               } else {
+               if (wc.numBytes == 0) {
                        timer->endTime();
+                       close(wc.fd);
                        return false;
                }
+
+
+               int salt_length = readURLInt(&wc);
+               Array<char> *tmp = new Array<char>(salt_length);
+               readURLData(&wc, tmp);
+               close(wc.fd);
+
+               salt = tmp;
+               timer->endTime();
+               return true;
        } catch (SocketTimeoutException *e) {
                timer->endTime();
                throw new ServerException("getSalt failed", ServerException_TypeInputTimeout);
@@ -225,12 +484,24 @@ Array<char> *CloudComm::createIV(int64_t machineId, int64_t localSequenceNumber)
        return buffer->array();
 }
 
+Array<char> *AESEncrypt(Array<char> *ivBytes, AESKey *key, Array<char> *data) {
+       Array<char> *output = new Array<char>(data->length());
+       aes_encrypt_ctr((BYTE *)data->internalArray(), data->length(), (BYTE *) output->internalArray(), (WORD *)key->getKeySchedule(), key->getKey()->length() * 8, (BYTE *)ivBytes->internalArray());
+       return output;
+}
+
+Array<char> *AESDecrypt(Array<char> *ivBytes, AESKey *key, Array<char> *data) {
+       Array<char> *output = new Array<char>(data->length());
+       aes_decrypt_ctr((BYTE *)data->internalArray(), data->length(), (BYTE *)output->internalArray(), (WORD *)key->getKeySchedule(), key->getKey()->length() * 8, (BYTE *)ivBytes->internalArray());
+       return output;
+}
+
 Array<char> *CloudComm::encryptSlotAndPrependIV(Array<char> *rawData, Array<char> *ivBytes) {
        try {
-               IvParameterSpec *ivSpec = new IvParameterSpec(ivBytes);
-               Cipher *cipher = Cipher_getInstance("AES/CTR/NoPadding");
-               cipher->init(Cipher_ENCRYPT_MODE, key, ivSpec);
-               Array<char> *encryptedBytes = cipher->doFinal(rawData);
+               Array<char> *encryptedBytes = AESEncrypt(ivBytes, key, rawData);
+               Array<char> *origBytes = AESDecrypt(ivBytes, key, encryptedBytes);
+               if (!rawData->equals(origBytes))
+                       throw new Error("BAD");
                Array<char> *chars = new Array<char>(encryptedBytes->length() + CloudComm_IV_SIZE);
                System_arraycopy(ivBytes, 0, chars, 0, ivBytes->length());
                System_arraycopy(encryptedBytes, 0, chars, CloudComm_IV_SIZE, encryptedBytes->length());
@@ -247,10 +518,7 @@ Array<char> *CloudComm::stripIVAndDecryptSlot(Array<char> *rawData) {
                Array<char> *encryptedBytes = new Array<char>(rawData->length() - CloudComm_IV_SIZE);
                System_arraycopy(rawData, 0, ivBytes, 0, CloudComm_IV_SIZE);
                System_arraycopy(rawData, CloudComm_IV_SIZE, encryptedBytes, 0, encryptedBytes->length());
-               IvParameterSpec *ivSpec = new IvParameterSpec(ivBytes);
-               Cipher *cipher = Cipher_getInstance("AES/CTR/NoPadding");
-               cipher->init(Cipher_DECRYPT_MODE, key, ivSpec);
-               return cipher->doFinal(encryptedBytes);
+               return AESDecrypt(ivBytes, key, encryptedBytes);
        } catch (Exception *e) {
                throw new Error("Failed To Decrypt");
        }
@@ -262,7 +530,7 @@ Array<char> *CloudComm::stripIVAndDecryptSlot(Array<char> *rawData) {
  * numbers.
  */
 Array<Slot *> *CloudComm::putSlot(Slot *slot, int max) {
-       int fd = -1;
+       WebConnection wc = {-1, -1};
        try {
                if (salt == NULL) {
                        if (!getSalt()) {
@@ -276,8 +544,8 @@ Array<Slot *> *CloudComm::putSlot(Slot *slot, int max) {
                Array<char> *chars = encryptSlotAndPrependIV(slotBytes, slot->getSlotCryptIV());
                IoTString *url = buildRequest(true, sequencenumber, max);
                timer->startTime();
-               fd = openURL(url, true);
-               writeURLData(fd, chars);
+               wc = openURL(url);
+               writeURLDataAndClose(&wc, chars);
                timer->endTime();
        } catch (ServerException *e) {
                timer->endTime();
@@ -290,20 +558,28 @@ Array<Slot *> *CloudComm::putSlot(Slot *slot, int max) {
        }
 
        try {
+               int respcode = getResponseCode(&wc);
+               readHeaders(&wc);
                timer->startTime();
                Array<char> *resptype = new Array<char>(7);
-               readURLData(fd, resptype);
+               readURLData(&wc, resptype);
                timer->endTime();
 
-               if (resptype->equals("getslot"->getBytes())) {
-                       return processSlots(fd);
-               } else if (resptype->equals("putslot"->getBytes())) {
+               if (resptype->equals(getslot)) {
+                       Array<Slot *> *tmp = processSlots(&wc);
+                       close(wc.fd);
+                       return tmp;
+               } else if (resptype->equals(putslot)) {
+                       close(wc.fd);
                        return NULL;
-               } else
+               } else {
+                       close(wc.fd);
                        throw new Error("Bad response to putslot");
+               }
        } catch (SocketTimeoutException *e) {
                timer->endTime();
-               throw new ServerException("putSlot failed", ServerException->TypeInputTimeout);
+               close(wc.fd);
+               throw new ServerException("putSlot failed", ServerException_TypeInputTimeout);
        } catch (Exception *e) {
                throw new Error("putSlot failed");
        }
@@ -314,6 +590,7 @@ Array<Slot *> *CloudComm::putSlot(Slot *slot, int max) {
  * sequencenumber or newer->
  */
 Array<Slot *> *CloudComm::getSlots(int64_t sequencenumber) {
+       WebConnection wc = {-1, -1};
        try {
                if (salt == NULL) {
                        if (!getSalt()) {
@@ -324,12 +601,8 @@ Array<Slot *> *CloudComm::getSlots(int64_t sequencenumber) {
 
                IoTString *url = buildRequest(false, sequencenumber, 0);
                timer->startTime();
-               URLConnection *con = url->openConnection();
-               HttpURLConnection *http = (HttpURLConnection *) con;
-               http->setRequestMethod("POST");
-               http->setConnectTimeout(CloudComm_TIMEOUT_MILLIS);
-               http->setReadTimeout(CloudComm_TIMEOUT_MILLIS);
-               http->connect();
+               wc = openURL(url);
+               closeURLReq(&wc);
                timer->endTime();
        } catch (SocketTimeoutException *e) {
                timer->endTime();
@@ -344,15 +617,20 @@ Array<Slot *> *CloudComm::getSlots(int64_t sequencenumber) {
 
        try {
                timer->startTime();
+               int responsecode = getResponseCode(&wc);
+               readHeaders(&wc);
                Array<char> *resptype = new Array<char>(7);
-               readURLData(fd, resptype);
+               readURLData(&wc, resptype);
                timer->endTime();
-               if (!resptype->equals("getslot"->getBytes()))
-                       throw new Error("Bad Response: " + new String(resptype));
+               if (!resptype->equals(getslot))
+                       throw new Error("Bad Response: ");
 
-               return processSlots(dis);
+               Array<Slot *> *tmp = processSlots(&wc);
+               close(wc.fd);
+               return tmp;
        } catch (SocketTimeoutException *e) {
                timer->endTime();
+               close(wc.fd);
                throw new ServerException("getSlots failed", ServerException_TypeInputTimeout);
        } catch (Exception *e) {
                throw new Error("getSlots failed");
@@ -363,23 +641,23 @@ Array<Slot *> *CloudComm::getSlots(int64_t sequencenumber) {
  * Method that actually handles building Slot objects from the
  * server response.  Shared by both putSlot and getSlots.
  */
-Array<Slot *> *CloudComm::processSlots(int fd) {
-       int numberofslots = readURLInt(fd);
+Array<Slot *> *CloudComm::processSlots(WebConnection *wc) {
+       int numberofslots = readURLInt(wc);
        Array<int> *sizesofslots = new Array<int>(numberofslots);
        Array<Slot *> *slots = new Array<Slot *>(numberofslots);
 
        for (int i = 0; i < numberofslots; i++)
-               sizesofslots->set(i, readURLInt(fd));
+               sizesofslots->set(i, readURLInt(wc));
        for (int i = 0; i < numberofslots; i++) {
                Array<char> *rawData = new Array<char>(sizesofslots->get(i));
-               readURLData(rawData);
+               readURLData(wc, rawData);
                Array<char> *data = stripIVAndDecryptSlot(rawData);
                slots->set(i, Slot_decode(table, data, mac));
        }
        return slots;
 }
 
-Array<char> *CloudComm::sendLocalData(Array<char> *sendData, int64_t localSequenceNumber, String host, int port) {
+Array<char> *CloudComm::sendLocalData(Array<char> *sendData, int64_t localSequenceNumber, IoTString *host, int port) {
        if (salt == NULL)
                return NULL;
        try {
@@ -395,25 +673,21 @@ Array<char> *CloudComm::sendLocalData(Array<char> *sendData, int64_t localSequen
                Array<char> *encryptedData = encryptSlotAndPrependIV(totalData, iv);
 
                // Open a TCP socket connection to a local device
-               Socket *socket = new Socket(host, port);
-               socket->setReuseAddress(true);
-               DataOutputStream *output = new DataOutputStream(socket->getOutputStream());
-               DataInputStream *input = new DataInputStream(socket->getInputStream());
+               int socket = createSocket(host, port);
 
                timer->startTime();
                // Send data to output (length of data, the data)
-               output->writeInt(encryptedData->length());
-               output->write(encryptedData, 0, encryptedData->length());
-               output->flush();
+               writeSocketInt(socket, encryptedData->length());
+               writeSocketData(socket, encryptedData);
 
-               int lengthOfReturnData = input->readInt();
+               int lengthOfReturnData = readSocketInt(socket);
                Array<char> *returnData = new Array<char>(lengthOfReturnData);
-               input->readFully(returnData);
+               readSocketData(socket, returnData);
                timer->endTime();
                returnData = stripIVAndDecryptSlot(returnData);
 
                // We are done with this socket
-               socket->close();
+               close(socket);
                mac->update(returnData, 0, returnData->length() - CloudComm_HMAC_SIZE);
                Array<char> *realmac = mac->doFinal();
                Array<char> *recmac = new Array<char>(CloudComm_HMAC_SIZE);
@@ -434,13 +708,11 @@ Array<char> *CloudComm::sendLocalData(Array<char> *sendData, int64_t localSequen
 }
 
 void CloudComm::localServerWorkerFunction() {
-       ServerSocket *inputSocket = NULL;
+       int inputSocket = -1;
 
        try {
                // Local server socket
-               inputSocket = new ServerSocket(listeningPort);
-               inputSocket->setReuseAddress(true);
-               inputSocket->setSoTimeout(CloudComm_TIMEOUT_MILLIS);
+               inputSocket = createSocket(listeningPort);
        } catch (Exception *e) {
                throw new Error("Local server setup failure...");
        }
@@ -448,14 +720,12 @@ void CloudComm::localServerWorkerFunction() {
        while (!doEnd) {
                try {
                        // Accept incoming socket
-                       Socket *socket = inputSocket->accept();
-                       DataInputStream *input = new DataInputStream(socket->getInputStream());
-                       DataOutputStream *output = new DataOutputStream(socket->getOutputStream());
+                       int socket = acceptSocket(inputSocket);
 
                        // Get the encrypted data from the server
-                       int dataSize = input->readInt();
+                       int dataSize = readSocketInt(socket);
                        Array<char> *readData = new Array<char>(dataSize);
-                       input->readFully(readData);
+                       readSocketData(socket, readData);
                        timer->endTime();
 
                        // Decrypt the data
@@ -485,29 +755,26 @@ void CloudComm::localServerWorkerFunction() {
 
                        timer->startTime();
                        // Send data to output (length of data, the data)
-                       output->writeInt(encryptedData->length());
-                       output->write(encryptedData, 0, encryptedData->length());
-                       output->flush();
-
-                       // close the socket
-                       socket->close();
+                       writeSocketInt(socket, encryptedData->length());
+                       writeSocketData(socket, encryptedData);
+                       close(socket);
                } catch (Exception *e) {
                }
        }
 
-       if (inputSocket != NULL) {
+       if (inputSocket != -1) {
                try {
-                       inputSocket->close();
+                       close(inputSocket);
                } catch (Exception *e) {
                        throw new Error("Local server close failure...");
                }
        }
 }
 
-void CloudComm::close() {
+void CloudComm::closeCloud() {
        doEnd = true;
 
-       if (localServerThread != NULL) {
+       if (listeningPort > 0) {
                if (pthread_join(localServerThread, NULL) != 0)
                        throw new Error("Local Server thread join issue...");
        }