更改视频下载逻辑
This commit is contained in:
mtvpls
@@ -0,0 +1,315 @@
|
||||
/**
|
||||
* AES 解密器
|
||||
* 代码来自 hls.js: https://github.com/video-dev/hls.js
|
||||
*/
|
||||
|
||||
function removePadding(buffer: ArrayBuffer): ArrayBuffer {
|
||||
const outputBytes = buffer.byteLength;
|
||||
const paddingBytes = outputBytes && new DataView(buffer).getUint8(outputBytes - 1);
|
||||
if (paddingBytes) {
|
||||
return buffer.slice(0, outputBytes - paddingBytes);
|
||||
} else {
|
||||
return buffer;
|
||||
}
|
||||
}
|
||||
|
||||
export class AESDecryptor {
|
||||
private rcon: number[];
|
||||
private subMix: Uint32Array[];
|
||||
private invSubMix: Uint32Array[];
|
||||
private sBox: Uint32Array;
|
||||
private invSBox: Uint32Array;
|
||||
private key: Uint32Array;
|
||||
private keySize?: number;
|
||||
private ksRows?: number;
|
||||
private keySchedule?: Uint32Array;
|
||||
private invKeySchedule?: Uint32Array;
|
||||
|
||||
constructor() {
|
||||
this.rcon = [0x0, 0x1, 0x2, 0x4, 0x8, 0x10, 0x20, 0x40, 0x80, 0x1b, 0x36];
|
||||
this.subMix = [new Uint32Array(256), new Uint32Array(256), new Uint32Array(256), new Uint32Array(256)];
|
||||
this.invSubMix = [new Uint32Array(256), new Uint32Array(256), new Uint32Array(256), new Uint32Array(256)];
|
||||
this.sBox = new Uint32Array(256);
|
||||
this.invSBox = new Uint32Array(256);
|
||||
this.key = new Uint32Array(0);
|
||||
|
||||
this.initTable();
|
||||
}
|
||||
|
||||
private uint8ArrayToUint32Array_(arrayBuffer: ArrayBuffer): Uint32Array {
|
||||
const view = new DataView(arrayBuffer);
|
||||
const newArray = new Uint32Array(4);
|
||||
for (let i = 0; i < 4; i++) {
|
||||
newArray[i] = view.getUint32(i * 4);
|
||||
}
|
||||
return newArray;
|
||||
}
|
||||
|
||||
private initTable(): void {
|
||||
const sBox = this.sBox;
|
||||
const invSBox = this.invSBox;
|
||||
const subMix = this.subMix;
|
||||
const subMix0 = subMix[0];
|
||||
const subMix1 = subMix[1];
|
||||
const subMix2 = subMix[2];
|
||||
const subMix3 = subMix[3];
|
||||
const invSubMix = this.invSubMix;
|
||||
const invSubMix0 = invSubMix[0];
|
||||
const invSubMix1 = invSubMix[1];
|
||||
const invSubMix2 = invSubMix[2];
|
||||
const invSubMix3 = invSubMix[3];
|
||||
|
||||
const d = new Uint32Array(256);
|
||||
let x = 0;
|
||||
let xi = 0;
|
||||
let i = 0;
|
||||
|
||||
for (i = 0; i < 256; i++) {
|
||||
if (i < 128) {
|
||||
d[i] = i << 1;
|
||||
} else {
|
||||
d[i] = (i << 1) ^ 0x11b;
|
||||
}
|
||||
}
|
||||
|
||||
for (i = 0; i < 256; i++) {
|
||||
let sx = xi ^ (xi << 1) ^ (xi << 2) ^ (xi << 3) ^ (xi << 4);
|
||||
sx = (sx >>> 8) ^ (sx & 0xff) ^ 0x63;
|
||||
sBox[x] = sx;
|
||||
invSBox[sx] = x;
|
||||
|
||||
const x2 = d[x];
|
||||
const x4 = d[x2];
|
||||
const x8 = d[x4];
|
||||
|
||||
let t = (d[sx] * 0x101) ^ (sx * 0x1010100);
|
||||
subMix0[x] = (t << 24) | (t >>> 8);
|
||||
subMix1[x] = (t << 16) | (t >>> 16);
|
||||
subMix2[x] = (t << 8) | (t >>> 24);
|
||||
subMix3[x] = t;
|
||||
|
||||
t = (x8 * 0x1010101) ^ (x4 * 0x10001) ^ (x2 * 0x101) ^ (x * 0x1010100);
|
||||
invSubMix0[sx] = (t << 24) | (t >>> 8);
|
||||
invSubMix1[sx] = (t << 16) | (t >>> 16);
|
||||
invSubMix2[sx] = (t << 8) | (t >>> 24);
|
||||
invSubMix3[sx] = t;
|
||||
|
||||
if (!x) {
|
||||
x = xi = 1;
|
||||
} else {
|
||||
x = x2 ^ d[d[d[x8 ^ x2]]];
|
||||
xi ^= d[d[xi]];
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
expandKey(keyBuffer: ArrayBuffer): void {
|
||||
const key = this.uint8ArrayToUint32Array_(keyBuffer);
|
||||
let sameKey = true;
|
||||
let offset = 0;
|
||||
|
||||
while (offset < key.length && sameKey) {
|
||||
sameKey = key[offset] === this.key[offset];
|
||||
offset++;
|
||||
}
|
||||
|
||||
if (sameKey) {
|
||||
return;
|
||||
}
|
||||
|
||||
this.key = key;
|
||||
const keySize = (this.keySize = key.length);
|
||||
|
||||
if (keySize !== 4 && keySize !== 6 && keySize !== 8) {
|
||||
throw new Error('Invalid aes key size=' + keySize);
|
||||
}
|
||||
|
||||
const ksRows = (this.ksRows = (keySize + 6 + 1) * 4);
|
||||
let ksRow;
|
||||
let invKsRow;
|
||||
|
||||
const keySchedule = (this.keySchedule = new Uint32Array(ksRows));
|
||||
const invKeySchedule = (this.invKeySchedule = new Uint32Array(ksRows));
|
||||
const sbox = this.sBox;
|
||||
const rcon = this.rcon;
|
||||
|
||||
const invSubMix = this.invSubMix;
|
||||
const invSubMix0 = invSubMix[0];
|
||||
const invSubMix1 = invSubMix[1];
|
||||
const invSubMix2 = invSubMix[2];
|
||||
const invSubMix3 = invSubMix[3];
|
||||
|
||||
let prev;
|
||||
let t;
|
||||
|
||||
for (ksRow = 0; ksRow < ksRows; ksRow++) {
|
||||
if (ksRow < keySize) {
|
||||
prev = keySchedule[ksRow] = key[ksRow];
|
||||
continue;
|
||||
}
|
||||
t = prev;
|
||||
|
||||
if (ksRow % keySize === 0) {
|
||||
t = (t << 8) | (t >>> 24);
|
||||
t = (sbox[t >>> 24] << 24) | (sbox[(t >>> 16) & 0xff] << 16) | (sbox[(t >>> 8) & 0xff] << 8) | sbox[t & 0xff];
|
||||
t ^= rcon[(ksRow / keySize) | 0] << 24;
|
||||
} else if (keySize > 6 && ksRow % keySize === 4) {
|
||||
t = (sbox[t >>> 24] << 24) | (sbox[(t >>> 16) & 0xff] << 16) | (sbox[(t >>> 8) & 0xff] << 8) | sbox[t & 0xff];
|
||||
}
|
||||
|
||||
keySchedule[ksRow] = prev = (keySchedule[ksRow - keySize] ^ t) >>> 0;
|
||||
}
|
||||
|
||||
for (invKsRow = 0; invKsRow < ksRows; invKsRow++) {
|
||||
ksRow = ksRows - invKsRow;
|
||||
if (invKsRow & 3) {
|
||||
t = keySchedule[ksRow];
|
||||
} else {
|
||||
t = keySchedule[ksRow - 4];
|
||||
}
|
||||
|
||||
if (invKsRow < 4 || ksRow <= 4) {
|
||||
invKeySchedule[invKsRow] = t;
|
||||
} else {
|
||||
invKeySchedule[invKsRow] =
|
||||
invSubMix0[sbox[t >>> 24]] ^
|
||||
invSubMix1[sbox[(t >>> 16) & 0xff]] ^
|
||||
invSubMix2[sbox[(t >>> 8) & 0xff]] ^
|
||||
invSubMix3[sbox[t & 0xff]];
|
||||
}
|
||||
|
||||
invKeySchedule[invKsRow] = invKeySchedule[invKsRow] >>> 0;
|
||||
}
|
||||
}
|
||||
|
||||
private networkToHostOrderSwap(word: number): number {
|
||||
return (word << 24) | ((word & 0xff00) << 8) | ((word & 0xff0000) >> 8) | (word >>> 24);
|
||||
}
|
||||
|
||||
decrypt(inputArrayBuffer: ArrayBuffer, offset: number, aesIV: ArrayBuffer, removePKCS7Padding: boolean): ArrayBuffer {
|
||||
if (!this.keySize || !this.ksRows || !this.keySchedule || !this.invKeySchedule) {
|
||||
throw new Error('AES key not expanded');
|
||||
}
|
||||
|
||||
const nRounds = this.keySize + 6;
|
||||
const invKeySchedule = this.invKeySchedule;
|
||||
const invSBOX = this.invSBox;
|
||||
|
||||
const invSubMix = this.invSubMix;
|
||||
const invSubMix0 = invSubMix[0];
|
||||
const invSubMix1 = invSubMix[1];
|
||||
const invSubMix2 = invSubMix[2];
|
||||
const invSubMix3 = invSubMix[3];
|
||||
|
||||
const initVector = this.uint8ArrayToUint32Array_(aesIV);
|
||||
let initVector0 = initVector[0];
|
||||
let initVector1 = initVector[1];
|
||||
let initVector2 = initVector[2];
|
||||
let initVector3 = initVector[3];
|
||||
|
||||
const inputInt32 = new Int32Array(inputArrayBuffer);
|
||||
const outputInt32 = new Int32Array(inputInt32.length);
|
||||
|
||||
let t0, t1, t2, t3;
|
||||
let s0, s1, s2, s3;
|
||||
let inputWords0, inputWords1, inputWords2, inputWords3;
|
||||
|
||||
let ksRow, i;
|
||||
const swapWord = this.networkToHostOrderSwap;
|
||||
|
||||
while (offset < inputInt32.length) {
|
||||
inputWords0 = swapWord(inputInt32[offset]);
|
||||
inputWords1 = swapWord(inputInt32[offset + 1]);
|
||||
inputWords2 = swapWord(inputInt32[offset + 2]);
|
||||
inputWords3 = swapWord(inputInt32[offset + 3]);
|
||||
|
||||
s0 = inputWords0 ^ invKeySchedule[0];
|
||||
s1 = inputWords3 ^ invKeySchedule[1];
|
||||
s2 = inputWords2 ^ invKeySchedule[2];
|
||||
s3 = inputWords1 ^ invKeySchedule[3];
|
||||
|
||||
ksRow = 4;
|
||||
|
||||
for (i = 1; i < nRounds; i++) {
|
||||
t0 =
|
||||
invSubMix0[s0 >>> 24] ^
|
||||
invSubMix1[(s1 >> 16) & 0xff] ^
|
||||
invSubMix2[(s2 >> 8) & 0xff] ^
|
||||
invSubMix3[s3 & 0xff] ^
|
||||
invKeySchedule[ksRow];
|
||||
t1 =
|
||||
invSubMix0[s1 >>> 24] ^
|
||||
invSubMix1[(s2 >> 16) & 0xff] ^
|
||||
invSubMix2[(s3 >> 8) & 0xff] ^
|
||||
invSubMix3[s0 & 0xff] ^
|
||||
invKeySchedule[ksRow + 1];
|
||||
t2 =
|
||||
invSubMix0[s2 >>> 24] ^
|
||||
invSubMix1[(s3 >> 16) & 0xff] ^
|
||||
invSubMix2[(s0 >> 8) & 0xff] ^
|
||||
invSubMix3[s1 & 0xff] ^
|
||||
invKeySchedule[ksRow + 2];
|
||||
t3 =
|
||||
invSubMix0[s3 >>> 24] ^
|
||||
invSubMix1[(s0 >> 16) & 0xff] ^
|
||||
invSubMix2[(s1 >> 8) & 0xff] ^
|
||||
invSubMix3[s2 & 0xff] ^
|
||||
invKeySchedule[ksRow + 3];
|
||||
|
||||
s0 = t0;
|
||||
s1 = t1;
|
||||
s2 = t2;
|
||||
s3 = t3;
|
||||
|
||||
ksRow = ksRow + 4;
|
||||
}
|
||||
|
||||
t0 =
|
||||
((invSBOX[s0 >>> 24] << 24) ^
|
||||
(invSBOX[(s1 >> 16) & 0xff] << 16) ^
|
||||
(invSBOX[(s2 >> 8) & 0xff] << 8) ^
|
||||
invSBOX[s3 & 0xff]) ^
|
||||
invKeySchedule[ksRow];
|
||||
t1 =
|
||||
((invSBOX[s1 >>> 24] << 24) ^
|
||||
(invSBOX[(s2 >> 16) & 0xff] << 16) ^
|
||||
(invSBOX[(s3 >> 8) & 0xff] << 8) ^
|
||||
invSBOX[s0 & 0xff]) ^
|
||||
invKeySchedule[ksRow + 1];
|
||||
t2 =
|
||||
((invSBOX[s2 >>> 24] << 24) ^
|
||||
(invSBOX[(s3 >> 16) & 0xff] << 16) ^
|
||||
(invSBOX[(s0 >> 8) & 0xff] << 8) ^
|
||||
invSBOX[s1 & 0xff]) ^
|
||||
invKeySchedule[ksRow + 2];
|
||||
t3 =
|
||||
((invSBOX[s3 >>> 24] << 24) ^
|
||||
(invSBOX[(s0 >> 16) & 0xff] << 16) ^
|
||||
(invSBOX[(s1 >> 8) & 0xff] << 8) ^
|
||||
invSBOX[s2 & 0xff]) ^
|
||||
invKeySchedule[ksRow + 3];
|
||||
|
||||
outputInt32[offset] = swapWord(t0 ^ initVector0);
|
||||
outputInt32[offset + 1] = swapWord(t3 ^ initVector1);
|
||||
outputInt32[offset + 2] = swapWord(t2 ^ initVector2);
|
||||
outputInt32[offset + 3] = swapWord(t1 ^ initVector3);
|
||||
|
||||
initVector0 = inputWords0;
|
||||
initVector1 = inputWords1;
|
||||
initVector2 = inputWords2;
|
||||
initVector3 = inputWords3;
|
||||
|
||||
offset = offset + 4;
|
||||
}
|
||||
|
||||
return removePKCS7Padding ? removePadding(outputInt32.buffer) : outputInt32.buffer;
|
||||
}
|
||||
|
||||
destroy(): void {
|
||||
this.key = new Uint32Array(0);
|
||||
this.keySize = undefined;
|
||||
this.ksRows = undefined;
|
||||
this.keySchedule = undefined;
|
||||
this.invKeySchedule = undefined;
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,558 @@
|
||||
/**
|
||||
* M3U8 下载器核心逻辑
|
||||
* 基于 M3U8Download 项目改造为 TypeScript 版本
|
||||
*/
|
||||
|
||||
import { AESDecryptor } from './aes-decryptor';
|
||||
// @ts-ignore - mux.js 没有类型定义
|
||||
import * as muxjs from 'mux.js';
|
||||
|
||||
export interface M3U8DownloadTask {
|
||||
id: string;
|
||||
url: string;
|
||||
title: string;
|
||||
type: 'TS' | 'MP4';
|
||||
status: 'ready' | 'downloading' | 'pause' | 'done' | 'error';
|
||||
finishList: Array<{ title: string; status: '' | 'is-downloading' | 'is-success' | 'is-error' }>;
|
||||
tsUrlList: string[];
|
||||
requests: XMLHttpRequest[];
|
||||
mediaFileList: ArrayBuffer[];
|
||||
downloadIndex: number;
|
||||
downloading: boolean;
|
||||
durationSecond: number;
|
||||
beginTime: Date;
|
||||
errorNum: number;
|
||||
finishNum: number;
|
||||
retryNum: number;
|
||||
retryCountdown: number;
|
||||
rangeDownload: {
|
||||
isShowRange: boolean;
|
||||
startSegment: number;
|
||||
endSegment: number;
|
||||
targetSegment: number;
|
||||
};
|
||||
aesConf: {
|
||||
method: string;
|
||||
uri: string;
|
||||
iv: Uint8Array | null;
|
||||
key: ArrayBuffer | null;
|
||||
decryption: AESDecryptor | null;
|
||||
};
|
||||
}
|
||||
|
||||
export interface M3U8DownloaderOptions {
|
||||
onProgress?: (task: M3U8DownloadTask) => void;
|
||||
onComplete?: (task: M3U8DownloadTask) => void;
|
||||
onError?: (task: M3U8DownloadTask, error: string) => void;
|
||||
}
|
||||
|
||||
export class M3U8Downloader {
|
||||
private tasks: Map<string, M3U8DownloadTask> = new Map();
|
||||
private currentTask: M3U8DownloadTask | null = null;
|
||||
private options: M3U8DownloaderOptions;
|
||||
|
||||
constructor(options: M3U8DownloaderOptions = {}) {
|
||||
this.options = options;
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建下载任务
|
||||
*/
|
||||
async createTask(url: string, title: string, type: 'TS' | 'MP4' = 'TS'): Promise<string> {
|
||||
const taskId = 't_' + Date.now() + Math.random().toString(36).substr(2, 9);
|
||||
|
||||
try {
|
||||
// 获取 m3u8 文件内容
|
||||
const m3u8Content = await this.fetchM3U8(url);
|
||||
|
||||
if (!m3u8Content.startsWith('#EXTM3U')) {
|
||||
throw new Error('无效的 m3u8 链接');
|
||||
}
|
||||
|
||||
// 检查是否是主播放列表
|
||||
if (this.isMasterPlaylist(m3u8Content)) {
|
||||
const streams = this.parseStreamInfo(m3u8Content, url);
|
||||
if (streams.length > 0) {
|
||||
// 自动选择最高清晰度
|
||||
url = streams[0].url;
|
||||
const subM3u8Content = await this.fetchM3U8(url);
|
||||
return this.processM3U8Content(taskId, url, title, type, subM3u8Content);
|
||||
}
|
||||
}
|
||||
|
||||
return this.processM3U8Content(taskId, url, title, type, m3u8Content);
|
||||
} catch (error) {
|
||||
throw new Error(`创建任务失败: ${error}`);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理 M3U8 内容
|
||||
*/
|
||||
private processM3U8Content(
|
||||
taskId: string,
|
||||
url: string,
|
||||
title: string,
|
||||
type: 'TS' | 'MP4',
|
||||
m3u8Content: string
|
||||
): string {
|
||||
const task: M3U8DownloadTask = {
|
||||
id: taskId,
|
||||
url,
|
||||
title,
|
||||
type,
|
||||
status: 'ready',
|
||||
finishList: [],
|
||||
tsUrlList: [],
|
||||
requests: [],
|
||||
mediaFileList: [],
|
||||
downloadIndex: 0,
|
||||
downloading: false,
|
||||
durationSecond: 0,
|
||||
beginTime: new Date(),
|
||||
errorNum: 0,
|
||||
finishNum: 0,
|
||||
retryNum: 3,
|
||||
retryCountdown: 0,
|
||||
rangeDownload: {
|
||||
isShowRange: false,
|
||||
startSegment: 1,
|
||||
endSegment: 0,
|
||||
targetSegment: 0,
|
||||
},
|
||||
aesConf: {
|
||||
method: '',
|
||||
uri: '',
|
||||
iv: null,
|
||||
key: null,
|
||||
decryption: null,
|
||||
},
|
||||
};
|
||||
|
||||
// 解析 TS 片段
|
||||
const lines = m3u8Content.split('\n');
|
||||
for (let i = 0; i < lines.length; i++) {
|
||||
const line = lines[i].trim();
|
||||
|
||||
if (line.startsWith('#EXTINF:')) {
|
||||
const duration = parseFloat(line.split('#EXTINF:')[1]);
|
||||
task.durationSecond += duration;
|
||||
} else if (line.startsWith('#EXT-X-KEY')) {
|
||||
const keyMatch = line.match(/METHOD=([^,]+)(?:,URI="([^"]+)")?(?:,IV=([^,]+))?/);
|
||||
if (keyMatch) {
|
||||
task.aesConf.method = keyMatch[1];
|
||||
task.aesConf.uri = keyMatch[2] ? this.applyURL(keyMatch[2], url) : '';
|
||||
task.aesConf.iv = keyMatch[3] ? this.parseIV(keyMatch[3]) : null;
|
||||
}
|
||||
} else if (line && !line.startsWith('#')) {
|
||||
task.tsUrlList.push(this.applyURL(line, url));
|
||||
task.finishList.push({ title: line, status: '' });
|
||||
}
|
||||
}
|
||||
|
||||
task.rangeDownload.endSegment = task.tsUrlList.length;
|
||||
task.rangeDownload.targetSegment = task.tsUrlList.length;
|
||||
|
||||
this.tasks.set(taskId, task);
|
||||
return taskId;
|
||||
}
|
||||
|
||||
/**
|
||||
* 开始下载任务
|
||||
*/
|
||||
async startTask(taskId: string): Promise<void> {
|
||||
const task = this.tasks.get(taskId);
|
||||
if (!task) {
|
||||
throw new Error('任务不存在');
|
||||
}
|
||||
|
||||
if (task.status === 'downloading') {
|
||||
return;
|
||||
}
|
||||
|
||||
// 如果需要 AES 解密，先获取密钥
|
||||
if (task.aesConf.method && task.aesConf.method !== 'NONE' && !task.aesConf.key) {
|
||||
await this.getAESKey(task);
|
||||
}
|
||||
|
||||
task.status = 'downloading';
|
||||
this.currentTask = task;
|
||||
this.downloadTS(task);
|
||||
}
|
||||
|
||||
/**
|
||||
* 暂停任务
|
||||
*/
|
||||
pauseTask(taskId: string): void {
|
||||
const task = this.tasks.get(taskId);
|
||||
if (!task) return;
|
||||
|
||||
task.status = 'pause';
|
||||
this.abortRequests(task);
|
||||
}
|
||||
|
||||
/**
|
||||
* 取消任务
|
||||
*/
|
||||
cancelTask(taskId: string): void {
|
||||
const task = this.tasks.get(taskId);
|
||||
if (!task) return;
|
||||
|
||||
this.abortRequests(task);
|
||||
this.tasks.delete(taskId);
|
||||
|
||||
if (this.currentTask?.id === taskId) {
|
||||
this.currentTask = null;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取任务信息
|
||||
*/
|
||||
getTask(taskId: string): M3U8DownloadTask | undefined {
|
||||
return this.tasks.get(taskId);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取所有任务
|
||||
*/
|
||||
getAllTasks(): M3U8DownloadTask[] {
|
||||
return Array.from(this.tasks.values());
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取下载进度
|
||||
*/
|
||||
getProgress(taskId: string): number {
|
||||
const task = this.tasks.get(taskId);
|
||||
if (!task) return 0;
|
||||
|
||||
if (task.rangeDownload.targetSegment === 0) return 0;
|
||||
return (task.finishNum / task.rangeDownload.targetSegment) * 100;
|
||||
}
|
||||
|
||||
/**
|
||||
* 下载 TS 片段
|
||||
*/
|
||||
private downloadTS(task: M3U8DownloadTask): void {
|
||||
const download = () => {
|
||||
const isPause = task.status === 'pause';
|
||||
const index = task.downloadIndex;
|
||||
|
||||
if (index >= task.rangeDownload.endSegment || isPause) {
|
||||
return;
|
||||
}
|
||||
|
||||
task.downloadIndex++;
|
||||
|
||||
if (task.finishList[index] && task.finishList[index].status === '') {
|
||||
task.finishList[index].status = 'is-downloading';
|
||||
|
||||
const xhr = new XMLHttpRequest();
|
||||
xhr.responseType = 'arraybuffer';
|
||||
xhr.onreadystatechange = () => {
|
||||
if (xhr.readyState === 4) {
|
||||
if (xhr.status >= 200 && xhr.status < 300) {
|
||||
this.dealTS(task, xhr.response, index, () => {
|
||||
if (task.downloadIndex < task.rangeDownload.endSegment && !isPause) {
|
||||
download();
|
||||
}
|
||||
});
|
||||
} else {
|
||||
task.errorNum++;
|
||||
task.finishList[index].status = 'is-error';
|
||||
this.options.onError?.(task, `片段 ${index} 下载失败`);
|
||||
|
||||
if (task.downloadIndex < task.rangeDownload.endSegment) {
|
||||
!isPause && download();
|
||||
}
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
xhr.open('GET', task.tsUrlList[index], true);
|
||||
xhr.send();
|
||||
task.requests.push(xhr);
|
||||
} else if (task.downloadIndex < task.rangeDownload.endSegment) {
|
||||
!isPause && download();
|
||||
}
|
||||
};
|
||||
|
||||
// 并发下载 6 个片段
|
||||
const concurrency = Math.min(6, task.rangeDownload.targetSegment - task.finishNum);
|
||||
for (let i = 0; i < concurrency; i++) {
|
||||
download();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理 TS 片段
|
||||
*/
|
||||
private dealTS(
|
||||
task: M3U8DownloadTask,
|
||||
file: ArrayBuffer,
|
||||
index: number,
|
||||
callback: () => void
|
||||
): void {
|
||||
let data = file;
|
||||
|
||||
// AES 解密
|
||||
if (task.aesConf.key) {
|
||||
data = this.aesDecrypt(task, data, index);
|
||||
}
|
||||
|
||||
// MP4 转码（如果需要）
|
||||
if (task.type === 'MP4') {
|
||||
this.conversionMp4(task, data, index, (convertedData) => {
|
||||
task.mediaFileList[index - task.rangeDownload.startSegment + 1] = convertedData;
|
||||
task.finishList[index].status = 'is-success';
|
||||
task.finishNum++;
|
||||
|
||||
this.options.onProgress?.(task);
|
||||
|
||||
if (task.finishNum === task.rangeDownload.targetSegment) {
|
||||
task.status = 'done';
|
||||
this.downloadFile(task);
|
||||
this.options.onComplete?.(task);
|
||||
}
|
||||
|
||||
callback();
|
||||
});
|
||||
} else {
|
||||
task.mediaFileList[index - task.rangeDownload.startSegment + 1] = data;
|
||||
task.finishList[index].status = 'is-success';
|
||||
task.finishNum++;
|
||||
|
||||
this.options.onProgress?.(task);
|
||||
|
||||
if (task.finishNum === task.rangeDownload.targetSegment) {
|
||||
task.status = 'done';
|
||||
this.downloadFile(task);
|
||||
this.options.onComplete?.(task);
|
||||
}
|
||||
|
||||
callback();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 下载文件
|
||||
*/
|
||||
private downloadFile(task: M3U8DownloadTask): void {
|
||||
const fileBlob = new Blob(task.mediaFileList, {
|
||||
type: task.type === 'MP4' ? 'video/mp4' : 'video/MP2T',
|
||||
});
|
||||
|
||||
const a = document.createElement('a');
|
||||
a.href = URL.createObjectURL(fileBlob);
|
||||
a.download = `${task.title}.${task.type === 'MP4' ? 'mp4' : 'ts'}`;
|
||||
a.style.display = 'none';
|
||||
document.body.appendChild(a);
|
||||
a.click();
|
||||
document.body.removeChild(a);
|
||||
URL.revokeObjectURL(a.href);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取 M3U8 文件
|
||||
*/
|
||||
private async fetchM3U8(url: string): Promise<string> {
|
||||
return new Promise((resolve, reject) => {
|
||||
const xhr = new XMLHttpRequest();
|
||||
xhr.onreadystatechange = () => {
|
||||
if (xhr.readyState === 4) {
|
||||
if (xhr.status >= 200 && xhr.status < 300) {
|
||||
resolve(xhr.responseText);
|
||||
} else {
|
||||
reject(new Error(`HTTP ${xhr.status}`));
|
||||
}
|
||||
}
|
||||
};
|
||||
xhr.open('GET', url, true);
|
||||
xhr.send();
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* 检测是否是主播放列表
|
||||
*/
|
||||
private isMasterPlaylist(m3u8Str: string): boolean {
|
||||
return m3u8Str.includes('#EXT-X-STREAM-INF');
|
||||
}
|
||||
|
||||
/**
|
||||
* 解析流信息
|
||||
*/
|
||||
private parseStreamInfo(m3u8Str: string, baseUrl: string): Array<{
|
||||
url: string;
|
||||
bandwidth: number;
|
||||
resolution: string;
|
||||
name: string;
|
||||
}> {
|
||||
const streams: Array<{
|
||||
url: string;
|
||||
bandwidth: number;
|
||||
resolution: string;
|
||||
name: string;
|
||||
}> = [];
|
||||
const lines = m3u8Str.split('\n');
|
||||
|
||||
for (let i = 0; i < lines.length; i++) {
|
||||
const line = lines[i].trim();
|
||||
if (line.startsWith('#EXT-X-STREAM-INF:')) {
|
||||
const bandwidth = line.match(/BANDWIDTH=(\d+)/)?.[1] || '';
|
||||
const resolution = line.match(/RESOLUTION=([^\s,]+)/)?.[1] || '';
|
||||
const name = line.match(/NAME="([^"]+)"/)?.[1] || '';
|
||||
|
||||
if (i + 1 < lines.length) {
|
||||
const url = lines[i + 1].trim();
|
||||
if (url && !url.startsWith('#')) {
|
||||
streams.push({
|
||||
url: this.applyURL(url, baseUrl),
|
||||
bandwidth: parseInt(bandwidth) || 0,
|
||||
resolution: resolution || 'Unknown',
|
||||
name: name || `${resolution || ''} ${bandwidth ? parseInt(bandwidth) / 1000 + 'kbps' : 'Unknown'}`,
|
||||
});
|
||||
i++;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
streams.sort((a, b) => b.bandwidth - a.bandwidth);
|
||||
return streams;
|
||||
}
|
||||
|
||||
/**
|
||||
* 合成 URL
|
||||
*/
|
||||
private applyURL(targetURL: string, baseURL: string): string {
|
||||
if (targetURL.indexOf('http') === 0) {
|
||||
return targetURL;
|
||||
} else if (targetURL[0] === '/') {
|
||||
const domain = baseURL.split('/');
|
||||
return domain[0] + '//' + domain[2] + targetURL;
|
||||
} else {
|
||||
const domain = baseURL.split('/');
|
||||
domain.pop();
|
||||
return domain.join('/') + '/' + targetURL;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 解析 IV
|
||||
*/
|
||||
private parseIV(ivString: string): Uint8Array {
|
||||
const hex = ivString.replace(/^0x/, '');
|
||||
return new Uint8Array(hex.match(/.{1,2}/g)!.map((byte) => parseInt(byte, 16)));
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取 AES 密钥
|
||||
*/
|
||||
private async getAESKey(task: M3U8DownloadTask): Promise<void> {
|
||||
return new Promise((resolve, reject) => {
|
||||
const xhr = new XMLHttpRequest();
|
||||
xhr.responseType = 'arraybuffer';
|
||||
xhr.onreadystatechange = () => {
|
||||
if (xhr.readyState === 4) {
|
||||
if (xhr.status >= 200 && xhr.status < 300) {
|
||||
task.aesConf.key = xhr.response;
|
||||
// 初始化 AES 解密器
|
||||
task.aesConf.decryption = new AESDecryptor();
|
||||
task.aesConf.decryption.expandKey(task.aesConf.key);
|
||||
resolve();
|
||||
} else {
|
||||
reject(new Error('获取 AES 密钥失败'));
|
||||
}
|
||||
}
|
||||
};
|
||||
xhr.open('GET', task.aesConf.uri, true);
|
||||
xhr.send();
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* AES 解密
|
||||
*/
|
||||
private aesDecrypt(task: M3U8DownloadTask, data: ArrayBuffer, index: number): ArrayBuffer {
|
||||
if (!task.aesConf.decryption) {
|
||||
return data;
|
||||
}
|
||||
|
||||
// 使用 IV 或默认 IV
|
||||
let iv: Uint8Array;
|
||||
if (task.aesConf.iv) {
|
||||
iv = task.aesConf.iv;
|
||||
} else {
|
||||
// 如果没有指定 IV，使用片段索引作为 IV
|
||||
iv = new Uint8Array(16);
|
||||
for (let i = 12; i < 16; i++) {
|
||||
iv[i] = (index >> (8 * (15 - i))) & 0xff;
|
||||
}
|
||||
}
|
||||
|
||||
try {
|
||||
return task.aesConf.decryption.decrypt(data, 0, iv.buffer, true);
|
||||
} catch (error) {
|
||||
console.error('AES 解密失败:', error);
|
||||
return data;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* MP4 转码
|
||||
*/
|
||||
private conversionMp4(
|
||||
task: M3U8DownloadTask,
|
||||
data: ArrayBuffer,
|
||||
index: number,
|
||||
callback: (data: ArrayBuffer) => void
|
||||
): void {
|
||||
if (task.type === 'MP4') {
|
||||
try {
|
||||
// @ts-ignore - mux.js 的 Transmuxer 在 mp4 子模块下
|
||||
const transMuxer = new muxjs.mp4.Transmuxer({
|
||||
keepOriginalTimestamps: true,
|
||||
duration: parseInt(task.durationSecond.toString()),
|
||||
});
|
||||
|
||||
transMuxer.on('data', (segment: any) => {
|
||||
// 第一个片段需要包含初始化段
|
||||
if (index === task.rangeDownload.startSegment - 1) {
|
||||
const combinedData = new Uint8Array(
|
||||
segment.initSegment.byteLength + segment.data.byteLength
|
||||
);
|
||||
combinedData.set(segment.initSegment, 0);
|
||||
combinedData.set(segment.data, segment.initSegment.byteLength);
|
||||
callback(combinedData.buffer);
|
||||
} else {
|
||||
callback(segment.data);
|
||||
}
|
||||
});
|
||||
|
||||
transMuxer.push(new Uint8Array(data));
|
||||
transMuxer.flush();
|
||||
} catch (error) {
|
||||
console.error('MP4 转码失败:', error);
|
||||
// 转码失败，返回原始数据
|
||||
callback(data);
|
||||
}
|
||||
} else {
|
||||
// TS 格式直接返回
|
||||
callback(data);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 终止请求
|
||||
*/
|
||||
private abortRequests(task: M3U8DownloadTask): void {
|
||||
task.requests.forEach((xhr) => {
|
||||
if (xhr.readyState !== 4) {
|
||||
xhr.abort();
|
||||
}
|
||||
});
|
||||
task.requests = [];
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user