mirror of https://github.com/XTLS/Xray-core
62 lines
1.7 KiB
Go
62 lines
1.7 KiB
Go
package grpc
|
|
|
|
import (
|
|
"net/url"
|
|
"strings"
|
|
|
|
"github.com/xtls/xray-core/common"
|
|
"github.com/xtls/xray-core/transport/internet"
|
|
)
|
|
|
|
const protocolName = "grpc"
|
|
|
|
func init() {
|
|
common.Must(internet.RegisterProtocolConfigCreator(protocolName, func() interface{} {
|
|
return new(Config)
|
|
}))
|
|
}
|
|
|
|
func (c *Config) getServiceName() string {
|
|
// Normal old school config
|
|
if !strings.HasPrefix(c.ServiceName, "/") {
|
|
return url.PathEscape(c.ServiceName)
|
|
}
|
|
|
|
// Otherwise new custom paths
|
|
lastIndex := strings.LastIndex(c.ServiceName, "/")
|
|
if lastIndex < 1 {
|
|
lastIndex = 1
|
|
}
|
|
rawServiceName := c.ServiceName[1:lastIndex] // trim from first to last '/'
|
|
serviceNameParts := strings.Split(rawServiceName, "/")
|
|
for i := range serviceNameParts {
|
|
serviceNameParts[i] = url.PathEscape(serviceNameParts[i])
|
|
}
|
|
return strings.Join(serviceNameParts, "/")
|
|
}
|
|
|
|
func (c *Config) getTunStreamName() string {
|
|
// Normal old school config
|
|
if !strings.HasPrefix(c.ServiceName, "/") {
|
|
return "Tun"
|
|
}
|
|
// Otherwise new custom paths
|
|
endingPath := c.ServiceName[strings.LastIndex(c.ServiceName, "/")+1:] // from the last '/' to end of string
|
|
return url.PathEscape(strings.Split(endingPath, "|")[0])
|
|
}
|
|
|
|
func (c *Config) getTunMultiStreamName() string {
|
|
// Normal old school config
|
|
if !strings.HasPrefix(c.ServiceName, "/") {
|
|
return "TunMulti"
|
|
}
|
|
// Otherwise new custom paths
|
|
endingPath := c.ServiceName[strings.LastIndex(c.ServiceName, "/")+1:] // from the last '/' to end of string
|
|
streamNames := strings.Split(endingPath, "|")
|
|
if len(streamNames) == 1 { // client side. Service name is the full path to multi tun
|
|
return url.PathEscape(streamNames[0])
|
|
} else { // server side. The second part is the path to multi tun
|
|
return url.PathEscape(streamNames[1])
|
|
}
|
|
}
|