Merge pull request #195 from nhooyr/example-cross-origin

nhooyr · web-flow · commit fa720b9bd47f · 2020-02-21T01:42:17.000-05:00
Add Example_crossOrigin
diff --git a/example_test.go b/example_test.go
@@ -6,6 +6,7 @@ import (
 	"context"
 	"log"
 	"net/http"
+	"net/url"
 	"time"
 
 	"nhooyr.io/websocket"
@@ -115,3 +116,30 @@ func Example_writeOnly() {
 	err := http.ListenAndServe("localhost:8080", fn)
 	log.Fatal(err)
 }
+
+// This example demonstrates how to safely accept cross origin WebSockets
+// from the origin example.com.
+func Example_crossOrigin() {
+	fn := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		origin := r.Header.Get("Origin")
+		if origin != "" {
+			u, err := url.Parse(origin)
+			if err != nil || u.Host != "example.com" {
+				http.Error(w, "bad origin header", http.StatusForbidden)
+				return
+			}
+		}
+
+		c, err := websocket.Accept(w, r, &websocket.AcceptOptions{
+			InsecureSkipVerify: true,
+		})
+		if err != nil {
+			log.Println(err)
+			return
+		}
+		c.Close(websocket.StatusNormalClosure, "cross origin WebSocket accepted")
+	})
+
+	err := http.ListenAndServe("localhost:8080", fn)
+	log.Fatal(err)
+}
