GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-05-11 18:11:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: 下载eapi文件限制郁闷

Browse Source
This commit is contained in:
余泓铮
2024-08-14 16:30:06 +08:00
parent ecb0b510c4
commit c6b6f3b2e5
1 changed files with 13 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
hrp/internal/builtin/utils.go
View File

@@ -13,6 +13,7 @@ import (
"math/rand"
"net"
"net/http"
"net/url"
"os"
"path/filepath"
"reflect"
@@ -512,8 +513,17 @@ func GetCurrentDay() string {
return formattedDate
}
func DownloadFile(filePath string, url string) error {
log.Info().Str("filePath", filePath).Str("url", url).Msg("download file")
func DownloadFile(filePath string, fileUrl string) error {
log.Info().Str("filePath", filePath).Str("url", fileUrl).Msg("download file")
parsedURL, err := url.Parse(fileUrl)
if err != nil {
return err
}
if parsedURL.Host != "gtf-eapi-cn.bytedance.com" {
return errors.New("invalid domain: must be gtf-eapi-cn.bytedance.com")
}
out, err := os.Create(filePath)
if err != nil {
return err
@@ -521,7 +531,7 @@ func DownloadFile(filePath string, url string) error {
defer out.Close()
// 创建一个新的 HTTP 请求
req, err := http.NewRequest("GET", url, nil)
req, err := http.NewRequest("GET", fileUrl, nil)
if err != nil {
return err
}