2020-11-01 16:50:02 +00:00
|
|
|
package tcp
|
|
|
|
|
|
|
|
import (
|
2020-11-03 20:56:51 +00:00
|
|
|
"log"
|
2020-11-01 16:50:02 +00:00
|
|
|
"mpbl3p/proxy"
|
|
|
|
"net"
|
|
|
|
)
|
|
|
|
|
|
|
|
func NewListener(p *proxy.Proxy, local string, v proxy.MacVerifier) error {
|
|
|
|
laddr, err := net.ResolveTCPAddr("tcp", local)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
listener, err := net.ListenTCP("tcp", laddr)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
go func() {
|
|
|
|
for {
|
|
|
|
conn, err := listener.AcceptTCP()
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
|
2020-11-03 20:56:51 +00:00
|
|
|
err = conn.SetWriteBuffer(0)
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
|
2020-11-01 18:34:47 +00:00
|
|
|
f := Flow{conn: conn, isAlive: true}
|
2020-11-02 17:24:15 +00:00
|
|
|
|
2020-11-26 22:10:37 +00:00
|
|
|
log.Printf("received new tcp connection: %v\n", f)
|
2020-11-02 17:24:15 +00:00
|
|
|
|
2020-11-01 16:50:02 +00:00
|
|
|
p.AddConsumer(&f)
|
|
|
|
p.AddProducer(&f, v)
|
|
|
|
}
|
|
|
|
}()
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|