Austin/mihomo
1
0
Fork 0
mirror of https://github.com/MetaCubeX/mihomo.git synced 2024-11-15 19:22:53 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: regexp

Browse Source
This commit is contained in:
Adlyq 2023-09-04 13:31:18 +08:00
parent 1f7a883bfc
commit a0fd6cfeea
2 changed files with 32 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
rewrite/handler.go
View File

@ -6,7 +6,6 @@ import (
"errors"
log "github.com/sirupsen/logrus"
"io"
"io/ioutil"
"net/http"
"net/textproto"
"strconv"
@ -150,19 +149,20 @@ func (*RewriteHandler) HandleResponse(session *mitm.Session) *http.Response {
return nil
}
body, err := mitm.DecodeLatin1(bytes.NewReader(b))
if err != nil {
return nil
}
//body, err := mitm.DecodeLatin1(bytes.NewReader(b))
//if err != nil {
// return nil
//}
newBody := rule.ReplaceSubPayload(body)
newBody := rule.ReplaceSubPayload(string(b))
modifiedBody, err := mitm.EncodeLatin1(newBody)
if err != nil {
return nil
}
//modifiedBody, err := mitm.EncodeLatin1(newBody)
//if err != nil {
// return nil
//}
modifiedBody := []byte(newBody)
response.Body = ioutil.NopCloser(bytes.NewReader(modifiedBody))
response.Body = io.NopCloser(bytes.NewReader(modifiedBody))
response.Header.Del("Content-Encoding")
response.ContentLength = int64(len(modifiedBody))
default:

19
rewrite/rewrite.go
View File

@ -61,16 +61,18 @@ func (r *RewriteRule) ReplaceURLPayload(matchSub []string) string {
func (r *RewriteRule) ReplaceSubPayload(oldData string) string {
payload := r.rulePayload
if r.ruleRegx == nil {
return oldData
}
sub, err := r.ruleRegx.FindStringMatch(oldData)
if err != nil {
return oldData
}
var groups []string
for err == nil && sub != nil {
var (
groups []string
sPayload = payload
)
for _, fg := range sub.Groups() {
groups = append(groups, fg.String())
}
@ -78,10 +80,15 @@ func (r *RewriteRule) ReplaceSubPayload(oldData string) string {
l := len(groups)
for i := 1; i < l; i++ {
payload = strings.ReplaceAll(payload, "$"+strconv.Itoa(i), groups[i])
sPayload = strings.Replace(payload, "$"+strconv.Itoa(i), groups[i], 1)
}
return strings.ReplaceAll(oldData, groups[0], payload)
oldData = strings.Replace(oldData, groups[0], sPayload, 1)
sub, err = r.ruleRegx.FindNextMatch(sub)
}
return oldData
}
func NewRewriteRule(urlRegx *regexp.Regexp, ruleType C.RewriteType, ruleRegx *regexp.Regexp, rulePayload string) *RewriteRule {