diff options
Diffstat (limited to 'cmd')
| -rw-r--r-- | cmd/vhost/main.go | 71 | ||||
| -rw-r--r-- | cmd/vrouter/main.go | 75 |
2 files changed, 139 insertions, 7 deletions
diff --git a/cmd/vhost/main.go b/cmd/vhost/main.go index 82b8195..27e073f 100644 --- a/cmd/vhost/main.go +++ b/cmd/vhost/main.go @@ -4,14 +4,83 @@ import ( "bufio" "fmt" "os" + "iptcp/pkg/ipstack" + "strings" ) func main() { + if len(os.Args) != 2 { + fmt.Printf("Usage: %s <configFile>\n", os.Args[0]) + os.Exit(1) + } + + fileName := os.Args[1] + + go ipstack.Initialize(fileName) + + scanner := bufio.NewScanner(os.Stdin) for scanner.Scan() { line := scanner.Text() - fmt.Println(line) + + if line == "li" { + fmt.Println("Name\tAddr/Prefix\tState") + ipstack.SprintInterfaces() + } + + if line == "ln" { + fmt.Println("Iface\tVIP\tUDPAddr") + ipstack.SprintNeighbors() + } + + if line == "lr" { + fmt.Println("T\tPrefix\tNext Hop\tCost") + ipstack.SprintRoutingTable() + } + + if len(line) > 4 { + if line[:4] == "down" { + // get interface name + ifaceName := line[5:] + ipstack.InterfaceDownREPL(ifaceName) + } + + if line[:4] == "send" { + // get IP address and message that follows it + listOfWords := strings.Split(line, " ") + ipAddr := listOfWords[1] + message := listOfWords[2:] + // combine message into one string + messageToSend := strings.Join(message, " ") + // convert message to byte array + messageToSendBytes := []byte(messageToSend) + // get interface by ipAddr + iface, err := ipstack.GetNeighborByIP(ipAddr) + if err != nil { + fmt.Println(err) + iface, err = ipstack.GetRouteByIP(ipAddr) + if err != nil { + fmt.Println(err) + continue + } + ipstack.SendIP(ipstack.GetMyVIP(), iface, 0, messageToSendBytes, ipAddr) + } + ipstack.SendIP(ipstack.GetMyVIP(), iface, 0, messageToSendBytes, ipAddr) + } + } + if len(line) > 2 { + if line[:2] == "up" { + // get interface name + ifaceName := line[3:] + ipstack.InterfaceUpREPL(ifaceName) + } + } + + if line == "q" { + ipstack.CleanUp() + os.Exit(0) + } } }
\ No newline at end of file diff --git a/cmd/vrouter/main.go b/cmd/vrouter/main.go index 158016a..abd0fc0 100644 --- a/cmd/vrouter/main.go +++ b/cmd/vrouter/main.go @@ -5,19 +5,25 @@ import ( "fmt" "os" "time" + "iptcp/pkg/ipstack" + "strings" + ) func SendUpdates() { + myInterfaces := ipstack.GetInterfaces() + myNeighbors := ipstack.GetNeighbors() for _, iface := range myInterfaces { // send RIP updates to all neighbors - for _, neighbor := range myNeighbors { - iface.udp.Write(neighbor, data) + for _, _ = range myNeighbors { + // iface.udp.Write(neighbor, data) + // iface.RecvSocket.Write(neighbor, data) // wait for response for 12 seconds response := make([]byte, 512) - iface.udp.Read(response) + iface.RecvSocket.Read(response) time.Sleep(12 * time.Second) if len(response) == 0 { - RemoveNeighbor(neighbor) + // ipstack.RemoveNeighbor(neighbor) } } } @@ -32,7 +38,12 @@ func main() { fileName := os.Args[1] - initialize(fileName) + // initialize(fileName) + go ipstack.Initialize(fileName) + // myInterfaces := ipstack.GetInterfaces() + // myNeighbors := ipstack.GetNeighbors() + // myRoutes := ipstack.GetRoutes() + go SendUpdates() @@ -40,6 +51,58 @@ func main() { for scanner.Scan() { line := scanner.Text() - fmt.Println(line) + switch line { + case "li": + fmt.Println("Name\tAddr/Prefix\tState") + ipstack.SprintInterfaces() + case "ln": + fmt.Println("Iface\tVIP\tUDPAddr") + ipstack.SprintNeighbors() + case "lr": + fmt.Println("T\tPrefix\tNext Hop\tCost") + ipstack.SprintRoutingTable() + case "q": + ipstack.CleanUp() + os.Exit(0) + default: + if len(line) > 4 { + if line[:4] == "down" { + // get interface name + ifaceName := line[5:] + ipstack.InterfaceDownREPL(ifaceName) + } + + if line[:4] == "send" { + // get IP address and message that follows it + listOfWords := strings.Split(line, " ") + ipAddr := listOfWords[1] + message := listOfWords[2:] + // combine message into one string + messageToSend := strings.Join(message, " ") + // convert message to byte array + messageToSendBytes := []byte(messageToSend) + // get interface by ipAddr + iface, err := ipstack.GetNeighborByIP(ipAddr) + if err != nil { + fmt.Println(err) + iface, err = ipstack.GetRouteByIP(ipAddr) + if err != nil { + fmt.Println(err) + continue + } + ipstack.SendIP(ipstack.GetMyVIP(), iface, 0, messageToSendBytes, ipAddr) + } + ipstack.SendIP(ipstack.GetMyVIP(), iface, 0, messageToSendBytes, ipAddr) + } + } + if len(line) > 2 { + if line[:2] == "up" { + // get interface name + ifaceName := line[3:] + ipstack.InterfaceUpREPL(ifaceName) + } + } + continue + } } }
\ No newline at end of file |