From ec7439ada1c23627112237afcf4f8e3a64869495 Mon Sep 17 00:00:00 2001
From: Brian Demsky <bdemsky@plrg.eecs.uci.edu>
Date: Sat, 23 Jul 2016 10:34:20 -0700
Subject: [PATCH] more code

---
 src/java/iotcloud/CloudComm.java | 18 +++++++++------
 src/java/iotcloud/IoTString.java | 38 ++++++++++++++++++++++++++++++++
 src/java/iotcloud/Slot.java      |  2 +-
 src/java/iotcloud/Table.java     |  5 ++++-
 4 files changed, 54 insertions(+), 9 deletions(-)
 create mode 100644 src/java/iotcloud/IoTString.java

diff --git a/src/java/iotcloud/CloudComm.java b/src/java/iotcloud/CloudComm.java
index 4bc223b..4f64036 100644
--- a/src/java/iotcloud/CloudComm.java
+++ b/src/java/iotcloud/CloudComm.java
@@ -8,11 +8,13 @@ class CloudComm {
 	String baseurl;
 	Cipher encryptcipher;
 	Cipher decryptcipher;
+	Mac mac;
 	
-	CloudComm(String _baseurl, Cipher _encrypt, Cipher _decrypt) {
+	CloudComm(String _baseurl, Cipher _encrypt, Cipher _decrypt, Mac _mac) {
 		this.baseurl=_baseurl;
 		this.encryptcipher = _encrypt;
 		this.decryptcipher = _decrypt;
+		this.mac = _mac;
 	}
 	
 	private URL buildRequest(boolean isput, long sequencenumber, long maxentries) throws IOException {
@@ -25,15 +27,17 @@ class CloudComm {
 
 	public Slot[] putSlot(Slot slot, int max) throws IOException{
 		long sequencenumber=slot.getSequenceNumber();
+		byte[] bytes=slot.encode(mac);
+		
 		URL url=buildRequest(true, sequencenumber, max);
 		URLConnection con=url.openConnection();
 		HttpURLConnection http = (HttpURLConnection) con;
 		http.setRequestMethod("POST");
-		http.setFixedLengthStreamingMode(slot.getBytes().length);
+		http.setFixedLengthStreamingMode(bytes.length);
 		http.setDoOutput(true);
 		http.connect();
 		OutputStream os=http.getOutputStream();
-		os.write(slot.getBytes());
+		os.write(bytes);
 		System.out.println(http.getResponseMessage());
 
 		InputStream is=http.getInputStream();
@@ -41,7 +45,7 @@ class CloudComm {
 		byte[] resptype=new byte[7];
 		dis.readFully(resptype);
 		if (Arrays.equals(resptype, "getslot".getBytes()))
-			return processSlots(dis, sequencenumber);
+			return processSlots(dis);
 		else if (Arrays.equals(resptype, "putslot".getBytes()))
 			return null;
 		else
@@ -63,10 +67,10 @@ class CloudComm {
 		if (!Arrays.equals(resptype, "getslot".getBytes()))
 			throw new Error("Bad Response: "+new String(resptype));
 		else
-			return processSlots(dis, sequencenumber);
+			return processSlots(dis);
 	}
 		
-	Slot[] processSlots(DataInputStream dis, long sequencenumber) throws IOException {
+	Slot[] processSlots(DataInputStream dis) throws IOException {
 		int numberofslots=dis.readInt();
 		int[] sizesofslots=new int[numberofslots];
 		Slot[] slots=new Slot[numberofslots];
@@ -77,7 +81,7 @@ class CloudComm {
 		for(int i=0;i<numberofslots;i++) {
 			byte[] data=new byte[sizesofslots[i]];
 			dis.readFully(data);
-			slots[i]=new Slot(sequencenumber+i, data);
+			slots[i]=Slot.decode(data, mac);
 		}
 		dis.close();
 		return slots;
diff --git a/src/java/iotcloud/IoTString.java b/src/java/iotcloud/IoTString.java
new file mode 100644
index 0000000..1fe7d0e
--- /dev/null
+++ b/src/java/iotcloud/IoTString.java
@@ -0,0 +1,38 @@
+package IoTCloud;
+
+import util.Arrays;
+
+public class IoTString {
+	byte[] array;
+	int hashcode;
+
+	public IoTString(byte[] _array) {
+		array=(byte[]) _array.clone();
+		hashcode=Arrays.hashCode(array);
+	}
+
+	public IoTString(String str) {
+		array=str.getBytes();
+		hashcode=Arrays.hashCode(array);
+	}
+
+	public int hashCode() {
+		return hashcode;
+	}
+
+	public String toString() {
+		return new String(array);
+	}
+
+	public byte[] getBytes() {
+		return (byte[]) array.clone();
+	}
+	
+	public boolean equals(Object o) {
+		if (o instanceof IoTString) {
+			IoTString i=(IoTString)o;
+			return Arrays.equals(array, i.array);
+		}
+		return false;
+	}
+}
diff --git a/src/java/iotcloud/Slot.java b/src/java/iotcloud/Slot.java
index 841e818..b0002b0 100644
--- a/src/java/iotcloud/Slot.java
+++ b/src/java/iotcloud/Slot.java
@@ -41,7 +41,7 @@ class Slot {
 		long seqnum=bb.getLong();
 		long machineid=bb.getLong();
 		int numentries=bb.getInt();
-		Vector<Entry> entries=new Vector();
+		Vector<Entry> entries=new Vector<Entry>();
 		for(int i=0;i<numentries;i++) {
 			entries.add(Entry.decode(bb));
 		}
diff --git a/src/java/iotcloud/Table.java b/src/java/iotcloud/Table.java
index f1087ae..5efe9b2 100644
--- a/src/java/iotcloud/Table.java
+++ b/src/java/iotcloud/Table.java
@@ -22,7 +22,7 @@ public class Table {
 			decryptCipher.init(Cipher.DECRYPT_MODE, secret);
 			hmac = Mac.getInstance("HmacSHA256");
 			hmac.init(secret);
-			cloud=new CloudComm(baseurl, encryptCipher, decryptCipher);
+			cloud=new CloudComm(baseurl, encryptCipher, decryptCipher, hmac);
 		} catch (Exception e) {
 			throw new Error("Failed To Initialize Ciphers");
 		}
@@ -38,4 +38,7 @@ public class Table {
 			throw new Error("Failed generating key.");
 		}
 	}
+
+
+
 }
-- 
2.34.1