@j3r3mcdev/oast-server 1.0.0 → 1.1.0

package/src/listeners/http/http.listener.ts

@@ -1,110 +1,110 @@
-import http, { IncomingMessage, ServerResponse } from "http";
-import { Logger } from "../../core/logger";
-import { CoreRouter } from "../../core/router";
-import { EventNormalizer } from "../../core/event.normalizer";
-
-export interface HttpListenerOptions {
-  port: number;
-  logger?: Logger;
-}
-
-export class HttpListener {
-  private server?: http.Server;
-  private logger: Logger;
-
-  constructor(
-    private router: CoreRouter,
-    private options: HttpListenerOptions,
-  ) {
-    this.logger = options.logger ?? new Logger({ context: "HttpListener" });
-  }
-
-  async start(): Promise<void> {
-    return new Promise((resolve) => {
-      this.server = http.createServer(async (req, res) => {
-        await this.handleRequest(req, res);
-      });
-
-      this.server.listen(this.options.port, () => {
-        this.logger.info(`HTTP Listener started on port ${this.options.port}`);
-        resolve();
-      });
-    });
-  }
-
-  async stop(): Promise<void> {
-    return new Promise((resolve, reject) => {
-      if (!this.server) return resolve();
-
-      this.server.close((err) => {
-        if (err) return reject(err);
-        this.logger.info("HTTP Listener stopped");
-        resolve();
-      });
-    });
-  }
-
-  private async handleRequest(req: IncomingMessage, res: ServerResponse) {
-    try {
-      const body = await this.readBody(req);
-
-      const ipHeader = req.headers["x-forwarded-for"];
-      const ip =
-        typeof ipHeader === "string"
-          ? ipHeader
-          : Array.isArray(ipHeader)
-            ? ipHeader[0]
-            : (req.socket.remoteAddress ?? "");
-
-      const rawEvent = {
-        ip,
-        method: req.method ?? "",
-        path: req.url ?? "",
-        headers: req.headers,
-        query: {}, // parsing optionnel
-        body,
-        raw: req,
-      };
-
-      const normalized = EventNormalizer.normalizeHttp(rawEvent);
-
-      await this.router.dispatch(normalized);
-
-      res.writeHead(200, { "Content-Type": "application/json" });
-      res.end(
-        JSON.stringify({
-          success: true,
-          eventId: normalized.id,
-        }),
-      );
-    } catch (err: any) {
-      this.logger.error("HTTP Listener error", err);
-
-      res.writeHead(500, { "Content-Type": "application/json" });
-      res.end(
-        JSON.stringify({
-          success: false,
-          error: err.message,
-        }),
-      );
-    }
-  }
-
-  private readBody(req: IncomingMessage): Promise<any> {
-    return new Promise((resolve) => {
-      let data = "";
-
-      req.on("data", (chunk) => {
-        data += chunk;
-      });
-
-      req.on("end", () => {
-        try {
-          resolve(JSON.parse(data || "{}"));
-        } catch {
-          resolve(data);
-        }
-      });
-    });
-  }
-}
+import http, { IncomingMessage, ServerResponse } from "http";
+import { Logger } from "../../core/logger";
+import { CoreRouter } from "../../core/router";
+import { EventNormalizer } from "../../core/event.normalizer";
+
+export interface HttpListenerOptions {
+  port: number;
+  logger?: Logger;
+}
+
+export class HttpListener {
+  private server?: http.Server;
+  private logger: Logger;
+
+  constructor(
+    private router: CoreRouter,
+    private options: HttpListenerOptions,
+  ) {
+    this.logger = options.logger ?? new Logger({ context: "HttpListener" });
+  }
+
+  async start(): Promise<void> {
+    return new Promise((resolve) => {
+      this.server = http.createServer(async (req, res) => {
+        await this.handleRequest(req, res);
+      });
+
+      this.server.listen(this.options.port, () => {
+        this.logger.info(`HTTP Listener started on port ${this.options.port}`);
+        resolve();
+      });
+    });
+  }
+
+  async stop(): Promise<void> {
+    return new Promise((resolve, reject) => {
+      if (!this.server) return resolve();
+
+      this.server.close((err) => {
+        if (err) return reject(err);
+        this.logger.info("HTTP Listener stopped");
+        resolve();
+      });
+    });
+  }
+
+  private async handleRequest(req: IncomingMessage, res: ServerResponse) {
+    try {
+      const body = await this.readBody(req);
+
+      const ipHeader = req.headers["x-forwarded-for"];
+      const ip =
+        typeof ipHeader === "string"
+          ? ipHeader
+          : Array.isArray(ipHeader)
+            ? ipHeader[0]
+            : (req.socket.remoteAddress ?? "");
+
+      const rawEvent = {
+        ip,
+        method: req.method ?? "",
+        path: req.url ?? "",
+        headers: req.headers,
+        query: {}, // parsing optionnel
+        body,
+        raw: req,
+      };
+
+      const normalized = EventNormalizer.normalizeHttp(rawEvent);
+
+      await this.router.dispatch(normalized);
+
+      res.writeHead(200, { "Content-Type": "application/json" });
+      res.end(
+        JSON.stringify({
+          success: true,
+          eventId: normalized.id,
+        }),
+      );
+    } catch (err: any) {
+      this.logger.error("HTTP Listener error", err);
+
+      res.writeHead(500, { "Content-Type": "application/json" });
+      res.end(
+        JSON.stringify({
+          success: false,
+          error: err.message,
+        }),
+      );
+    }
+  }
+
+  private readBody(req: IncomingMessage): Promise<any> {
+    return new Promise((resolve) => {
+      let data = "";
+
+      req.on("data", (chunk) => {
+        data += chunk;
+      });
+
+      req.on("end", () => {
+        try {
+          resolve(JSON.parse(data || "{}"));
+        } catch {
+          resolve(data);
+        }
+      });
+    });
+  }
+}

package/src/listeners/listener.interface.ts

@@ -1,4 +1,4 @@
-export interface Listener {
-  start(): Promise<void>;
-  stop(): Promise<void>;
-}
+export interface Listener {
+  start(): Promise<void>;
+  stop(): Promise<void>;
+}

package/src/listeners/smtp/__tests__/smtp.extractor.test.ts

@@ -1,69 +1,69 @@
-import { describe, it, expect } from "@jest/globals";
-import { SmtpExtractor } from "../smtp.extractor";
-
-describe("SmtpExtractor", () => {
-  it("extrait correctement un RawSmtpEvent valide", () => {
-    const session = {
-      remoteAddress: "1.2.3.4",
-      envelope: {
-        mailFrom: { address: "attacker@example.com" },
-        rcptTo: [{ address: "victim@example.com" }],
-      },
-      headers: { subject: "Hello World" },
-    };
-
-    const body = "This is the email body";
-
-    const result = SmtpExtractor.extract(session, body);
-
-    expect(result).toEqual({
-      ip: "1.2.3.4",
-      from: "attacker@example.com",
-      to: "victim@example.com",
-      subject: "Hello World",
-      raw: {
-        session,
-        body,
-      },
-    });
-  });
-
-  it("gère les champs manquants proprement", () => {
-    const session = {
-      remoteAddress: undefined,
-      envelope: {
-        mailFrom: {},
-        rcptTo: [{}],
-      },
-      headers: {},
-    };
-
-    const result = SmtpExtractor.extract(session, "");
-
-    expect(result).toEqual({
-      ip: "",
-      from: "",
-      to: "",
-      subject: "",
-      raw: {
-        session,
-        body: "",
-      },
-    });
-  });
-
-  it("ne plante pas si session est vide", () => {
-    const result = SmtpExtractor.extract({}, "BODY");
-
-    expect(result).toEqual({
-      ip: "",
-      from: "",
-      to: "",
-      subject: "",
-      raw: {
-        session: {},
-        body: "BODY",
-      },
-    });
-  });
-});
+import { describe, it, expect } from "@jest/globals";
+import { SmtpExtractor } from "../smtp.extractor";
+
+describe("SmtpExtractor", () => {
+  it("extrait correctement un RawSmtpEvent valide", () => {
+    const session = {
+      remoteAddress: "1.2.3.4",
+      envelope: {
+        mailFrom: { address: "attacker@example.com" },
+        rcptTo: [{ address: "victim@example.com" }],
+      },
+      headers: { subject: "Hello World" },
+    };
+
+    const body = "This is the email body";
+
+    const result = SmtpExtractor.extract(session, body);
+
+    expect(result).toEqual({
+      ip: "1.2.3.4",
+      from: "attacker@example.com",
+      to: "victim@example.com",
+      subject: "Hello World",
+      raw: {
+        session,
+        body,
+      },
+    });
+  });
+
+  it("gère les champs manquants proprement", () => {
+    const session = {
+      remoteAddress: undefined,
+      envelope: {
+        mailFrom: {},
+        rcptTo: [{}],
+      },
+      headers: {},
+    };
+
+    const result = SmtpExtractor.extract(session, "");
+
+    expect(result).toEqual({
+      ip: "",
+      from: "",
+      to: "",
+      subject: "",
+      raw: {
+        session,
+        body: "",
+      },
+    });
+  });
+
+  it("ne plante pas si session est vide", () => {
+    const result = SmtpExtractor.extract({}, "BODY");
+
+    expect(result).toEqual({
+      ip: "",
+      from: "",
+      to: "",
+      subject: "",
+      raw: {
+        session: {},
+        body: "BODY",
+      },
+    });
+  });
+});

package/src/listeners/smtp/__tests__/smtp.listener.test.ts

@@ -1,150 +1,150 @@
-import { describe, it, expect, jest, beforeEach } from "@jest/globals";
-import { SmtpListener } from "../smtp.listener";
-import { EventNormalizer } from "../../../core/event.normalizer";
-import { CoreRouter } from "../../../core/router";
-
-// Mock EventNormalizer
-jest.mock("../../../core/event.normalizer", () => ({
-  EventNormalizer: {
-    normalizeSmtp: jest.fn(),
-  },
-}));
-
-describe("SmtpListener", () => {
-  let router: CoreRouter;
-  let server: any;
-  let listener: SmtpListener;
-
-  beforeEach(() => {
-    router = {
-      dispatch: jest.fn(async (_event: any) => {}),
-    } as any;
-
-    server = {
-      onData: null,
-      close: jest.fn(),
-    };
-
-    listener = new SmtpListener(router, { server });
-  });
-
-  it("traite un email SMTP valide et appelle dispatch", async () => {
-    const callback = jest.fn();
-
-    // NormalizedSmtpEvent VALIDE
-    (EventNormalizer.normalizeSmtp as jest.Mock).mockReturnValue({
-      id: "smtp-123",
-      type: "smtp",
-      timestamp: Date.now(),
-      ip: "1.2.3.4",
-      from: "attacker@example.com",
-      to: ["victim@example.com"],
-      subject: "Test Email",
-      body: "Hello world",
-      raw: {},
-    });
-
-    await listener.start();
-
-    const stream = {
-      on: jest.fn((event: string, handler: (chunk?: any) => void) => {
-        if (event === "data") handler("Hello world");
-        if (event === "end") handler();
-      }),
-    };
-
-    const session = {
-      remoteAddress: "1.2.3.4",
-      envelope: {
-        mailFrom: { address: "attacker@example.com" },
-        rcptTo: [{ address: "victim@example.com" }],
-      },
-      headers: { subject: "Test Email" },
-    };
-
-    await server.onData(stream, session, callback);
-
-    expect(EventNormalizer.normalizeSmtp).toHaveBeenCalled();
-    expect(router.dispatch).toHaveBeenCalled();
-    expect(callback).toHaveBeenCalledWith(null);
-  });
-
-  it("renvoie une erreur si normalizeSmtp échoue", async () => {
-    const callback = jest.fn();
-
-    (EventNormalizer.normalizeSmtp as jest.Mock).mockImplementation(() => {
-      throw new Error("bad smtp");
-    });
-
-    await listener.start();
-
-    const stream = {
-      on: jest.fn((event: string, handler: (chunk?: any) => void) => {
-        if (event === "data") handler("DATA");
-        if (event === "end") handler();
-      }),
-    };
-
-    const session = {
-      remoteAddress: "1.2.3.4",
-      envelope: {
-        mailFrom: { address: "a@b.com" },
-        rcptTo: [{ address: "c@d.com" }],
-      },
-      headers: { subject: "X" },
-    };
-
-    await server.onData(stream, session, callback);
-
-    expect(callback).toHaveBeenCalledWith(new Error("bad smtp"));
-  });
-
-  it("renvoie une erreur si router.dispatch échoue", async () => {
-    const callback = jest.fn();
-
-    (EventNormalizer.normalizeSmtp as jest.Mock).mockReturnValue({
-      id: "smtp-999",
-      type: "smtp",
-      timestamp: Date.now(),
-      ip: "1.2.3.4",
-      from: "a@b.com",
-      to: ["c@d.com"],
-      subject: "X",
-      body: "DATA",
-      raw: {},
-    });
-
-    listener["router"] = {
-      dispatch: jest.fn(async () => {
-        throw new Error("dispatch failed");
-      }),
-    } as any;
-
-    await listener.start();
-
-    const stream = {
-      on: jest.fn((event: string, handler: (chunk?: any) => void) => {
-        if (event === "data") handler("DATA");
-        if (event === "end") handler();
-      }),
-    };
-
-    const session = {
-      remoteAddress: "1.2.3.4",
-      envelope: {
-        mailFrom: { address: "a@b.com" },
-        rcptTo: [{ address: "c@d.com" }],
-      },
-      headers: { subject: "X" },
-    };
-
-    await server.onData(stream, session, callback);
-
-    expect(callback).toHaveBeenCalledWith(new Error("dispatch failed"));
-  });
-
-  it("stop() appelle server.close si disponible", async () => {
-    await listener.stop();
-    expect(server.close).toHaveBeenCalled();
-  });
-});
+import { describe, it, expect, jest, beforeEach } from "@jest/globals";
+import { SmtpListener } from "../smtp.listener";
+import { EventNormalizer } from "../../../core/event.normalizer";
+import { CoreRouter } from "../../../core/router";
+
+// Mock EventNormalizer
+jest.mock("../../../core/event.normalizer", () => ({
+  EventNormalizer: {
+    normalizeSmtp: jest.fn(),
+  },
+}));
+
+describe("SmtpListener", () => {
+  let router: CoreRouter;
+  let server: any;
+  let listener: SmtpListener;
+
+  beforeEach(() => {
+    router = {
+      dispatch: jest.fn(async (_event: any) => {}),
+    } as any;
+
+    server = {
+      onData: null,
+      close: jest.fn(),
+    };
+
+    listener = new SmtpListener(router, { server });
+  });
+
+  it("traite un email SMTP valide et appelle dispatch", async () => {
+    const callback = jest.fn();
+
+    // NormalizedSmtpEvent VALIDE
+    (EventNormalizer.normalizeSmtp as jest.Mock).mockReturnValue({
+      id: "smtp-123",
+      type: "smtp",
+      timestamp: Date.now(),
+      ip: "1.2.3.4",
+      from: "attacker@example.com",
+      to: ["victim@example.com"],
+      subject: "Test Email",
+      body: "Hello world",
+      raw: {},
+    });
+
+    await listener.start();
+
+    const stream = {
+      on: jest.fn((event: string, handler: (chunk?: any) => void) => {
+        if (event === "data") handler("Hello world");
+        if (event === "end") handler();
+      }),
+    };
+
+    const session = {
+      remoteAddress: "1.2.3.4",
+      envelope: {
+        mailFrom: { address: "attacker@example.com" },
+        rcptTo: [{ address: "victim@example.com" }],
+      },
+      headers: { subject: "Test Email" },
+    };
+
+    await server.onData(stream, session, callback);
+
+    expect(EventNormalizer.normalizeSmtp).toHaveBeenCalled();
+    expect(router.dispatch).toHaveBeenCalled();
+    expect(callback).toHaveBeenCalledWith(null);
+  });
+
+  it("renvoie une erreur si normalizeSmtp échoue", async () => {
+    const callback = jest.fn();
+
+    (EventNormalizer.normalizeSmtp as jest.Mock).mockImplementation(() => {
+      throw new Error("bad smtp");
+    });
+
+    await listener.start();
+
+    const stream = {
+      on: jest.fn((event: string, handler: (chunk?: any) => void) => {
+        if (event === "data") handler("DATA");
+        if (event === "end") handler();
+      }),
+    };
+
+    const session = {
+      remoteAddress: "1.2.3.4",
+      envelope: {
+        mailFrom: { address: "a@b.com" },
+        rcptTo: [{ address: "c@d.com" }],
+      },
+      headers: { subject: "X" },
+    };
+
+    await server.onData(stream, session, callback);
+
+    expect(callback).toHaveBeenCalledWith(new Error("bad smtp"));
+  });
+
+  it("renvoie une erreur si router.dispatch échoue", async () => {
+    const callback = jest.fn();
+
+    (EventNormalizer.normalizeSmtp as jest.Mock).mockReturnValue({
+      id: "smtp-999",
+      type: "smtp",
+      timestamp: Date.now(),
+      ip: "1.2.3.4",
+      from: "a@b.com",
+      to: ["c@d.com"],
+      subject: "X",
+      body: "DATA",
+      raw: {},
+    });
+
+    listener["router"] = {
+      dispatch: jest.fn(async () => {
+        throw new Error("dispatch failed");
+      }),
+    } as any;
+
+    await listener.start();
+
+    const stream = {
+      on: jest.fn((event: string, handler: (chunk?: any) => void) => {
+        if (event === "data") handler("DATA");
+        if (event === "end") handler();
+      }),
+    };
+
+    const session = {
+      remoteAddress: "1.2.3.4",
+      envelope: {
+        mailFrom: { address: "a@b.com" },
+        rcptTo: [{ address: "c@d.com" }],
+      },
+      headers: { subject: "X" },
+    };
+
+    await server.onData(stream, session, callback);
+
+    expect(callback).toHaveBeenCalledWith(new Error("dispatch failed"));
+  });
+
+  it("stop() appelle server.close si disponible", async () => {
+    await listener.stop();
+    expect(server.close).toHaveBeenCalled();
+  });
+});