feat(fw): Models + Repos für Firewall-v2 (6 Entities)
Models (internal/models/): * FirewallAddressObject (host|network|range|fqdn) * FirewallAddressGroup mit MemberIDs gorm:"-"-Slice * FirewallService (proto+ports, builtin-Flag) * FirewallServiceGroup mit MemberIDs * FirewallRule (v2-Shape, src/dst nullable refs, exactly-one-of-Validation in Handler-Layer) * FirewallNATRule (kind=dnat|snat|masquerade, alle nullable) Repos (internal/services/firewall/, ein Paket): * AddressObjectsRepo, AddressGroupsRepo (mit Members-Junction-Ops) * ServicesRepo (refused Update/Delete für builtin=TRUE Rows), ServiceGroupsRepo * RulesRepo, NATRulesRepo Jeweils Standard-CRUD; Group-Repos handhaben Members atomic in einer TX (Update ersetzt komplette Membership). Handler + Renderer-Rewrite + Frontend folgen in den nächsten Commits. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
This commit is contained in:
185
internal/services/firewall/addressgroups.go
Normal file
185
internal/services/firewall/addressgroups.go
Normal file
@@ -0,0 +1,185 @@
|
||||
package firewall
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
|
||||
"github.com/jackc/pgx/v5"
|
||||
"github.com/jackc/pgx/v5/pgxpool"
|
||||
|
||||
"git.netcell-it.de/projekte/edgeguard-native/internal/models"
|
||||
)
|
||||
|
||||
var ErrAddressGroupNotFound = errors.New("address group not found")
|
||||
|
||||
type AddressGroupsRepo struct {
|
||||
Pool *pgxpool.Pool
|
||||
}
|
||||
|
||||
func NewAddressGroupsRepo(pool *pgxpool.Pool) *AddressGroupsRepo {
|
||||
return &AddressGroupsRepo{Pool: pool}
|
||||
}
|
||||
|
||||
const addrGrpBaseSelect = `
|
||||
SELECT id, name, description, created_at, updated_at
|
||||
FROM firewall_address_groups
|
||||
`
|
||||
|
||||
// List returns all groups with their MemberIDs populated via a
|
||||
// single follow-up query. Two roundtrips total — keeps the SQL
|
||||
// simple at the cost of one extra query for an inherently small
|
||||
// table.
|
||||
func (r *AddressGroupsRepo) List(ctx context.Context) ([]models.FirewallAddressGroup, error) {
|
||||
rows, err := r.Pool.Query(ctx, addrGrpBaseSelect+" ORDER BY name ASC")
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer rows.Close()
|
||||
groups := []models.FirewallAddressGroup{}
|
||||
byID := map[int64]int{}
|
||||
for rows.Next() {
|
||||
g, err := scanAddrGrp(rows)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
byID[g.ID] = len(groups)
|
||||
groups = append(groups, *g)
|
||||
}
|
||||
if err := rows.Err(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
mRows, err := r.Pool.Query(ctx, `SELECT group_id, object_id FROM firewall_address_group_members`)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer mRows.Close()
|
||||
for mRows.Next() {
|
||||
var gid, oid int64
|
||||
if err := mRows.Scan(&gid, &oid); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if idx, ok := byID[gid]; ok {
|
||||
groups[idx].MemberIDs = append(groups[idx].MemberIDs, oid)
|
||||
}
|
||||
}
|
||||
return groups, mRows.Err()
|
||||
}
|
||||
|
||||
func (r *AddressGroupsRepo) Get(ctx context.Context, id int64) (*models.FirewallAddressGroup, error) {
|
||||
row := r.Pool.QueryRow(ctx, addrGrpBaseSelect+" WHERE id = $1", id)
|
||||
g, err := scanAddrGrp(row)
|
||||
if err != nil {
|
||||
if errors.Is(err, pgx.ErrNoRows) {
|
||||
return nil, ErrAddressGroupNotFound
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
mRows, err := r.Pool.Query(ctx,
|
||||
`SELECT object_id FROM firewall_address_group_members WHERE group_id = $1 ORDER BY object_id`, id)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer mRows.Close()
|
||||
for mRows.Next() {
|
||||
var oid int64
|
||||
if err := mRows.Scan(&oid); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
g.MemberIDs = append(g.MemberIDs, oid)
|
||||
}
|
||||
return g, mRows.Err()
|
||||
}
|
||||
|
||||
// Create inserts the group and (optionally) its members atomically.
|
||||
func (r *AddressGroupsRepo) Create(ctx context.Context, g models.FirewallAddressGroup) (*models.FirewallAddressGroup, error) {
|
||||
tx, err := r.Pool.Begin(ctx)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer tx.Rollback(ctx)
|
||||
|
||||
row := tx.QueryRow(ctx, `
|
||||
INSERT INTO firewall_address_groups (name, description)
|
||||
VALUES ($1, $2)
|
||||
RETURNING id, name, description, created_at, updated_at`,
|
||||
g.Name, g.Description)
|
||||
out, err := scanAddrGrp(row)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if len(g.MemberIDs) > 0 {
|
||||
if err := insertAddrGrpMembers(ctx, tx, out.ID, g.MemberIDs); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
out.MemberIDs = append([]int64{}, g.MemberIDs...)
|
||||
}
|
||||
return out, tx.Commit(ctx)
|
||||
}
|
||||
|
||||
// Update replaces both the metadata and the membership set
|
||||
// atomically. Pass the desired complete member list; absent IDs are
|
||||
// removed.
|
||||
func (r *AddressGroupsRepo) Update(ctx context.Context, id int64, g models.FirewallAddressGroup) (*models.FirewallAddressGroup, error) {
|
||||
tx, err := r.Pool.Begin(ctx)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer tx.Rollback(ctx)
|
||||
|
||||
row := tx.QueryRow(ctx, `
|
||||
UPDATE firewall_address_groups SET name = $1, description = $2, updated_at = NOW()
|
||||
WHERE id = $3
|
||||
RETURNING id, name, description, created_at, updated_at`,
|
||||
g.Name, g.Description, id)
|
||||
out, err := scanAddrGrp(row)
|
||||
if err != nil {
|
||||
if errors.Is(err, pgx.ErrNoRows) {
|
||||
return nil, ErrAddressGroupNotFound
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
if _, err := tx.Exec(ctx, `DELETE FROM firewall_address_group_members WHERE group_id = $1`, id); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if len(g.MemberIDs) > 0 {
|
||||
if err := insertAddrGrpMembers(ctx, tx, id, g.MemberIDs); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
out.MemberIDs = append([]int64{}, g.MemberIDs...)
|
||||
}
|
||||
return out, tx.Commit(ctx)
|
||||
}
|
||||
|
||||
func (r *AddressGroupsRepo) Delete(ctx context.Context, id int64) error {
|
||||
tag, err := r.Pool.Exec(ctx, `DELETE FROM firewall_address_groups WHERE id = $1`, id)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if tag.RowsAffected() == 0 {
|
||||
return ErrAddressGroupNotFound
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func insertAddrGrpMembers(ctx context.Context, tx pgx.Tx, gid int64, members []int64) error {
|
||||
for _, oid := range members {
|
||||
if _, err := tx.Exec(ctx,
|
||||
`INSERT INTO firewall_address_group_members (group_id, object_id) VALUES ($1, $2) ON CONFLICT DO NOTHING`,
|
||||
gid, oid); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func scanAddrGrp(row interface{ Scan(...any) error }) (*models.FirewallAddressGroup, error) {
|
||||
var g models.FirewallAddressGroup
|
||||
if err := row.Scan(
|
||||
&g.ID, &g.Name, &g.Description,
|
||||
&g.CreatedAt, &g.UpdatedAt,
|
||||
); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &g, nil
|
||||
}
|
||||
103
internal/services/firewall/addressobjects.go
Normal file
103
internal/services/firewall/addressobjects.go
Normal file
@@ -0,0 +1,103 @@
|
||||
package firewall
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
|
||||
"github.com/jackc/pgx/v5"
|
||||
"github.com/jackc/pgx/v5/pgxpool"
|
||||
|
||||
"git.netcell-it.de/projekte/edgeguard-native/internal/models"
|
||||
)
|
||||
|
||||
var ErrAddressObjectNotFound = errors.New("address object not found")
|
||||
|
||||
type AddressObjectsRepo struct {
|
||||
Pool *pgxpool.Pool
|
||||
}
|
||||
|
||||
func NewAddressObjectsRepo(pool *pgxpool.Pool) *AddressObjectsRepo {
|
||||
return &AddressObjectsRepo{Pool: pool}
|
||||
}
|
||||
|
||||
const addrObjBaseSelect = `
|
||||
SELECT id, name, kind, value, description, created_at, updated_at
|
||||
FROM firewall_address_objects
|
||||
`
|
||||
|
||||
func (r *AddressObjectsRepo) List(ctx context.Context) ([]models.FirewallAddressObject, error) {
|
||||
rows, err := r.Pool.Query(ctx, addrObjBaseSelect+" ORDER BY name ASC")
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer rows.Close()
|
||||
out := make([]models.FirewallAddressObject, 0, 8)
|
||||
for rows.Next() {
|
||||
o, err := scanAddrObj(rows)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
out = append(out, *o)
|
||||
}
|
||||
return out, rows.Err()
|
||||
}
|
||||
|
||||
func (r *AddressObjectsRepo) Get(ctx context.Context, id int64) (*models.FirewallAddressObject, error) {
|
||||
row := r.Pool.QueryRow(ctx, addrObjBaseSelect+" WHERE id = $1", id)
|
||||
o, err := scanAddrObj(row)
|
||||
if err != nil {
|
||||
if errors.Is(err, pgx.ErrNoRows) {
|
||||
return nil, ErrAddressObjectNotFound
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
return o, nil
|
||||
}
|
||||
|
||||
func (r *AddressObjectsRepo) Create(ctx context.Context, o models.FirewallAddressObject) (*models.FirewallAddressObject, error) {
|
||||
row := r.Pool.QueryRow(ctx, `
|
||||
INSERT INTO firewall_address_objects (name, kind, value, description)
|
||||
VALUES ($1, $2, $3, $4)
|
||||
RETURNING id, name, kind, value, description, created_at, updated_at`,
|
||||
o.Name, o.Kind, o.Value, o.Description)
|
||||
return scanAddrObj(row)
|
||||
}
|
||||
|
||||
func (r *AddressObjectsRepo) Update(ctx context.Context, id int64, o models.FirewallAddressObject) (*models.FirewallAddressObject, error) {
|
||||
row := r.Pool.QueryRow(ctx, `
|
||||
UPDATE firewall_address_objects SET
|
||||
name = $1, kind = $2, value = $3, description = $4, updated_at = NOW()
|
||||
WHERE id = $5
|
||||
RETURNING id, name, kind, value, description, created_at, updated_at`,
|
||||
o.Name, o.Kind, o.Value, o.Description, id)
|
||||
out, err := scanAddrObj(row)
|
||||
if err != nil {
|
||||
if errors.Is(err, pgx.ErrNoRows) {
|
||||
return nil, ErrAddressObjectNotFound
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
return out, nil
|
||||
}
|
||||
|
||||
func (r *AddressObjectsRepo) Delete(ctx context.Context, id int64) error {
|
||||
tag, err := r.Pool.Exec(ctx, `DELETE FROM firewall_address_objects WHERE id = $1`, id)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if tag.RowsAffected() == 0 {
|
||||
return ErrAddressObjectNotFound
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func scanAddrObj(row interface{ Scan(...any) error }) (*models.FirewallAddressObject, error) {
|
||||
var o models.FirewallAddressObject
|
||||
if err := row.Scan(
|
||||
&o.ID, &o.Name, &o.Kind, &o.Value,
|
||||
&o.Description, &o.CreatedAt, &o.UpdatedAt,
|
||||
); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &o, nil
|
||||
}
|
||||
7
internal/services/firewall/doc.go
Normal file
7
internal/services/firewall/doc.go
Normal file
@@ -0,0 +1,7 @@
|
||||
// Package firewall holds the v2 (Fortigate-style) firewall data
|
||||
// repos: address objects + groups, services + groups, policy rules,
|
||||
// and NAT rules. Each entity has its own *.go file; the public
|
||||
// surface is one Repo per entity, all sharing the same *pgxpool.Pool.
|
||||
//
|
||||
// Render-Logik (Joins zu nftables) wohnt in internal/firewall/.
|
||||
package firewall
|
||||
139
internal/services/firewall/natrules.go
Normal file
139
internal/services/firewall/natrules.go
Normal file
@@ -0,0 +1,139 @@
|
||||
package firewall
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
|
||||
"github.com/jackc/pgx/v5"
|
||||
"github.com/jackc/pgx/v5/pgxpool"
|
||||
|
||||
"git.netcell-it.de/projekte/edgeguard-native/internal/models"
|
||||
)
|
||||
|
||||
var ErrNATRuleNotFound = errors.New("nat rule not found")
|
||||
|
||||
type NATRulesRepo struct {
|
||||
Pool *pgxpool.Pool
|
||||
}
|
||||
|
||||
func NewNATRulesRepo(pool *pgxpool.Pool) *NATRulesRepo { return &NATRulesRepo{Pool: pool} }
|
||||
|
||||
const natRuleBaseSelect = `
|
||||
SELECT id, name, priority, enabled, kind,
|
||||
in_zone, out_zone, proto,
|
||||
match_src_cidr, match_dst_cidr, match_dport_start, match_dport_end,
|
||||
target_addr, target_port_start, target_port_end,
|
||||
comment, created_at, updated_at
|
||||
FROM firewall_nat_rules
|
||||
`
|
||||
|
||||
func (r *NATRulesRepo) List(ctx context.Context) ([]models.FirewallNATRule, error) {
|
||||
rows, err := r.Pool.Query(ctx, natRuleBaseSelect+" ORDER BY priority DESC, id ASC")
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer rows.Close()
|
||||
out := make([]models.FirewallNATRule, 0, 8)
|
||||
for rows.Next() {
|
||||
x, err := scanNATRule(rows)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
out = append(out, *x)
|
||||
}
|
||||
return out, rows.Err()
|
||||
}
|
||||
|
||||
func (r *NATRulesRepo) Get(ctx context.Context, id int64) (*models.FirewallNATRule, error) {
|
||||
row := r.Pool.QueryRow(ctx, natRuleBaseSelect+" WHERE id = $1", id)
|
||||
x, err := scanNATRule(row)
|
||||
if err != nil {
|
||||
if errors.Is(err, pgx.ErrNoRows) {
|
||||
return nil, ErrNATRuleNotFound
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
return x, nil
|
||||
}
|
||||
|
||||
func (r *NATRulesRepo) Create(ctx context.Context, x models.FirewallNATRule) (*models.FirewallNATRule, error) {
|
||||
row := r.Pool.QueryRow(ctx, `
|
||||
INSERT INTO firewall_nat_rules (
|
||||
name, priority, enabled, kind,
|
||||
in_zone, out_zone, proto,
|
||||
match_src_cidr, match_dst_cidr, match_dport_start, match_dport_end,
|
||||
target_addr, target_port_start, target_port_end,
|
||||
comment
|
||||
) VALUES (
|
||||
$1, $2, $3, $4,
|
||||
$5, $6, $7,
|
||||
$8, $9, $10, $11,
|
||||
$12, $13, $14,
|
||||
$15
|
||||
)
|
||||
RETURNING id, name, priority, enabled, kind,
|
||||
in_zone, out_zone, proto,
|
||||
match_src_cidr, match_dst_cidr, match_dport_start, match_dport_end,
|
||||
target_addr, target_port_start, target_port_end,
|
||||
comment, created_at, updated_at`,
|
||||
x.Name, x.Priority, x.Enabled, x.Kind,
|
||||
x.InZone, x.OutZone, x.Proto,
|
||||
x.MatchSrcCIDR, x.MatchDstCIDR, x.MatchDPortStart, x.MatchDPortEnd,
|
||||
x.TargetAddr, x.TargetPortStart, x.TargetPortEnd,
|
||||
x.Comment)
|
||||
return scanNATRule(row)
|
||||
}
|
||||
|
||||
func (r *NATRulesRepo) Update(ctx context.Context, id int64, x models.FirewallNATRule) (*models.FirewallNATRule, error) {
|
||||
row := r.Pool.QueryRow(ctx, `
|
||||
UPDATE firewall_nat_rules SET
|
||||
name = $1, priority = $2, enabled = $3, kind = $4,
|
||||
in_zone = $5, out_zone = $6, proto = $7,
|
||||
match_src_cidr = $8, match_dst_cidr = $9, match_dport_start = $10, match_dport_end = $11,
|
||||
target_addr = $12, target_port_start = $13, target_port_end = $14,
|
||||
comment = $15, updated_at = NOW()
|
||||
WHERE id = $16
|
||||
RETURNING id, name, priority, enabled, kind,
|
||||
in_zone, out_zone, proto,
|
||||
match_src_cidr, match_dst_cidr, match_dport_start, match_dport_end,
|
||||
target_addr, target_port_start, target_port_end,
|
||||
comment, created_at, updated_at`,
|
||||
x.Name, x.Priority, x.Enabled, x.Kind,
|
||||
x.InZone, x.OutZone, x.Proto,
|
||||
x.MatchSrcCIDR, x.MatchDstCIDR, x.MatchDPortStart, x.MatchDPortEnd,
|
||||
x.TargetAddr, x.TargetPortStart, x.TargetPortEnd,
|
||||
x.Comment, id)
|
||||
out, err := scanNATRule(row)
|
||||
if err != nil {
|
||||
if errors.Is(err, pgx.ErrNoRows) {
|
||||
return nil, ErrNATRuleNotFound
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
return out, nil
|
||||
}
|
||||
|
||||
func (r *NATRulesRepo) Delete(ctx context.Context, id int64) error {
|
||||
tag, err := r.Pool.Exec(ctx, `DELETE FROM firewall_nat_rules WHERE id = $1`, id)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if tag.RowsAffected() == 0 {
|
||||
return ErrNATRuleNotFound
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func scanNATRule(row interface{ Scan(...any) error }) (*models.FirewallNATRule, error) {
|
||||
var x models.FirewallNATRule
|
||||
if err := row.Scan(
|
||||
&x.ID, &x.Name, &x.Priority, &x.Enabled, &x.Kind,
|
||||
&x.InZone, &x.OutZone, &x.Proto,
|
||||
&x.MatchSrcCIDR, &x.MatchDstCIDR, &x.MatchDPortStart, &x.MatchDPortEnd,
|
||||
&x.TargetAddr, &x.TargetPortStart, &x.TargetPortEnd,
|
||||
&x.Comment, &x.CreatedAt, &x.UpdatedAt,
|
||||
); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &x, nil
|
||||
}
|
||||
139
internal/services/firewall/rules.go
Normal file
139
internal/services/firewall/rules.go
Normal file
@@ -0,0 +1,139 @@
|
||||
package firewall
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
|
||||
"github.com/jackc/pgx/v5"
|
||||
"github.com/jackc/pgx/v5/pgxpool"
|
||||
|
||||
"git.netcell-it.de/projekte/edgeguard-native/internal/models"
|
||||
)
|
||||
|
||||
var ErrRuleNotFound = errors.New("firewall rule not found")
|
||||
|
||||
type RulesRepo struct {
|
||||
Pool *pgxpool.Pool
|
||||
}
|
||||
|
||||
func NewRulesRepo(pool *pgxpool.Pool) *RulesRepo { return &RulesRepo{Pool: pool} }
|
||||
|
||||
const ruleBaseSelect = `
|
||||
SELECT id, name, priority, enabled, action,
|
||||
src_zone, src_address_object_id, src_address_group_id, src_cidr,
|
||||
dst_zone, dst_address_object_id, dst_address_group_id, dst_cidr,
|
||||
service_object_id, service_group_id,
|
||||
log, comment, created_at, updated_at
|
||||
FROM firewall_rules
|
||||
`
|
||||
|
||||
func (r *RulesRepo) List(ctx context.Context) ([]models.FirewallRule, error) {
|
||||
rows, err := r.Pool.Query(ctx, ruleBaseSelect+" ORDER BY priority DESC, id ASC")
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer rows.Close()
|
||||
out := make([]models.FirewallRule, 0, 16)
|
||||
for rows.Next() {
|
||||
x, err := scanRule(rows)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
out = append(out, *x)
|
||||
}
|
||||
return out, rows.Err()
|
||||
}
|
||||
|
||||
func (r *RulesRepo) Get(ctx context.Context, id int64) (*models.FirewallRule, error) {
|
||||
row := r.Pool.QueryRow(ctx, ruleBaseSelect+" WHERE id = $1", id)
|
||||
x, err := scanRule(row)
|
||||
if err != nil {
|
||||
if errors.Is(err, pgx.ErrNoRows) {
|
||||
return nil, ErrRuleNotFound
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
return x, nil
|
||||
}
|
||||
|
||||
func (r *RulesRepo) Create(ctx context.Context, x models.FirewallRule) (*models.FirewallRule, error) {
|
||||
row := r.Pool.QueryRow(ctx, `
|
||||
INSERT INTO firewall_rules (
|
||||
name, priority, enabled, action,
|
||||
src_zone, src_address_object_id, src_address_group_id, src_cidr,
|
||||
dst_zone, dst_address_object_id, dst_address_group_id, dst_cidr,
|
||||
service_object_id, service_group_id,
|
||||
log, comment
|
||||
) VALUES (
|
||||
$1, $2, $3, $4,
|
||||
$5, $6, $7, $8,
|
||||
$9, $10, $11, $12,
|
||||
$13, $14,
|
||||
$15, $16
|
||||
)
|
||||
RETURNING id, name, priority, enabled, action,
|
||||
src_zone, src_address_object_id, src_address_group_id, src_cidr,
|
||||
dst_zone, dst_address_object_id, dst_address_group_id, dst_cidr,
|
||||
service_object_id, service_group_id,
|
||||
log, comment, created_at, updated_at`,
|
||||
x.Name, x.Priority, x.Enabled, x.Action,
|
||||
x.SrcZone, x.SrcAddressObjectID, x.SrcAddressGroupID, x.SrcCIDR,
|
||||
x.DstZone, x.DstAddressObjectID, x.DstAddressGroupID, x.DstCIDR,
|
||||
x.ServiceObjectID, x.ServiceGroupID,
|
||||
x.Log, x.Comment)
|
||||
return scanRule(row)
|
||||
}
|
||||
|
||||
func (r *RulesRepo) Update(ctx context.Context, id int64, x models.FirewallRule) (*models.FirewallRule, error) {
|
||||
row := r.Pool.QueryRow(ctx, `
|
||||
UPDATE firewall_rules SET
|
||||
name = $1, priority = $2, enabled = $3, action = $4,
|
||||
src_zone = $5, src_address_object_id = $6, src_address_group_id = $7, src_cidr = $8,
|
||||
dst_zone = $9, dst_address_object_id = $10, dst_address_group_id = $11, dst_cidr = $12,
|
||||
service_object_id = $13, service_group_id = $14,
|
||||
log = $15, comment = $16, updated_at = NOW()
|
||||
WHERE id = $17
|
||||
RETURNING id, name, priority, enabled, action,
|
||||
src_zone, src_address_object_id, src_address_group_id, src_cidr,
|
||||
dst_zone, dst_address_object_id, dst_address_group_id, dst_cidr,
|
||||
service_object_id, service_group_id,
|
||||
log, comment, created_at, updated_at`,
|
||||
x.Name, x.Priority, x.Enabled, x.Action,
|
||||
x.SrcZone, x.SrcAddressObjectID, x.SrcAddressGroupID, x.SrcCIDR,
|
||||
x.DstZone, x.DstAddressObjectID, x.DstAddressGroupID, x.DstCIDR,
|
||||
x.ServiceObjectID, x.ServiceGroupID,
|
||||
x.Log, x.Comment, id)
|
||||
out, err := scanRule(row)
|
||||
if err != nil {
|
||||
if errors.Is(err, pgx.ErrNoRows) {
|
||||
return nil, ErrRuleNotFound
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
return out, nil
|
||||
}
|
||||
|
||||
func (r *RulesRepo) Delete(ctx context.Context, id int64) error {
|
||||
tag, err := r.Pool.Exec(ctx, `DELETE FROM firewall_rules WHERE id = $1`, id)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if tag.RowsAffected() == 0 {
|
||||
return ErrRuleNotFound
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func scanRule(row interface{ Scan(...any) error }) (*models.FirewallRule, error) {
|
||||
var x models.FirewallRule
|
||||
if err := row.Scan(
|
||||
&x.ID, &x.Name, &x.Priority, &x.Enabled, &x.Action,
|
||||
&x.SrcZone, &x.SrcAddressObjectID, &x.SrcAddressGroupID, &x.SrcCIDR,
|
||||
&x.DstZone, &x.DstAddressObjectID, &x.DstAddressGroupID, &x.DstCIDR,
|
||||
&x.ServiceObjectID, &x.ServiceGroupID,
|
||||
&x.Log, &x.Comment, &x.CreatedAt, &x.UpdatedAt,
|
||||
); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &x, nil
|
||||
}
|
||||
175
internal/services/firewall/servicegroups.go
Normal file
175
internal/services/firewall/servicegroups.go
Normal file
@@ -0,0 +1,175 @@
|
||||
package firewall
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
|
||||
"github.com/jackc/pgx/v5"
|
||||
"github.com/jackc/pgx/v5/pgxpool"
|
||||
|
||||
"git.netcell-it.de/projekte/edgeguard-native/internal/models"
|
||||
)
|
||||
|
||||
var ErrServiceGroupNotFound = errors.New("service group not found")
|
||||
|
||||
type ServiceGroupsRepo struct {
|
||||
Pool *pgxpool.Pool
|
||||
}
|
||||
|
||||
func NewServiceGroupsRepo(pool *pgxpool.Pool) *ServiceGroupsRepo {
|
||||
return &ServiceGroupsRepo{Pool: pool}
|
||||
}
|
||||
|
||||
const svcGrpBaseSelect = `
|
||||
SELECT id, name, description, created_at, updated_at
|
||||
FROM firewall_service_groups
|
||||
`
|
||||
|
||||
func (r *ServiceGroupsRepo) List(ctx context.Context) ([]models.FirewallServiceGroup, error) {
|
||||
rows, err := r.Pool.Query(ctx, svcGrpBaseSelect+" ORDER BY name ASC")
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer rows.Close()
|
||||
groups := []models.FirewallServiceGroup{}
|
||||
byID := map[int64]int{}
|
||||
for rows.Next() {
|
||||
g, err := scanSvcGrp(rows)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
byID[g.ID] = len(groups)
|
||||
groups = append(groups, *g)
|
||||
}
|
||||
if err := rows.Err(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
mRows, err := r.Pool.Query(ctx, `SELECT group_id, service_id FROM firewall_service_group_members`)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer mRows.Close()
|
||||
for mRows.Next() {
|
||||
var gid, sid int64
|
||||
if err := mRows.Scan(&gid, &sid); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if idx, ok := byID[gid]; ok {
|
||||
groups[idx].MemberIDs = append(groups[idx].MemberIDs, sid)
|
||||
}
|
||||
}
|
||||
return groups, mRows.Err()
|
||||
}
|
||||
|
||||
func (r *ServiceGroupsRepo) Get(ctx context.Context, id int64) (*models.FirewallServiceGroup, error) {
|
||||
row := r.Pool.QueryRow(ctx, svcGrpBaseSelect+" WHERE id = $1", id)
|
||||
g, err := scanSvcGrp(row)
|
||||
if err != nil {
|
||||
if errors.Is(err, pgx.ErrNoRows) {
|
||||
return nil, ErrServiceGroupNotFound
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
mRows, err := r.Pool.Query(ctx,
|
||||
`SELECT service_id FROM firewall_service_group_members WHERE group_id = $1 ORDER BY service_id`, id)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer mRows.Close()
|
||||
for mRows.Next() {
|
||||
var sid int64
|
||||
if err := mRows.Scan(&sid); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
g.MemberIDs = append(g.MemberIDs, sid)
|
||||
}
|
||||
return g, mRows.Err()
|
||||
}
|
||||
|
||||
func (r *ServiceGroupsRepo) Create(ctx context.Context, g models.FirewallServiceGroup) (*models.FirewallServiceGroup, error) {
|
||||
tx, err := r.Pool.Begin(ctx)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer tx.Rollback(ctx)
|
||||
|
||||
row := tx.QueryRow(ctx, `
|
||||
INSERT INTO firewall_service_groups (name, description) VALUES ($1, $2)
|
||||
RETURNING id, name, description, created_at, updated_at`,
|
||||
g.Name, g.Description)
|
||||
out, err := scanSvcGrp(row)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if len(g.MemberIDs) > 0 {
|
||||
if err := insertSvcGrpMembers(ctx, tx, out.ID, g.MemberIDs); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
out.MemberIDs = append([]int64{}, g.MemberIDs...)
|
||||
}
|
||||
return out, tx.Commit(ctx)
|
||||
}
|
||||
|
||||
func (r *ServiceGroupsRepo) Update(ctx context.Context, id int64, g models.FirewallServiceGroup) (*models.FirewallServiceGroup, error) {
|
||||
tx, err := r.Pool.Begin(ctx)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer tx.Rollback(ctx)
|
||||
|
||||
row := tx.QueryRow(ctx, `
|
||||
UPDATE firewall_service_groups SET name = $1, description = $2, updated_at = NOW()
|
||||
WHERE id = $3
|
||||
RETURNING id, name, description, created_at, updated_at`,
|
||||
g.Name, g.Description, id)
|
||||
out, err := scanSvcGrp(row)
|
||||
if err != nil {
|
||||
if errors.Is(err, pgx.ErrNoRows) {
|
||||
return nil, ErrServiceGroupNotFound
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
if _, err := tx.Exec(ctx, `DELETE FROM firewall_service_group_members WHERE group_id = $1`, id); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if len(g.MemberIDs) > 0 {
|
||||
if err := insertSvcGrpMembers(ctx, tx, id, g.MemberIDs); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
out.MemberIDs = append([]int64{}, g.MemberIDs...)
|
||||
}
|
||||
return out, tx.Commit(ctx)
|
||||
}
|
||||
|
||||
func (r *ServiceGroupsRepo) Delete(ctx context.Context, id int64) error {
|
||||
tag, err := r.Pool.Exec(ctx, `DELETE FROM firewall_service_groups WHERE id = $1`, id)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if tag.RowsAffected() == 0 {
|
||||
return ErrServiceGroupNotFound
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func insertSvcGrpMembers(ctx context.Context, tx pgx.Tx, gid int64, members []int64) error {
|
||||
for _, sid := range members {
|
||||
if _, err := tx.Exec(ctx,
|
||||
`INSERT INTO firewall_service_group_members (group_id, service_id) VALUES ($1, $2) ON CONFLICT DO NOTHING`,
|
||||
gid, sid); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func scanSvcGrp(row interface{ Scan(...any) error }) (*models.FirewallServiceGroup, error) {
|
||||
var g models.FirewallServiceGroup
|
||||
if err := row.Scan(
|
||||
&g.ID, &g.Name, &g.Description,
|
||||
&g.CreatedAt, &g.UpdatedAt,
|
||||
); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &g, nil
|
||||
}
|
||||
113
internal/services/firewall/services.go
Normal file
113
internal/services/firewall/services.go
Normal file
@@ -0,0 +1,113 @@
|
||||
package firewall
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
|
||||
"github.com/jackc/pgx/v5"
|
||||
"github.com/jackc/pgx/v5/pgxpool"
|
||||
|
||||
"git.netcell-it.de/projekte/edgeguard-native/internal/models"
|
||||
)
|
||||
|
||||
var ErrServiceNotFound = errors.New("service not found")
|
||||
|
||||
type ServicesRepo struct {
|
||||
Pool *pgxpool.Pool
|
||||
}
|
||||
|
||||
func NewServicesRepo(pool *pgxpool.Pool) *ServicesRepo { return &ServicesRepo{Pool: pool} }
|
||||
|
||||
const svcBaseSelect = `
|
||||
SELECT id, name, proto, port_start, port_end, builtin, description,
|
||||
created_at, updated_at
|
||||
FROM firewall_services
|
||||
`
|
||||
|
||||
func (r *ServicesRepo) List(ctx context.Context) ([]models.FirewallService, error) {
|
||||
rows, err := r.Pool.Query(ctx, svcBaseSelect+" ORDER BY name ASC")
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer rows.Close()
|
||||
out := make([]models.FirewallService, 0, 16)
|
||||
for rows.Next() {
|
||||
s, err := scanService(rows)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
out = append(out, *s)
|
||||
}
|
||||
return out, rows.Err()
|
||||
}
|
||||
|
||||
func (r *ServicesRepo) Get(ctx context.Context, id int64) (*models.FirewallService, error) {
|
||||
row := r.Pool.QueryRow(ctx, svcBaseSelect+" WHERE id = $1", id)
|
||||
s, err := scanService(row)
|
||||
if err != nil {
|
||||
if errors.Is(err, pgx.ErrNoRows) {
|
||||
return nil, ErrServiceNotFound
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
return s, nil
|
||||
}
|
||||
|
||||
func (r *ServicesRepo) Create(ctx context.Context, s models.FirewallService) (*models.FirewallService, error) {
|
||||
row := r.Pool.QueryRow(ctx, `
|
||||
INSERT INTO firewall_services (name, proto, port_start, port_end, builtin, description)
|
||||
VALUES ($1, $2, $3, $4, FALSE, $5)
|
||||
RETURNING id, name, proto, port_start, port_end, builtin, description, created_at, updated_at`,
|
||||
s.Name, s.Proto, s.PortStart, s.PortEnd, s.Description)
|
||||
return scanService(row)
|
||||
}
|
||||
|
||||
func (r *ServicesRepo) Update(ctx context.Context, id int64, s models.FirewallService) (*models.FirewallService, error) {
|
||||
// Forbid editing builtin services — they're guaranteed by the
|
||||
// migration set; users that want a tweak can clone with a new name.
|
||||
cur, err := r.Get(ctx, id)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if cur.Builtin {
|
||||
return nil, errors.New("builtin service cannot be edited — clone it under a new name")
|
||||
}
|
||||
row := r.Pool.QueryRow(ctx, `
|
||||
UPDATE firewall_services SET
|
||||
name = $1, proto = $2, port_start = $3, port_end = $4,
|
||||
description = $5, updated_at = NOW()
|
||||
WHERE id = $6
|
||||
RETURNING id, name, proto, port_start, port_end, builtin, description, created_at, updated_at`,
|
||||
s.Name, s.Proto, s.PortStart, s.PortEnd, s.Description, id)
|
||||
return scanService(row)
|
||||
}
|
||||
|
||||
func (r *ServicesRepo) Delete(ctx context.Context, id int64) error {
|
||||
cur, err := r.Get(ctx, id)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if cur.Builtin {
|
||||
return errors.New("builtin service cannot be deleted")
|
||||
}
|
||||
tag, err := r.Pool.Exec(ctx, `DELETE FROM firewall_services WHERE id = $1`, id)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if tag.RowsAffected() == 0 {
|
||||
return ErrServiceNotFound
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func scanService(row interface{ Scan(...any) error }) (*models.FirewallService, error) {
|
||||
var s models.FirewallService
|
||||
if err := row.Scan(
|
||||
&s.ID, &s.Name, &s.Proto, &s.PortStart, &s.PortEnd,
|
||||
&s.Builtin, &s.Description,
|
||||
&s.CreatedAt, &s.UpdatedAt,
|
||||
); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &s, nil
|
||||
}
|
||||
Reference in New Issue
Block a user