Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Huge performance boost by reducing operations with arrays #102

Open
wants to merge 4 commits into
base: master
Choose a base branch
from
Open

Huge performance boost by reducing operations with arrays #102

Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
03cf18b
Reducing number of operations with arrays
NikitaCartes Jul 6, 2020
5e46882
Delete convertToInt32 from encrypt and decrypt
NikitaCartes Jul 6, 2020
6d16f9c
Aligning in one style with the rest of file
NikitaCartes Jul 8, 2020
3b4b5da
Make Ke and Kd one-dimensional array
NikitaCartes Jul 8, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
224 changes: 160 additions & 64 deletions index.js
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -206,29 +206,28 @@
throw new Error('invalid key size (must be 16, 24 or 32 bytes)');
}

var roundKeyCount = (rounds + 1) * 4;
var KC = this.key.length / 4;

// encryption round keys
this._Ke = [];
this._Ke = new Array(roundKeyCount);

// decryption round keys
this._Kd = [];
this._Kd = new Array(roundKeyCount);

for (var i = 0; i <= rounds; i++) {
this._Ke.push([0, 0, 0, 0]);
this._Kd.push([0, 0, 0, 0]);
for (var i = 0; i < roundKeyCount; i++) {
this._Ke[i] = 0;
this._Kd[i] = 0;
}

var roundKeyCount = (rounds + 1) * 4;
var KC = this.key.length / 4;

// convert the key into ints
var tk = convertToInt32(this.key);

// copy values into round key arrays
var index;
for (var i = 0; i < KC; i++) {
index = i >> 2;
this._Ke[index][i % 4] = tk[i];
this._Kd[rounds - index][i % 4] = tk[i];
this._Ke[i] = tk[i];
this._Kd[(rounds - (i >> 2))*4 + (i % 4)] = tk[i];
}

// key expansion (fips-197 section 5.2)
Expand Down Expand Up @@ -267,24 +266,22 @@
}

// copy values into round key arrays
var i = 0, r, c;
var i = 0;
while (i < KC && t < roundKeyCount) {
r = t >> 2;
c = t % 4;
this._Ke[r][c] = tk[i];
this._Kd[rounds - r][c] = tk[i++];
this._Ke[t] = tk[i];
this._Kd[(rounds - (t >> 2))*4 + (t % 4)] = tk[i++];
t++;
}
}

// inverse-cipher-ify the decryption round key (fips-197 section 5.3)
for (var r = 1; r < rounds; r++) {
for (var c = 0; c < 4; c++) {
tt = this._Kd[r][c];
this._Kd[r][c] = (U1[(tt >> 24) & 0xFF] ^
U2[(tt >> 16) & 0xFF] ^
U3[(tt >> 8) & 0xFF] ^
U4[ tt & 0xFF]);
tt = this._Kd[4 * r + c];
this._Kd[4 * r + c] = (U1[(tt >> 24) & 0xFF] ^
U2[(tt >> 16) & 0xFF] ^
U3[(tt >> 8) & 0xFF] ^
U4[ tt & 0xFF]);
}
}
}
Expand All @@ -294,36 +291,85 @@
throw new Error('invalid plaintext size (must be 16 bytes)');
}

var rounds = this._Ke.length - 1;
var a = [0, 0, 0, 0];
var rounds = this._Ke.length / 4 - 1;
var a0 = 0;
var a1 = 0;
var a2 = 0;
var a3 = 0;

// convert plaintext to (ints ^ key)
var t = convertToInt32(plaintext);
for (var i = 0; i < 4; i++) {
t[i] ^= this._Ke[0][i];
}

var t0 = ((plaintext[0] << 24) |
(plaintext[1] << 16) |
(plaintext[2] << 8) |
plaintext[3]) ^ this._Ke[0];
var t1 = ((plaintext[4] << 24) |
(plaintext[5] << 16) |
(plaintext[6] << 8) |
plaintext[7]) ^ this._Ke[1];
var t2 = ((plaintext[8] << 24) |
(plaintext[9] << 16) |
(plaintext[10] << 8) |
plaintext[11]) ^ this._Ke[2];
var t3 = ((plaintext[12] << 24) |
(plaintext[13] << 16) |
(plaintext[14] << 8) |
plaintext[15]) ^ this._Ke[3];
// apply round transforms
for (var r = 1; r < rounds; r++) {
for (var i = 0; i < 4; i++) {
a[i] = (T1[(t[ i ] >> 24) & 0xff] ^
T2[(t[(i + 1) % 4] >> 16) & 0xff] ^
T3[(t[(i + 2) % 4] >> 8) & 0xff] ^
T4[ t[(i + 3) % 4] & 0xff] ^
this._Ke[r][i]);
}
t = a.slice();
a0 = (T1[(t0 >> 24) & 0xff] ^
T2[(t1 >> 16) & 0xff] ^
T3[(t2 >> 8) & 0xff] ^
T4[ t3 & 0xff] ^
this._Ke[4 * r]);

a1 = (T1[(t1 >> 24) & 0xff] ^
T2[(t2 >> 16) & 0xff] ^
T3[(t3 >> 8) & 0xff] ^
T4[ t0 & 0xff] ^
this._Ke[4 * r + 1]);

a2 = (T1[(t2 >> 24) & 0xff] ^
T2[(t3 >> 16) & 0xff] ^
T3[(t0 >> 8) & 0xff] ^
T4[ t1 & 0xff] ^
this._Ke[4 * r + 2]);

a3 = (T1[(t3 >> 24) & 0xff] ^
T2[(t0 >> 16) & 0xff] ^
T3[(t1 >> 8) & 0xff] ^
T4[ t2 & 0xff] ^
this._Ke[4 * r + 3]);
t0 = a0;
t1 = a1;
t2 = a2;
t3 = a3;
}

// the last round is special
var result = createArray(16), tt;
for (var i = 0; i < 4; i++) {
tt = this._Ke[rounds][i];
result[4 * i ] = (S[(t[ i ] >> 24) & 0xff] ^ (tt >> 24)) & 0xff;
result[4 * i + 1] = (S[(t[(i + 1) % 4] >> 16) & 0xff] ^ (tt >> 16)) & 0xff;
result[4 * i + 2] = (S[(t[(i + 2) % 4] >> 8) & 0xff] ^ (tt >> 8)) & 0xff;
result[4 * i + 3] = (S[ t[(i + 3) % 4] & 0xff] ^ tt ) & 0xff;
}
tt = this._Ke[4 * rounds];
result[0] = (S[(t0 >> 24) & 0xff] ^ (tt >> 24)) & 0xff;
result[1] = (S[(t1 >> 16) & 0xff] ^ (tt >> 16)) & 0xff;
result[2] = (S[(t2 >> 8) & 0xff] ^ (tt >> 8)) & 0xff;
result[3] = (S[ t3 & 0xff] ^ tt ) & 0xff;

tt = this._Ke[4 * rounds + 1];
result[4] = (S[(t1 >> 24) & 0xff] ^ (tt >> 24)) & 0xff;
result[5] = (S[(t2 >> 16) & 0xff] ^ (tt >> 16)) & 0xff;
result[6] = (S[(t3 >> 8) & 0xff] ^ (tt >> 8)) & 0xff;
result[7] = (S[ t0 & 0xff] ^ tt ) & 0xff;

tt = this._Ke[4 * rounds + 2];
result[8] = (S[(t2 >> 24) & 0xff] ^ (tt >> 24)) & 0xff;
result[9] = (S[(t3 >> 16) & 0xff] ^ (tt >> 16)) & 0xff;
result[10] = (S[(t0 >> 8) & 0xff] ^ (tt >> 8)) & 0xff;
result[11] = (S[t1 & 0xff] ^ tt ) & 0xff;

tt = this._Ke[4 * rounds + 3];
result[12] = (S[(t3 >> 24) & 0xff] ^ (tt >> 24)) & 0xff;
result[13] = (S[(t0 >> 16) & 0xff] ^ (tt >> 16)) & 0xff;
result[14] = (S[(t1 >> 8) & 0xff] ^ (tt >> 8)) & 0xff;
result[15] = (S[ t2 & 0xff] ^ tt ) & 0xff;

return result;
}
Expand All @@ -333,36 +379,86 @@
throw new Error('invalid ciphertext size (must be 16 bytes)');
}

var rounds = this._Kd.length - 1;
var a = [0, 0, 0, 0];
var rounds = this._Kd.length / 4 - 1;
var a0 = 0;
var a1 = 0;
var a2 = 0;
var a3 = 0;

// convert plaintext to (ints ^ key)
var t = convertToInt32(ciphertext);
for (var i = 0; i < 4; i++) {
t[i] ^= this._Kd[0][i];
}
var t0 = ((ciphertext[0] << 24) |
(ciphertext[1] << 16) |
(ciphertext[2] << 8) |
ciphertext[3]) ^ this._Kd[0];
var t1 = ((ciphertext[4] << 24) |
(ciphertext[5] << 16) |
(ciphertext[6] << 8) |
ciphertext[7]) ^ this._Kd[1];
var t2 = ((ciphertext[8] << 24) |
(ciphertext[9] << 16) |
(ciphertext[10] << 8) |
ciphertext[11]) ^ this._Kd[2];
var t3 = ((ciphertext[12] << 24) |
(ciphertext[13] << 16) |
(ciphertext[14] << 8) |
ciphertext[15]) ^ this._Kd[3];

// apply round transforms
for (var r = 1; r < rounds; r++) {
for (var i = 0; i < 4; i++) {
a[i] = (T5[(t[ i ] >> 24) & 0xff] ^
T6[(t[(i + 3) % 4] >> 16) & 0xff] ^
T7[(t[(i + 2) % 4] >> 8) & 0xff] ^
T8[ t[(i + 1) % 4] & 0xff] ^
this._Kd[r][i]);
}
t = a.slice();
a0 = (T5[(t0 >> 24) & 0xff] ^
T6[(t3 >> 16) & 0xff] ^
T7[(t2 >> 8) & 0xff] ^
T8[ t1 & 0xff] ^
this._Kd[4 * r]);

a1 = (T5[(t1 >> 24) & 0xff] ^
T6[(t0 >> 16) & 0xff] ^
T7[(t3 >> 8) & 0xff] ^
T8[ t2 & 0xff] ^
this._Kd[4 * r + 1]);

a2 = (T5[(t2 >> 24) & 0xff] ^
T6[(t1 >> 16) & 0xff] ^
T7[(t0 >> 8) & 0xff] ^
T8[ t3 & 0xff] ^
this._Kd[4 * r + 2]);

a3 = (T5[(t3 >> 24) & 0xff] ^
T6[(t2 >> 16) & 0xff] ^
T7[(t1 >> 8) & 0xff] ^
T8[ t0 & 0xff] ^
this._Kd[4 * r + 3]);
t0 = a0;
t1 = a1;
t2 = a2;
t3 = a3;
}

// the last round is special
var result = createArray(16), tt;
for (var i = 0; i < 4; i++) {
tt = this._Kd[rounds][i];
result[4 * i ] = (Si[(t[ i ] >> 24) & 0xff] ^ (tt >> 24)) & 0xff;
result[4 * i + 1] = (Si[(t[(i + 3) % 4] >> 16) & 0xff] ^ (tt >> 16)) & 0xff;
result[4 * i + 2] = (Si[(t[(i + 2) % 4] >> 8) & 0xff] ^ (tt >> 8)) & 0xff;
result[4 * i + 3] = (Si[ t[(i + 1) % 4] & 0xff] ^ tt ) & 0xff;
}
tt = this._Kd[4 * rounds];
result[0] = (Si[(t0 >> 24) & 0xff] ^ (tt >> 24)) & 0xff;
result[1] = (Si[(t3 >> 16) & 0xff] ^ (tt >> 16)) & 0xff;
result[2] = (Si[(t2 >> 8) & 0xff] ^ (tt >> 8)) & 0xff;
result[3] = (Si[ t1 & 0xff] ^ tt ) & 0xff;

tt = this._Kd[4 * rounds + 1];
result[4] = (Si[(t1 >> 24) & 0xff] ^ (tt >> 24)) & 0xff;
result[5] = (Si[(t0 >> 16) & 0xff] ^ (tt >> 16)) & 0xff;
result[6] = (Si[(t3 >> 8) & 0xff] ^ (tt >> 8)) & 0xff;
result[7] = (Si[ t2 & 0xff] ^ tt ) & 0xff;

tt = this._Kd[4 * rounds + 2];
result[8] = (Si[(t2 >> 24) & 0xff] ^ (tt >> 24)) & 0xff;
result[9] = (Si[(t1 >> 16) & 0xff] ^ (tt >> 16)) & 0xff;
result[10] = (Si[(t0 >> 8) & 0xff] ^ (tt >> 8)) & 0xff;
result[11] = (Si[t3 & 0xff] ^ tt ) & 0xff;

tt = this._Kd[4 * rounds + 3];
result[12] = (Si[(t3 >> 24) & 0xff] ^ (tt >> 24)) & 0xff;
result[13] = (Si[(t2 >> 16) & 0xff] ^ (tt >> 16)) & 0xff;
result[14] = (Si[(t1 >> 8) & 0xff] ^ (tt >> 8)) & 0xff;
result[15] = (Si[ t0 & 0xff] ^ tt ) & 0xff;

return result;
}
Expand Down