mirrors/XTLS_REALITY
0
0
Fork 0
mirror of https://github.com/XTLS/REALITY.git synced 2025-08-22 14:38:35 +00:00
Code

Fix hanging issue for some randomized fingerprints

Browse Source
This commit is contained in:
yuhan6665 2025-06-15 00:30:40 -04:00
parent 0d942695e3
commit f13edcc98b
2 changed files with 41 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
record_detect.go
View File

@ -26,13 +26,20 @@ func InitAllRecords(config *Config) {
func DetectPostHandshakeRecordsLens(config *Config, fingerprint string) map[string][]int {
GlobalPostHandshakeRecordsLock.Lock()
defer GlobalPostHandshakeRecordsLock.Unlock()
if GlobalPostHandshakeRecordsLens == nil {
GlobalPostHandshakeRecordsLens = make(map[string]map[string][]int)
}
if GlobalPostHandshakeRecordsLens[fingerprint] == nil {
GlobalPostHandshakeRecordsLens[fingerprint] = make(map[string][]int)
}
var pending []string
for sni := range config.ServerNames {
if (GlobalPostHandshakeRecordsLens[fingerprint][sni] == nil) {
pending = append(pending, sni)
}
}
GlobalPostHandshakeRecordsLock.Unlock()
for _, sni := range pending {
target, err := net.Dial("tcp", config.Dest)
if err != nil {
continue
@ -56,7 +63,6 @@ func DetectPostHandshakeRecordsLens(config *Config, fingerprint string) map[stri
}
io.Copy(io.Discard, uConn)
}
}
return GlobalPostHandshakeRecordsLens[fingerprint]
}
@ -77,10 +83,12 @@ func (c *DetectConn) Write(b []byte) (n int, err error) {
func (c *DetectConn) Read(b []byte) (n int, err error) {
if !c.CcsSent {
c.Conn.SetReadDeadline(time.Now().Add(2 * time.Second))
return c.Conn.Read(b)
}
c.Conn.SetReadDeadline(time.Now().Add(5 * time.Second))
c.Conn.SetReadDeadline(time.Now().Add(2 * time.Second))
data, _ := io.ReadAll(c.Conn)
GlobalPostHandshakeRecordsLock.Lock()
for {
if len(data) >= 5 && bytes.Equal(data[:3], []byte{23, 3, 3}) {
length := int(binary.BigEndian.Uint16(data[3:5])) + 5
@ -90,6 +98,10 @@ func (c *DetectConn) Read(b []byte) (n int, err error) {
break
}
}
if len(c.PostHandshakeRecordsLens[c.Sni]) == 0 {
c.PostHandshakeRecordsLens[c.Sni] = append(c.PostHandshakeRecordsLens[c.Sni], 0)
}
GlobalPostHandshakeRecordsLock.Unlock()
fmt.Printf("REALITY fingerprint probe: %v\tSni: %v\tlen(postHandshakeRecord): %v\n", c.Fingerprint, c.Sni, c.PostHandshakeRecordsLens[c.Sni])
return 0, io.EOF
}
@ -115,6 +127,7 @@ var ModernFingerprints = map[string]*utls.ClientHelloID{
"hellosafari_16_0": &utls.HelloSafari_16_0,
"hello360_11_0": &utls.Hello360_11_0,
"helloqq_11_1": &utls.HelloQQ_11_1,
"hellogolang": &utls.HelloGolang,
"hellorandomized": &utls.HelloRandomized,
"hellorandomizedalpn": &utls.HelloRandomizedALPN,

3
tls.go
View File

@ -375,6 +375,9 @@ func Server(ctx context.Context, conn net.Conn, config *Config) (*Conn, error) {
break
}
for _, length := range postHandshakeRecordsLens[hs.clientHello.serverName] {
if length == 0 {
break;
}
plainText := make([]byte, length-16)
plainText[0] = 23
plainText[1] = 3