mirror of https://github.com/XTLS/Xray-core
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
55 lines
1.2 KiB
55 lines
1.2 KiB
// +build !confonly |
|
|
|
package dispatcher |
|
|
|
import ( |
|
"github.com/xtls/xray-core/v1/common" |
|
"github.com/xtls/xray-core/v1/common/protocol/bittorrent" |
|
"github.com/xtls/xray-core/v1/common/protocol/http" |
|
"github.com/xtls/xray-core/v1/common/protocol/tls" |
|
) |
|
|
|
type SniffResult interface { |
|
Protocol() string |
|
Domain() string |
|
} |
|
|
|
type protocolSniffer func([]byte) (SniffResult, error) |
|
|
|
type Sniffer struct { |
|
sniffer []protocolSniffer |
|
} |
|
|
|
func NewSniffer() *Sniffer { |
|
return &Sniffer{ |
|
sniffer: []protocolSniffer{ |
|
func(b []byte) (SniffResult, error) { return http.SniffHTTP(b) }, |
|
func(b []byte) (SniffResult, error) { return tls.SniffTLS(b) }, |
|
func(b []byte) (SniffResult, error) { return bittorrent.SniffBittorrent(b) }, |
|
}, |
|
} |
|
} |
|
|
|
var errUnknownContent = newError("unknown content") |
|
|
|
func (s *Sniffer) Sniff(payload []byte) (SniffResult, error) { |
|
var pendingSniffer []protocolSniffer |
|
for _, s := range s.sniffer { |
|
result, err := s(payload) |
|
if err == common.ErrNoClue { |
|
pendingSniffer = append(pendingSniffer, s) |
|
continue |
|
} |
|
|
|
if err == nil && result != nil { |
|
return result, nil |
|
} |
|
} |
|
|
|
if len(pendingSniffer) > 0 { |
|
s.sniffer = pendingSniffer |
|
return nil, common.ErrNoClue |
|
} |
|
|
|
return nil, errUnknownContent |
|
}
|
|
|