v2ray-core/net/freedom/freedom.go

66 lines
1.2 KiB
Go
Raw Normal View History

2015-09-09 17:39:06 +02:00
package tcp
import (
2015-09-09 17:39:25 +02:00
"io"
"net"
2015-09-10 00:50:21 +02:00
"github.com/v2ray/v2ray-core"
2015-09-09 17:39:06 +02:00
)
type VFreeConnection struct {
2015-09-09 17:39:25 +02:00
network string
address string
2015-09-09 17:39:06 +02:00
}
func NewVFreeConnection(network string, address string) *VFreeConnection {
2015-09-09 17:39:25 +02:00
conn := new(VFreeConnection)
conn.network = network
conn.address = address
return conn
2015-09-09 17:39:06 +02:00
}
2015-09-10 00:50:21 +02:00
func (vconn *VFreeConnection) Start(vRay core.OutboundVRay) error {
input := vRay.OutboundInput()
output := vRay.OutboundOutput()
2015-09-09 17:39:25 +02:00
conn, err := net.Dial(vconn.network, vconn.address)
if err != nil {
2015-09-10 00:50:21 +02:00
return err
2015-09-09 17:39:25 +02:00
}
finish := make(chan bool, 2)
go vconn.DumpInput(conn, input, finish)
go vconn.DumpOutput(conn, output, finish)
go vconn.CloseConn(conn, finish)
2015-09-10 00:50:21 +02:00
return nil
2015-09-09 17:39:06 +02:00
}
func (vconn *VFreeConnection) DumpInput(conn net.Conn, input <-chan []byte, finish chan<- bool) {
2015-09-09 17:39:25 +02:00
for {
data, open := <-input
if !open {
finish <- true
break
}
conn.Write(data)
}
2015-09-09 17:39:06 +02:00
}
func (vconn *VFreeConnection) DumpOutput(conn net.Conn, output chan<- []byte, finish chan<- bool) {
2015-09-09 17:39:25 +02:00
for {
buffer := make([]byte, 128)
nBytes, err := conn.Read(buffer)
if err == io.EOF {
finish <- true
break
}
output <- buffer[:nBytes]
}
2015-09-09 17:39:06 +02:00
}
func (vconn *VFreeConnection) CloseConn(conn net.Conn, finish <-chan bool) {
2015-09-09 17:39:25 +02:00
for i := 0; i < 2; i++ {
<-finish
}
conn.Close()
2015-09-09 17:39:06 +02:00
}