Fix sha2 check, can be dynamic

paulmillr · Dec 2, 2022 · 770f8a3 · 770f8a3
1 parent 0d37cca
commit 770f8a3
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/_sha2.ts b/src/_sha2.ts
@@ -38,7 +38,6 @@ export abstract class SHA2<T extends SHA2<T>> extends Hash<T> {
     super();
     this.buffer = new Uint8Array(blockLen);
     this.view = createView(this.buffer);
-    if (this.outputLen % 4) throw new Error('_sha2: outputLen should be aligned to 32bit');
   }
   update(data: Input): this {
     assert.exists(this);
@@ -90,7 +89,9 @@ export abstract class SHA2<T extends SHA2<T>> extends Hash<T> {
     setBigUint64(view, blockLen - 8, BigInt(this.length * 8), isLE);
     this.process(view, 0);
     const oview = createView(out);
-    const outLen = this.outputLen / 4;
+    const oLen = this.outputLen;
+    if (oLen % 4) throw new Error('_sha2: outputLen should be aligned to 32bit');
+    const outLen = oLen / 4;
     const state = this.get();
     if (outLen > state.length) throw new Error('_sha2: outputLen bigger than state');
     for (let i = 0; i < outLen; i++) oview.setUint32(4 * i, state[i], isLE);