Added TLS / secure cert for Websockets (#42)

* Added TLS / secure cert for Websockets

* Added exclusion of current id to group broadcast.

* bugfix for listen interface

* Revert broadcastExcludeId, change not needed.

Author: brcoding

app.go

@@ -37,6 +37,10 @@ import (
 var running int32
 
 func listen(addr string, isWs bool, opts ...Option) {
+	listenTLS(addr, isWs, "", "", opts...);
+}
+
+func listenTLS(addr string, isWs bool, certificate string, key string, opts ...Option) {
 	// mark application running
 	if atomic.AddInt32(&running, 1) != 1 {
 		logger.Println("Nano has running")
@@ -62,7 +66,11 @@ func listen(addr string, isWs bool, opts ...Option) {
 
 	go func() {
 		if isWs {
-			listenAndServeWS(addr)
+			if len(certificate) != 0 {
+				listenAndServeWSTLS(addr, certificate, key)
+			} else {
+				listenAndServeWS(addr)
+			}
 		} else {
 			listenAndServe(addr)
 		}
@@ -128,3 +136,25 @@ func listenAndServeWS(addr string) {
 		logger.Fatal(err.Error())
 	}
 }
+
+func listenAndServeWSTLS(addr string, certificate string, key string) {
+	var upgrader = websocket.Upgrader{
+		ReadBufferSize:  1024,
+		WriteBufferSize: 1024,
+		CheckOrigin:     env.checkOrigin,
+	}
+
+	http.HandleFunc("/"+strings.TrimPrefix(env.wsPath, "/"), func(w http.ResponseWriter, r *http.Request) {
+		conn, err := upgrader.Upgrade(w, r, nil)
+		if err != nil {
+			logger.Println(fmt.Sprintf("Upgrade failure, URI=%s, Error=%s", r.RequestURI, err.Error()))
+			return
+		}
+
+		handler.handleWS(conn)
+	})
+
+	if err := http.ListenAndServeTLS(addr, certificate, key, nil); err != nil {
+		logger.Fatal(err.Error())
+	}
+}

group.go

@@ -224,4 +224,4 @@ func (c *Group) Close() error {
 	// release all reference
 	c.sessions = make(map[int64]*session.Session)
 	return nil
-}
+}

interface.go

@@ -42,6 +42,13 @@ func ListenWS(addr string, opts ...Option) {
 	listen(addr, true, opts...)
 }
 
+// ListenWS listens on the TCP network address addr
+// and then upgrades the HTTP server connection to the WebSocket protocol
+// to handle requests on incoming connections.
+func ListenWSTLS(addr string, certificate string, key string, opts ...Option) {
+	listenTLS(addr, true, certificate, key, opts...)
+}
+
 // Register register a component with options
 func Register(c component.Component, options ...component.Option) {
 	comps = append(comps, regComp{c, options})