SpoofDPI/proxy/proxy.go

67 lines
1.1 KiB
Go
Raw Normal View History

2021-12-29 17:08:30 +00:00
package proxy
import (
"os"
2022-01-03 07:24:39 +00:00
2022-01-11 15:05:16 +00:00
log "github.com/sirupsen/logrus"
"github.com/xvzc/SpoofDPI/net"
2022-01-08 06:35:32 +00:00
"github.com/xvzc/SpoofDPI/packet"
2021-12-29 17:08:30 +00:00
)
2022-01-08 15:48:19 +00:00
type Proxy struct {
2022-01-11 17:15:45 +00:00
port string
2022-01-08 15:48:19 +00:00
}
func New(port string) *Proxy {
2022-01-08 15:48:19 +00:00
return &Proxy{
2022-01-11 17:15:45 +00:00
port: port,
2022-01-08 15:48:19 +00:00
}
}
2022-01-11 17:15:45 +00:00
func (p *Proxy) Port() string {
return p.port
}
2022-01-08 15:48:19 +00:00
func (p *Proxy) Start() {
2022-01-11 17:15:45 +00:00
l, err := net.Listen("tcp", ":"+p.Port())
2021-12-29 17:08:30 +00:00
if err != nil {
2022-01-04 16:47:18 +00:00
log.Fatal("Error creating listener: ", err)
os.Exit(1)
2021-12-29 17:08:30 +00:00
}
2022-01-11 17:15:45 +00:00
log.Println("Created a listener on :", p.Port())
2021-12-29 17:08:30 +00:00
2022-01-04 16:47:18 +00:00
for {
2022-01-10 19:27:12 +00:00
conn, err := l.Accept()
2021-12-29 17:08:30 +00:00
if err != nil {
2022-01-04 16:47:18 +00:00
log.Fatal("Error accepting connection: ", err)
2021-12-29 17:08:30 +00:00
continue
}
2022-01-11 15:05:16 +00:00
log.Debug("Accepted a new connection.", conn.RemoteAddr())
2021-12-29 17:08:30 +00:00
2022-01-04 16:47:18 +00:00
go func() {
2022-01-10 19:27:12 +00:00
b, err := conn.ReadBytes()
2022-01-04 16:47:18 +00:00
if err != nil {
return
}
2022-01-11 15:05:16 +00:00
log.Debug("Client sent data: ", len(b))
2021-12-29 17:08:30 +00:00
2022-01-11 17:15:45 +00:00
pkt := packet.NewHttpPacket(b)
2022-01-07 15:39:58 +00:00
2022-01-11 17:15:45 +00:00
if !pkt.IsValidMethod() {
log.Println("Unsupported method: ", pkt.Method())
2022-01-07 14:04:09 +00:00
return
}
2022-01-11 17:15:45 +00:00
if pkt.IsConnectMethod() {
2022-01-11 18:06:14 +00:00
log.Debug("[HTTPS] Start")
2022-03-04 15:46:33 +00:00
conn.HandleHttps(pkt)
2022-01-04 16:47:18 +00:00
} else {
2022-01-11 18:06:14 +00:00
log.Debug("[HTTP] Start")
2022-03-04 15:46:33 +00:00
conn.HandleHttp(pkt)
2022-01-04 16:47:18 +00:00
}
}()
}
2021-12-29 17:08:30 +00:00
}