From 369d8f3ccfcf0c118843e6b13cd4ada1ea2ada69 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Miroslav=20=C5=A0ediv=C3=BD?= <sedivy.miro@gmail.com>
Date: Thu, 10 Feb 2022 00:12:30 +0100
Subject: [PATCH] peer webrtc connection check.

---
 internal/types/webrtc.go |  1 +
 internal/webrtc/peer.go  | 26 ++++++++++++++++++++++++++
 2 files changed, 27 insertions(+)

diff --git a/internal/types/webrtc.go b/internal/types/webrtc.go
index ce53b003..54a7c6e6 100644
--- a/internal/types/webrtc.go
+++ b/internal/types/webrtc.go
@@ -9,6 +9,7 @@ import (
 var (
 	ErrWebRTCVideoNotFound       = errors.New("webrtc video not found")
 	ErrWebRTCDataChannelNotFound = errors.New("webrtc data channel not found")
+	ErrWebRTCConnectionNotFound  = errors.New("webrtc connection not found")
 )
 
 type ICEServer struct {
diff --git a/internal/webrtc/peer.go b/internal/webrtc/peer.go
index 97be3cb2..b338204d 100644
--- a/internal/webrtc/peer.go
+++ b/internal/webrtc/peer.go
@@ -5,6 +5,8 @@ import (
 
 	"github.com/pion/webrtc/v3"
 	"github.com/rs/zerolog"
+
+	"demodesk/neko/internal/types"
 )
 
 type WebRTCPeerCtx struct {
@@ -20,6 +22,10 @@ func (peer *WebRTCPeerCtx) CreateOffer(ICERestart bool) (*webrtc.SessionDescript
 	peer.mu.Lock()
 	defer peer.mu.Unlock()
 
+	if peer.connection == nil {
+		return nil, types.ErrWebRTCConnectionNotFound
+	}
+
 	offer, err := peer.connection.CreateOffer(&webrtc.OfferOptions{
 		ICERestart: ICERestart,
 	})
@@ -34,6 +40,10 @@ func (peer *WebRTCPeerCtx) CreateAnswer() (*webrtc.SessionDescription, error) {
 	peer.mu.Lock()
 	defer peer.mu.Unlock()
 
+	if peer.connection == nil {
+		return nil, types.ErrWebRTCConnectionNotFound
+	}
+
 	answer, err := peer.connection.CreateAnswer(nil)
 	if err != nil {
 		return nil, err
@@ -65,6 +75,10 @@ func (peer *WebRTCPeerCtx) SetOffer(sdp string) error {
 	peer.mu.Lock()
 	defer peer.mu.Unlock()
 
+	if peer.connection == nil {
+		return types.ErrWebRTCConnectionNotFound
+	}
+
 	return peer.connection.SetRemoteDescription(webrtc.SessionDescription{
 		SDP:  sdp,
 		Type: webrtc.SDPTypeOffer,
@@ -75,6 +89,10 @@ func (peer *WebRTCPeerCtx) SetAnswer(sdp string) error {
 	peer.mu.Lock()
 	defer peer.mu.Unlock()
 
+	if peer.connection == nil {
+		return types.ErrWebRTCConnectionNotFound
+	}
+
 	return peer.connection.SetRemoteDescription(webrtc.SessionDescription{
 		SDP:  sdp,
 		Type: webrtc.SDPTypeAnswer,
@@ -85,6 +103,10 @@ func (peer *WebRTCPeerCtx) SetCandidate(candidate webrtc.ICECandidateInit) error
 	peer.mu.Lock()
 	defer peer.mu.Unlock()
 
+	if peer.connection == nil {
+		return types.ErrWebRTCConnectionNotFound
+	}
+
 	return peer.connection.AddICECandidate(candidate)
 }
 
@@ -92,6 +114,10 @@ func (peer *WebRTCPeerCtx) SetVideoID(videoID string) error {
 	peer.mu.Lock()
 	defer peer.mu.Unlock()
 
+	if peer.connection == nil {
+		return types.ErrWebRTCConnectionNotFound
+	}
+
 	peer.logger.Info().Str("video_id", videoID).Msg("change video id")
 	return peer.changeVideo(videoID)
 }