2015-09-10 18:24:18 -04:00
|
|
|
package vmess
|
|
|
|
|
|
|
|
import (
|
|
|
|
"crypto/md5"
|
|
|
|
"io"
|
|
|
|
"net"
|
2015-09-23 08:14:53 -04:00
|
|
|
"sync"
|
2015-09-10 18:24:18 -04:00
|
|
|
|
2015-10-14 08:51:19 -04:00
|
|
|
"github.com/v2ray/v2ray-core/app"
|
2015-10-08 08:46:18 -04:00
|
|
|
"github.com/v2ray/v2ray-core/common/alloc"
|
2015-11-03 15:26:16 -05:00
|
|
|
v2crypto "github.com/v2ray/v2ray-core/common/crypto"
|
2015-09-19 18:50:21 -04:00
|
|
|
"github.com/v2ray/v2ray-core/common/log"
|
2015-09-19 17:54:36 -04:00
|
|
|
v2net "github.com/v2ray/v2ray-core/common/net"
|
2015-10-15 07:42:43 -04:00
|
|
|
"github.com/v2ray/v2ray-core/common/retry"
|
2015-10-29 19:11:29 -04:00
|
|
|
"github.com/v2ray/v2ray-core/proxy/common/connhandler"
|
2015-10-16 06:03:22 -04:00
|
|
|
"github.com/v2ray/v2ray-core/proxy/vmess/config"
|
2015-09-19 18:11:14 -04:00
|
|
|
"github.com/v2ray/v2ray-core/proxy/vmess/protocol"
|
2015-09-19 18:50:21 -04:00
|
|
|
"github.com/v2ray/v2ray-core/proxy/vmess/protocol/user"
|
2015-09-10 18:24:18 -04:00
|
|
|
)
|
|
|
|
|
|
|
|
type VMessInboundHandler struct {
|
2015-10-14 08:51:19 -04:00
|
|
|
dispatcher app.PacketDispatcher
|
2015-10-03 05:34:01 -04:00
|
|
|
clients user.UserSet
|
|
|
|
accepting bool
|
2015-09-10 18:24:18 -04:00
|
|
|
}
|
|
|
|
|
2015-11-04 15:52:48 -05:00
|
|
|
func NewVMessInboundHandler(dispatcher app.PacketDispatcher, clients user.UserSet) *VMessInboundHandler {
|
2015-09-16 10:27:36 -04:00
|
|
|
return &VMessInboundHandler{
|
2015-10-14 08:51:19 -04:00
|
|
|
dispatcher: dispatcher,
|
2015-10-03 05:34:01 -04:00
|
|
|
clients: clients,
|
2015-09-16 10:27:36 -04:00
|
|
|
}
|
2015-09-10 18:24:18 -04:00
|
|
|
}
|
|
|
|
|
2015-11-27 15:57:15 -05:00
|
|
|
func (this *VMessInboundHandler) Listen(port uint16) error {
|
2015-10-07 19:39:50 -04:00
|
|
|
listener, err := net.ListenTCP("tcp", &net.TCPAddr{
|
|
|
|
IP: []byte{0, 0, 0, 0},
|
|
|
|
Port: int(port),
|
|
|
|
Zone: "",
|
|
|
|
})
|
2015-09-10 18:24:18 -04:00
|
|
|
if err != nil {
|
2015-10-13 12:27:29 -04:00
|
|
|
log.Error("Unable to listen tcp port %d: %v", port, err)
|
|
|
|
return err
|
2015-09-10 18:24:18 -04:00
|
|
|
}
|
2015-11-27 15:57:15 -05:00
|
|
|
this.accepting = true
|
|
|
|
go this.AcceptConnections(listener)
|
2015-09-10 18:24:18 -04:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2015-11-27 15:57:15 -05:00
|
|
|
func (this *VMessInboundHandler) AcceptConnections(listener *net.TCPListener) error {
|
|
|
|
for this.accepting {
|
2015-10-15 07:42:43 -04:00
|
|
|
retry.Timed(100 /* times */, 100 /* ms */).On(func() error {
|
|
|
|
connection, err := listener.AcceptTCP()
|
|
|
|
if err != nil {
|
|
|
|
log.Error("Failed to accpet connection: %s", err.Error())
|
|
|
|
return err
|
|
|
|
}
|
2015-11-27 15:57:15 -05:00
|
|
|
go this.HandleConnection(connection)
|
2015-10-15 07:42:43 -04:00
|
|
|
return nil
|
|
|
|
})
|
|
|
|
|
2015-09-10 18:24:18 -04:00
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2015-11-27 15:57:15 -05:00
|
|
|
func (this *VMessInboundHandler) HandleConnection(connection *net.TCPConn) error {
|
2015-09-10 18:24:18 -04:00
|
|
|
defer connection.Close()
|
2015-09-14 12:19:17 -04:00
|
|
|
|
2015-10-21 17:10:03 -04:00
|
|
|
connReader := v2net.NewTimeOutReader(16, connection)
|
2015-11-27 15:57:15 -05:00
|
|
|
requestReader := protocol.NewVMessRequestReader(this.clients)
|
2015-09-10 18:24:18 -04:00
|
|
|
|
2015-09-24 08:51:19 -04:00
|
|
|
request, err := requestReader.Read(connReader)
|
2015-09-10 18:24:18 -04:00
|
|
|
if err != nil {
|
2015-10-09 11:49:59 -04:00
|
|
|
log.Access(connection.RemoteAddr().String(), "", log.AccessRejected, err.Error())
|
2015-09-18 07:19:12 -04:00
|
|
|
log.Warning("VMessIn: Invalid request from (%s): %v", connection.RemoteAddr().String(), err)
|
2015-09-10 18:24:18 -04:00
|
|
|
return err
|
|
|
|
}
|
2015-10-09 11:43:27 -04:00
|
|
|
log.Access(connection.RemoteAddr().String(), request.Address.String(), log.AccessAccepted, "")
|
2015-09-18 07:19:12 -04:00
|
|
|
log.Debug("VMessIn: Received request for %s", request.Address.String())
|
2015-09-10 18:24:18 -04:00
|
|
|
|
2015-11-27 15:57:15 -05:00
|
|
|
ray := this.dispatcher.DispatchToOutbound(v2net.NewPacket(request.Destination(), nil, true))
|
2015-09-17 17:00:17 -04:00
|
|
|
input := ray.InboundInput()
|
|
|
|
output := ray.InboundOutput()
|
2015-09-23 08:14:53 -04:00
|
|
|
var readFinish, writeFinish sync.Mutex
|
|
|
|
readFinish.Lock()
|
|
|
|
writeFinish.Lock()
|
2015-09-10 18:24:18 -04:00
|
|
|
|
2015-10-31 09:08:13 -04:00
|
|
|
userSettings := config.GetUserSettings(request.User.Level())
|
2015-10-31 04:39:45 -04:00
|
|
|
connReader.SetTimeOut(userSettings.PayloadReadTimeout)
|
2015-09-24 08:51:19 -04:00
|
|
|
go handleInput(request, connReader, input, &readFinish)
|
2015-09-17 17:00:17 -04:00
|
|
|
|
2015-10-07 08:58:31 -04:00
|
|
|
responseKey := md5.Sum(request.RequestKey)
|
|
|
|
responseIV := md5.Sum(request.RequestIV)
|
2015-09-10 18:24:18 -04:00
|
|
|
|
2015-11-03 15:26:16 -05:00
|
|
|
aesStream, err := v2crypto.NewAesEncryptionStream(responseKey[:], responseIV[:])
|
2015-09-10 18:24:18 -04:00
|
|
|
if err != nil {
|
2015-11-03 15:26:16 -05:00
|
|
|
log.Error("VMessIn: Failed to create AES decryption stream: %v", err)
|
2015-10-13 12:27:29 -04:00
|
|
|
return err
|
2015-09-10 18:24:18 -04:00
|
|
|
}
|
|
|
|
|
2015-11-03 15:26:16 -05:00
|
|
|
responseWriter := v2crypto.NewCryptionWriter(aesStream, connection)
|
|
|
|
|
2015-09-17 17:00:17 -04:00
|
|
|
// Optimize for small response packet
|
2015-10-10 10:50:19 -04:00
|
|
|
buffer := alloc.NewLargeBuffer().Clear()
|
2015-11-28 04:11:56 -05:00
|
|
|
buffer.AppendBytes(request.ResponseHeader[0] | request.ResponseHeader[1])
|
|
|
|
buffer.AppendBytes(request.ResponseHeader[2] | request.ResponseHeader[3])
|
|
|
|
buffer.AppendBytes(byte(0), byte(0))
|
2015-09-18 06:31:42 -04:00
|
|
|
|
2015-09-18 06:31:26 -04:00
|
|
|
if data, open := <-output; open {
|
2015-10-08 11:41:38 -04:00
|
|
|
buffer.Append(data.Value)
|
|
|
|
data.Release()
|
|
|
|
responseWriter.Write(buffer.Value)
|
|
|
|
buffer.Release()
|
2015-09-23 11:13:50 -04:00
|
|
|
go handleOutput(request, responseWriter, output, &writeFinish)
|
2015-09-23 08:14:53 -04:00
|
|
|
writeFinish.Lock()
|
2015-09-17 17:00:17 -04:00
|
|
|
}
|
2015-09-18 06:31:42 -04:00
|
|
|
|
2015-10-07 19:39:50 -04:00
|
|
|
connection.CloseWrite()
|
2015-09-23 08:14:53 -04:00
|
|
|
readFinish.Lock()
|
2015-09-10 18:24:18 -04:00
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2015-10-08 08:46:18 -04:00
|
|
|
func handleInput(request *protocol.VMessRequest, reader io.Reader, input chan<- *alloc.Buffer, finish *sync.Mutex) {
|
2015-09-17 17:00:17 -04:00
|
|
|
defer close(input)
|
2015-09-23 08:14:53 -04:00
|
|
|
defer finish.Unlock()
|
2015-09-17 17:00:17 -04:00
|
|
|
|
2015-11-03 15:26:16 -05:00
|
|
|
aesStream, err := v2crypto.NewAesDecryptionStream(request.RequestKey, request.RequestIV)
|
2015-09-17 17:00:17 -04:00
|
|
|
if err != nil {
|
2015-11-03 15:26:16 -05:00
|
|
|
log.Error("VMessIn: Failed to create AES decryption stream: %v", err)
|
2015-09-17 17:00:17 -04:00
|
|
|
return
|
|
|
|
}
|
2015-11-03 15:26:16 -05:00
|
|
|
requestReader := v2crypto.NewCryptionReader(aesStream, reader)
|
2015-09-17 17:00:17 -04:00
|
|
|
v2net.ReaderToChan(input, requestReader)
|
2015-09-10 18:24:18 -04:00
|
|
|
}
|
|
|
|
|
2015-10-08 08:46:18 -04:00
|
|
|
func handleOutput(request *protocol.VMessRequest, writer io.Writer, output <-chan *alloc.Buffer, finish *sync.Mutex) {
|
2015-09-13 14:01:50 -04:00
|
|
|
v2net.ChanToWriter(writer, output)
|
2015-09-23 08:14:53 -04:00
|
|
|
finish.Unlock()
|
2015-09-10 18:24:18 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
type VMessInboundHandlerFactory struct {
|
2015-09-11 11:27:36 -04:00
|
|
|
}
|
|
|
|
|
2015-11-27 15:57:15 -05:00
|
|
|
func (this *VMessInboundHandlerFactory) Create(dispatcher app.PacketDispatcher, rawConfig interface{}) (connhandler.InboundConnectionHandler, error) {
|
2015-10-16 06:03:22 -04:00
|
|
|
config := rawConfig.(config.Inbound)
|
2015-10-06 17:11:08 -04:00
|
|
|
|
2015-09-19 18:11:14 -04:00
|
|
|
allowedClients := user.NewTimedUserSet()
|
2015-10-16 06:03:22 -04:00
|
|
|
for _, user := range config.AllowedUsers() {
|
2015-09-12 05:51:42 -04:00
|
|
|
allowedClients.AddUser(user)
|
|
|
|
}
|
2015-10-03 05:34:01 -04:00
|
|
|
|
2015-11-04 15:52:48 -05:00
|
|
|
return NewVMessInboundHandler(dispatcher, allowedClients), nil
|
2015-09-12 14:36:21 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
func init() {
|
2015-10-29 19:11:29 -04:00
|
|
|
connhandler.RegisterInboundConnectionHandlerFactory("vmess", &VMessInboundHandlerFactory{})
|
2015-09-10 18:24:18 -04:00
|
|
|
}
|