Skip to content
This repository has been archived by the owner on Apr 26, 2024. It is now read-only.

Commit

Permalink
Revert "Replace string with concret address struct"
Browse files Browse the repository at this point in the history
This reverts commit 3a44eb8.
  • Loading branch information
bzEq committed Apr 26, 2024
1 parent 3a44eb8 commit 3d18294
Show file tree
Hide file tree
Showing 2 changed files with 9 additions and 17 deletions.
18 changes: 5 additions & 13 deletions proxy/intrinsic/client.go
Original file line number Diff line number Diff line change
Expand Up @@ -78,16 +78,12 @@ func (self *ClientContext) Dial(network string, addr string) (net.Conn, error) {
}

func (self *ClientContext) dialUDP(network, addr string) (net.Conn, error) {
raddr, err := net.ResolveUDPAddr(network, addr)
if err != nil {
return nil, err
}
local := core.MakePipe()
c := local[1]
go func() {
defer c.Close()
disp := self.router.C.(*UDPDispatcher)
id := disp.NewEntry(raddr)
id := disp.NewEntry(addr)
defer disp.DeleteEntry(id)
cp := core.NewSyncPortWithTimeout(c, nil, core.DEFAULT_UDP_TIMEOUT)
r, err := self.router.NewRoute(id, cp)
Expand All @@ -101,10 +97,6 @@ func (self *ClientContext) dialUDP(network, addr string) (net.Conn, error) {
}

func (self *ClientContext) dialTCP(network, addr string) (net.Conn, error) {
raddr, err := net.ResolveTCPAddr(network, addr)
if err != nil {
return nil, err
}
local := core.MakePipe()
go func() {
defer local[1].Close()
Expand All @@ -117,7 +109,7 @@ func (self *ClientContext) dialTCP(network, addr string) (net.Conn, error) {
i := Intrinsic{Func: RELAY_TCP}
{
data := &bytes.Buffer{}
req := TCPRequest{Addr: *raddr}
req := TCPRequest{Addr: addr}
enc := gob.NewEncoder(data)
if err := enc.Encode(&req); err != nil {
log.Println(err)
Expand All @@ -140,11 +132,11 @@ func (self *ClientContext) dialTCP(network, addr string) (net.Conn, error) {
}

type UDPDispatcher struct {
t core.Map[core.RouteId, *net.UDPAddr]
t core.Map[core.RouteId, string]
c uint64
}

func (self *UDPDispatcher) NewEntry(addr *net.UDPAddr) core.RouteId {
func (self *UDPDispatcher) NewEntry(addr string) core.RouteId {
id := core.RouteId(atomic.AddUint64(&self.c, 1))
self.t.Store(id, addr)
return id
Expand All @@ -159,7 +151,7 @@ func (self *UDPDispatcher) Encode(id core.RouteId, data *iovec.IoVec) error {
if !in {
return fmt.Errorf("Remote address of RouteId #%d doesn't exist", id)
}
msg := UDPMessage{Id: id, Addr: *raddr, Data: data.Consume()}
msg := UDPMessage{Id: id, Addr: raddr, Data: data.Consume()}
var buf bytes.Buffer
enc := gob.NewEncoder(&buf)
if err := enc.Encode(&msg); err != nil {
Expand Down
8 changes: 4 additions & 4 deletions proxy/intrinsic/server.go
Original file line number Diff line number Diff line change
Expand Up @@ -25,12 +25,12 @@ const (
)

type TCPRequest struct {
Addr net.TCPAddr
Addr string
}

type UDPMessage struct {
Id core.RouteId
Addr net.UDPAddr
Addr string
Data []byte
}

Expand Down Expand Up @@ -64,7 +64,7 @@ func (self *Server) relayUDP() error {
log.Println(err)
return
}
c, err := net.Dial("udp", msg.Addr.String())
c, err := net.Dial("udp", msg.Addr)
if err != nil {
log.Println(err)
return
Expand Down Expand Up @@ -128,7 +128,7 @@ func (self *Server) Run() {
log.Println(err)
return
}
if err := self.relayTCP(req.Addr.String()); err != nil {
if err := self.relayTCP(req.Addr); err != nil {
log.Println(err)
return
}
Expand Down

0 comments on commit 3d18294

Please sign in to comment.