Fixed availability check for Linux Telemetry providers

Author: petehayes102

core/src/target/linux.rs

@@ -9,6 +9,47 @@ use regex::Regex;
 use std::{fs, process, str};
 use std::io::Read;
 
+#[derive(Eq, PartialEq)]
+pub enum LinuxFlavour {
+    Centos,
+    Debian,
+    Fedora,
+    Redhat,
+    Ubuntu,
+    NixOs,
+}
+
+pub fn fingerprint_os() -> Option<LinuxFlavour> {
+    // @todo Cache this result
+
+    // CentOS
+    if let Ok(_) = fs::metadata("/etc/centos-release") {
+        Some(LinuxFlavour::Centos)
+    }
+    // Ubuntu
+    else if let Ok(_) = fs::metadata("/etc/lsb-release") {
+        Some(LinuxFlavour::Ubuntu)
+    }
+    // Debian
+    else if let Ok(_) = fs::metadata("/etc/debian_version") {
+        Some(LinuxFlavour::Debian)
+    }
+    // Fedora
+    else if let Ok(_) = fs::metadata("/etc/fedora-release") {
+        Some(LinuxFlavour::Fedora)
+    }
+    // RedHat
+    else if let Ok(_) = fs::metadata("/etc/redhat-release") {
+        Some(LinuxFlavour::Redhat)
+    }
+    // NixOS
+    else if let Ok(_) = fs::metadata("/etc/nixos/configuration.nix") {
+        Some(LinuxFlavour::NixOs)
+    } else {
+        None
+    }
+}
+
 pub fn cpu_vendor() -> Result<String> {
     get_cpu_item("vendor_id")
 }

core/src/telemetry/providers/centos.rs

@@ -11,6 +11,7 @@ use host::*;
 use pnet::datalink::interfaces;
 use std::{env, str};
 use target::{default, linux, redhat};
+use target::linux::LinuxFlavour;
 use telemetry::{Cpu, Os, OsFamily, OsPlatform, Telemetry, TelemetryProvider, serializable};
 
 pub struct Centos;
@@ -36,7 +37,7 @@ impl <'de>ExecutableProvider<'de> for RemoteProvider {
 impl TelemetryProvider for Centos {
     fn available(host: &Host) -> bool {
         if host.is_local() {
-            cfg!(target_os="linux")
+            cfg!(target_os="linux") && linux::fingerprint_os() == Some(LinuxFlavour::Centos)
         } else {
             unimplemented!();
             // let r = RemoteProvider::Available;

core/src/telemetry/providers/debian.rs

@@ -11,6 +11,7 @@ use host::*;
 use pnet::datalink::interfaces;
 use std::{env, process, str};
 use target::{default, linux};
+use target::linux::LinuxFlavour;
 use telemetry::{Cpu, Os, OsFamily, OsPlatform, Telemetry, TelemetryProvider, serializable};
 
 pub struct Debian;
@@ -36,7 +37,7 @@ impl <'de>ExecutableProvider<'de> for RemoteProvider {
 impl TelemetryProvider for Debian {
     fn available(host: &Host) -> bool {
         if host.is_local() {
-            cfg!(target_os="linux")
+            cfg!(target_os="linux") && linux::fingerprint_os() == Some(LinuxFlavour::Debian)
         } else {
             unimplemented!();
             // let r = RemoteProvider::Available;