fix(dkim): Store byteLength in BodyHashStream

postalsys · Oct 2, 2024 · 081f823 · 081f823
1 parent a322180
commit 081f823
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/lib/dkim/body/index.js b/lib/dkim/body/index.js
@@ -43,6 +43,8 @@ class BodyHashStream extends Transform {
         this.finished = false;
         this.finishCb = null;
 
+        this.byteLength = 0;
+
         this.messageHasher = new MessageHasher(canonicalization, ...options);
         this.bodyHash = null;
         this.messageHasher.once('finish', () => this.finishHashing());
@@ -73,12 +75,15 @@ class BodyHashStream extends Transform {
             chunk = Buffer.from(chunk, encoding);
         }
 
+        this.byteLength += chunk.length;
+
+        this.push(chunk);
+
         if (this.messageHasher.write(chunk) === false) {
             // wait for drain
             return this.messageHasher.once('drain', done);
         }
 
-        this.push(chunk);
         done();
     }