mirror of
https://github.com/syssecfsu/witty.git
synced 2024-12-24 11:42:34 +01:00
123 lines
2.4 KiB
Go
123 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"log"
|
|
"net/http"
|
|
"net/url"
|
|
"os"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/syssecfsu/web_terminal/term_conn"
|
|
)
|
|
|
|
// command line options
|
|
var cmdToExec = []string{"bash"}
|
|
|
|
var host *string = nil
|
|
|
|
// simple function to check origin
|
|
func checkOrigin(r *http.Request) bool {
|
|
org := r.Header.Get("Origin")
|
|
h, err := url.Parse(org)
|
|
|
|
if err != nil {
|
|
return false
|
|
}
|
|
|
|
if (host == nil) || (*host != h.Host) {
|
|
log.Println("Failed origin check of ", org)
|
|
}
|
|
|
|
return (host != nil) && (*host == h.Host)
|
|
}
|
|
|
|
type InteractiveSession struct {
|
|
Ip string
|
|
Cmd string
|
|
Name string
|
|
}
|
|
|
|
func fillIndex(c *gin.Context) {
|
|
var players []InteractiveSession
|
|
|
|
term_conn.ForEachSession(func(tc *term_conn.TermConn) {
|
|
players = append(players, InteractiveSession{
|
|
Name: tc.Name,
|
|
Ip: tc.Ip,
|
|
Cmd: cmdToExec[0],
|
|
})
|
|
})
|
|
|
|
c.HTML(http.StatusOK, "index.html", gin.H{
|
|
"title": "interactive terminal",
|
|
"path": "/ws_do",
|
|
"players": players,
|
|
})
|
|
}
|
|
|
|
func main() {
|
|
fp, err := os.OpenFile("witty.log", os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0644)
|
|
|
|
if err == nil {
|
|
defer fp.Close()
|
|
log.SetOutput(fp)
|
|
gin.DefaultWriter = fp
|
|
}
|
|
|
|
// parse the arguments. User can pass the command to execute
|
|
// by default, we use bash, but macos users might want to use zsh
|
|
// you can also run single program, such as pstree, htop...
|
|
// but program might misbehave (htop seems to be fine)
|
|
args := os.Args
|
|
|
|
if len(args) > 1 {
|
|
cmdToExec = args[1:]
|
|
log.Println(cmdToExec)
|
|
}
|
|
|
|
rt := gin.Default()
|
|
|
|
rt.SetTrustedProxies(nil)
|
|
rt.LoadHTMLGlob("./assets/*.html")
|
|
|
|
rt.GET("/view/:sname", func(c *gin.Context) {
|
|
sname := c.Param("sname")
|
|
c.HTML(http.StatusOK, "term.html", gin.H{
|
|
"title": "viewer terminal",
|
|
"path": "/ws_view/" + sname,
|
|
})
|
|
})
|
|
|
|
rt.GET("/new", func(c *gin.Context) {
|
|
if host == nil {
|
|
host = &c.Request.Host
|
|
}
|
|
|
|
c.HTML(http.StatusOK, "term.html", gin.H{
|
|
"title": "interactive terminal",
|
|
"path": "/ws_new",
|
|
})
|
|
})
|
|
|
|
rt.GET("/ws_new", func(c *gin.Context) {
|
|
term_conn.ConnectTerm(c.Writer, c.Request, false, "", cmdToExec)
|
|
})
|
|
|
|
rt.GET("/ws_view/:sname", func(c *gin.Context) {
|
|
path := c.Param("sname")
|
|
term_conn.ConnectTerm(c.Writer, c.Request, true, path, nil)
|
|
})
|
|
|
|
// handle static files
|
|
rt.Static("/assets", "./assets")
|
|
|
|
rt.GET("/", func(c *gin.Context) {
|
|
host = &c.Request.Host
|
|
fillIndex(c)
|
|
})
|
|
|
|
term_conn.Init(checkOrigin)
|
|
|
|
rt.RunTLS(":8080", "./tls/cert.pem", "./tls/private-key.pem")
|
|
}
|