diff --git a/src/webrtc/pkg/session/receiver/receiver.go b/src/webrtc/pkg/session/receiver/receiver.go index 7d8de2cd..7f1d953a 100644 --- a/src/webrtc/pkg/session/receiver/receiver.go +++ b/src/webrtc/pkg/session/receiver/receiver.go @@ -138,12 +138,12 @@ func (s *Session) CreateDataHandler() { }) } -func (s *Session) ReceiveData(pathToFile string, fileSize int64) { - s.receiveData(pathToFile, fileSize) +func (s *Session) ReceiveData(pathToFile string, fileSize int64, sharedSecret []byte) { + s.receiveData(pathToFile, fileSize, sharedSecret) s.sess.OnCompletion() } -func (s *Session) receiveData(pathToFile string, fileSize int64) error { +func (s *Session) receiveData(pathToFile string, fileSize int64, sharedSecret []byte) error { log.Debugln("Starting to receive data...") log.Debugf("receiving %s", pathToFile) @@ -195,7 +195,7 @@ func (s *Session) receiveData(pathToFile string, fileSize int64) error { log.Debug("closed gracefully") return nil case msg := <-s.msgChannel: - buff, errDecrypt := crypt.DecryptFromBytes(msg.Data, []byte{1, 2, 3, 4}) + buff, errDecrypt := crypt.DecryptFromBytes(msg.Data, sharedSecret) if errDecrypt != nil { log.Error(errDecrypt) return errDecrypt diff --git a/src/webrtc/pkg/session/sender/sender.go b/src/webrtc/pkg/session/sender/sender.go index 02b1ab75..1dbfcf3b 100644 --- a/src/webrtc/pkg/session/sender/sender.go +++ b/src/webrtc/pkg/session/sender/sender.go @@ -117,8 +117,8 @@ func (s *Session) SetSDP(sdp string) error { return s.sess.SetSDP(sdp) } -func (s *Session) TransferFile(pathToFile string) { - s.readFile(pathToFile) +func (s *Session) TransferFile(pathToFile string, sharedSecret []byte) { + s.readFile(pathToFile, sharedSecret) s.sess.OnCompletion() } @@ -186,7 +186,7 @@ func (s *Session) CreateDataChannel() error { return nil } -func (s *Session) readFile(pathToFile string) error { +func (s *Session) readFile(pathToFile string, sharedSecret []byte) error { f, err := os.Open(pathToFile) if err != nil { log.Error(err) @@ -232,7 +232,7 @@ func (s *Session) readFile(pathToFile string) error { } buff = append(buff, s.dataBuff[:n-i]...) buff = compress.Compress(buff) - buff = crypt.EncryptToBytes(buff, []byte{1, 2, 3, 4}) + buff = crypt.EncryptToBytes(buff, sharedSecret) if len(buff) < maxPacketSize { if n-i > 0 { lastBytes = append([]byte(nil), s.dataBuff[n-i:]...)