Merge pull request #44 from jamesmcm/pulseaudio

Add PulseAudio support

Cargo.toml

@@ -1,7 +1,7 @@
 [package]
 name = "vopono"
 description = "Launch applications via VPN tunnels using temporary network namespaces"
-version = "0.5.3"
+version = "0.6.0"
 authors = ["James McMurray <[email protected]>"]
 edition = "2018"
 license = "GPL-3.0-or-later"

src/main.rs

@@ -11,6 +11,7 @@ mod network_interface;
 mod openconnect;
 mod openvpn;
 mod providers;
+mod pulseaudio;
 mod shadowsocks;
 mod sync;
 mod sysctl;
@@ -21,13 +22,14 @@ mod wireguard;
 
 use list::output_list;
 use list_configs::print_configs;
-use log::LevelFilter;
+use log::{debug, LevelFilter};
 use netns::NetworkNamespace;
 use structopt::StructOpt;
 use sync::{sync_menu, synch};
 use util::clean_dead_locks;
 use util::clean_dead_namespaces;
 use util::elevate_privileges;
+use which::which;
 
 // TODO:
 // - Allow for not saving OpenVPN creds to config
@@ -49,7 +51,12 @@ fn main() -> anyhow::Result<()> {
     match app.cmd {
         args::Command::Exec(cmd) => {
             clean_dead_locks()?;
-
+            if which("pactl").is_ok() {
+                let pa = pulseaudio::get_pulseaudio_server()?;
+                std::env::set_var("PULSE_SERVER", pa);
+            } else {
+                debug!("pactl not found, will not set PULSE_SERVER");
+            }
             elevate_privileges()?;
             clean_dead_namespaces()?;
             exec::exec(cmd)?

src/netns.rs

@@ -100,8 +100,8 @@ impl NetworkNamespace {
             handle.current_dir(cdir);
         }
         let sudo_string = if user.is_some() {
-            handle.args(&["sudo", "-u", user.as_ref().unwrap()]);
-            Some(format!(" sudo -u {}", user.as_ref().unwrap()))
+            handle.args(&["sudo", "-Eu", user.as_ref().unwrap()]);
+            Some(format!(" sudo -Eu {}", user.as_ref().unwrap()))
         } else {
             None
         };

src/pulseaudio.rs

@@ -0,0 +1,23 @@
+use anyhow::anyhow;
+use log::debug;
+use regex::Regex;
+use std::process::Command;
+
+pub fn get_pulseaudio_server() -> anyhow::Result<String> {
+    let output = Command::new("pactl").args(&["info"]).output()?.stdout;
+    let re = Regex::new(r"Server String: ([^\n]+)").unwrap();
+    let output = std::str::from_utf8(&output)?;
+
+    let caps = re.captures(output);
+    if caps.is_none() {
+        return Err(anyhow!("Could not parse pactl output!"));
+    }
+    let caps = caps.unwrap().get(1);
+    if caps.is_none() {
+        return Err(anyhow!("Could not parse pactl output!"));
+    }
+
+    let out = caps.unwrap().as_str().to_string();
+    debug!("Setting PULSE_SERVER to {}", out);
+    Ok(out)
+}