65 lines
2.1 KiB
JavaScript
65 lines
2.1 KiB
JavaScript
"use strict";
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
exports.ChecksumStream = void 0;
|
|
const util_base64_1 = require("@smithy/util-base64");
|
|
const stream_1 = require("stream");
|
|
class ChecksumStream extends stream_1.Duplex {
|
|
expectedChecksum;
|
|
checksumSourceLocation;
|
|
checksum;
|
|
source;
|
|
base64Encoder;
|
|
pendingCallback = null;
|
|
constructor({ expectedChecksum, checksum, source, checksumSourceLocation, base64Encoder, }) {
|
|
super();
|
|
if (typeof source.pipe === "function") {
|
|
this.source = source;
|
|
}
|
|
else {
|
|
throw new Error(`@smithy/util-stream: unsupported source type ${source?.constructor?.name ?? source} in ChecksumStream.`);
|
|
}
|
|
this.base64Encoder = base64Encoder ?? util_base64_1.toBase64;
|
|
this.expectedChecksum = expectedChecksum;
|
|
this.checksum = checksum;
|
|
this.checksumSourceLocation = checksumSourceLocation;
|
|
this.source.pipe(this);
|
|
}
|
|
_read(size) {
|
|
if (this.pendingCallback) {
|
|
const callback = this.pendingCallback;
|
|
this.pendingCallback = null;
|
|
callback();
|
|
}
|
|
}
|
|
_write(chunk, encoding, callback) {
|
|
try {
|
|
this.checksum.update(chunk);
|
|
const canPushMore = this.push(chunk);
|
|
if (!canPushMore) {
|
|
this.pendingCallback = callback;
|
|
return;
|
|
}
|
|
}
|
|
catch (e) {
|
|
return callback(e);
|
|
}
|
|
return callback();
|
|
}
|
|
async _final(callback) {
|
|
try {
|
|
const digest = await this.checksum.digest();
|
|
const received = this.base64Encoder(digest);
|
|
if (this.expectedChecksum !== received) {
|
|
return callback(new Error(`Checksum mismatch: expected "${this.expectedChecksum}" but received "${received}"` +
|
|
` in response header "${this.checksumSourceLocation}".`));
|
|
}
|
|
}
|
|
catch (e) {
|
|
return callback(e);
|
|
}
|
|
this.push(null);
|
|
return callback();
|
|
}
|
|
}
|
|
exports.ChecksumStream = ChecksumStream;
|