/* dhasetup - DhaHelper setup program
*
* Copyright (c) 2004 Sascha Sommer (MPlayer)
* Copyright (C) 2007 Romain Lievin
*
* Some parts from dhasetup.c source code
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
*/
#include
#include
#include
#include "dhahelper.h"
#define DRV_NAME "DhaHelper"
#define DRV_FILENAME "dhahelper.sys"
#define DRV_VERSION "1.0"
static void print_last_error(char *s)
{
LPTSTR lpMsgBuf;
FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER |
FORMAT_MESSAGE_FROM_SYSTEM |
FORMAT_MESSAGE_IGNORE_INSERTS,
NULL, GetLastError(),
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
(LPTSTR) & lpMsgBuf, 0, NULL);
lpMsgBuf[strlen(lpMsgBuf)-2] = '\0';
printf("%s (%i -> %s)\n", s, GetLastError(), lpMsgBuf);
}
int dha_start(void)
{
SC_HANDLE hSCManager = NULL;
SC_HANDLE hService = NULL;
int result;
hSCManager = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
hService = OpenService(hSCManager, DRV_NAME, SERVICE_ALL_ACCESS);
result = StartService(hService, 0, NULL);
if(!result) print_last_error("Error while starting service");
CloseServiceHandle(hService);
CloseServiceHandle(hSCManager);
return 0;
}
int dha_stop(void)
{
SC_HANDLE hSCManager = NULL;
SC_HANDLE hService = NULL;
SERVICE_STATUS ServiceStatus;
int result;
hSCManager = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
hService = OpenService(hSCManager, DRV_NAME, SERVICE_ALL_ACCESS);
result = ControlService(hService, SERVICE_CONTROL_STOP, &ServiceStatus);
if(!result) print_last_error("Error while stopping service");
CloseServiceHandle(hService);
CloseServiceHandle(hSCManager);
return 0;
}
int dha_install(void)
{
SC_HANDLE hSCManager = NULL;
SC_HANDLE hService = NULL;
char szPath[MAX_PATH];
int result = 0;
GetWindowsDirectory(szPath, MAX_PATH);
strcpy(szPath + strlen(szPath), "\\system32\\drivers\\" DRV_FILENAME);
if(!CopyFile(DRV_FILENAME, szPath, FALSE))
{
printf("Copying " DRV_FILENAME " failed.\nEither " DRV_FILENAME " is not in the current directory or you lack sufficient\nprivileges to write to %s.", szPath);
return -1;
}
// Install the driver
hSCManager = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
hService = CreateService(hSCManager,
DRV_NAME,
DRV_NAME,
SERVICE_ALL_ACCESS,
SERVICE_KERNEL_DRIVER,
SERVICE_SYSTEM_START,
SERVICE_ERROR_NORMAL,
szPath,
NULL,
NULL,
NULL,
NULL,
NULL);
if(!hService)
{
print_last_error("Unable to register DhaHelper Service");
result = -1;
}
else
{
printf("Success!\n");
result = 0;
}
CloseServiceHandle(hService);
CloseServiceHandle(hSCManager);
return result;
}
int dha_uninstall(void)
{
SC_HANDLE hSCManager = NULL;
SC_HANDLE hService = NULL;
char szPath[MAX_PATH];
int result = 0;
hSCManager = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
hService = OpenService(hSCManager, DRV_NAME, SERVICE_ALL_ACCESS);
dha_stop();
result = DeleteService(hService);
if(!result) print_last_error("Error while deleting service");
CloseServiceHandle(hService);
CloseServiceHandle(hSCManager);
GetWindowsDirectory(szPath, MAX_PATH);
strcpy(szPath + strlen(szPath), "\\system32\\drivers\\" DRV_FILENAME);
DeleteFile(szPath);
return 0;
}
int dha_detect(int* result)
{
HANDLE hDriver;
*result = 0;
hDriver = CreateFile("\\\\.\\DhaHelper", GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
if(hDriver == INVALID_HANDLE_VALUE)
{
hDriver = CreateFile("\\\\.\\DhaHelper", GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
if(hDriver != INVALID_HANDLE_VALUE)
{
CloseHandle(hDriver);
hDriver = INVALID_HANDLE_VALUE;
*result = 1;
}
}
else
{
CloseHandle(hDriver);
hDriver = INVALID_HANDLE_VALUE;
*result = 1;
}
printf(DRV_NAME "%sfound.\n", *result ? " " : " not ");
return 0;
}
int dha_enable(void)
{
HANDLE hDriver;
DWORD BytesReturned;
int iError;
int result = 0;
hDriver = CreateFile("\\\\.\\DhaHelper",
GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
if(hDriver == INVALID_HANDLE_VALUE)
{
printf("Couldn't access " DRV_NAME " driver; Please ensure driver is installed/loaded.");
return -1;
}
iError = DeviceIoControl(hDriver, IOCTL_DHAHELPER_ENABLEDIRECTIO, NULL, 0, NULL, 0, &BytesReturned, NULL);
if(!iError)
{
print_last_error("DhaHelper: error %d occured in IOCTL_DHAHELPER_ENABLEDIRECTIO");
result = -1;
}
else
printf("I/O ports have been enabled.\n");
CloseHandle(hDriver);
hDriver = INVALID_HANDLE_VALUE;
return result;
}
int dha_disable(void)
{
HANDLE hDriver;
DWORD BytesReturned;
int iError;
int result = 0;
hDriver = CreateFile("\\\\.\\DhaHelper",
GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
if(hDriver == INVALID_HANDLE_VALUE)
{
printf("Couldn't access " DRV_NAME " driver; Please ensure driver is installed/loaded.");
return -1;
}
iError = DeviceIoControl(hDriver, IOCTL_DHAHELPER_DISABLEDIRECTIO, NULL, 0, NULL, 0, &BytesReturned, NULL);
if(!iError)
{
print_last_error("DhaHelper: error %d occured in IOCTL_DHAHELPER_DISABLEDIRECTIO");
result = -1;
}
else
printf("I/O ports have been disabled.\n");
CloseHandle(hDriver);
hDriver = INVALID_HANDLE_VALUE;
return result;
}
int main(int argc,char* argv[])
{
SC_HANDLE hSCManager = NULL;
SC_HANDLE hService = NULL;
printf("dhasetup (c) 2004 Sascha Sommer\n");
if(argc==1)
{
char szPath[MAX_PATH];
int result = 0;
GetWindowsDirectory(szPath, MAX_PATH);
strcpy(szPath + strlen(szPath), "\\system32\\drivers\\");
printf("Usage:\n");
printf("dhasetup install - Copies dhahelper.sys from the current directory to system \n directory and configures it to start at boot.\n");
printf("dhasetup remove - Removes the DhaHelper driver.\n");
printf("dhasetup start - Start driver.\n");
printf("dhasetup stop - Stop driver.\n");
printf("dhasetup check - Check if driver is present and loaded.\n");
printf("\n");
printf("System directory: %s\n", szPath);
return 0;
}
if(!strcmp(argv[1], "install"))
{
printf("Installing DhaHelper...\n");
dha_install();
dha_start();
}
else if(!strcmp(argv[1], "remove"))
{
printf("Removing DhaHelper...\n");
dha_stop();
dha_uninstall();
}
else if(!strcmp(argv[1], "start"))
{
printf("Starting DhaHelper...\n");
dha_start();
printf("Done!\n");
}
else if(!strcmp(argv[1], "stop"))
{
printf("Stopping DhaHelper...\n");
dha_stop();
printf("Done!\n");
}
else if(!strcmp(argv[1], "check"))
{
int result;
dha_detect(&result);
printf("DhaHelper driver is%spresent.\n", result ? " " : " not ");
}
else
{
printf("unknown parameter: %s\n",argv[1]);
}
return 0;
}