From 373af959548091b1294bc385a457a3070bb1358b Mon Sep 17 00:00:00 2001
From: Stuart Golightly <stuart.golightly@comparethemarket.com>
Date: Wed, 7 Feb 2024 09:32:00 +0000
Subject: [PATCH] Add flag for csm port

Extending commit 05be059 by @clebio
---
 iamlivecore/csm.go     | 2 +-
 iamlivecore/service.go | 5 ++++-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/iamlivecore/csm.go b/iamlivecore/csm.go
index 009a8a2f..4c4016eb 100644
--- a/iamlivecore/csm.go
+++ b/iamlivecore/csm.go
@@ -154,7 +154,7 @@ func listenForEvents() {
 	var iamMap iamMapBase
 
 	addr := net.UDPAddr{
-		Port: 31000,
+		Port: *csmPortFlag,
 		IP:   net.ParseIP(*hostFlag),
 	}
 	conn, err := net.ListenUDP("udp", &addr)
diff --git a/iamlivecore/service.go b/iamlivecore/service.go
index f9dc7a5f..b1d23daa 100644
--- a/iamlivecore/service.go
+++ b/iamlivecore/service.go
@@ -30,6 +30,7 @@ var backgroundFlag *bool
 var debugFlag *bool
 var forceWildcardResourceFlag *bool
 var cpuProfileFlag = flag.String("cpu-profile", "", "write a CPU profile to this file (for performance testing purposes)")
+var csmPortFlag *int
 
 func parseConfig() {
 	provider := "aws"
@@ -48,6 +49,7 @@ func parseConfig() {
 	background := false
 	debug := false
 	forceWildcardResource := false
+	csmPort := 31000
 
 	cfgfile, err := homedir.Expand("~/.iamlive/config")
 	if err == nil {
@@ -120,6 +122,7 @@ func parseConfig() {
 	backgroundFlag = flag.Bool("background", background, "when set, the process will return the current PID and run in the background without output")
 	debugFlag = flag.Bool("debug", debug, "dumps associated HTTP requests when set in proxy mode")
 	forceWildcardResourceFlag = flag.Bool("force-wildcard-resource", forceWildcardResource, "when set, the Resource will always be a wildcard")
+	csmPortFlag = flag.Int("csm-port", csmPort, "port to listen on for CSM")
 }
 
 func Run() {
@@ -161,7 +164,7 @@ func Run() {
 	if *providerFlag == "aws" {
 		setINIConfigAndFileFlush()
 	}
-	
+
 	loadMaps()
 
 	if *modeFlag == "csm" && *providerFlag == "aws" {