rohit/probo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove code who has been merged

Browse Source 
Signed-off-by: Bryan Frimin <bryan@getprobo.com>
This commit is contained in:
Bryan Frimin
2026-02-02 10:53:53 +01:00
parent cd8e905802
commit 86c0d6a94c
1 changed files with 1 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
pkg/probod/probod.go
View File

@@ -563,7 +563,7 @@ func (impl *Implm) runApiServer(
} }
if len(impl.cfg.Api.ProxyProtocol.TrustedProxies) > 0 { if len(impl.cfg.Api.ProxyProtocol.TrustedProxies) > 0 {
policy := rejectProxyHeaderFrom(impl.cfg.Api.ProxyProtocol.TrustedProxies...) policy := proxyproto.TrustProxyHeaderFrom(impl.cfg.Api.ProxyProtocol.TrustedProxies...)
listener = &proxyproto.Listener{ listener = &proxyproto.Listener{
Listener: listener, Listener: listener,
@@ -861,34 +861,3 @@ func (impl *Implm) runTrustCenterServer(
return ctx.Err() return ctx.Err()
} }
func rejectProxyHeaderFrom(trustedIPs ...net.IP) proxyproto.ConnPolicyFunc {
return func(connOpts proxyproto.ConnPolicyOptions) (proxyproto.Policy, error) {
ip, err := ipFromAddr(connOpts.Upstream)
if err != nil {
return proxyproto.REJECT, err
}
for _, trustedIP := range trustedIPs {
if trustedIP.Equal(ip) {
return proxyproto.USE, nil
}
}
return proxyproto.REJECT, nil
}
}
func ipFromAddr(upstream net.Addr) (net.IP, error) {
upstreamString, _, err := net.SplitHostPort(upstream.String())
if err != nil {
return nil, err
}
upstreamIP := net.ParseIP(upstreamString)
if nil == upstreamIP {
return nil, fmt.Errorf("proxyproto: invalid IP address")
}
return upstreamIP, nil
}