speatzle/nfsense
1
0
Fork 0
mirror of https://github.com/speatzle/nfsense.git synced 2025-05-10 18:38:22 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add Service Port Helpers

Browse source
This commit is contained in:
Samuel Lorch 2023-03-02 00:05:58 +01:00
parent 16a9fec479
commit 2afe7341c1
1 changed files with 23 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
pkg/definitions/service.go
View file

@ -1,6 +1,9 @@
package definitions
import "encoding/json"
import (
"encoding/json"
"fmt"
)
type Service struct {
Type ServiceType `json:"type"`
@ -10,6 +13,25 @@ type Service struct {
DPortStart *uint32 `json:"dport_start,omitempty"`
DPortEnd *uint32 `json:"dport_end,omitempty"`
ICMPCode *uint32 `json:"icmp_code,omitempty"`
Children *[]string `json:"children,omitempty"`
}
func (s Service) GetSPort() string {
if *s.SPortStart == 0 {
return ""
} else if *s.SPortEnd == 0 {
return fmt.Sprintf("sport %d", *s.SPortStart)
}
return fmt.Sprintf("sport %d - %d", *s.SPortStart, *s.SPortEnd)
}
func (s Service) GetDPort() string {
if *s.DPortStart == 0 {
return ""
} else if *s.DPortEnd == 0 {
return fmt.Sprintf("dport %d", *s.DPortStart)
}
return fmt.Sprintf("dport %d - %d", *s.DPortStart, *s.DPortEnd)
}
type ServiceType int