clash/hub/route/server.go

237 lines
4.8 KiB
Go
Raw Normal View History

2018-11-21 05:47:46 +00:00
package route
2018-06-18 03:31:49 +00:00
import (
"bytes"
2018-06-18 03:31:49 +00:00
"encoding/json"
"net/http"
2018-10-06 05:15:02 +00:00
"strings"
2018-06-18 03:31:49 +00:00
"time"
2019-09-13 09:44:30 +00:00
C "github.com/Dreamacro/clash/constant"
2018-11-21 05:47:46 +00:00
"github.com/Dreamacro/clash/log"
2018-07-15 14:23:20 +00:00
T "github.com/Dreamacro/clash/tunnel"
2018-06-18 03:31:49 +00:00
"github.com/go-chi/chi"
2018-07-19 01:31:53 +00:00
"github.com/go-chi/cors"
2018-06-18 03:31:49 +00:00
"github.com/go-chi/render"
"github.com/gorilla/websocket"
2018-06-18 03:31:49 +00:00
)
2018-11-21 05:47:46 +00:00
var (
serverSecret = ""
serverAddr = ""
2018-12-19 17:29:13 +00:00
uiPath = ""
2019-09-30 06:13:29 +00:00
upgrader = websocket.Upgrader{
CheckOrigin: func(r *http.Request) bool {
return true
},
}
2018-11-21 05:47:46 +00:00
)
2018-10-06 05:15:02 +00:00
2018-06-18 03:31:49 +00:00
type Traffic struct {
Up int64 `json:"up"`
Down int64 `json:"down"`
}
2018-12-19 17:29:13 +00:00
func SetUIPath(path string) {
uiPath = path
}
2018-11-21 05:47:46 +00:00
func Start(addr string, secret string) {
if serverAddr != "" {
return
}
2018-11-21 05:47:46 +00:00
serverAddr = addr
serverSecret = secret
2018-06-18 03:31:49 +00:00
r := chi.NewRouter()
2018-07-19 01:31:53 +00:00
cors := cors.New(cors.Options{
AllowedOrigins: []string{"*"},
2018-12-29 06:11:54 +00:00
AllowedMethods: []string{"GET", "POST", "PUT", "PATCH", "DELETE"},
2018-10-06 05:15:02 +00:00
AllowedHeaders: []string{"Content-Type", "Authorization"},
2018-07-19 01:31:53 +00:00
MaxAge: 300,
})
r.Use(cors.Handler)
2018-12-29 06:11:54 +00:00
r.Get("/", hello)
r.Group(func(r chi.Router) {
r.Use(authentication)
r.Get("/logs", getLogs)
r.Get("/traffic", traffic)
2019-09-13 09:44:30 +00:00
r.Get("/version", version)
r.Mount("/configs", configRouter())
r.Mount("/proxies", proxyRouter())
r.Mount("/rules", ruleRouter())
r.Mount("/connections", connectionRouter())
r.Mount("/providers/proxies", proxyProviderRouter())
})
2018-06-18 03:31:49 +00:00
2018-12-19 17:29:13 +00:00
if uiPath != "" {
r.Group(func(r chi.Router) {
fs := http.StripPrefix("/ui", http.FileServer(http.Dir(uiPath)))
2019-09-08 14:33:52 +00:00
r.Get("/ui", http.RedirectHandler("/ui/", http.StatusTemporaryRedirect).ServeHTTP)
r.Get("/ui/*", func(w http.ResponseWriter, r *http.Request) {
fs.ServeHTTP(w, r)
})
2018-12-19 17:29:13 +00:00
})
}
2018-11-21 05:47:46 +00:00
log.Infoln("RESTful API listening at: %s", addr)
2018-06-18 03:31:49 +00:00
err := http.ListenAndServe(addr, r)
if err != nil {
2018-11-21 05:47:46 +00:00
log.Errorln("External controller error: %s", err.Error())
2018-06-18 03:31:49 +00:00
}
}
2018-10-06 05:15:02 +00:00
func authentication(next http.Handler) http.Handler {
fn := func(w http.ResponseWriter, r *http.Request) {
2018-11-21 05:47:46 +00:00
if serverSecret == "" {
2018-10-06 05:15:02 +00:00
next.ServeHTTP(w, r)
return
}
2019-09-30 06:13:29 +00:00
// Browser websocket not support custom header
if websocket.IsWebSocketUpgrade(r) && r.URL.Query().Get("token") != "" {
token := r.URL.Query().Get("token")
if token != serverSecret {
render.Status(r, http.StatusUnauthorized)
render.JSON(w, r, ErrUnauthorized)
return
}
next.ServeHTTP(w, r)
return
}
header := r.Header.Get("Authorization")
text := strings.SplitN(header, " ", 2)
2018-10-06 05:15:02 +00:00
hasUnvalidHeader := text[0] != "Bearer"
2018-11-21 05:47:46 +00:00
hasUnvalidSecret := len(text) == 2 && text[1] != serverSecret
2018-10-06 05:15:02 +00:00
if hasUnvalidHeader || hasUnvalidSecret {
2018-12-10 03:33:37 +00:00
render.Status(r, http.StatusUnauthorized)
render.JSON(w, r, ErrUnauthorized)
2018-10-06 05:15:02 +00:00
return
}
next.ServeHTTP(w, r)
}
return http.HandlerFunc(fn)
}
2018-12-29 06:11:54 +00:00
func hello(w http.ResponseWriter, r *http.Request) {
render.JSON(w, r, render.M{"hello": "clash"})
}
2018-06-18 03:31:49 +00:00
func traffic(w http.ResponseWriter, r *http.Request) {
var wsConn *websocket.Conn
if websocket.IsWebSocketUpgrade(r) {
var err error
wsConn, err = upgrader.Upgrade(w, r, nil)
if err != nil {
return
}
}
if wsConn == nil {
w.Header().Set("Content-Type", "application/json")
render.Status(r, http.StatusOK)
}
2018-06-18 03:31:49 +00:00
tick := time.NewTicker(time.Second)
t := T.DefaultManager
buf := &bytes.Buffer{}
var err error
2018-06-18 03:31:49 +00:00
for range tick.C {
buf.Reset()
2018-06-18 03:31:49 +00:00
up, down := t.Now()
if err := json.NewEncoder(buf).Encode(Traffic{
2018-06-18 03:31:49 +00:00
Up: up,
Down: down,
}); err != nil {
break
}
if wsConn == nil {
_, err = w.Write(buf.Bytes())
w.(http.Flusher).Flush()
} else {
err = wsConn.WriteMessage(websocket.TextMessage, buf.Bytes())
}
if err != nil {
break
}
2018-06-18 03:31:49 +00:00
}
}
2018-06-22 05:15:20 +00:00
type Log struct {
Type string `json:"type"`
Payload string `json:"payload"`
}
2018-06-18 03:31:49 +00:00
func getLogs(w http.ResponseWriter, r *http.Request) {
2018-08-12 11:35:13 +00:00
levelText := r.URL.Query().Get("level")
if levelText == "" {
levelText = "info"
2018-06-22 05:15:20 +00:00
}
2018-11-21 05:47:46 +00:00
level, ok := log.LogLevelMapping[levelText]
2018-06-22 05:15:20 +00:00
if !ok {
2018-12-10 03:33:37 +00:00
render.Status(r, http.StatusBadRequest)
render.JSON(w, r, ErrBadRequest)
2018-06-22 05:15:20 +00:00
return
}
var wsConn *websocket.Conn
if websocket.IsWebSocketUpgrade(r) {
var err error
wsConn, err = upgrader.Upgrade(w, r, nil)
if err != nil {
return
}
}
if wsConn == nil {
w.Header().Set("Content-Type", "application/json")
render.Status(r, http.StatusOK)
}
2018-11-21 05:47:46 +00:00
sub := log.Subscribe()
defer log.UnSubscribe(sub)
buf := &bytes.Buffer{}
var err error
2018-06-18 03:31:49 +00:00
for elm := range sub {
buf.Reset()
2018-11-21 05:47:46 +00:00
log := elm.(*log.Event)
if log.LogLevel < level {
2018-06-22 05:15:20 +00:00
continue
}
if err := json.NewEncoder(buf).Encode(Log{
2018-06-18 03:31:49 +00:00
Type: log.Type(),
Payload: log.Payload,
}); err != nil {
break
}
if wsConn == nil {
_, err = w.Write(buf.Bytes())
w.(http.Flusher).Flush()
} else {
err = wsConn.WriteMessage(websocket.TextMessage, buf.Bytes())
}
if err != nil {
break
}
2018-06-18 03:31:49 +00:00
}
}
2019-09-13 09:44:30 +00:00
func version(w http.ResponseWriter, r *http.Request) {
render.JSON(w, r, render.M{"version": C.Version})
}