SpoofDPI/proxy/proxy.go
2022-01-12 02:15:45 +09:00

70 lines
1.2 KiB
Go

package proxy
import (
"os"
log "github.com/sirupsen/logrus"
"github.com/xvzc/SpoofDPI/net"
"github.com/xvzc/SpoofDPI/packet"
)
type Proxy struct {
port string
}
func New(port string) *Proxy {
return &Proxy{
port: port,
}
}
func (p *Proxy) Port() string {
return p.port
}
func (p *Proxy) Start() {
l, err := net.Listen("tcp", ":"+p.Port())
if err != nil {
log.Fatal("Error creating listener: ", err)
os.Exit(1)
}
log.Println("Created a listener on :", p.Port())
for {
conn, err := l.Accept()
if err != nil {
log.Fatal("Error accepting connection: ", err)
continue
}
log.Debug("Accepted a new connection.", conn.RemoteAddr())
go func() {
defer conn.Close()
b, err := conn.ReadBytes()
if err != nil {
return
}
log.Debug("Client sent data: ", len(b))
pkt := packet.NewHttpPacket(b)
log.Debug("New request: \n\n" + string(pkt.Raw()))
if !pkt.IsValidMethod() {
log.Println("Unsupported method: ", pkt.Method())
return
}
if pkt.IsConnectMethod() {
log.Debug("HTTPS Requested")
conn.HandleHttps(pkt)
} else {
log.Debug("HTTP Requested.")
conn.HandleHttp(pkt)
}
}()
}
}