|
| 1 | +/** |
| 2 | + * Performance benchmarks for HB Scrub. |
| 3 | + * |
| 4 | + * Run: |
| 5 | + * npx tsx bench/bench.ts |
| 6 | + * |
| 7 | + * Generates synthetic test data of varying sizes and measures throughput |
| 8 | + * for the core operations: detectFormat, removeMetadata, verifyClean. |
| 9 | + */ |
| 10 | + |
| 11 | +import { removeMetadataSync } from '../src/node.js'; |
| 12 | +import { detectFormat } from '../src/detect.js'; |
| 13 | +import { verifyCleanSync } from '../src/operations/verify.js'; |
| 14 | + |
| 15 | +// ─── Synthetic test image generators ───────────────────────────────────────── |
| 16 | + |
| 17 | +/** Minimal valid JPEG with APP1 EXIF marker */ |
| 18 | +function makeJpeg(payloadKB: number): Uint8Array { |
| 19 | + const exifPayload = new Uint8Array(payloadKB * 1024).fill(0x41); |
| 20 | + // SOI + APP1 marker |
| 21 | + const header = new Uint8Array([ |
| 22 | + 0xff, 0xd8, // SOI |
| 23 | + 0xff, 0xe1, // APP1 |
| 24 | + ...uint16BE(exifPayload.length + 2 + 6), // length |
| 25 | + 0x45, 0x78, 0x69, 0x66, 0x00, 0x00, // "Exif\0\0" |
| 26 | + ]); |
| 27 | + // Minimal SOS + EOI |
| 28 | + const tail = new Uint8Array([0xff, 0xda, 0x00, 0x02, 0xff, 0xd9]); |
| 29 | + return concat(header, exifPayload, tail); |
| 30 | +} |
| 31 | + |
| 32 | +/** Minimal valid PNG with tEXt chunk */ |
| 33 | +function makePng(payloadKB: number): Uint8Array { |
| 34 | + const sig = new Uint8Array([137, 80, 78, 71, 13, 10, 26, 10]); |
| 35 | + // Minimal IHDR |
| 36 | + const ihdr = pngChunk('IHDR', new Uint8Array([ |
| 37 | + 0, 0, 0, 1, 0, 0, 0, 1, 8, 2, 0, 0, 0, |
| 38 | + ])); |
| 39 | + // tEXt chunk with payload |
| 40 | + const keyword = new TextEncoder().encode('Comment\0'); |
| 41 | + const textData = new Uint8Array(payloadKB * 1024).fill(0x41); |
| 42 | + const text = pngChunk('tEXt', concat(keyword, textData)); |
| 43 | + // IEND |
| 44 | + const iend = pngChunk('IEND', new Uint8Array(0)); |
| 45 | + return concat(sig, ihdr, text, iend); |
| 46 | +} |
| 47 | + |
| 48 | +// ─── Helpers ───────────────────────────────────────────────────────────────── |
| 49 | + |
| 50 | +function uint16BE(n: number): number[] { |
| 51 | + return [(n >> 8) & 0xff, n & 0xff]; |
| 52 | +} |
| 53 | + |
| 54 | +function uint32BE(n: number): Uint8Array { |
| 55 | + return new Uint8Array([(n >>> 24) & 0xff, (n >>> 16) & 0xff, (n >>> 8) & 0xff, n & 0xff]); |
| 56 | +} |
| 57 | + |
| 58 | +function concat(...bufs: Uint8Array[]): Uint8Array { |
| 59 | + const len = bufs.reduce((s, b) => s + b.length, 0); |
| 60 | + const out = new Uint8Array(len); |
| 61 | + let off = 0; |
| 62 | + for (const b of bufs) { |
| 63 | + out.set(b, off); |
| 64 | + off += b.length; |
| 65 | + } |
| 66 | + return out; |
| 67 | +} |
| 68 | + |
| 69 | +function pngChunk(type: string, payload: Uint8Array): Uint8Array { |
| 70 | + const typeBytes = new TextEncoder().encode(type); |
| 71 | + const len = uint32BE(payload.length); |
| 72 | + // CRC placeholder (good enough for benchmarking) |
| 73 | + const crc = uint32BE(0); |
| 74 | + return concat(len, typeBytes, payload, crc); |
| 75 | +} |
| 76 | + |
| 77 | +// ─── Benchmark runner ──────────────────────────────────────────────────────── |
| 78 | + |
| 79 | +interface BenchResult { |
| 80 | + name: string; |
| 81 | + opsPerSec: number; |
| 82 | + avgMs: number; |
| 83 | + runs: number; |
| 84 | +} |
| 85 | + |
| 86 | +function bench(name: string, fn: () => void, durationMs = 2000): BenchResult { |
| 87 | + // Warm up |
| 88 | + for (let i = 0; i < 5; i++) fn(); |
| 89 | + |
| 90 | + let runs = 0; |
| 91 | + const start = performance.now(); |
| 92 | + while (performance.now() - start < durationMs) { |
| 93 | + fn(); |
| 94 | + runs++; |
| 95 | + } |
| 96 | + const elapsed = performance.now() - start; |
| 97 | + return { |
| 98 | + name, |
| 99 | + opsPerSec: Math.round((runs / elapsed) * 1000), |
| 100 | + avgMs: Number((elapsed / runs).toFixed(3)), |
| 101 | + runs, |
| 102 | + }; |
| 103 | +} |
| 104 | + |
| 105 | +function printResults(results: BenchResult[]): void { |
| 106 | + console.log('\n┌────────────────────────────────────────┬────────────┬──────────┬────────┐'); |
| 107 | + console.log('│ Benchmark │ ops/sec │ avg ms │ runs │'); |
| 108 | + console.log('├────────────────────────────────────────┼────────────┼──────────┼────────┤'); |
| 109 | + for (const r of results) { |
| 110 | + console.log( |
| 111 | + `│ ${r.name.padEnd(38)} │ ${String(r.opsPerSec).padStart(10)} │ ${String(r.avgMs).padStart(8)} │ ${String(r.runs).padStart(6)} │` |
| 112 | + ); |
| 113 | + } |
| 114 | + console.log('└────────────────────────────────────────┴────────────┴──────────┴────────┘\n'); |
| 115 | +} |
| 116 | + |
| 117 | +// ─── Main ──────────────────────────────────────────────────────────────────── |
| 118 | + |
| 119 | +const sizes = [1, 10, 100, 500]; |
| 120 | +const results: BenchResult[] = []; |
| 121 | + |
| 122 | +for (const kb of sizes) { |
| 123 | + const jpegData = makeJpeg(kb); |
| 124 | + const pngData = makePng(kb); |
| 125 | + |
| 126 | + results.push(bench(`detectFormat JPEG ${kb}KB`, () => detectFormat(jpegData))); |
| 127 | + results.push(bench(`detectFormat PNG ${kb}KB`, () => detectFormat(pngData))); |
| 128 | + results.push(bench(`removeSync JPEG ${kb}KB`, () => removeMetadataSync(jpegData))); |
| 129 | + results.push(bench(`removeSync PNG ${kb}KB`, () => removeMetadataSync(pngData))); |
| 130 | + results.push(bench(`verifyClean JPEG ${kb}KB`, () => verifyCleanSync(jpegData))); |
| 131 | + results.push(bench(`verifyClean PNG ${kb}KB`, () => verifyCleanSync(pngData))); |
| 132 | +} |
| 133 | + |
| 134 | +printResults(results); |
0 commit comments