Add AES encryption to AUTH application.

Author: ankitbhrdwj

db/src/master.rs

@@ -300,7 +300,7 @@ impl Master {
         }
 
         // Next, populate the actual records.
-        for i in 1..records {
+        for i in 0..records {
             let mut key = vec![0; K_LEN as usize];
             let mut val = vec![0; V_LEN as usize];
 

ext/auth/Cargo.toml

@@ -9,3 +9,4 @@ crate-type = ["dylib"]
 [dependencies]
 sandstorm = { path = "../../sandstorm" }
 rust-crypto = "0.2.36"
+openssl = "0.10.24"

ext/auth/src/lib.rs

@@ -20,6 +20,10 @@
 extern crate crypto;
 #[macro_use]
 extern crate sandstorm;
+extern crate openssl;
+
+use openssl::aes::{aes_ige, AesKey};
+use openssl::symm::Mode;
 
 use crypto::bcrypt::bcrypt;
 
@@ -55,6 +59,11 @@ pub fn init(db: Rc<DB>) -> Box<Generator<Yield = u64, Return = u64>> {
         let mut status = INVALIDARG;
         let mut username: Vec<u8> = Vec::with_capacity(30);
         let mut password: Vec<u8> = Vec::with_capacity(72);
+        let key = b"\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0A\x0B\x0C\x0D\x0E\x0F";
+        let mut iv = *b"\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0A\x0B\x0C\x0D\x0E\x0F\
+                \x10\x11\x12\x13\x14\x15\x16\x17\x18\x19\x1A\x1B\x1C\x1D\x1E\x1F";
+        let ekey = AesKey::new_decrypt(key).unwrap();
+        let mut output = [0u8; 16];
 
         {
             // First off, retrieve the arguments to the extension.
@@ -76,6 +85,8 @@ pub fn init(db: Rc<DB>) -> Box<Generator<Yield = u64, Return = u64>> {
             let (userid, pass) = remain_args.split_at(30);
             username.extend_from_slice(userid);
             password.extend_from_slice(pass);
+            aes_ige(&password[0..16], &mut output, &ekey, &mut iv, Mode::Decrypt);
+            password[0..16].copy_from_slice(&output[0..16]);
 
             // Get the table id from the unwrapped arguments.
             for (idx, e) in s_table.iter().enumerate() {

splinter/Cargo.toml

@@ -73,6 +73,7 @@ spin         = "0.4.7"  # Consider using parking lot?
 bytes        = "0.4.7"
 env_logger   = "0.3"
 libloading   = "0.3"
+openssl      = "0.10.24"
 rustlearn    = "0.5.0"
 serde        = "1.0.37"
 serde_derive = "1.0.37"

splinter/src/bin/client/auth.rs

@@ -17,13 +17,17 @@
 
 extern crate crypto;
 extern crate db;
+extern crate openssl;
 extern crate rand;
 extern crate sandstorm;
 extern crate spin;
 extern crate splinter;
 extern crate time;
 extern crate zipf;
 
+use openssl::aes::{aes_ige, AesKey};
+use openssl::symm::Mode;
+
 mod setup;
 
 use std::cell::RefCell;
@@ -308,6 +312,12 @@ where
             return;
         }
 
+        let key = b"\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0A\x0B\x0C\x0D\x0E\x0F";
+        let mut iv = *b"\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0A\x0B\x0C\x0D\x0E\x0F\
+                \x10\x11\x12\x13\x14\x15\x16\x17\x18\x19\x1A\x1B\x1C\x1D\x1E\x1F";
+        let ekey = AesKey::new_encrypt(key).unwrap();
+        let mut output = [0u8; 16];
+
         while self.outstanding < 32 && self.manager.borrow().get_queue_len() < 32 {
             // Get the current time stamp so that we can determine if it is time to issue the next RPC.
             let curr = cycles::rdtsc();
@@ -338,7 +348,8 @@ where
                         // extension name (4 bytes), the table id (8 bytes), Just write
                         // in the first 4 bytes of the key and first 4 bytes of value.
                         p_get[12..16].copy_from_slice(&key[0..4]);
-                        p_get[42..46].copy_from_slice(&key[0..4]);
+                        aes_ige(&key[0..16], &mut output, &ekey, &mut iv, Mode::Encrypt);
+                        p_get[42..58].copy_from_slice(&output[0..16]);
                         self.manager.borrow_mut().create_task(
                             curr,
                             &p_get,