X-Git-Url: http://plrg.eecs.uci.edu/git/?p=iotcloud.git;a=blobdiff_plain;f=version2%2Fsrc%2FC%2FCrypto.h;h=5334335f08e0d97c5a60fdc16d04ed80828cc017;hp=8f64573eb7c8639360452df3acd6b0f3bbd4de26;hb=b2bc9b5c707bd7d932d60cd4e8c1cb580b36b5b4;hpb=15acb678f516190fbf7364a41a1f570e90dc09d0

diff --git a/version2/src/C/Crypto.h b/version2/src/C/Crypto.h
index 8f64573..5334335 100644
--- a/version2/src/C/Crypto.h
+++ b/version2/src/C/Crypto.h
@@ -1,15 +1,18 @@
 #ifndef CRYPTO_H
 #define CRYPTO_H
 #include "common.h"
+#include "aes.h"
 
 class AESKey {
 public:
 	AESKey(Array<char> *password, Array<char> *salt, int iterationCount, int keyLength);
 	~AESKey();
-	Array<char> * getKey();
-	
+	Array<char> *getKey();
+	WORD *getKeySchedule();
+
 private:
-	Array<char> * key;
+	Array<char> *key;
+	WORD key_schedule[60];
 };
 
 #endif