qga/service-win32.c

   1 /*
   2  * QEMU Guest Agent helpers for win32 service management
   3  *
   4  * Copyright IBM Corp. 2012
   5  *
   6  * Authors:
   7  *  Gal Hammer        <[email protected]>
   8  *  Michael Roth      <[email protected]>
   9  *
  10  * This work is licensed under the terms of the GNU GPL, version 2 or later.
  11  * See the COPYING file in the top-level directory.
  12  */
  13 #include <stdlib.h>
  14 #include <stdio.h>
  15 #include <glib.h>
  16 #include <windows.h>
  17 #include "qga/service-win32.h"
  18
  19 static int printf_win_error(const char *text)
  20 {
  21     DWORD err = GetLastError();
  22     char *message;
  23     int n;
  24
  25     FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER |
  26         FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
  27         NULL,
  28         err,
  29         MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
  30         (char *)&message, 0,
  31         NULL);
  32     n = printf("%s. (Error: %d) %s", text, err, message);
  33     LocalFree(message);
  34
  35     return n;
  36 }
  37
  38 int ga_install_service(const char *path, const char *logfile)
  39 {
  40     SC_HANDLE manager;
  41     SC_HANDLE service;
  42     TCHAR cmdline[MAX_PATH];
  43
  44     if (GetModuleFileName(NULL, cmdline, MAX_PATH) == 0) {
  45         printf_win_error("No full path to service's executable");
  46         return EXIT_FAILURE;
  47     }
  48
  49     _snprintf(cmdline, MAX_PATH - strlen(cmdline), "%s -d", cmdline);
  50
  51     if (path) {
  52         _snprintf(cmdline, MAX_PATH - strlen(cmdline), "%s -p %s", cmdline, path);
  53     }
  54     if (logfile) {
  55         _snprintf(cmdline, MAX_PATH - strlen(cmdline), "%s -l %s -v",
  56             cmdline, logfile);
  57     }
  58
  59     g_debug("service's cmdline: %s", cmdline);
  60
  61     manager = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
  62     if (manager == NULL) {
  63         printf_win_error("No handle to service control manager");
  64         return EXIT_FAILURE;
  65     }
  66
  67     service = CreateService(manager, QGA_SERVICE_NAME, QGA_SERVICE_DISPLAY_NAME,
  68         SERVICE_ALL_ACCESS, SERVICE_WIN32_OWN_PROCESS, SERVICE_AUTO_START,
  69         SERVICE_ERROR_NORMAL, cmdline, NULL, NULL, NULL, NULL, NULL);
  70
  71     if (service) {
  72         SERVICE_DESCRIPTION desc = { (char *)QGA_SERVICE_DESCRIPTION };
  73         ChangeServiceConfig2(service, SERVICE_CONFIG_DESCRIPTION, &desc);
  74
  75         printf("Service was installed successfully.\n");
  76     } else {
  77         printf_win_error("Failed to install service");
  78     }
  79
  80     CloseServiceHandle(service);
  81     CloseServiceHandle(manager);
  82
  83     return (service == NULL);
  84 }
  85
  86 int ga_uninstall_service(void)
  87 {
  88     SC_HANDLE manager;
  89     SC_HANDLE service;
  90
  91     manager = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
  92     if (manager == NULL) {
  93         printf_win_error("No handle to service control manager");
  94         return EXIT_FAILURE;
  95     }
  96
  97     service = OpenService(manager, QGA_SERVICE_NAME, DELETE);
  98     if (service == NULL) {
  99         printf_win_error("No handle to service");
 100         CloseServiceHandle(manager);
 101         return EXIT_FAILURE;
 102     }
 103
 104     if (DeleteService(service) == FALSE) {
 105         printf_win_error("Failed to delete service");
 106     } else {
 107         printf("Service was deleted successfully.\n");
 108     }
 109
 110     CloseServiceHandle(service);
 111     CloseServiceHandle(manager);
 112
 113     return EXIT_SUCCESS;
 114 }