diff options
Diffstat (limited to 'pkg/ipstack/ipstack_test.go')
| -rw-r--r-- | pkg/ipstack/ipstack_test.go | 517 |
1 files changed, 286 insertions, 231 deletions
diff --git a/pkg/ipstack/ipstack_test.go b/pkg/ipstack/ipstack_test.go index 941c4e9..5cef903 100644 --- a/pkg/ipstack/ipstack_test.go +++ b/pkg/ipstack/ipstack_test.go @@ -2,252 +2,307 @@ package ipstack import ( "fmt" - ipv4header "github.com/brown-csci1680/iptcp-headers" - "net" "net/netip" "testing" - "time" ) -func TestInitialize(t *testing.T) { - lnxFilePath := "../../doc-example/r2.lnx" - err := Initialize(lnxFilePath) - if err != nil { - t.Error(err) - } - fmt.Printf("Interfaces:\n%s\n\n", SprintInterfaces()) - fmt.Printf("Neighbors:\n%s\n", SprintNeighbors()) - fmt.Printf("RoutingTable:\n%s\n", SprintRoutingTable()) - - fmt.Println("TestInitialize successful") - t.Cleanup(func() { CleanUp() }) -} - -func TestInterfaceUpThenDown(t *testing.T) { - lnxFilePath := "../../doc-example/r2.lnx" - err := Initialize(lnxFilePath) - if err != nil { - t.Error(err) - } - - iface, err := GetInterfaceByName("if0") - if err != nil { - t.Error(err) - } - - InterfaceUp(iface) - if iface.State == false { - t.Error("iface state should be true") - } - - fmt.Printf("Interfaces:\n%s\n", SprintInterfaces()) - - time.Sleep(5 * time.Millisecond) // allow time to print - - InterfaceDown(iface) - if iface.State == true { - t.Error("iface state should be false") - } - - time.Sleep(5 * time.Millisecond) // allow time to print - - fmt.Printf("Interfaces:\n%s\n", SprintInterfaces()) - - fmt.Println("TestInterfaceUpThenDown successful") - t.Cleanup(func() { CleanUp() }) -} - -func TestInterfaceUpThenDownTwice(t *testing.T) { - lnxFilePath := "../../doc-example/r2.lnx" - err := Initialize(lnxFilePath) - if err != nil { - t.Error(err) - } - - iface, err := GetInterfaceByName("if0") - if err != nil { - t.Error(err) - } - - InterfaceUp(iface) - if iface.State == false { - t.Error("iface state should be true") - } - - fmt.Printf("Interfaces:\n%s\n", SprintInterfaces()) - - time.Sleep(5 * time.Millisecond) // allow time to print - - fmt.Println("putting interface down") - InterfaceDown(iface) - if iface.State == true { - t.Error("iface state should be false") - } - - time.Sleep(3 * time.Millisecond) - - fmt.Println("putting interface back up for 3 iterations") - InterfaceUp(iface) - if iface.State == false { - t.Error("iface state should be true") - } - time.Sleep(3 * time.Millisecond) // allow time to print - - fmt.Println("putting interface down") - InterfaceDown(iface) - if iface.State == true { - t.Error("iface state should be false") - } - - time.Sleep(5 * time.Millisecond) // allow time to print - - fmt.Printf("Interfaces:\n%s\n", SprintInterfaces()) - - fmt.Println("TestInterfaceUpThenDownTwice successful") - t.Cleanup(func() { CleanUp() }) -} - -func TestSendIPToNeighbor(t *testing.T) { - lnxFilePath := "../../doc-example/r2.lnx" - err := Initialize(lnxFilePath) - if err != nil { - t.Error(err) - } - - // get the first neighbor of this interface - iface, err := GetInterfaceByName("if0") - if err != nil { - t.Error(err) - } - neighbors, err := GetNeighborsToInterface("if0") - if err != nil { - t.Error(err) - } - - // setup a neighbor listener socket - testNeighbor := neighbors[0] - // close the socket so we can listen on it - err = testNeighbor.SendSocket.Close() - if err != nil { - t.Error(err) - } - - fmt.Printf("Interfaces:\n%s\n", SprintInterfaces()) - fmt.Printf("Neighbors:\n%s\n", SprintNeighbors()) - - listenString := testNeighbor.UdpAddr.String() - fmt.Println("listening on " + listenString) - listenAddr, err := net.ResolveUDPAddr("udp4", listenString) - if err != nil { - t.Error(err) +//func TestInitialize(t *testing.T) { +// lnxFilePath := "../../doc-example/r2.lnx" +// err := Initialize(lnxFilePath) +// if err != nil { +// t.Error(err) +// } +// fmt.Printf("Interfaces:\n%s\n\n", SprintInterfaces()) +// fmt.Printf("Neighbors:\n%s\n", SprintNeighbors()) +// fmt.Printf("RoutingTable:\n%s\n", SprintRoutingTable()) +// +// fmt.Println("TestInitialize successful") +// t.Cleanup(func() { CleanUp() }) +//} +// +//func TestInterfaceUpThenDown(t *testing.T) { +// lnxFilePath := "../../doc-example/r2.lnx" +// err := Initialize(lnxFilePath) +// if err != nil { +// t.Error(err) +// } +// +// iface, err := GetInterfaceByName("if0") +// if err != nil { +// t.Error(err) +// } +// +// InterfaceUp(iface) +// if iface.State == false { +// t.Error("iface state should be true") +// } +// +// fmt.Printf("Interfaces:\n%s\n", SprintInterfaces()) +// +// time.Sleep(5 * time.Millisecond) // allow time to print +// +// InterfaceDown(iface) +// if iface.State == true { +// t.Error("iface state should be false") +// } +// +// time.Sleep(5 * time.Millisecond) // allow time to print +// +// fmt.Printf("Interfaces:\n%s\n", SprintInterfaces()) +// +// fmt.Println("TestInterfaceUpThenDown successful") +// t.Cleanup(func() { CleanUp() }) +//} +// +//func TestInterfaceUpThenDownTwice(t *testing.T) { +// lnxFilePath := "../../doc-example/r2.lnx" +// err := Initialize(lnxFilePath) +// if err != nil { +// t.Error(err) +// } +// +// iface, err := GetInterfaceByName("if0") +// if err != nil { +// t.Error(err) +// } +// +// InterfaceUp(iface) +// if iface.State == false { +// t.Error("iface state should be true") +// } +// +// fmt.Printf("Interfaces:\n%s\n", SprintInterfaces()) +// +// time.Sleep(5 * time.Millisecond) // allow time to print +// +// fmt.Println("putting interface down") +// InterfaceDown(iface) +// if iface.State == true { +// t.Error("iface state should be false") +// } +// +// time.Sleep(3 * time.Millisecond) +// +// fmt.Println("putting interface back up for 3 iterations") +// InterfaceUp(iface) +// if iface.State == false { +// t.Error("iface state should be true") +// } +// time.Sleep(3 * time.Millisecond) // allow time to print +// +// fmt.Println("putting interface down") +// InterfaceDown(iface) +// if iface.State == true { +// t.Error("iface state should be false") +// } +// +// time.Sleep(5 * time.Millisecond) // allow time to print +// +// fmt.Printf("Interfaces:\n%s\n", SprintInterfaces()) +// +// fmt.Println("TestInterfaceUpThenDownTwice successful") +// t.Cleanup(func() { CleanUp() }) +//} +// +//func TestSendIPToNeighbor(t *testing.T) { +// lnxFilePath := "../../doc-example/r2.lnx" +// err := Initialize(lnxFilePath) +// if err != nil { +// t.Error(err) +// } +// +// // get the first neighbor of this interface +// iface, err := GetInterfaceByName("if0") +// if err != nil { +// t.Error(err) +// } +// neighbors, err := GetNeighborsToInterface("if0") +// if err != nil { +// t.Error(err) +// } +// +// // setup a neighbor listener socket +// testNeighbor := neighbors[0] +// // close the socket so we can listen on it +// err = testNeighbor.SendSocket.Close() +// if err != nil { +// t.Error(err) +// } +// +// fmt.Printf("Interfaces:\n%s\n", SprintInterfaces()) +// fmt.Printf("Neighbors:\n%s\n", SprintNeighbors()) +// +// listenString := testNeighbor.UdpAddr.String() +// fmt.Println("listening on " + listenString) +// listenAddr, err := net.ResolveUDPAddr("udp4", listenString) +// if err != nil { +// t.Error(err) +// } +// recvSocket, err := net.ListenUDP("udp4", listenAddr) +// if err != nil { +// t.Error(err) +// } +// testNeighbor.SendSocket = *recvSocket +// +// sent := false +// go func() { +// buffer := make([]byte, MAX_IP_PACKET_SIZE) +// fmt.Println("wating to read from UDP socket") +// _, sourceAddr, err := recvSocket.ReadFromUDP(buffer) +// if err != nil { +// t.Error(err) +// } +// fmt.Println("read from UDP socket") +// hdr, err := ipv4header.ParseHeader(buffer) +// if err != nil { +// t.Error(err) +// } +// headerSize := hdr.Len +// headerBytes := buffer[:headerSize] +// checksumFromHeader := uint16(hdr.Checksum) +// computedChecksum := ValidateChecksum(headerBytes, checksumFromHeader) +// +// var checksumState string +// if computedChecksum == checksumFromHeader { +// checksumState = "OK" +// } else { +// checksumState = "FAIL" +// } +// message := buffer[headerSize:] +// fmt.Printf("Received IP packet from %s\nHeader: %v\nChecksum: %s\nMessage: %s\n", +// sourceAddr.String(), hdr, checksumState, string(message)) +// if err != nil { +// t.Error(err) +// } +// +// sent = true +// }() +// +// time.Sleep(10 * time.Millisecond) +// +// // send a message to the neighbor +// fmt.Printf("sending message to neighbor\t%t\n", sent) +// err = SendIP(*iface, *testNeighbor, 0, []byte("You are my firest neighbor!")) +// if err != nil { +// t.Error(err) +// } +// +// fmt.Printf("SENT message to neighbor\t%t\n", sent) +// // give a little time for the message to be sent +// time.Sleep(1000 * time.Millisecond) +// if !sent { +// t.Error("Message not sent") +// t.Fail() +// } +// +// fmt.Println("TestSendIPToNeighbor successful") +// t.Cleanup(func() { CleanUp() }) +//} +// +//func TestRecvIP(t *testing.T) { +// lnxFilePath := "../../doc-example/r2.lnx" +// err := Initialize(lnxFilePath) +// if err != nil { +// t.Error(err) +// } +// +// // get the first neighbor of this interface to RecvIP from +// iface, err := GetInterfaceByName("if0") +// if err != nil { +// t.Error(err) +// } +// InterfaceUp(iface) +// +// // setup a random socket to send an ip packet from +// listenAddr, err := net.ResolveUDPAddr("udp4", "127.0.0.1:6969") +// sendSocket, err := net.ListenUDP("udp4", listenAddr) +// +// // send a message to the neighbor +// ifaceAsNeighbor := Neighbor{ +// VipAddr: iface.IpPrefix.Addr(), +// UdpAddr: iface.UdpAddr, +// SendSocket: iface.RecvSocket, +// SocketChannel: iface.SocketChannel, +// } +// fakeIface := Interface{ +// Name: "if69", +// IpPrefix: netip.MustParsePrefix("10.69.0.1/24"), +// UdpAddr: netip.MustParseAddrPort("127.0.0.1:6969"), +// RecvSocket: net.UDPConn{}, +// SocketChannel: nil, +// State: true, +// } +// err = SendIP(fakeIface, ifaceAsNeighbor, 0, []byte("hello")) +// if err != nil { +// return +// } +// +// time.Sleep(10 * time.Millisecond) +// +// // TODO: potenially make this a channel, so it actually checks values. +// // For now, you must read the message from the console. +// +// err = sendSocket.Close() +// if err != nil { +// t.Error(err) +// } +// t.Cleanup(func() { CleanUp() }) +//} + +func TestIntersect(t *testing.T) { + net1 := netip.MustParsePrefix("1.1.1.1/24") + net2 := netip.MustParsePrefix("1.1.1.2/24") + net3 := netip.MustParsePrefix("1.0.0.1/24") + net4 := netip.MustParsePrefix("0.0.0.0/0") // default route + net5 := netip.MustParsePrefix("1.1.1.1/32") + + res00 := intersect(net1, net5) + if !res00 { + t.Error("net1 and net2 should intersect") + t.Fail() } - recvSocket, err := net.ListenUDP("udp4", listenAddr) - if err != nil { - t.Error(err) + res01 := intersect(net5, net1) + if res01 { + t.Error("net1 and net2 should not intersect") + t.Fail() } - testNeighbor.SendSocket = *recvSocket - - sent := false - go func() { - buffer := make([]byte, MAX_IP_PACKET_SIZE) - fmt.Println("wating to read from UDP socket") - _, sourceAddr, err := recvSocket.ReadFromUDP(buffer) - if err != nil { - t.Error(err) - } - fmt.Println("read from UDP socket") - hdr, err := ipv4header.ParseHeader(buffer) - if err != nil { - t.Error(err) - } - headerSize := hdr.Len - headerBytes := buffer[:headerSize] - checksumFromHeader := uint16(hdr.Checksum) - computedChecksum := ValidateChecksum(headerBytes, checksumFromHeader) - - var checksumState string - if computedChecksum == checksumFromHeader { - checksumState = "OK" - } else { - checksumState = "FAIL" - } - message := buffer[headerSize:] - fmt.Printf("Received IP packet from %s\nHeader: %v\nChecksum: %s\nMessage: %s\n", - sourceAddr.String(), hdr, checksumState, string(message)) - if err != nil { - t.Error(err) - } - - sent = true - }() - time.Sleep(10 * time.Millisecond) - - // send a message to the neighbor - fmt.Printf("sending message to neighbor\t%t\n", sent) - err = SendIP(*iface, *testNeighbor, 0, []byte("You are my firest neighbor!")) - if err != nil { - t.Error(err) + res1 := intersect(net1, net2) + if !res1 { + t.Error("net1 and net2 should intersect") + t.Fail() } - - fmt.Printf("SENT message to neighbor\t%t\n", sent) - // give a little time for the message to be sent - time.Sleep(1000 * time.Millisecond) - if !sent { - t.Error("Message not sent") + res0 := intersect(net2, net1) + if !res0 { + t.Error("net1 and net2 should intersect") t.Fail() } - fmt.Println("TestSendIPToNeighbor successful") - t.Cleanup(func() { CleanUp() }) -} - -func TestRecvIP(t *testing.T) { - lnxFilePath := "../../doc-example/r2.lnx" - err := Initialize(lnxFilePath) - if err != nil { - t.Error(err) + res6 := intersect(net1, net3) + if res6 { + t.Error("net1 and net3 should not intersect") + t.Fail() } - - // get the first neighbor of this interface to RecvIP from - iface, err := GetInterfaceByName("if0") - if err != nil { - t.Error(err) + res2 := intersect(net2, net3) + if res2 { + t.Error("net2 and net3 should not intersect") + t.Fail() } - InterfaceUp(iface) - - // setup a random socket to send an ip packet from - listenAddr, err := net.ResolveUDPAddr("udp4", "127.0.0.1:6969") - sendSocket, err := net.ListenUDP("udp4", listenAddr) - - // send a message to the neighbor - ifaceAsNeighbor := Neighbor{ - VipAddr: iface.IpPrefix.Addr(), - UdpAddr: iface.UdpAddr, - SendSocket: iface.RecvSocket, - SocketChannel: iface.SocketChannel, + res3 := intersect(net1, net4) + if !res3 { + t.Error("net1 and net4 should intersect") + t.Fail() } - fakeIface := Interface{ - Name: "if69", - IpPrefix: netip.MustParsePrefix("10.69.0.1/24"), - UdpAddr: netip.MustParseAddrPort("127.0.0.1:6969"), - RecvSocket: net.UDPConn{}, - SocketChannel: nil, - State: true, + res4 := intersect(net2, net4) + if !res4 { + t.Error("net2 and net4 should intersect") + t.Fail() } - err = SendIP(fakeIface, ifaceAsNeighbor, 0, []byte("hello")) - if err != nil { - return + res5 := intersect(net3, net4) + if !res5 { + t.Error("net3 and net4 should intersect") + t.Fail() } - time.Sleep(10 * time.Millisecond) - - // TODO: potenially make this a channel, so it actually checks values. - // For now, you must read the message from the console. - - err = sendSocket.Close() - if err != nil { - t.Error(err) - } - t.Cleanup(func() { CleanUp() }) + fmt.Println("TestIntersect successful") } |