106 lines
2.4 KiB
Go
106 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"container/ring"
|
|
"fmt"
|
|
"github.com/hpcloud/tail"
|
|
"encoding/json"
|
|
"net/http"
|
|
"os"
|
|
"regexp"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
func fileHandler(w http.ResponseWriter, r *http.Request, s string) {
|
|
fn := r.URL.Path[5:len(r.URL.Path)]
|
|
|
|
for _, v := range whitelist {
|
|
if !strings.Contains(v, fn) {
|
|
sendFile(w, fn, "", "Permission denied")
|
|
return
|
|
}
|
|
}
|
|
|
|
if _, err := os.Stat(fn); err != nil {
|
|
sendFile(w, fn, "", "No such file or directory")
|
|
return
|
|
}
|
|
fmt.Println("watch: %#v", watch)
|
|
if (watch == false) {
|
|
t, err := tail.TailFile(fn, tail.Config{Follow: false, MustExist: true})
|
|
watch = true
|
|
if (err == nil) {
|
|
fmt.Println(err)
|
|
}
|
|
}
|
|
|
|
var m = ""
|
|
|
|
w.WriteHeader(http.StatusOK)
|
|
for line := range t.Lines {
|
|
m += fmt.Sprintf("%s\n", line.Text)
|
|
logring.Value = line.Text
|
|
logring = logring.Next()
|
|
}
|
|
if m != m_prev {
|
|
sendFile(w, fn, m, "")
|
|
m_prev = m
|
|
}
|
|
}
|
|
|
|
func sendFile(w http.ResponseWriter, fn string, m string, err string) {
|
|
json.NewEncoder(w).Encode(map[string]string{"time": strconv.FormatInt(getMsTimestamp(), 16), "Filename": fn, "Text": m, "error": err})
|
|
}
|
|
|
|
func getMsTimestamp() (int64) {
|
|
now := time.Now()
|
|
ns := now.UnixNano()
|
|
ms := ns / 1000000
|
|
return ms
|
|
}
|
|
|
|
func makeHandler(fn func(http.ResponseWriter, *http.Request, string)) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
m := validPath.FindStringSubmatch(r.URL.Path)
|
|
if m == nil {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
fmt.Println("makeHandler::request::" + r.RequestURI)
|
|
fn(w, r, m[2])
|
|
}
|
|
}
|
|
|
|
var prj = "httptail"
|
|
var version = "0.1.0"
|
|
var server_inet6 = "[fdef:c181:4234:7978:6::1]:8080"
|
|
var whitelist = map[int]string{
|
|
0: "/var/log",
|
|
}
|
|
|
|
var logring = ring.New(1000)
|
|
var validPath = regexp.MustCompile("^/(tail*|auth)/(.*)$")
|
|
var watch bool = false
|
|
var m_prev = ""
|
|
|
|
func main() {
|
|
uid := os.Getuid()
|
|
|
|
if (uid == -1) {
|
|
fmt.Println("Windows is unsupported")
|
|
}
|
|
|
|
if (uid == 0) {
|
|
fmt.Println("Refusing to run as root")
|
|
return
|
|
}
|
|
|
|
http.HandleFunc("/tail/", makeHandler(fileHandler))
|
|
|
|
fmt.Println("Serving " + prj + " " + version + " on " + server_inet6)
|
|
http.ListenAndServe(server_inet6, nil)
|
|
//t.Cleanup()
|
|
}
|