Hunter0x7c7
2022-08-11 b8230139fb40edea387617b6accd8371e37eda58
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
package tcp
 
import (
    "net/http"
 
    "github.com/v2fly/v2ray-core/v5/common/net"
)
 
type Server struct {
    Port        net.Port
    PathHandler map[string]http.HandlerFunc
    server      *http.Server
}
 
func (s *Server) ServeHTTP(resp http.ResponseWriter, req *http.Request) {
    if req.URL.Path == "/" {
        resp.Header().Set("Content-Type", "text/plain; charset=utf-8")
        resp.WriteHeader(http.StatusOK)
        resp.Write([]byte("Home"))
        return
    }
 
    handler, found := s.PathHandler[req.URL.Path]
    if found {
        handler(resp, req)
    }
}
 
func (s *Server) Start() (net.Destination, error) {
    s.server = &http.Server{
        Addr:    "127.0.0.1:" + s.Port.String(),
        Handler: s,
    }
    go s.server.ListenAndServe()
    return net.TCPDestination(net.LocalHostIP, s.Port), nil
}
 
func (s *Server) Close() error {
    return s.server.Close()
}