ModNet.vb 99 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082
  1. Imports System.Net.Http.Headers
  2. Imports System.Net.Sockets
  3. Imports System.Runtime.CompilerServices
  4. Imports System.Threading.Tasks
  5. Public Module ModNet
  6. #Region "网络请求"
  7. ''' <summary>
  8. ''' 发送一次网络请求并获取返回内容。
  9. ''' </summary>
  10. Public Function NetRequestByClient(Url As String, Optional Method As HttpMethod = Nothing,
  11. Optional Content As Object = Nothing, Optional ContentType As String = Nothing, Optional Accept As String = "*/*",
  12. Optional Timeout As Integer = 25000, Optional Headers As String(,) = Nothing, Optional RequireJson As Boolean = False,
  13. Optional Encoding As Encoding = Nothing, Optional SimulateBrowserHeaders As Boolean = False, Optional MakeLog As Boolean = True) As String
  14. If Method Is Nothing Then Method = HttpMethod.Get
  15. If MakeLog Then Log("[Net] 发起网络请求(" & Method.Method & "," & Url & "),最大超时 " & Timeout)
  16. Try
  17. Dim HeaderDictionary = If(Headers, {}).ToDictionary
  18. If ContentType IsNot Nothing Then HeaderDictionary("Content-Type") = ContentType
  19. If Accept IsNot Nothing Then HeaderDictionary("Accept") = Accept
  20. Dim Result = If(Encoding, Encoding.UTF8).GetString(SendRequest(Url, Method, Content, HeaderDictionary, Timeout:=Timeout, Encoding:=Encoding, SimulateBrowserHeaders:=SimulateBrowserHeaders))
  21. '检查结果是否为完整 JSON(在 #6683 中可能返回空,在被 GFW 截断时可能不完整)
  22. '在此处检查并报错,可以让 NetRequestByClientRetry 等方法进行重试
  23. If RequireJson Then
  24. Result = Result.Trim((vbCrLf & " ").ToCharArray) 'Authlib-Injector 的结果会在末尾附带一个空行
  25. If Not (Result.StartsWithF("{") AndAlso Result.EndsWithF("}")) AndAlso
  26. Not (Result.StartsWithF("[") AndAlso Result.EndsWithF("]")) Then
  27. Throw New FormatException("返回结果并非 JSON 格式:" &
  28. If(Result.Length > 2000, Result.Substring(0, 1000) & "..." & Result.Substring(Result.Length - 1000), Result))
  29. End If
  30. End If
  31. Return Result
  32. Catch ex As FormatException
  33. Throw
  34. Catch ex As ThreadInterruptedException
  35. Throw
  36. Catch ex As Exception
  37. If MakeLog Then Log(ex, "网络请求失败", LogLevel.Developer)
  38. Throw
  39. End Try
  40. End Function
  41. ''' <summary>
  42. ''' 发送网络请求并获取返回内容。
  43. ''' 会进行至多 45 秒 3 次的尝试,允许最长 30s 的超时。
  44. ''' </summary>
  45. ''' <param name="BackupUrl">如果第一次尝试失败,换用的备用 URL。</param>
  46. Public Function NetRequestByClientRetry(Url As String, Optional Method As HttpMethod = Nothing, Optional BackupUrl As String = Nothing,
  47. Optional Content As Object = Nothing, Optional ContentType As String = Nothing, Optional Accept As String = "*/*",
  48. Optional Headers As String(,) = Nothing, Optional RequireJson As Boolean = False,
  49. Optional Encoding As Encoding = Nothing, Optional SimulateBrowserHeaders As Boolean = False) As String
  50. Dim RetryCount As Integer = 0
  51. Dim RetryException As Exception = Nothing
  52. Dim StartTime As Long = GetTimeMs()
  53. Try
  54. Retry:
  55. Select Case RetryCount
  56. Case 0 '正常尝试
  57. Return NetRequestByClient(Url, Method, Content, ContentType, Accept, 10000, Headers, RequireJson, Encoding, SimulateBrowserHeaders)
  58. Case 1 '慢速重试
  59. Thread.Sleep(500)
  60. Return NetRequestByClient(Url, Method, Content, ContentType, Accept, 30000, Headers, RequireJson, Encoding, SimulateBrowserHeaders)
  61. Case Else '快速重试
  62. If GetTimeMs() - StartTime > 5500 Then
  63. '若前两次加载耗费 5 秒以上,才进行重试
  64. Thread.Sleep(500)
  65. Return NetRequestByClient(Url, Method, Content, ContentType, Accept, 4000, Headers, RequireJson, Encoding, SimulateBrowserHeaders)
  66. Else
  67. Throw RetryException
  68. End If
  69. End Select
  70. Catch ex As ThreadInterruptedException
  71. Throw
  72. Catch ex As Exception
  73. If TypeOf ex Is ResponsedWebException Then
  74. If CType(ex, ResponsedWebException).StatusCode = HttpStatusCode.Forbidden Then Throw
  75. If CType(ex, ResponsedWebException).StatusCode = 429 Then Thread.Sleep(10000) 'Too Many Requests
  76. End If
  77. '重试
  78. Select Case RetryCount
  79. Case 0
  80. RetryException = ex
  81. RetryCount += 1
  82. GoTo Retry
  83. Case 1
  84. RetryCount += 1
  85. GoTo Retry
  86. Case Else
  87. Throw
  88. End Select
  89. End Try
  90. End Function
  91. ''' <summary>
  92. ''' 生成数个线程,同时发送网络请求并获取返回内容。
  93. ''' 每个线程开始之前会略有延迟。
  94. ''' </summary>
  95. Public Function NetRequestByClientMultiple(Url As String, Optional Method As HttpMethod = Nothing, Optional ThreadCount As Integer = 3,
  96. Optional Content As Object = Nothing, Optional ContentType As String = Nothing, Optional Accept As String = "*/*",
  97. Optional Headers As String(,) = Nothing, Optional RequireJson As Boolean = False, Optional Timeout As Integer = 25000,
  98. Optional Encoding As Encoding = Nothing, Optional SimulateBrowserHeaders As Boolean = False) As String
  99. Dim Threads As New List(Of Thread)
  100. Dim RequestResult = Nothing
  101. Dim RequestEx As Exception = Nothing
  102. Dim FailCount As Integer = 0
  103. For i = 1 To ThreadCount
  104. Dim th As New Thread(
  105. Sub()
  106. Try
  107. RequestResult = NetRequestByClient(Url, Method, Content, ContentType, Accept, Timeout, Headers, RequireJson, Encoding, SimulateBrowserHeaders)
  108. Catch ex As Exception
  109. FailCount += 1
  110. RequestEx = ex
  111. End Try
  112. End Sub)
  113. th.Start()
  114. Threads.Add(th)
  115. Thread.Sleep(i * 250)
  116. If RequestResult IsNot Nothing Then GoTo RequestFinished
  117. Next
  118. Do While True
  119. If RequestResult IsNot Nothing Then
  120. RequestFinished:
  121. Try
  122. For Each th In Threads
  123. If th.IsAlive Then th.Interrupt()
  124. Next
  125. Catch
  126. End Try
  127. Return RequestResult
  128. ElseIf FailCount = ThreadCount Then
  129. Try
  130. For Each th In Threads
  131. If th.IsAlive Then th.Interrupt()
  132. Next
  133. Catch
  134. End Try
  135. Throw RequestEx
  136. End If
  137. Thread.Sleep(20)
  138. Loop
  139. Return Nothing
  140. End Function
  141. ''' <summary>
  142. ''' 以多线程下载网页文件的方式获取内容。
  143. ''' 不支持缓存协商;若需缓存协商,需要换用 NetRequestByClient。
  144. ''' </summary>
  145. Public Function NetRequestByLoader(Url As String, Optional Timeout As Integer = 45000, Optional IsJson As Boolean = False, Optional SimulateBrowserHeaders As Boolean = False) As String
  146. Return NetRequestByLoader({Url}, Timeout, IsJson, SimulateBrowserHeaders)
  147. End Function
  148. ''' <summary>
  149. ''' 以多线程下载网页文件的方式获取内容。
  150. ''' 不支持缓存协商;若需缓存协商,需要换用 NetRequestByClient。
  151. ''' </summary>
  152. Public Function NetRequestByLoader(Urls As IEnumerable(Of String), Optional Timeout As Integer = 45000, Optional IsJson As Boolean = False, Optional SimulateBrowserHeaders As Boolean = False) As String
  153. Dim Temp As String = RequestTaskTempFolder() & "download.txt"
  154. Dim NewTask As New LoaderDownload("源码获取 " & GetUuid() & "#", New List(Of NetFile) From {New NetFile(Urls, Temp, New FileChecker With {.IsJson = IsJson}, SimulateBrowserHeaders)})
  155. Try
  156. NewTask.WaitForExitTime(Timeout, TimeoutMessage:="连接服务器超时(第一下载源:" & Urls.First & ")")
  157. NetRequestByLoader = ReadFile(Temp)
  158. File.Delete(Temp)
  159. Finally
  160. NewTask.Abort()
  161. End Try
  162. End Function
  163. #End Region
  164. #Region "文件下载"
  165. ''' <summary>
  166. ''' 通过网络请求直接下载小文件,若文件已存在将被覆盖。
  167. ''' 不建议用于下载大文件。
  168. ''' </summary>
  169. Public Sub NetDownloadByClient(Url As String, LocalPath As String, Optional SimulateBrowserHeaders As Boolean = False)
  170. Log($"[Net] 通过网络请求直接下载小文件:{Url} → {LocalPath}")
  171. '初始化文件路径
  172. Try
  173. Directory.CreateDirectory(GetPathFromFullPath(LocalPath))
  174. File.Delete(LocalPath)
  175. Catch ex As Exception
  176. Throw New Exception($"预处理下载文件路径失败({LocalPath})", ex)
  177. End Try
  178. '下载
  179. Try
  180. File.WriteAllBytes(LocalPath, SendRequest(Url, HttpMethod.Get, SimulateBrowserHeaders:=SimulateBrowserHeaders))
  181. Catch ex As Exception
  182. File.Delete(LocalPath)
  183. Throw New WebException($"通过网络请求直接下载小文件失败({LocalPath})", ex)
  184. End Try
  185. End Sub
  186. ''' <summary>
  187. ''' 通过多线程下载引擎下载文件。
  188. ''' </summary>
  189. Public Sub NetDownloadByLoader(Url As String, LocalPath As String, Optional LoaderToSyncProgress As LoaderBase = Nothing, Optional Check As FileChecker = Nothing, Optional SimulateBrowserHeaders As Boolean = False)
  190. NetDownloadByLoader({Url}, LocalPath, LoaderToSyncProgress, Check, SimulateBrowserHeaders)
  191. End Sub
  192. ''' <summary>
  193. ''' 通过多线程下载引擎下载文件。
  194. ''' </summary>
  195. ''' <param name="Urls">文件的 Url 列表。</param>
  196. ''' <param name="LocalPath">下载的本地地址。</param>
  197. Public Sub NetDownloadByLoader(Urls As IEnumerable(Of String), LocalPath As String, Optional LoaderToSyncProgress As LoaderBase = Nothing, Optional Check As FileChecker = Nothing, Optional SimulateBrowserHeaders As Boolean = False)
  198. Dim NewTask As New LoaderDownload($"文件下载 {GetUuid()}#", New List(Of NetFile) From {New NetFile(Urls, LocalPath, Check, SimulateBrowserHeaders)})
  199. Try
  200. NewTask.WaitForExit(LoaderToSyncProgress:=LoaderToSyncProgress)
  201. Catch ex As Exception
  202. Throw New WebException($"多线程直接下载文件失败(第一下载源:" & Urls.First() & ")", ex)
  203. Finally
  204. NewTask.Abort()
  205. End Try
  206. End Sub
  207. #End Region
  208. #Region "基础请求引擎"
  209. '基础请求函数
  210. Private Function SendRequest(Url As String, Method As HttpMethod,
  211. Optional Content As Object = Nothing, Optional Headers As Dictionary(Of String, String) = Nothing,
  212. Optional SimulateBrowserHeaders As Boolean = False, Optional Timeout As Integer = 25000, Optional Encoding As Encoding = Nothing) As Byte()
  213. If RunInUi() AndAlso Not Url.Contains("//127.") Then Throw New Exception("在 UI 线程执行了网络请求")
  214. Dim Request As HttpRequestMessage = Nothing, CancelToken As CancellationTokenSource = Nothing,
  215. Response As HttpResponseMessage = Nothing, HostIp As String = Nothing
  216. Try
  217. Url = SecretCdnSign(Url)
  218. Request = New HttpRequestMessage(Method, Url)
  219. '写入 Content
  220. If Content IsNot Nothing AndAlso Request.Method <> HttpMethod.Get AndAlso Request.Method <> HttpMethod.Head Then
  221. If TypeOf Content Is HttpContent Then
  222. Request.Content = DirectCast(Content, HttpContent)
  223. ElseIf TypeOf Content Is Byte() Then
  224. Request.Content = New ByteArrayContent(DirectCast(Content, Byte()))
  225. ElseIf Content IsNot Nothing Then
  226. Request.Content = New ByteArrayContent(If(Encoding, Encoding.UTF8).GetBytes(Content.ToString()))
  227. End If
  228. End If
  229. '写入 Headers
  230. For Each Header In If(Headers, New Dictionary(Of String, String))
  231. If Header.Key.ToLower = "content-type" Then
  232. Request.Content?.Headers.TryAddWithoutValidation(Header.Key, Header.Value)
  233. Else
  234. Request.Headers.TryAddWithoutValidation(Header.Key, Header.Value)
  235. End If
  236. Next
  237. SecretHeadersSign(Url, Request, SimulateBrowserHeaders)
  238. 'DNS 解析
  239. CancelToken = New CancellationTokenSource(Timeout)
  240. HostIp = DNSLookup(Request, CancelToken)
  241. If HostIp IsNot Nothing AndAlso Not IPReliability.ContainsKey(HostIp) Then
  242. IPReliability(HostIp) = -0.01 '预先降低一点,这样快速的重复请求会使用不同的 IP 以提高成功率
  243. End If
  244. '发送请求
  245. SyncLock RequestClientLock
  246. '延迟初始化,以避免在程序启动前加载 CacheCow 导致 DLL 加载失败
  247. If RequestClient Is Nothing Then
  248. RequestClient = CacheCow.Client.ClientExtensions.CreateClient(New CacheCow.Client.FileCacheStore.FileStore(PathTemp & "Cache/Http/"), New HttpClientHandler With {
  249. .AutomaticDecompression = DecompressionMethods.Deflate Or DecompressionMethods.GZip,
  250. .UseCookies = False '不设为 False 就不能从 Header 手动传入 Cookies
  251. })
  252. End If
  253. End SyncLock
  254. Response = RequestClient.SendAsync(Request, HttpCompletionOption.ResponseHeadersRead, CancelToken.Token).GetResultWithTimeout(CancelToken, Timeout)
  255. Dim ResponseStream = Response.Content.ReadAsStreamAsync().GetAwaiter().GetResult()
  256. Dim ResponseBytes As Byte()
  257. Using Stream As New MemoryStream
  258. ResponseStream.CopyToAsync(Stream, 81920, CancelToken.Token).GetResultWithTimeout(CancelToken, Timeout)
  259. ResponseBytes = Stream.ToArray()
  260. End Using
  261. '输出
  262. If Response.IsSuccessStatusCode Then
  263. RecordIPReliability(HostIp, 0.5)
  264. Return ResponseBytes
  265. Else
  266. RecordIPReliability(HostIp, -0.7)
  267. Dim ResponseMessage = If(Encoding, Encoding.UTF8).GetString(ResponseBytes)
  268. Throw New ResponsedWebException(
  269. $"错误码 {Response.StatusCode} ({CInt(Response.StatusCode)}),{Method},{Url},{HostIp}" &
  270. If(String.IsNullOrEmpty(ResponseMessage), "", vbCrLf & ResponseMessage), Response.StatusCode, ResponseMessage)
  271. End If
  272. Catch ex As ThreadInterruptedException
  273. Throw
  274. Catch ex As ResponsedWebException
  275. Throw
  276. Catch ex As Exception
  277. RecordIPReliability(HostIp, -1)
  278. If TypeOf ex Is OperationCanceledException OrElse TypeOf ex Is TimeoutException Then 'CancellationToken 超时
  279. Throw New WebException($"连接服务器超时,请稍后再试,或使用 VPN 改善网络环境({Method}, {Url},IP:{HostIp})", WebExceptionStatus.Timeout)
  280. ElseIf ex.IsNetworkRelated Then
  281. Throw New WebException($"网络请求失败,请稍后再试,或使用 VPN 改善网络环境({Method}, {Url},IP:{HostIp})", WebExceptionStatus.Timeout)
  282. Else
  283. Throw New Exception($"网络请求出现意外异常({Method}, {Url},{HostIp})", ex)
  284. End If
  285. Finally
  286. Request?.Dispose()
  287. CancelToken?.Dispose()
  288. Response?.Dispose()
  289. End Try
  290. End Function
  291. Private RequestClient As HttpClient = Nothing
  292. Private RequestClientLock As New Object
  293. ''' <summary>
  294. ''' 进行 DNS 解析。它仅在选择的 IP 与系统默认的不一致时才对 URL 中的 Host 进行替换。
  295. ''' 返回请求时应使用的 IP;若为 IPv6,则加上了方括号。
  296. ''' 若解析失败,则返回 Nothing。
  297. ''' </summary>
  298. Private Function DNSLookup(Request As HttpRequestMessage, CancelToken As CancellationTokenSource) As String
  299. '若一分钟内已失败过,则不再重复尝试解析,以减少断网时的 Exception 数量
  300. Dim Host = Request.RequestUri.Host
  301. SyncLock DNSFailureRecord.SyncRoot
  302. If DNSFailureRecord.ContainsKey(Host) Then
  303. If Date.Now - DNSFailureRecord(Host) < New TimeSpan(0, 1, 0) Then
  304. Return Nothing
  305. Else
  306. DNSFailureRecord.Remove(Host)
  307. End If
  308. End If
  309. End SyncLock
  310. '初步 DNS 解析
  311. Dim DnsTask As Task(Of IPAddress()) = Nothing
  312. Try
  313. DnsTask = Dns.GetHostAddressesAsync(Host)
  314. DnsTask.Wait(CancelToken.Token)
  315. Catch ex As Exception
  316. Log(ex, $"DNS 解析失败({Host})")
  317. DNSFailureRecord(Host) = Date.Now
  318. Return Nothing
  319. End Try
  320. Dim Candidates As IPAddress() = DnsTask.Result.ToArray
  321. If Not Candidates.Any Then
  322. Log($"[Net] DNS 解析无结果({Host})")
  323. Return Nothing
  324. End If
  325. '若同时存在 IPv4 和 IPv6 地址,仅选择其中一类(因为 GFW 可能只屏蔽了 IPv4 或 IPv6)
  326. Dim GetIPReliability = Function(Key) IPReliability.GetOrDefault(Key.ToString, 0)
  327. Dim IPv4Targets = Candidates.Where(Function(i) i.AddressFamily = AddressFamily.InterNetwork).ToArray
  328. Dim IPv6Targets = Candidates.Where(Function(i) i.AddressFamily = AddressFamily.InterNetworkV6).ToArray
  329. If IPv4Targets.Any AndAlso IPv6Targets.Any Then
  330. Dim IPv4Reliability = IPv4Targets.Average(Function(i) GetIPReliability(i))
  331. Dim IPv6Reliability = IPv6Targets.Average(Function(i) GetIPReliability(i))
  332. If Host = "api.modrinth.com" Then IPv6Reliability -= 0.1 '让 Modrinth 优先使用 IPv4 地址(#6887)
  333. Candidates = If(IPv4Reliability >= IPv6Reliability, IPv4Targets, IPv6Targets)
  334. End If
  335. '选择可靠度最高的 IP
  336. Dim MaxReliability As Double = Candidates.Max(Function(i) GetIPReliability(i))
  337. Dim Target As IPAddress = Candidates.First(Function(i) GetIPReliability(i) = MaxReliability)
  338. Dim TargetIp As String = If(Target.AddressFamily = AddressFamily.InterNetworkV6, $"[{Target}]", Target.ToString)
  339. If Target IsNot Candidates.First Then
  340. '优选结果与系统默认 IP 不一致,替换域名并设置 Host 头
  341. If ModeDebug Then Log($"[Net] DNS 解析结果:{Host} → {Target},所有可能的 IP 与可靠度:{DnsTask.Result.Select(Function(i) $"{i} → {GetIPReliability(i):0.000}").Join(";")}")
  342. Request.Headers.Host = Host
  343. Dim Builder As New UriBuilder(Request.RequestUri)
  344. Builder.Host = TargetIp
  345. Request.RequestUri = Builder.Uri
  346. End If
  347. Return TargetIp
  348. End Function
  349. Private DNSFailureRecord As New SafeDictionary(Of String, Date)
  350. ''' <summary>
  351. ''' 记录每个 IP 地址的请求可靠度。
  352. ''' 通常取值 -1 ~ +0.5,越高越好。未尝试过的 IP 应视为 0。
  353. ''' </summary>
  354. Private IPReliability As New SafeDictionary(Of String, Double)
  355. ''' <summary>
  356. ''' 根据请求结果,记录 IP 地址的可靠度。
  357. ''' </summary>
  358. Private Sub RecordIPReliability(IP As String, Result As Double)
  359. If IP Is Nothing Then Return
  360. If Not IPReliability.ContainsKey(IP) Then IPReliability(IP) = 0
  361. IPReliability(IP) = (IPReliability(IP) + Result) / 2
  362. End Sub
  363. ''' <summary>
  364. ''' 当 HTTP 状态码不指示成功时引发的异常。
  365. ''' 附带额外属性,可用于获取远程服务器给予的回复以及 HTTP 状态码。
  366. ''' </summary>
  367. Public Class ResponsedWebException
  368. Inherits WebException
  369. ''' <summary>
  370. ''' HTTP 状态码。
  371. ''' </summary>
  372. Public StatusCode As HttpStatusCode
  373. ''' <summary>
  374. ''' 远程服务器给予的回复。
  375. ''' </summary>
  376. Public Overloads Property Response As String
  377. Public Sub New(Message As String, StatusCode As HttpStatusCode, Response As String)
  378. MyBase.New(Message)
  379. Me.StatusCode = StatusCode
  380. Me.Response = Response
  381. End Sub
  382. End Class
  383. #End Region
  384. #Region "多线程下载引擎"
  385. Private ThreadClient As New HttpClient(New HttpClientHandler With {
  386. .AutomaticDecompression = DecompressionMethods.Deflate Or DecompressionMethods.GZip
  387. })
  388. ''' <summary>
  389. ''' 最大线程数。
  390. ''' </summary>
  391. Public NetTaskThreadLimit As Integer
  392. ''' <summary>
  393. ''' 速度下限。
  394. ''' </summary>
  395. Public NetTaskSpeedLimitLow As Long = 256 * 1024L '256K/s
  396. ''' <summary>
  397. ''' 速度上限。若无限制则为 -1。
  398. ''' </summary>
  399. Public NetTaskSpeedLimitHigh As Long = -1
  400. ''' <summary>
  401. ''' 基于限速,当前可以下载的剩余量。
  402. ''' </summary>
  403. Public NetTaskSpeedLimitLeft As Long = -1
  404. Private ReadOnly NetTaskSpeedLimitLeftLock As New Object
  405. ''' <summary>
  406. ''' 正在运行中的线程数。
  407. ''' </summary>
  408. Public NetTaskThreadCount As Integer = 0
  409. Private ReadOnly NetTaskThreadCountLock As New Object
  410. ''' <summary>
  411. ''' 下载源。
  412. ''' </summary>
  413. Public Class NetSource
  414. Public Id As Integer
  415. Public Url As String
  416. Public FailCount As Integer
  417. Public Ex As Exception
  418. ''' <summary>
  419. ''' 若该下载源正在进行强制单线程下载,标记这个唯一的线程。
  420. ''' </summary>
  421. Public SingleThread As NetThread
  422. Public IsFailed As Boolean
  423. Public Overrides Function ToString() As String
  424. Return Url
  425. End Function
  426. End Class
  427. ''' <summary>
  428. ''' 下载进度标示。
  429. ''' </summary>
  430. Public Enum NetState
  431. ''' <summary>
  432. ''' 尚未进行已存在检查。
  433. ''' </summary>
  434. WaitingToCheck = -1
  435. ''' <summary>
  436. ''' 尚未开始。
  437. ''' </summary>
  438. WaitingToDownload = 0
  439. ''' <summary>
  440. ''' 正在连接,尚未获取文件大小。
  441. ''' </summary>
  442. Connecting = 1
  443. ''' <summary>
  444. ''' 已连接,正在尝试接收首个下载数据包。
  445. ''' </summary>
  446. Reading = 2
  447. ''' <summary>
  448. ''' 正在下载。
  449. ''' </summary>
  450. Downloading = 3
  451. ''' <summary>
  452. ''' 正在合并文件。
  453. ''' </summary>
  454. Merging = 4
  455. ''' <summary>
  456. ''' 已完成。
  457. ''' </summary>
  458. Finished = 5
  459. ''' <summary>
  460. ''' 已失败或中断。
  461. ''' </summary>
  462. Interrupted = 6
  463. End Enum
  464. ''' <summary>
  465. ''' 预下载检查行为。
  466. ''' </summary>
  467. Public Enum NetPreDownloadBehaviour
  468. ''' <summary>
  469. ''' 当文件已存在时,显示提示以提醒用户是否继续下载。
  470. ''' </summary>
  471. HintWhileExists
  472. ''' <summary>
  473. ''' 当文件已存在或正在下载时,直接退出下载函数执行,不对用户进行提示。
  474. ''' </summary>
  475. ExitWhileExistsOrDownloading
  476. ''' <summary>
  477. ''' 不进行已存在检查。
  478. ''' </summary>
  479. IgnoreCheck
  480. End Enum
  481. ''' <summary>
  482. ''' 下载线程。
  483. ''' </summary>
  484. Public Class NetThread
  485. Implements IEnumerable(Of NetThread), IEquatable(Of NetThread)
  486. ''' <summary>
  487. ''' 对应的下载任务。
  488. ''' </summary>
  489. Public Task As NetFile
  490. ''' <summary>
  491. ''' 对应的线程。
  492. ''' </summary>
  493. Public Thread As Thread
  494. ''' <summary>
  495. ''' 链表中的下一个线程。
  496. ''' </summary>
  497. Public NextThread As NetThread
  498. Private ReadOnly Iterator Property [Next]() As IEnumerable(Of NetThread)
  499. Get
  500. Dim CurrentChain As NetThread = Me
  501. While CurrentChain IsNot Nothing
  502. Yield CurrentChain
  503. CurrentChain = CurrentChain.NextThread
  504. End While
  505. End Get
  506. End Property
  507. Public Function GetEnumerator() As IEnumerator(Of NetThread) Implements IEnumerable(Of NetThread).GetEnumerator
  508. Return [Next].GetEnumerator()
  509. End Function
  510. Private Function IEnumerable_GetEnumerator() As IEnumerator Implements IEnumerable.GetEnumerator
  511. Return [Next].GetEnumerator()
  512. End Function
  513. ''' <summary>
  514. ''' 分配给任务中每个线程(无论其是否失败)的编号。
  515. ''' </summary>
  516. Public Uuid As Integer
  517. ''' <summary>
  518. ''' 是否为第一个线程。
  519. ''' </summary>
  520. Public ReadOnly Property IsFirstThread As Boolean
  521. Get
  522. Return DownloadStart = 0 AndAlso Task.FileSize = -2
  523. End Get
  524. End Property
  525. ''' <summary>
  526. ''' 该线程的缓存文件。
  527. ''' </summary>
  528. Public Temp As String
  529. ''' <summary>
  530. ''' 线程下载起始位置。
  531. ''' </summary>
  532. Public DownloadStart As Long
  533. ''' <summary>
  534. ''' 线程下载结束位置。
  535. ''' </summary>
  536. Public ReadOnly Property DownloadEnd As Long
  537. Get
  538. SyncLock Task.LockChain
  539. If NextThread Is Nothing Then
  540. If Task.IsUnknownSize Then
  541. Return 1000 * 1000 * 1000 * 1000L '约 1T
  542. Else
  543. Return Task.FileSize - 1
  544. End If
  545. Else
  546. Return NextThread.DownloadStart - 1
  547. End If
  548. End SyncLock
  549. End Get
  550. End Property
  551. ''' <summary>
  552. ''' 线程未下载的文件大小。
  553. ''' </summary>
  554. Public ReadOnly Property DownloadUndone As Long
  555. Get
  556. Return DownloadEnd - (DownloadStart + DownloadDone) + 1
  557. End Get
  558. End Property
  559. ''' <summary>
  560. ''' 线程已下载的文件大小。
  561. ''' </summary>
  562. Public DownloadDone As Long = 0
  563. ''' <summary>
  564. ''' 上次记速时的时间。
  565. ''' </summary>
  566. Private SpeedLastTime As Long = GetTimeMs()
  567. ''' <summary>
  568. ''' 上次记速时的已下载大小。
  569. ''' </summary>
  570. Private SpeedLastDone As Long = 0
  571. ''' <summary>
  572. ''' 当前的下载速度,单位为 Byte / 秒。
  573. ''' </summary>
  574. Public ReadOnly Property Speed As Long
  575. Get
  576. If GetTimeMs() - SpeedLastTime > 200 Then
  577. Dim DeltaTime As Long = GetTimeMs() - SpeedLastTime
  578. _Speed = (DownloadDone - SpeedLastDone) / (DeltaTime / 1000)
  579. SpeedLastDone = DownloadDone
  580. SpeedLastTime += DeltaTime
  581. End If
  582. Return _Speed
  583. End Get
  584. End Property
  585. Private _Speed As Long = 0
  586. ''' <summary>
  587. ''' 线程初始化时的时间。
  588. ''' </summary>
  589. Public InitTime As Long = GetTimeMs()
  590. ''' <summary>
  591. ''' 上次接受到有效数据的时间,-1 表示尚未有有效数据。
  592. ''' </summary>
  593. Public LastReceiveTime As Long = -1
  594. ''' <summary>
  595. ''' 当前线程的状态。
  596. ''' </summary>
  597. Public State As NetState = NetState.WaitingToDownload
  598. ''' <summary>
  599. ''' 是否已经结束。
  600. ''' </summary>
  601. Public ReadOnly Property IsEnded As Boolean
  602. Get
  603. Return State = NetState.Finished OrElse State = NetState.Interrupted
  604. End Get
  605. End Property
  606. ''' <summary>
  607. ''' 当前选取的是哪一个 Url。
  608. ''' </summary>
  609. Public Source As NetSource
  610. '允许进行 UUID 比较
  611. Public Overloads Function Equals(other As NetThread) As Boolean Implements IEquatable(Of NetThread).Equals
  612. Return other IsNot Nothing AndAlso Uuid = other.Uuid
  613. End Function
  614. Public Overrides Function Equals(obj As Object) As Boolean
  615. Return Equals(TryCast(obj, NetThread))
  616. End Function
  617. Public Shared Operator =(left As NetThread, right As NetThread) As Boolean
  618. Return EqualityComparer(Of NetThread).Default.Equals(left, right)
  619. End Operator
  620. Public Shared Operator <>(left As NetThread, right As NetThread) As Boolean
  621. Return Not left = right
  622. End Operator
  623. End Class
  624. ''' <summary>
  625. ''' 下载单个文件。
  626. ''' </summary>
  627. Public Class NetFile
  628. #Region "属性"
  629. ''' <summary>
  630. ''' 所属的文件列表任务。
  631. ''' </summary>
  632. Public Tasks As New SafeList(Of LoaderDownload)
  633. ''' <summary>
  634. ''' 所有下载源。
  635. ''' </summary>
  636. Public Sources As SafeList(Of NetSource)
  637. ''' <summary>
  638. ''' 用于在第一个线程出错时切换下载源。
  639. ''' </summary>
  640. Private FirstThreadSource As Integer = 0
  641. ''' <summary>
  642. ''' 所有已经被标记为失败的,但未完整尝试过的,不允许断点续传的下载源。
  643. ''' </summary>
  644. Public SourcesOnce As New SafeList(Of NetSource)
  645. ''' <summary>
  646. ''' 仅当合并失败或首次下载失败时,会将所有下载源重新标记为不允许断点续传的下载源,逐个重新尝试下载。
  647. ''' 这一策略可以兼容多个下载源中的一部分返回错误的文件的情况,以及部分在多线程下载时会抽风的源。
  648. ''' </summary>
  649. Private Retried As Boolean = False
  650. ''' <summary>
  651. ''' 获取从某个源开始,第一个可用的源。
  652. ''' </summary>
  653. Private Function GetSource(Optional Id As Integer = 0) As NetSource
  654. If Id >= Sources.Count OrElse Id < 0 Then Id = 0
  655. SyncLock LockSource
  656. If HasAvailableSource(False) Then
  657. '存在多线程可用源
  658. Dim CurrentSource As NetSource = Sources(Id)
  659. While CurrentSource.IsFailed
  660. Id += 1
  661. If Id >= Sources.Count Then Id = 0
  662. CurrentSource = Sources(Id)
  663. End While
  664. Return CurrentSource
  665. ElseIf SourcesOnce.Any Then
  666. '仅存在单线程可用源
  667. Return SourcesOnce.First
  668. Else
  669. '没有可用源
  670. Return Nothing
  671. End If
  672. End SyncLock
  673. End Function
  674. ''' <summary>
  675. ''' 是否存在可用源。
  676. ''' </summary>
  677. Public Function HasAvailableSource(Optional AllowOnceSource As Boolean = True) As Boolean
  678. SyncLock LockSource
  679. If Sources.Any(Function(s) Not s.IsFailed) Then Return True '存在多线程可用源
  680. If AllowOnceSource AndAlso SourcesOnce.Any Then Return True '存在单线程可用源
  681. End SyncLock
  682. Return False
  683. End Function
  684. ''' <summary>
  685. ''' 存储在本地的带文件名的地址。
  686. ''' </summary>
  687. Public LocalPath As String = Nothing
  688. ''' <summary>
  689. ''' 存储在本地的文件名。
  690. ''' </summary>
  691. Public LocalName As String = Nothing
  692. ''' <summary>
  693. ''' 当前的下载状态。
  694. ''' </summary>
  695. Public State As NetState = NetState.WaitingToCheck
  696. ''' <summary>
  697. ''' 导致下载失败的原因。
  698. ''' </summary>
  699. Public Ex As New List(Of Exception)
  700. ''' <summary>
  701. ''' 作为文件组成部分的线程链表。
  702. ''' 如果没有线程,可以为 Nothing。
  703. ''' </summary>
  704. Public Threads As NetThread
  705. ''' <summary>
  706. ''' 文件的总大小。若为 -2 则为未获取,若为 -1 则为无法获取准确大小。
  707. ''' </summary>
  708. Public FileSize As Long = -2
  709. ''' <summary>
  710. ''' 该文件是否无法获取准确大小。
  711. ''' </summary>
  712. Public IsUnknownSize As Boolean = False
  713. ''' <summary>
  714. ''' 该文件是否不需要分割。
  715. ''' </summary>
  716. Public ReadOnly Property IsNoSplit As Boolean
  717. Get
  718. Return IsUnknownSize OrElse FileSize < 1024 * 1024 '小于 1MB 的文件不分割
  719. End Get
  720. End Property
  721. ''' <summary>
  722. ''' 为不需要分割的小文件在内存中提供临时存储。
  723. ''' </summary>
  724. Private Cache As MemoryStream
  725. ''' <summary>
  726. ''' 文件的已下载大小。
  727. ''' </summary>
  728. Public DownloadDone As Long = 0
  729. Private ReadOnly LockDone As New Object
  730. ''' <summary>
  731. ''' 文件的校验规则。
  732. ''' </summary>
  733. Public Check As FileChecker
  734. ''' <summary>
  735. ''' 是否模拟浏览器的 UserAgent 和 Referer。
  736. ''' </summary>
  737. Public SimulateBrowserHeaders As Boolean
  738. ''' <summary>
  739. ''' 上次记速时的时间。
  740. ''' </summary>
  741. Private SpeedLastTime As Long = GetTimeMs()
  742. ''' <summary>
  743. ''' 上次记速时的已下载大小。
  744. ''' </summary>
  745. Private SpeedLastDone As Long = 0
  746. ''' <summary>
  747. ''' 当前的下载速度,单位为 Byte / 秒。
  748. ''' </summary>
  749. Public ReadOnly Property Speed As Long
  750. Get
  751. If GetTimeMs() - SpeedLastTime > 200 Then
  752. Dim DeltaTime As Long = GetTimeMs() - SpeedLastTime
  753. _Speed = (DownloadDone - SpeedLastDone) / (DeltaTime / 1000)
  754. SpeedLastDone = DownloadDone
  755. SpeedLastTime += DeltaTime
  756. End If
  757. Return _Speed
  758. End Get
  759. End Property
  760. Private _Speed As Long = 0
  761. ''' <summary>
  762. ''' 该文件是否由本地文件直接拷贝完成。
  763. ''' </summary>
  764. Public IsCopy As Boolean = False
  765. ''' <summary>
  766. ''' 本文件的显示进度。
  767. ''' </summary>
  768. Public ReadOnly Property Progress As Double
  769. Get
  770. Select Case State
  771. Case NetState.WaitingToCheck
  772. Return 0
  773. Case NetState.WaitingToDownload
  774. Return 0.01
  775. Case NetState.Connecting
  776. Return 0.02
  777. Case NetState.Reading
  778. Return 0.04
  779. Case NetState.Downloading
  780. '正在下载中,对应 5% ~ 98%
  781. Dim OriginalProgress As Double = If(IsUnknownSize, 0.5, DownloadDone / Math.Max(FileSize, 1))
  782. OriginalProgress = 1 - (1 - OriginalProgress) ^ 0.9
  783. Return OriginalProgress * 0.93 + 0.05
  784. Case NetState.Merging
  785. Return 0.99
  786. Case NetState.Finished, NetState.Interrupted
  787. Return 1
  788. Case Else
  789. Throw New ArgumentOutOfRangeException("文件状态未知:" & State)
  790. End Select
  791. End Get
  792. End Property
  793. ''' <summary>
  794. ''' 各个线程建立连接成功的总次数。
  795. ''' </summary>
  796. Private ConnectCount As Integer = 0
  797. ''' <summary>
  798. ''' 各个线程建立连接成功的总时间。
  799. ''' </summary>
  800. Private ConnectTime As Long = 0
  801. ''' <summary>
  802. ''' 各个线程建立连接成功的平均时间,单位为毫秒,-1 代表尚未有成功连接。
  803. ''' </summary>
  804. Private ReadOnly Property ConnectAverage As Integer
  805. Get
  806. SyncLock LockCount
  807. Return If(ConnectCount = 0, -1, ConnectTime / ConnectCount)
  808. End SyncLock
  809. End Get
  810. End Property
  811. Private Const FilePieceLimit As Long = 256 * 1024
  812. Public ReadOnly LockCount As New Object
  813. Public ReadOnly LockState As New Object
  814. Public ReadOnly LockChain As New Object
  815. Public ReadOnly LockSource As New Object
  816. Public ReadOnly Uuid As Integer = GetUuid()
  817. Public Overrides Function Equals(obj As Object) As Boolean
  818. Dim file = TryCast(obj, NetFile)
  819. Return file IsNot Nothing AndAlso Uuid = file.Uuid
  820. End Function
  821. #End Region
  822. ''' <summary>
  823. ''' 新建一个需要下载的文件。
  824. ''' </summary>
  825. ''' <param name="LocalPath">包含文件名的本地地址。</param>
  826. Public Sub New(Urls As IEnumerable(Of String), LocalPath As String, Optional Checker As FileChecker = Nothing, Optional SimulateBrowserHeaders As Boolean = False)
  827. Dim Sources As New List(Of NetSource)
  828. Dim Count As Integer = 0
  829. Urls = Urls.Distinct.ToArray
  830. For Each Source As String In Urls
  831. Sources.Add(New NetSource With {.FailCount = 0, .Url = SecretCdnSign(Source.Replace(vbCr, "").Replace(vbLf, "").Trim), .Id = Count, .IsFailed = False, .Ex = Nothing})
  832. Count += 1
  833. Next
  834. Me.Sources = Sources
  835. Me.LocalPath = LocalPath
  836. Me.Check = Checker
  837. Me.SimulateBrowserHeaders = SimulateBrowserHeaders
  838. Me.LocalName = GetFileNameFromPath(LocalPath)
  839. End Sub
  840. ''' <summary>
  841. ''' 尝试开始一个新的下载线程。
  842. ''' 如果失败,返回 Nothing。
  843. ''' </summary>
  844. Public Function TryBeginThread() As NetThread
  845. Try
  846. Dim StartPosition As Long, StartSource As NetSource = Nothing
  847. Dim Th As Thread, ThreadInfo As NetThread
  848. '条件检测
  849. SyncLock LockSource
  850. If NetTaskThreadCount >= NetTaskThreadLimit OrElse Not HasAvailableSource() OrElse
  851. (IsNoSplit AndAlso Threads IsNot Nothing AndAlso Threads.State <> NetState.Interrupted) Then Return Nothing
  852. If State >= NetState.Merging OrElse State = NetState.WaitingToCheck Then Return Nothing
  853. SyncLock LockState
  854. If State < NetState.Connecting Then State = NetState.Connecting
  855. End SyncLock
  856. '初始化参数
  857. '获取线程起点与下载源
  858. '不分割
  859. If IsNoSplit Then GoTo StartSingleThreadDownload
  860. '只剩下单线程可用点
  861. If Not HasAvailableSource(False) Then
  862. If SourcesOnce.First.SingleThread IsNot Nothing AndAlso SourcesOnce.First.SingleThread.State <> NetState.Interrupted Then Return Nothing
  863. StartSingleThreadDownload:
  864. Reset()
  865. State = NetState.Reading
  866. End If
  867. End SyncLock
  868. SyncLock LockChain
  869. '首个开始点
  870. If Threads Is Nothing Then
  871. StartPosition = 0
  872. StartSource = GetSource(FirstThreadSource)
  873. FirstThreadSource = StartSource.Id + 1
  874. GoTo StartThread
  875. End If
  876. '寻找失败点
  877. For Each Thread As NetThread In Threads
  878. If Thread.State = NetState.Interrupted AndAlso Thread.DownloadUndone > 0 Then
  879. StartPosition = Thread.DownloadStart + Thread.DownloadDone
  880. StartSource = GetSource(Thread.Source.Id + 1)
  881. GoTo StartThread
  882. End If
  883. Next
  884. '是否禁用多线程,以及规定碎片大小
  885. Dim Source As NetSource = GetSource()
  886. If Source Is Nothing Then Return Nothing
  887. Dim TargetUrl As String = Source.Url
  888. If TargetUrl.Contains("pcl2-server") OrElse TargetUrl.Contains("bmclapi") OrElse TargetUrl.Contains("github.com") OrElse
  889. TargetUrl.Contains("optifine.net") OrElse TargetUrl.Contains("modrinth") OrElse TargetUrl.Contains("momot.rs") Then Return Nothing
  890. '寻找最大碎片
  891. 'FUTURE: 下载引擎重做,计算下载源平均链接时间和线程下载速度,按最高时间节省来开启多线程
  892. Dim FilePieceMax As NetThread = Threads
  893. For Each Thread As NetThread In Threads
  894. If Thread.DownloadUndone > FilePieceMax.DownloadUndone Then FilePieceMax = Thread
  895. Next
  896. If FilePieceMax Is Nothing OrElse FilePieceMax.DownloadUndone < FilePieceLimit Then Return Nothing
  897. StartPosition = FilePieceMax.DownloadEnd - FilePieceMax.DownloadUndone * 0.4
  898. StartSource = GetSource()
  899. '开始线程
  900. StartThread:
  901. If (StartPosition > FileSize AndAlso FileSize >= 0 AndAlso Not IsUnknownSize) OrElse StartPosition < 0 OrElse IsNothing(StartSource) Then Return Nothing
  902. '构建线程
  903. Dim ThreadUuid As Integer = GetUuid()
  904. If Not Tasks.Any() Then Return Nothing '由于中断,已没有可用任务
  905. Th = New Thread(AddressOf Thread) With {.Name = $"下载 {Tasks(0).Uuid}/{Uuid}/{ThreadUuid}#", .Priority = ThreadPriority.BelowNormal}
  906. ThreadInfo = New NetThread With {.Uuid = ThreadUuid, .DownloadStart = StartPosition, .Thread = Th, .Source = StartSource, .Task = Me, .State = NetState.WaitingToDownload}
  907. '链表处理
  908. If ThreadInfo.IsFirstThread OrElse Threads Is Nothing Then
  909. Threads = ThreadInfo
  910. Else
  911. Dim CurrentChain As NetThread = Threads
  912. While CurrentChain.DownloadEnd <= StartPosition
  913. CurrentChain = CurrentChain.NextThread
  914. End While
  915. ThreadInfo.NextThread = CurrentChain.NextThread
  916. CurrentChain.NextThread = ThreadInfo
  917. End If
  918. End SyncLock
  919. '开始线程
  920. SyncLock NetTaskThreadCountLock
  921. NetTaskThreadCount += 1
  922. End SyncLock
  923. SyncLock LockSource
  924. If Not HasAvailableSource(False) Then SourcesOnce.First.SingleThread = ThreadInfo
  925. End SyncLock
  926. Th.Start(ThreadInfo)
  927. Return ThreadInfo
  928. Catch ex As Exception
  929. Fail(New Exception($"尝试开始下载线程失败({LocalName})", ex))
  930. Return Nothing
  931. End Try
  932. End Function
  933. ''' <summary>
  934. ''' 每个下载线程执行的代码。
  935. ''' </summary>
  936. Private Sub Thread(Th As NetThread)
  937. If ModeDebug OrElse Th.DownloadStart = 0 Then Log($"[Download] {LocalName}:开始,起始点 {Th.DownloadStart},{Th.Source.Url}")
  938. Dim ResultStream As Stream = Nothing, HttpRequest As HttpRequestMessage = Nothing,
  939. Response As HttpResponseMessage = Nothing, ResponseStream As Stream = Nothing,
  940. CancelToken As CancellationTokenSource = Nothing, HostIp As String = Nothing
  941. '部分下载源真的特别慢,并且只需要一个请求,例如 Ping 为 20s,如果增长太慢,就会造成类似 2.5s 5s 7.5s 10s 12.5s... 的极大延迟
  942. '延迟过长会导致某些特别慢的链接迟迟不被掐死
  943. Dim Timeout As Integer = Math.Min(Math.Max(ConnectAverage, 10000) * (1 + Th.Source.FailCount), 30000)
  944. Th.State = NetState.Connecting
  945. Try
  946. Dim HttpDataCount As Integer = 0
  947. If SourcesOnce.Contains(Th.Source) AndAlso Th <> Th.Source.SingleThread Then GoTo SourceBreak
  948. HttpRequest = New HttpRequestMessage(HttpMethod.Get, Th.Source.Url)
  949. SecretHeadersSign(Th.Source.Url, HttpRequest, SimulateBrowserHeaders)
  950. CancelToken = New CancellationTokenSource(Timeout)
  951. HostIp = DNSLookup(HttpRequest, CancelToken) 'DNS 预解析
  952. If Not Th.IsFirstThread Then HttpRequest.Headers.Range = New RangeHeaderValue(Th.DownloadStart, Nothing)
  953. Dim ContentLength As Long = 0
  954. Response = ThreadClient.SendAsync(HttpRequest, HttpCompletionOption.ResponseHeadersRead, CancelToken.Token).GetResultWithTimeout(CancelToken, Timeout)
  955. If Not Response.IsSuccessStatusCode Then Throw New Exception($"错误码 {Response.StatusCode} ({CInt(Response.StatusCode)}),{Th.Source.Url}") '状态码检查
  956. If State = NetState.Interrupted Then GoTo SourceBreak '快速中断
  957. If ModeDebug AndAlso Response.RequestMessage.RequestUri.ToString <> Th.Source.Url Then Log($"[Download] {LocalName}:重定向至 {Response.RequestMessage.RequestUri}")
  958. '文件大小校验
  959. ContentLength = Response.Content.Headers.ContentLength.GetValueOrDefault(-1)
  960. If ContentLength < 0 Then
  961. If FileSize > 1 Then
  962. If Th.DownloadStart = 0 Then
  963. Log($"[Download] {LocalName}:文件大小未知,但已从其他下载源获取,不作处理")
  964. Else
  965. Log($"[Download] {LocalName}:ContentLength 返回了 {ContentLength},无法确定是否支持分段下载,视作不支持")
  966. GoTo NotSupportRange
  967. End If
  968. Else
  969. FileSize = -1 : IsUnknownSize = True
  970. Log($"[Download] {LocalName}:文件大小未知")
  971. End If
  972. ElseIf Th.IsFirstThread Then
  973. If Check IsNot Nothing Then
  974. If ContentLength < Check.MinSize AndAlso Check.MinSize > 0 Then
  975. Throw New Exception($"文件大小不足,获取结果为 {ContentLength},要求至少为 {Check.MinSize}。")
  976. End If
  977. If ContentLength <> Check.ActualSize AndAlso Check.ActualSize > 0 Then
  978. Throw New Exception($"文件大小不一致,获取结果为 {ContentLength},要求必须为 {Check.ActualSize}。")
  979. End If
  980. End If
  981. FileSize = ContentLength : IsUnknownSize = False
  982. Log($"[Download] {LocalName}:文件大小 {ContentLength}({GetString(ContentLength)})")
  983. '若文件大小大于 50 M,进行剩余磁盘空间校验
  984. If ContentLength > 50 * 1024 * 1024 Then
  985. For Each Drive As DriveInfo In DriveInfo.GetDrives
  986. Dim DriveName As String = Drive.Name.First.ToString
  987. Dim RequiredSpace = If(PathTemp.StartsWithF(DriveName), ContentLength * 1.1, 0) +
  988. If(LocalPath.StartsWithF(DriveName), ContentLength + 5 * 1024 * 1024, 0)
  989. If Drive.TotalFreeSpace < RequiredSpace Then
  990. Throw New Exception(DriveName & " 盘空间不足,无法进行下载。" & vbCrLf & "需要至少 " & GetString(RequiredSpace) & " 空间,但当前仅剩余 " & GetString(Drive.TotalFreeSpace) & "。" &
  991. If(PathTemp.StartsWithF(DriveName), vbCrLf & vbCrLf & "下载时需要与文件同等大小的空间存放缓存,你可以在设置中调整缓存文件夹的位置。", ""))
  992. End If
  993. Next
  994. End If
  995. ElseIf FileSize < 0 Then
  996. Throw New Exception("非首线程运行时,尚未获取文件大小")
  997. ElseIf Th.DownloadStart > 0 AndAlso ContentLength = FileSize Then
  998. NotSupportRange:
  999. SyncLock LockSource
  1000. If SourcesOnce.Contains(Th.Source) Then GoTo SourceBreak
  1001. End SyncLock
  1002. Throw New RangeNotSupportedException($"该下载源不支持分段下载:Range 起始于 {Th.DownloadStart},预期 ContentLength 为 {FileSize - Th.DownloadStart},返回 ContentLength 为 {ContentLength},总文件大小 {FileSize}")
  1003. ElseIf Not FileSize - Th.DownloadStart = ContentLength Then
  1004. Throw New RangeNotSupportedException($"获取到的分段大小不一致:Range 起始于 {Th.DownloadStart},预期 ContentLength 为 {FileSize - Th.DownloadStart},返回 ContentLength 为 {ContentLength},总文件大小 {FileSize}")
  1005. End If
  1006. 'Log($"[Download] {LocalName} {Info.Uuid}#:通过大小检查,文件大小 {FileSize},起始点 {Info.DownloadStart},ContentLength {ContentLength}")
  1007. Th.State = NetState.Reading
  1008. SyncLock LockState
  1009. If State < NetState.Reading Then State = NetState.Reading
  1010. End SyncLock
  1011. '创建缓存文件
  1012. If IsNoSplit Then
  1013. Th.Temp = Nothing
  1014. Cache = New MemoryStream
  1015. ResultStream = Cache
  1016. Else
  1017. Directory.CreateDirectory(PathTemp & "Download")
  1018. Th.Temp = $"{PathTemp}Download\{Uuid}_{Th.Uuid}_{RandomInteger(0, 999999)}.tmp"
  1019. ResultStream = New FileStream(Th.Temp, FileMode.Create, FileAccess.Write, FileShare.Read)
  1020. End If
  1021. '开始下载
  1022. ResponseStream = Response.Content.ReadAsStreamAsync().GetAwaiter().GetResult()
  1023. If Setup.Get("SystemDebugDelay") Then Threading.Thread.Sleep(RandomInteger(50, 3000))
  1024. Dim ResponseBytes As Byte() = New Byte(16384) {}
  1025. HttpDataCount = ResponseStream.ReadAsync(ResponseBytes, 0, 16384, CancelToken.Token).GetResultWithTimeout(CancelToken, Timeout)
  1026. While (IsUnknownSize OrElse Th.DownloadUndone > 0) AndAlso '判断是否下载完成
  1027. HttpDataCount > 0 AndAlso Not IsProgramEnded AndAlso State < NetState.Merging AndAlso (Not Th.Source.IsFailed OrElse Th.Source.SingleThread = Th)
  1028. '限速
  1029. While NetTaskSpeedLimitHigh > 0 AndAlso NetTaskSpeedLimitLeft <= 0
  1030. Threading.Thread.Sleep(16)
  1031. End While
  1032. Dim RealDataCount As Integer = If(IsUnknownSize, HttpDataCount, Math.Min(HttpDataCount, Th.DownloadUndone))
  1033. SyncLock NetTaskSpeedLimitLeftLock
  1034. If NetTaskSpeedLimitHigh > 0 Then NetTaskSpeedLimitLeft -= RealDataCount
  1035. End SyncLock
  1036. Dim DeltaTime = GetTimeMs() - Th.LastReceiveTime
  1037. If DeltaTime > 1000000 Then DeltaTime = 1 '避免时间刻反转导致出现极大值
  1038. If RealDataCount > 0 Then
  1039. '有数据
  1040. If Th.DownloadDone = 0 Then
  1041. '第一次接受到数据
  1042. Th.State = NetState.Downloading
  1043. SyncLock LockState
  1044. If State < NetState.Downloading Then State = NetState.Downloading
  1045. End SyncLock
  1046. SyncLock LockCount
  1047. ConnectCount += 1
  1048. ConnectTime += GetTimeMs() - Th.InitTime
  1049. End SyncLock
  1050. End If
  1051. SyncLock LockCount
  1052. Th.Source.FailCount = 0
  1053. For Each Task In Tasks
  1054. Task.FailCount = 0
  1055. Next
  1056. End SyncLock
  1057. NetManager.DownloadDone += RealDataCount
  1058. SyncLock LockDone
  1059. DownloadDone += RealDataCount
  1060. End SyncLock
  1061. Th.DownloadDone += RealDataCount
  1062. ResultStream.Write(ResponseBytes, 0, RealDataCount)
  1063. '已完成
  1064. If Th.DownloadUndone = 0 AndAlso Not IsUnknownSize Then Exit While
  1065. '检查速度是否过慢
  1066. If DeltaTime > 5000 AndAlso DeltaTime > RealDataCount AndAlso '数据包间隔大于 5s,且速度小于 1K/s
  1067. Th.Source.SingleThread Is Nothing Then '且并非单线程下载
  1068. Throw New TimeoutException("由于速度过慢断开链接,下载 " & RealDataCount & " B,消耗 " & DeltaTime & " ms。")
  1069. End If
  1070. Th.LastReceiveTime = GetTimeMs()
  1071. ElseIf Th.LastReceiveTime > 0 AndAlso DeltaTime > Timeout Then
  1072. '无数据,且已超时
  1073. Throw New TimeoutException("操作超时,无数据。")
  1074. End If
  1075. HttpDataCount = ResponseStream.ReadAsync(ResponseBytes, 0, 16384, CancelToken.Token).GetResultWithTimeout(CancelToken, Timeout)
  1076. End While
  1077. SourceBreak:
  1078. If State = NetState.Interrupted OrElse (Th.Source.IsFailed AndAlso Th.Source.SingleThread <> Th) OrElse (Th.DownloadUndone > 0 AndAlso Not IsUnknownSize) Then
  1079. '被外部中断
  1080. Th.State = NetState.Interrupted
  1081. Log($"[Download] {LocalName}:中断")
  1082. ElseIf HttpDataCount = 0 AndAlso Th.DownloadUndone > 0 AndAlso Not IsUnknownSize Then
  1083. '服务器无返回数据
  1084. Throw New Exception($"返回的 ContentLength 过多:ContentLength 为 {ContentLength},但获取到的总数据量仅为 {Th.DownloadDone}(全文件总数据量 {DownloadDone})")
  1085. Else
  1086. '本线程完成
  1087. Th.State = NetState.Finished
  1088. If ModeDebug Then Log($"[Download] {LocalName}:完成,已下载 {Th.DownloadDone}({Th.DownloadStart}~{Th.DownloadEnd})")
  1089. RecordIPReliability(HostIp, 0.5)
  1090. End If
  1091. Catch ex As Exception
  1092. Log($"[Download] {LocalName}:出错,{If(TypeOf ex Is OperationCanceledException OrElse TypeOf ex Is TimeoutException,
  1093. $"已超时({Timeout}ms)", ex.GetDetail())},IP:{HostIp}")
  1094. RecordIPReliability(HostIp, -0.7)
  1095. SourceFail(Th, ex, False)
  1096. Finally
  1097. '释放资源
  1098. HttpRequest?.Dispose()
  1099. Response?.Dispose()
  1100. ResponseStream?.Dispose()
  1101. CancelToken?.Dispose()
  1102. If Not IsNoSplit Then ResultStream?.Dispose()
  1103. '改变计数
  1104. SyncLock NetTaskThreadCountLock
  1105. NetTaskThreadCount -= 1
  1106. End SyncLock
  1107. '合并
  1108. If ((FileSize >= 0 AndAlso DownloadDone >= FileSize) OrElse (FileSize = -1 AndAlso DownloadDone > 0)) AndAlso
  1109. State < NetState.Merging AndAlso Th.State <> NetState.Interrupted Then Merge(Th)
  1110. End Try
  1111. End Sub
  1112. Private Sub SourceFail(Th As NetThread, ex As Exception, IsMergeFailure As Boolean)
  1113. '状态变更
  1114. SyncLock LockCount
  1115. Th.Source.FailCount += 1
  1116. For Each Task In Tasks
  1117. Task.FailCount += 1
  1118. Next
  1119. End SyncLock
  1120. Th.State = NetState.Interrupted
  1121. Th.Source.Ex = ex
  1122. '根据情况判断,是否在多线程下禁用下载源(连续错误过多,或不支持断点续传)
  1123. Dim IsRangeNotSupported As Boolean = TypeOf ex Is RangeNotSupportedException OrElse ex.Message.Contains("(416)")
  1124. If IsMergeFailure OrElse IsRangeNotSupported OrElse
  1125. ex.Message.Contains("(502)") OrElse ex.Message.Contains("(404)") OrElse
  1126. ex.Message.Contains("未能解析") OrElse ex.Message.Contains("无返回数据") OrElse ex.Message.Contains("空间不足") OrElse
  1127. ((ex.Message.Contains("(403)") OrElse ex.Message.Contains("(429)")) AndAlso Not Th.Source.Url.ContainsF("bmclapi")) OrElse 'BMCLAPI 的部分源在高频率请求下会返回 403/429,所以不应因此禁用下载源
  1128. (Th.Source.FailCount >= MathClamp(NetTaskThreadLimit, 5, 30) AndAlso DownloadDone < 1) OrElse Th.Source.FailCount > NetTaskThreadLimit + 2 Then
  1129. '当一个下载源有多个线程在下载时,只选择其中一个线程进行后续处理
  1130. Dim IsThisFail As Boolean = False
  1131. SyncLock LockSource
  1132. If Not Th.Source.IsFailed OrElse Th.Source.SingleThread = Th Then
  1133. IsThisFail = True
  1134. Th.Source.IsFailed = True
  1135. End If
  1136. End SyncLock
  1137. '……后续处理
  1138. If IsThisFail Then
  1139. Log($"[Download] {LocalName}:下载源被禁用({Th.Source.Id},Range 问题:{IsRangeNotSupported}):{Th.Source.Url}")
  1140. Log(ex, $"{If(SourcesOnce.FirstOrDefault?.SingleThread Is Nothing, "", "单线程")}下载源 {Th.Source.Id} 已被禁用",
  1141. If(IsRangeNotSupported OrElse ex.Message.Contains("(404)"), LogLevel.Developer, LogLevel.Debug))
  1142. SyncLock LockSource
  1143. SourcesOnce.Remove(Th.Source)
  1144. End SyncLock
  1145. If ex.Message.Contains("空间不足") Then
  1146. '硬盘空间不足:强制失败
  1147. Fail(ex)
  1148. ElseIf HasAvailableSource() AndAlso Not IsMergeFailure Then
  1149. '当前源失败,但还有下载源:正常地继续执行
  1150. ElseIf Not Retried Then
  1151. '合并失败或首次下载失败,未重试:将所有下载源重新标记为不允许断点续传的下载源,逐个重新尝试下载
  1152. '若所有源均不支持 Range,也会走到这里重试
  1153. If Not IsRangeNotSupported Then Log($"[Download] {LocalName}:文件下载失败,正在自动重试……", LogLevel.Debug)
  1154. Retried = True
  1155. SyncLock LockSource
  1156. SourcesOnce.Clear()
  1157. For Each Source In Sources
  1158. SourcesOnce.Add(Source)
  1159. Source.IsFailed = True
  1160. Next
  1161. End SyncLock
  1162. Reset()
  1163. SyncLock LockState
  1164. State = NetState.WaitingToDownload
  1165. End SyncLock
  1166. ElseIf HasAvailableSource() AndAlso IsMergeFailure Then
  1167. '合并失败且单个源失败:继续下一个源
  1168. Reset()
  1169. SyncLock LockState
  1170. State = NetState.WaitingToDownload
  1171. End SyncLock
  1172. Else
  1173. '失败
  1174. Log($"[Download] {LocalName}:已无可用下载源,下载失败")
  1175. Dim ExampleEx As Exception = Nothing
  1176. SyncLock LockSource
  1177. For Each Source As NetSource In Sources
  1178. Log("[Download] 已禁用的下载源:" & Source.Url)
  1179. If Source.Ex IsNot Nothing Then
  1180. ExampleEx = Source.Ex
  1181. Log(Source.Ex, "下载源禁用原因", LogLevel.Developer)
  1182. End If
  1183. Next
  1184. End SyncLock
  1185. Fail(ExampleEx)
  1186. End If
  1187. End If
  1188. End If
  1189. '清理当前已下载的内容
  1190. If FileSize = -2 Then Reset()
  1191. End Sub
  1192. Private Sub Reset()
  1193. FileSize = -2
  1194. Cache?.Dispose() : Cache = Nothing
  1195. SyncLock LockChain
  1196. Threads = Nothing
  1197. End SyncLock
  1198. NetManager.DownloadDone -= DownloadDone
  1199. SyncLock LockDone
  1200. DownloadDone = 0
  1201. End SyncLock
  1202. SpeedLastDone = 0
  1203. End Sub
  1204. '最终收束事件
  1205. ''' <summary>
  1206. ''' 下载完成。合并文件。
  1207. ''' </summary>
  1208. Private Sub Merge(Th As NetThread)
  1209. '状态判断
  1210. SyncLock LockState
  1211. If State < NetState.Merging Then
  1212. State = NetState.Merging
  1213. Else
  1214. Return
  1215. End If
  1216. End SyncLock
  1217. Dim RetryCount As Integer = 0
  1218. Try
  1219. Retry:
  1220. '创建文件夹
  1221. If File.Exists(LocalPath) Then File.Delete(LocalPath)
  1222. Directory.CreateDirectory(GetPathFromFullPath(LocalPath))
  1223. SyncLock LockChain
  1224. '合并文件
  1225. If IsNoSplit Then
  1226. '仅有一个线程,从缓存中输出
  1227. If ModeDebug Then Log($"[Download] {LocalName}:下载结束,从内存输出文件")
  1228. Cache.Position = 0
  1229. WriteFile(LocalPath, Cache)
  1230. ElseIf Threads.DownloadDone = DownloadDone AndAlso Threads.Temp IsNot Nothing Then
  1231. '仅有一个文件,直接复制
  1232. If ModeDebug Then Log($"[Download] {LocalName}:下载结束,仅有一个文件,无需合并")
  1233. CopyFile(Threads.Temp, LocalPath)
  1234. Else
  1235. '有多个线程,合并
  1236. If ModeDebug Then Log($"[Download] {LocalName}:下载结束,开始合并文件")
  1237. Using MergeFile As New FileStream(LocalPath, FileMode.Create)
  1238. Using AddWriter As New BinaryWriter(MergeFile)
  1239. For Each Thread As NetThread In Threads
  1240. If Thread.DownloadDone = 0 OrElse Thread.Temp Is Nothing Then Continue For
  1241. Using fs As New FileStream(Thread.Temp, FileMode.Open, FileAccess.Read, FileShare.Read)
  1242. Using TempReader As New BinaryReader(fs)
  1243. AddWriter.Write(TempReader.ReadBytes(Thread.DownloadDone))
  1244. End Using
  1245. End Using
  1246. Next
  1247. End Using
  1248. End Using
  1249. End If
  1250. '检查文件
  1251. Dim CheckResult As String = Check?.Check(LocalPath)
  1252. If CheckResult Is Nothing AndAlso Not IsUnknownSize AndAlso Check IsNot Nothing Then
  1253. If Check.ActualSize = -1 Then
  1254. CheckResult = (New FileChecker With {.ActualSize = FileSize}).Check(LocalPath) '不修改原始的 Checker,以免污染原始实例
  1255. ElseIf Check.ActualSize <> FileSize Then
  1256. CheckResult = $"文件大小不一致:任务校验要求为 {Check.ActualSize},请求结果为 {FileSize}"
  1257. End If
  1258. End If
  1259. If CheckResult IsNot Nothing Then
  1260. Log($"[Download] {LocalName} 文件校验失败,下载线程细节:")
  1261. For Each T As NetThread In Threads
  1262. Log($"[Download] - {T.Uuid}#,状态 {GetStringFromEnum(T.State)},范围 {T.DownloadStart}~{T.DownloadEnd},已下载 {T.DownloadDone},未下载 {T.DownloadUndone}")
  1263. Next
  1264. Throw New Exception(CheckResult)
  1265. End If
  1266. '后处理
  1267. If Not IsNoSplit Then
  1268. For Each Thread As NetThread In Threads
  1269. If Thread.Temp IsNot Nothing Then File.Delete(Thread.Temp)
  1270. Next
  1271. End If
  1272. Finish()
  1273. End SyncLock
  1274. Catch ex As Exception
  1275. RetryCount += 1
  1276. Log(ex, $"合并文件出错,第 {RetryCount} 次尝试({LocalName})")
  1277. '重新尝试合并
  1278. If RetryCount < 3 Then
  1279. Threading.Thread.Sleep(500 * RetryCount)
  1280. GoTo Retry
  1281. End If
  1282. '失败,禁用当前下载源并重启下载
  1283. Try
  1284. If File.Exists(LocalPath) Then File.Delete(LocalPath)
  1285. Catch exx As Exception
  1286. Log(ex, $"删除合并失败的文件出错({LocalPath})")
  1287. End Try
  1288. SourceFail(Th, ex, True)
  1289. Finally
  1290. Cache?.Dispose()
  1291. End Try
  1292. End Sub
  1293. ''' <summary>
  1294. ''' 下载失败。
  1295. ''' </summary>
  1296. Private Sub Fail(Optional RaiseEx As Exception = Nothing)
  1297. SyncLock LockState
  1298. If State >= NetState.Finished Then Return
  1299. If RaiseEx IsNot Nothing Then Ex.Add(RaiseEx)
  1300. '凉凉
  1301. Log($"[Download] {LocalName}:已失败,当前状态 {State}")
  1302. State = NetState.Interrupted
  1303. End SyncLock
  1304. AbortInternal()
  1305. For Each Task In Tasks
  1306. Task.OnFileFail(Me)
  1307. Next
  1308. End Sub
  1309. ''' <summary>
  1310. ''' 下载中断。
  1311. ''' </summary>
  1312. Public Sub Abort(AbortedTask As LoaderDownload)
  1313. '从特定任务中移除,如果它还属于其他任务,则继续下载
  1314. Tasks.Remove(AbortedTask)
  1315. If Tasks.Any Then Return
  1316. '确认中断
  1317. SyncLock LockState
  1318. If State >= NetState.Finished Then Return
  1319. Log($"[Download] {LocalName}:已中断,当前状态 {State}")
  1320. State = NetState.Interrupted
  1321. End SyncLock
  1322. AbortInternal()
  1323. End Sub
  1324. Private Sub AbortInternal()
  1325. On Error Resume Next
  1326. Reset()
  1327. SyncLock NetManager.LockRemain
  1328. NetManager.FileRemain -= 1
  1329. End SyncLock
  1330. End Sub
  1331. '状态改变接口
  1332. ''' <summary>
  1333. ''' 将该文件设置为已下载完成。
  1334. ''' </summary>
  1335. Public Sub Finish(Optional PrintLog As Boolean = True)
  1336. SyncLock LockState
  1337. If State >= NetState.Finished Then Return
  1338. State = NetState.Finished
  1339. End SyncLock
  1340. SyncLock NetManager.LockRemain
  1341. NetManager.FileRemain -= 1
  1342. End SyncLock
  1343. If PrintLog Then Log($"[Download] {LocalName}:已完成")
  1344. For Each Task In Tasks
  1345. Task.OnFileFinish(Me)
  1346. Next
  1347. End Sub
  1348. End Class
  1349. Private Class RangeNotSupportedException
  1350. Inherits WebException
  1351. Public Sub New(Message As String)
  1352. MyBase.New(Message)
  1353. End Sub
  1354. End Class
  1355. ''' <summary>
  1356. ''' 下载一系列文件的加载器。
  1357. ''' </summary>
  1358. Public Class LoaderDownload
  1359. Inherits LoaderBase
  1360. #Region "属性"
  1361. ''' <summary>
  1362. ''' 需要下载的文件。
  1363. ''' </summary>
  1364. Public Files As SafeList(Of NetFile)
  1365. ''' <summary>
  1366. ''' 剩余未完成的文件数。(用于减轻 FilesLock 的占用)
  1367. ''' </summary>
  1368. Private FileRemain As Integer
  1369. Private ReadOnly FileRemainLock As New Object
  1370. ''' <summary>
  1371. ''' 用于显示的百分比进度。
  1372. ''' </summary>
  1373. Public Overrides Property Progress As Double
  1374. Get
  1375. If State >= LoadState.Finished Then Return 1
  1376. If Not Files.Any() Then Return 0 '必须返回 0,否则在获取列表的时候会错觉已经下载完了
  1377. Return _Progress
  1378. End Get
  1379. Set(value As Double)
  1380. Throw New Exception("文件下载不允许指定进度")
  1381. End Set
  1382. End Property
  1383. Private _Progress As Double = 0
  1384. ''' <summary>
  1385. ''' 任务中的文件的连续失败计数。
  1386. ''' </summary>
  1387. Public Property FailCount As Integer
  1388. Get
  1389. Return _FailCount
  1390. End Get
  1391. Set(value As Integer)
  1392. _FailCount = value
  1393. If State = LoadState.Loading AndAlso value >= Math.Min(10000, Math.Max(FileRemain * 5.5, NetTaskThreadLimit * 5.5 + 3)) Then
  1394. Log("[Download] 由于同加载器中失败次数过多引发强制失败:连续失败了 " & value & " 次", LogLevel.Debug)
  1395. On Error Resume Next
  1396. Dim ExList As New List(Of Exception)
  1397. For Each File In Files
  1398. For Each Source In File.Sources
  1399. If Source.Ex IsNot Nothing Then
  1400. ExList.Add(Source.Ex)
  1401. If ExList.Count > 10 Then GoTo FinishExCatch
  1402. End If
  1403. Next
  1404. Next
  1405. FinishExCatch:
  1406. OnFail(ExList)
  1407. End If
  1408. End Set
  1409. End Property
  1410. Private _FailCount As Integer = 0
  1411. #End Region
  1412. ''' <summary>
  1413. ''' 刷新公开属性。由 NetManager 每 0.1 秒调用一次。
  1414. ''' </summary>
  1415. Public Sub RefreshStat()
  1416. '计算进度
  1417. Dim NewProgress As Double = 0
  1418. Dim TotalProgress As Double = 0
  1419. For Each File In Files
  1420. If File.IsCopy Then
  1421. NewProgress += File.Progress * 0.2
  1422. TotalProgress += 0.2
  1423. Else
  1424. NewProgress += File.Progress
  1425. TotalProgress += 1
  1426. End If
  1427. Next
  1428. If TotalProgress > 0 AndAlso Not Double.IsNaN(TotalProgress) Then NewProgress /= TotalProgress
  1429. '刷新进度
  1430. _Progress = NewProgress
  1431. End Sub
  1432. Public Sub New(Name As String, FileTasks As List(Of NetFile))
  1433. Me.Name = Name
  1434. Files = New SafeList(Of NetFile)(FileTasks)
  1435. End Sub
  1436. Public Overrides Sub Start(Optional Input As Object = Nothing, Optional IsForceRestart As Boolean = False)
  1437. If Input IsNot Nothing Then Files = New SafeList(Of NetFile)(Input)
  1438. '去重
  1439. Files = New SafeList(Of NetFile)(Files.Distinct(Function(a, b) a.LocalPath = b.LocalPath))
  1440. '设置剩余文件数
  1441. SyncLock FileRemainLock
  1442. FileRemain += Files.Where(Function(f) f.State <> NetState.Finished).Count
  1443. End SyncLock
  1444. State = LoadState.Loading
  1445. '开始执行
  1446. RunInNewThread(
  1447. Sub()
  1448. Try
  1449. '输入检测
  1450. If Not Files.Any() Then
  1451. OnFinish()
  1452. Return
  1453. End If
  1454. For Each File As NetFile In Files
  1455. If File Is Nothing Then Throw New ArgumentException("存在空文件请求!")
  1456. For Each Source As NetSource In File.Sources
  1457. If Not (Source.Url.StartsWithF("https://", True) OrElse Source.Url.StartsWithF("http://", True)) Then
  1458. Source.Ex = New ArgumentException("输入的下载链接不正确:" & Source.Url)
  1459. Source.IsFailed = True
  1460. End If
  1461. Next
  1462. If Not File.HasAvailableSource() Then Throw New ArgumentException("输入的下载链接不正确!")
  1463. File.LocalPath = File.LocalPath.Replace("/", "\")
  1464. If Not File.LocalPath.ToLower.Contains(":\") Then Throw New ArgumentException("输入的本地文件地址不正确:" & File.LocalPath)
  1465. If File.LocalPath.EndsWithF("\") Then Throw New ArgumentException("请输入含文件名的完整文件路径:" & File.LocalPath)
  1466. Directory.CreateDirectory(GetPathFromFullPath(File.LocalPath)) '创建目标文件夹
  1467. Next
  1468. '接入下载管理器
  1469. NetManager.Start(Me)
  1470. '====================================
  1471. ' 已存在文件查找
  1472. '====================================
  1473. '整理允许进行查找的文件
  1474. Dim FilesToCheck As New List(Of NetFile)
  1475. Dim DisabledCopy As Boolean = Setup.Get("SystemDebugSkipCopy") '在设置中禁用了复制
  1476. For Each File In Files
  1477. If Not DisabledCopy AndAlso File.Check?.CanUseExistsFile Then
  1478. FilesToCheck.Add(File)
  1479. Else '不允许,直接开始下载
  1480. SyncLock LockState
  1481. File.State = NetState.WaitingToDownload
  1482. File.IsCopy = False
  1483. End SyncLock
  1484. End If
  1485. Next
  1486. If Not FilesToCheck.Any Then Return
  1487. '获取 MC 文件夹列表
  1488. Dim Folders As New List(Of String)
  1489. Folders.Add(Environment.GetFolderPath(Environment.SpecialFolder.ApplicationData) & "\.minecraft\") '总是添加官启文件夹,因为 HMCL 会把所有文件存在这里
  1490. Folders.AddRange(McFolderList.Select(Function(f) f.Location))
  1491. Folders = Folders.Distinct.Where(Function(f) Directory.Exists(f)).ToList
  1492. '平均分配到多个检查线程
  1493. Dim ThreadCount As Integer = MathClamp(FilesToCheck.Count \ 40, 1, 8) '每个线程至少 40 个文件,最多 8 线程
  1494. If ThreadCount = 1 Then '只有一个线程,直接执行
  1495. CheckExistingFiles(FilesToCheck, Folders)
  1496. Else
  1497. Dim BaseSize = FilesToCheck.Count \ ThreadCount
  1498. Dim Remainder = FilesToCheck.Count Mod ThreadCount
  1499. Dim Index = 0
  1500. For i = 0 To ThreadCount - 1
  1501. Dim Size = BaseSize + If(i < Remainder, 1, 0)
  1502. Dim ThreadFiles = FilesToCheck.GetRange(Index, Size)
  1503. Index += Size
  1504. RunInNewThread(Sub() CheckExistingFiles(ThreadFiles, Folders), $"下载 文件复制 {Uuid}/{GetUuid()}")
  1505. Next
  1506. End If
  1507. Catch ex As Exception
  1508. OnFail(New List(Of Exception) From {New Exception("下载初始化失败", ex)})
  1509. End Try
  1510. End Sub, "L/下载 " & Uuid)
  1511. End Sub
  1512. Private Sub CheckExistingFiles(Files As List(Of NetFile), FolderList As List(Of String))
  1513. Try
  1514. If ModeDebug Then Log($"[Download] 文件检查线程已启动,分配的文件数:{Files.Count}")
  1515. '列出 MC 文件夹中的各个版本文件夹
  1516. Dim VersionFolders As New List(Of String)
  1517. For Each McFolder In FolderList
  1518. Dim VersionsFolder As New DirectoryInfo(McFolder & "versions\")
  1519. If VersionsFolder.Exists() Then
  1520. For Each VersionFolder In VersionsFolder.GetDirectories
  1521. VersionFolders.Add(VersionFolder.FullName & "\")
  1522. Next
  1523. End If
  1524. Next
  1525. '处理每个文件
  1526. For Each File As NetFile In Files
  1527. Dim Target As String = CheckExistingFile(FolderList, VersionFolders, File)
  1528. If File.State >= NetState.WaitingToDownload Then Return '中断
  1529. '已找到相同文件
  1530. If Target IsNot Nothing Then
  1531. File.IsCopy = True
  1532. Dim RetryCount As Integer = 0
  1533. Retry:
  1534. Try
  1535. If Target <> File.LocalPath Then
  1536. Log($"[Download] 复制已存在的文件:{Target} → {File.LocalPath}")
  1537. CopyFile(Target, File.LocalPath)
  1538. End If
  1539. File.Finish(False)
  1540. Continue For
  1541. Catch ex As Exception
  1542. RetryCount += 1
  1543. Log(ex, $"复制已存在的文件失败,第 {RetryCount} 次重试({Target} → {File.LocalPath})")
  1544. If RetryCount < 3 Then
  1545. Thread.Sleep(2000)
  1546. GoTo Retry
  1547. End If
  1548. End Try
  1549. End If
  1550. '回退到下载
  1551. SyncLock LockState
  1552. File.State = NetState.WaitingToDownload
  1553. File.IsCopy = False
  1554. End SyncLock
  1555. Next
  1556. Catch ex As Exception
  1557. OnFail(New List(Of Exception) From {New Exception("下载已存在文件查找失败", ex)})
  1558. End Try
  1559. End Sub
  1560. Private Function CheckExistingFile(FolderList As List(Of String), VersionFolders As List(Of String), File As NetFile) As String
  1561. '目标文件已存在
  1562. If File.Check.Check(File.LocalPath) Is Nothing Then Return File.LocalPath
  1563. '没有可用的检查规则,只能开始下载
  1564. If File.Check.Hash Is Nothing AndAlso File.Check.ActualSize < 0 Then Return Nothing
  1565. '大致判断文件类别
  1566. Dim TypeIndexes =
  1567. {"\assets\", "\libraries\", "\versions\", "\mods\", "\coremods\", "\lib\", "\resourcepacks\", "\texturepacks\", "\shaderpacks\"}.
  1568. Select(Function(FolderName) (FolderName, File.LocalPath.IndexOfF(FolderName, True))).
  1569. Where(Function(kv) kv.Item2 >= 0).ToList
  1570. If Not TypeIndexes.Any Then
  1571. If File.LocalName.EndsWithF(".jar") Then
  1572. TypeIndexes.Add(("\versions\", 1)) '总是对 jar 进行版本文件检查,以包括另存为 jar 的情况
  1573. Else
  1574. Return Nothing
  1575. End If
  1576. End If
  1577. Dim Type = TypeIndexes.MaxOf(Function(kv) kv.Item2).FolderName.TrimStart("\"c)
  1578. '根据类别进行查找
  1579. Static Sizes As New SafeDictionary(Of String, Long)
  1580. Select Case Type
  1581. Case "assets\", "libraries\"
  1582. 'assets/libraries:查找 MC 文件夹下的相同路径
  1583. For Each Folder In FolderList
  1584. Dim Candidate = Folder & Type & File.LocalPath.AfterFirst(Type)
  1585. If File.Check.Check(Candidate) Is Nothing Then Return Candidate
  1586. Next
  1587. Case "versions\"
  1588. '版本 jar 或 json:查找 MC 文件夹下的各个版本文件夹
  1589. For Each VersionFolder In VersionFolders
  1590. For Each Candidate In Directory.GetFiles(VersionFolder,
  1591. "*." & GetFileNameFromPath(File.LocalPath).AfterLast(".").ToLower, SearchOption.TopDirectoryOnly)
  1592. '快速进行大小校验
  1593. If Not Sizes.ContainsKey(Candidate) Then Sizes(Candidate) = New FileInfo(Candidate).Length
  1594. If File.Check.ActualSize <> Sizes(Candidate) Then Continue For
  1595. 'Hash 校验
  1596. If File.Check.Check(Candidate) Is Nothing Then Return Candidate
  1597. Next
  1598. Next
  1599. Case Else
  1600. '社区资源
  1601. If File.Check.ActualSize < 0 OrElse File.Check.Hash Is Nothing Then Return Nothing '必须要求指定了文件大小和 Hash
  1602. For Each Folder In FolderList.Concat(VersionFolders)
  1603. Dim TargetFolder = Folder & Type
  1604. If Not Directory.Exists(TargetFolder) Then Continue For
  1605. For Each Candidate In Directory.GetFiles(TargetFolder)
  1606. '快速进行大小校验
  1607. If Not Sizes.ContainsKey(Candidate) Then Sizes(Candidate) = New FileInfo(Candidate).Length
  1608. If File.Check.ActualSize <> Sizes(Candidate) Then Continue For
  1609. 'Hash 校验
  1610. If File.Check.Check(Candidate) Is Nothing Then Return Candidate
  1611. Next
  1612. Next
  1613. End Select
  1614. Return Nothing
  1615. End Function
  1616. Public Sub OnFileFinish(File As NetFile)
  1617. '要求全部文件完成
  1618. SyncLock FileRemainLock
  1619. FileRemain -= 1
  1620. If FileRemain > 0 Then Return
  1621. End SyncLock
  1622. OnFinish()
  1623. End Sub
  1624. Public Sub OnFinish()
  1625. RaisePreviewFinish()
  1626. SyncLock LockState
  1627. If State > LoadState.Loading Then Return
  1628. State = LoadState.Finished
  1629. End SyncLock
  1630. End Sub
  1631. Public Sub OnFileFail(File As NetFile)
  1632. '将下载源的错误加入主错误列表
  1633. For Each Source In File.Sources
  1634. If Not IsNothing(Source.Ex) Then File.Ex.Add(Source.Ex)
  1635. Next
  1636. OnFail(File.Ex)
  1637. End Sub
  1638. Public Overrides Sub Failed(Ex As Exception)
  1639. OnFail(New List(Of Exception) From {Ex})
  1640. End Sub
  1641. Public Sub OnFail(ExList As List(Of Exception))
  1642. SyncLock LockState
  1643. If State >= LoadState.Finished Then Return
  1644. If ExList Is Nothing OrElse Not ExList.Any() Then ExList = New List(Of Exception) From {New Exception("未知错误!")}
  1645. '寻找有效的错误信息
  1646. Dim UsefulExs = ExList.Where(Function(e) TypeOf e IsNot OperationCanceledException AndAlso TypeOf e IsNot TimeoutException AndAlso TypeOf e IsNot ThreadInterruptedException).ToList
  1647. [Error] = If(UsefulExs.FirstOrDefault, ExList.FirstOrDefault)
  1648. '获取实际失败的文件
  1649. For Each File In Files
  1650. If File.State <> NetState.Interrupted Then Continue For
  1651. If File.Sources.All(Function(s) TypeOf s.Ex Is OperationCanceledException OrElse TypeOf s.Ex Is TimeoutException OrElse TypeOf s.Ex Is ThreadInterruptedException) Then Continue For
  1652. Dim Detail As String = Join(File.Sources.Select(Function(s) $"{If(s.Ex Is Nothing, "无错误信息。", s.Ex.GetBrief())}({s.Url})"), vbCrLf)
  1653. [Error] = New Exception("文件下载失败:" & File.LocalPath & vbCrLf &
  1654. "各下载源的错误如下:" & vbCrLf & Detail, [Error])
  1655. '上报
  1656. Telemetry("文件下载失败",
  1657. "FileName", File.LocalName,
  1658. "Exception", Detail)
  1659. Exit For
  1660. Next
  1661. '在设置 Error 对象后再更改为失败,避免 WaitForExit 无法捕获错误
  1662. State = LoadState.Failed
  1663. End SyncLock
  1664. '中断所有文件
  1665. For Each TaskFile In Files
  1666. TaskFile.Abort(Me)
  1667. Next
  1668. '在退出同步锁后再进行日志输出
  1669. Dim ErrOutput As New List(Of String)
  1670. For Each Ex As Exception In ExList
  1671. ErrOutput.Add(Ex.GetDetail())
  1672. Next
  1673. Log("[Download] " & Join(ErrOutput.Distinct.ToArray, vbCrLf))
  1674. End Sub
  1675. Public Overrides Sub Abort()
  1676. SyncLock LockState
  1677. If State >= LoadState.Finished Then Return
  1678. State = LoadState.Aborted
  1679. End SyncLock
  1680. Log("[Download] " & Name & " 已取消!")
  1681. '中断所有文件
  1682. For Each TaskFile In Files
  1683. TaskFile.Abort(Me)
  1684. Next
  1685. End Sub
  1686. End Class
  1687. Public NetManager As New NetManagerClass
  1688. ''' <summary>
  1689. ''' 下载文件管理。
  1690. ''' </summary>
  1691. Public Class NetManagerClass
  1692. #Region "属性"
  1693. ''' <summary>
  1694. ''' 需要下载的文件。为“本地地址 - 文件对象”键值对。
  1695. ''' </summary>
  1696. Public Files As New Dictionary(Of String, NetFile)
  1697. Public ReadOnly LockFiles As New Object
  1698. ''' <summary>
  1699. ''' 当前的所有下载任务。
  1700. ''' </summary>
  1701. Public Tasks As New SafeList(Of LoaderDownload)
  1702. ''' <summary>
  1703. ''' 已下载完成的大小。
  1704. ''' </summary>
  1705. Public Property DownloadDone As Long
  1706. Get
  1707. Return _DownloadDone
  1708. End Get
  1709. Set(value As Long)
  1710. SyncLock LockDone
  1711. _DownloadDone = value
  1712. End SyncLock
  1713. End Set
  1714. End Property
  1715. Private _DownloadDone As Long = 0
  1716. Private ReadOnly LockDone As New Object
  1717. ''' <summary>
  1718. ''' 尚未完成下载的文件数。
  1719. ''' </summary>
  1720. Public FileRemain As Integer = 0
  1721. Public ReadOnly LockRemain As New Object
  1722. '这些属性由 RefreshStat 刷新
  1723. ''' <summary>
  1724. ''' 当前的全局下载速度,单位为 Byte / 秒。
  1725. ''' </summary>
  1726. Public Speed As Long = 0
  1727. Public ReadOnly Uuid As Integer = GetUuid()
  1728. #End Region
  1729. ''' <summary>
  1730. ''' 进度与下载速度由下载管理线程每隔约 0.1 秒刷新一次。
  1731. ''' </summary>
  1732. Private Sub RefreshStat()
  1733. Try
  1734. Dim DeltaTime As Long = GetTimeMs() - RefreshStatLast
  1735. If DeltaTime = 0 Then Return
  1736. RefreshStatLast += DeltaTime
  1737. #Region "刷新整体速度"
  1738. '计算瞬时速度
  1739. Static SpeedLast As New List(Of Long) '记录至多最近 30 次下载速度的记录,较新的在前面
  1740. Static SpeedLastDone As Long = 0 '上次记速时的已下载大小
  1741. Dim ActualSpeed As Double = Math.Max(0, (DownloadDone - SpeedLastDone) / (DeltaTime / 1000))
  1742. SpeedLast.Insert(0, ActualSpeed)
  1743. If SpeedLast.Count >= 31 Then SpeedLast.RemoveAt(30)
  1744. SpeedLastDone = DownloadDone
  1745. '计算用于显示的速度
  1746. Dim SpeedSum As Long = 0, SpeedDiv As Long = 0, Weight = SpeedLast.Count
  1747. For Each SpeedRecord In SpeedLast
  1748. SpeedSum += SpeedRecord * Weight
  1749. SpeedDiv += Weight
  1750. Weight -= 1
  1751. Next
  1752. Speed = If(SpeedDiv > 0, SpeedSum / SpeedDiv, 0)
  1753. '计算新的速度下限
  1754. Dim Limit As Long = 0
  1755. If SpeedLast.Count >= 10 Then Limit = SpeedLast.Take(10).Average * 0.85 '取近 1 秒的平均速度的 85%
  1756. If Limit > NetTaskSpeedLimitLow Then
  1757. NetTaskSpeedLimitLow = Limit
  1758. Log("[Download] " & "速度下限已提升到 " & GetString(Limit))
  1759. End If
  1760. #End Region
  1761. #Region "刷新下载任务属性"
  1762. For Each Task In Tasks
  1763. Task.RefreshStat()
  1764. Next
  1765. #End Region
  1766. Catch ex As Exception
  1767. Log(ex, "刷新下载公开属性失败")
  1768. End Try
  1769. End Sub
  1770. Private RefreshStatLast As Long
  1771. ''' <summary>
  1772. ''' 启动监控线程,用于新增下载线程。
  1773. ''' </summary>
  1774. Private Sub StartManager()
  1775. Static IsStarted As Boolean = False
  1776. If IsStarted Then Return
  1777. IsStarted = True
  1778. Dim ThreadStarter =
  1779. Sub(Id As Integer) '0 或 1
  1780. Try
  1781. While True
  1782. Thread.Sleep(20)
  1783. '获取文件列表
  1784. Dim AllFiles As List(Of NetFile)
  1785. SyncLock LockFiles
  1786. If Id = 0 AndAlso FileRemain = 0 AndAlso Files.Any() Then Files.Clear() '若已完成,则清空
  1787. AllFiles = Files.Values.ToList()
  1788. End SyncLock
  1789. Dim WaitingFiles As New List(Of NetFile)
  1790. Dim OngoingFiles As New List(Of NetFile)
  1791. For Each File As NetFile In AllFiles
  1792. If File.Uuid Mod 2 = Id Then Continue For
  1793. If File.State = NetState.WaitingToDownload Then
  1794. WaitingFiles.Add(File)
  1795. ElseIf File.State < NetState.Merging Then
  1796. OngoingFiles.Add(File)
  1797. End If
  1798. Next
  1799. '为等待中的文件开始线程
  1800. For Each File As NetFile In WaitingFiles
  1801. If NetTaskThreadCount >= NetTaskThreadLimit Then Continue While '最大线程数检查
  1802. Dim NewThread = File.TryBeginThread()
  1803. If NewThread IsNot Nothing AndAlso NewThread.Source.Url.Contains("bmclapi") Then Thread.Sleep(100) '减少 BMCLAPI 请求频率
  1804. Next
  1805. '为进行中的文件追加线程
  1806. If Speed >= NetTaskSpeedLimitLow Then Continue While '下载速度足够,无需新增
  1807. For Each File As NetFile In OngoingFiles
  1808. If NetTaskThreadCount >= NetTaskThreadLimit Then Continue While '最大线程数检查
  1809. '线程种类计数
  1810. Dim PreparingCount = 0, DownloadingCount = 0
  1811. If File.Threads IsNot Nothing Then
  1812. For Each Thread As NetThread In File.Threads.ToList
  1813. If Thread.State < NetState.Downloading Then
  1814. PreparingCount += 1
  1815. ElseIf Thread.State = NetState.Downloading Then
  1816. DownloadingCount += 1
  1817. End If
  1818. Next
  1819. End If
  1820. '新增线程
  1821. If PreparingCount > DownloadingCount Then Continue For '准备中的线程已多于下载中的线程,不再新增
  1822. Dim NewThread = File.TryBeginThread()
  1823. If NewThread IsNot Nothing AndAlso NewThread.Source.Url.Contains("bmclapi") Then Thread.Sleep(100) '减少 BMCLAPI 请求频率
  1824. Next
  1825. End While
  1826. Catch ex As Exception
  1827. Log(ex, $"下载管理启动线程 {Id} 出错", LogLevel.Critical)
  1828. End Try
  1829. End Sub
  1830. RunInNewThread(Sub() ThreadStarter(0), "NetManager ThreadStarter 0")
  1831. RunInNewThread(Sub() ThreadStarter(1), "NetManager ThreadStarter 1")
  1832. RunInNewThread(
  1833. Sub()
  1834. Try
  1835. Dim NextTick As Long = GetTimeMs()
  1836. While True
  1837. '增加限速余量
  1838. If NetTaskSpeedLimitHigh > 0 Then NetTaskSpeedLimitLeft = NetTaskSpeedLimitHigh / 10
  1839. '刷新公开属性
  1840. RefreshStat()
  1841. '等待 100 ms
  1842. NextTick += 100
  1843. Dim SleepTime = NextTick - GetTimeMs()
  1844. If SleepTime > 0 Then
  1845. Thread.Sleep(SleepTime)
  1846. Else
  1847. NextTick = GetTimeMs() '超时,直接追帧,不等待
  1848. End If
  1849. End While
  1850. Catch ex As Exception
  1851. Log(ex, "下载管理刷新线程出错", LogLevel.Critical)
  1852. End Try
  1853. End Sub, "NetManager StatRefresher")
  1854. End Sub
  1855. Private DownloadCacheLock As New Object
  1856. Private IsDownloadCacheCleared As Boolean = False
  1857. ''' <summary>
  1858. ''' 开始一个下载任务。
  1859. ''' </summary>
  1860. Public Sub Start(Task As LoaderDownload)
  1861. StartManager()
  1862. '清理缓存
  1863. SyncLock DownloadCacheLock '防止同时开启多个下载任务时重复清理
  1864. If Not IsDownloadCacheCleared Then
  1865. Try
  1866. Log("[Net] 开始清理下载缓存")
  1867. DeleteDirectory(PathTemp & "Download")
  1868. Log("[Net] 下载缓存已清理")
  1869. Catch ex As Exception
  1870. Log(ex, "清理下载缓存失败")
  1871. End Try
  1872. End If
  1873. IsDownloadCacheCleared = True
  1874. End SyncLock
  1875. '文件处理
  1876. SyncLock LockFiles
  1877. '添加每个文件
  1878. For i = 0 To Task.Files.Count - 1
  1879. Dim File = Task.Files(i)
  1880. If Files.ContainsKey(File.LocalPath) Then
  1881. '已有该文件
  1882. If Files(File.LocalPath).State >= NetState.Finished Then
  1883. '该文件已经下载过一次,且下载完成
  1884. '将已下载的文件替换成当前文件,重新下载
  1885. File.Tasks.Add(Task)
  1886. Files(File.LocalPath) = File
  1887. SyncLock LockRemain
  1888. FileRemain += 1
  1889. End SyncLock
  1890. If ModeDebug Then Log($"[Download] {File.LocalName}:重新加入下载列表,目标 {File.LocalPath}")
  1891. Else
  1892. '该文件正在下载中
  1893. '将当前文件替换成下载中的文件,即两个任务指向同一个文件
  1894. File = Files(File.LocalPath)
  1895. File.Tasks.Add(Task)
  1896. End If
  1897. Else
  1898. '没有该文件
  1899. File.Tasks.Add(Task)
  1900. Files.Add(File.LocalPath, File)
  1901. SyncLock LockRemain
  1902. FileRemain += 1
  1903. End SyncLock
  1904. If ModeDebug Then Log($"[Download] {File.LocalName}:已加入下载列表,目标 {File.LocalPath}")
  1905. End If
  1906. Task.Files(i) = File '回设
  1907. Next
  1908. End SyncLock
  1909. Tasks.Add(Task)
  1910. End Sub
  1911. End Class
  1912. ''' <summary>
  1913. ''' 是否有正在进行中、需要在下载管理页面显示的下载任务?
  1914. ''' </summary>
  1915. Public Function HasDownloadingTask(Optional IgnoreCustomDownload As Boolean = False) As Boolean
  1916. For Each Task In LoaderTaskbar.ToList()
  1917. If (Task.Show AndAlso Task.State = LoadState.Loading) AndAlso
  1918. (Not IgnoreCustomDownload OrElse Not Task.Name.ToString.Contains("自定义下载")) Then
  1919. Return True
  1920. End If
  1921. Next
  1922. Return False
  1923. End Function
  1924. #End Region
  1925. #Region "端口"
  1926. ''' <summary>
  1927. ''' 随机获取单个可用的端口。
  1928. ''' </summary>
  1929. Public Function FindFreePort() As Integer
  1930. Dim Listener As New TcpListener(IPAddress.Loopback, 0)
  1931. Listener.Start()
  1932. Dim port As Integer = CType(Listener.LocalEndpoint, IPEndPoint).Port
  1933. Listener.Stop()
  1934. Return port
  1935. End Function
  1936. ''' <summary>
  1937. ''' 获取当前已被占用的端口列表。
  1938. ''' </summary>
  1939. Public Function GetUsedPorts() As List(Of Integer)
  1940. Dim IPProperties = NetworkInformation.IPGlobalProperties.GetIPGlobalProperties()
  1941. Dim UsedPorts As New List(Of Integer)
  1942. UsedPorts.AddRange(IPProperties.GetActiveTcpListeners().Select(Function(ep) ep.Port))
  1943. UsedPorts.AddRange(IPProperties.GetActiveUdpListeners().Select(Function(ep) ep.Port))
  1944. UsedPorts.AddRange(IPProperties.GetActiveTcpConnections().Select(Function(conn) conn.LocalEndPoint.Port))
  1945. Return UsedPorts.Distinct().ToList()
  1946. End Function
  1947. ''' <summary>
  1948. ''' 寻找数个连续编号的可用端口。
  1949. ''' </summary>
  1950. Public Function FindFreePorts(ConsecutiveCount As Integer, ParamArray ExtraBlackLists As Integer()) As List(Of Integer)
  1951. Dim UsedPorts = GetUsedPorts().Concat(ExtraBlackLists)
  1952. For port = 12000 + RandomInteger(0, 1000) To 65000 - ConsecutiveCount
  1953. Dim Range = Enumerable.Range(port, ConsecutiveCount)
  1954. If Not Range.Any(Function(p) UsedPorts.Contains(p)) Then Return Range.ToList
  1955. Next
  1956. Throw New Exception("未能找到可用的端口!")
  1957. End Function
  1958. #End Region
  1959. ''' <summary>
  1960. ''' 测试 Ping。失败则返回 -1。
  1961. ''' </summary>
  1962. Public Function Ping(Ip As String, Optional Timeout As Integer = 10000, Optional MakeLog As Boolean = True) As Integer
  1963. Dim PingResult As NetworkInformation.PingReply
  1964. Try
  1965. PingResult = (New NetworkInformation.Ping).Send(Ip)
  1966. Catch ex As Exception
  1967. If MakeLog Then Log("[Net] Ping " & Ip & " 失败:" & ex.Message)
  1968. Return -1
  1969. End Try
  1970. If PingResult.Status = NetworkInformation.IPStatus.Success Then
  1971. If MakeLog Then Log("[Net] Ping " & Ip & " 结束:" & PingResult.RoundtripTime & "ms")
  1972. Return PingResult.RoundtripTime
  1973. Else
  1974. If MakeLog Then Log("[Net] Ping " & Ip & " 失败")
  1975. Return -1
  1976. End If
  1977. End Function
  1978. ''' <summary>
  1979. ''' 判断某个 Exception 是否为网络问题所导致。
  1980. ''' </summary>
  1981. <Extension> Public Function IsNetworkRelated(Ex As Exception) As Boolean
  1982. '提取 Message(不能用 GetDetail,因为堆栈的方法参数中就有 timeout 字样)
  1983. Dim Detail As String = ""
  1984. Dim InnerEx As Exception = Ex
  1985. Detail += InnerEx.Message
  1986. Do Until InnerEx.InnerException Is Nothing
  1987. InnerEx = InnerEx.InnerException
  1988. Detail += InnerEx.Message
  1989. Loop
  1990. '判断
  1991. If Detail.Contains("(403)") Then Return False
  1992. Return {
  1993. "(408)", "超时", "timeout", "网络请求失败", "连接尝试失败", "远程主机强迫关闭了", "远程方已关闭传输流", "未能解析此远程名称",
  1994. "由于目标计算机积极拒绝", "基础连接已经关闭"
  1995. }.Any(Function(k) Detail.ContainsF(k, True))
  1996. End Function
  1997. End Module