GuanM/hysteria-dev
1
0
Fork 0
mirror of https://github.com/cmz0228/hysteria-dev.git synced 2025-06-22 12:29:57 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: HTTP traffic stats server

Browse Source
This commit is contained in:
Toby 2023-08-19 17:19:36 -07:00
parent a47285896a
commit e602ec6169
4 changed files with 143 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
app/cmd/server.go
View File

@ -20,6 +20,7 @@ import (
"github.com/apernet/hysteria/extras/auth" "github.com/apernet/hysteria/extras/auth"
"github.com/apernet/hysteria/extras/obfs" "github.com/apernet/hysteria/extras/obfs"
"github.com/apernet/hysteria/extras/outbounds" "github.com/apernet/hysteria/extras/outbounds"
"github.com/apernet/hysteria/extras/trafficlogger"
) )
var serverCmd = &cobra.Command{ var serverCmd = &cobra.Command{
@ -46,6 +47,7 @@ type serverConfig struct {
Resolver serverConfigResolver `mapstructure:"resolver"` Resolver serverConfigResolver `mapstructure:"resolver"`
ACL serverConfigACL `mapstructure:"acl"` ACL serverConfigACL `mapstructure:"acl"`
Outbounds []serverConfigOutboundEntry `mapstructure:"outbounds"` Outbounds []serverConfigOutboundEntry `mapstructure:"outbounds"`
TrafficStats serverConfigTrafficStats `mapstructure:"trafficStats"`
Masquerade serverConfigMasquerade `mapstructure:"masquerade"` Masquerade serverConfigMasquerade `mapstructure:"masquerade"`
} }
@ -160,6 +162,10 @@ type serverConfigOutboundEntry struct {
SOCKS5 serverConfigOutboundSOCKS5 `mapstructure:"socks5"` SOCKS5 serverConfigOutboundSOCKS5 `mapstructure:"socks5"`
} }
type serverConfigTrafficStats struct {
Listen string `mapstructure:"listen"`
}
type serverConfigMasqueradeFile struct { type serverConfigMasqueradeFile struct {
Dir string `mapstructure:"dir"` Dir string `mapstructure:"dir"`
} }
@ -504,6 +510,15 @@ func (c *serverConfig) fillEventLogger(hyConfig *server.Config) error {
return nil return nil
} }
func (c *serverConfig) fillTrafficLogger(hyConfig *server.Config) error {
if c.TrafficStats.Listen != "" {
tss := trafficlogger.NewTrafficStatsServer()
hyConfig.TrafficLogger = tss
go runTrafficStatsServer(c.TrafficStats.Listen, tss)
}
return nil
}
func (c *serverConfig) fillMasqHandler(hyConfig *server.Config) error { func (c *serverConfig) fillMasqHandler(hyConfig *server.Config) error {
switch strings.ToLower(c.Masquerade.Type) { switch strings.ToLower(c.Masquerade.Type) {
case "", "404": case "", "404":
@ -557,6 +572,7 @@ func (c *serverConfig) Config() (*server.Config, error) {
c.fillUDPIdleTimeout, c.fillUDPIdleTimeout,
c.fillAuthenticator, c.fillAuthenticator,
c.fillEventLogger, c.fillEventLogger,
c.fillTrafficLogger,
c.fillMasqHandler, c.fillMasqHandler,
} }
for _, f := range fillers { for _, f := range fillers {
@ -594,6 +610,13 @@ func runServer(cmd *cobra.Command, args []string) {
} }
} }
func runTrafficStatsServer(listen string, handler http.Handler) {
logger.Info("traffic stats server up and running", zap.String("listen", listen))
if err := http.ListenAndServe(listen, handler); err != nil {
logger.Fatal("failed to serve traffic stats", zap.Error(err))
}
}
func geoipDownloadFunc(filename, url string) { func geoipDownloadFunc(filename, url string) {
logger.Info("downloading GeoIP database", zap.String("filename", filename), zap.String("url", url)) logger.Info("downloading GeoIP database", zap.String("filename", filename), zap.String("url", url))
} }

3
app/cmd/server_test.go
View File

@ -124,6 +124,9 @@ func TestServerConfig(t *testing.T) {
}, },
}, },
}, },
TrafficStats: serverConfigTrafficStats{
Listen: ":9999",
},
Masquerade: serverConfigMasquerade{ Masquerade: serverConfigMasquerade{
Type: "proxy", Type: "proxy",
File: serverConfigMasqueradeFile{ File: serverConfigMasqueradeFile{

3
app/cmd/server_test.yaml
View File

@ -92,6 +92,9 @@ outbounds:
username: hackerman username: hackerman
password: Elliot Alderson password: Elliot Alderson
trafficStats:
listen: :9999
masquerade: masquerade:
type: proxy type: proxy
file: file:

114
extras/trafficlogger/http.go Normal file
View File

@ -0,0 +1,114 @@
package trafficlogger
import (
"encoding/json"
"net/http"
"strconv"
"sync"
"github.com/apernet/hysteria/core/server"
)
const (
indexHTML = `<!DOCTYPE html><html lang="en"><head> <meta charset="UTF-8"> <meta name="viewport" content="width=device-width, initial-scale=1.0"> <title>Hysteria Traffic Stats API Server</title> <style>body{font-family: Arial, sans-serif; display: flex; justify-content: center; align-items: center; height: 100vh; margin: 0; padding: 0; background-color: #f4f4f4;}.container{padding: 20px; background-color: #fff; box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1); border-radius: 5px;}</style></head><body> <div class="container"> <p>This is a Hysteria Traffic Stats API server.</p><p>Check the documentation for usage.</p></div></body></html>`
)
// TrafficStatsServer implements both server.TrafficLogger and http.Handler
// to provide a simple HTTP API to get the traffic stats per user.
type TrafficStatsServer interface {
server.TrafficLogger
http.Handler
}
func NewTrafficStatsServer() TrafficStatsServer {
return &trafficStatsServerImpl{
StatsMap: make(map[string]*trafficStatsEntry),
KickMap: make(map[string]struct{}),
}
}
type trafficStatsServerImpl struct {
Mutex sync.RWMutex
StatsMap map[string]*trafficStatsEntry
KickMap map[string]struct{}
}
type trafficStatsEntry struct {
Tx uint64 `json:"tx"`
Rx uint64 `json:"rx"`
}
func (s *trafficStatsServerImpl) Log(id string, tx, rx uint64) (ok bool) {
s.Mutex.Lock()
defer s.Mutex.Unlock()
_, ok = s.KickMap[id]
if ok {
delete(s.KickMap, id)
return false
}
entry, ok := s.StatsMap[id]
if !ok {
entry = &trafficStatsEntry{}
s.StatsMap[id] = entry
}
entry.Tx += tx
entry.Rx += rx
return true
}
func (s *trafficStatsServerImpl) ServeHTTP(w http.ResponseWriter, r *http.Request) {
if r.Method == http.MethodGet && r.URL.Path == "/" {
_, _ = w.Write([]byte(indexHTML))
return
}
if r.Method == http.MethodGet && r.URL.Path == "/traffic" {
s.getTraffic(w, r)
return
}
if r.Method == http.MethodPost && r.URL.Path == "/kick" {
s.kick(w, r)
return
}
http.NotFound(w, r)
}
func (s *trafficStatsServerImpl) getTraffic(w http.ResponseWriter, r *http.Request) {
bClear, _ := strconv.ParseBool(r.URL.Query().Get("clear"))
var jb []byte
var err error
if bClear {
s.Mutex.Lock()
jb, err = json.Marshal(s.StatsMap)
s.StatsMap = make(map[string]*trafficStatsEntry)
s.Mutex.Unlock()
} else {
s.Mutex.RLock()
jb, err = json.Marshal(s.StatsMap)
s.Mutex.RUnlock()
}
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
w.Header().Set("Content-Type", "application/json; charset=utf-8")
_, _ = w.Write(jb)
}
func (s *trafficStatsServerImpl) kick(w http.ResponseWriter, r *http.Request) {
var ids []string
err := json.NewDecoder(r.Body).Decode(&ids)
if err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
s.Mutex.Lock()
for _, id := range ids {
s.KickMap[id] = struct{}{}
}
s.Mutex.Unlock()
w.WriteHeader(http.StatusOK)
}