Fixed bug in counter

playerxq · Jan 28, 2016 · 0810b0e · 0810b0e
1 parent 6272778
commit 0810b0e
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/src/util/ot/opemasking.h b/src/util/ot/opemasking.h
@@ -82,10 +82,10 @@ class OPEMasking : public MaskingFunction
 		uint32_t hashinbytes = m_nCodeWordBytes + sizeof(uint64_t);
 		uint8_t *Mptr = matrix.GetArr();
 		CBitVector mask(m_nCodeWordBits * m_nExpansionFactor);
-		uint8_t* hash_buf = (uint8_t*) malloc(m_nCodeWordBytes * m_nExpansionFactor);
+		uint8_t* hash_buf = (uint8_t*) mlloc(m_nCodeWordBytes * m_nExpansionFactor);
 		uint8_t* mask_ptr;
 		uint8_t* hash_ptr;
-		uint8_t* tmpbuf = (uint8_t*) malloc(hashinbytes);
+		uint8_t* tmpbuf = (uint8_t*) calloc(hashinbytes, sizeof(uint8_t));
 
 #ifdef AES256_HASH
 	AES_KEY tk_aeskey;
@@ -103,6 +103,10 @@ class OPEMasking : public MaskingFunction
 
 			mask_ptr = mask.GetArr();
 
+#ifdef AES256_HASH
+			memcpy(tmpbuf, (uint8_t*) &ot_id, sizeof(uint64_t));
+#endif
+
 			if(m_vNumEleInBin[bin_id] < m_nCodeWordBits) {
 				//cout << "Choice for ot_id = " << ot_id << ": ";
 				for(u = 0; u < m_vNumEleInBin[bin_id]; u++)
@@ -192,17 +196,21 @@ class OPEMasking : public MaskingFunction
 		uint32_t hashinbytes = m_nCodeWordBytes + sizeof(uint64_t);
 		uint8_t *Mptr = matrix.GetArr();
 		uint8_t* hash_buf = (uint8_t*) malloc(m_nCodeWordBytes);
-		uint8_t* tmpbuf = (uint8_t*) malloc(hashinbytes);
+		uint8_t* tmpbuf = (uint8_t*) calloc(hashinbytes, sizeof(uint8_t));
 
 #ifdef AES256_HASH
 	AES_KEY tk_aeskey;
 	block inblock, outblock;
 	tk_aeskey.rounds = 14;
+
 #endif
 
 		for(ot_id = ot_begin_id; ot_id < ot_begin_id+processedOTs; ot_id++, Mptr+=m_nCodeWordBytes)
 		{
 
+#ifdef AES256_HASH
+			memcpy(tmpbuf, (uint8_t*) &ot_id, sizeof(uint64_t));
+#endif
 			bin_id = ot_id/m_nOTsPerElement;
 			if(m_vNumEleInBin[bin_id] > 0) {
 #ifdef DEBUG_HASH_INPUT