nodes/pythongosssss-ComfyUI-Custom-Scripts
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

replaced crypto.subtle.digest with simple, unsafe hash to avoid browser error under insecure context

Browse Source
This commit is contained in:
cho
2024-03-11 11:40:13 -07:00
parent ee9599827b
commit 854bc6cd2d
1 changed files with 19 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files
web/js/imageFeed.js
+19 -22
View File
@@ -450,6 +450,7 @@ app.registerExtension({
if (deduplicateFeed.value) {
// deduplicate by ignoring images with the same filename/type/subfolder
const startTime = performance.now();
const fingerprint = JSON.stringify({ filename: src.filename, type: src.type, subfolder: src.subfolder });
if (seenImages.has(fingerprint)) {
// NOOP: image is a duplicate
@@ -458,6 +459,10 @@ app.registerExtension({
// deduplicate by ignoring images with the same content
let img = $el("img", { src: href })
img.onerror = () => {
// fall back to default behavior
addImageToFeed(href);
}
img.onload = () => {
// redraw the image onto a canvas to strip metadata
var imgCanvas = document.createElement("canvas");
@@ -469,29 +474,21 @@ app.registerExtension({
// and then hash the stripped image for memoization
var strippedImgB64 = imgCanvas.toDataURL("image/png");
var buf = new TextEncoder("utf-8").encode(strippedImgB64);
crypto.subtle.digest("SHA-1", buf).then((hash) => {
let hexes = [], view = new DataView(hash)
for (let i = 0; i < view.byteLength; i += 4) {
hexes.push(('00000000' + view.getUint32(i).toString(16)).slice(-8));
}
const hexHash = hexes.join('');
if (seenImages.has(hexHash)) {
// NOOP: image is a duplicate
} else {
// if we got to here, then the image is unique--so add to feed
seenImages.set(hexHash, true);
addImageToFeed(href);
}
}).catch((err) => {
// fall back to default behavior
console.error(err);
var simpleHash = Array.from(strippedImgB64).reduce((a, b) => (((a << 5) - a) + b.charCodeAt(0)) | 0, 0);
if (seenImages.has(simpleHash)) {
// NOOP: image is a duplicate
} else {
// if we got to here, then the image is unique--so add to feed
seenImages.set(simpleHash, true);
addImageToFeed(href);
});
}
img.onerror = () => {
// fall back to default behavior
addImageToFeed(href);
}
// calculate the size of the image based on the stripped base64 data: ((4 * n / 3) + 3) & ~3 bytes, for n b64 chars
// source: https://stackoverflow.com/a/32140193
const imgSizeBytes = ((4 * strippedImgB64.length / 3) + 3) & ~3;
const imgSizeMB = (imgSizeBytes / 1048576).toFixed(2);
const endTime = performance.now();
console.log("%c[🐍 pysssss]", "color: limegreen", `Image deduplication (${imgSizeMB} MB) took ${endTime - startTime}ms: ${fingerprint}`);
}
}
} else {