2020-06-15 10:04:08 +00:00
|
|
|
package tls_interceptor
|
2020-04-01 02:08:21 +00:00
|
|
|
|
|
|
|
import (
|
|
|
|
"net"
|
2020-10-02 09:56:48 +00:00
|
|
|
"sync"
|
|
|
|
)
|
|
|
|
|
|
|
|
var (
|
|
|
|
bufferPool = sync.Pool{
|
|
|
|
New: func() interface{} {
|
|
|
|
return make([]byte, 1024)
|
|
|
|
},
|
|
|
|
}
|
2020-04-01 02:08:21 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
func chanFromConn(conn net.Conn) chan []byte {
|
|
|
|
c := make(chan []byte)
|
|
|
|
|
|
|
|
go func() {
|
2020-10-02 09:56:48 +00:00
|
|
|
b := bufferPool.Get().([]byte)
|
2020-04-01 02:08:21 +00:00
|
|
|
for {
|
|
|
|
n, err := conn.Read(b)
|
|
|
|
if n > 0 {
|
2020-12-26 13:11:49 +00:00
|
|
|
res := bufferPool.Get().([]byte)
|
2020-04-01 02:08:21 +00:00
|
|
|
// Copy the buffer so it doesn't get changed while read by the recipient.
|
|
|
|
copy(res, b[:n])
|
2020-12-26 13:11:49 +00:00
|
|
|
c <- res[:n]
|
2020-04-01 02:08:21 +00:00
|
|
|
}
|
|
|
|
if err != nil {
|
|
|
|
c <- nil
|
|
|
|
break
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}()
|
|
|
|
|
|
|
|
return c
|
|
|
|
}
|
|
|
|
|
|
|
|
func Pipe(conn1 net.Conn, conn2 net.Conn) {
|
|
|
|
chan1 := chanFromConn(conn1)
|
|
|
|
chan2 := chanFromConn(conn2)
|
|
|
|
|
|
|
|
for {
|
|
|
|
select {
|
|
|
|
case b1 := <-chan1:
|
|
|
|
if b1 == nil {
|
|
|
|
return
|
|
|
|
} else {
|
|
|
|
conn2.Write(b1)
|
|
|
|
}
|
|
|
|
case b2 := <-chan2:
|
|
|
|
if b2 == nil {
|
|
|
|
return
|
|
|
|
} else {
|
|
|
|
conn1.Write(b2)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|