private void SetMaxValue()
{
lock (TotalSizeTemp)
{
SetMaxValue(TotalSizeTemp.Sum(ab => ab.Value) * TotalCount / DownCount);
}
//try
//{
//}
//catch
//{ }
}
/// <summary>
/// M3u8下载
/// </summary>
/// <param name="threads">线程数</param>
/// <param name="savePath">保存文件</param>
/// <param name="workPath">工作目录</param>
/// <param name="errmsg"></param>
/// <returns></returns>
bool DownUrlByM3u8(int threads, string savePath, string workPath, out string errmsg)
{
//解析M3u8
Parser parser = new Parser
{
DownName = _Name,
M3u8Url = uri.AbsoluteUri,
DownDir = workPath
};
string Err;
if (parser.Parse(out Err)) //开始解析
{
if (File.Exists(parser.jsonSavePath))
{
JObject initJson = JObject.Parse(File.ReadAllText(parser.jsonSavePath));
bool isVOD = Convert.ToBoolean(initJson["m3u8Info"]["vod"].ToString());
JArray parts = JArray.Parse(initJson["m3u8Info"]["segments"].ToString()); //大分组
string segCount = initJson["m3u8Info"]["count"].ToString();
string oriCount = initJson["m3u8Info"]["originalCount"].ToString(); //原始分片数量
int total = Convert.ToInt32(segCount);
int PartsCount = parts.Count;
string segsPadZero = string.Empty.PadRight(oriCount.Length, '0');
string partsPadZero = string.Empty.PadRight(Convert.ToString(parts.Count).Length, '0');
TotalCount = total;
//点播
if (isVOD)
{
List <FilesResult> files = new List <FilesResult>();
workPath.CreateDirectory(true);//新建文件夹
#region 任务分配
//构造包含所有分片的新的segments
JArray segments = new JArray();
for (int i = 0; i < parts.Count; i++)
{
var tmp = JArray.Parse(parts[i].ToString());
for (int j = 0; j < tmp.Count; j++)
{
JObject t = (JObject)tmp[j];
t.Add("part", i);
segments.Add(t);
}
}
bool isVTT = false;
for (int i = 0; i < segments.Count; i++)
{
JToken info = segments[i];
string TsUrl = info["segUri"].Value <string>();
//VTT字幕
if (isVTT == false && TsUrl.Trim('\"').EndsWith(".vtt"))
{
isVTT = true;
}
string Method = info["method"].Value <string>();
string Iv = null;
long StartByte = 0, ExpectByte = 0;
if (Method != "NONE")
{
Iv = info["iv"].Value <string>();
}
try
{
ExpectByte = info["expectByte"].Value <long>();
}
catch { }
try
{
StartByte = info["startByte"].Value <long>();
}
catch { }
string Dir = workPath + "Part_" + info["part"].Value <int>().ToString(partsPadZero) + "\\";
Dir.CreateDirectory(true);
string filename_temp = info["index"].Value <int>().ToString(segsPadZero) + ".ts";
files.Add(new FilesResult
{
index = i,
tsurl = TsUrl,
path = Dir + filename_temp,
start_position = StartByte,
end_position = ExpectByte,
});
lock (DownSizeTemp)
{
DownSizeTemp.Add(i, 0);
}
}
#endregion
SetState(DownState.Downloading);
TestTime(true);
bool isComplete = false;
Action _action = () =>
{
foreach (FilesResult item in files)
{
lock (files)
{
//int counts = files.Count(ab => ab.run);
if (files.Count(ab => ab.run) > threads)
{
resetEvents.Reset();//暂停
}
}
resetEvents.WaitOne();
double _downvalueTemp = 0;
item.run = item.runTask = true;
Action _actions = () =>
{
if (tokenSource.Token.IsCancellationRequested)
{
return;
}
//阻止当前线程
resetEvent.WaitOne();
Uri TSuri = new Uri(item.tsurl);
int ErrCount = 0;
while (item.runTask)
{
try
{
HttpWebRequest request = (HttpWebRequest)WebRequest.Create(TSuri);
request.Host = TSuri.Host;
request.Accept = "*/*";
request.UserAgent = "Mozilla/5.0";
request.Method = "GET";
if (SystemSettings.DownloadTimeOut > 0)
{
request.Timeout = SystemSettings.DownloadTimeOut;
}
request.ReadWriteTimeout = request.Timeout; //重要
request.AllowAutoRedirect = true;
request.KeepAlive = false;
using (HttpWebResponse p = (HttpWebResponse)request.GetResponse())
{
lock (TotalSizeTemp)
{
if (!TotalSizeTemp.ContainsKey(item.index))
{
using (FileStream file = new FileStream(item.path, FileMode.OpenOrCreate))
{
long fileLong = p.ContentLength;
if (fileLong > 0 && file.Length >= fileLong)
{
_downvalueTemp += fileLong;
lock (DownSizeTemp)
{
DownSizeTemp[item.index] = fileLong;
}
TotalSizeTemp.Add(item.index, fileLong);
DownCount++;
SetValue();
return;
}
}
}
}
using (FileStream file = new FileStream(item.path, FileMode.Create))
{
if (item.start_position > 0 && item.end_position > 0)
{
request.AddRange((long)item.start_position, (long)item.end_position);
}
lock (TotalSizeTemp)
{
if (!TotalSizeTemp.ContainsKey(item.index))
{
TotalSizeTemp.Add(item.index, p.ContentLength);
}
}
using (Stream stream = p.GetResponseStream())
{
byte[] _cache = new byte[SystemSettings.DownloadCacheCount];
int osize = stream.Read(_cache, 0, _cache.Length);
bool isRun = true;
while (isRun)
{
if (osize > 0)
{
if (tokenSource.Token.IsCancellationRequested)
{
return;
}
//阻止当前线程
resetEvent.WaitOne();
_downvalueTemp += osize;
lock (DownSizeTemp)
{
DownSizeTemp[item.index] = _downvalueTemp;
}
SetValue();
file.Write(_cache, 0, osize);
osize = stream.Read(_cache, 0, _cache.Length);
}
else
{
isRun = false;
}
}
}
}
}
item.complete = true;
DownCount++;
item.runTask = false;
}
catch
{
if (isStop)
{
item.runTask = false;
return;
}
ErrCount++;
_downvalueTemp = 0;
lock (DownSizeTemp)
{
DownSizeTemp[item.index] = 0;
}
SetValue();
//System.Diagnostics.Debug.WriteLine("序号：" + item.index + "|次数：" + ErrCount);
if (ErrCount > SystemSettings.DownloadRetryCount)
{
item.runTask = false;
}
}
}
};
var task = Task.Run(_actions, tokenSource.Token).ContinueWith(t =>
{
item.run = false;
lock (files)
{
int tackCount = files.Count(ab => ab.run);
if (tackCount < threads)
{
resetEvents.Set();//继续
}
}
}, TaskContinuationOptions.OnlyOnRanToCompletion | TaskContinuationOptions.ExecuteSynchronously);
ParallelTasks.Add(task);
resetEvents.WaitOne();
}
_task.ContinueWhenAll(ParallelTasks.ToArray(), (action =>
{
isdown = false;
if (tokenSource.Token.IsCancellationRequested)
{
return;
}
List <string> _files = new List <string>();
lock (files)
{
foreach (FilesResult item in files)
{
if (!item.complete)
{
return;
}
_files.Add(item.path);
}
}
bool externalAudio = false, externalSub = false;
string externalAudioUrl = null, externalSubUrl = null;
try
{
if (initJson["m3u8Info"]["audio"].ToString() != "")
{
externalAudio = true;
}
externalAudioUrl = initJson["m3u8Info"]["audio"].ToString();
System.Diagnostics.Debug.WriteLine("识别到外挂音频轨道");
}
catch { }
try
{
if (initJson["m3u8Info"]["sub"].ToString() != "")
{
externalSub = true;
}
externalSubUrl = initJson["m3u8Info"]["sub"].ToString();
System.Diagnostics.Debug.WriteLine("识别到外挂字幕轨道");
}
catch { }
CombineMultipleFilesByM3u8(savePath, PartsCount, workPath, partsPadZero, externalAudio, externalAudioUrl, externalSub, externalSubUrl);
SetMaxValue();
while (isTimeTask)
{
Thread.Sleep(500);
}
//Thread.Sleep(500);
Directory.Delete(workPath, true);
isComplete = true;
SetState(DownState.Complete);
})).Wait();
};
_task.StartNew(_action).Wait();
if (isComplete)
{
errmsg = null;
return(true);
}
else
{
SetState(DownState.Fail);
errmsg = "文件下载失败";
return(false);
}
}
else
{
SetState(DownState.Fail);
errmsg = "无法下载直播";
return(false);
}
}
else
{
SetState(DownState.Fail);
errmsg = "M3u8解析失败:地址无效";
return(false);
}
}
else
{
SetState(DownState.Fail);
errmsg = "M3u8解析失败";
return(false);
}
}
