1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
package main
import (
"bufio"
"fmt"
"iptcp/pkg/ipstack"
"net/netip"
"os"
"strings"
)
func main() {
if len(os.Args) == 1 {
fmt.Printf("Usage: %s <configFile>\n", os.Args[0])
os.Exit(1)
}
fileName := os.Args[2]
err := ipstack.Initialize(fileName)
if err != nil {
return
}
ipstack.RegisterProtocolHandler(ipstack.TEST_PROTOCOL)
scanner := bufio.NewScanner(os.Stdin)
for scanner.Scan() {
line := scanner.Text()
switch line {
case "li":
fmt.Println("Name\tAddr/Prefix\tState")
fmt.Println(ipstack.SprintInterfaces())
case "ln":
fmt.Println("Iface\tVIP\tUDPAddr")
fmt.Println(ipstack.SprintNeighbors())
case "lr":
fmt.Println("T\tPrefix\tNext Hop\tCost")
fmt.Println(ipstack.SprintRoutingTable())
case "q":
ipstack.CleanUp()
os.Exit(0)
case "exit":
ipstack.CleanUp()
os.Exit(0)
default:
if len(line) > 4 {
if line[:4] == "down" {
ifaceName := line[5:]
ipstack.InterfaceDownREPL(ifaceName)
}
if line[:4] == "send" {
// get IP address and message that follows it
IPAndMessage := strings.Split(line, " ")
ipAddr := IPAndMessage[1]
message := IPAndMessage[2:]
// combine message into one string
messageToSend := strings.Join(message, " ")
messageToSendBytes := []byte(messageToSend)
address, _ := netip.ParseAddr(ipAddr)
hop, err := ipstack.LongestPrefix(address)
if err != nil {
fmt.Println(err)
continue
}
myAddr := hop.Interface.IpPrefix.Addr()
for _, neighbor := range ipstack.GetNeighbors()[hop.Interface.Name] {
// TODO: fix multiple send bug here on static route
if neighbor.VipAddr == address ||
neighbor.VipAddr == hop.VIP && hop.Type == "S" {
bytesWritten, err := ipstack.SendIP(&myAddr, neighbor, ipstack.TEST_PROTOCOL, messageToSendBytes, ipAddr, nil)
if err != nil {
fmt.Println(err)
} else {
fmt.Printf("Sent %d bytes to %s\n", bytesWritten, neighbor.VipAddr.String())
}
}
}
}
}
if len(line) > 2 {
if line[:2] == "up" {
// get interface name
ifaceName := line[3:]
ipstack.InterfaceUpREPL(ifaceName)
}
} else {
fmt.Println("Invalid command: ", line)
fmt.Println("Commands: ")
fmt.Println(" exit Terminate this program")
fmt.Println(" li List interfaces")
fmt.Println(" lr List routes")
fmt.Println(" ln List available neighbors")
fmt.Println(" up Enable an interface")
fmt.Println(" down Disable an interface")
fmt.Println(" send Send test packet")
}
continue
}
}
}
|
