Improved SHA-1 performance with Rusha library

AADSTS900144 · Oct 23, 2014 · 1089af2 · 1089af2
1 parent 94238f3
commit 1089af2
Show file tree

Hide file tree

Showing 6 changed files with 417 additions and 16 deletions.
diff --git a/app/index.html b/app/index.html
@@ -56,6 +56,7 @@
 
   <script type="text/javascript" src="vendor/jsbn/jsbn_combined.js"></script>
   <script type="text/javascript" src="vendor/cryptoJS/crypto.js"></script>
+  <script type="text/javascript" src="vendor/rusha/rusha.js"></script>
   <script type="text/javascript" src="vendor/zlib/gunzip.min.js"></script>
   <script type="text/javascript" src="vendor/closure/long.js"></script>
 

diff --git a/app/js/lib/bin_utils.js b/app/js/lib/bin_utils.js
@@ -196,9 +196,10 @@ function uintToInt (val) {
 }
 
 function sha1Hash (bytes) {
-  // console.log('SHA-1 hash start');
-  var hashBytes = sha1.hash(bytes, true);
-  // console.log('SHA-1 hash finish');
+  this.rushaInstance = this.rushaInstance || new Rusha(1024 * 1024);
+  // console.log(dT(), 'SHA-1 hash start', bytes.byteLength || bytes.length);
+  var hashBytes = bytesFromArrayBuffer(rushaInstance.rawDigest(bytes).buffer);
+  // console.log(dT(), 'SHA-1 hash finish');
 
   return hashBytes;
 }

diff --git a/app/js/lib/crypto_worker.js b/app/js/lib/crypto_worker.js
@@ -11,7 +11,8 @@ importScripts(
   '../../vendor/jsbn/jsbn_combined.js',
   '../../vendor/leemon_bigint/bigint.js',
   '../../vendor/closure/long.js',
-  '../../vendor/cryptoJS/crypto.js'
+  '../../vendor/cryptoJS/crypto.js',
+  '../../vendor/rusha/rusha.js'
 );
 
 onmessage = function (e) {

diff --git a/app/js/lib/tl_utils.js b/app/js/lib/tl_utils.js
@@ -36,7 +36,16 @@ TLSerialization.prototype.getBuffer = function () {
   return this.getArray().buffer;
 };
 
-TLSerialization.prototype.getBytes = function () {
+TLSerialization.prototype.getBytes = function (typed) {
+  if (typed) {
+    var resultBuffer = new ArrayBuffer(this.offset);
+    var resultArray  = new Uint8Array(resultBuffer);
+
+    resultArray.set(this.byteView.subarray(0, this.offset));
+
+    return resultArray;
+  }
+
   var bytes = [];
   for (var i = 0; i < this.offset; i++) {
     bytes.push(this.byteView[i]);
@@ -141,9 +150,8 @@ TLSerialization.prototype.storeString = function (s, field) {
 TLSerialization.prototype.storeBytes = function (bytes, field) {
   this.debug && console.log('>>>', bytesToHex(bytes), (field || '') + ':bytes');
 
-  this.checkLength(bytes.length + 8);
-
-  var len = bytes.length;
+  var len = bytes.byteLength || bytes.length;
+  this.checkLength(len + 8);
   if (len <= 253) {
     this.byteView[this.offset++] = len;
   } else {
@@ -152,9 +160,8 @@ TLSerialization.prototype.storeBytes = function (bytes, field) {
     this.byteView[this.offset++] = (len & 0xFF00) >> 8;
     this.byteView[this.offset++] = (len & 0xFF0000) >> 16;
   }
-  for (var i = 0; i < len; i++) {
-    this.byteView[this.offset++] = bytes[i];
-  }
+  this.byteView.set(bytes, this.offset);
+  this.offset += len;
 
   // Padding
   while (this.offset % 4) {
@@ -177,14 +184,13 @@ TLSerialization.prototype.storeIntBytes = function (bytes, bits, field) {
 };
 
 TLSerialization.prototype.storeRawBytes = function (bytes, field) {
-  var len = bytes.length;
+  var len = bytes.byteLength || bytes.length;
 
   this.debug && console.log('>>>', bytesToHex(bytes), (field || ''));
   this.checkLength(len);
 
-  for (var i = 0; i < len; i++) {
-    this.byteView[this.offset++] = bytes[i];
-  }
+  this.byteView.set(bytes, this.offset);
+  this.offset += len;
 };