12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182 |
- 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
- }
|