yggdrasil-go/src/yggdrasil/admin.go

88 lines
1.7 KiB
Go
Raw Normal View History

2018-01-21 03:17:15 +03:00
package yggdrasil
import "net"
import "os"
import "bytes"
import "fmt"
// TODO: Make all of this JSON
// TODO: Add authentication
// TODO: Is any of this thread safe?
2018-01-21 03:17:15 +03:00
type admin struct {
core *Core
listenaddr string
}
func (a *admin) init(c *Core, listenaddr string) {
a.core = c
a.listenaddr = listenaddr
go a.listen()
}
func (a *admin) listen() {
l, err := net.Listen("tcp", a.listenaddr)
if err != nil {
a.core.log.Printf("Admin socket failed to listen: %v", err)
os.Exit(1)
}
defer l.Close()
a.core.log.Printf("Admin socket listening on %s", l.Addr().String())
for {
conn, err := l.Accept()
if err == nil {
a.handleRequest(conn)
}
}
}
func (a *admin) handleRequest(conn net.Conn) {
buf := make([]byte, 1024)
_, err := conn.Read(buf)
if err != nil {
a.core.log.Printf("Admin socket failed to read: %v", err)
conn.Close()
return
}
buf = bytes.Trim(buf, "\x00\r\n\t")
switch string(buf) {
case "switch table":
table := a.core.switchTable.table.Load().(lookupTable)
conn.Write([]byte(fmt.Sprintf(
"port 0 -> %+v\n",
table.self.coords)))
for _, v := range table.elems {
conn.Write([]byte(fmt.Sprintf(
"port %d -> %+v\n",
v.port,
v.locator.coords)))
}
break
case "dht":
n := a.core.dht.nBuckets()
for i := 0; i < n; i++ {
b := a.core.dht.getBucket(i)
if len(b.infos) == 0 {
continue
}
for _, v := range b.infos {
addr := address_addrForNodeID(v.nodeID_hidden)
ip := net.IP(addr[:]).String()
conn.Write([]byte(fmt.Sprintf("%+v -> %+v\n",
ip,
v.coords)))
}
2018-01-21 03:17:15 +03:00
}
break
default:
conn.Write([]byte("I didn't understand that!\n"))
}
if err != nil {
a.core.log.Printf("Admin socket error: %v", err)
}
conn.Close()
}