Skip to content

Commit

Permalink
Fix IP Address Order Bug
Browse files Browse the repository at this point in the history
  • Loading branch information
qzydustin authored and kradalby committed Jul 23, 2023
1 parent 23a3adf commit 6567af7
Show file tree
Hide file tree
Showing 2 changed files with 43 additions and 0 deletions.
17 changes: 17 additions & 0 deletions hscontrol/types/machine.go
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@ import (
"errors"
"fmt"
"net/netip"
"sort"
"strings"
"time"

Expand Down Expand Up @@ -73,7 +74,23 @@ type (

type MachineAddresses []netip.Addr

func (ma MachineAddresses) Sort() {
sort.Slice(ma, func(index1, index2 int) bool {
if ma[index1].Is4() && ma[index2].Is6() {

return true
}
if ma[index1].Is6() && ma[index2].Is4() {

return false
}

return ma[index1].Compare(ma[index2]) < 0
})
}

func (ma MachineAddresses) StringSlice() []string {
ma.Sort()
strSlice := make([]string, 0, len(ma))
for _, addr := range ma {
strSlice = append(strSlice, addr.String())
Expand Down
26 changes: 26 additions & 0 deletions hscontrol/types/machine_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -113,3 +113,29 @@ func Test_MachineCanAccess(t *testing.T) {
})
}
}

func TestMachineAddressesOrder(t *testing.T) {
machineAddresses := MachineAddresses{
netip.MustParseAddr("2001:db8::2"),
netip.MustParseAddr("100.64.0.2"),
netip.MustParseAddr("2001:db8::1"),
netip.MustParseAddr("100.64.0.1"),
}

strSlice := machineAddresses.StringSlice()
expected := []string{
"100.64.0.1",
"100.64.0.2",
"2001:db8::1",
"2001:db8::2",
}

if len(strSlice) != len(expected) {
t.Fatalf("unexpected slice length: got %v, want %v", len(strSlice), len(expected))
}
for i, addr := range strSlice {
if addr != expected[i] {
t.Errorf("unexpected address at index %v: got %v, want %v", i, addr, expected[i])
}
}
}

0 comments on commit 6567af7

Please sign in to comment.