qga/vss-win32.c

   1 /*
   2  * QEMU Guest Agent VSS utility functions
   3  *
   4  * Copyright Hitachi Data Systems Corp. 2013
   5  *
   6  * Authors:
   7  *  Tomoki Sekiyama   <tomoki.sekiyama@hds.com>
   8  *
   9  * This work is licensed under the terms of the GNU GPL, version 2 or later.
  10  * See the COPYING file in the top-level directory.
  11  */
  12
  13 #include <stdio.h>
  14 #include <windows.h>
  15 #include "qga/guest-agent-core.h"
  16 #include "qga/vss-win32.h"
  17 #include "qga/vss-win32/requester.h"
  18
  19 #define QGA_VSS_DLL "qga-vss.dll"
  20
  21 static HMODULE provider_lib;
  22
  23 /* Call a function in qga-vss.dll with the specified name */
  24 static HRESULT call_vss_provider_func(const char *func_name)
  25 {
  26     FARPROC WINAPI func;
  27
  28     g_assert(provider_lib);
  29
  30     func = GetProcAddress(provider_lib, func_name);
  31     if (!func) {
  32         char *msg;
  33         FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER |
  34                       FORMAT_MESSAGE_FROM_SYSTEM, NULL, GetLastError(),
  35                       MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
  36                       (char *)&msg, 0, NULL);
  37         fprintf(stderr, "failed to load %s from %s: %s",
  38                 func_name, QGA_VSS_DLL, msg);
  39         LocalFree(msg);
  40         return E_FAIL;
  41     }
  42
  43     return func();
  44 }
  45
  46 /* Check whether this OS version supports VSS providers */
  47 static bool vss_check_os_version(void)
  48 {
  49     OSVERSIONINFO OSver;
  50
  51     OSver.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);
  52     GetVersionEx(&OSver);
  53     if ((OSver.dwMajorVersion == 5 && OSver.dwMinorVersion >= 2) ||
  54        OSver.dwMajorVersion > 5) {
  55         BOOL wow64 = false;
  56 #ifndef _WIN64
  57         /* Provider doesn't work under WOW64 (32bit agent on 64bit OS) */
  58         if (!IsWow64Process(GetCurrentProcess(), &wow64)) {
  59             fprintf(stderr, "failed to IsWow64Process (Error: %lx\n)\n",
  60                     GetLastError());
  61             return false;
  62         }
  63         if (wow64) {
  64             fprintf(stderr, "Warning: Running under WOW64\n");
  65         }
  66 #endif
  67         return !wow64;
  68     }
  69     return false;
  70 }
  71
  72 /* Load qga-vss.dll */
  73 bool vss_init(bool init_requester)
  74 {
  75     if (!vss_check_os_version()) {
  76         /* Do nothing if OS doesn't support providers. */
  77         fprintf(stderr, "VSS provider is not supported in this OS version: "
  78                 "fsfreeze is disabled.\n");
  79         return false;
  80     }
  81
  82     provider_lib = LoadLibraryA(QGA_VSS_DLL);
  83     if (!provider_lib) {
  84         char *msg;
  85         FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER |
  86                       FORMAT_MESSAGE_FROM_SYSTEM, NULL, GetLastError(),
  87                       MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
  88                       (char *)&msg, 0, NULL);
  89         fprintf(stderr, "failed to load %s: %sfsfreeze is disabled\n",
  90                 QGA_VSS_DLL, msg);
  91         LocalFree(msg);
  92         return false;
  93     }
  94
  95     if (init_requester) {
  96         HRESULT hr = call_vss_provider_func("requester_init");
  97         if (FAILED(hr)) {
  98             fprintf(stderr, "fsfreeze is disabled.\n");
  99             vss_deinit(false);
 100             return false;
 101         }
 102     }
 103
 104     return true;
 105 }
 106
 107 /* Unload qga-provider.dll */
 108 void vss_deinit(bool deinit_requester)
 109 {
 110     if (deinit_requester) {
 111         call_vss_provider_func("requester_deinit");
 112     }
 113     FreeLibrary(provider_lib);
 114     provider_lib = NULL;
 115 }
 116
 117 bool vss_initialized(void)
 118 {
 119     return !!provider_lib;
 120 }
 121
 122 int ga_install_vss_provider(void)
 123 {
 124     HRESULT hr;
 125
 126     if (!vss_init(false)) {
 127         fprintf(stderr, "Installation of VSS provider is skipped. "
 128                 "fsfreeze will be disabled.\n");
 129         return 0;
 130     }
 131     hr = call_vss_provider_func("COMRegister");
 132     vss_deinit(false);
 133
 134     return SUCCEEDED(hr) ? 0 : EXIT_FAILURE;
 135 }
 136
 137 void ga_uninstall_vss_provider(void)
 138 {
 139     if (!vss_init(false)) {
 140         fprintf(stderr, "Removal of VSS provider is skipped.\n");
 141         return;
 142     }
 143     call_vss_provider_func("COMUnregister");
 144     vss_deinit(false);
 145 }
 146
 147 /* Call VSS requester and freeze/thaw filesystems and applications */
 148 void qga_vss_fsfreeze(int *nr_volume, Error **errp, bool freeze)
 149 {
 150     const char *func_name = freeze ? "requester_freeze" : "requester_thaw";
 151     QGAVSSRequesterFunc func;
 152     ErrorSet errset = {
 153         .error_setg_win32 = error_setg_win32_internal,
 154         .errp = errp,
 155     };
 156
 157     g_assert(errp);             /* requester.cpp requires it */
 158     func = (QGAVSSRequesterFunc)GetProcAddress(provider_lib, func_name);
 159     if (!func) {
 160         error_setg_win32(errp, GetLastError(), "failed to load %s from %s",
 161                          func_name, QGA_VSS_DLL);
 162         return;
 163     }
 164
 165     func(nr_volume, &errset);
 166 }