first commit
This commit is contained in:
+101
@@ -0,0 +1,101 @@
|
||||
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
|
||||
}
|
||||
Reference in New Issue
Block a user