Add validateHTTPSegment function (#467)

Author: Chocobozzz

packages/p2p-media-loader-core/src/core.ts

@@ -65,6 +65,7 @@ export class Core<TStream extends Stream = Stream> {
       ],
     },
     validateP2PSegment: undefined,
+    validateHTTPSegment: undefined,
     httpRequestSetup: undefined,
     swarmId: undefined,
   };

packages/p2p-media-loader-core/src/http-loader.ts

@@ -12,7 +12,7 @@ import { EventTarget } from "./utils/event-target.js";
 
 type HttpConfig = Pick<
   CoreConfig,
-  "httpNotReceivingBytesTimeoutMs" | "httpRequestSetup"
+  "httpNotReceivingBytesTimeoutMs" | "httpRequestSetup" | "validateHTTPSegment"
 >;
 
 export class HttpRequestExecutor {
@@ -101,6 +101,21 @@ export class HttpRequestExecutor {
         this.requestControls.addLoadedChunk(chunk);
         this.onChunkDownloaded(chunk.byteLength, "http");
       }
+
+      const isValid =
+        (await this.httpConfig.validateHTTPSegment?.(
+          segment.url,
+          segment.byteRange,
+          this.request.data,
+        )) ?? true;
+
+    if (!isValid) {
+      this.request.clearLoadedBytes();
+      throw new RequestError<"http-segment-validation-failed">(
+        "http-segment-validation-failed"
+      );
+    }
+
       requestControls.completeOnSuccess();
     } catch (error) {
       this.handleError(error);

packages/p2p-media-loader-core/src/types.ts

@@ -408,6 +408,24 @@ export type StreamConfig = {
     data: ArrayBuffer,
   ) => Promise<boolean>;
 
+  /**
+   * Optional function to validate a HTTP segment before fully integrating it into the playback buffer.
+   * @param url URL of the segment to validate.
+   * @param byteRange Optional byte range of the segment.
+   * @param data Downloaded segment data.
+   * @returns A promise that resolves with a boolean indicating if the segment is valid.
+   *
+   * @default
+   * ```typescript
+   * validateHTTPSegment: undefined
+   * ```
+   */
+  validateHTTPSegment?: (
+    url: string,
+    byteRange: ByteRange | undefined,
+    data: ArrayBuffer,
+  ) => Promise<boolean>;
+
   /**
    * Optional function to customize the setup of HTTP requests for segment downloads.
    * @param segmentUrl URL of the segment.
@@ -631,7 +649,8 @@ export type RequestAbortErrorType = "abort" | "bytes-receiving-timeout";
 export type HttpRequestErrorType =
   | "http-error"
   | "http-bytes-mismatch"
-  | "http-unexpected-status-code";
+  | "http-unexpected-status-code"
+  | "http-segment-validation-failed";
 
 /** Defines the types of errors specific to peer-to-peer requests. */
 export type PeerRequestErrorType =