Check return value of gstreamer_init

awslabs · Aug 22, 2023 · 8df715a · 8df715a
1 parent 7842b4c
commit 8df715a
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 4 deletions.
diff --git a/samples/kvs_gstreamer_sample.cpp b/samples/kvs_gstreamer_sample.cpp
@@ -1158,7 +1158,10 @@ int main(int argc, char* argv[]) {
                 LOG_DEBUG("Attempt to upload file: " << data.file_list[i].path);
 
                 // control will return after gstreamer_init after file eos or any GST_ERROR was put on the bus.
-                gstreamer_init(argc, argv, &data);
+                if (gstreamer_init(argc, argv, &data) != 0) {
+                    LOG_ERROR("Failed to initialize gstreamer");
+                    return 1;
+                }
 
                 // check if any stream error occurred.
                 stream_status = data.stream_status.load();
@@ -1208,7 +1211,10 @@ int main(int argc, char* argv[]) {
 
     } else {
         // non file uploading scenario
-        gstreamer_init(argc, argv, &data);
+        if (gstreamer_init(argc, argv, &data) != 0) {
+            LOG_ERROR("Failed to initialize gstreamer");
+            return 1;
+        }
         if (STATUS_SUCCEEDED(stream_status)) {
             // if stream_status is success after eos, send out remaining frames.
             data.kinesis_video_stream->stopSync();

diff --git a/samples/kvssink_gstreamer_sample.cpp b/samples/kvssink_gstreamer_sample.cpp
@@ -816,7 +816,9 @@ int main(int argc, char *argv[]) {
                 LOG_DEBUG("Attempt to upload file: " << data_global.file_list[i].path);
 
                 // control will return after gstreamer_init after file eos or any GST_ERROR was put on the bus.
-                gstreamer_init(argc, argv, &data_global);
+                if (gstreamer_init(argc, argv, &data_global) != 0) {
+                    return 1;
+                }
 
                 // check if any stream error occurred.
                 stream_status = data_global.stream_status.load();
@@ -860,7 +862,9 @@ int main(int argc, char *argv[]) {
 
     } else {
         // non file uploading scenario
-        gstreamer_init(argc, argv, &data_global);
+        if (gstreamer_init(argc, argv, &data_global) != 0) {
+            return 1;
+        }
 	stream_status = data_global.stream_status.load();
         if (STATUS_SUCCEEDED(stream_status)) {
             LOG_INFO("Stream succeeded");