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

diff --git a/version2/src/C/Crypto.h b/version2/src/C/Crypto.h
index 40a587f..5334335 100644
--- a/version2/src/C/Crypto.h
+++ b/version2/src/C/Crypto.h
@@ -1,12 +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);
-	
- private:
+public:
+	AESKey(Array<char> *password, Array<char> *salt, int iterationCount, int keyLength);
+	~AESKey();
+	Array<char> *getKey();
+	WORD *getKeySchedule();
+
+private:
+	Array<char> *key;
+	WORD key_schedule[60];
 };
 
 #endif