adelorenzo/kilo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ipv4 and ipv6 detection using regex

Browse Source
This commit is contained in:
Paulo Nascimento 2021-03-20 19:35:35 -03:00
parent 832360a02c
commit 4101e3fc44
1 changed files with 3 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
pkg/iptables/iptables.go Normal file → Executable file
View File

@ -73,7 +73,7 @@ type rule struct {
proto Protocol proto Protocol
} }
var ipv6Regex,_ = regexp.Compile("[-]d\\s(.*:.*\\s[-]m\\scomment)") var ipv6Regex, _ = regexp.Compile("[-]d\\s(.*:.*\\s[-]m\\scomment)")
// NewRule creates a new iptables or ip6tables rule in the given table and chain // NewRule creates a new iptables or ip6tables rule in the given table and chain
// depending on the given protocol. // depending on the given protocol.
@ -380,21 +380,10 @@ func (c *Controller) Set(rules []Rule) error {
} }
} }
if i >= len(c.rules) { if i >= len(c.rules) {
proto := rules[i].Proto() if err := rules[i].Add(c.client(rules[i].Proto())); err != nil {
protocolName := "ipv4"
if proto == ProtocolIPv6 {
protocolName = "ipv6"
}
var ruleString = rules[i].String()
level.Debug(c.logger).Log("msg", "Applying Firewall Rule...", "Rule", ruleString, "Protocol", protocolName)
if err := rules[i].Add(c.v4); err != nil {
return fmt.Errorf("failed to add rule: %v", err) return fmt.Errorf("failed to add rule: %v", err)
} }
level.Debug(c.logger).Log("msg", "Firewall Rule applied.", "Rule", ruleString, "Protocol", protocolName) c.rules = append(c.rulIes, rules[i])
c.rules = append(c.rules, rules[i])
} }
} }