mirror of https://github.com/Chocobozzz/PeerTube
83 lines
2.6 KiB
TypeScript
83 lines
2.6 KiB
TypeScript
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
|
|
}
|