/*  dhasetup - dhahelper setup program
 *
 *  Copyright (c) 2004 - 2007 Sascha Sommer (MPlayer)
 *
 *  Some parts from dhasetup.c source code <http://svn.tilp.info/cgi-bin/viewcvs.cgi/libticables/trunk/src/win32/dha/>
 *
 *  Copyright (C) 2007 Romain Lievin (tilp)
 *
 *  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 <windows.h>
#include <stdio.h>
#include <winioctl.h>

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);
	if(strlen(lpMsgBuf) >= 2)
		lpMsgBuf[strlen(lpMsgBuf)-2] = 0;

        printf("%s (%i -> %s)\n", s, GetLastError(), lpMsgBuf);
		LocalFree(lpMsgBuf);
}

int main(int argc,char* argv[]){
  SC_HANDLE hSCManager = NULL;
  SC_HANDLE hService = NULL;
  char path[MAX_PATH];
  printf("dhasetup (c) 2004 Sascha Sommer\n");
  GetWindowsDirectory(path,MAX_PATH);
  strcpy(path+strlen(path),"\\system32\\drivers\\dhahelper.sys");
  if(argc==1){
    printf("Usage:\n");
    printf("dhasetup install  -  Copies dhahelper.sys from the current directory to\n%s and configures it to start at boot.\n", path);
    printf("dhasetup remove  -  Removes the dhahelper utility.\n");
    return 0;
  }
  hSCManager = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
  if(!strcmp(argv[1],"install")){
    printf("Installing dhahelper...");
    if(!CopyFile("dhahelper.sys",path,FALSE)){
      printf("Copying dhahelper.sys failed.\nEither dhahelper.sys is not in the current directory or you lack sufficient\nprivileges to write to %s.", path);
      return 1;
    }
    // Install the driver
    hService = CreateService(hSCManager,
                             "DHAHELPER",
                             "DHAHELPER",
                             SERVICE_ALL_ACCESS,
                             SERVICE_KERNEL_DRIVER,
                             SERVICE_SYSTEM_START,
                             SERVICE_ERROR_NORMAL,
                             path,
                             NULL,
                             NULL,
                             NULL,
                             NULL,
                             NULL);
    if(!hService){
      print_last_error("Unable to register DhaHelper Service");
      return 1;
    }
	
    if(!StartService(hService, 0, NULL)){
       print_last_error("Error while starting service");
       return 1;
    }

    printf("Success!\n");
  }
  else if(!strcmp(argv[1],"remove")){
    SERVICE_STATUS ServiceStatus;

    printf("Removing dhahelper... ");
    hService = OpenService(hSCManager, "DHAHELPER", SERVICE_ALL_ACCESS);
    if(!hService){
      print_last_error("Error opening dhahelper service");
      return 1;
    }
    if(!ControlService(hService, SERVICE_CONTROL_STOP, &ServiceStatus))
      print_last_error("Error while stopping service");
    if(!DeleteService(hService))
      print_last_error("Error while deleting service");
    DeleteFile(path);
    printf("Done!\n");
  }
  else {
    printf("unknown parameter: %s\n",argv[1]);
  }
  CloseServiceHandle(hService);
  CloseServiceHandle(hSCManager);
  return 0;
}