From 4543f1c5484feab960151997e2e59431191c9d41 Mon Sep 17 00:00:00 2001
From: Peter Barker <pbarker@barker.dropbear.id.au>
Date: Wed, 16 Nov 2016 13:18:34 +1100
Subject: [PATCH] Replay: check return value from fopen() of output files

---
 Tools/Replay/Replay.cpp | 22 ++++++++++++++++------
 Tools/Replay/Replay.h   |  2 ++
 2 files changed, 18 insertions(+), 6 deletions(-)

diff --git a/Tools/Replay/Replay.cpp b/Tools/Replay/Replay.cpp
index 01dfba63cf..c9499cb53d 100644
--- a/Tools/Replay/Replay.cpp
+++ b/Tools/Replay/Replay.cpp
@@ -482,6 +482,16 @@ static void _replay_sig_fpe(int signum)
     abort();
 }
 
+FILE *Replay::xfopen(const char *filename, const char *mode)
+{
+    FILE *ret = fopen(filename, mode);
+    if (ret == nullptr) {
+        ::fprintf(stderr, "Failed to open (%s): %m\n", filename);
+        abort();
+    }
+    return ret;
+}
+
 void Replay::setup()
 {
     ::printf("Starting\n");
@@ -532,12 +542,12 @@ void Replay::setup()
     
     set_ins_update_rate(log_info.update_rate);
 
-    plotf = fopen("plot.dat", "w");
-    plotf2 = fopen("plot2.dat", "w");
-    ekf1f = fopen("EKF1.dat", "w");
-    ekf2f = fopen("EKF2.dat", "w");
-    ekf3f = fopen("EKF3.dat", "w");
-    ekf4f = fopen("EKF4.dat", "w");
+    plotf = xfopen("plot.dat", "w");
+    plotf2 = xfopen("plot2.dat", "w");
+    ekf1f = xfopen("EKF1.dat", "w");
+    ekf2f = xfopen("EKF2.dat", "w");
+    ekf3f = xfopen("EKF3.dat", "w");
+    ekf4f = xfopen("EKF4.dat", "w");
 
     fprintf(plotf, "time SIM.Roll SIM.Pitch SIM.Yaw BAR.Alt FLIGHT.Roll FLIGHT.Pitch FLIGHT.Yaw FLIGHT.dN FLIGHT.dE AHR2.Roll AHR2.Pitch AHR2.Yaw DCM.Roll DCM.Pitch DCM.Yaw EKF.Roll EKF.Pitch EKF.Yaw INAV.dN INAV.dE INAV.Alt EKF.dN EKF.dE EKF.Alt\n");
     fprintf(plotf2, "time E1 E2 E3 VN VE VD PN PE PD GX GY GZ WN WE MN ME MD MX MY MZ E1ref E2ref E3ref\n");
diff --git a/Tools/Replay/Replay.h b/Tools/Replay/Replay.h
index fd2b937740..29f49c2411 100644
--- a/Tools/Replay/Replay.h
+++ b/Tools/Replay/Replay.h
@@ -177,6 +177,8 @@ private:
     void load_param_file(const char *filename);
     void set_signal_handlers(void);
     void flush_and_exit();
+
+    FILE *xfopen(const char *f, const char *mode);
 };
 
 enum {