102 lines
3.0 KiB
Go
102 lines
3.0 KiB
Go
|
package app
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"fmt"
|
||
|
|
"html/template"
|
||
|
|
"log"
|
||
|
|
"net/http"
|
||
|
|
"os"
|
||
|
|
"strings"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
var tpl = template.Must(template.New("base").Parse(`
|
||
|
|
<!doctype html>
|
||
|
|
<html>
|
||
|
|
<head><meta charset="utf-8"><title>Honeypot Dashboard</title></head>
|
||
|
|
<body>
|
||
|
|
<h1>Honeypot Dashboard</h1>
|
||
|
|
<p><a href="/logs">Logs</a></p>
|
||
|
|
{{ .Body }}
|
||
|
|
</body>
|
||
|
|
</html>
|
||
|
|
`))
|
||
|
|
|
||
|
|
func (a *App) startWeb() {
|
||
|
|
bind := a.cfg.Web.Bind
|
||
|
|
port := a.cfg.Web.Port
|
||
|
|
addr := fmt.Sprintf("%s:%d", bind, port)
|
||
|
|
mux := http.NewServeMux()
|
||
|
|
mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
tpl.Execute(w, map[string]interface{}{"Body": template.HTML("<p><a href=\"/logs\">View logs</a></p>")})
|
||
|
|
})
|
||
|
|
mux.HandleFunc("/logs", func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
// display last 200 logs
|
||
|
|
var rows []Record
|
||
|
|
if a.logger != nil && a.logger.mode == "sqlite" && a.logger.db != nil {
|
||
|
|
// query sqlite
|
||
|
|
q := `SELECT timestamp, remote_addr, remote_port, service, details, raw_payload FROM logs ORDER BY id DESC LIMIT 200`
|
||
|
|
rs, err := a.logger.db.Query(q)
|
||
|
|
if err != nil {
|
||
|
|
http.Error(w, "db query failed", 500)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
defer rs.Close()
|
||
|
|
for rs.Next() {
|
||
|
|
var ts, ra, rp, svc, detailsS, raw string
|
||
|
|
if err := rs.Scan(&ts, &ra, &rp, &svc, &detailsS, &raw); err != nil {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
var det map[string]string
|
||
|
|
_ = json.Unmarshal([]byte(detailsS), &det)
|
||
|
|
rows = append(rows, Record{Timestamp: parseTime(ts), RemoteAddr: ra, RemotePort: rp, Service: svc, Details: det, RawPayload: raw})
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
// try to read file based JSON-lines
|
||
|
|
path := a.cfg.LogPath
|
||
|
|
b, err := os.ReadFile(path)
|
||
|
|
if err == nil {
|
||
|
|
lines := strings.Split(string(b), "\n")
|
||
|
|
for i := len(lines) - 1; i >= 0 && len(rows) < 200; i-- {
|
||
|
|
line := strings.TrimSpace(lines[i])
|
||
|
|
if line == "" {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
var rec Record
|
||
|
|
if err := json.Unmarshal([]byte(line), &rec); err != nil {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
rows = append(rows, rec)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// render simple table
|
||
|
|
var sb strings.Builder
|
||
|
|
sb.WriteString("<table border=1 cellpadding=4><tr><th>Time</th><th>Remote</th><th>Service</th><th>Details</th><th>Payload</th></tr>")
|
||
|
|
for _, r := range rows {
|
||
|
|
detB, _ := json.Marshal(r.Details)
|
||
|
|
sb.WriteString(fmt.Sprintf("<tr><td>%s</td><td>%s:%s</td><td>%s</td><td>%s</td><td><pre>%s</pre></td></tr>", r.Timestamp.Format("2006-01-02 15:04:05"), r.RemoteAddr, r.RemotePort, r.Service, template.HTMLEscapeString(string(detB)), template.HTMLEscapeString(r.RawPayload)))
|
||
|
|
}
|
||
|
|
sb.WriteString("</table>")
|
||
|
|
tpl.Execute(w, map[string]interface{}{"Body": template.HTML(sb.String())})
|
||
|
|
})
|
||
|
|
|
||
|
|
srv := &http.Server{Addr: addr, Handler: mux}
|
||
|
|
a.httpSrv = srv
|
||
|
|
log.Printf("Dashboard listening on http://%s", addr)
|
||
|
|
if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
||
|
|
log.Printf("dashboard error: %v", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func parseTime(s string) (t time.Time) {
|
||
|
|
t, _ = time.Parse(time.RFC3339Nano, s)
|
||
|
|
if t.IsZero() {
|
||
|
|
// fallback current time
|
||
|
|
t = time.Now()
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|