ZacharyZhang-NYandClaude Opus 5.5
7a9059fbea
Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
44 lines
3.1 KiB
TypeScript
44 lines
3.1 KiB
TypeScript
import { describe, expect, test } from "vitest";
|
|
import { scanFileContent } from "../../src/main/services/files/file-object-store";
|
|
|
|
const MIB = 1024 * 1024;
|
|
|
|
describe("the file scan", () => {
|
|
test("checks the whole file, not just its first MiB, including a secret that straddles a window edge", () => {
|
|
const padding = "x".repeat(3 * MIB);
|
|
expect(scanFileContent("notes.txt", Buffer.from(`${padding}\nAKIAABCDEFGHIJKLMNOP\n`)).state).toBe("blocked");
|
|
const straddling = Buffer.concat([Buffer.alloc(MIB - 10, "y"), Buffer.from("-----BEGIN RSA PRIVATE KEY-----")]);
|
|
expect(scanFileContent("key.txt", straddling).labels).toContain("dlp:private-key");
|
|
expect(scanFileContent("eicar.txt", Buffer.from(`${padding}X5O!P%@AP[4\\PZX54(P^)7CC)7}$EICAR-STANDARD-ANTIVIRUS-TEST-FILE!$H+H*`)).labels).toContain("av:eicar");
|
|
});
|
|
|
|
test("no pattern match can stretch past the window overlap", () => {
|
|
const edge = Buffer.concat([Buffer.alloc(MIB - 20, "z"), Buffer.from(`password:${"\n".repeat(5000)}${"A".repeat(32)}`)]);
|
|
// Whitespace between a key name and its value is bounded; a value 5,000 lines away is not that key's value.
|
|
expect(scanFileContent("far.txt", edge).labels).not.toContain("dlp:credential");
|
|
const long = Buffer.concat([Buffer.alloc(MIB - 10, "z"), Buffer.from(` ghp_${"a".repeat(9000)} `)]);
|
|
expect(scanFileContent("token.txt", long).labels).toContain("dlp:github-token");
|
|
expect(scanFileContent("plan.md", Buffer.from("# Plan")).labels).toContain("scan:whole-file");
|
|
});
|
|
|
|
test("a window cut does not invent a word boundary", () => {
|
|
const inWord = Buffer.concat([Buffer.alloc(MIB - 4, "z"), Buffer.from("NOT_AKIAABCDEFGHIJKLMNOP\n")]);
|
|
expect(scanFileContent("a.txt", inWord).state).toBe("approved");
|
|
const alone = Buffer.concat([Buffer.alloc(MIB - 1, "z"), Buffer.from(" AKIAABCDEFGHIJKLMNOP\n")]);
|
|
expect(scanFileContent("b.txt", alone).state).toBe("blocked");
|
|
// The far edge too: a longer word whose first 20 characters end exactly at a window's last byte is not a key.
|
|
const atTail = Buffer.concat([Buffer.alloc(MIB + 4096 - 20, "z"), Buffer.from(" AKIAABCDEFGHIJKLMNOPZ\n")]);
|
|
expect(scanFileContent("c.txt", atTail).state).toBe("approved");
|
|
const atOwnEnd = Buffer.concat([Buffer.alloc(MIB - 20, "z"), Buffer.from(" AKIAABCDEFGHIJKLMNOPZ\n")]);
|
|
expect(scanFileContent("d.txt", atOwnEnd).state).toBe("approved");
|
|
});
|
|
|
|
test("a container whose contents are compressed or encoded is labelled as not inspected inside", () => {
|
|
expect(scanFileContent("report.docx", Buffer.from("PK\u0003\u0004rest")).labels).toContain("scan:contents-not-inspected");
|
|
expect(scanFileContent("deck.pdf", Buffer.from("%PDF-1.7 ...")).labels).toContain("scan:contents-not-inspected");
|
|
expect(scanFileContent("archive.gz", Buffer.from([0x1f, 0x8b, 0x08, 0x00])).labels).toContain("scan:contents-not-inspected");
|
|
expect(scanFileContent("photo.png", Buffer.from([0x89, 0x50, 0x4e, 0x47, 0x0d, 0x0a])).labels).toContain("scan:contents-not-inspected");
|
|
expect(scanFileContent("plan.md", Buffer.from("# Plan")).labels).not.toContain("scan:contents-not-inspected");
|
|
});
|
|
});
|