aboutsummaryrefslogblamecommitdiffhomepage
path: root/shared/extra-utils/videos/video-streaming-playlists.ts
blob: 99c2e1880d2e1a1ec47ba778a20a1aa889ab19bd (plain) (tree)
1
2
3
4
5
6
7
8
9
10



                                                                                           
                                                                                  
 
                                                                                


                                                
                                                                                               


                                                       
                                                                                     




















                                                                                     
                                                                                                                                    


                                                                      
                                                    


                                                  













                                                                                













                                                                                                                             




                                                                              
                                   
                   
                       

                  
import { makeRawRequest } from '../requests/requests'
import { sha256 } from '../../../server/helpers/core-utils'
import { VideoStreamingPlaylist } from '../../models/videos/video-streaming-playlist.model'
import { expect } from 'chai'
import { HttpStatusCode } from '../../../shared/core-utils/miscs/http-error-codes'

function getPlaylist (url: string, statusCodeExpected = HttpStatusCode.OK_200) {
  return makeRawRequest(url, statusCodeExpected)
}

function getSegment (url: string, statusCodeExpected = HttpStatusCode.OK_200, range?: string) {
  return makeRawRequest(url, statusCodeExpected, range)
}

function getSegmentSha256 (url: string, statusCodeExpected = HttpStatusCode.OK_200) {
  return makeRawRequest(url, statusCodeExpected)
}

async function checkSegmentHash (
  baseUrlPlaylist: string,
  baseUrlSegment: string,
  videoUUID: string,
  resolution: number,
  hlsPlaylist: VideoStreamingPlaylist
) {
  const res = await getPlaylist(`${baseUrlPlaylist}/${videoUUID}/${resolution}.m3u8`)
  const playlist = res.text

  const videoName = `${videoUUID}-${resolution}-fragmented.mp4`

  const matches = /#EXT-X-BYTERANGE:(\d+)@(\d+)/.exec(playlist)

  const length = parseInt(matches[1], 10)
  const offset = parseInt(matches[2], 10)
  const range = `${offset}-${offset + length - 1}`

  const res2 = await getSegment(`${baseUrlSegment}/${videoUUID}/${videoName}`, HttpStatusCode.PARTIAL_CONTENT_206, `bytes=${range}`)

  const resSha = await getSegmentSha256(hlsPlaylist.segmentsSha256Url)

  const sha256Server = resSha.body[videoName][range]
  expect(sha256(res2.body)).to.equal(sha256Server)
}

async function checkLiveSegmentHash (
  baseUrlSegment: string,
  videoUUID: string,
  segmentName: string,
  hlsPlaylist: VideoStreamingPlaylist
) {
  const res2 = await getSegment(`${baseUrlSegment}/${videoUUID}/${segmentName}`)

  const resSha = await getSegmentSha256(hlsPlaylist.segmentsSha256Url)

  const sha256Server = resSha.body[segmentName]
  expect(sha256(res2.body)).to.equal(sha256Server)
}

async function checkResolutionsInMasterPlaylist (playlistUrl: string, resolutions: number[]) {
  const res = await getPlaylist(playlistUrl)

  const masterPlaylist = res.text

  for (const resolution of resolutions) {
    const reg = new RegExp(
      '#EXT-X-STREAM-INF:BANDWIDTH=\\d+,RESOLUTION=\\d+x' + resolution + ',(FRAME-RATE=\\d+,)?CODECS="avc1.64001f,mp4a.40.2"'
    )

    expect(masterPlaylist).to.match(reg)
  }
}

// ---------------------------------------------------------------------------

export {
  getPlaylist,
  getSegment,
  checkResolutionsInMasterPlaylist,
  getSegmentSha256,
  checkLiveSegmentHash,
  checkSegmentHash
}