#include <windows.h>
#include <iostream>
#include <string>
#include <algorithm>
using namespace std;

int help()
{
    cout << "usage: giveio-install [option]" << endl;
    cout << "    [option] is one of the following:" << endl;
    cout << "        --install    Installs GiveIO driver" << endl;
    cout << "        --uninstall  Uninstalls GiveIO driver" << endl;
    cout << "        --help       Prints this help message" << endl;
    return 1;
}

int error(const char* msg)
{
    cout << "giveio-install: " << msg << endl;
    return 0;
}

class auto_sc_handle {
public:
    auto_sc_handle(SC_HANDLE h_arg) : h(h_arg) { }
    ~auto_sc_handle() { CloseServiceHandle(h); }

    operator SC_HANDLE() { return h; }
private:
    SC_HANDLE h;
};

bool is_installed(SC_HANDLE scmh)
{
    bool result = false;
    size_t ess_size = sizeof(ENUM_SERVICE_STATUS) + 256 + 256;
    ENUM_SERVICE_STATUS* essp =
	reinterpret_cast<ENUM_SERVICE_STATUS*>(new char[ess_size]);
    DWORD bytes_needed = 0;
    DWORD services_returned = 0;
    DWORD resume_handle = 0;
    do {
	BOOL success =
	    EnumServicesStatus(scmh,
			       SERVICE_DRIVER/* | SERVICE_WIN32*/,
			       SERVICE_STATE_ALL,
			       essp,
			       ess_size,
			       &bytes_needed,
			       &services_returned,
			       &resume_handle);
	if (success == FALSE && GetLastError() != ERROR_MORE_DATA)
	    break;
	
	for (DWORD i = 0; i < services_returned; i++) {
	    // cout << essp[i].lpServiceName << endl;
	    if (strcasecmp(essp[i].lpServiceName, "GiveIO") == 0
		|| strcasecmp(essp[i].lpDisplayName, "GiveIO") == 0) {
		result = true;
		break;
	    }
		
	}
    } while (bytes_needed != 0 && ! result);
    delete[] essp;
    return result;
}

int install()
{
    // Taking out the input from stdin -- eases installation and if
    // the user chooses N, TOS won't work right. 
    // -lkw 1/25/2002
    //cout << "giveio-install:" << endl;
    //cout << "You are about to install the GiveIO driver" << endl;
    //cout << "on your computer!  This driver can give applications" << endl;
    //cout << "direct access to I/O, circumventing" << endl;
    //cout << "Windows NT/Windows 2000 prohibitions." << endl << endl;
    //cout << "Do you want to continue?  (Y/N) ";
    char c;
    //cin >> c;
    //cout << endl;
    //if (c != 'y' && c != 'Y')
	//return error("Installation aborted");

    auto_sc_handle scmh = OpenSCManager(0, 0, GENERIC_WRITE | GENERIC_READ);
    if (! scmh)
	return error("Could not connect to the Service Control Manager");

    if (is_installed(scmh))
	return error("Driver already installed");

    bool copy_failed = false;
    string WINDIR = getenv("WINDIR");
    if (WINDIR.empty()) {
	copy_failed = true;
	cout << "giveio-install: warning: "
	     << "WINDIR environment variable is not defined." << endl;
    } else {
	replace(WINDIR.begin(), WINDIR.end(), '\\', '/');
	const string cp =
	    "cp giveio.sys " + WINDIR + "/system32/drivers/giveio.sys";
	cout << cp << endl;
	if (system(cp.c_str()) != 0) {
	    copy_failed = true;
	    cout << "giveio-install: warning: "
		 << "Copy failed." << endl;
	}
    }
    if (copy_failed)
	cout << "giveio-install: warning: "
	     << "Please copy giveio.sys to the "
	     << "$WINDIR/system32/drivers directory." << endl;

    auto_sc_handle csh =
	CreateService(scmh,
		      "GiveIO",
		      "GiveIO Port Access",
		      SERVICE_ALL_ACCESS,
		      SERVICE_KERNEL_DRIVER,
		      SERVICE_AUTO_START,
		      SERVICE_ERROR_NORMAL,
		      "system32\\drivers\\giveio.sys",
		      0,
		      0,
		      0,
		      0,
		      0);
    if (csh == 0)
      return error("Could not create service");
    BOOL result = StartService(csh,0,NULL);
    if (!result)
      return error("Could not start service");
    return 0;
}

int uninstall()
{
  SERVICE_STATUS status;
    auto_sc_handle scmh = OpenSCManager(0, 0, GENERIC_WRITE | GENERIC_READ);
    if (! scmh)
	return error("Could not connect to the Service Control Manager");

    if (! is_installed(scmh))
	return error("Driver has not been installed");
    auto_sc_handle csh =
	OpenService(scmh,
		    "GiveIO",
		    SERVICE_ALL_ACCESS);
    if (!csh)
	return error("Could not access GiveIO driver");
    BOOL result = ControlService(csh,
				 SERVICE_CONTROL_STOP,
				 &status);
    if (!result) {
      cout << "giveio-install: warning:"
	   << "The GiveIO service could not be stopped. "
	   << "Uninstall will complete when the system restarts." << endl;
    }
    DeleteService(csh);
    return 0;
}

int main(int argc, char* argv[])
{
    enum actions {
	a_install,
	a_uninstall,
	a_help
    } action = a_install;

    for (int i = 1; i < argc; i++) {
	if (strcmp(argv[i], "--install") == 0)
	    action = a_install;
	else if (strcmp(argv[i], "--uninstall") == 0)
	    action = a_uninstall;
	else if (strcmp(argv[i], "--help") == 0)
	    action = a_help;
	else
	    action = a_help;
    }
    switch (action) {
    case a_help:
	return help();
    case a_install:
	return install();
    case a_uninstall:
	return uninstall();
    }
    return 0;
}