package main import ( "fmt" "net" "strconv" "strings" "unicode" "github.com/subgraph/fw-daemon/sgfw" "github.com/gotk3/gotk3/gtk" ) const ( editDialogCancel = 1 editDialogOk = 2 ) type ruleEdit struct { row *ruleRow dialog *gtk.Dialog pathLabel *gtk.Label verbCombo *gtk.ComboBoxText hostEntry *gtk.Entry portEntry *gtk.Entry ok *gtk.Button } func newRuleEdit(rr *ruleRow, saveasnew bool) *ruleEdit { redit := &ruleEdit{row: rr} b := newBuilder("RuleEdit") b.getItems( "dialog", &redit.dialog, "path_label", &redit.pathLabel, "verb_combo", &redit.verbCombo, "host_entry", &redit.hostEntry, "port_entry", &redit.portEntry, "ok_button", &redit.ok, ) b.ConnectSignals(map[string]interface{}{ "on_port_insert_text": redit.onPortInsertText, "on_port_changed": redit.onChanged, "on_host_changed": redit.onChanged, }) if saveasnew { redit.ok.SetLabel("Save As New") } return redit } func (re *ruleEdit) updateDialogFields() { r := re.row.rule re.pathLabel.SetText(r.Path) if sgfw.RuleAction(r.Verb) == sgfw.RULE_ACTION_ALLOW { re.verbCombo.SetActiveID("allow") } else { re.verbCombo.SetActiveID("deny") } target := strings.Split(r.Target, ":") if len(target) != 2 { return } re.hostEntry.SetText(target[0]) re.portEntry.SetText(target[1]) } func (re *ruleEdit) validateFields() bool { id := re.verbCombo.GetActiveID() if id != "allow" && id != "deny" { return false } host, _ := re.hostEntry.GetText() port, _ := re.portEntry.GetText() if !isValidHost(host) { return false } if !isValidPort(port) { return false } return true } func isValidHost(host string) bool { if host == "*" { return true } if net.ParseIP(host) != nil { return true } parts := strings.Split(host, ".") if len(parts) < 2 { return false } for _, part := range parts { if part == "" { return false } } return true } func isValidPort(port string) bool { if port == "*" { return true } pval, err := strconv.Atoi(port) if err != nil { return false } return pval > 0 && pval <= 0xFFFF } func (re *ruleEdit) updateRow() { if !re.validateFields() { return } r := re.row.rule switch re.verbCombo.GetActiveID() { case "allow": r.Verb = uint16(sgfw.RULE_ACTION_ALLOW) case "deny": r.Verb = uint16(sgfw.RULE_ACTION_DENY) } host, _ := re.hostEntry.GetText() port, _ := re.portEntry.GetText() r.Target = fmt.Sprintf("%s:%s", host, port) re.row.update() } func (re *ruleEdit) run(saveasnew bool) { re.dialog.SetTransientFor(re.row.rl.win) if re.dialog.Run() == editDialogOk { if saveasnew { re.row.rule.Mode = uint16(sgfw.RULE_MODE_PERMANENT) } re.updateRow() re.row.rl.dbus.updateRule(re.row.rule) if saveasnew { re.row.widget.Hide() } } re.dialog.Destroy() } func (rr *ruleRow) runEditor(saveasnew bool) { redit := newRuleEdit(rr, saveasnew) redit.updateDialogFields() redit.run(saveasnew) } func (re *ruleEdit) onPortInsertText(entry *gtk.Entry, text string) { current, _ := entry.GetText() if current == "" && text == "*" { return } if current == "*" { entry.StopEmission("insert-text") return } for _, c := range text { if !unicode.IsDigit(c) { entry.StopEmission("insert-text") return } } } func (re *ruleEdit) onChanged() { re.ok.SetSensitive(re.validateFields()) }