96 lines
2.5 KiB
Go
96 lines
2.5 KiB
Go
package owrt
|
|
|
|
import (
|
|
"fmt"
|
|
)
|
|
|
|
// UCIFirewallRedirect is the description of an Wireless interface (cf Openwrt doc) on top of an Wireless Device
|
|
type UCIFirewallRedirect struct {
|
|
UCI *UCI
|
|
Name string
|
|
Index int
|
|
Src string
|
|
Dest string
|
|
Target string
|
|
Proto string
|
|
SrcDIP string
|
|
SrcDPort string
|
|
DestIP string
|
|
DestPort string
|
|
}
|
|
|
|
// NewUCIFirewallRedirect builds a new UCIFirewallRedirect instance
|
|
func NewUCIFirewallRedirect(uci *UCI) *UCIFirewallRedirect {
|
|
return &UCIFirewallRedirect{UCI: uci}
|
|
}
|
|
|
|
// Create add a new firewall rule in UCI Configuration
|
|
func (rd *UCIFirewallRedirect) Create() *Action {
|
|
uci := rd.UCI
|
|
confPrefix := fmt.Sprintf("firewall.@redirect[%d]", rd.Index)
|
|
|
|
conf := make(map[string][]string)
|
|
|
|
conf["name"] = append(conf["name"], fmt.Sprintf("%s.name", confPrefix), rd.Name)
|
|
conf["src"] = append(conf["src"], fmt.Sprintf("%s.src", confPrefix), rd.Src)
|
|
conf["target"] = append(conf["target"], fmt.Sprintf("%s.target", confPrefix), rd.Target)
|
|
conf["proto"] = append(conf["proto"], fmt.Sprintf("%s.proto", confPrefix), rd.Proto)
|
|
conf["src_dip"] = append(conf["src_dip"], fmt.Sprintf("%s.src_dip", confPrefix), rd.SrcDIP)
|
|
conf["src_dport"] = append(conf["src_dport"], fmt.Sprintf("%s.src_dport", confPrefix), rd.SrcDPort)
|
|
conf["dest_ip"] = append(conf["dest_ip"], fmt.Sprintf("%s.dest_ip", confPrefix), rd.DestIP)
|
|
conf["dest_port"] = append(conf["dest_port"], fmt.Sprintf("%s.dest_port", confPrefix), rd.DestPort)
|
|
|
|
uci.Add("firewall", "redirect")
|
|
|
|
for _, value := range conf {
|
|
if value[1] != "" {
|
|
result := uci.Set(value[0], value[1])
|
|
if result.ReturnCode != 0 {
|
|
return result
|
|
}
|
|
}
|
|
}
|
|
|
|
return &Action{
|
|
CommandResult: &CommandResult{
|
|
Stdout: "",
|
|
Stderr: "",
|
|
ReturnCode: 0,
|
|
},
|
|
}
|
|
}
|
|
|
|
// Save commit and relaod configuration (writes it to files !)
|
|
func (rd *UCIFirewallRedirect) Save() *Action {
|
|
uci := rd.UCI
|
|
commitRes := uci.Commit()
|
|
if commitRes.ReturnCode != 0 {
|
|
return commitRes
|
|
}
|
|
|
|
reload := uci.Reload()
|
|
return reload
|
|
}
|
|
|
|
// Delete remove wifi interface from UCI Configuration
|
|
func (rd *UCIFirewallRedirect) Delete() *Action {
|
|
uci := rd.UCI
|
|
toDelete := fmt.Sprintf("firewall.@redirect[%d]", rd.Index)
|
|
del := uci.Delete(toDelete)
|
|
if del.ReturnCode != 0 {
|
|
return del
|
|
}
|
|
return uci.Commit()
|
|
}
|
|
|
|
// Update add a new entry for wifi interface in UCI Configuration
|
|
func (rd *UCIFirewallRedirect) Update() *Action {
|
|
uci := rd.UCI
|
|
rd.Delete()
|
|
create := rd.Create()
|
|
if create.ReturnCode != 0 {
|
|
return create
|
|
}
|
|
return uci.Commit()
|
|
}
|