mirror of
https://github.com/netbirdio/netbird.git
synced 2026-04-16 15:26:40 +00:00
55126f990c
- Move `util/grpc` and `util/net` to `client` so `internal` packages can be accessed - Add methods to return the next best interface after the NetBird interface. - Use `IP_UNICAST_IF` sock opt to force the outgoing interface for the NetBird `net.Dialer` and `net.ListenerConfig` to avoid routing loops. The interface is picked by the new route lookup method. - Some refactoring to avoid import cycles - Old behavior is available through `NB_USE_LEGACY_ROUTING=true` env var
50 lines
1.2 KiB
Go
50 lines
1.2 KiB
Go
//go:build !ios
|
|
|
|
package net
|
|
|
|
import (
|
|
"io"
|
|
"net"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/netbirdio/netbird/client/net/hooks"
|
|
)
|
|
|
|
// Conn wraps a net.Conn to override the Close method
|
|
type Conn struct {
|
|
net.Conn
|
|
ID hooks.ConnectionID
|
|
}
|
|
|
|
// Close overrides the net.Conn Close method to execute all registered hooks after closing the connection
|
|
// Close overrides the net.Conn Close method to execute all registered hooks before closing the connection.
|
|
func (c *Conn) Close() error {
|
|
return closeConn(c.ID, c.Conn)
|
|
}
|
|
|
|
// TCPConn wraps net.TCPConn to override its Close method to include hook functionality.
|
|
type TCPConn struct {
|
|
*net.TCPConn
|
|
ID hooks.ConnectionID
|
|
}
|
|
|
|
// Close overrides the net.TCPConn Close method to execute all registered hooks before closing the connection.
|
|
func (c *TCPConn) Close() error {
|
|
return closeConn(c.ID, c.TCPConn)
|
|
}
|
|
|
|
// closeConn is a helper function to close connections and execute close hooks.
|
|
func closeConn(id hooks.ConnectionID, conn io.Closer) error {
|
|
err := conn.Close()
|
|
|
|
closeHooks := hooks.GetCloseHooks()
|
|
for _, hook := range closeHooks {
|
|
if err := hook(id); err != nil {
|
|
log.Errorf("Error executing close hook: %v", err)
|
|
}
|
|
}
|
|
|
|
return err
|
|
}
|