telebit/listener_client.go

package main

import (
	"net/http"

	jwt "github.com/dgrijalva/jwt-go"
)

//launchClientListener - starts up http listeners and handles various URI paths
func launchClientListener() {
	loginfo.Println("starting WebRequestExternal Listener ", *argServerBinding)

	mux := http.NewServeMux()
	mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		switch url := r.URL.Path; url {
		case "/":
			handleConnectionWebSocket(connectionTable, w, r, false)

		default:
			http.Error(w, "Not Found", 404)

		}

	})

	s := &http.Server{
		Addr:    *argServerBinding,
		Handler: mux,
	}

	err := s.ListenAndServeTLS("certs/fullchain.pem", "certs/privkey.pem")
	if err != nil {
		logfatal.Println("ListenAndServeTLS: ", err)
		panic(err)
	}
}

// handleConnectionWebSocket handles websocket requests from the peer.
func handleConnectionWebSocket(connectionTable *ConnectionTable, w http.ResponseWriter, r *http.Request, admin bool) {
	loginfo.Println("websocket opening ", r.RemoteAddr, " ", r.Host)

	tokenString := r.URL.Query().Get("access_token")
	result, err := jwt.Parse(tokenString, func(token *jwt.Token) (interface{}, error) {
		return []byte(secretKey), nil
	})

	if err != nil || !result.Valid {
		w.WriteHeader(http.StatusForbidden)
		w.Write([]byte("Not Authorized"))
		loginfo.Println("access_token invalid...closing connection")
		return
	}

	loginfo.Println("access_token valid")

	claims := result.Claims.(jwt.MapClaims)
	loginfo.Println("processing domains", claims["domains"])

	if admin == true {
		loginfo.Println("Recognized Admin connection, waiting authentication")
	} else {
		loginfo.Println("Recognized connection, waiting authentication")
	}

	conn, err := upgrader.Upgrade(w, r, nil)
	if err != nil {
		loginfo.Println("WebSocket upgrade failed", err)
		return
	}
	loginfo.Println("before connection table")
	connection := &Connection{connectionTable: connectionTable, conn: conn, send: make(chan []byte, 256), source: r.RemoteAddr, admin: admin}
	connection.connectionTable.register <- connection
	go connection.writer()
	//go connection.sender()
	connection.reader()
}