func (handler *VMessOutboundHandler) startCommunicate(request *vmessio.VMessRequest, dest v2net.Address, ray core.OutboundRay) error { conn, err := net.DialTCP("tcp", nil, &net.TCPAddr{dest.IP, int(dest.Port), ""}) log.Debug("VMessOutbound dialing tcp: %s", dest.String()) if err != nil { log.Error("Failed to open tcp (%s): %v", dest.String(), err) return err } defer conn.Close() input := ray.OutboundInput() output := ray.OutboundOutput() requestWriter := vmessio.NewVMessRequestWriter() err = requestWriter.Write(conn, request) if err != nil { log.Error("Failed to write VMess request: %v", err) close(output) return err } requestKey := request.RequestKey[:] requestIV := request.RequestIV[:] responseKey := md5.Sum(requestKey) responseIV := md5.Sum(requestIV) response := vmessio.VMessResponse{} nBytes, err := conn.Read(response[:]) if err != nil { close(output) log.Error("Failed to read VMess response (%d bytes): %v", nBytes, err) return err } log.Debug("Got response %v", response) // TODO: check response encryptRequestWriter, err := v2io.NewAesEncryptWriter(requestKey, requestIV, conn) if err != nil { close(output) log.Error("Failed to create encrypt writer: %v", err) return err } decryptResponseReader, err := v2io.NewAesDecryptReader(responseKey[:], responseIV[:], conn) if err != nil { close(output) log.Error("Failed to create decrypt reader: %v", err) return err } readFinish := make(chan bool) writeFinish := make(chan bool) go handler.dumpInput(encryptRequestWriter, input, readFinish) go handler.dumpOutput(decryptResponseReader, output, writeFinish) <-readFinish conn.CloseWrite() log.Debug("VMessOut closing write") <-writeFinish return nil }
func handleRequest(conn *net.TCPConn, request *vmessio.VMessRequest, input <-chan []byte, finish chan<- bool) error { defer close(finish) requestWriter := vmessio.NewVMessRequestWriter(v2hash.NewTimeHash(v2hash.HMACHash{}), v2math.GenerateRandomInt64InRange) err := requestWriter.Write(conn, request) if err != nil { log.Error("Failed to write VMess request: %v", err) return err } encryptRequestWriter, err := v2io.NewAesEncryptWriter(request.RequestKey[:], request.RequestIV[:], conn) if err != nil { log.Error("Failed to create encrypt writer: %v", err) return err } v2net.ChanToWriter(encryptRequestWriter, input) return nil }
func (handler *VMessOutboundHandler) Start(ray core.OutboundVRay) error { vNextAddress, vNextUser := handler.pickVNext() request := new(vmessio.VMessRequest) request.Version = vmessio.Version request.UserId = vNextUser.Id rand.Read(request.RequestIV[:]) rand.Read(request.RequestKey[:]) rand.Read(request.ResponseHeader[:]) request.Command = byte(0x01) request.Address = handler.dest conn, err := net.Dial("tcp", vNextAddress.String()) if err != nil { return err } defer conn.Close() requestWriter := vmessio.NewVMessRequestWriter() requestWriter.Write(conn, request) requestKey := request.RequestKey[:] requestIV := request.RequestIV[:] responseKey := md5.Sum(requestKey) responseIV := md5.Sum(requestIV) encryptRequestWriter, err := v2io.NewAesEncryptWriter(requestKey, requestIV, conn) if err != nil { return err } responseReader, err := v2io.NewAesDecryptReader(responseKey[:], responseIV[:], conn) if err != nil { return err } input := ray.OutboundInput() output := ray.OutboundOutput() finish := make(chan bool, 2) go handler.dumpInput(encryptRequestWriter, input, finish) go handler.dumpOutput(responseReader, output, finish) handler.waitForFinish(finish) return nil }