Skip to content

Commit

Permalink
fix huya (#764)
Browse files Browse the repository at this point in the history 
* fix huya

* fix huya 获取真实roomid
  • Loading branch information
@redstone1994
redstone1994 authored Jul 9, 2024
1 parent 649285c commit c5b74de
Showing 1 changed file with 65 additions and 149 deletions.
214 changes: 65 additions & 149 deletions src/live/huya/huya.go
View file
Original file line number Diff line number Diff line change
@@ -1,31 +1,23 @@
package huya

import (
"bytes"
"crypto/md5"
"encoding/base64"
"encoding/hex"
"fmt"
"html/template"
"math/rand"
"net/http"
"net/url"
"strconv"
"strings"
"time"

"github.com/hr3lxphr6j/requests"
"github.com/tidwall/gjson"

"github.com/hr3lxphr6j/bililive-go/src/live"
"github.com/hr3lxphr6j/bililive-go/src/live/internal"
"github.com/hr3lxphr6j/bililive-go/src/pkg/utils"
"github.com/hr3lxphr6j/requests"
"github.com/tidwall/gjson"
"net/http"
"net/url"
)

const (
domain = "www.huya.com"
cnName = "虎牙"
userAgent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/125.0.0.0 Safari/537.36"
uaApp = "Mozilla/5.0 (iPhone; CPU iPhone OS 17_0 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Mobile/15E148 MicroMessenger/8.0.49(0x18003137) NetType/WIFI Language/zh_CN WeChat/8.0.49.33 CFNetwork/1474 Darwin/23.0.0"
)

func init() {
Expand All @@ -45,8 +37,35 @@ type Live struct {
lastCdnIndex int
}

func (l *Live) GetInfo() (info *live.Info, err error) {
resp, err := requests.Get(l.Url.String(), live.CommonUserAgent)
func (l *Live) getDate() (result *gjson.Result, err error) {
html, err := requests.Get(l.Url.String(), live.CommonUserAgent)
if err != nil {
return nil, err
}
if html.StatusCode != http.StatusOK {
return nil, live.ErrRoomNotExist
}
htmlBody, err := html.Text()
if err != nil {
return nil, err
}
strFilter := utils.NewStringFilterChain(utils.ParseUnicode, utils.UnescapeHTMLEntity)
rjson := strFilter.Do(utils.Match1(`stream: (\{"data".*?),"iWebDefaultBitRate"`, htmlBody)) + "}"
gj := gjson.Parse(rjson)

roomId := gj.Get("data.0.gameLiveInfo.profileRoom").String()
params := make(map[string]string)
params["m"] = "Live"
params["do"] = "profileRoom"
params["roomid"] = roomId
params["showSecret"] = "1"

headers := make(map[string]interface{})
headers["User-Agent"] = uaApp
headers["xweb_xhr"] = "1"
headers["referer"] = "https://servicewechat.com/wx74767bf0b684f7d3/301/page-frame.html"
headers["accept-language"] = "zh-CN,zh;q=0.9"
resp, err := requests.Get("https://mp.huya.com/cache.php", requests.Headers(headers), requests.Queries(params), requests.UserAgent(uaApp))
if err != nil {
return nil, err
}
Expand All @@ -57,25 +76,32 @@ func (l *Live) GetInfo() (info *live.Info, err error) {
if err != nil {
return nil, err
}
res := gjson.Parse(body)
return &res, nil
}

if res := utils.Match1("哎呀，虎牙君找不到这个主播，要不搜索看看？", body); res != "" {
func (l *Live) GetInfo() (info *live.Info, err error) {
res, err := l.getDate()
if err != nil {
return nil, err
}
if res := utils.Match1("该主播不存在！", res.String()); res != "" {
return nil, live.ErrRoomNotExist
}

if strings.Contains(body, "该主播涉嫌违规，正在整改中") {
return &live.Info{
Live: l,
HostName: "该主播涉嫌违规，正在整改中",
RoomName: "该主播涉嫌违规，正在整改中",
Status: false,
}, nil
}
//if strings.Contains(body, "该主播涉嫌违规，正在整改中") {
// return &live.Info{
// Live: l,
// HostName: "该主播涉嫌违规，正在整改中",
// RoomName: "该主播涉嫌违规，正在整改中",
// Status: false,
// }, nil
//}

var (
strFilter = utils.NewStringFilterChain(utils.ParseUnicode, utils.UnescapeHTMLEntity)
hostName = strFilter.Do(utils.Match1(`"nick":"([^"]*)"`, body))
roomName = strFilter.Do(utils.Match1(`"introduction":"([^"]*)"`, body))
status = strFilter.Do(utils.Match1(`"isOn":([^,]*),`, body))
hostName = res.Get("data.liveData.nick").String()
roomName = res.Get("data.liveData.introduction").String()
status = res.Get("data.realLiveStatus").String()
)

if hostName == "" || roomName == "" || status == "" {
Expand All @@ -86,7 +112,7 @@ func (l *Live) GetInfo() (info *live.Info, err error) {
Live: l,
HostName: hostName,
RoomName: roomName,
Status: status == "true",
Status: status == "ON",
}
return info, nil
}
Expand All @@ -96,133 +122,23 @@ func GetMD5Hash(text string) string {
return hex.EncodeToString(hash[:])
}

type urlQueryParams struct {
WsSecret string
WsTime string
Seqid string
Ctype string
Ver string
Fs string
U string
T string
Sv string
Sdk_sid string
Codec string
}

func parseAntiCode(anticode string, uid int64, streamName string) (string, error) {
qr, err := url.ParseQuery(anticode)
if err != nil {
return "", err
}
resultTemplate := template.Must(template.New("urlQuery").Parse(
"wsSecret={{.WsSecret}}" +
"&wsTime={{.WsTime}}" +
"&seqid={{.Seqid}}" +
"&ctype={{.Ctype}}" +
"&ver={{.Ver}}" +
"&fs={{.Fs}}" +
"&u={{.U}}" +
"&t={{.T}}" +
"&sv={{.Sv}}" +
"&sdk_sid={{.Sdk_sid}}" +
"&codec={{.Codec}}",
))
timeNow := time.Now().Unix() * 1000
resultParams := urlQueryParams{
WsSecret: "",
WsTime: qr.Get("wsTime"),
Seqid: strconv.FormatInt(timeNow+uid, 10),
Ctype: qr.Get("ctype"),
Ver: "1",
Fs: qr.Get("fs"),
U: strconv.FormatInt(uid, 10),
T: "100",
Sv: "2405220949",
Sdk_sid: strconv.FormatInt(uid, 10),
Codec: "264",
}
ss := GetMD5Hash(fmt.Sprintf("%s|%s|%s", resultParams.Seqid, resultParams.Ctype, resultParams.T))

decodeString, _ := base64.StdEncoding.DecodeString(qr.Get("fm"))
fm := string(decodeString)
fm = strings.ReplaceAll(fm, "$0", resultParams.U)
fm = strings.ReplaceAll(fm, "$1", streamName)
fm = strings.ReplaceAll(fm, "$2", ss)
fm = strings.ReplaceAll(fm, "$3", resultParams.WsTime)

resultParams.WsSecret = GetMD5Hash(fm)
var buf bytes.Buffer
if err := resultTemplate.Execute(&buf, resultParams); err != nil {
return "", err
}
return buf.String(), nil
}

func (l *Live) GetStreamUrls() (us []*url.URL, err error) {
resp, err := requests.Get(l.Url.String(), requests.UserAgent(userAgent))
if err != nil {
return nil, err
}
if resp.StatusCode != http.StatusOK {
return nil, fmt.Errorf("status code: %d", resp.StatusCode)
}
body, err := resp.Text()
data, err := l.getDate()
if err != nil {
return nil, err
}
sFlvUrl := data.Get("data.stream.baseSteamInfoList.0.sFlvUrl").String()
sStreamName := data.Get("data.stream.baseSteamInfoList.0.sStreamName").String()
sFlvUrlSuffix := data.Get("data.stream.baseSteamInfoList.0.sFlvUrlSuffix").String()
sFlvAntiCode := data.Get("data.stream.baseSteamInfoList.0.sFlvAntiCode").String()
streamUrl := fmt.Sprintf("%s/%s.%s?%s", sFlvUrl, sStreamName, sFlvUrlSuffix, sFlvAntiCode)

tmpStrings := strings.Split(body, `stream: `)
if len(tmpStrings) < 2 {
return nil, fmt.Errorf("stream json info not found")
}
tmpStreamJsonRawString := strings.Split(tmpStrings[1], `};`)
if len(tmpStreamJsonRawString) < 1 {
return nil, fmt.Errorf("stream json info end not found. stream text: %s", tmpStrings[1])
}
streamJsonRawString := tmpStreamJsonRawString[0]
if !gjson.Valid(streamJsonRawString) {
return nil, fmt.Errorf("streamJsonRawString not valid")
}
streamJson := gjson.Parse(streamJsonRawString)
vMultiStreamInfoJson := streamJson.Get("vMultiStreamInfo").Array()
if len(vMultiStreamInfoJson) == 0 {
return nil, fmt.Errorf("vMultiStreamInfo not found")
}

streamInfoJsons := streamJson.Get("data.0.gameStreamInfoList").Array()
if len(streamInfoJsons) == 0 {
return nil, fmt.Errorf("gameStreamInfoList not found")
}
index := l.lastCdnIndex
if index >= len(streamInfoJsons) {
index = 0
}
l.lastCdnIndex = index + 1
gameStreamInfoJson := streamInfoJsons[index]
return getStreamUrlsFromGameStreamInfoJson(gameStreamInfoJson)
}

func getStreamUrlsFromGameStreamInfoJson(gameStreamInfoJson gjson.Result) (us []*url.URL, err error) {
// get streamName
sStreamName := gameStreamInfoJson.Get("sStreamName").String()
// get sFlvAntiCode
sFlvAntiCode := gameStreamInfoJson.Get("sFlvAntiCode").String()
// get sFlvUrl
sFlvUrl := gameStreamInfoJson.Get("sFlvUrl").String()
// get random uid
uid := rand.Int63n(99999999999) + 1200000000000

query, err := parseAntiCode(sFlvAntiCode, uid, sStreamName)
if err != nil {
return nil, err
}
tmpUrlString := fmt.Sprintf("%s/%s.flv?%s", sFlvUrl, sStreamName, query)
u, err := url.Parse(tmpUrlString)
res, err := utils.GenUrls(streamUrl)
if err != nil {
return nil, err
}
return []*url.URL{u}, nil
return res, nil

}

func (l *Live) GetPlatformCNName() string {
Expand All @@ -231,7 +147,7 @@ func (l *Live) GetPlatformCNName() string {

func (l *Live) GetHeadersForDownloader() map[string]string {
return map[string]string{
"User-Agent": userAgent,
"User-Agent": uaApp,
"Accept": `text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8`,
"Accept-Encoding": `gzip, deflate`,
"Accept-Language": `zh-CN,zh;q=0.8,en-US;q=0.5,en;q=0.3`,
Expand Down

0 comments on commit c5b74de

Please sign in to comment.