format code

This commit is contained in:
V2Ray committed 2015-09-09 17:39:25 +02:00
1 parent 19308037e5
commit f09d65b2e5
3 files changed
+92 -92

No files matched your search

+40 -40
View File
@@ -1,62 +1,62 @@
package tcp
import (
"net"
"io"
"io"
"net"
)
type VFreeConnection struct {
network string
address string
network string
address string
}
func NewVFreeConnection(network string, address string) *VFreeConnection {
conn := new(VFreeConnection)
conn.network = network
conn.address = address
return conn
conn := new(VFreeConnection)
conn.network = network
conn.address = address
return conn
}
func (vconn *VFreeConnection) Start(input <-chan []byte) chan<- []byte {
output := make(chan []byte, 128)
conn, err := net.Dial(vconn.network, vconn.address)
if err != nil {
panic(err)
}
finish := make(chan bool, 2)
go vconn.DumpInput(conn, input, finish)
go vconn.DumpOutput(conn, output, finish)
go vconn.CloseConn(conn, finish)
return output
output := make(chan []byte, 128)
conn, err := net.Dial(vconn.network, vconn.address)
if err != nil {
panic(err)
}
finish := make(chan bool, 2)
go vconn.DumpInput(conn, input, finish)
go vconn.DumpOutput(conn, output, finish)
go vconn.CloseConn(conn, finish)
return output
}
func (vconn *VFreeConnection) DumpInput(conn net.Conn, input <-chan []byte, finish chan<- bool) {
for {
data, open := <-input
if !open {
finish <- true
break
}
conn.Write(data)
}
for {
data, open := <-input
if !open {
finish <- true
break
}
conn.Write(data)
}
}
func (vconn *VFreeConnection) DumpOutput(conn net.Conn, output chan<- []byte, finish chan<- bool) {
for {
buffer := make([]byte, 128)
nBytes, err := conn.Read(buffer)
if err == io.EOF {
finish <- true
break
}
output <- buffer[:nBytes]
}
for {
buffer := make([]byte, 128)
nBytes, err := conn.Read(buffer)
if err == io.EOF {
finish <- true
break
}
output <- buffer[:nBytes]
}
}
func (vconn *VFreeConnection) CloseConn(conn net.Conn, finish <-chan bool) {
for i := 0; i < 2; i++ {
<-finish
}
conn.Close()
for i := 0; i < 2; i++ {
<-finish
}
conn.Close()
}