|  | 
| 7 | 7 | 	"github.com/Sora233/DDBOT/proxy_pool" | 
| 8 | 8 | 	"github.com/Sora233/DDBOT/requests" | 
| 9 | 9 | 	"github.com/Sora233/DDBOT/utils" | 
| 10 |  | -	"github.com/guonaihong/gout" | 
| 11 | 10 | 	"github.com/samber/lo" | 
| 12 | 11 | 	"sort" | 
|  | 12 | +	"strconv" | 
| 13 | 13 | 	"strings" | 
| 14 | 14 | 	"time" | 
| 15 | 15 | ) | 
| @@ -40,30 +40,36 @@ func getWbi() (imgKey string, subKey string) { | 
| 40 | 40 | 	subKey = getKey(wbi.SubUrl) | 
| 41 | 41 | 	return | 
| 42 | 42 | } | 
|  | 43 | +func getMixinKey(orig string) string { | 
|  | 44 | +	var str strings.Builder | 
|  | 45 | +	for _, v := range mixinKeyEncTab { | 
|  | 46 | +		if v < len(orig) { | 
|  | 47 | +			str.WriteByte(orig[v]) | 
|  | 48 | +		} | 
|  | 49 | +	} | 
|  | 50 | +	return str.String()[:32] | 
|  | 51 | +} | 
| 43 | 52 | 
 | 
| 44 |  | -func signWbi(params gout.H) { | 
|  | 53 | +func signWbi(params map[string]string) map[string]string { | 
| 45 | 54 | 	imgKey, subKey := getWbi() | 
| 46 |  | -	var sb strings.Builder | 
| 47 |  | -	var orig = imgKey + subKey | 
| 48 |  | -	for _, r := range mixinKeyEncTab { | 
| 49 |  | -		sb.WriteRune(rune(orig[r])) | 
|  | 55 | +	mixinKey := getMixinKey(imgKey + subKey) | 
|  | 56 | +	currTime := strconv.FormatInt(time.Now().Unix(), 10) | 
|  | 57 | +	params["wts"] = currTime | 
|  | 58 | +	// Sort keys | 
|  | 59 | +	keys := make([]string, 0, len(params)) | 
|  | 60 | +	for k := range params { | 
|  | 61 | +		keys = append(keys, k) | 
| 50 | 62 | 	} | 
| 51 |  | -	salt := sb.String()[:32] | 
| 52 |  | -	params["wts"] = time.Now().Unix() | 
| 53 |  | -	p := lo.MapToSlice(params, func(key string, value any) lo.Tuple2[string, any] { | 
| 54 |  | -		return lo.Tuple2[string, any]{A: key, B: value} | 
| 55 |  | -	}) | 
| 56 |  | -	sort.Slice(p, func(i, j int) bool { | 
| 57 |  | -		return p[i].A < p[j].A | 
| 58 |  | -	}) | 
| 59 |  | - | 
| 60 |  | -	query := strings.Join(lo.Map(p, func(item lo.Tuple2[string, any], _ int) string { | 
| 61 |  | -		return fmt.Sprintf("%v=%v", item.A, item.B) | 
| 62 |  | -	}), "&") | 
| 63 |  | -	hash := md5.New() | 
| 64 |  | -	hash.Write([]byte(query + salt)) | 
| 65 |  | -	wbiSign := hex.EncodeToString(hash.Sum(nil)) | 
| 66 |  | -	params["w_rid"] = wbiSign | 
|  | 63 | +	sort.Strings(keys) | 
|  | 64 | +	// Build URL parameters | 
|  | 65 | +	var str strings.Builder | 
|  | 66 | +	for _, k := range keys { | 
|  | 67 | +		str.WriteString(fmt.Sprintf("%s=%s&", k, params[k])) | 
|  | 68 | +	} | 
|  | 69 | +	query := strings.TrimSuffix(str.String(), "&") | 
|  | 70 | +	hash := md5.Sum([]byte(query + mixinKey)) | 
|  | 71 | +	params["w_rid"] = hex.EncodeToString(hash[:]) | 
|  | 72 | +	return params | 
| 67 | 73 | } | 
| 68 | 74 | 
 | 
| 69 | 75 | func XWebInterfaceNav(login bool) (*WebInterfaceNavResponse, error) { | 
|  | 
0 commit comments