mirror of
https://github.com/morluto/rea.git
synced 2026-11-14 10:43:30 +00:00
morluto
b138f52e12
* refactor: prune internal seams and preserve configuration ownership * fix(setup): keep YAML aliases and inherited registration settings on refresh - Reuse an unchanged YAML collection only while every alias in it still resolves after the edit; otherwise rebuild it so the alias becomes its value. A sibling anchor replaced or moved in the same refresh no longer leaves an unresolved alias in Hermes or Goose output. - Keep a connection selector whose existing value already chooses stdio (for example an inherited Hermes transport: stdio) instead of superseding it; remote selectors are still replaced. Explicit stdio selectors are therefore no longer rewritten (Kimi expectation updated). - When the whole Hermes registration is inherited through a merge, override it with that merge and write only the changed fields. * fix(setup): edit registration fields whose YAML key is an alias A key written as an alias to a string scalar (for example *field with field: &field env) was treated as a source-only key, so any change to its value was refused in Hermes and Goose. Resolve such keys to their string, keep the aliased spelling while its anchor survives the edit, and write the plain key when the same edit rebuilds the anchor. * test(process): bound cleanup-boundary waits and always release gates Both cancellation tests awaited a bare started promise and released their gates only on success, so a missed boundary waited for the outer deadline and could leave cleanup parked. Use awaitAcknowledgement and release and join the cleanup in finally. * fix(setup): repair only aliases an edit leaves unresolved Replace the pre-edit alias survival guess, which expanded aliases to sibling anchors that survive unchanged, with an exact repair: record what each alias in the replaced value resolves to, write the edit, then replace only aliases left without a preceding anchor by that value. - Keep an aliased key's source position, so its anchor stays ahead of later aliases. - Treat an aliased key as its string only when no other key in the map resolves to the same string; an ambiguous one stays guarded. - Override an inherited Hermes registration only when the requested entry keeps every inherited field, so an explicit replacement can drop remote selectors. * fix(setup): repair aliases whose binding an edit changes, keeping comments An alias whose own anchor was removed could resolve to an earlier anchor with the same name and silently change value. Record the node each alias resolved to and repair any alias that no longer resolves to that node. Carry the alias's comments and spacing onto the repaired value. * test(setup): split the Hermes anchor and merge refresh cases * fix(setup): refuse removing an overridden key its merge would restore The merge-retention check skipped keys the mapping also overrides, so dropping such a key (for example a remote url during an explicit stdio replacement) let the retained merge restore it. Check removal before skipping overridden keys, so the edit is refused with the existing message instead of producing a registration that cannot be applied.
1433 lines
47 KiB
TypeScript
1433 lines
47 KiB
TypeScript
import { expect, it } from "vitest";
|
|
import { writeFile, unlink } from "node:fs/promises";
|
|
import { join } from "node:path";
|
|
import { createHash } from "node:crypto";
|
|
import { deflateSync } from "node:zlib";
|
|
import { toolContract } from "../../src/contracts/toolContracts.js";
|
|
import { gpuCodeObjectsSchema } from "../../src/domain/gpu/gpuCodeObjects.js";
|
|
import {
|
|
gpuCodeObjectsPageViewResultSchema,
|
|
gpuCodeObjectsSummaryViewResultSchema,
|
|
} from "../../src/domain/analysisView/analysisViewResult.js";
|
|
import { parseEvidence, createEvidence } from "../../src/domain/evidence.js";
|
|
import { createTestTempDirectory } from "../fixtures/temporaryDirectory.js";
|
|
import { connectLocalToolsMcp } from "../fixtures/localToolsMcp.js";
|
|
import { parseMcpToolError } from "../fixtures/mcpToolError.js";
|
|
import { cliTest } from "../support/cli/cliFixture.js";
|
|
import {
|
|
cudaFixture,
|
|
peGpuFixture,
|
|
ccobFixture,
|
|
ptxFixture,
|
|
amdFixture,
|
|
amdNoteFixture,
|
|
cubinFixture,
|
|
hipFixture,
|
|
elfFixture,
|
|
llvmBitcodeFixture,
|
|
} from "../fixtures/gpu/codeObjects.js";
|
|
|
|
cliTest(
|
|
"reads CUDA/PTX and compressed HIP through the public CLI",
|
|
async ({ cli }) => {
|
|
const root = await createTestTempDirectory("rea-gpu-cli-");
|
|
for (const [name, bytes] of [
|
|
["cuda", cudaFixture("zstd")],
|
|
["hip", ccobFixture()],
|
|
["amd", amdFixture()],
|
|
] as const) {
|
|
const path = join(root, name);
|
|
await writeFile(path, bytes);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"kernels",
|
|
"--detail",
|
|
"complete",
|
|
...(name === "amd"
|
|
? ["--targets", '["amdgcn-amd-amdhsa--gfx900"]']
|
|
: []),
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const evidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
response.json,
|
|
),
|
|
);
|
|
const report = gpuCodeObjectsSchema.parse(evidence.normalized_result);
|
|
expect(report.artifact.sha256).toBe(
|
|
createHash("sha256").update(bytes).digest("hex"),
|
|
);
|
|
expect(report.kernels).toHaveLength(name === "cuda" ? 2 : 1);
|
|
expect(report.kernels[0]!.parameter_bytes).toBe(12);
|
|
expect(report.coverage.payloads).toBe("complete");
|
|
}
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"reports an exhausted ELF symbol string scan as capacity instead of corruption",
|
|
async ({ cli }) => {
|
|
const root = await createTestTempDirectory("rea-cubin-string-limit-");
|
|
const bytes = cubinFixture({ name: "k".repeat(65_536) });
|
|
const path = join(root, "long-symbol.cubin");
|
|
await writeFile(path, bytes);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"kernels",
|
|
"--detail",
|
|
"complete",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const evidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
response.json,
|
|
),
|
|
);
|
|
const report = gpuCodeObjectsSchema.parse(evidence.normalized_result);
|
|
expect(report.objects[0]).toMatchObject({
|
|
kind: "cubin",
|
|
inspection: "capacity",
|
|
});
|
|
expect(report.artifact.sha256).toBe(
|
|
createHash("sha256").update(bytes).digest("hex"),
|
|
);
|
|
expect(report.coverage.payloads).toBe("partial");
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"inventories a complete CCOB stream containing ordinary CCOB payload text",
|
|
async ({ cli }) => {
|
|
const root = await createTestTempDirectory("rea-ccob-embedded-marker-");
|
|
const bundle = hipFixture(
|
|
Buffer.from("ordinary CCOB marker"),
|
|
"host-other",
|
|
),
|
|
compressed = deflateSync(bundle, { level: 0 });
|
|
expect(compressed.indexOf("CCOB")).toBeGreaterThan(0);
|
|
for (const version of [1, 2] as const) {
|
|
const width = version === 1 ? 20 : 24,
|
|
header = Buffer.from(
|
|
ccobFixture(version, "zlib", bundle).subarray(0, width),
|
|
);
|
|
if (version === 2) header.writeUInt32LE(width + compressed.length, 8);
|
|
const bytes = Buffer.concat([header, compressed]),
|
|
path = join(root, `v${version}`);
|
|
await writeFile(path, bytes);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"inventory",
|
|
"--detail",
|
|
"complete",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const evidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
response.json,
|
|
),
|
|
);
|
|
const report = gpuCodeObjectsSchema.parse(evidence.normalized_result);
|
|
expect(report.coverage).toMatchObject({
|
|
discovery: "complete",
|
|
payloads: "not-requested",
|
|
});
|
|
expect(report.objects).toHaveLength(3);
|
|
expect(report.objects[0]).toMatchObject({
|
|
kind: "hip-bundle",
|
|
inspection: "complete",
|
|
encoded_size: compressed.length,
|
|
decoded_size: bundle.length,
|
|
sha256: createHash("sha256").update(bundle).digest("hex"),
|
|
location: { offset: width, size: compressed.length },
|
|
});
|
|
expect(
|
|
report.objects
|
|
.slice(1)
|
|
.map(({ bundle_id, inspection }) => [bundle_id, inspection]),
|
|
).toEqual([
|
|
["host-x86_64-unknown-linux", "complete"],
|
|
["host-other", "complete"],
|
|
]);
|
|
}
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"distinguishes unsupported HIP bitcode from corrupted ELF through the public CLI",
|
|
async ({ cli }) => {
|
|
const root = await createTestTempDirectory("rea-hip-bitcode-cli-"),
|
|
bitcode = llvmBitcodeFixture(),
|
|
corruptedElf = amdFixture(),
|
|
bundleId = "hip-amdgcn-amd-amdhsa--gfx900";
|
|
corruptedElf.writeBigUInt64LE(BigInt(corruptedElf.length + 1), 40);
|
|
for (const [name, payload, analysis, inspection] of [
|
|
["inventory", bitcode, "inventory", "inventory"],
|
|
["bitcode", bitcode, "kernels", "unsupported"],
|
|
["corrupt-elf", corruptedElf, "kernels", "malformed"],
|
|
] as const) {
|
|
const bytes = hipFixture(payload, bundleId),
|
|
path = join(root, name);
|
|
await writeFile(path, bytes);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
analysis,
|
|
"--targets",
|
|
'["gfx900"]',
|
|
"--detail",
|
|
"complete",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const evidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
response.json,
|
|
),
|
|
);
|
|
const report = gpuCodeObjectsSchema.parse(evidence.normalized_result);
|
|
const row = report.objects.find(({ kind }) => kind === "amdgpu")!;
|
|
expect(row).toMatchObject({
|
|
bundle_id: bundleId,
|
|
target: "gfx900",
|
|
payload_target: null,
|
|
inspection,
|
|
encoded_size: payload.length,
|
|
location: {
|
|
container_id: null,
|
|
offset: bytes.length - payload.length,
|
|
size: payload.length,
|
|
},
|
|
sha256:
|
|
analysis === "inventory"
|
|
? null
|
|
: createHash("sha256").update(payload).digest("hex"),
|
|
});
|
|
expect(report.coverage.payloads).toBe(
|
|
analysis === "inventory" ? "not-requested" : "partial",
|
|
);
|
|
expect(report.kernels).toEqual([]);
|
|
if (name === "bitcode") {
|
|
expect(row.limitations.join(" ")).toContain("LLVM bitcode");
|
|
expect(row.limitations.join(" ")).toContain(
|
|
"validity and inner target were not inspected",
|
|
);
|
|
}
|
|
}
|
|
const path = join(root, "raw-bitcode");
|
|
await writeFile(path, bitcode);
|
|
const raw = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"kernels",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(raw.exitCode).toBe(1);
|
|
expect(raw.json).toMatchObject({ code: "unsupported_target" });
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"keeps embedded Clang bundle reads inside their native section",
|
|
async ({ cli }) => {
|
|
const root = await createTestTempDirectory("rea-gpu-section-range-");
|
|
const magic = Buffer.from("__CLANG_OFFLOAD_BUNDLE__"),
|
|
truncated = magic,
|
|
complete = Buffer.concat([magic, Buffer.alloc(8)]);
|
|
const cases = [
|
|
["truncated-zero-neighbor", truncated, Buffer.alloc(8)],
|
|
[
|
|
"truncated-one-neighbor",
|
|
truncated,
|
|
Buffer.from([1, ...Array(7).fill(0)]),
|
|
],
|
|
["complete", complete, Buffer.alloc(8)],
|
|
] as const;
|
|
const reports = [];
|
|
for (const [name, bundle, adjacent] of cases) {
|
|
const bytes = elfFixture(62, [
|
|
{ name: ".hip_fatbin", bytes: bundle },
|
|
{ name: ".data", bytes: adjacent },
|
|
]),
|
|
path = join(root, name);
|
|
await writeFile(path, bytes);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"kernels",
|
|
"--detail",
|
|
"complete",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const evidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
response.json,
|
|
),
|
|
);
|
|
const report = gpuCodeObjectsSchema.parse(evidence.normalized_result);
|
|
expect(report.artifact.sha256).toBe(
|
|
createHash("sha256").update(bytes).digest("hex"),
|
|
);
|
|
reports.push(report);
|
|
}
|
|
const [zeroNeighbor, oneNeighbor, completeReport] = reports;
|
|
expect(zeroNeighbor!.coverage).toMatchObject({
|
|
discovery: "partial",
|
|
payloads: "partial",
|
|
});
|
|
expect(oneNeighbor!.coverage).toMatchObject({
|
|
discovery: "partial",
|
|
payloads: "partial",
|
|
});
|
|
expect(zeroNeighbor!.objects[0]).toMatchObject({
|
|
kind: "native-section",
|
|
inspection: "malformed",
|
|
location: { size: 24, section_name: ".hip_fatbin" },
|
|
});
|
|
expect(oneNeighbor!.objects[0]).toMatchObject({
|
|
kind: "native-section",
|
|
inspection: "malformed",
|
|
location: { size: 24, section_name: ".hip_fatbin" },
|
|
});
|
|
expect(completeReport!.coverage).toMatchObject({
|
|
discovery: "complete",
|
|
payloads: "complete",
|
|
});
|
|
expect(completeReport!.objects[0]).toMatchObject({
|
|
kind: "native-section",
|
|
inspection: "complete",
|
|
location: { size: 32, section_name: ".hip_fatbin" },
|
|
});
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"compares equivalent PTX targets and retains family-specific CUDA targets",
|
|
async ({ cli }) => {
|
|
const root = await createTestTempDirectory("rea-gpu-target-spelling-");
|
|
for (const [name, target, payloadTarget, ptx, flags] of [
|
|
[
|
|
"compute",
|
|
"sm_80",
|
|
"compute_80",
|
|
ptxFixture.replace(".target sm_80", ".target compute_80"),
|
|
0,
|
|
],
|
|
[
|
|
"family",
|
|
"sm_100f",
|
|
"sm_100f",
|
|
ptxFixture.replace(".target sm_80", ".target sm_100f"),
|
|
0x200000,
|
|
],
|
|
] as const) {
|
|
const bytes = cudaFixture("none", [
|
|
{
|
|
kind: 1,
|
|
target: name === "compute" ? 80 : 100,
|
|
bytes: Buffer.from(ptx),
|
|
flags,
|
|
},
|
|
]),
|
|
path = join(root, name);
|
|
await writeFile(path, bytes);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"kernels",
|
|
"--detail",
|
|
"complete",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const evidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
response.json,
|
|
),
|
|
);
|
|
const report = gpuCodeObjectsSchema.parse(evidence.normalized_result);
|
|
expect(report.objects[0]).toMatchObject({
|
|
target,
|
|
payload_target: payloadTarget,
|
|
target_consistency: "consistent",
|
|
});
|
|
expect(report.coverage.payloads).toBe("complete");
|
|
}
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"reads CUDA V2 parameter records through the public CLI",
|
|
async ({ cli }) => {
|
|
const root = await createTestTempDirectory("rea-cubin-v2-parameter-"),
|
|
bytes = cudaFixture("none", [
|
|
{
|
|
kind: 2,
|
|
target: 80,
|
|
bytes: cubinFixture({ architecture: 80, v2ParameterSize: 20000 }),
|
|
},
|
|
]),
|
|
path = join(root, "large-parameter.cubin");
|
|
await writeFile(path, bytes);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"kernels",
|
|
"--detail",
|
|
"complete",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const evidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
response.json,
|
|
),
|
|
);
|
|
const report = gpuCodeObjectsSchema.parse(evidence.normalized_result);
|
|
expect(report.kernels[0]).toMatchObject({
|
|
parameter_count: 1,
|
|
parameter_bytes: 20000,
|
|
parameters: [{ ordinal: 0, offset: 0, size: 20000 }],
|
|
});
|
|
expect(report.coverage.payloads).toBe("complete");
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"preserves HIP v3 feature target IDs for exact selection",
|
|
async ({ cli }) => {
|
|
const target = "amdgcn-amd-amdhsa--gfx906+xnack",
|
|
root = await createTestTempDirectory("rea-hip-v3-target-id-"),
|
|
payload = amdFixture(false, undefined, target);
|
|
payload.writeUInt32LE(0x2f, 48); // EF_AMDGPU_MACH_AMDGCN_GFX906
|
|
const bytes = hipFixture(payload, "hip-amdgcn-amd-amdhsa--gfx906+xnack"),
|
|
path = join(root, "feature.bundle");
|
|
await writeFile(path, bytes);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"kernels",
|
|
"--targets",
|
|
JSON.stringify(["gfx906+xnack"]),
|
|
"--detail",
|
|
"complete",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const evidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
response.json,
|
|
),
|
|
);
|
|
const report = gpuCodeObjectsSchema.parse(evidence.normalized_result);
|
|
expect(report.objects.find((row) => row.kind === "amdgpu")!.target).toBe(
|
|
"gfx906+xnack",
|
|
);
|
|
expect(report.kernels).toHaveLength(1);
|
|
expect(report.coverage.payloads).toBe("complete");
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"keeps PTX qualifier layouts and compact registers while excluding call prototypes",
|
|
async ({ cli }) => {
|
|
const source = ptxFixture
|
|
.replace(".u64 output", ".u64 .ptr .global output")
|
|
.replace(
|
|
".u32 value",
|
|
".u64 .ptr .global .align 4 ptr_arg, .param .u64 .ptr .align 1 increment_param_0, .param .u64 .ptr.global.align 4 joined_arg, .param .u64 .ptr .global.align 16 partly_joined_arg, .param .u64 .ptr.global .align 16 joined_space_arg, .param .b8 .align 8 payload[8], .param .u32 value",
|
|
)
|
|
.replace(".reg .b32", ".reg.b32")
|
|
.replace(
|
|
".reg .b64 %rd<5>;",
|
|
".reg .b64 %rd<5>; F: .callprototype _ (.reg.b32 _);",
|
|
),
|
|
root = await createTestTempDirectory("rea-ptx-callprototype-"),
|
|
path = join(root, "prototype.ptx");
|
|
await writeFile(path, source);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"kernels",
|
|
"--detail",
|
|
"complete",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const evidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
response.json,
|
|
),
|
|
);
|
|
const report = gpuCodeObjectsSchema.parse(evidence.normalized_result);
|
|
expect(report.kernels[0]).toMatchObject({
|
|
parameter_count: 8,
|
|
parameter_bytes: 60,
|
|
parameters: [
|
|
{ name: "output", offset: 0, size: 8 },
|
|
{ name: "ptr_arg", offset: 8, size: 8 },
|
|
{ name: "increment_param_0", offset: 16, size: 8 },
|
|
{ name: "joined_arg", offset: 24, size: 8 },
|
|
{ name: "partly_joined_arg", offset: 32, size: 8 },
|
|
{ name: "joined_space_arg", offset: 40, size: 8 },
|
|
{ name: "payload", offset: 48, size: 8 },
|
|
{ name: "value", offset: 56, size: 4 },
|
|
],
|
|
registers: {
|
|
virtual: [
|
|
{ type: ".b32", count: 9 },
|
|
{ type: ".b64", count: 5 },
|
|
],
|
|
},
|
|
});
|
|
expect(report.coverage.payloads).toBe("complete");
|
|
expect(report.ptx_artifacts[0]!.text).toBe(source);
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"resolves CUBIN SHN_XINDEX symbols through their ELF extension table",
|
|
async ({ cli }) => {
|
|
const root = await createTestTempDirectory("rea-cubin-shndx-cli-");
|
|
const bytes = cubinFixture({ extendedSectionIndex: true });
|
|
const path = join(root, "extended.cubin");
|
|
await writeFile(path, bytes);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"kernels",
|
|
"--detail",
|
|
"complete",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const evidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
response.json,
|
|
),
|
|
);
|
|
const report = gpuCodeObjectsSchema.parse(evidence.normalized_result);
|
|
const sha256 = createHash("sha256").update(bytes).digest("hex");
|
|
expect(report.artifact.sha256).toBe(sha256);
|
|
expect(report.objects[0]!.sha256).toBe(sha256);
|
|
expect(report.kernels).toHaveLength(1);
|
|
expect(report.kernels[0]).toMatchObject({
|
|
name: "add_one",
|
|
parameter_count: 2,
|
|
parameter_bytes: 12,
|
|
shared_bytes: 1056,
|
|
sources: {
|
|
parameters: expect.objectContaining({ section_index: 65537 }),
|
|
shared: expect.objectContaining({ section_index: 65538 }),
|
|
},
|
|
});
|
|
|
|
const malformedBytes = Buffer.from(bytes);
|
|
const sectionTable = Number(malformedBytes.readBigUInt64LE(40));
|
|
const sectionCount = Number(
|
|
malformedBytes.readBigUInt64LE(sectionTable + 32),
|
|
);
|
|
let indexTable: number | undefined;
|
|
for (let index = 0; index < sectionCount; index++)
|
|
if (malformedBytes.readUInt32LE(sectionTable + index * 64 + 4) === 18) {
|
|
indexTable = index;
|
|
break;
|
|
}
|
|
expect(indexTable).toBeDefined();
|
|
malformedBytes.writeBigUInt64LE(4n, sectionTable + indexTable! * 64 + 32);
|
|
const malformedPath = join(root, "short-shndx.cubin");
|
|
await writeFile(malformedPath, malformedBytes);
|
|
const malformed = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
malformedPath,
|
|
"--analysis",
|
|
"kernels",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(malformed.exitCode, malformed.stderr).toBe(0);
|
|
const malformedEvidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
malformed.json,
|
|
),
|
|
);
|
|
const malformedReport = gpuCodeObjectsSchema.parse(
|
|
malformedEvidence.normalized_result,
|
|
);
|
|
expect(malformedReport.objects[0]!.inspection).toBe("malformed");
|
|
expect(malformedReport.coverage.payloads).toBe("partial");
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"counts numeric percent-prefixed PTX register declarations",
|
|
async ({ cli }) => {
|
|
const root = await createTestTempDirectory("rea-ptx-numeric-register-");
|
|
for (const register of ["%0", "%r0"] as const) {
|
|
const text = `.version 7.0\n.target sm_80\n.address_size 64\n.visible .entry kernel() {\n.reg .b32 ${register};\nmov.u32 ${register}, 0;\nret;\n}\n`;
|
|
const path = join(root, `${register.slice(1)}.ptx`);
|
|
await writeFile(path, text);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"kernels",
|
|
"--detail",
|
|
"complete",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const report = gpuCodeObjectsSchema.parse(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
response.json,
|
|
).normalized_result,
|
|
);
|
|
expect(report.kernels).toMatchObject([
|
|
{
|
|
name: "kernel",
|
|
registers: { virtual: [{ type: ".b32", count: 1 }] },
|
|
},
|
|
]);
|
|
expect(report.ptx_artifacts[0]!.sha256).toBe(
|
|
createHash("sha256").update(text).digest("hex"),
|
|
);
|
|
}
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"preserves PTX target spelling and percent-prefixed entry names",
|
|
async ({ cli }) => {
|
|
const root = await createTestTempDirectory("rea-ptx-identifiers-cli-");
|
|
const text = `.version 8.8\n.target sm_80\n.address_size 64\n.entry %kernel(.param .u32 input) {\nret;\n}\n`;
|
|
const path = join(root, "identifiers.ptx");
|
|
await writeFile(path, text);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"kernels",
|
|
"--detail",
|
|
"complete",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const evidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
response.json,
|
|
),
|
|
);
|
|
const report = gpuCodeObjectsSchema.parse(evidence.normalized_result);
|
|
expect(report.objects[0]!.target).toBe("sm_80");
|
|
expect(report.kernels).toEqual([
|
|
expect.objectContaining({
|
|
name: "%kernel",
|
|
parameter_count: 1,
|
|
parameter_bytes: 4,
|
|
parameters: [
|
|
expect.objectContaining({
|
|
ordinal: 0,
|
|
name: "input",
|
|
offset: 0,
|
|
size: 4,
|
|
type: ".u32",
|
|
}),
|
|
],
|
|
}),
|
|
]);
|
|
expect(report.ptx_artifacts[0]!.sha256).toBe(
|
|
createHash("sha256").update(text).digest("hex"),
|
|
);
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"keeps entry-scoped PTX pragmas outside the body boundary",
|
|
async ({ cli }) => {
|
|
const root = await createTestTempDirectory("rea-ptx-entry-pragma-");
|
|
const preamble = ".version 8.8\n.target sm_80\n.address_size 64\n",
|
|
body = "{ .reg .b32 %r<3>; .shared .align 4 .b8 scratch[16]; ret; }",
|
|
declarations = [
|
|
["plain", `.visible .entry k(.param .u32 input) ${body}`],
|
|
[
|
|
"entry-pragma",
|
|
`.visible .entry k(.param .u32 input) .pragma "nounroll"; ${body}`,
|
|
],
|
|
[
|
|
"body-pragma",
|
|
`.visible .entry k(.param .u32 input) { .pragma "nounroll"; .reg .b32 %r<3>; .shared .align 4 .b8 scratch[16]; ret; }`,
|
|
],
|
|
] as const;
|
|
const reports = new Map();
|
|
for (const [name, declaration] of declarations) {
|
|
const text = `${preamble}${declaration}\n`,
|
|
path = join(root, `${name}.ptx`);
|
|
await writeFile(path, text);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"kernels",
|
|
"--detail",
|
|
"complete",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const evidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
response.json,
|
|
),
|
|
);
|
|
const report = gpuCodeObjectsSchema.parse(evidence.normalized_result);
|
|
expect(report.ptx_artifacts[0]!.sha256).toBe(
|
|
createHash("sha256").update(text).digest("hex"),
|
|
);
|
|
reports.set(name, report);
|
|
}
|
|
const expected = {
|
|
name: "k",
|
|
parameter_count: 1,
|
|
parameter_bytes: 4,
|
|
registers: { virtual: [{ type: ".b32", count: 3 }] },
|
|
shared_bytes: 16,
|
|
};
|
|
for (const [name, report] of reports) {
|
|
expect(report.kernels[0]).toMatchObject(expected);
|
|
expect(report.coverage.payloads).toBe("complete");
|
|
const declaration = declarations.find(
|
|
([caseName]) => caseName === name,
|
|
)![1];
|
|
expect(report.kernels[0]!.location.size).toBe(
|
|
Buffer.byteLength(declaration.slice(declaration.indexOf(".entry"))),
|
|
);
|
|
}
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"accepts the PTX compute_xx .target synonym verbatim",
|
|
async ({ cli }) => {
|
|
const root = await createTestTempDirectory("rea-ptx-target-cli-");
|
|
const text = `.version 8.8\n.target compute_80\n.address_size 64\n.entry kernel() {\nret;\n}\n`;
|
|
const path = join(root, "target.ptx");
|
|
await writeFile(path, text);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"kernels",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const evidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
response.json,
|
|
),
|
|
);
|
|
const report = gpuCodeObjectsSchema.parse(evidence.normalized_result);
|
|
expect(report.objects[0]!.target).toBe("compute_80");
|
|
expect(report.ptx_artifacts[0]!.sha256).toBe(
|
|
createHash("sha256").update(text).digest("hex"),
|
|
);
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"extracts entry parameters declared in PTX 1.3 bodies",
|
|
async ({ cli }) => {
|
|
const root = await createTestTempDirectory("rea-ptx-legacy-cli-");
|
|
const text = `.version 1.3\n.target sm_80\n.address_size 64\n.entry kernel {\n.param .u32 input;\nret;\n}\n`;
|
|
const path = join(root, "legacy.ptx");
|
|
await writeFile(path, text);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"kernels",
|
|
"--detail",
|
|
"complete",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const evidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
response.json,
|
|
),
|
|
);
|
|
const report = gpuCodeObjectsSchema.parse(evidence.normalized_result);
|
|
expect(report.kernels).toEqual([
|
|
expect.objectContaining({
|
|
name: "kernel",
|
|
parameter_count: 1,
|
|
parameter_bytes: 4,
|
|
parameters: [
|
|
expect.objectContaining({
|
|
ordinal: 0,
|
|
name: "input",
|
|
offset: 0,
|
|
size: 4,
|
|
type: ".u32",
|
|
}),
|
|
],
|
|
}),
|
|
]);
|
|
const malformedPath = join(root, "truncated.ptx");
|
|
await writeFile(
|
|
malformedPath,
|
|
`.version 1.3\n.target sm_80\n.entry kernel {\n.param .u32 input\n`,
|
|
);
|
|
const malformed = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
malformedPath,
|
|
"--analysis",
|
|
"kernels",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(malformed.exitCode, malformed.stderr).toBe(0);
|
|
const malformedEvidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
malformed.json,
|
|
),
|
|
);
|
|
const malformedReport = gpuCodeObjectsSchema.parse(
|
|
malformedEvidence.normalized_result,
|
|
);
|
|
expect(malformedReport.objects[0]!.inspection).toBe("malformed");
|
|
expect(malformedReport.coverage.payloads).toBe("partial");
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"attributes malformed selector JSON to its actual CLI option",
|
|
async ({ cli }) => {
|
|
for (const name of ["object_ids", "targets", "view"]) {
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
"/not-opened-for-invalid-selectors",
|
|
`--${name}`,
|
|
"[",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode).not.toBe(0);
|
|
expect(response.json).toMatchObject({
|
|
code: "invalid_request",
|
|
details: {
|
|
operation: "inspect_gpu_code_objects",
|
|
issues: [
|
|
{
|
|
path: [name],
|
|
reason: "invalid_format",
|
|
message: `${name} must be valid JSON`,
|
|
},
|
|
],
|
|
},
|
|
});
|
|
}
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"publishes and enforces the lowercase --expected_sha256 digest",
|
|
async ({ cli }) => {
|
|
const schema = await cli.run({
|
|
arguments: ["inspect-gpu-code-objects", "--schema", "--json"],
|
|
});
|
|
expect(schema.exitCode, schema.stderr).toBe(0);
|
|
expect(schema.json).toMatchObject({
|
|
options: {
|
|
properties: { expected_sha256: { pattern: "^[a-f0-9]{64}$" } },
|
|
},
|
|
});
|
|
// PowerShell's Get-FileHash prints the digest in upper case.
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
"/not-opened-for-invalid-selectors",
|
|
"--expected_sha256",
|
|
"AB".repeat(32),
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode).toBe(1);
|
|
expect(response.json).toMatchObject({
|
|
code: "VALIDATION_ERROR",
|
|
fieldErrors: [expect.objectContaining({ path: "expected_sha256" })],
|
|
});
|
|
},
|
|
);
|
|
|
|
it("preserves incomplete decoder coverage through MCP target and object pages", async () => {
|
|
const root = await createTestTempDirectory("rea-gpu-partial-"),
|
|
{ call } = await connectLocalToolsMcp();
|
|
const cases = {
|
|
"section-names": () => {
|
|
const bytes = cubinFixture();
|
|
bytes.writeUInt16LE(0, 62);
|
|
return bytes;
|
|
},
|
|
symbols: () => {
|
|
const bytes = cubinFixture();
|
|
bytes.writeUInt32LE(1, Number(bytes.readBigUInt64LE(40)) + 3 * 64 + 4);
|
|
return bytes;
|
|
},
|
|
"later-amd-note": () => amdFixture(false, Buffer.alloc(3)),
|
|
"modern-amd-target-conflict": () =>
|
|
hipFixture(
|
|
amdFixture(false, undefined, "amdgpu9.00-amd-amdhsa--gfx900"),
|
|
"hipv4-amdgpu11.00-amd-amdhsa--gfx1100",
|
|
),
|
|
"target-conflict": () =>
|
|
cudaFixture("none", [
|
|
{
|
|
kind: 2,
|
|
target: 80,
|
|
bytes: cubinFixture({ architecture: 90 }),
|
|
},
|
|
]),
|
|
"register-conflict": () => cubinFixture({ textRegisters: 11 }),
|
|
"ptx-parameter": () =>
|
|
Buffer.from(ptxFixture.replace(".u64 output", ".u64 .vendor output")),
|
|
"ptx-shared": () =>
|
|
Buffer.from(ptxFixture.replace("scratch[32]", "scratch[]")),
|
|
};
|
|
for (const [missing, fixture] of Object.entries(cases)) {
|
|
const bytes = fixture(),
|
|
path = join(root, missing);
|
|
await writeFile(path, bytes);
|
|
const reply = await call("inspect_gpu_code_objects", {
|
|
path,
|
|
analysis: "kernels",
|
|
view: { collection: "targets" },
|
|
});
|
|
expect(reply.isError, JSON.stringify(reply)).not.toBe(true);
|
|
const targets = gpuCodeObjectsPageViewResultSchema.parse(
|
|
parseEvidence(reply.structuredContent).normalized_result,
|
|
);
|
|
expect(targets.source_coverage.payloads).toBe("partial");
|
|
expect(targets.items).toEqual([
|
|
expect.objectContaining({
|
|
target:
|
|
missing === "later-amd-note"
|
|
? "amdgcn-amd-amdhsa--gfx900"
|
|
: missing === "modern-amd-target-conflict"
|
|
? "gfx1100"
|
|
: "sm_80",
|
|
kernel_count: missing === "symbols" ? 0 : 1,
|
|
kernel_coverage: "partial",
|
|
}),
|
|
]);
|
|
const objectReply = await call("inspect_analysis_view", {
|
|
source: {
|
|
kind: "retained-evidence",
|
|
evidence_id: targets.parent_evidence_id,
|
|
},
|
|
view: {
|
|
kind: "gpu-code-objects",
|
|
collection: "objects",
|
|
fields: ["kind"],
|
|
},
|
|
});
|
|
expect(objectReply.isError, JSON.stringify(objectReply)).not.toBe(true);
|
|
const objects = gpuCodeObjectsPageViewResultSchema.parse(
|
|
parseEvidence(objectReply.structuredContent).normalized_result,
|
|
);
|
|
expect(objects.items).toEqual([
|
|
...(missing === "modern-amd-target-conflict"
|
|
? [expect.objectContaining({ kind: "host", inspection: "complete" })]
|
|
: []),
|
|
expect.objectContaining({
|
|
inspection: missing === "later-amd-note" ? "malformed" : "partial",
|
|
}),
|
|
]);
|
|
if (missing === "target-conflict")
|
|
expect(objects.items[0]).toMatchObject({
|
|
target: "sm_80",
|
|
payload_target: "sm_90",
|
|
target_consistency: "conflict",
|
|
});
|
|
if (missing === "modern-amd-target-conflict")
|
|
expect(objects.items[1]).toMatchObject({
|
|
kind: "amdgpu",
|
|
target: "gfx1100",
|
|
payload_target: "amdgpu9.00-amd-amdhsa--gfx900",
|
|
target_consistency: "conflict",
|
|
});
|
|
}
|
|
});
|
|
|
|
it("keeps unknown target discovery visible in the default MCP summary", async () => {
|
|
const root = await createTestTempDirectory("rea-gpu-unknown-target-"),
|
|
path = join(root, "cuda");
|
|
await writeFile(path, cubinFixture({ osAbi: 0x41, flags: 0x06016402 }));
|
|
const { call } = await connectLocalToolsMcp();
|
|
const reply = await call("inspect_gpu_code_objects", { path });
|
|
expect(reply.isError, JSON.stringify(reply)).not.toBe(true);
|
|
const result = gpuCodeObjectsSummaryViewResultSchema.parse(
|
|
parseEvidence(reply.structuredContent).normalized_result,
|
|
);
|
|
expect(result.summary.coverage.discovery).toBe("partial");
|
|
expect(result.summary.counts).toMatchObject({ objects: 1, targets: 0 });
|
|
const detailedReply = await call("inspect_gpu_code_objects", {
|
|
path,
|
|
analysis: "kernels",
|
|
view: { collection: "objects" },
|
|
});
|
|
expect(detailedReply.isError, JSON.stringify(detailedReply)).not.toBe(true);
|
|
const detailed = gpuCodeObjectsPageViewResultSchema.parse(
|
|
parseEvidence(detailedReply.structuredContent).normalized_result,
|
|
);
|
|
expect(detailed.items).toHaveLength(1);
|
|
const source = {
|
|
kind: "retained-evidence",
|
|
evidence_id: detailed.parent_evidence_id,
|
|
} as const;
|
|
for (const collection of ["objects", "targets", "kernels"] as const) {
|
|
const filteredReply = await call("inspect_analysis_view", {
|
|
source,
|
|
view: {
|
|
kind: "gpu-code-objects",
|
|
collection,
|
|
targets: [""],
|
|
},
|
|
});
|
|
expect(filteredReply.isError, JSON.stringify(filteredReply)).not.toBe(true);
|
|
const filtered = gpuCodeObjectsPageViewResultSchema.parse(
|
|
parseEvidence(filteredReply.structuredContent).normalized_result,
|
|
);
|
|
expect(filtered.items).toEqual([]);
|
|
}
|
|
});
|
|
|
|
it("preserves omitted AMD metadata in selected MCP object pages", async () => {
|
|
const root = await createTestTempDirectory("rea-gpu-metadata-columns-"),
|
|
path = join(root, "amd");
|
|
await writeFile(path, amdFixture());
|
|
const { call } = await connectLocalToolsMcp();
|
|
const inspected = await call("inspect_gpu_code_objects", {
|
|
path,
|
|
analysis: "kernels",
|
|
view: { collection: "objects", fields: ["amd_target_metadata"] },
|
|
});
|
|
expect(inspected.isError, JSON.stringify(inspected)).not.toBe(true);
|
|
const full = gpuCodeObjectsPageViewResultSchema.parse(
|
|
parseEvidence(inspected.structuredContent).normalized_result,
|
|
);
|
|
expect(full.items[0]).toHaveProperty(
|
|
"amd_target_metadata.processor",
|
|
"gfx900",
|
|
);
|
|
const selected = await call("inspect_analysis_view", {
|
|
source: { kind: "retained-evidence", evidence_id: full.parent_evidence_id },
|
|
view: { kind: "gpu-code-objects", collection: "objects", fields: ["kind"] },
|
|
});
|
|
expect(selected.isError, JSON.stringify(selected)).not.toBe(true);
|
|
const page = gpuCodeObjectsPageViewResultSchema.parse(
|
|
parseEvidence(selected.structuredContent).normalized_result,
|
|
);
|
|
expect(page.items).toHaveLength(1);
|
|
expect(page.items[0]).not.toHaveProperty("amd_target_metadata");
|
|
});
|
|
|
|
it("retains selected kernel facts and complete PTX for historical MCP pages after source removal", async () => {
|
|
const root = await createTestTempDirectory("rea-gpu-mcp-"),
|
|
path = join(root, "application");
|
|
await writeFile(path, cudaFixture("zstd"));
|
|
const { call } = await connectLocalToolsMcp();
|
|
const inventoryReply = await call("inspect_gpu_code_objects", {
|
|
path,
|
|
view: { collection: "objects", fields: ["kind", "target"] },
|
|
});
|
|
expect(inventoryReply.isError, JSON.stringify(inventoryReply)).not.toBe(true);
|
|
const inventory = gpuCodeObjectsPageViewResultSchema.parse(
|
|
parseEvidence(inventoryReply.structuredContent).normalized_result,
|
|
);
|
|
expect(inventory.items).toHaveLength(2);
|
|
expect(inventory.items[0]).not.toHaveProperty("sha256");
|
|
expect(inventory.items[0]).toHaveProperty("metadata_location");
|
|
const targetsReply = await call("inspect_analysis_view", {
|
|
source: {
|
|
kind: "retained-evidence",
|
|
evidence_id: inventory.parent_evidence_id,
|
|
},
|
|
view: { kind: "gpu-code-objects", collection: "targets" },
|
|
});
|
|
expect(targetsReply.isError, JSON.stringify(targetsReply)).not.toBe(true);
|
|
const targets = gpuCodeObjectsPageViewResultSchema.parse(
|
|
parseEvidence(targetsReply.structuredContent).normalized_result,
|
|
);
|
|
expect(targets.items).toHaveLength(2);
|
|
expect(targets.items).toEqual(
|
|
expect.arrayContaining([
|
|
expect.objectContaining({
|
|
target: "sm_80",
|
|
object_count: 1,
|
|
kernel_count: 0,
|
|
kernel_coverage: "not-requested",
|
|
}),
|
|
expect.objectContaining({
|
|
target: "sm_90",
|
|
object_count: 1,
|
|
kernel_count: 0,
|
|
kernel_coverage: "not-requested",
|
|
}),
|
|
]),
|
|
);
|
|
const inspectedReply = await call("inspect_gpu_code_objects", {
|
|
path,
|
|
analysis: "kernels",
|
|
view: {
|
|
collection: "kernels",
|
|
names: ["add_one"],
|
|
fields: ["parameter_count", "registers"],
|
|
},
|
|
});
|
|
expect(inspectedReply.isError, JSON.stringify(inspectedReply)).not.toBe(true);
|
|
const inspected = gpuCodeObjectsPageViewResultSchema.parse(
|
|
parseEvidence(inspectedReply.structuredContent).normalized_result,
|
|
);
|
|
expect(inspected.items).toHaveLength(2);
|
|
expect(inspected.items[0]).not.toHaveProperty("parameters");
|
|
expect(inspected.items[0]).toHaveProperty("sources");
|
|
const objectId =
|
|
"object_id" in inspected.items[0]!
|
|
? inspected.items[0]!.object_id
|
|
: undefined;
|
|
expect(objectId).toBeTypeOf("string");
|
|
await unlink(path);
|
|
const source = {
|
|
kind: "retained-evidence",
|
|
evidence_id: inspected.parent_evidence_id,
|
|
};
|
|
const textReply = await call("inspect_analysis_view", {
|
|
source,
|
|
view: {
|
|
kind: "gpu-code-objects",
|
|
collection: "ptx",
|
|
object_id: objectId,
|
|
offset: 0,
|
|
limit: 4096,
|
|
},
|
|
});
|
|
expect(textReply.isError, JSON.stringify(textReply)).not.toBe(true);
|
|
const text = gpuCodeObjectsPageViewResultSchema.parse(
|
|
parseEvidence(textReply.structuredContent).normalized_result,
|
|
);
|
|
expect(text.ptx!.text).toBe(ptxFixture);
|
|
expect(text.ptx!.sha256).toBe(
|
|
createHash("sha256").update(ptxFixture).digest("hex"),
|
|
);
|
|
const summaryReply = await call("inspect_analysis_view", {
|
|
source,
|
|
view: { kind: "summary" },
|
|
});
|
|
expect(summaryReply.isError, JSON.stringify(summaryReply)).not.toBe(true);
|
|
const summary = gpuCodeObjectsSummaryViewResultSchema.parse(
|
|
parseEvidence(summaryReply.structuredContent).normalized_result,
|
|
);
|
|
expect(summary.summary.counts).toMatchObject({
|
|
objects: 2,
|
|
kernels: 2,
|
|
ptx_artifacts: 1,
|
|
});
|
|
expect(summary.summary).not.toHaveProperty("ptx_artifacts");
|
|
});
|
|
|
|
it("rejects forged duplicate object IDs and mismatched Evidence subjects through MCP", async () => {
|
|
const root = await createTestTempDirectory("rea-gpu-import-"),
|
|
path = join(root, "cuda");
|
|
await writeFile(path, cudaFixture());
|
|
const { call } = await connectLocalToolsMcp();
|
|
const reply = await call("inspect_gpu_code_objects", {
|
|
path,
|
|
analysis: "kernels",
|
|
detail: "complete",
|
|
});
|
|
expect(reply.isError, JSON.stringify(reply)).not.toBe(true);
|
|
const parent = parseEvidence(reply.structuredContent),
|
|
report = gpuCodeObjectsSchema.parse(parent.normalized_result);
|
|
const forgedHeaderReport = gpuCodeObjectsSchema.parse(
|
|
parent.normalized_result,
|
|
);
|
|
forgedHeaderReport.objects[1]!.cuda_target_header!.location.container_id =
|
|
forgedHeaderReport.objects[0]!.id;
|
|
const forgedHeader = createEvidence(undefined, parent.provider, {
|
|
operation: parent.operation,
|
|
parameters: parent.parameters,
|
|
result: forgedHeaderReport,
|
|
});
|
|
expect(
|
|
parseMcpToolError(
|
|
await call("inspect_analysis_view", {
|
|
source: { kind: "inline", evidence: forgedHeader },
|
|
view: { kind: "gpu-code-objects", collection: "objects" },
|
|
}),
|
|
).error,
|
|
).toMatchObject({ code: "invalid_request" });
|
|
report.objects[1]!.id = report.objects[0]!.id;
|
|
const forged = createEvidence(undefined, parent.provider, {
|
|
operation: "inspect_gpu_code_objects",
|
|
parameters: parent.parameters,
|
|
result: report,
|
|
});
|
|
const failure = parseMcpToolError(
|
|
await call("inspect_analysis_view", {
|
|
source: { kind: "inline", evidence: forged },
|
|
view: { kind: "gpu-code-objects", collection: "objects" },
|
|
}),
|
|
);
|
|
expect(failure.error).toMatchObject({ code: "invalid_request" });
|
|
const mismatched = createEvidence(
|
|
{ path: join(root, "other-file"), sha256: "f".repeat(64), format: "file" },
|
|
parent.provider,
|
|
{
|
|
operation: parent.operation,
|
|
parameters: parent.parameters,
|
|
result: parent.normalized_result,
|
|
},
|
|
);
|
|
const identityFailure = parseMcpToolError(
|
|
await call("inspect_analysis_view", {
|
|
source: { kind: "inline", evidence: mismatched },
|
|
view: { kind: "summary" },
|
|
}),
|
|
);
|
|
expect(identityFailure.error).toMatchObject({ code: "invalid_request" });
|
|
});
|
|
|
|
cliTest(
|
|
"uses DOS relocation ownership before GPU section discovery through the CLI",
|
|
async ({ cli }) => {
|
|
const root = await createTestTempDirectory("rea-gpu-dos-relocations-cli-");
|
|
const path = join(root, "selected.exe");
|
|
const bytes = peGpuFixture();
|
|
bytes.writeUInt16LE(bytes.length % 512, 2);
|
|
bytes.writeUInt16LE(Math.ceil(bytes.length / 512), 4);
|
|
bytes.writeUInt16LE(2, 8);
|
|
await writeFile(path, bytes);
|
|
const args = [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"inventory",
|
|
"--json",
|
|
];
|
|
const valid = await cli.run({ arguments: args });
|
|
expect(valid.exitCode, valid.stderr).toBe(0);
|
|
bytes.writeUInt16LE(1, 6);
|
|
bytes.writeUInt16LE(4, 8);
|
|
bytes.writeUInt16LE(60, 24);
|
|
await writeFile(path, bytes);
|
|
const refused = await cli.run({ arguments: args });
|
|
expect(refused.exitCode).toBe(1);
|
|
expect(refused.json).toMatchObject({
|
|
code: "target_unavailable",
|
|
details: {
|
|
path,
|
|
constraint: "file_format_rejected",
|
|
reason: "MZ header does not declare a PE image",
|
|
},
|
|
});
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"retains AMD target declarations and examined note coverage through CLI and MCP",
|
|
async ({ cli }) => {
|
|
const root = await createTestTempDirectory("rea-amd-target-sources-");
|
|
const { call } = await connectLocalToolsMcp();
|
|
const first = "amdgcn-amd-amdhsa--gfx900";
|
|
const second = "amdgcn-amd-amdhsa--gfx1100";
|
|
const legacyNote = Buffer.alloc(20);
|
|
legacyNote.writeUInt32LE(4, 0);
|
|
legacyNote.writeUInt32LE(4, 4);
|
|
legacyNote.writeUInt32LE(10, 8);
|
|
legacyNote.write("AMD\0", 12);
|
|
const headerConflict = amdFixture();
|
|
headerConflict.writeUInt32LE(0x141, 48);
|
|
for (const [name, bytes, declarations, coverage, effective] of [
|
|
[
|
|
"note-conflict",
|
|
amdFixture(false, amdNoteFixture(false, second)),
|
|
[first, second],
|
|
"complete",
|
|
null,
|
|
],
|
|
["header-conflict", headerConflict, [first], "complete", null],
|
|
[
|
|
"duplicate",
|
|
amdFixture(false, amdNoteFixture()),
|
|
[first, first],
|
|
"complete",
|
|
first,
|
|
],
|
|
["legacy-tail", amdFixture(false, legacyNote), [first], "partial", first],
|
|
[
|
|
"invalid-target",
|
|
amdFixture(false, undefined, 123),
|
|
[null],
|
|
"partial",
|
|
null,
|
|
],
|
|
[
|
|
"absent-target",
|
|
amdFixture(false, undefined, null),
|
|
[null],
|
|
"complete",
|
|
"gfx900",
|
|
],
|
|
[
|
|
"truncated-tail",
|
|
amdFixture(false, Buffer.alloc(3)),
|
|
[first],
|
|
"partial",
|
|
first,
|
|
],
|
|
] as const) {
|
|
const path = join(root, name);
|
|
await writeFile(path, bytes);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"kernels",
|
|
"--detail",
|
|
"complete",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const reply = await call("inspect_gpu_code_objects", {
|
|
path,
|
|
analysis: "kernels",
|
|
detail: "complete",
|
|
});
|
|
expect(reply.isError, JSON.stringify(reply)).not.toBe(true);
|
|
for (const value of [response.json, reply.structuredContent]) {
|
|
const report = gpuCodeObjectsSchema.parse(
|
|
parseEvidence(value).normalized_result,
|
|
);
|
|
const row = report.objects[0]!;
|
|
expect(row).toMatchObject({
|
|
target: effective,
|
|
payload_target: effective,
|
|
});
|
|
const metadata = row.amd_target_metadata!;
|
|
expect(metadata).toMatchObject({
|
|
flags: name === "header-conflict" ? "0x00000141" : "0x0000012c",
|
|
processor: name === "header-conflict" ? "gfx1100" : "gfx900",
|
|
header_location: { container_id: row.id, offset: 0, size: 64 },
|
|
note_coverage: coverage,
|
|
});
|
|
expect(metadata.notes.map(({ target }) => target)).toEqual(
|
|
declarations,
|
|
);
|
|
expect(
|
|
new Set(metadata.notes.map(({ location }) => location.offset)).size,
|
|
).toBe(declarations.length);
|
|
for (const note of metadata.notes) {
|
|
expect(note.location.container_id).toBe(row.id);
|
|
expect(note.location.section_name).toBe(".note");
|
|
const source = bytes.subarray(
|
|
note.location.offset,
|
|
note.location.offset + note.location.size,
|
|
);
|
|
if (note.target !== null)
|
|
expect(source.includes(Buffer.from(note.target))).toBe(true);
|
|
expect(note.target_state).toBe(
|
|
name === "invalid-target"
|
|
? "invalid"
|
|
: name === "absent-target"
|
|
? "absent"
|
|
: "declared",
|
|
);
|
|
}
|
|
expect(report.kernels).toHaveLength(declarations.length);
|
|
if (coverage === "partial")
|
|
expect(report.coverage.payloads).toBe("partial");
|
|
}
|
|
}
|
|
},
|
|
);
|
|
|
|
it("does not interpret non-AMD ELF flags as AMD processor declarations through MCP", async () => {
|
|
const root = await createTestTempDirectory("rea-amd-machine-");
|
|
const path = join(root, "wrong-machine");
|
|
const payload = amdFixture();
|
|
payload.writeUInt16LE(62, 18);
|
|
await writeFile(path, hipFixture(payload, "hip-amdgcn-amd-amdhsa--"));
|
|
const { call } = await connectLocalToolsMcp();
|
|
const reply = await call("inspect_gpu_code_objects", {
|
|
path,
|
|
analysis: "kernels",
|
|
detail: "complete",
|
|
});
|
|
expect(reply.isError, JSON.stringify(reply)).not.toBe(true);
|
|
const report = gpuCodeObjectsSchema.parse(
|
|
parseEvidence(reply.structuredContent).normalized_result,
|
|
);
|
|
expect(report.objects.find(({ kind }) => kind === "amdgpu")).toMatchObject({
|
|
target: null,
|
|
payload_target: null,
|
|
amd_target_metadata: null,
|
|
target_consistency: "unknown",
|
|
inspection: "malformed",
|
|
});
|
|
expect(report.kernels).toEqual([]);
|
|
expect(report.coverage.payloads).toBe("partial");
|
|
});
|