Start capturing screen when connection established

src/main_window/main_window.cpp

@@ -225,11 +225,10 @@ int MainWindow::Run() {
                 }
               });
 
-          if (0 == screen_capturer_init_ret) {
-            screen_capturer_->Start();
-          } else {
+          if (0 != screen_capturer_init_ret) {
             screen_capturer_->Destroy();
             screen_capturer_ = nullptr;
+            LOG_ERROR("Create screen capturer failed");
           }
 
           // Mouse control

src/main_window/main_window_callback_func.cpp

@@ -153,12 +153,14 @@ void MainWindow::OnConnectionStatusCb(ConnectionStatus status,
   } else if (ConnectionStatus::Connected == status) {
     main_window->connection_status_str_ = "Connected";
     main_window->connection_established_ = true;
+    main_window->screen_capturer_->Start();
   } else if (ConnectionStatus::Disconnected == status) {
     main_window->connection_status_str_ = "Disconnected";
   } else if (ConnectionStatus::Failed == status) {
     main_window->connection_status_str_ = "Failed";
   } else if (ConnectionStatus::Closed == status) {
     main_window->connection_status_str_ = "Closed";
+    main_window->screen_capturer_->Stop();
   } else if (ConnectionStatus::IncorrectPassword == status) {
     main_window->connection_status_str_ = "Incorrect password";
     if (main_window->connect_button_pressed_) {

src/screen_capturer/linux/screen_capturer_x11.cpp

@@ -133,12 +133,12 @@ int ScreenCapturerX11::Start() {
   return 0;
 }
 
+int ScreenCapturerX11::Stop() { return 0; }
+
 int ScreenCapturerX11::Pause() { return 0; }
 
 int ScreenCapturerX11::Resume() { return 0; }
 
-int ScreenCapturerX11::Stop() { return 0; }
-
 void ScreenCapturerX11::OnFrame() {}
 
 void ScreenCapturerX11::CleanUp() {}

src/screen_capturer/linux/screen_capturer_x11.h

@@ -32,9 +32,10 @@ class ScreenCapturerX11 : public ScreenCapturer {
 
   virtual int Start();
 
+  virtual int Stop();
+
   int Pause();
   int Resume();
-  int Stop();
 
   void OnFrame();
 

src/screen_capturer/macosx/screen_capturer_avf.cpp

@@ -134,12 +134,12 @@ int ScreenCapturerAvf::Start() {
   return 0;
 }
 
+int ScreenCapturerAvf::Stop() { return 0; }
+
 int ScreenCapturerAvf::Pause() { return 0; }
 
 int ScreenCapturerAvf::Resume() { return 0; }
 
-int ScreenCapturerAvf::Stop() { return 0; }
-
 void ScreenCapturerAvf::OnFrame() {}
 
 void ScreenCapturerAvf::CleanUp() {}

src/screen_capturer/macosx/screen_capturer_avf.h

@@ -38,9 +38,10 @@ class ScreenCapturerAvf : public ScreenCapturer {
 
   virtual int Start();
 
+  virtual int Stop();
+
   int Pause();
   int Resume();
-  int Stop();
 
   void OnFrame();
 

src/screen_capturer/screen_capturer.h

@@ -28,6 +28,8 @@ class ScreenCapturer {
   virtual int Destroy() = 0;
 
   virtual int Start() = 0;
+
+  virtual int Stop() = 0;
 };
 
 #endif

src/screen_capturer/windows/screen_capturer_wgc.h

@@ -27,7 +27,7 @@ class ScreenCapturerWgc : public ScreenCapturer,
 
   int Pause();
   int Resume();
-  int Stop();
+  virtual int Stop();
 
   void OnFrame(const WgcSession::wgc_session_frame &frame);