him411/UA3F
1
0
Fork 0
mirror of https://github.com/SunBK201/UA3F.git synced 2025-12-16 16:57:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

style: code tidy

Browse Source
This commit is contained in:
SunBK201 2025-11-08 17:05:20 +08:00
parent e17478f612
commit 1c97a0bdf0
2 changed files with 7 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/internal/sniff/packet.go → src/internal/rewrite/http.go
View File

@ -1,4 +1,4 @@
package sniff
package rewrite
import (
"bytes"
@ -17,9 +17,9 @@ func toLowerASCII(b byte) byte {
return b
}
// IndexFoldASCII performs case-insensitive search for needle in haystack (ASCII only)
// indexFoldASCII performs case-insensitive search for needle in haystack (ASCII only)
// Returns the first occurrence index or -1 if not found
func IndexFoldASCII(haystack, needle []byte) int {
func indexFoldASCII(haystack, needle []byte) int {
if len(needle) == 0 {
return 0
}
@ -46,10 +46,10 @@ func IndexFoldASCII(haystack, needle []byte) int {
return -1
}
// FindUserAgentInPayload searches for User-Agent header(s) in raw HTTP payload
// findUserAgentInPayload searches for User-Agent header(s) in raw HTTP payload
// Returns slice of (startPos, endPos) pairs for each User-Agent value found
// Returns empty slice if no User-Agent found, or if any UA is unterminated (missing \r)
func FindUserAgentInPayload(payload []byte) (positions [][2]int, unterminated bool) {
func findUserAgentInPayload(payload []byte) (positions [][2]int, unterminated bool) {
if len(payload) < len(uaTag) {
return nil, false
}
@ -60,7 +60,7 @@ func FindUserAgentInPayload(payload []byte) (positions [][2]int, unterminated bo
break
}
idx := IndexFoldASCII(payload[searchStart:], uaTag)
idx := indexFoldASCII(payload[searchStart:], uaTag)
if idx < 0 {
break
}

3
src/internal/rewrite/packet.go
View File

@ -5,7 +5,6 @@ import (
"github.com/google/gopacket/layers"
"github.com/sunbk201/ua3f/internal/log"
"github.com/sunbk201/ua3f/internal/sniff"
"github.com/sunbk201/ua3f/internal/statistics"
)
@ -72,7 +71,7 @@ func (r *Rewriter) buildReplacement(srcAddr, dstAddr string, originalUA string,
// Returns the rewritten payload and metadata about the operation
func (r *Rewriter) RewritePacketUserAgent(payload []byte, srcAddr, dstAddr string) (newPayload []byte, hasUA, modified, inWhitelist bool) {
// Find all User-Agent positions
positions, unterm := sniff.FindUserAgentInPayload(payload)
positions, unterm := findUserAgentInPayload(payload)
if unterm {
log.LogInfoWithAddr(srcAddr, dstAddr, "Unterminated User-Agent found, not rewriting")