|
|
@ -163,14 +163,19 @@ func (d *DokodemoDoor) Process(ctx context.Context, network net.Network, conn in
|
|
|
|
if !destinationOverridden {
|
|
|
|
if !destinationOverridden {
|
|
|
|
writer = &buf.SequentialWriter{Writer: conn}
|
|
|
|
writer = &buf.SequentialWriter{Writer: conn}
|
|
|
|
} else {
|
|
|
|
} else {
|
|
|
|
var addr *net.UDPAddr
|
|
|
|
back := conn.RemoteAddr().(*net.UDPAddr)
|
|
|
|
var mark int
|
|
|
|
if !dest.Address.Family().IsIP() {
|
|
|
|
if dest.Address.Family().IsIP() {
|
|
|
|
if len(back.IP) == 4 {
|
|
|
|
addr = &net.UDPAddr{
|
|
|
|
dest.Address = net.AnyIP
|
|
|
|
IP: dest.Address.IP(),
|
|
|
|
} else {
|
|
|
|
Port: int(dest.Port),
|
|
|
|
dest.Address = net.AnyIPv6
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
addr := &net.UDPAddr{
|
|
|
|
|
|
|
|
IP: dest.Address.IP(),
|
|
|
|
|
|
|
|
Port: int(dest.Port),
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
var mark int
|
|
|
|
if d.sockopt != nil {
|
|
|
|
if d.sockopt != nil {
|
|
|
|
mark = int(d.sockopt.Mark)
|
|
|
|
mark = int(d.sockopt.Mark)
|
|
|
|
}
|
|
|
|
}
|
|
|
@ -178,8 +183,7 @@ func (d *DokodemoDoor) Process(ctx context.Context, network net.Network, conn in
|
|
|
|
if err != nil {
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
back := net.DestinationFromAddr(conn.RemoteAddr())
|
|
|
|
writer = NewPacketWriter(pConn, &dest, mark, back)
|
|
|
|
writer = NewPacketWriter(pConn, &dest, mark, &back)
|
|
|
|
|
|
|
|
defer writer.(*PacketWriter).Close()
|
|
|
|
defer writer.(*PacketWriter).Close()
|
|
|
|
/*
|
|
|
|
/*
|
|
|
|
sockopt := &internet.SocketConfig{
|
|
|
|
sockopt := &internet.SocketConfig{
|
|
|
@ -236,15 +240,12 @@ func (d *DokodemoDoor) Process(ctx context.Context, network net.Network, conn in
|
|
|
|
return nil
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
func NewPacketWriter(conn net.PacketConn, d *net.Destination, mark int, back *net.Destination) buf.Writer {
|
|
|
|
func NewPacketWriter(conn net.PacketConn, d *net.Destination, mark int, back *net.UDPAddr) buf.Writer {
|
|
|
|
writer := &PacketWriter{
|
|
|
|
writer := &PacketWriter{
|
|
|
|
conn: conn,
|
|
|
|
conn: conn,
|
|
|
|
conns: make(map[net.Destination]net.PacketConn),
|
|
|
|
conns: make(map[net.Destination]net.PacketConn),
|
|
|
|
mark: mark,
|
|
|
|
mark: mark,
|
|
|
|
back: &net.UDPAddr{
|
|
|
|
back: back,
|
|
|
|
IP: back.Address.IP(),
|
|
|
|
|
|
|
|
Port: int(back.Port),
|
|
|
|
|
|
|
|
},
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
writer.conns[*d] = conn
|
|
|
|
writer.conns[*d] = conn
|
|
|
|
return writer
|
|
|
|
return writer
|
|
|
|