2017-04-08 19:43:25 -04:00
|
|
|
package tls
|
|
|
|
|
2017-04-20 05:00:15 -04:00
|
|
|
import (
|
|
|
|
"crypto/tls"
|
|
|
|
"net"
|
|
|
|
|
|
|
|
"v2ray.com/core/common/buf"
|
|
|
|
)
|
|
|
|
|
2017-04-08 19:43:25 -04:00
|
|
|
//go:generate go run $GOPATH/src/v2ray.com/core/tools/generrorgen/main.go -pkg tls -path Transport,Internet,TLS
|
2017-04-20 05:00:15 -04:00
|
|
|
|
|
|
|
type conn struct {
|
|
|
|
net.Conn
|
|
|
|
|
|
|
|
mergingReader buf.Reader
|
|
|
|
mergingWriter buf.Writer
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *conn) ReadMultiBuffer() (buf.MultiBuffer, error) {
|
|
|
|
if c.mergingReader == nil {
|
|
|
|
c.mergingReader = buf.NewMergingReaderSize(c.Conn, 16*1024)
|
|
|
|
}
|
|
|
|
return c.mergingReader.Read()
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *conn) WriteMultiBuffer(mb buf.MultiBuffer) (int, error) {
|
|
|
|
if c.mergingWriter == nil {
|
|
|
|
c.mergingWriter = buf.NewMergingWriter(c.Conn)
|
|
|
|
}
|
|
|
|
return mb.Len(), c.mergingWriter.Write(mb)
|
|
|
|
}
|
|
|
|
|
|
|
|
func Client(c net.Conn, config *tls.Config) net.Conn {
|
|
|
|
tlsConn := tls.Client(c, config)
|
|
|
|
return &conn{Conn: tlsConn}
|
|
|
|
}
|
|
|
|
|
|
|
|
func Server(c net.Conn, config *tls.Config) net.Conn {
|
|
|
|
tlsConn := tls.Server(c, config)
|
|
|
|
return &conn{Conn: tlsConn}
|
|
|
|
}
|