Update management.py (#92)

Fix bug with windows binary path management
osquery · Jul 23, 2024 · 15c1d09 · 15c1d09
1 parent 705e5b9
commit 15c1d09
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/osquery/management.py b/osquery/management.py
@@ -42,10 +42,12 @@ def emit(self, record):
     # We bootleg our own version of Windows pipe coms
     from osquery.TPipe import TPipe
     from osquery.TPipe import TPipeServer
-    if os.path.exists(os.environ["PROGRAMDATA"] + "\\osquery\\osqueryd\\osqueryd.exe"):
-        WINDOWS_BINARY_PATH = os.environ["PROGRAMDATA"] + "\\osquery\\osqueryd\\osqueryd.exe"
-    if os.path.exists(os.environ["PROGRAMW6432"] + "\\osquery\\osqueryd\\osqueryd.exe"):
-        WINDOWS_BINARY_PATH = os.environ["PROGRAMW6432"] + "\\osquery\\osqueryd\\osqueryd.exe"
+    PTH = os.path.join(os.environ["PROGRAMDATA"], "osquery", "osqueryd", "osqueryd.exe")
+    if os.path.exists(PTH):
+        WINDOWS_BINARY_PATH = PTH
+    PTH = os.path.join(os.environ["PROGRAMW6432"], "osquery", "osqueryd", "osqueryd.exe")
+    if os.path.exists(PTH):
+        WINDOWS_BINARY_PATH = PTH
 
 DARWIN_BINARY_PATH = "/usr/local/bin/osqueryd"
 LINUX_BINARY_PATH = "/usr/bin/osqueryd"