mirror of
https://github.com/netbirdio/netbird.git
synced 2026-04-20 01:06:45 +00:00
47 lines
1.2 KiB
Go
47 lines
1.2 KiB
Go
package server
|
|
|
|
import (
|
|
"context"
|
|
|
|
firewall "github.com/netbirdio/netbird/client/firewall/types"
|
|
"github.com/netbirdio/netbird/client/proto"
|
|
)
|
|
|
|
func (s *Server) ForwardingRules(context.Context, *proto.EmptyRequest) (*proto.ForwardingRulesResponse, error) {
|
|
s.mutex.Lock()
|
|
defer s.mutex.Unlock()
|
|
|
|
rules := s.statusRecorder.ForwardingRules()
|
|
|
|
responseRules := make([]*proto.ForwardingRule, 0, len(rules))
|
|
for _, rule := range rules {
|
|
|
|
respRule := &proto.ForwardingRule{
|
|
Protocol: string(rule.Protocol),
|
|
DestinationPort: portToProto(rule.DestinationPort),
|
|
TranslatedAddress: rule.TranslatedAddress.String(),
|
|
TranslatedPort: portToProto(rule.TranslatedPort),
|
|
}
|
|
responseRules = append(responseRules, respRule)
|
|
|
|
}
|
|
|
|
return &proto.ForwardingRulesResponse{Rules: responseRules}, nil
|
|
}
|
|
|
|
func portToProto(port firewall.Port) *proto.PortInfo {
|
|
var portInfo proto.PortInfo
|
|
|
|
if !port.IsRange {
|
|
portInfo.PortSelection = &proto.PortInfo_Port{Port: uint32(port.Values[0])}
|
|
} else {
|
|
portInfo.PortSelection = &proto.PortInfo_Range_{
|
|
Range: &proto.PortInfo_Range{
|
|
Start: uint32(port.Values[0]),
|
|
End: uint32(port.Values[1]),
|
|
},
|
|
}
|
|
}
|
|
return &portInfo
|
|
}
|