@memrov2/pentcloud-reference 0.1.0 → 0.1.1-dev.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@memrov2/pentcloud-reference",
-  "version": "0.1.0",
+  "version": "0.1.1-dev.0",
   "type": "module",
   "license": "MIT",
   "description": "Cloudflare-native reference tool pack",
@@ -33,6 +33,6 @@
     "access": "public"
   },
   "dependencies": {
-    "@memrov2/pentcloud-tool-runtime": "0.1.0"
+    "@memrov2/pentcloud-tool-runtime": "0.1.1-dev.0"
   }
 }

package/src/index.d.mts

@@ -12,3 +12,5 @@ export declare function fetchReference(
   sha256: string;
   bytes: Uint8Array;
 }>;
+
+export declare function referencePath(path: string): string;

package/src/index.mjs

@@ -1,6 +1,6 @@
 import lock from "../tools.lock.json" with { type: "json" };
 export { lock };
-export async function fetchReference(path, fetcher = fetch) {
+export function referencePath(path) {
   if (typeof path === "string" && path.startsWith("/opt/exploitdb/"))
     path = path.slice("/opt/exploitdb/".length);
   if (
@@ -10,6 +10,10 @@ export async function fetchReference(path, fetcher = fetch) {
     path.split("/").some((p) => !p || p === "." || p === "..")
   )
     throw new TypeError("Select a pinned exploit reference path");
+  return path;
+}
+export async function fetchReference(path, fetcher = fetch) {
+  path = referencePath(path);
   const url =
     "https://gitlab.com/exploit-database/exploitdb/-/raw/" +
     lock.dataRevision +

package/src/worker.d.mts

@@ -2,7 +2,9 @@ import { DurableObject } from "cloudflare:workers";
 import type { ToolResult } from "@memrov2/pentcloud-tool-runtime";
 import type { PackRequest } from "./index.mjs";
 export declare class ReferenceRunner extends DurableObject {
+  ready(): Promise<{ image: string; instance: "lite"; readyMs: number }>;
   run(input: PackRequest): Promise<ToolResult>;
+  retire(): Promise<{ stopped: boolean; retired: true }>;
   stop(): Promise<{ stopped: boolean }>;
   readSource(
     path: string,

package/src/worker.mjs

@@ -2,7 +2,8 @@ import { createRunner } from "@memrov2/pentcloud-tool-runtime/worker";
 import { lock, buildCommand, fetchReference } from "./index.mjs";
 const Runtime = createRunner(lock, buildCommand);
 export class ReferenceRunner extends Runtime {
-  readSource(path) {
+  async readSource(path) {
+    if(await this.ctx.storage.get("retired"))throw new Error("Owned tool runtime is retired");
     return fetchReference(path);
   }
 }