-
-
Notifications
You must be signed in to change notification settings - Fork 1.4k
/
http_server.go
125 lines (98 loc) · 2.69 KB
/
http_server.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
package http_server
import (
"context"
"fmt"
"net/http"
"strings"
"time"
"github.com/bettercap/bettercap/session"
"github.com/evilsocket/islazy/tui"
)
type HttpServer struct {
session.SessionModule
server *http.Server
}
func NewHttpServer(s *session.Session) *HttpServer {
mod := &HttpServer{
SessionModule: session.NewSessionModule("http.server", s),
server: &http.Server{},
}
mod.AddParam(session.NewStringParameter("http.server.path",
".",
"",
"Server folder."))
mod.AddParam(session.NewStringParameter("http.server.address",
session.ParamIfaceAddress,
session.IPv4Validator,
"Address to bind the http server to."))
mod.AddParam(session.NewIntParameter("http.server.port",
"80",
"Port to bind the http server to."))
mod.AddHandler(session.NewModuleHandler("http.server on", "",
"Start httpd server.",
func(args []string) error {
return mod.Start()
}))
mod.AddHandler(session.NewModuleHandler("http.server off", "",
"Stop httpd server.",
func(args []string) error {
return mod.Stop()
}))
return mod
}
func (mod *HttpServer) Name() string {
return "http.server"
}
func (mod *HttpServer) Description() string {
return "A simple HTTP server, to be used to serve files and scripts across the network."
}
func (mod *HttpServer) Author() string {
return "Simone Margaritelli <evilsocket@gmail.com>"
}
func (mod *HttpServer) Configure() error {
var err error
var path string
var address string
var port int
if mod.Running() {
return session.ErrAlreadyStarted(mod.Name())
}
if err, path = mod.StringParam("http.server.path"); err != nil {
return err
}
router := http.NewServeMux()
fileServer := http.FileServer(http.Dir(path))
router.HandleFunc("/", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
mod.Debug("%s %s %s%s", tui.Bold(strings.Split(r.RemoteAddr, ":")[0]), r.Method, r.Host, r.URL.Path)
fileServer.ServeHTTP(w, r)
}))
mod.server.Handler = router
if err, address = mod.StringParam("http.server.address"); err != nil {
return err
}
if err, port = mod.IntParam("http.server.port"); err != nil {
return err
}
mod.server.Addr = fmt.Sprintf("%s:%d", address, port)
return nil
}
func (mod *HttpServer) Start() error {
if err := mod.Configure(); err != nil {
return err
}
return mod.SetRunning(true, func() {
var err error
mod.Info("starting on http://%s", mod.server.Addr)
if err = mod.server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
mod.Error("%v", err)
mod.Stop()
}
})
}
func (mod *HttpServer) Stop() error {
return mod.SetRunning(false, func() {
ctx, cancel := context.WithTimeout(context.Background(), 60*time.Second)
defer cancel()
mod.server.Shutdown(ctx)
})
}