From dfc5d861c727ba63adabf3a5edc7fe76ba60b0b7 Mon Sep 17 00:00:00 2001
From: Juan Font <juanfontalonso@gmail.com>
Date: Tue, 4 Apr 2023 22:41:00 +0000
Subject: [PATCH] Fix CIDR calculation in expandACLPeerAddr

---
 acls.go      |  4 +--
 acls_test.go | 71 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 73 insertions(+), 2 deletions(-)

diff --git a/acls.go b/acls.go
index 6e23e0a..64f18fe 100644
--- a/acls.go
+++ b/acls.go
@@ -213,11 +213,11 @@ func expandACLPeerAddr(srcIP string) []string {
 			return []string{from.String()}
 		}
 
-		for from != too {
+		for from != too && from.Less(too) {
 			addrs = append(addrs, from.String())
-
 			from = from.Next()
 		}
+		addrs = append(addrs, too.String()) // Add the last IP address in the range
 
 		return addrs
 	}
diff --git a/acls_test.go b/acls_test.go
index c4d619d..7d464c7 100644
--- a/acls_test.go
+++ b/acls_test.go
@@ -1596,6 +1596,7 @@ func Test_expandACLPeerAddr(t *testing.T) {
 				"10.0.0.0",
 				"10.0.0.1",
 				"10.0.0.2",
+				"10.0.0.3",
 			},
 		},
 		{
@@ -1609,6 +1610,76 @@ func Test_expandACLPeerAddr(t *testing.T) {
 				"192.168.0.134", "192.168.0.135", "192.168.0.136",
 				"192.168.0.137", "192.168.0.138", "192.168.0.139",
 				"192.168.0.140", "192.168.0.141", "192.168.0.142",
+				"192.168.0.143",
+			},
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := expandACLPeerAddr(tt.args.srcIP); !reflect.DeepEqual(got, tt.want) {
+				t.Errorf("expandACLPeerAddr() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}
+
+func Test_expandACLPeerAddrV6(t *testing.T) {
+	type args struct {
+		srcIP string
+	}
+	tests := []struct {
+		name string
+		args args
+		want []string
+	}{
+		{
+			name: "asterix",
+			args: args{
+				srcIP: "*",
+			},
+			want: []string{"*"},
+		},
+		{
+			name: "ipfull",
+			args: args{
+				srcIP: "fd7a:115c:a1e0:ab12:4943:cd96:624c:3166",
+			},
+			want: []string{"fd7a:115c:a1e0:ab12:4943:cd96:624c:3166"},
+		},
+		{
+			name: "ipzerocompression",
+			args: args{
+				srcIP: "fd7a:115c:a1e0:ab12:4943:cd96:624c::",
+			},
+			want: []string{"fd7a:115c:a1e0:ab12:4943:cd96:624c:0"},
+		},
+		{
+			name: "ip/128",
+			args: args{
+				srcIP: "fd7a:115c:a1e0:ab12:4943:cd96:624c:3166/128",
+			},
+			want: []string{"fd7a:115c:a1e0:ab12:4943:cd96:624c:3166"},
+		},
+		{
+			name: "ip/127",
+			args: args{
+				srcIP: "fd7a:115c:a1e0:ab12:4943:cd96:624c:0000/127",
+			},
+			want: []string{
+				"fd7a:115c:a1e0:ab12:4943:cd96:624c:0",
+				"fd7a:115c:a1e0:ab12:4943:cd96:624c:1",
+			},
+		},
+		{
+			name: "ip/126",
+			args: args{
+				srcIP: "fd7a:115c:a1e0:ab12:4943:cd96:624c:0000/126",
+			},
+			want: []string{
+				"fd7a:115c:a1e0:ab12:4943:cd96:624c:0",
+				"fd7a:115c:a1e0:ab12:4943:cd96:624c:1",
+				"fd7a:115c:a1e0:ab12:4943:cd96:624c:2",
+				"fd7a:115c:a1e0:ab12:4943:cd96:624c:3",
 			},
 		},
 	}