Feat: Segment validation

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

@@ -1,11 +1,14 @@
 import { PeerConnection } from "bittorrent-tracker";
-import * as Command from "./commands";
-import * as Utils from "../utils/utils";
 import { Settings } from "../types";
+import * as Utils from "../utils/utils";
+import * as Command from "./commands";
 
 export type PeerSettings = Pick<
   Settings,
-  "p2pNotReceivingBytesTimeoutMs" | "webRtcMaxMessageSize" | "p2pErrorRetries"
+  | "p2pNotReceivingBytesTimeoutMs"
+  | "webRtcMaxMessageSize"
+  | "p2pErrorRetries"
+  | "validateP2PSegment"
 >;
 
 export class PeerProtocol {

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

@@ -1,16 +1,16 @@
 import { PeerConnection } from "bittorrent-tracker";
-import { PeerProtocol, PeerSettings } from "./peer-protocol";
+import debug from "debug";
 import {
+  PeerRequestErrorType,
   Request,
   RequestControls,
   RequestError,
-  PeerRequestErrorType,
   RequestInnerErrorType,
 } from "../requests/request";
-import * as Command from "./commands";
 import { Segment } from "../types";
 import * as Utils from "../utils/utils";
-import debug from "debug";
+import * as Command from "./commands";
+import { PeerProtocol, PeerSettings } from "./peer-protocol";
 
 const { PeerCommandType } = Command;
 type PeerEventHandlers = {
@@ -45,6 +45,7 @@ export class Peer {
     this.id = Peer.getPeerIdFromConnection(connection);
     this.peerProtocol = new PeerProtocol(connection, settings, {
       onSegmentChunkReceived: this.onSegmentChunkReceived,
+      // eslint-disable-next-line @typescript-eslint/no-misused-promises
       onCommandReceived: this.onCommandReceived,
     });
 
@@ -62,7 +63,7 @@ export class Peer {
     if (this.httpLoadingSegments.has(externalId)) return "http-loading";
   }
 
-  private onCommandReceived = (command: Command.PeerCommand) => {
+  private onCommandReceived = async (command: Command.PeerCommand) => {
     switch (command.c) {
       case PeerCommandType.SegmentsAnnouncement:
         this.loadedSegments = new Set(command.l);
@@ -85,7 +86,9 @@ export class Peer {
             request.setTotalBytes(command.s);
           } else if (request.totalBytes - request.loadedBytes !== command.s) {
             request.clearLoadedBytes();
-            this.cancelSegmentDownloading("peer-response-bytes-mismatch");
+            this.cancelSegmentDownloading(
+              "peer-response-bytes-length-mismatch",
+            );
             this.destroy();
           }
         }
@@ -110,7 +113,20 @@ export class Peer {
 
         if (isWrongBytes) {
           request.clearLoadedBytes();
-          this.cancelSegmentDownloading("peer-response-bytes-mismatch");
+          this.cancelSegmentDownloading("peer-response-bytes-length-mismatch");
+          this.destroy();
+          return;
+        }
+
+        const isValid =
+          (await this.settings.validateP2PSegment?.(
+            request.segment.url,
+            request.segment.byteRange,
+          )) ?? true;
+
+        if (!isValid) {
+          request.clearLoadedBytes();
+          this.cancelSegmentDownloading("p2p-segment-validation-failed");
           this.destroy();
           return;
         }
@@ -145,7 +161,7 @@ export class Peer {
 
     if (isOverflow) {
       request.clearLoadedBytes();
-      this.cancelSegmentDownloading("peer-response-bytes-mismatch");
+      this.cancelSegmentDownloading("peer-response-bytes-length-mismatch");
       this.destroy();
       return;
     }

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

@@ -1,8 +1,8 @@
-import { Segment, Playback, BandwidthCalculators } from "../types";
+import debug from "debug";
+import { BandwidthCalculators, Playback, Segment } from "../types";
+import * as LoggerUtils from "../utils/logger";
 import * as StreamUtils from "../utils/stream";
 import * as Utils from "../utils/utils";
-import * as LoggerUtils from "../utils/logger";
-import debug from "debug";
 
 export type LoadProgress = {
   startTimestamp: number;
@@ -323,10 +323,11 @@ export type HttpRequestErrorType =
   | "http-unexpected-status-code";
 
 export type PeerRequestErrorType =
-  | "peer-response-bytes-mismatch"
+  | "peer-response-bytes-length-mismatch"
   | "peer-protocol-violation"
   | "peer-segment-absent"
-  | "peer-closed";
+  | "peer-closed"
+  | "p2p-segment-validation-failed";
 
 type RequestErrorType =
   | RequestInnerErrorType

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

@@ -1,5 +1,5 @@
-import { RequestAttempt } from "./requests/request";
 import { BandwidthCalculator } from "./bandwidth-calculator";
+import { RequestAttempt } from "./requests/request";
 
 export type StreamType = "main" | "secondary";
 
@@ -53,6 +53,7 @@ export type Settings = {
   httpNotReceivingBytesTimeoutMs: number;
   httpErrorRetries: number;
   p2pErrorRetries: number;
+  validateP2PSegment?: (url: string, byteRange?: ByteRange) => Promise<boolean>;
 };
 
 export type CoreEventHandlers = {