Serge Bazanski | 8d7843c | 2018-10-04 10:37:36 +0100 | [diff] [blame] | 1 | package main |
| 2 | |
Serge Bazanski | 4676508 | 2018-10-06 12:32:01 +0100 | [diff] [blame] | 3 | import ( |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 4 | "context" |
| 5 | "fmt" |
Serge Bazanski | 4676508 | 2018-10-06 12:32:01 +0100 | [diff] [blame] | 6 | "net/http" |
Serge Bazanski | 69518ab | 2018-10-06 17:48:23 +0100 | [diff] [blame] | 7 | "sort" |
Serge Bazanski | de869df | 2018-10-06 13:55:49 +0100 | [diff] [blame] | 8 | "strings" |
Serge Bazanski | 8d7843c | 2018-10-04 10:37:36 +0100 | [diff] [blame] | 9 | |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 10 | "github.com/gobuffalo/packr" |
Serge Bazanski | 4676508 | 2018-10-06 12:32:01 +0100 | [diff] [blame] | 11 | "github.com/golang/glog" |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 12 | "github.com/q3k/statusz" |
Serge Bazanski | 69518ab | 2018-10-06 17:48:23 +0100 | [diff] [blame] | 13 | "vbom.ml/util/sortorder" |
Serge Bazanski | a758ef5 | 2018-10-06 17:54:25 +0100 | [diff] [blame] | 14 | |
| 15 | "code.hackerspace.pl/q3k/topo/graph" |
| 16 | "code.hackerspace.pl/q3k/topo/state" |
Serge Bazanski | 16e4ba2 | 2018-10-07 00:22:52 +0100 | [diff] [blame^] | 17 | |
| 18 | pb "code.hackerspace.pl/q3k/topo/proto/control" |
Serge Bazanski | 4676508 | 2018-10-06 12:32:01 +0100 | [diff] [blame] | 19 | ) |
| 20 | |
| 21 | type ServiceConfig struct { |
| 22 | DebugListen string |
| 23 | } |
| 24 | |
| 25 | type Service struct { |
| 26 | gr *graph.Graph |
Serge Bazanski | a758ef5 | 2018-10-06 17:54:25 +0100 | [diff] [blame] | 27 | stm *state.StateManager |
Serge Bazanski | 4676508 | 2018-10-06 12:32:01 +0100 | [diff] [blame] | 28 | config ServiceConfig |
| 29 | } |
| 30 | |
Serge Bazanski | a758ef5 | 2018-10-06 17:54:25 +0100 | [diff] [blame] | 31 | func NewService(gr *graph.Graph, stm *state.StateManager, c ServiceConfig) *Service { |
Serge Bazanski | 4676508 | 2018-10-06 12:32:01 +0100 | [diff] [blame] | 32 | return &Service{ |
| 33 | gr: gr, |
Serge Bazanski | a758ef5 | 2018-10-06 17:54:25 +0100 | [diff] [blame] | 34 | stm: stm, |
Serge Bazanski | 4676508 | 2018-10-06 12:32:01 +0100 | [diff] [blame] | 35 | config: c, |
| 36 | } |
| 37 | } |
| 38 | |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 39 | const topologyFragment = ` |
| 40 | <script src="/assets/viz.js"></script> |
| 41 | <script> |
| 42 | |
| 43 | var viz = new Viz({ workerURL: '/assets/full.render.js' }); |
| 44 | var xmlhttp = new XMLHttpRequest(); |
| 45 | xmlhttp.onreadystatechange = function() { |
| 46 | if (this.readyState == 4 && this.status == 200) { |
| 47 | var dot = this.responseText; |
| 48 | viz.renderSVGElement(dot) |
| 49 | .then(function(element) { |
Serge Bazanski | 69518ab | 2018-10-06 17:48:23 +0100 | [diff] [blame] | 50 | document.getElementById("graph").appendChild(element); |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 51 | }); |
| 52 | } |
| 53 | }; |
| 54 | xmlhttp.open('GET', '/debug/graphviz'); |
| 55 | xmlhttp.send(); |
| 56 | |
| 57 | </script> |
Serge Bazanski | 69518ab | 2018-10-06 17:48:23 +0100 | [diff] [blame] | 58 | <div id="graph" style="text-align: center;"></div> |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 59 | ` |
| 60 | |
Serge Bazanski | 16e4ba2 | 2018-10-07 00:22:52 +0100 | [diff] [blame^] | 61 | const switchportsFragment = ` |
| 62 | <style type="text/css"> |
| 63 | .table td,th { |
| 64 | background-color: #eee; |
| 65 | padding: 0.2em 0.4em 0.2em 0.4em; |
| 66 | } |
| 67 | .table th { |
| 68 | background-color: #c0c0c0; |
| 69 | } |
| 70 | .table { |
| 71 | background-color: #fff; |
| 72 | border-spacing: 0.2em; |
| 73 | margin-left: auto; |
| 74 | margin-right: auto; |
| 75 | } |
| 76 | </style> |
| 77 | <div> |
| 78 | <table class="table"> |
| 79 | <tr> |
| 80 | <th>Switch</th> |
| 81 | <th>Port</th> |
| 82 | <th>Link State</th> |
| 83 | <th>Port Mode</th> |
| 84 | <th>MTU</th> |
| 85 | <th>Sync Status</th> |
| 86 | </tr> |
| 87 | {{range .Ports }} |
| 88 | {{ if .Managed }} |
| 89 | <tr> |
| 90 | {{ else }} |
| 91 | <tr style="opacity: 0.5"> |
| 92 | {{ end}} |
| 93 | <td style="text-align: right;">{{ .Switch }}</td> |
| 94 | <td>{{ .Name }}</td> |
| 95 | {{ if eq .State "DOWN" }} |
| 96 | <td style="background-color: #ff3030;">{{ .State }}</td> |
| 97 | {{ else }} |
| 98 | <td>{{ .State }}</td> |
| 99 | {{ end }} |
| 100 | <td>{{ .Mode }}</td> |
| 101 | <td>{{ .MTU }}</td> |
| 102 | {{ if .Managed }} |
| 103 | <td style="background-color: #30ff30;">OK</td> |
| 104 | {{ else }} |
| 105 | <td><i>Unmanaged</i></td> |
| 106 | {{ end }} |
| 107 | </tr> |
| 108 | {{end}} |
| 109 | </table> |
| 110 | </div> |
| 111 | ` |
| 112 | |
Serge Bazanski | 4676508 | 2018-10-06 12:32:01 +0100 | [diff] [blame] | 113 | func (s *Service) Run() { |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 114 | assets := packr.NewBox("./assets") |
| 115 | http.Handle("/assets/", http.StripPrefix("/assets/", http.FileServer(assets))) |
Serge Bazanski | 4676508 | 2018-10-06 12:32:01 +0100 | [diff] [blame] | 116 | http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) { |
| 117 | http.Redirect(w, r, "/debug/status", http.StatusSeeOther) |
| 118 | }) |
Serge Bazanski | 16e4ba2 | 2018-10-07 00:22:52 +0100 | [diff] [blame^] | 119 | http.HandleFunc("/debug/graphviz", s.httpHandleGraphviz) |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 120 | |
Serge Bazanski | 16e4ba2 | 2018-10-07 00:22:52 +0100 | [diff] [blame^] | 121 | statusz.AddStatusPart("Switch Ports", switchportsFragment, s.statusHandleSwitchports) |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 122 | statusz.AddStatusPart("Topology", topologyFragment, func(ctx context.Context) interface{} { |
Serge Bazanski | 16e4ba2 | 2018-10-07 00:22:52 +0100 | [diff] [blame^] | 123 | return nil |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 124 | }) |
Serge Bazanski | 4676508 | 2018-10-06 12:32:01 +0100 | [diff] [blame] | 125 | glog.Infof("Debug listening on %s....", s.config.DebugListen) |
| 126 | http.ListenAndServe(s.config.DebugListen, nil) |
Serge Bazanski | 8d7843c | 2018-10-04 10:37:36 +0100 | [diff] [blame] | 127 | } |
Serge Bazanski | 16e4ba2 | 2018-10-07 00:22:52 +0100 | [diff] [blame^] | 128 | |
| 129 | func (s *Service) statusHandleSwitchports(ctx context.Context) interface{} { |
| 130 | managedPorts := make(map[string]bool) |
| 131 | s.gr.Mu.RLock() |
| 132 | for _, sw := range s.gr.Switches { |
| 133 | for _, port := range sw.Ports { |
| 134 | managedPorts[sw.Name+"|"+port.Name] = true |
| 135 | } |
| 136 | } |
| 137 | s.gr.Mu.RUnlock() |
| 138 | |
| 139 | s.stm.Mu.RLock() |
| 140 | defer s.stm.Mu.RUnlock() |
| 141 | |
| 142 | res := struct { |
| 143 | Ports []*struct { |
| 144 | Switch string |
| 145 | Name string |
| 146 | State string |
| 147 | Mode string |
| 148 | Managed bool |
| 149 | MTU string |
| 150 | } |
| 151 | }{} |
| 152 | for _, sw := range s.stm.Switches { |
| 153 | for _, po := range sw.Ports { |
| 154 | state := "INVALID" |
| 155 | switch po.Proto.LinkState { |
| 156 | case pb.SwitchPort_LINKSTATE_DOWN: |
| 157 | state = "DOWN" |
| 158 | case pb.SwitchPort_LINKSTATE_UP: |
| 159 | state = "UP" |
| 160 | } |
| 161 | mode := "INVALID" |
| 162 | switch po.Proto.PortMode { |
| 163 | case pb.SwitchPort_PORTMODE_SWITCHPORT_UNTAGGED: |
| 164 | mode = fmt.Sprintf("UNTAGGED (%d)", po.Proto.VlanNative) |
| 165 | case pb.SwitchPort_PORTMODE_SWITCHPORT_TAGGED: |
| 166 | mode = fmt.Sprintf("TAGGED (%v)", po.Proto.VlanTagged) |
| 167 | case pb.SwitchPort_PORTMODE_SWITCHPORT_GENERIC: |
| 168 | mode = "GENERIC" |
| 169 | case pb.SwitchPort_PORTMODE_ROUTED: |
| 170 | mode = "ROUTED" |
| 171 | case pb.SwitchPort_PORTMODE_MANGLED: |
| 172 | mode = "MANGLED" |
| 173 | } |
| 174 | |
| 175 | managed := managedPorts[sw.Name+"|"+po.Proto.Name] |
| 176 | res.Ports = append(res.Ports, &struct { |
| 177 | Switch string |
| 178 | Name string |
| 179 | State string |
| 180 | Mode string |
| 181 | Managed bool |
| 182 | MTU string |
| 183 | }{ |
| 184 | Switch: sw.Name, |
| 185 | Name: po.Proto.Name, |
| 186 | State: state, |
| 187 | Mode: mode, |
| 188 | Managed: managed, |
| 189 | MTU: fmt.Sprintf("%d", po.Proto.Mtu), |
| 190 | }) |
| 191 | } |
| 192 | } |
| 193 | |
| 194 | return res |
| 195 | } |
| 196 | |
| 197 | func (s *Service) httpHandleGraphviz(w http.ResponseWriter, r *http.Request) { |
| 198 | fmt.Fprintf(w, "graph G {\n") |
| 199 | fmt.Fprintf(w, " ranksep = 2\n") |
| 200 | fmt.Fprintf(w, " splines = polyline\n") |
| 201 | fmt.Fprintf(w, " rankdir = LR\n") |
| 202 | for _, machine := range s.gr.Machines { |
| 203 | portNames := []string{} |
| 204 | for _, port := range machine.Ports { |
| 205 | name := fmt.Sprintf("<%s> %s", port.Name, port.Name) |
| 206 | portNames = append(portNames, name) |
| 207 | } |
| 208 | ports := strings.Join(portNames, "|") |
| 209 | fmt.Fprintf(w, " %s [shape=record label=\"{ %s | { %s }}\"]\n", machine.Name, machine.Name, ports) |
| 210 | } |
| 211 | for _, sw := range s.gr.Switches { |
| 212 | portNames := []string{} |
| 213 | portsOrdered := []*graph.SwitchPort{} |
| 214 | for _, port := range sw.Ports { |
| 215 | portsOrdered = append(portsOrdered, port) |
| 216 | } |
| 217 | sort.Slice(portsOrdered, func(i, j int) bool { |
| 218 | return sortorder.NaturalLess(portsOrdered[i].Name, portsOrdered[j].Name) |
| 219 | }) |
| 220 | for _, port := range portsOrdered { |
| 221 | name := fmt.Sprintf("<%s> %s", port.Name, port.Name) |
| 222 | portNames = append(portNames, name) |
| 223 | } |
| 224 | ports := strings.Join(portNames, "|") |
| 225 | fmt.Fprintf(w, " %s [shape=record label=\"{{ %s } | %s}\"]\n", sw.Name, ports, sw.Name) |
| 226 | } |
| 227 | for _, machine := range s.gr.Machines { |
| 228 | for _, port := range machine.Ports { |
| 229 | if port.OtherEnd == nil { |
| 230 | continue |
| 231 | } |
| 232 | fmt.Fprintf(w, " %s:%q:e -- %s:%q:w\n", machine.Name, port.Name, port.OtherEnd.Switch.Name, port.OtherEnd.Name) |
| 233 | } |
| 234 | } |
| 235 | fmt.Fprintf(w, "}\n") |
| 236 | } |