fw-daemon/fw-settings/rules.go

package main

import (
	"fmt"
	"os"
	"strings"

	"github.com/subgraph/fw-daemon/sgfw"

	"github.com/gotk3/gotk3/gtk"
)

type ruleList struct {
	dbus *dbusObject
	win  *gtk.Window
	list *gtk.ListBox
	col1 *gtk.SizeGroup
	col2 *gtk.SizeGroup
	col3 *gtk.SizeGroup
	col4 *gtk.SizeGroup
	col5 *gtk.SizeGroup
}

type ruleRow struct {
	rl              *ruleList
	rule            *sgfw.DbusRule
	widget          *gtk.ListBoxRow
	gtkLabelApp     *gtk.Label
	gtkLabelVerb    *gtk.Label
	gtkLabelOrigin  *gtk.Label
	gtkLabelPrivs   *gtk.Label
	gtkLabelTarget  *gtk.Label
	gtkButtonEdit   *gtk.Button
	gtkButtonSave   *gtk.Button
	gtkButtonDelete *gtk.Button
}

func newRuleList(dbus *dbusObject, win *gtk.Window, list *gtk.ListBox) *ruleList {
	rl := &ruleList{dbus: dbus, win: win, list: list}
	rl.list.SetSelectionMode(gtk.SELECTION_NONE)
	rl.col1, _ = gtk.SizeGroupNew(gtk.SIZE_GROUP_HORIZONTAL)
	rl.col2, _ = gtk.SizeGroupNew(gtk.SIZE_GROUP_HORIZONTAL)
	rl.col3, _ = gtk.SizeGroupNew(gtk.SIZE_GROUP_HORIZONTAL)
	rl.col4, _ = gtk.SizeGroupNew(gtk.SIZE_GROUP_HORIZONTAL)
	rl.col5, _ = gtk.SizeGroupNew(gtk.SIZE_GROUP_HORIZONTAL)
	return rl
}

func (rl *ruleList) loadRules(mode sgfw.RuleMode) error {
	rules, err := rl.dbus.listRules()
	if err != nil {
		fmt.Fprintf(os.Stderr, "ERROR: %+v\n", err)
		return err
	}
	rl.addRules(rules, mode)
	return nil
}

func (rl *ruleList) addRules(rules []sgfw.DbusRule, mode sgfw.RuleMode) {
	for i := 0; i < len(rules); i++ {
		if sgfw.RuleMode(rules[i].Mode) != mode {
			continue
		}
		row := createWidget(&rules[i])
		row.rl = rl
		rl.col1.AddWidget(row.gtkLabelApp)
		rl.col2.AddWidget(row.gtkLabelVerb)
		rl.col3.AddWidget(row.gtkLabelOrigin)
		rl.col4.AddWidget(row.gtkLabelPrivs)
		rl.col5.AddWidget(row.gtkLabelTarget)
		rl.list.Add(row.widget)
	}
}

func createWidget(rule *sgfw.DbusRule) *ruleRow {
	row := &ruleRow{}
	row.rule = rule
	builder := newBuilder("RuleItem")
	var grid *gtk.Grid
	builder.getItems(
		"grid", &grid,
		"app_label", &row.gtkLabelApp,
		"verb_label", &row.gtkLabelVerb,
		"origin_label", &row.gtkLabelOrigin,
		"privs_label", &row.gtkLabelPrivs,
		"target_label", &row.gtkLabelTarget,
		"edit_button", &row.gtkButtonEdit,
		"save_button", &row.gtkButtonSave,
		"delete_button", &row.gtkButtonDelete,
	)
	switch sgfw.RuleMode(rule.Mode) {
	case sgfw.RULE_MODE_SYSTEM:
		row.gtkButtonEdit.SetVisible(false)
		row.gtkButtonEdit.SetNoShowAll(true)
		row.gtkButtonDelete.SetSensitive(false)
		row.gtkButtonDelete.SetTooltipText("Cannot delete system rules")
		break
	case sgfw.RULE_MODE_SESSION:
		row.gtkButtonSave.SetSensitive(true)
		row.gtkButtonSave.SetNoShowAll(false)
		break
	}

	builder.ConnectSignals(map[string]interface{}{
		"on_edit_rule":   row.onEdit,
		"on_save_rule":   row.onSaveAsNew,
		"on_delete_rule": row.onDelete,
	})
	row.widget, _ = gtk.ListBoxRowNew()
	row.widget.Add(grid)
	row.update()
	return row
}

func (rr *ruleRow) update() {
	rr.gtkLabelApp.SetText(rr.rule.App)
	rr.gtkLabelApp.SetTooltipText(rr.rule.Path)
	rr.gtkLabelVerb.SetText(getVerbText(rr.rule))
	if (rr.rule.Proto == "tcp") {
		rr.gtkLabelOrigin.SetText(rr.rule.Origin)
	} else {
		rr.gtkLabelOrigin.SetText(rr.rule.Origin + " (" + rr.rule.Proto + ")")
	}
	rr.gtkLabelPrivs.SetText(rr.rule.Privs)
	rr.gtkLabelTarget.SetText(getTargetText(rr.rule))
}

func getVerbText(rule *sgfw.DbusRule) string {
	if sgfw.RuleAction(rule.Verb) == sgfw.RULE_ACTION_ALLOW {
		return sgfw.RuleActionString[sgfw.RULE_ACTION_ALLOW] + ":"
	}
	return sgfw.RuleActionString[sgfw.RULE_ACTION_DENY] + ":"
}

func getTargetText(rule *sgfw.DbusRule) string {
	if rule.Target == "*:*" {
		return "All connections"
	}
	items := strings.Split(rule.Target, ":")

	if len(items) != 2 {
		return rule.Target
	}

	if items[0] == "*" {
		if rule.Proto == "tcp" {
			return fmt.Sprintf("Connections to ALL hosts on port %s", items[1])
		} else if rule.Proto == "icmp" {
			return fmt.Sprintf("Data to ALL hosts with ICMP code %s", items[1])
		}
		return fmt.Sprintf("Data to ALL hosts on port %s", items[1])
	}
	if items[1] == "*" {
		if rule.Proto == "tcp" {
			return fmt.Sprintf("All connections to host %s", items[0])
		}
		return fmt.Sprintf("All data to host %s", items[0])
	}

	if rule.Proto == "tcp" {
		return fmt.Sprintf("Connections to %s on port %s", items[0], items[1])
	} else if rule.Proto == "icmp" {
		return fmt.Sprintf("Data to %s with ICMP code %s", items[0], items[1])
	}
	return fmt.Sprintf("Data to %s on port %s", items[0], items[1])
}

func (rr *ruleRow) onSaveAsNew() {
	rr.runEditor(true)
}

func (rr *ruleRow) onEdit() {
	rr.runEditor(false)
}

func (rr *ruleRow) onDelete() {
	body := fmt.Sprintf(`Are you sure you want to delete this rule:

	<b>Path:</b>   %s

	<b>Rule:</b>   %s %s`, rr.rule.Path, getVerbText(rr.rule), getTargetText(rr.rule))
	d := gtk.MessageDialogNewWithMarkup(
		rr.rl.win,
		gtk.DIALOG_DESTROY_WITH_PARENT,
		gtk.MESSAGE_QUESTION,
		gtk.BUTTONS_OK_CANCEL,
		"")
	d.SetMarkup(body)
	if d.Run() == (int)(gtk.RESPONSE_OK) {
		rr.delete()
	}
	d.Destroy()

}

func (rl *ruleList) remove(rr *ruleRow) {
	rl.col1.RemoveWidget(rr.gtkLabelApp)
	rl.col2.RemoveWidget(rr.gtkLabelVerb)
	rl.col3.RemoveWidget(rr.gtkLabelOrigin)
	rl.col4.RemoveWidget(rr.gtkLabelPrivs)
	rl.col5.RemoveWidget(rr.gtkLabelTarget)
	rl.list.Remove(rr.widget)
}

func (rr *ruleRow) delete() {
	rr.rl.remove(rr)
	rr.rl.dbus.deleteRule(rr.rule.ID)
}