Be a bit more careful with fwrite.
authorCarl Hetherington <cth@carlh.net>
Sun, 23 Dec 2018 21:38:44 +0000 (21:38 +0000)
committerCarl Hetherington <cth@carlh.net>
Sun, 23 Dec 2018 21:38:44 +0000 (21:38 +0000)
src/lib/config.cc
src/lib/internet.cc
src/lib/reel_writer.cc
src/lib/util.cc
src/lib/util.h
src/lib/writer.cc
src/wx/config_dialog.cc
src/wx/swaroop_controls.cc

index 172890d..eb26713 100644 (file)
@@ -987,11 +987,7 @@ Config::write_config () const
                if (!f) {
                        throw FileError (_("Could not open file for writing"), cf);
                }
-               size_t const w = fwrite (s.c_str(), 1, s.length(), f);
-               if (w != s.length()) {
-                       fclose (f);
-                       throw FileError (_("Could not write whole file"), cf);
-               }
+               checked_fwrite (s.c_str(), s.length(), f, cf);
                fclose (f);
        } catch (xmlpp::exception& e) {
                string s = e.what ();
index ad313bd..4eba1ef 100644 (file)
@@ -22,6 +22,7 @@
 #include "compose.hpp"
 #include "exceptions.h"
 #include "cross.h"
+#include "util.h"
 #include <curl/curl.h>
 #include <zip.h>
 #include <boost/function.hpp>
@@ -140,7 +141,7 @@ get_from_zip_url (string url, string file, bool pasv, function<void (boost::file
        char buffer[4096];
        while (true) {
                int const N = zip_fread (file_in_zip, buffer, sizeof (buffer));
-               fwrite (buffer, 1, N, f);
+               checked_fwrite (buffer, N, f, temp_cert.file());
                if (N < int (sizeof (buffer))) {
                        break;
                }
index f645d6e..e34874a 100644 (file)
@@ -143,9 +143,9 @@ ReelWriter::write_frame_info (Frame frame, Eyes eyes, dcp::FrameInfo info) const
                throw OpenFileError (info_file, errno, read);
        }
        dcpomatic_fseek (file, frame_info_position (frame, eyes), SEEK_SET);
-       fwrite (&info.offset, sizeof (info.offset), 1, file);
-       fwrite (&info.size, sizeof (info.size), 1, file);
-       fwrite (info.hash.c_str(), 1, info.hash.size(), file);
+       checked_fwrite (&info.offset, sizeof (info.offset), file, info_file);
+       checked_fwrite (&info.size, sizeof (info.size), file, info_file);
+       checked_fwrite (info.hash.c_str(), info.hash.size(), file, info_file);
        fclose (file);
 }
 
index 5eba8d7..595c7e7 100644 (file)
@@ -784,13 +784,30 @@ increment_eyes (Eyes e)
 }
 
 void
+checked_fwrite (void const * ptr, size_t size, FILE* stream, boost::filesystem::path path)
+{
+       size_t N = fwrite (ptr, 1, size, stream);
+       if (N != size) {
+               if (ferror(stream)) {
+                       fclose (stream);
+                       throw FileError (String::compose("fwrite error %1", errno), path);
+               } else {
+                       fclose (stream);
+                       throw FileError ("Unexpected short write", path);
+               }
+       }
+}
+
+void
 checked_fread (void* ptr, size_t size, FILE* stream, boost::filesystem::path path)
 {
        size_t N = fread (ptr, 1, size, stream);
        if (N != size) {
                if (ferror(stream)) {
+                       fclose (stream);
                        throw FileError (String::compose("fread error %1", errno), path);
                } else {
+                       fclose (stream);
                        throw FileError ("Unexpected short read", path);
                }
        }
index f62b2c4..94e9e37 100644 (file)
@@ -99,5 +99,6 @@ extern std::pair<int, int> audio_channel_types (std::list<int> mapped, int chann
 extern boost::shared_ptr<AudioBuffers> remap (boost::shared_ptr<const AudioBuffers> input, int output_channels, AudioMapping map);
 extern Eyes increment_eyes (Eyes e);
 extern void checked_fread (void* ptr, size_t size, FILE* stream, boost::filesystem::path path);
+extern void checked_fwrite (void const * ptr, size_t size, FILE* stream, boost::filesystem::path path);
 
 #endif
index c31ae2a..45a7462 100644 (file)
@@ -634,7 +634,7 @@ Writer::write_cover_sheet ()
 
        boost::algorithm::replace_all (text, "$LENGTH", length);
 
-       fwrite (text.c_str(), 1, text.length(), f);
+       checked_fwrite (text.c_str(), text.length(), f, cover);
        fclose (f);
 }
 
index ccf1202..8b6d215 100644 (file)
@@ -536,7 +536,7 @@ CertificateChainEditor::export_certificate ()
                }
 
                string const s = j->certificate (true);
-               fwrite (s.c_str(), 1, s.length(), f);
+               checked_fwrite (s.c_str(), s.length(), f, path);
                fclose (f);
        }
        d->Destroy ();
@@ -709,7 +709,7 @@ CertificateChainEditor::export_private_key ()
                }
 
                string const s = _get()->key().get ();
-               fwrite (s.c_str(), 1, s.length(), f);
+               checked_fwrite (s.c_str(), s.length(), f, path);
                fclose (f);
        }
        d->Destroy ();
@@ -805,10 +805,10 @@ KeysPage::export_decryption_chain_and_key ()
                }
 
                string const chain = Config::instance()->decryption_chain()->chain();
-               fwrite (chain.c_str(), 1, chain.length(), f);
+               checked_fwrite (chain.c_str(), chain.length(), f, path);
                optional<string> const key = Config::instance()->decryption_chain()->key();
                DCPOMATIC_ASSERT (key);
-               fwrite (key->c_str(), 1, key->length(), f);
+               checked_fwrite (key->c_str(), key->length(), f, path);
                fclose (f);
        }
        d->Destroy ();
@@ -883,7 +883,7 @@ KeysPage::export_decryption_chain ()
                }
 
                string const s = Config::instance()->decryption_chain()->chain();
-               fwrite (s.c_str(), 1, s.length(), f);
+               checked_fwrite (s.c_str(), s.length(), f, path);
                fclose (f);
        }
        d->Destroy ();
@@ -905,7 +905,7 @@ KeysPage::export_decryption_certificate ()
                }
 
                string const s = Config::instance()->decryption_chain()->leaf().certificate (true);
-               fwrite (s.c_str(), 1, s.length(), f);
+               checked_fwrite (s.c_str(), s.length(), f, path);
                fclose (f);
        }
 
index 36aac7c..d6ab5c3 100644 (file)
@@ -158,7 +158,7 @@ SwaroopControls::viewer_position_changed ()
                        + " " + dcp::raw_convert<string>(_selected_playlist_position)
                        + " " + dcp::raw_convert<string>(_viewer->position().get());
 
-               fwrite (p.c_str(), p.length(), 1, f);
+               checked_fwrite (p.c_str(), p.length(), f, Config::path("position"));
                fclose (f);
        }
 }