mirror of
https://github.com/denoland/deno.git
synced 2024-11-30 16:40:57 -05:00
Improve & refactor SHA-1 inside ws package (#137)
This commit is contained in:
parent
d1a9605e53
commit
0a10ee9f93
2 changed files with 118 additions and 115 deletions
218
ws/sha1.ts
218
ws/sha1.ts
|
@ -9,77 +9,78 @@
|
|||
/*jslint bitwise: true */
|
||||
|
||||
const HEX_CHARS = "0123456789abcdef".split("");
|
||||
const EXTRA = [-2147483648, 8388608, 32768, 128];
|
||||
const SHIFT = [24, 16, 8, 0];
|
||||
const EXTRA = Uint32Array.of(-2147483648, 8388608, 32768, 128);
|
||||
const SHIFT = Uint32Array.of(24, 16, 8, 0);
|
||||
|
||||
const blocks = [];
|
||||
const blocks = new Uint32Array(80);
|
||||
|
||||
export class Sha1 {
|
||||
blocks;
|
||||
block;
|
||||
start;
|
||||
bytes;
|
||||
hBytes;
|
||||
finalized;
|
||||
hashed;
|
||||
first;
|
||||
private _blocks: Uint32Array;
|
||||
private _block: number;
|
||||
private _start: number;
|
||||
private _bytes: number;
|
||||
private _hBytes: number;
|
||||
private _finalized: boolean;
|
||||
private _hashed: boolean;
|
||||
|
||||
h0 = 0x67452301;
|
||||
h1 = 0xefcdab89;
|
||||
h2 = 0x98badcfe;
|
||||
h3 = 0x10325476;
|
||||
h4 = 0xc3d2e1f0;
|
||||
lastByteIndex = 0;
|
||||
private _h0 = 0x67452301;
|
||||
private _h1 = 0xefcdab89;
|
||||
private _h2 = 0x98badcfe;
|
||||
private _h3 = 0x10325476;
|
||||
private _h4 = 0xc3d2e1f0;
|
||||
private _lastByteIndex = 0;
|
||||
|
||||
constructor(sharedMemory: boolean = false) {
|
||||
constructor(sharedMemory = false) {
|
||||
if (sharedMemory) {
|
||||
blocks[0] = blocks[16] = blocks[1] = blocks[2] = blocks[3] = blocks[4] = blocks[5] = blocks[6] = blocks[7] = blocks[8] = blocks[9] = blocks[10] = blocks[11] = blocks[12] = blocks[13] = blocks[14] = blocks[15] = 0;
|
||||
this.blocks = blocks;
|
||||
this._blocks = blocks.fill(0, 0, 17);
|
||||
} else {
|
||||
this.blocks = [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0];
|
||||
this._blocks = new Uint32Array(80);
|
||||
}
|
||||
|
||||
this.h0 = 0x67452301;
|
||||
this.h1 = 0xefcdab89;
|
||||
this.h2 = 0x98badcfe;
|
||||
this.h3 = 0x10325476;
|
||||
this.h4 = 0xc3d2e1f0;
|
||||
this._h0 = 0x67452301;
|
||||
this._h1 = 0xefcdab89;
|
||||
this._h2 = 0x98badcfe;
|
||||
this._h3 = 0x10325476;
|
||||
this._h4 = 0xc3d2e1f0;
|
||||
|
||||
this.block = this.start = this.bytes = this.hBytes = 0;
|
||||
this.finalized = this.hashed = false;
|
||||
this.first = true;
|
||||
this._block = this._start = this._bytes = this._hBytes = 0;
|
||||
this._finalized = this._hashed = false;
|
||||
}
|
||||
|
||||
update(data: string | ArrayBuffer) {
|
||||
if (this.finalized) {
|
||||
update(data: string | ArrayBuffer | ArrayBufferView) {
|
||||
if (this._finalized) {
|
||||
return;
|
||||
}
|
||||
let notString = true;
|
||||
let message;
|
||||
let notString = typeof data !== "string";
|
||||
if (notString && data instanceof ArrayBuffer) {
|
||||
if (data instanceof ArrayBuffer) {
|
||||
message = new Uint8Array(data);
|
||||
} else if (ArrayBuffer.isView(data)) {
|
||||
message = new Uint8Array(data.buffer);
|
||||
} else {
|
||||
message = data;
|
||||
notString = false;
|
||||
message = String(data);
|
||||
}
|
||||
let code,
|
||||
index = 0,
|
||||
i,
|
||||
start = this._start,
|
||||
length = message.length || 0,
|
||||
blocks = this.blocks;
|
||||
blocks = this._blocks;
|
||||
|
||||
while (index < length) {
|
||||
if (this.hashed) {
|
||||
this.hashed = false;
|
||||
blocks[0] = this.block;
|
||||
blocks[16] = blocks[1] = blocks[2] = blocks[3] = blocks[4] = blocks[5] = blocks[6] = blocks[7] = blocks[8] = blocks[9] = blocks[10] = blocks[11] = blocks[12] = blocks[13] = blocks[14] = blocks[15] = 0;
|
||||
if (this._hashed) {
|
||||
this._hashed = false;
|
||||
blocks[0] = this._block;
|
||||
blocks.fill(0, 1, 17);
|
||||
}
|
||||
|
||||
if (notString) {
|
||||
for (i = this.start; index < length && i < 64; ++index) {
|
||||
for (i = start; index < length && i < 64; ++index) {
|
||||
blocks[i >> 2] |= message[index] << SHIFT[i++ & 3];
|
||||
}
|
||||
} else {
|
||||
for (i = this.start; index < length && i < 64; ++index) {
|
||||
for (i = start; index < length && i < 64; ++index) {
|
||||
code = message.charCodeAt(index);
|
||||
if (code < 0x80) {
|
||||
blocks[i >> 2] |= code << SHIFT[i++ & 3];
|
||||
|
@ -102,56 +103,55 @@ export class Sha1 {
|
|||
}
|
||||
}
|
||||
|
||||
this.lastByteIndex = i;
|
||||
this.bytes += i - this.start;
|
||||
this._lastByteIndex = i;
|
||||
this._bytes += i - start;
|
||||
if (i >= 64) {
|
||||
this.block = blocks[16];
|
||||
this.start = i - 64;
|
||||
this._block = blocks[16];
|
||||
this._start = i - 64;
|
||||
this.hash();
|
||||
this.hashed = true;
|
||||
this._hashed = true;
|
||||
} else {
|
||||
this.start = i;
|
||||
this._start = i;
|
||||
}
|
||||
}
|
||||
if (this.bytes > 4294967295) {
|
||||
this.hBytes += (this.bytes / 4294967296) << 0;
|
||||
this.bytes = this.bytes % 4294967296;
|
||||
if (this._bytes > 4294967295) {
|
||||
this._hBytes += (this._bytes / 4294967296) >>> 0;
|
||||
this._bytes = this._bytes >>> 0;
|
||||
}
|
||||
return this;
|
||||
}
|
||||
|
||||
finalize() {
|
||||
if (this.finalized) {
|
||||
if (this._finalized) {
|
||||
return;
|
||||
}
|
||||
this.finalized = true;
|
||||
let blocks = this.blocks,
|
||||
i = this.lastByteIndex;
|
||||
blocks[16] = this.block;
|
||||
this._finalized = true;
|
||||
let blocks = this._blocks,
|
||||
i = this._lastByteIndex;
|
||||
blocks[16] = this._block;
|
||||
blocks[i >> 2] |= EXTRA[i & 3];
|
||||
this.block = blocks[16];
|
||||
this._block = blocks[16];
|
||||
if (i >= 56) {
|
||||
if (!this.hashed) {
|
||||
if (!this._hashed) {
|
||||
this.hash();
|
||||
}
|
||||
blocks[0] = this.block;
|
||||
blocks[16] = blocks[1] = blocks[2] = blocks[3] = blocks[4] = blocks[5] = blocks[6] = blocks[7] = blocks[8] = blocks[9] = blocks[10] = blocks[11] = blocks[12] = blocks[13] = blocks[14] = blocks[15] = 0;
|
||||
blocks[0] = this._block;
|
||||
blocks.fill(0, 1, 17);
|
||||
}
|
||||
blocks[14] = (this.hBytes << 3) | (this.bytes >>> 29);
|
||||
blocks[15] = this.bytes << 3;
|
||||
blocks[14] = (this._hBytes << 3) | (this._bytes >>> 29);
|
||||
blocks[15] = this._bytes << 3;
|
||||
this.hash();
|
||||
}
|
||||
|
||||
hash() {
|
||||
let a = this.h0,
|
||||
b = this.h1,
|
||||
c = this.h2,
|
||||
d = this.h3,
|
||||
e = this.h4;
|
||||
let a = this._h0,
|
||||
b = this._h1,
|
||||
c = this._h2,
|
||||
d = this._h3,
|
||||
e = this._h4;
|
||||
let f,
|
||||
j,
|
||||
t,
|
||||
blocks = this.blocks;
|
||||
blocks = this._blocks;
|
||||
|
||||
for (j = 16; j < 80; ++j) {
|
||||
t = blocks[j - 3] ^ blocks[j - 8] ^ blocks[j - 14] ^ blocks[j - 16];
|
||||
|
@ -161,126 +161,126 @@ export class Sha1 {
|
|||
for (j = 0; j < 20; j += 5) {
|
||||
f = (b & c) | (~b & d);
|
||||
t = (a << 5) | (a >>> 27);
|
||||
e = (t + f + e + 1518500249 + blocks[j]) << 0;
|
||||
e = (t + f + e + 1518500249 + blocks[j]) >>> 0;
|
||||
b = (b << 30) | (b >>> 2);
|
||||
|
||||
f = (a & b) | (~a & c);
|
||||
t = (e << 5) | (e >>> 27);
|
||||
d = (t + f + d + 1518500249 + blocks[j + 1]) << 0;
|
||||
d = (t + f + d + 1518500249 + blocks[j + 1]) >>> 0;
|
||||
a = (a << 30) | (a >>> 2);
|
||||
|
||||
f = (e & a) | (~e & b);
|
||||
t = (d << 5) | (d >>> 27);
|
||||
c = (t + f + c + 1518500249 + blocks[j + 2]) << 0;
|
||||
c = (t + f + c + 1518500249 + blocks[j + 2]) >>> 0;
|
||||
e = (e << 30) | (e >>> 2);
|
||||
|
||||
f = (d & e) | (~d & a);
|
||||
t = (c << 5) | (c >>> 27);
|
||||
b = (t + f + b + 1518500249 + blocks[j + 3]) << 0;
|
||||
b = (t + f + b + 1518500249 + blocks[j + 3]) >>> 0;
|
||||
d = (d << 30) | (d >>> 2);
|
||||
|
||||
f = (c & d) | (~c & e);
|
||||
t = (b << 5) | (b >>> 27);
|
||||
a = (t + f + a + 1518500249 + blocks[j + 4]) << 0;
|
||||
a = (t + f + a + 1518500249 + blocks[j + 4]) >>> 0;
|
||||
c = (c << 30) | (c >>> 2);
|
||||
}
|
||||
|
||||
for (; j < 40; j += 5) {
|
||||
f = b ^ c ^ d;
|
||||
t = (a << 5) | (a >>> 27);
|
||||
e = (t + f + e + 1859775393 + blocks[j]) << 0;
|
||||
e = (t + f + e + 1859775393 + blocks[j]) >>> 0;
|
||||
b = (b << 30) | (b >>> 2);
|
||||
|
||||
f = a ^ b ^ c;
|
||||
t = (e << 5) | (e >>> 27);
|
||||
d = (t + f + d + 1859775393 + blocks[j + 1]) << 0;
|
||||
d = (t + f + d + 1859775393 + blocks[j + 1]) >>> 0;
|
||||
a = (a << 30) | (a >>> 2);
|
||||
|
||||
f = e ^ a ^ b;
|
||||
t = (d << 5) | (d >>> 27);
|
||||
c = (t + f + c + 1859775393 + blocks[j + 2]) << 0;
|
||||
c = (t + f + c + 1859775393 + blocks[j + 2]) >>> 0;
|
||||
e = (e << 30) | (e >>> 2);
|
||||
|
||||
f = d ^ e ^ a;
|
||||
t = (c << 5) | (c >>> 27);
|
||||
b = (t + f + b + 1859775393 + blocks[j + 3]) << 0;
|
||||
b = (t + f + b + 1859775393 + blocks[j + 3]) >>> 0;
|
||||
d = (d << 30) | (d >>> 2);
|
||||
|
||||
f = c ^ d ^ e;
|
||||
t = (b << 5) | (b >>> 27);
|
||||
a = (t + f + a + 1859775393 + blocks[j + 4]) << 0;
|
||||
a = (t + f + a + 1859775393 + blocks[j + 4]) >>> 0;
|
||||
c = (c << 30) | (c >>> 2);
|
||||
}
|
||||
|
||||
for (; j < 60; j += 5) {
|
||||
f = (b & c) | (b & d) | (c & d);
|
||||
t = (a << 5) | (a >>> 27);
|
||||
e = (t + f + e - 1894007588 + blocks[j]) << 0;
|
||||
e = (t + f + e - 1894007588 + blocks[j]) >>> 0;
|
||||
b = (b << 30) | (b >>> 2);
|
||||
|
||||
f = (a & b) | (a & c) | (b & c);
|
||||
t = (e << 5) | (e >>> 27);
|
||||
d = (t + f + d - 1894007588 + blocks[j + 1]) << 0;
|
||||
d = (t + f + d - 1894007588 + blocks[j + 1]) >>> 0;
|
||||
a = (a << 30) | (a >>> 2);
|
||||
|
||||
f = (e & a) | (e & b) | (a & b);
|
||||
t = (d << 5) | (d >>> 27);
|
||||
c = (t + f + c - 1894007588 + blocks[j + 2]) << 0;
|
||||
c = (t + f + c - 1894007588 + blocks[j + 2]) >>> 0;
|
||||
e = (e << 30) | (e >>> 2);
|
||||
|
||||
f = (d & e) | (d & a) | (e & a);
|
||||
t = (c << 5) | (c >>> 27);
|
||||
b = (t + f + b - 1894007588 + blocks[j + 3]) << 0;
|
||||
b = (t + f + b - 1894007588 + blocks[j + 3]) >>> 0;
|
||||
d = (d << 30) | (d >>> 2);
|
||||
|
||||
f = (c & d) | (c & e) | (d & e);
|
||||
t = (b << 5) | (b >>> 27);
|
||||
a = (t + f + a - 1894007588 + blocks[j + 4]) << 0;
|
||||
a = (t + f + a - 1894007588 + blocks[j + 4]) >>> 0;
|
||||
c = (c << 30) | (c >>> 2);
|
||||
}
|
||||
|
||||
for (; j < 80; j += 5) {
|
||||
f = b ^ c ^ d;
|
||||
t = (a << 5) | (a >>> 27);
|
||||
e = (t + f + e - 899497514 + blocks[j]) << 0;
|
||||
e = (t + f + e - 899497514 + blocks[j]) >>> 0;
|
||||
b = (b << 30) | (b >>> 2);
|
||||
|
||||
f = a ^ b ^ c;
|
||||
t = (e << 5) | (e >>> 27);
|
||||
d = (t + f + d - 899497514 + blocks[j + 1]) << 0;
|
||||
d = (t + f + d - 899497514 + blocks[j + 1]) >>> 0;
|
||||
a = (a << 30) | (a >>> 2);
|
||||
|
||||
f = e ^ a ^ b;
|
||||
t = (d << 5) | (d >>> 27);
|
||||
c = (t + f + c - 899497514 + blocks[j + 2]) << 0;
|
||||
c = (t + f + c - 899497514 + blocks[j + 2]) >>> 0;
|
||||
e = (e << 30) | (e >>> 2);
|
||||
|
||||
f = d ^ e ^ a;
|
||||
t = (c << 5) | (c >>> 27);
|
||||
b = (t + f + b - 899497514 + blocks[j + 3]) << 0;
|
||||
b = (t + f + b - 899497514 + blocks[j + 3]) >>> 0;
|
||||
d = (d << 30) | (d >>> 2);
|
||||
|
||||
f = c ^ d ^ e;
|
||||
t = (b << 5) | (b >>> 27);
|
||||
a = (t + f + a - 899497514 + blocks[j + 4]) << 0;
|
||||
a = (t + f + a - 899497514 + blocks[j + 4]) >>> 0;
|
||||
c = (c << 30) | (c >>> 2);
|
||||
}
|
||||
|
||||
this.h0 = (this.h0 + a) << 0;
|
||||
this.h1 = (this.h1 + b) << 0;
|
||||
this.h2 = (this.h2 + c) << 0;
|
||||
this.h3 = (this.h3 + d) << 0;
|
||||
this.h4 = (this.h4 + e) << 0;
|
||||
this._h0 = (this._h0 + a) >>> 0;
|
||||
this._h1 = (this._h1 + b) >>> 0;
|
||||
this._h2 = (this._h2 + c) >>> 0;
|
||||
this._h3 = (this._h3 + d) >>> 0;
|
||||
this._h4 = (this._h4 + e) >>> 0;
|
||||
}
|
||||
|
||||
hex() {
|
||||
this.finalize();
|
||||
|
||||
let h0 = this.h0,
|
||||
h1 = this.h1,
|
||||
h2 = this.h2,
|
||||
h3 = this.h3,
|
||||
h4 = this.h4;
|
||||
let h0 = this._h0,
|
||||
h1 = this._h1,
|
||||
h2 = this._h2,
|
||||
h3 = this._h3,
|
||||
h4 = this._h4;
|
||||
|
||||
return (
|
||||
HEX_CHARS[(h0 >> 28) & 0x0f] +
|
||||
|
@ -333,11 +333,11 @@ export class Sha1 {
|
|||
digest() {
|
||||
this.finalize();
|
||||
|
||||
let h0 = this.h0,
|
||||
h1 = this.h1,
|
||||
h2 = this.h2,
|
||||
h3 = this.h3,
|
||||
h4 = this.h4;
|
||||
let h0 = this._h0,
|
||||
h1 = this._h1,
|
||||
h2 = this._h2,
|
||||
h3 = this._h3,
|
||||
h4 = this._h4;
|
||||
|
||||
return [
|
||||
(h0 >> 24) & 0xff,
|
||||
|
@ -369,14 +369,8 @@ export class Sha1 {
|
|||
|
||||
arrayBuffer() {
|
||||
this.finalize();
|
||||
|
||||
let buffer = new ArrayBuffer(20);
|
||||
let dataView = new DataView(buffer);
|
||||
dataView.setUint32(0, this.h0);
|
||||
dataView.setUint32(4, this.h1);
|
||||
dataView.setUint32(8, this.h2);
|
||||
dataView.setUint32(12, this.h3);
|
||||
dataView.setUint32(16, this.h4);
|
||||
return buffer;
|
||||
return Uint32Array.of(
|
||||
this._h0, this._h1, this._h2, this._h3, this._h4
|
||||
).buffer;
|
||||
}
|
||||
}
|
||||
|
|
|
@ -1,8 +1,17 @@
|
|||
import { assertEqual, test } from "../testing/mod.ts";
|
||||
import { assert, test } from "../testing/mod.ts";
|
||||
import { Sha1 } from "./sha1.ts";
|
||||
|
||||
test(function testSha1() {
|
||||
const sha1 = new Sha1();
|
||||
let sha1 = new Sha1();
|
||||
sha1.update("abcde");
|
||||
assertEqual(sha1.toString(), "03de6c570bfe24bfc328ccd7ca46b76eadaf4334");
|
||||
assert.equal(sha1.toString(), "03de6c570bfe24bfc328ccd7ca46b76eadaf4334");
|
||||
|
||||
const data = Uint8Array.of(0x61, 0x62, 0x63, 0x64, 0x65);
|
||||
sha1 = new Sha1();
|
||||
sha1.update(data);
|
||||
assert.equal(sha1.toString(), "03de6c570bfe24bfc328ccd7ca46b76eadaf4334");
|
||||
|
||||
sha1 = new Sha1();
|
||||
sha1.update(data.buffer);
|
||||
assert.equal(sha1.toString(), "03de6c570bfe24bfc328ccd7ca46b76eadaf4334");
|
||||
});
|
||||
|
|
Loading…
Reference in a new issue