v0.1.8: IPTABLES_SAVE=iptables-nft-save support

andreas-p · Jun 7, 2021 · 6c90ef3 · 6c90ef3
1 parent ba8876e
commit 6c90ef3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/Makefile b/Makefile
@@ -1,7 +1,7 @@
 all: build
 
 ENVVAR = GOOS=linux GOARCH=amd64
-TAG = v0.1.7
+TAG = v0.1.8
 APP_NAME = calico-accountant
 REPO = andreas-p
 IMAGE_TAG = $(REPO)/$(APP_NAME):$(TAG)

diff --git a/iptables/iptables.go b/iptables/iptables.go
@@ -5,6 +5,7 @@ import (
 	"bytes"
 	"errors"
 	"io"
+	"os"
 	"os/exec"
 	"regexp"
 	"sort"
@@ -88,7 +89,12 @@ var (
 )
 
 func iptablesSave(interfaceToWorkload map[string]*apiv3.WorkloadEndpoint) ([]*Result, error) {
-	cmd := exec.Command("iptables-save", "-t", "filter", "-c")
+	saveCommand, ok := os.LookupEnv("IPTABLES_SAVE")
+	if !ok {
+		saveCommand = "iptables-save"
+	}
+
+	cmd := exec.Command(saveCommand, "-t", "filter", "-c")
 
 	stdout, err := cmd.StdoutPipe()
 	if err != nil {