Partially working auto installer.

Author: AngriestSCV

CMakeLists.txt

@@ -4,6 +4,7 @@ project (OpenVR-SpaceCalibrator VERSION 1.0)
 add_definitions(-D__linux__)
 
 set(CMAKE_BUILD_TYPE Debug)
+set(DRIVER_LOG_FILE "/tmp/spaceCalDriverLog.txt" CACHE PATH "Driver Log Path" )
 
 add_library(imgui 
 	lib/imgui/imgui.cpp
@@ -63,14 +64,25 @@ set_target_properties(driver_01spacecalibrator PROPERTIES
     PREFIX ""
 )
 
+#######################################################################################
+set( APP_MANIFEST_PATH ${CMAKE_INSTALL_PREFIX}/share/openvr-spacecalibrator/manifest.vrmanifest )
+set( DRIVER_INSTALLER_PATH ${CMAKE_INSTALL_PREFIX}/share/openvr-spacecalibrator )
+set( DRIVER_MANIFEST_PATH ${CMAKE_INSTALL_PREFIX}/lib/steamvr/OpenVR-SpaceCalibrator/01spacecalibrator )
+configure_file( StaticConfig.h.in StaticConfig.h )
+
+include_directories("${CMAKE_CURRENT_BINARY_DIR}")
 
 configure_file( ./OpenVR-SpaceCalibrator/manifest.vrmanifest ${CMAKE_CURRENT_BINARY_DIR}/manifest.vrmanifest )
 configure_file( ./OpenVR-SpaceCalibratorDriver/01spacecalibrator/driver.vrdrivermanifest ${CMAKE_CURRENT_BINARY_DIR}/driver.vrdrivermanifest )
+configure_file( ./driverInstall.py ${CMAKE_CURRENT_BINARY_DIR}/driverInstall.py )
+
 
+#######################################################################################
 install(
     FILES ${CMAKE_CURRENT_BINARY_DIR}/manifest.vrmanifest
     DESTINATION ${CMAKE_INSTALL_PREFIX}/share/openvr-spacecalibrator
 )
+
 install(
     FILES ${CMAKE_CURRENT_BINARY_DIR}/driver.vrdrivermanifest
     DESTINATION lib/steamvr/OpenVR-SpaceCalibrator/01spacecalibrator
@@ -84,6 +96,11 @@ install(
     DESTINATION lib/steamvr/OpenVR-SpaceCalibrator/01spacecalibrator/bin/linux64/
 )
 
+install(
+    FILES driverInstall.py
+    DESTINATION "${DRIVER_INSTALLER_PATH}"
+)
+
 find_package(Git QUIET)
 
 if(GIT_FOUND AND EXISTS "${PROJECT_SOURCE_DIR}/.git")

OpenVR-SpaceCalibrator/OpenVR-SpaceCalibrator.cpp

@@ -7,7 +7,7 @@
 #include <string>
 #include <codecvt>
 #include <locale>
-#include <GL/gl3w.h>
+#include "GL/gl3w.h"
 
 #ifdef __linux__
 #include <unistd.h>
@@ -24,7 +24,7 @@
 #include <openvr.h>
 
 #ifdef __linux__
-//NOP
+#include "StaticConfig.h"
 #else
 #pragma comment(linker,"\"/manifestdependency:type='win32' \
 name='Microsoft.Windows.Common-Controls' version='6.0.0.0' \
@@ -40,9 +40,9 @@ extern "C" __declspec(dllexport) DWORD AmdPowerXpressRequestHighPerformance = 0x
 #endif
 
 #ifdef __linux__
-#define MANIFEST_END_PATH "/manifest.vrmanifest"
+#define MANIFEST_PATH(CWD)  (APP_MANIFEST_PATH)
 #else
-#define MANIFEST_END_PATH "\\manifest.vrmanifest"
+#define MANIFEST_PATH(CWD) (CWD + "\\manifest.vrmanifest")
 #endif
 
 
@@ -474,6 +474,10 @@ static void HandleCommandLine(LPWSTR lpCmdLine)
         std::cout << "-installmanifest             install the application vrmanifest" << std::endl;
         std::cout << "-removemanifest              remove the application vrmanifest" << std::endl;
         std::cout << "-activatemultipledrivers     enable multiple drivers in steamvr" << std::endl;
+#ifdef __linux__
+        std::cout << "-installdriver               install the steam vr driver." << std::endl;
+        std::cout << "-uninstalldriver             uninstall the steam vr driver." << std::endl;
+#endif
         std::cout << "-help -h                     print this message" << std::endl;
         exit(0);
     }
@@ -513,13 +517,14 @@ static void HandleCommandLine(LPWSTR lpCmdLine)
 				else
 				{
 					std::string manifestPath = oldWd;
-					manifestPath += MANIFEST_END_PATH;
+					manifestPath = MANIFEST_PATH(manifestPath);
 					std::cout << "Removing old manifest path: " << manifestPath << std::endl;
 					vr::VRApplications()->RemoveApplicationManifest(manifestPath.c_str());
 				}
 			}
 			std::string manifestPath = cwd;
-			manifestPath += MANIFEST_END_PATH;
+            manifestPath = MANIFEST_PATH(manifestPath);
+
 			std::cout << "Adding manifest path: " << manifestPath << std::endl;
 			auto vrAppErr = vr::VRApplications()->AddApplicationManifest(manifestPath.c_str());
 			if (vrAppErr != vr::VRApplicationError_None)
@@ -545,8 +550,9 @@ static void HandleCommandLine(LPWSTR lpCmdLine)
 		{
 			if (vr::VRApplications()->IsApplicationInstalled(OPENVR_APPLICATION_KEY))
 			{
-				std::string manifestPath = cwd;
-				manifestPath += MANIFEST_END_PATH;
+                std::string manifestPath = cwd;
+                manifestPath = MANIFEST_PATH(manifestPath);
+
 				std::cout << "Removing manifest path: " << manifestPath << std::endl;
 				vr::VRApplications()->RemoveApplicationManifest(manifestPath.c_str());
 			}
@@ -581,4 +587,56 @@ static void HandleCommandLine(LPWSTR lpCmdLine)
 		vr::VR_Shutdown();
 		exit(ret);
 	}
+#ifdef __linux__
+	else if (StringMatch(lpCmdLine, L"-installdriver"))
+	{
+		auto vrErr = vr::VRInitError_None;
+		vr::VR_Init(&vrErr, vr::VRApplication_Utility);
+		if (vrErr != vr::VRInitError_None)
+		{
+            fprintf(stderr, "Failed to initialize OpenVR: %s\n", vr::VR_GetVRInitErrorAsEnglishDescription(vrErr));
+            vr::VR_Shutdown();
+            exit(-2);
+        }
+
+        char cruntimePath[MAX_PATH] = { 0 };
+        unsigned int pathLen;
+        vr::VR_GetRuntimePath(cruntimePath, MAX_PATH, &pathLen);
+
+        const int cmdLength = 8196;
+        char cmd[cmdLength];
+
+        snprintf(cmd, cmdLength, "python " DRIVER_INSTALLER_PATH "/driverInstall.py --toInstall " DRIVER_MANIFEST_PATH " --vrpathreg %s/bin/vrpathreg.sh", cruntimePath);
+        printf("cmd: %s\n", cmd);
+        system(cmd);
+
+        vr::VR_Shutdown();
+        exit(0);
+	}
+	else if (StringMatch(lpCmdLine, L"-uninstalldriver"))
+	{
+		auto vrErr = vr::VRInitError_None;
+		vr::VR_Init(&vrErr, vr::VRApplication_Utility);
+		if (vrErr != vr::VRInitError_None)
+		{
+            fprintf(stderr, "Failed to initialize OpenVR: %s\n", vr::VR_GetVRInitErrorAsEnglishDescription(vrErr));
+            vr::VR_Shutdown();
+            exit(-2);
+        }
+
+        char cruntimePath[MAX_PATH] = { 0 };
+        unsigned int pathLen;
+        vr::VR_GetRuntimePath(cruntimePath, MAX_PATH, &pathLen);
+
+        const int cmdLength = 8196;
+        char cmd[cmdLength];
+
+        snprintf(cmd, cmdLength,  "\"%s/bin/vrpathreg.sh\" removedriverwithname 01spacecalibrator", cruntimePath);
+        printf("cmd: %s\n", cmd);
+
+        vr::VR_Shutdown();
+        exit(0);
+	}
+#endif
+
 }

OpenVR-SpaceCalibratorDriver/Comms.h

@@ -33,13 +33,13 @@ class UDPServerSocket{
             //TODO verify that 0 is valid for UDP.
             sock = socket(AF_INET, SOCK_DGRAM, 0);
             if(sock == -1){
-                LOG("%s", "Failed to create socket");
+                LOG("Failed to create socket: %s", strerror(errno));
                 return false;
             }
 
             int err = bind(sock, (sockaddr*)&my_addr, sizeof(my_addr));
             if(err){
-                LOG("%s", "Failed to bind socket");
+                LOG("Failed to bind socket: %s", strerror(errno));
                 return false;
             }
             return true;

OpenVR-SpaceCalibratorDriver/Logging.cpp

@@ -3,11 +3,18 @@
 #include <chrono>
 #include <ctime>
 
+#ifdef  __linux__
+#include "StaticConfig.h"
+#else
+#define DRIVER_LOG_FILE "space_calibrator_driver.log"
+#endif
+
+
 FILE *LogFile;
 
 void OpenLogFile()
 {
-	LogFile = fopen("/tmp/space_calibrator_driver.log", "w");
+	LogFile = fopen(DRIVER_LOG_FILE, "w");
 	if (LogFile == nullptr)
 	{
 		LogFile = stderr;

OpenVR-SpaceCalibratorDriver/build.sh

@@ -0,0 +1,4 @@
+#cmakedefine DRIVER_LOG_FILE "@DRIVER_LOG_FILE@"
+#cmakedefine APP_MANIFEST_PATH "@APP_MANIFEST_PATH@"
+#cmakedefine DRIVER_MANIFEST_PATH "@DRIVER_MANIFEST_PATH@"
+#cmakedefine DRIVER_INSTALLER_PATH "@DRIVER_INSTALLER_PATH@"

StaticConfig.h.in

@@ -0,0 +1,76 @@
+#!/usr/bin/env python3
+import os, sys
+import subprocess
+import re
+import argparse
+
+#vrpathreg = "/home/zack/.local/share/Steam/steamapps/common/SteamVR/bin/vrpathreg.sh"
+
+def GetDrivers(vrpathreg):
+    proc = subprocess.Popen([vrpathreg], stdout=subprocess.PIPE, stderr=subprocess.STDOUT, encoding='utf-8')
+    (stdout, _) = proc.communicate()
+
+    lineno = 0
+    externalReg = re.compile('.*External Drivers:.*')
+    lines = stdout.split('\n')
+    for line in lines:
+        lineno += 1
+        if externalReg.match(line):
+            break
+
+    lines = [ x.strip().split(':')[1].strip() for x in lines[lineno:] if len(x) > 0]
+    return lines
+
+def RemoveDriver(vrpathreg, driver):
+    cmd = [vrpathreg, 'removedriver', driver]
+    print(cmd)
+    proc = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, encoding='utf-8')
+    (stdout, _) = proc.communicate()
+
+def AddDriver(vrpathreg, driver):
+    cmd = [vrpathreg, 'adddriver', driver]
+    print(cmd)
+    proc = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, encoding='utf-8')
+    (stdout, _) = proc.communicate()
+
+def main():
+    parser = argparse.ArgumentParser()
+
+    parser.add_argument('--toInstall', help="Driver to install (folder path)", required=True)
+    parser.add_argument('--vrpathreg', help="Path to vrpathreg.sh", required=True)
+
+    args = parser.parse_args()
+
+    fail = False
+    if not os.path.exists(args.toInstall):
+        print("Driver path [{args.toInstall}] does not exist")
+        fail = True
+
+    if not os.path.exists(args.vrpathreg):
+        print("Path to vrpathreg.sh [{args.toInstall}] does not exist")
+        fail = True
+
+    if fail: 
+        return
+
+    drivers = GetDrivers(args.vrpathreg)
+
+    spaceCalReg = re.compile('.*01spacecalibrator.*')
+
+    for driver in drivers:
+        RemoveDriver(args.vrpathreg, driver)
+
+    AddDriver(args.vrpathreg, args.toInstall)
+
+    for driver in drivers[::-1]:
+        if spaceCalReg.match(driver):
+            continue
+
+        AddDriver(args.vrpathreg, driver)
+
+    pass
+
+
+if __name__ == "__main__":
+    main()
+