mirror of
https://github.com/netbirdio/netbird.git
synced 2026-04-19 16:56:39 +00:00
ca33849f31
- Add debug endpoint for monitoring netbird clients - Add types package with AccountID type - Refactor netbird roundtrip to key clients by AccountID - Multiple domains can share the same client per account - Add status notifier for tunnel connection updates - Add OIDC flags to CLI - Add tests for netbird client management
74 lines
1.6 KiB
Go
74 lines
1.6 KiB
Go
package proxy
|
|
|
|
import (
|
|
"net"
|
|
"net/http"
|
|
"net/url"
|
|
"sort"
|
|
"strings"
|
|
|
|
"github.com/netbirdio/netbird/proxy/internal/types"
|
|
)
|
|
|
|
type Mapping struct {
|
|
ID string
|
|
AccountID types.AccountID
|
|
Host string
|
|
Paths map[string]*url.URL
|
|
}
|
|
|
|
func (p *ReverseProxy) findTargetForRequest(req *http.Request) (*url.URL, string, types.AccountID, bool) {
|
|
p.mappingsMux.RLock()
|
|
if p.mappings == nil {
|
|
p.mappingsMux.RUnlock()
|
|
p.mappingsMux.Lock()
|
|
defer p.mappingsMux.Unlock()
|
|
p.mappings = make(map[string]Mapping)
|
|
// There cannot be any loaded Mappings as we have only just initialized.
|
|
return nil, "", "", false
|
|
}
|
|
defer p.mappingsMux.RUnlock()
|
|
|
|
// Strip port from host if present (e.g., "external.test:8443" -> "external.test")
|
|
host := req.Host
|
|
if h, _, err := net.SplitHostPort(host); err == nil {
|
|
host = h
|
|
}
|
|
|
|
m, exists := p.mappings[host]
|
|
if !exists {
|
|
return nil, "", "", false
|
|
}
|
|
|
|
// Sort paths by length (longest first) in a naive attempt to match the most specific route first.
|
|
paths := make([]string, 0, len(m.Paths))
|
|
for path := range m.Paths {
|
|
paths = append(paths, path)
|
|
}
|
|
sort.Slice(paths, func(i, j int) bool {
|
|
return len(paths[i]) > len(paths[j])
|
|
})
|
|
|
|
for _, path := range paths {
|
|
if strings.HasPrefix(req.URL.Path, path) {
|
|
return m.Paths[path], m.ID, m.AccountID, true
|
|
}
|
|
}
|
|
return nil, "", "", false
|
|
}
|
|
|
|
func (p *ReverseProxy) AddMapping(m Mapping) {
|
|
p.mappingsMux.Lock()
|
|
defer p.mappingsMux.Unlock()
|
|
if p.mappings == nil {
|
|
p.mappings = make(map[string]Mapping)
|
|
}
|
|
p.mappings[m.Host] = m
|
|
}
|
|
|
|
func (p *ReverseProxy) RemoveMapping(m Mapping) {
|
|
p.mappingsMux.Lock()
|
|
defer p.mappingsMux.Unlock()
|
|
delete(p.mappings, m.Host)
|
|
}
|