93 lines
2.5 KiB
Go
93 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 {
|
||
|
|
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() *UCIFirewallRedirect {
|
||
|
|
return &UCIFirewallRedirect{}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Create add a new firewall rule in UCI Configuration
|
||
|
|
func (rd *UCIFirewallRedirect) Create(uci *UCI) *Action {
|
||
|
|
var confPrefix string
|
||
|
|
|
||
|
|
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(uci *UCI) *Action {
|
||
|
|
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(uci *UCI) *Action {
|
||
|
|
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(uci *UCI) *Action {
|
||
|
|
rd.Delete(uci)
|
||
|
|
create := rd.Create(uci)
|
||
|
|
if create.ReturnCode != 0 {
|
||
|
|
return create
|
||
|
|
}
|
||
|
|
return uci.Commit()
|
||
|
|
}
|