forked from SagerNet/sing-quic
-
Notifications
You must be signed in to change notification settings - Fork 0
/
service_packet.go
43 lines (39 loc) · 1016 Bytes
/
service_packet.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
package hysteria
import (
"github.com/sagernet/sing/common"
E "github.com/sagernet/sing/common/exceptions"
)
func (s *serverSession[U]) loopMessages() {
for {
message, err := s.quicConn.ReceiveDatagram(s.ctx)
if err != nil {
s.closeWithError(E.Cause(err, "receive message"))
return
}
hErr := s.handleMessage(message)
if hErr != nil {
s.closeWithError(E.Cause(hErr, "handle message"))
return
}
}
}
func (s *serverSession[U]) handleMessage(data []byte) error {
message := allocMessage()
err := decodeUDPMessage(message, data)
if err != nil {
message.release()
return E.Cause(err, "decode UDP message")
}
return s.handleUDPMessage(message)
}
func (s *serverSession[U]) handleUDPMessage(message *udpMessage) error {
s.udpAccess.RLock()
udpConn, loaded := s.udpConnMap[message.sessionID]
s.udpAccess.RUnlock()
if !loaded || common.Done(udpConn.ctx) {
message.release()
return E.New("unknown session iD: ", message.sessionID)
}
udpConn.inputPacket(message)
return nil
}