fix: keep plugins working after client-only updates (#4077)

2 files changed, 25 insertions, 0 deletions

diff --git a/src/platform/linux.rs b/src/platform/linux.rs
index 7b0755e1..d563512d 100644
--- a/src/platform/linux.rs
+++ b/src/platform/linux.rs
@@ -82,6 +82,26 @@ struct ProcGroupMember {
     state: char,
 }
 
+pub(crate) fn launch_executable() -> std::io::Result<PathBuf> {
+    use std::os::unix::ffi::OsStrExt;
+
+    let executable = std::env::current_exe()?;
+    if !executable.is_file() {
+        // Linux marks the old inode as deleted after an update replaces the binary.
+        if let Some(path) = executable
+            .as_os_str()
+            .as_bytes()
+            .strip_suffix(b" (deleted)")
+        {
+            let replacement = PathBuf::from(std::ffi::OsStr::from_bytes(path));
+            if replacement.is_file() {
+                return Ok(replacement);
+            }
+        }
+    }
+    Ok(executable)
+}
+
 pub fn raise_server_nofile_limit() {}
 
 pub(crate) fn should_draw_host_cursor_by_default() -> bool {
diff --git a/src/platform/mod.rs b/src/platform/mod.rs
index 67e474ff..3efad45e 100644
--- a/src/platform/mod.rs
+++ b/src/platform/mod.rs
@@ -55,6 +55,11 @@ pub(crate) fn classify_child_exit(_status: &portable_pty::ExitStatus) -> ChildEx
     ChildExitReason::Exited
 }
 
+#[cfg(not(target_os = "linux"))]
+pub(crate) fn launch_executable() -> std::io::Result<std::path::PathBuf> {
+    std::env::current_exe()
+}
+
 pub(crate) fn detached_custom_command_process(command: &str) -> std::process::Command {
     let mut process = detached_custom_command_process_platform(command);
     configure_background_command(&mut process);