stored client-side peer info to ServerConnPeer when PEER_STARTED is received

This commit is contained in:
hyung-hwan 2025-02-17 00:52:29 +09:00
parent f2536a0acc
commit be7f4f4da5
4 changed files with 23 additions and 7 deletions

View File

@ -1267,6 +1267,7 @@ func (c *Client) wrap_http_handler(handler ClientHttpHandler) http.Handler {
if handler.Cors(req) {
w.Header().Set("Access-Control-Allow-Origin", "*")
w.Header().Set("Access-Control-Allow-Headers", "*")
w.Header().Set("Access-Control-Allow-Methods", "*")
}
if req.Method == http.MethodOptions {
status_code = WriteEmptyRespHeader(w, http.StatusOK)

View File

@ -33,10 +33,10 @@ type json_out_server_route struct {
type json_out_server_peer struct {
Id PeerId `json:"id"`
ClientPeerAddr string `json:"client-peer-addr"`
ClientLocalAddr string `json:"client-local-addr"`
ServerPeerAddr string `json:"server-peer-addr"`
ServerLocalAddr string `json:"server-local-addr"`
ClientPeerAddr string `json:"client-peer-addr"`
ClientLocalAddr string `json:"client-local-addr"`
}
type json_out_server_stats struct {
@ -465,8 +465,8 @@ func (ctl *server_ctl_server_conns_id_routes_id_peers) ServeHTTP(w http.Response
Id: p.conn_id,
ServerPeerAddr: p.conn.RemoteAddr().String(),
ServerLocalAddr: p.conn.LocalAddr().String(),
ClientPeerAddr: p.cts.remote_addr,
ClientLocalAddr: p.cts.local_addr,
ClientPeerAddr: p.client_peer_raddr,
ClientLocalAddr: p.client_peer_laddr,
})
}
r.pts_mtx.Unlock()
@ -522,8 +522,8 @@ func (ctl *server_ctl_server_conns_id_routes_id_peers_id) ServeHTTP(w http.Respo
Id: p.conn_id,
ServerPeerAddr: p.conn.RemoteAddr().String(),
ServerLocalAddr: p.conn.LocalAddr().String(),
ClientPeerAddr: p.cts.remote_addr,
ClientLocalAddr: p.cts.local_addr,
ClientPeerAddr: p.client_peer_raddr,
ClientLocalAddr: p.client_peer_laddr,
}
status_code = WriteJsonRespHeader(w, http.StatusOK)

View File

@ -12,7 +12,6 @@ import "time"
type ServerPeerConn struct {
route *ServerRoute
conn_id PeerId
cts *ClientConn
conn *net.TCPConn
stop_chan chan bool
@ -22,6 +21,8 @@ type ServerPeerConn struct {
client_peer_started atomic.Bool
client_peer_stopped atomic.Bool
client_peer_eof atomic.Bool
client_peer_laddr string
client_peer_raddr string
}
func NewServerPeerConn(r *ServerRoute, c *net.TCPConn, id PeerId) *ServerPeerConn {
@ -163,6 +164,19 @@ func (spc *ServerPeerConn) ReportEvent(event_type PACKET_KIND, event_data interf
switch event_type {
case PACKET_KIND_PEER_STARTED:
var ok bool
var pd *PeerDesc
pd, ok = event_data.(*PeerDesc)
if !ok {
// something wrong. leave it unknown.
spc.client_peer_laddr = "";
spc.client_peer_raddr = "";
} else {
spc.client_peer_laddr = pd.LocalAddrStr
spc.client_peer_raddr = pd.RemoteAddrStr
}
if spc.client_peer_started.CompareAndSwap(false, true) {
spc.client_peer_status_chan <- true
}

View File

@ -1004,6 +1004,7 @@ func (s *Server) wrap_http_handler(handler ServerHttpHandler) http.Handler {
if handler.Cors(req) {
w.Header().Set("Access-Control-Allow-Origin", "*")
w.Header().Set("Access-Control-Allow-Headers", "*")
w.Header().Set("Access-Control-Allow-Methods", "*")
}
if req.Method == http.MethodOptions {
status_code = WriteEmptyRespHeader(w, http.StatusOK)