--- /dev/null
+#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;
+}
projects / tinyos-2.x.git / blobdiff