diff --git a/lix/libstore/local-store.cc b/lix/libstore/local-store.cc index e0334c558..a78956fe5 100644 --- a/lix/libstore/local-store.cc +++ b/lix/libstore/local-store.cc @@ -169,7 +169,7 @@ void migrateCASchema(SQLite& db, Path schemaPath, AutoCloseFD& lockFd) txn.commit(); } - writeFile(schemaPath, fmt("%d", nixCASchemaVersion), 0666, true); + writeFileAndSync(schemaPath, fmt("%d", nixCASchemaVersion), 0666); lockFile(lockFd.get(), ltRead); } } @@ -303,7 +303,7 @@ LocalStore::LocalStore(LocalStoreConfig config) else if (curSchema == 0) { /* new store */ curSchema = nixSchemaVersion; openDB(*state, true); - writeFile(schemaPath, fmt("%1%", nixSchemaVersion), 0666, true); + writeFileAndSync(schemaPath, fmt("%1%", nixSchemaVersion), 0666); } else if (curSchema < nixSchemaVersion) { @@ -352,7 +352,7 @@ LocalStore::LocalStore(LocalStoreConfig config) txn.commit(); } - writeFile(schemaPath, fmt("%1%", nixSchemaVersion), 0666, true); + writeFileAndSync(schemaPath, fmt("%1%", nixSchemaVersion), 0666); lockFile(globalLock.get(), ltRead); } diff --git a/lix/libutil/file-system.cc b/lix/libutil/file-system.cc index 9322e96c1..4d9c3076e 100644 --- a/lix/libutil/file-system.cc +++ b/lix/libutil/file-system.cc @@ -354,17 +354,47 @@ Generator readFileSource(const Path & path) } -void writeFile(const Path & path, std::string_view s, mode_t mode, bool sync) +void writeFile(const Path & path, std::string_view s, mode_t mode) { AutoCloseFD fd{open(path.c_str(), O_WRONLY | O_TRUNC | O_CREAT | O_CLOEXEC, mode)}; if (!fd) throw SysError("opening file '%1%'", path); + + writeFile(fd, s, mode); + + /* Close explicitly to propagate the exceptions. */ + fd.close(); +} + +void writeFile(AutoCloseFD & fd, std::string_view s, mode_t mode) +{ + assert(fd); try { writeFull(fd.get(), s); } catch (Error & e) { - e.addTrace({}, "writing file '%1%'", path); + e.addTrace({}, "writing file '%1%'", fd.guessOrInventPath()); throw; } +} + +void writeFileAndSync(const Path & path, std::string_view s, mode_t mode) +{ + { + AutoCloseFD fd{open(path.c_str(), O_WRONLY | O_TRUNC | O_CREAT | O_CLOEXEC, mode)}; + if (!fd) + throw SysError("opening file '%1%'", path); + + writeFile(fd, s, mode); + fd.fsync(); + /* Close explicitly to ensure that exceptions are propagated. */ + fd.close(); + } + + syncParent(path); +} + +static void closeForWrite(const Path & path, AutoCloseFD & fd, bool sync) +{ if (sync) fd.fsync(); // Explicitly close to make sure exceptions are propagated. @@ -373,8 +403,7 @@ void writeFile(const Path & path, std::string_view s, mode_t mode, bool sync) syncParent(path); } - -void writeFile(const Path & path, Source & source, mode_t mode, bool sync) +void writeFile(const Path & path, Source & source, mode_t mode) { AutoCloseFD fd{open(path.c_str(), O_WRONLY | O_TRUNC | O_CREAT | O_CLOEXEC, mode)}; if (!fd) @@ -393,12 +422,8 @@ void writeFile(const Path & path, Source & source, mode_t mode, bool sync) e.addTrace({}, "writing file '%1%'", path); throw; } - if (sync) - fd.fsync(); - // Explicitly close to make sure exceptions are propagated. - fd.close(); - if (sync) - syncParent(path); + + closeForWrite(path, fd, false); } void syncParent(const Path & path) diff --git a/lix/libutil/file-system.hh b/lix/libutil/file-system.hh index d878d5fc3..9674b6289 100644 --- a/lix/libutil/file-system.hh +++ b/lix/libutil/file-system.hh @@ -188,9 +188,16 @@ Generator readFileSource(const Path & path); /** * Write a string to a file. */ -void writeFile(const Path & path, std::string_view s, mode_t mode = 0666, bool sync = false); +void writeFile(const Path & path, std::string_view s, mode_t mode = 0666); -void writeFile(const Path & path, Source & source, mode_t mode = 0666, bool sync = false); +void writeFile(const Path & path, Source & source, mode_t mode = 0666); + +void writeFile(AutoCloseFD & fd, std::string_view s, mode_t mode = 0666); + +/** + * Write a string to a file and flush the file and its parents direcotry to disk. + */ +void writeFileAndSync(const Path & path, std::string_view s, mode_t mode = 0666); /** * Flush a file's parent directory to disk