From d7be8e9f967a65992f43d08d1767098715e46c68 Mon Sep 17 00:00:00 2001
From: Tarald <taraldv@gmail.com>
Date: Thu, 13 Aug 2020 13:44:23 +0200
Subject: [PATCH] fixed QIODevice with waitforbytes

---
 handlers/audiohandler.cpp        | 15 ++++++++-------
 handlers/outputstreamhandler.cpp |  6 +++---
 handlers/tcpsockethandler.cpp    |  3 ++-
 handlers/videohandler.cpp        |  2 +-
 4 files changed, 14 insertions(+), 12 deletions(-)

diff --git a/handlers/audiohandler.cpp b/handlers/audiohandler.cpp
index 4ef3235..8669db4 100644
--- a/handlers/audiohandler.cpp
+++ b/handlers/audiohandler.cpp
@@ -818,7 +818,7 @@ int AudioHandler::encodeAudioFrame(AVFrame *frame,int *data_present)
     if(error<0){
         char* errbuff = (char *)malloc((1000)*sizeof(char));
         av_strerror(error,errbuff,1000);
-        qDebug() << "Failed playbackhandler av_buffersrc_add_frame_flags: code "<<error<< " meaning: " << errbuff;
+        qDebug() << "Failed playbackhandler av_buffersrc_add_frame_flags: code " <<error<< " meaning: " << errbuff;
         exit(1);
     }
 
@@ -827,7 +827,7 @@ int AudioHandler::encodeAudioFrame(AVFrame *frame,int *data_present)
     if(error<0){
         char* errbuff = (char *)malloc((1000)*sizeof(char));
         av_strerror(error,errbuff,1000);
-        qDebug() << "Failed playbackhandler av_buffersink_get_frame: code "<<error<< " meaning: " << errbuff;
+        qDebug() << "Failed playbackhandler av_buffersink_get_frame: code " <<error<< " meaning: " << errbuff;
         exit(1);
     }
 
@@ -853,9 +853,8 @@ int AudioHandler::encodeAudioFrame(AVFrame *frame,int *data_present)
 
 
     /* Packet used for temporary storage. */
-    AVPacket output_packet;
+    AVPacket* output_packet = av_packet_alloc();
 
-    initPacket(&output_packet);
 
     /* Set a timestamp based on the sample rate for the container. */
     if(first)
@@ -887,7 +886,7 @@ int AudioHandler::encodeAudioFrame(AVFrame *frame,int *data_present)
     }
 
     /* Receive one encoded frame from the encoder. */
-    error = avcodec_receive_packet(mOutputCodecContext, &output_packet);
+    error = avcodec_receive_packet(mOutputCodecContext, output_packet);
     /* If the encoder asks for more data to be able to provide an
      * encoded frame, return indicating that no data is present. */
     if (error == AVERROR(EAGAIN))
@@ -915,7 +914,7 @@ int AudioHandler::encodeAudioFrame(AVFrame *frame,int *data_present)
     /* Write one audio frame from the temporary packet to the output file. */
     mWriteLock->lock();
     if (*data_present &&
-            (error = av_interleaved_write_frame(mOutputFormatContext, &output_packet)) < 0)
+            (error = av_interleaved_write_frame(mOutputFormatContext, output_packet)) < 0)
     {
         fprintf(stderr, "Could not write audio frame");
         mWriteLock->unlock();
@@ -923,9 +922,11 @@ int AudioHandler::encodeAudioFrame(AVFrame *frame,int *data_present)
     }
     //qDebug() << error;
     mWriteLock->unlock();
+    av_packet_free(&output_packet);
 
 cleanup:
-    av_packet_unref(&output_packet);
+    av_packet_unref(output_packet);
+    av_packet_free(&output_packet);
     return error;
 }
 
diff --git a/handlers/outputstreamhandler.cpp b/handlers/outputstreamhandler.cpp
index a6acaf2..cfc8e72 100644
--- a/handlers/outputstreamhandler.cpp
+++ b/handlers/outputstreamhandler.cpp
@@ -144,14 +144,14 @@ int OutputStreamHandler::enableVideo(bool screenShare)
     }
 
     const int error = mVideoHandler->init();
-    mTcpSocketHandler->writeHeader();
-    /*if(error < 0)
+    //mTcpSocketHandler->writeHeader();
+    if(error < 0)
     {
         fprintf(stderr, "Could not init videohandler");
         errorHandler->giveErrorDialog("Could not stream video");
         qDebug() << "Error: " << error;
         return (int)error;
-    }*/
+    }
 
     mVideoHandler->toggleGrabFrames(mVideoEnabled);
     qDebug() << "Before grab frames";
diff --git a/handlers/tcpsockethandler.cpp b/handlers/tcpsockethandler.cpp
index 5792b29..bbe5816 100644
--- a/handlers/tcpsockethandler.cpp
+++ b/handlers/tcpsockethandler.cpp
@@ -182,6 +182,7 @@ void TcpSocketHandler::writeHeader()
     qDebug() << "My Header: " << mHeader.length() << "\n" << mHeader;
 
     mSocket->write(mHeader);
+    qDebug() << "after write";
     mHeader.clear();
 }
 
@@ -209,7 +210,7 @@ void TcpSocketHandler::sendDisabledVideoSignal()
     qDebug() << "My Header: " << header.length() << "\n" << header;
 
     mSocket->write(header);
-    header.clear();
+    mSocket->waitForBytesWritten();
 }
 
 void TcpSocketHandler::sendDisabledAudioSignal()
diff --git a/handlers/videohandler.cpp b/handlers/videohandler.cpp
index dd2cd15..1283b96 100644
--- a/handlers/videohandler.cpp
+++ b/handlers/videohandler.cpp
@@ -341,7 +341,7 @@ void VideoHandler::grabFrames()
 {
     //init();
 
-    //emit callWriteHeader();
+    emit callWriteHeader();
 
     mActive = true;
     AVPacket* pkt = av_packet_alloc();