Flush file writes when saving config and backup files

2 files changed, 13 insertions, 3 deletions

diff --git a/talpid-core/src/firewall/linux/dns.rs b/talpid-core/src/firewall/linux/dns.rs
index 9353476d56..840ffd609e 100644
--- a/talpid-core/src/firewall/linux/dns.rs
+++ b/talpid-core/src/firewall/linux/dns.rs
@@ -1,11 +1,12 @@
 extern crate notify;
 extern crate resolv_conf;
 
+use std::io::{self, Write};
 use std::net::IpAddr;
 use std::ops::DerefMut;
 use std::path::Path;
 use std::sync::{mpsc, Arc, Mutex, MutexGuard};
-use std::{fs, io, thread};
+use std::{fs, thread};
 
 use error_chain::ChainedError;
 
@@ -106,7 +107,14 @@ impl DnsSettings {
         match fs::read(&backup_file) {
             Ok(backup) => {
                 info!("Restoring DNS state from backup");
-                fs::write(RESOLV_CONF_PATH, &backup).chain_err(|| ErrorKind::RestoreResolvConf)?;
+                let mut conf_file =
+                    fs::File::create(RESOLV_CONF_PATH).chain_err(|| ErrorKind::RestoreResolvConf)?;
+                conf_file
+                    .write_all(&backup)
+                    .chain_err(|| ErrorKind::RestoreResolvConf)?;
+                conf_file
+                    .sync_all()
+                    .chain_err(|| ErrorKind::RestoreResolvConf)?;
                 fs::remove_file(&backup_file).chain_err(|| ErrorKind::RemoveBackup)?;
             }
             Err(ref error) if error.kind() == io::ErrorKind::NotFound => {
diff --git a/talpid-core/src/firewall/windows/system_state.rs b/talpid-core/src/firewall/windows/system_state.rs
index 4695c13bc4..8c310667bb 100644
--- a/talpid-core/src/firewall/windows/system_state.rs
+++ b/talpid-core/src/firewall/windows/system_state.rs
@@ -24,7 +24,9 @@ impl SystemStateWriter {
     /// Writes a binary blob representing the system state to the backup location before any
     /// security policies are applied.
     pub fn write_backup(&self, data: &[u8]) -> io::Result<()> {
-        fs::write(&self.backup_path, &data)
+        let mut backup_file = File::create(&self.backup_path)?;
+        backup_file.write_all(data)?;
+        backup_file.sync_all()
     }
 
     pub fn read_backup(&self) -> io::Result<Option<Vec<u8>>> {