kinect_upload_fw: check the return value of get_first_reply() and get_reply()

diff --git a/kinect_upload_fw/kinect_upload_fw.c b/kinect_upload_fw/kinect_upload_fw.c
index 5faf8e1..a88d3aa 100644 (file)
--- a/kinect_upload_fw/kinect_upload_fw.c
+++ b/kinect_upload_fw/kinect_upload_fw.c
@@ -197,7 +197,16 @@ int main(int argc, char** argv) {
                goto cleanup;
        }
        res = get_first_reply(); // This first one doesn't have the usual magic bytes at the beginning, and is 96 bytes long - much longer than the usual 12-byte replies.
                goto cleanup;
        }
        res = get_first_reply(); // This first one doesn't have the usual magic bytes at the beginning, and is 96 bytes long - much longer than the usual 12-byte replies.

+       if (res < 0) {
+               LOG("get_first_reply() failed");
+               goto cleanup;
+       }
+

        res = get_reply(); // I'm not sure why we do this twice here, but maybe it'll make sense later.
        res = get_reply(); // I'm not sure why we do this twice here, but maybe it'll make sense later.

+       if (res < 0) {
+               LOG("First get_reply() failed");
+               goto cleanup;
+       }

        seq++;
 
        // Split addr declaration and assignment in order to compile as C++,
        seq++;
 
        // Split addr declaration and assignment in order to compile as C++,


@@ -238,6 +247,10 @@ int main(int argc, char** argv) {
                        bytes_sent += to_send;
                }
                res = get_reply();
                        bytes_sent += to_send;
                }
                res = get_reply();

+               if (res < 0) {
+                       LOG("get_reply failed");
+                       goto cleanup;
+               }

 
                addr += (uint32_t)read;
                seq++;
 
                addr += (uint32_t)read;
                seq++;