|
|
@@ -228,6 +228,7 @@ func main() {
|
|
|
adderss := flag.String("b", "127.0.0.1:8288", "Bind address")
|
|
|
flag.StringVar(&docDir, "d", "documents", "Document directory")
|
|
|
distDir := flag.String("w", "dist", "Web dist directory")
|
|
|
+ allowAddr := flag.String("a", "", "Allow other host to connect to websocket")
|
|
|
flag.Parse()
|
|
|
var err error
|
|
|
e = echo.New()
|
|
|
@@ -292,8 +293,12 @@ func main() {
|
|
|
e.GET("/*", func(c echo.Context) error {
|
|
|
return c.File(filepath.Join(*distDir, "index.html"))
|
|
|
})
|
|
|
-
|
|
|
go hub.run()
|
|
|
+ upgrader.CheckOrigin = func(r *http.Request) bool {
|
|
|
+ fmt.Printf("Host: %s\n", r.Host)
|
|
|
+ return r.Host == *adderss || r.Host == *allowAddr
|
|
|
+ }
|
|
|
+
|
|
|
e.Logger.Fatal(e.Start(*adderss))
|
|
|
|
|
|
}
|
