mirror of
				https://github.com/oneclickvirt/backtrace.git
				synced 2025-11-04 15:52:37 +08:00 
			
		
		
		
	fix:修复cdn检测可用性的漏洞
This commit is contained in:
		
							parent
							
								
									d8148ff279
								
							
						
					
					
						commit
						e3e938e152
					
				
							
								
								
									
										74
									
								
								bk/utils.go
									
									
									
									
									
								
							
							
						
						
									
										74
									
								
								bk/utils.go
									
									
									
									
									
								
							@ -33,6 +33,41 @@ func removeDuplicates(elements []string) []string {
 | 
				
			|||||||
	return result
 | 
						return result
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// checkCdn 检查CDN可用性,参考shell脚本的测试逻辑
 | 
				
			||||||
 | 
					func checkCdn(testUrl string) string {
 | 
				
			||||||
 | 
						client := req.C()
 | 
				
			||||||
 | 
						client.SetTimeout(6 * time.Second)
 | 
				
			||||||
 | 
						if model.EnableLoger {
 | 
				
			||||||
 | 
							InitLogger()
 | 
				
			||||||
 | 
							defer Logger.Sync()
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						for _, cdnUrl := range model.CdnList {
 | 
				
			||||||
 | 
							url := cdnUrl + testUrl
 | 
				
			||||||
 | 
							if model.EnableLoger {
 | 
				
			||||||
 | 
								Logger.Info(fmt.Sprintf("Testing CDN: %s", url))
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							resp, err := client.R().Get(url)
 | 
				
			||||||
 | 
							if err == nil {
 | 
				
			||||||
 | 
								b, err := io.ReadAll(resp.Body)
 | 
				
			||||||
 | 
								resp.Body.Close()
 | 
				
			||||||
 | 
								if err == nil && strings.Contains(string(b), "success") {
 | 
				
			||||||
 | 
									if model.EnableLoger {
 | 
				
			||||||
 | 
										Logger.Info(fmt.Sprintf("CDN available: %s", cdnUrl))
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
									return cdnUrl
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							if model.EnableLoger {
 | 
				
			||||||
 | 
								Logger.Info(fmt.Sprintf("CDN test failed: %s, error: %v", cdnUrl, err))
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							time.Sleep(500 * time.Millisecond)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						if model.EnableLoger {
 | 
				
			||||||
 | 
							Logger.Info("No CDN available, using direct connection")
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						return ""
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// getData 获取目标地址的文本内容
 | 
					// getData 获取目标地址的文本内容
 | 
				
			||||||
func getData(endpoint string) string {
 | 
					func getData(endpoint string) string {
 | 
				
			||||||
	client := req.C()
 | 
						client := req.C()
 | 
				
			||||||
@ -45,16 +80,22 @@ func getData(endpoint string) string {
 | 
				
			|||||||
		InitLogger()
 | 
							InitLogger()
 | 
				
			||||||
		defer Logger.Sync()
 | 
							defer Logger.Sync()
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	for _, baseUrl := range model.CdnList {
 | 
						
 | 
				
			||||||
		url := baseUrl + endpoint
 | 
						// 先测试CDN可用性
 | 
				
			||||||
 | 
						testUrl := "https://raw.githubusercontent.com/spiritLHLS/ecs/main/back/test"
 | 
				
			||||||
 | 
						cdnUrl := checkCdn(testUrl)
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						// 如果有可用的CDN,使用CDN获取数据
 | 
				
			||||||
 | 
						if cdnUrl != "" {
 | 
				
			||||||
 | 
							url := cdnUrl + endpoint
 | 
				
			||||||
 | 
							if model.EnableLoger {
 | 
				
			||||||
 | 
								Logger.Info(fmt.Sprintf("Using CDN: %s", url))
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
		resp, err := client.R().Get(url)
 | 
							resp, err := client.R().Get(url)
 | 
				
			||||||
		if err == nil {
 | 
							if err == nil {
 | 
				
			||||||
			defer resp.Body.Close()
 | 
								defer resp.Body.Close()
 | 
				
			||||||
			b, err := io.ReadAll(resp.Body)
 | 
								b, err := io.ReadAll(resp.Body)
 | 
				
			||||||
			if strings.Contains(string(b), "error") {
 | 
								if err == nil && !strings.Contains(string(b), "error") {
 | 
				
			||||||
				continue
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
			if err == nil {
 | 
					 | 
				
			||||||
				if model.EnableLoger {
 | 
									if model.EnableLoger {
 | 
				
			||||||
					Logger.Info(fmt.Sprintf("Received data length: %d", len(b)))
 | 
										Logger.Info(fmt.Sprintf("Received data length: %d", len(b)))
 | 
				
			||||||
				}
 | 
									}
 | 
				
			||||||
@ -62,9 +103,28 @@ func getData(endpoint string) string {
 | 
				
			|||||||
			}
 | 
								}
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
		if model.EnableLoger {
 | 
							if model.EnableLoger {
 | 
				
			||||||
			Logger.Info(fmt.Sprintf("HTTP request failed: %v", err))
 | 
								Logger.Info(fmt.Sprintf("CDN request failed: %v", err))
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						// CDN不可用,尝试直连
 | 
				
			||||||
 | 
						if model.EnableLoger {
 | 
				
			||||||
 | 
							Logger.Info(fmt.Sprintf("Trying direct connection: %s", endpoint))
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						resp, err := client.R().Get(endpoint)
 | 
				
			||||||
 | 
						if err == nil {
 | 
				
			||||||
 | 
							defer resp.Body.Close()
 | 
				
			||||||
 | 
							b, err := io.ReadAll(resp.Body)
 | 
				
			||||||
 | 
							if err == nil {
 | 
				
			||||||
 | 
								if model.EnableLoger {
 | 
				
			||||||
 | 
									Logger.Info(fmt.Sprintf("Received data length: %d", len(b)))
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								return string(b)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						if model.EnableLoger {
 | 
				
			||||||
 | 
							Logger.Info(fmt.Sprintf("Direct connection failed: %v", err))
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
	return ""
 | 
						return ""
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
				
			|||||||
		Loading…
	
		Reference in New Issue
	
	Block a user