mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-12-10 20:26:29 -05:00
49 lines
1002 B
Go
49 lines
1002 B
Go
package mocks
|
|
|
|
import (
|
|
"bytes"
|
|
|
|
"github.com/v2ray/v2ray-core"
|
|
v2net "github.com/v2ray/v2ray-core/common/net"
|
|
)
|
|
|
|
type OutboundConnectionHandler struct {
|
|
Data2Send *bytes.Buffer
|
|
Data2Return []byte
|
|
Destination v2net.Destination
|
|
}
|
|
|
|
func (handler *OutboundConnectionHandler) Start(ray core.OutboundRay) error {
|
|
input := ray.OutboundInput()
|
|
output := ray.OutboundOutput()
|
|
|
|
go func() {
|
|
for {
|
|
data, open := <-input
|
|
if !open {
|
|
break
|
|
}
|
|
handler.Data2Send.Write(data)
|
|
}
|
|
dataCopy := make([]byte, len(handler.Data2Return))
|
|
copy(dataCopy, handler.Data2Return)
|
|
output <- dataCopy
|
|
close(output)
|
|
}()
|
|
|
|
return nil
|
|
}
|
|
|
|
func (handler *OutboundConnectionHandler) Initialize(config []byte) error {
|
|
return nil
|
|
}
|
|
|
|
func (handler *OutboundConnectionHandler) Create(point *core.Point, packet v2net.Packet) (core.OutboundConnectionHandler, error) {
|
|
handler.Destination = packet.Destination()
|
|
if packet.Chunk() != nil {
|
|
handler.Data2Send.Write(packet.Chunk())
|
|
}
|
|
|
|
return handler, nil
|
|
}
|