In case this script runs from a CI pipeline its possible the
executing netns has a conntrack_max setting that isn't enough
for the default 128k size that is autogenerated in the script.

We can't change conntrack_max either, as we might be
running via unshare(1).  Fetch the max allowed setting
and then cap the generated output once that is reached.

Furthermore add a assert() helper to emit PASS/FAIL
messages so CI logs are more grep friedly.

Signed-off-by: Florian Westphal <fw@strlen.de>
---
 tests/conntrack/bulk-load-stress.sh | 47 +++++++++++++++++++++--------
 1 file changed, 35 insertions(+), 12 deletions(-)

diff --git a/tests/conntrack/bulk-load-stress.sh b/tests/conntrack/bulk-load-stress.sh
index 64f51b9c75e8..31ad1086887f 100755
--- a/tests/conntrack/bulk-load-stress.sh
+++ b/tests/conntrack/bulk-load-stress.sh
@@ -14,12 +14,27 @@ TMP_FILE="$(mktemp)"
 CT_ZONE=$DEFAULT_CT_ZONE
 GEN_ONLY=$DEFAULT_GEN_ONLY
 CLEANUP_INDIVIDUAL=$DEFAULT_CLEANUP_INDIVIDUAL
+ret=0
+
+ct_max=0
+read ct_max < /proc/sys/net/netfilter/nf_conntrack_max
 
 cleanup() {
 	rm -f "$TMP_FILE"
 }
 trap cleanup EXIT
 
+assert() {
+	local r="$1"
+
+	if [ $r -ne 0 ]; then
+		[ "$ret" -eq 0 ] && ret="$r"
+		echo "FAIL: bulk-load-stress.sh: $@"
+	else
+		echo "PASS: bulk-load-stress.sh: $@"
+	fi
+}
+
 print_help()
 {
 	me=$(basename "$0")
@@ -52,9 +67,18 @@ print_help()
 
 function ct_data_gen()
 {
+	local cnt=1
+
 	for (( d = 1; d <= $DPORT_COUNT; d++ )) do
 		for (( s = 1; s <= $SPORT_COUNT; s++ )) do
 			echo "-I -w $CT_ZONE -s 1.1.1.1 -d 2.2.2.2 -p tcp --sport ${s} --dport ${d} --state LISTEN -u SEEN_REPLY -t 50"
+
+			cnt=$((cnt+1))
+			[ $ct_max -eq 0 ] && continue
+			if [ $cnt -ge $ct_max ]; then
+				echo "WARN: Generated only $cnt entries ct_max ($ct_max) reached" 1>&2
+				return
+			fi
 		done
 	done
 }
@@ -135,7 +159,7 @@ ct_data_gen > $TMP_FILE
 
 NUM_ENTRIES=$(cat ${TMP_FILE} | wc -l)
 
-echo "File ${TMP_FILE} is generated, number of entries: ${NUM_ENTRIES}."
+echo "File ${TMP_FILE} is generated, number of entries: ${NUM_ENTRIES}. ct_max is $ct_max."
 
 if [ "$GEN_ONLY" -eq "1" ]; then
 	# Retain tmpfile in this mode
@@ -151,8 +175,9 @@ fi
 
 echo "Loading ${NUM_ENTRIES} entries from ${TMP_FILE} .."
 time -p ${CT} -R $TMP_FILE
-lret=$?
-ret=$lret
+assert $? "$CT -R $TMP_FILE"
+$CT -C
+assert $? "$CT -C"
 
 if [ "$CLEANUP_INDIVIDUAL" -eq "1" ]; then
 	sed -i -e "s/-I/-D/g" -e "s/-t 50//g" $TMP_FILE
@@ -163,15 +188,13 @@ if [ "$CLEANUP_INDIVIDUAL" -eq "1" ]; then
 
 	echo "Cleaning ${NUM_ENTRIES} entries from ${TMP_FILE} .."
 	time -p ${CT} -R $TMP_FILE
-	lret=$?
-	[ $ret -eq 0 ] && ret=$lret
+	assert $? "$CT -R $TMP_FILE - cleaning"
+else
+	echo "Cleaning up zone ${CT_ZONE}.."
+	time -p ${CT} -D -w $CT_ZONE > /dev/null
+	assert $? "$CT -D -w $CT_ZONE"
+	conntrack -C
+	assert $? "$CT -C"
 fi
 
-
-echo "Cleaning up zone ${CT_ZONE}.."
-time -p ${CT} -D -w $CT_ZONE > /dev/null
-lret=$?
-[ $ret -eq 0 ] && ret=$lret
-rm $TMP_FILE
-
 exit $ret
-- 
2.54.0