serve tls traffic and static files to view webcam locally and open websocket conn

master
bel 2020-05-03 23:09:46 -06:00
parent dda64da85e
commit 2388723fa5
7 changed files with 267 additions and 0 deletions

2
.gitignore vendored
View File

@ -1,6 +1,8 @@
entropy
vendor
exec-entropy
**.sw*
**/**.sw*
*.sw*
**/*.sw*
testdata

17
config.go Normal file
View File

@ -0,0 +1,17 @@
package main
import "local/args"
func config() *args.ArgSet {
as := args.NewArgSet()
as.Append(args.INT, "p", "port to listen on", "58080")
as.Append(args.STRING, "d", "root dir to serve static", ".")
as.Append(args.STRING, "crt", "path to crt", "./cert.crt")
as.Append(args.STRING, "key", "path to key", "./cert.key")
if err := as.Parse(); err != nil {
panic(err)
}
return as
}

34
main.go Normal file
View File

@ -0,0 +1,34 @@
package main
import (
"crypto/tls"
"fmt"
"log"
"net/http"
)
func main() {
as := config()
httpsServer := &http.Server{
Addr: fmt.Sprintf(":%d", as.GetInt("p")),
Handler: New(),
TLSConfig: &tls.Config{
MinVersion: tls.VersionTLS12,
CurvePreferences: []tls.CurveID{tls.CurveP521, tls.CurveP384, tls.CurveP256},
PreferServerCipherSuites: true,
CipherSuites: []uint16{
tls.TLS_ECDHE_RSA_WITH_AES_256_GCM_SHA384,
tls.TLS_ECDHE_RSA_WITH_AES_256_CBC_SHA,
tls.TLS_RSA_WITH_AES_256_GCM_SHA384,
tls.TLS_RSA_WITH_AES_256_CBC_SHA,
},
},
TLSNextProto: make(map[string]func(*http.Server, *tls.Conn, http.Handler), 0),
}
c := as.GetString("crt")
k := as.GetString("key")
log.Printf("listening on %q", httpsServer.Addr)
if err := httpsServer.ListenAndServeTLS(c, k); err != nil {
panic(err)
}
}

27
public/index.html Executable file
View File

@ -0,0 +1,27 @@
<!DOCTYPE html>
<html>
<head>
<script src="https://webrtc.github.io/adapter/adapter-latest.js"></script>
<script src="webrtc.js"></script>
<style>
video {
max-width: 100%;
min-width: 100%;
border: 1px solid black;
}
</style>
</head>
<body>
<video id="localVideo" autoplay muted ></video>
<video id="remoteVideo" autoplay ></video>
<br />
<input type="button" id="start" onclick="start(true)" value="Start Video"></input>
<script type="text/javascript">
pageReady();
</script>
</body>
</html>

103
public/webrtc.js Executable file
View File

@ -0,0 +1,103 @@
var localVideo;
var localStream;
var remoteVideo;
var peerConnection;
var uuid;
var serverConnection;
var peerConnectionConfig = {
'iceServers': [
{'urls': 'stun:stun.stunprotocol.org:3478'},
{'urls': 'stun:stun.l.google.com:19302'},
]
};
function pageReady() {
uuid = createUUID();
localVideo = document.getElementById('localVideo');
remoteVideo = document.getElementById('remoteVideo');
serverConnection = new WebSocket('wss://' + window.location.hostname + '/abc');
serverConnection.onmessage = gotMessageFromServer;
var constraints = {
video: true,
audio: true,
};
if(navigator.mediaDevices.getUserMedia) {
navigator.mediaDevices.getUserMedia(constraints).then(getUserMediaSuccess).catch(errorHandler);
} else {
alert('Your browser does not support getUserMedia API');
}
}
function getUserMediaSuccess(stream) {
localStream = stream;
localVideo.srcObject = stream;
}
function start(isCaller) {
peerConnection = new RTCPeerConnection(peerConnectionConfig);
peerConnection.onicecandidate = gotIceCandidate;
peerConnection.ontrack = gotRemoteStream;
peerConnection.addStream(localStream);
if(isCaller) {
peerConnection.createOffer().then(createdDescription).catch(errorHandler);
}
}
function gotMessageFromServer(message) {
if(!peerConnection) start(false);
var signal = JSON.parse(message.data);
// Ignore messages from ourself
if(signal.uuid == uuid) return;
if(signal.sdp) {
peerConnection.setRemoteDescription(new RTCSessionDescription(signal.sdp)).then(function() {
// Only create answers in response to offers
if(signal.sdp.type == 'offer') {
peerConnection.createAnswer().then(createdDescription).catch(errorHandler);
}
}).catch(errorHandler);
} else if(signal.ice) {
peerConnection.addIceCandidate(new RTCIceCandidate(signal.ice)).catch(errorHandler);
}
}
function gotIceCandidate(event) {
if(event.candidate != null) {
serverConnection.send(JSON.stringify({'ice': event.candidate, 'uuid': uuid}));
}
}
function createdDescription(description) {
console.log('got description');
peerConnection.setLocalDescription(description).then(function() {
serverConnection.send(JSON.stringify({'sdp': peerConnection.localDescription, 'uuid': uuid}));
}).catch(errorHandler);
}
function gotRemoteStream(event) {
console.log('got remote stream');
remoteVideo.srcObject = event.streams[0];
}
function errorHandler(error) {
console.log(error);
}
// Taken from http://stackoverflow.com/a/105074/515584
// Strictly speaking, it's not a real UUID, but it gets the job done here
function createUUID() {
function s4() {
return Math.floor((1 + Math.random()) * 0x10000).toString(16).substring(1);
}
return s4() + s4() + '-' + s4() + '-' + s4() + '-' + s4() + '-' + s4() + s4() + s4();
}

31
server.go Normal file
View File

@ -0,0 +1,31 @@
package main
import (
"log"
"net/http"
"os"
"path"
)
type Server struct {
fs http.Handler
ws *WS
}
func New() *Server {
fs := http.FileServer(http.Dir(config().GetString("d")))
return &Server{
fs: fs,
ws: NewWS(),
}
}
func (s *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
log.Println("serving", r.URL)
if _, err := os.Stat(path.Join(config().GetString("d"), r.URL.Path[1:])); os.IsNotExist(err) {
s.ws.ServeHTTP(w, r)
} else {
log.Printf("Serving static %q from %q", r.URL.Path, config().GetString("d"))
s.fs.ServeHTTP(w, r)
}
}

53
ws.go Normal file
View File

@ -0,0 +1,53 @@
package main
import (
"io"
"log"
"net/http"
"github.com/gorilla/websocket"
)
type WS struct {
upgrader websocket.Upgrader
}
func NewWS() *WS {
return &WS{
upgrader: websocket.Upgrader{
ReadBufferSize: 10240,
WriteBufferSize: 10240,
CheckOrigin: func(_ *http.Request) bool { return true },
},
}
}
func (ws *WS) ServeHTTP(w http.ResponseWriter, r *http.Request) {
if err := ws.serveHTTP(w, r); err != nil {
log.Println(r.URL.Path, err)
http.Error(w, err.Error(), http.StatusInternalServerError)
}
}
func (ws *WS) serveHTTP(w http.ResponseWriter, r *http.Request) error {
conn, err := ws.upgrader.Upgrade(w, r, nil)
if err != nil {
return err
}
for {
mt, r, err := conn.NextReader()
if err != nil {
return err
}
w, err := conn.NextWriter(mt)
if err != nil {
return err
}
if _, err := io.Copy(w, r); err != nil { // todo impl broadcast to channel;; sync map to all channels, goes to a forking reader-writer pipe, all listeners to broadcast read from pipe
return err
}
if err := w.Close(); err != nil {
return err
}
}
}