shared/utils/videos/video-streaming-playlists.ts

   1 import { makeRawRequest } from '../requests/requests'
   2 import { sha256 } from '../../../server/helpers/core-utils'
   3 import { VideoStreamingPlaylist } from '../../models/videos/video-streaming-playlist.model'
   4 import { expect } from 'chai'
   5
   6 function getPlaylist (url: string, statusCodeExpected = 200) {
   7   return makeRawRequest(url, statusCodeExpected)
   8 }
   9
  10 function getSegment (url: string, statusCodeExpected = 200, range?: string) {
  11   return makeRawRequest(url, statusCodeExpected, range)
  12 }
  13
  14 function getSegmentSha256 (url: string, statusCodeExpected = 200) {
  15   return makeRawRequest(url, statusCodeExpected)
  16 }
  17
  18 async function checkSegmentHash (
  19   baseUrlPlaylist: string,
  20   baseUrlSegment: string,
  21   videoUUID: string,
  22   resolution: number,
  23   hlsPlaylist: VideoStreamingPlaylist
  24 ) {
  25   const res = await getPlaylist(`${baseUrlPlaylist}/${videoUUID}/${resolution}.m3u8`)
  26   const playlist = res.text
  27
  28   const videoName = `${videoUUID}-${resolution}-fragmented.mp4`
  29
  30   const matches = /#EXT-X-BYTERANGE:(\d+)@(\d+)/.exec(playlist)
  31
  32   const length = parseInt(matches[1], 10)
  33   const offset = parseInt(matches[2], 10)
  34   const range = `${offset}-${offset + length - 1}`
  35
  36   const res2 = await getSegment(`${baseUrlSegment}/${videoUUID}/${videoName}`, 206, `bytes=${range}`)
  37
  38   const resSha = await getSegmentSha256(hlsPlaylist.segmentsSha256Url)
  39
  40   const sha256Server = resSha.body[ videoName ][range]
  41   expect(sha256(res2.body)).to.equal(sha256Server)
  42 }
  43
  44 // ---------------------------------------------------------------------------
  45
  46 export {
  47   getPlaylist,
  48   getSegment,
  49   getSegmentSha256,
  50   checkSegmentHash
  51 }