summaryrefslogtreecommitdiff
path: root/pyparallel/src
diff options
context:
space:
mode:
Diffstat (limited to 'pyparallel/src')
-rw-r--r--pyparallel/src/win32/install_giveio.bat5
-rw-r--r--pyparallel/src/win32/loaddrv_console/loaddrv.c269
-rw-r--r--pyparallel/src/win32/loaddrv_console/loaddrv.h3
3 files changed, 267 insertions, 10 deletions
diff --git a/pyparallel/src/win32/install_giveio.bat b/pyparallel/src/win32/install_giveio.bat
index 3e575d8..79f1383 100644
--- a/pyparallel/src/win32/install_giveio.bat
+++ b/pyparallel/src/win32/install_giveio.bat
@@ -7,8 +7,11 @@
@loaddrv start %DIRVERNAME%
@if errorlevel 1 goto error
-@goto exit
+@loaddrv starttype %DIRVERNAME% auto
+@if errorlevel 1 goto error
+
+@goto exit
:error
@echo ERROR: Installation of %DIRVERNAME% failed
:exit
diff --git a/pyparallel/src/win32/loaddrv_console/loaddrv.c b/pyparallel/src/win32/loaddrv_console/loaddrv.c
index 9d3bc9a..55bb046 100644
--- a/pyparallel/src/win32/loaddrv_console/loaddrv.c
+++ b/pyparallel/src/win32/loaddrv_console/loaddrv.c
@@ -46,11 +46,33 @@ int exists(char *filename) {
return pFile != NULL;
}
+void usage(void) {
+ printf("USGAE: loaddrv command drivername [args...]\n\n"
+ "NT/2k/XP Driver and Service modification tool.\n"
+ "(C)2002 Chris Liechti <cliechti@gmx.net>\n\n"
+ "Suported commands:\n\n"
+ " install [fullpathforinstall]\n"
+ " Install new service. Loaded from given path. If path is not present,\n"
+ " the local directory is searched for a .sys file. If the service\n"
+ " already exists, it must be removed first.\n"
+ " start\n"
+ " Start service. It must be installed in advance.\n"
+ " stop\n"
+ " Stop service.\n"
+ " remove\n"
+ " Remove service. It must be stopped in advance.\n"
+ " status\n"
+ " Show status information about service.\n"
+ " starttype auto|manual|system|disable\n"
+ " Change startup type to the given type.\n"
+ );
+}
+
int main(int argc, char *argv[]) {
DWORD status = 0;
int level = 0;
if (argc < 3) {
- printf("USGAE: loaddrv start|stop|install|remove drivername [fullpathforinstall]");
+ usage();
exit(1);
}
LoadDriverInit();
@@ -103,8 +125,48 @@ int main(int argc, char *argv[]) {
} else {
printf("ok.\n");
}
+ } else if (strcmp(argv[1], "status") == 0) {
+ printf("status of %s:\n", argv[2]);
+ status = DriverStatus(argv[2]);
+ if ( status != OKAY) {
+ printf("stat failed (status %ld):\n", status);
+ level = 1;
+ } else {
+ printf("ok.\n");
+ }
+ } else if (strcmp(argv[1], "starttype") == 0) {
+ if (argc < 4) {
+ printf("Error: need start type (string) as argument.\n");
+ level = 2;
+ } else {
+ DWORD type = 0;
+ printf("set start type %s to %s... ", argv[2], argv[3]);
+ if (strcmp(argv[1], "boot") == 0) {
+ type = SERVICE_BOOT_START;
+ } else if (strcmp(argv[3], "system") == 0) {
+ type = SERVICE_SYSTEM_START;
+ } else if (strcmp(argv[3], "auto") == 0) {
+ type = SERVICE_AUTO_START;
+ } else if (strcmp(argv[3], "manual") == 0) {
+ type = SERVICE_DEMAND_START;
+ } else if (strcmp(argv[3], "disabled") == 0) {
+ type = SERVICE_DISABLED;
+ } else {
+ printf("unknown type\n");
+ level = 1;
+ }
+ if (level == 0) {
+ status = DriverStartType(argv[2], type);
+ if ( status != OKAY) {
+ printf("set start type failed (status %ld):\n", status);
+ level = 1;
+ } else {
+ printf("ok.\n");
+ }
+ }
+ }
} else {
- printf("USGAE: loaddrv start|stop|install|remove drivername [fullpathforinstall]");
+ usage();
level = 1;
}
if (status) DisplayErrorText(status);
@@ -128,8 +190,7 @@ void LoadDriverCleanup(void) {
}
/**-----------------------------------------------------**/
-DWORD DriverInstall(LPSTR lpPath, LPSTR lpDriver)
-{
+DWORD DriverInstall(LPSTR lpPath, LPSTR lpDriver) {
BOOL dwStatus = OKAY;
SC_HANDLE hService = NULL;
@@ -145,8 +206,7 @@ DWORD DriverInstall(LPSTR lpPath, LPSTR lpDriver)
} // DriverInstall
/**-----------------------------------------------------**/
-DWORD DriverStart(LPSTR lpDriver)
-{
+DWORD DriverStart(LPSTR lpDriver) {
BOOL dwStatus = OKAY;
SC_HANDLE hService = NULL;
@@ -193,8 +253,7 @@ DWORD DriverRemove(LPSTR lpDriver)
// get a handle to the service
if ((hService = OpenService(hSCMan, lpDriver,
SERVICE_ALL_ACCESS)) != NULL)
- {
- // remove the driver
+ { // remove the driver
if (!DeleteService(hService))
dwStatus = GetLastError();
} else dwStatus = GetLastError();
@@ -202,3 +261,197 @@ DWORD DriverRemove(LPSTR lpDriver)
if (hService != NULL) CloseServiceHandle(hService);
return dwStatus;
} // DriverRemove
+
+/**-----------------------------------------------------**/
+////extensions by Lch
+/**-----------------------------------------------------**/
+DWORD DriverStatus(LPSTR lpDriver) {
+ BOOL dwStatus = OKAY;
+ SC_HANDLE hService = NULL;
+ DWORD dwBytesNeeded;
+
+ // get a handle to the service
+ if ((hService = OpenService(hSCMan, lpDriver,
+ SERVICE_ALL_ACCESS)) != NULL)
+ {
+ LPQUERY_SERVICE_CONFIG lpqscBuf;
+ LPSERVICE_DESCRIPTION lpqscBuf2;
+ // Allocate a buffer for the configuration information.
+ if ((lpqscBuf = (LPQUERY_SERVICE_CONFIG) LocalAlloc(
+ LPTR, 4096)) != NULL)
+ {
+ if ((lpqscBuf2 = (LPSERVICE_DESCRIPTION) LocalAlloc(
+ LPTR, 4096)) != NULL)
+ {
+ // Get the configuration information.
+ if (QueryServiceConfig(
+ hService,
+ lpqscBuf,
+ 4096,
+ &dwBytesNeeded) &&
+ QueryServiceConfig2(
+ hService,
+ SERVICE_CONFIG_DESCRIPTION,
+ lpqscBuf2,
+ 4096,
+ &dwBytesNeeded) )
+ {
+ // Print the configuration information.
+ printf("Type: [0x%02lx] ", lpqscBuf->dwServiceType);
+ switch (lpqscBuf->dwServiceType) {
+ case SERVICE_WIN32_OWN_PROCESS:
+ printf("The service runs in its own process.");
+ break;
+ case SERVICE_WIN32_SHARE_PROCESS:
+ printf("The service shares a process with other services.");
+ break;
+ case SERVICE_KERNEL_DRIVER:
+ printf("Kernel driver.");
+ break;
+ case SERVICE_FILE_SYSTEM_DRIVER:
+ printf("File system driver.");
+ break;
+ case SERVICE_INTERACTIVE_PROCESS:
+ printf("The service can interact with the desktop.");
+ break;
+ default:
+ printf("Unknown type.");
+ }
+ printf("\nStart Type: [0x%02lx] ", lpqscBuf->dwStartType);
+ switch (lpqscBuf->dwStartType) {
+ case SERVICE_BOOT_START:
+ printf("Boot");
+ break;
+ case SERVICE_SYSTEM_START:
+ printf("System");
+ break;
+ case SERVICE_AUTO_START:
+ printf("Automatic");
+ break;
+ case SERVICE_DEMAND_START:
+ printf("Manual");
+ break;
+ case SERVICE_DISABLED:
+ printf("Disabled");
+ break;
+ default:
+ printf("Unknown.");
+ }
+ printf("\nError Control: [0x%02lx] ", lpqscBuf->dwErrorControl);
+ switch (lpqscBuf->dwErrorControl) {
+ case SERVICE_ERROR_IGNORE:
+ printf("IGNORE: Ignore.");
+ break;
+ case SERVICE_ERROR_NORMAL:
+ printf("NORMAL: Display a message box.");
+ break;
+ case SERVICE_ERROR_SEVERE:
+ printf("SEVERE: Restart with last-known-good config.");
+ break;
+ case SERVICE_ERROR_CRITICAL:
+ printf("CRITICAL: Restart w/ last-known-good config.");
+ break;
+ default:
+ printf("Unknown.");
+ }
+ printf("\nBinary path: %s\n", lpqscBuf->lpBinaryPathName);
+
+ if (lpqscBuf->lpLoadOrderGroup != NULL)
+ printf("Load order grp: %s\n", lpqscBuf->lpLoadOrderGroup);
+ if (lpqscBuf->dwTagId != 0)
+ printf("Tag ID: %ld\n", lpqscBuf->dwTagId);
+ if (lpqscBuf->lpDependencies != NULL)
+ printf("Dependencies: %s\n", lpqscBuf->lpDependencies);
+ if (lpqscBuf->lpServiceStartName != NULL)
+ printf("Start Name: %s\n", lpqscBuf->lpServiceStartName);
+ if (lpqscBuf2->lpDescription != NULL)
+ printf("Description: %s\n", lpqscBuf2->lpDescription);
+ }
+ LocalFree(lpqscBuf2);
+ }
+ LocalFree(lpqscBuf);
+ } else {
+ dwStatus = GetLastError();
+ }
+ } else {
+ dwStatus = GetLastError();
+ }
+
+ if (hService != NULL) CloseServiceHandle(hService);
+ return dwStatus;
+} // DriverStatus
+
+/**-----------------------------------------------------**/
+DWORD DriverStartType(LPSTR lpDriver, DWORD dwStartType) {
+ BOOL dwStatus = OKAY;
+ SC_HANDLE hService = NULL;
+
+ SC_LOCK sclLock;
+ LPQUERY_SERVICE_LOCK_STATUS lpqslsBuf;
+ DWORD dwBytesNeeded;
+
+ // Need to acquire database lock before reconfiguring.
+ sclLock = LockServiceDatabase(hSCMan);
+
+ // If the database cannot be locked, report the details.
+ if (sclLock == NULL) {
+ // Exit if the database is not locked by another process.
+ if (GetLastError() == ERROR_SERVICE_DATABASE_LOCKED) {
+
+ // Allocate a buffer to get details about the lock.
+ lpqslsBuf = (LPQUERY_SERVICE_LOCK_STATUS) LocalAlloc(
+ LPTR, sizeof(QUERY_SERVICE_LOCK_STATUS)+256);
+ if (lpqslsBuf != NULL) {
+ // Get and print the lock status information.
+ if (QueryServiceLockStatus(
+ hSCMan,
+ lpqslsBuf,
+ sizeof(QUERY_SERVICE_LOCK_STATUS)+256,
+ &dwBytesNeeded) )
+ {
+ if (lpqslsBuf->fIsLocked) {
+ printf("Locked by: %s, duration: %ld seconds\n",
+ lpqslsBuf->lpLockOwner,
+ lpqslsBuf->dwLockDuration
+ );
+ } else {
+ printf("No longer locked\n");
+ }
+ }
+ LocalFree(lpqslsBuf);
+ }
+ }
+ dwStatus = GetLastError();
+ } else {
+ // The database is locked, so it is safe to make changes.
+ // Open a handle to the service.
+ hService = OpenService(
+ hSCMan, // SCManager database
+ lpDriver, // name of service
+ SERVICE_CHANGE_CONFIG
+ ); // need CHANGE access
+ if (hService != NULL) {
+ // Make the changes.
+ if (!ChangeServiceConfig(
+ hService, // handle of service
+ SERVICE_NO_CHANGE, // service type: no change
+ dwStartType, // change service start type
+ SERVICE_NO_CHANGE, // error control: no change
+ NULL, // binary path: no change
+ NULL, // load order group: no change
+ NULL, // tag ID: no change
+ NULL, // dependencies: no change
+ NULL, // account name: no change
+ NULL, // password: no change
+ NULL) ) // display name: no change
+ {
+ dwStatus = GetLastError();
+ }
+ }
+ // Release the database lock.
+ UnlockServiceDatabase(sclLock);
+ }
+
+ if (hService != NULL) CloseServiceHandle(hService);
+ return dwStatus;
+} // DriverStartType
diff --git a/pyparallel/src/win32/loaddrv_console/loaddrv.h b/pyparallel/src/win32/loaddrv_console/loaddrv.h
index 4f874da..e09b276 100644
--- a/pyparallel/src/win32/loaddrv_console/loaddrv.h
+++ b/pyparallel/src/win32/loaddrv_console/loaddrv.h
@@ -13,5 +13,6 @@ DWORD DriverInstall(LPSTR, LPSTR);
DWORD DriverStart(LPSTR);
DWORD DriverStop(LPSTR);
DWORD DriverRemove(LPSTR);
-
+DWORD DriverStatus(LPSTR);
+DWORD DriverStartType(LPSTR, DWORD);
#endif //LOADDRV_H \ No newline at end of file