src/util/winproc.c

   1 /*
   2      This file is part of GNUnet.
   3      Copyright (C) 2001, 2002, 2003, 2004, 2005, 2006 GNUnet e.V.
   4
   5      GNUnet is free software; you can redistribute it and/or modify
   6      it under the terms of the GNU General Public License as published
   7      by the Free Software Foundation; either version 3, or (at your
   8      option) any later version.
   9
  10      GNUnet is distributed in the hope that it will be useful, but
  11      WITHOUT ANY WARRANTY; without even the implied warranty of
  12      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13      General Public License for more details.
  14
  15      You should have received a copy of the GNU General Public License
  16      along with GNUnet; see the file COPYING.  If not, write to the
  17      Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
  18      Boston, MA 02110-1301, USA.
  19 */
  20
  21 /**
  22  * @file util/winproc.c
  23  * @brief Functions for MS Windows
  24  * @author Nils Durner
  25  */
  26
  27 #include "platform.h"
  28 #include "gnunet_crypto_lib.h"
  29 #include "gnunet_common.h"
  30
  31
  32 #ifdef MINGW
  33
  34 static HINSTANCE hNTDLL, hIphlpapi, hAdvapi, hNetapi;
  35 #ifdef W32_VEH
  36 static void *GNWinVEH_handle = NULL;
  37 #endif
  38
  39 TNtQuerySystemInformation GNNtQuerySystemInformation;
  40 TGetIfEntry GNGetIfEntry;
  41 TGetIpAddrTable GNGetIpAddrTable;
  42 TGetIfTable GNGetIfTable;
  43 TOpenSCManager GNOpenSCManager;
  44 TCreateService GNCreateService;
  45 TCloseServiceHandle GNCloseServiceHandle;
  46 TDeleteService GNDeleteService;
  47 TRegisterServiceCtrlHandler GNRegisterServiceCtrlHandler;
  48 TSetServiceStatus GNSetServiceStatus;
  49 TStartServiceCtrlDispatcher GNStartServiceCtrlDispatcher;
  50 TControlService GNControlService;
  51 TOpenService GNOpenService;
  52 TGetBestInterfaceEx GNGetBestInterfaceEx;
  53 TGetAdaptersInfo GNGetAdaptersInfo;
  54 TNetUserAdd GNNetUserAdd;
  55 TNetUserSetInfo GNNetUserSetInfo;
  56 TLsaOpenPolicy GNLsaOpenPolicy;
  57 TLsaAddAccountRights GNLsaAddAccountRights;
  58 TLsaRemoveAccountRights GNLsaRemoveAccountRights;
  59 TLsaClose GNLsaClose;
  60 TLookupAccountName GNLookupAccountName;
  61 TGetFileSecurity GNGetFileSecurity;
  62 TInitializeSecurityDescriptor GNInitializeSecurityDescriptor;
  63 TGetSecurityDescriptorDacl GNGetSecurityDescriptorDacl;
  64 TGetAclInformation GNGetAclInformation;
  65 TInitializeAcl GNInitializeAcl;
  66 TGetAce GNGetAce;
  67 TEqualSid GNEqualSid;
  68 TAddAce GNAddAce;
  69 TAddAccessAllowedAce GNAddAccessAllowedAce;
  70 TSetNamedSecurityInfo GNSetNamedSecurityInfo;
  71
  72 #define LOG(kind,...) GNUNET_log_from (kind, "util-winproc", __VA_ARGS__)
  73 /**
  74  * Log (panic) messages from PlibC
  75  */
  76 void
  77 plibc_panic (int err, char *msg)
  78 {
  79   LOG (((err == INT_MAX) ? GNUNET_ERROR_TYPE_DEBUG : GNUNET_ERROR_TYPE_ERROR),
  80        "%s", msg);
  81 }
  82
  83 #ifdef W32_VEH
  84 /**
  85  * Handles exceptions (useful for debugging).
  86  * Issues a DebugBreak() call if the process is being debugged (not really
  87  * useful - if the process is being debugged, this handler won't be invoked
  88  * anyway). If it is not, runs a debugger from GNUNET_DEBUGGER env var,
  89  * substituting first %u in it for PID, and the second one for the event,
  90  * that should be set once the debugger attaches itself (otherwise the
  91  * only way out of WaitForSingleObject() is to time out after 1 minute).
  92  */
  93 LONG __stdcall
  94 GNWinVEH (PEXCEPTION_POINTERS ExceptionInfo)
  95 {
  96   char debugger[MAX_PATH + 1];
  97   char *debugger_env = NULL;
  98   if (IsDebuggerPresent ())
  99   {
 100     DebugBreak ();
 101     return EXCEPTION_CONTINUE_EXECUTION;
 102   }
 103   debugger_env = getenv ("GNUNET_DEBUGGER");
 104   if (debugger_env != NULL)
 105   {
 106     STARTUPINFO si;
 107     PROCESS_INFORMATION pi;
 108     HANDLE event;
 109     SECURITY_ATTRIBUTES sa;
 110     memset (&si, 0, sizeof (si));
 111     si.cb = sizeof (si);
 112     memset (&pi, 0, sizeof (pi));
 113     memset (&sa, 0, sizeof (sa));
 114     sa.nLength = sizeof (sa);
 115     sa.bInheritHandle = TRUE;
 116     event = CreateEvent (&sa, FALSE, FALSE, NULL);
 117     snprintf (debugger, MAX_PATH + 1, debugger_env, GetCurrentProcessId (), (uintptr_t) event);
 118     debugger[MAX_PATH] = '\0';
 119     if (0 != CreateProcessA (NULL, debugger, NULL, NULL, TRUE, CREATE_NEW_CONSOLE, NULL, NULL, &si, &pi))
 120     {
 121       CloseHandle (pi.hProcess);
 122       CloseHandle (pi.hThread);
 123       WaitForSingleObject (event, 60000);
 124       CloseHandle (event);
 125       if (IsDebuggerPresent ())
 126       {
 127         return EXCEPTION_CONTINUE_EXECUTION;
 128       }
 129     }
 130     else
 131       CloseHandle (event);
 132   }
 133   return EXCEPTION_CONTINUE_SEARCH;
 134 }
 135 #endif
 136
 137 /**
 138  * @brief Initialize PlibC and set up Windows environment
 139  * @param logging context, NULL means stderr
 140  * @return Error code from winerror.h, ERROR_SUCCESS on success
 141 */
 142 int
 143 GNInitWinEnv ()
 144 {
 145   int ret;
 146
 147   plibc_initialized ();
 148   plibc_set_panic_proc (plibc_panic);
 149   ret = plibc_init_utf8 ("GNU", PACKAGE, 1);
 150   plibc_set_stat_size_size (sizeof (((struct stat *) 0)->st_size));
 151   plibc_set_stat_time_size (sizeof (((struct stat *) 0)->st_mtime));
 152   /* don't load other DLLs twice */
 153   if (hNTDLL)
 154     return ret;
 155
 156 #ifdef W32_VEH
 157   if (GNWinVEH_handle == NULL)
 158   {
 159     GNWinVEH_handle = AddVectoredExceptionHandler (1, &GNWinVEH);
 160     if (GNWinVEH_handle == NULL)
 161     {
 162       /* This is bad, but what can we do? */
 163       printf ("Failed to set up an exception handler!\n");
 164     }
 165   }
 166 #endif
 167
 168   hNTDLL = LoadLibrary ("ntdll.dll");
 169
 170   /* Function to get CPU usage under Win NT */
 171   if (hNTDLL)
 172   {
 173     GNNtQuerySystemInformation =
 174         (TNtQuerySystemInformation) GetProcAddress (hNTDLL,
 175                                                     "NtQuerySystemInformation");
 176   }
 177   else
 178   {
 179     GNNtQuerySystemInformation = NULL;
 180   }
 181
 182   /* Functions to get information about a network adapter */
 183   hIphlpapi = LoadLibrary ("iphlpapi.dll");
 184   if (hIphlpapi)
 185   {
 186     GNGetIfEntry = (TGetIfEntry) GetProcAddress (hIphlpapi, "GetIfEntry");
 187     GNGetIpAddrTable =
 188         (TGetIpAddrTable) GetProcAddress (hIphlpapi, "GetIpAddrTable");
 189     GNGetIfTable = (TGetIfTable) GetProcAddress (hIphlpapi, "GetIfTable");
 190     GNGetBestInterfaceEx =
 191         (TGetBestInterfaceEx) GetProcAddress (hIphlpapi, "GetBestInterfaceEx");
 192     GNGetAdaptersInfo =
 193         (TGetAdaptersInfo) GetProcAddress (hIphlpapi, "GetAdaptersInfo");
 194   }
 195   else
 196   {
 197     GNGetIfEntry = NULL;
 198     GNGetIpAddrTable = NULL;
 199     GNGetIfTable = NULL;
 200     GNGetBestInterfaceEx = NULL;
 201     GNGetAdaptersInfo = NULL;
 202   }
 203
 204   /* Service & Account functions */
 205   hAdvapi = LoadLibrary ("advapi32.dll");
 206   if (hAdvapi)
 207   {
 208     GNOpenSCManager =
 209         (TOpenSCManager) GetProcAddress (hAdvapi, "OpenSCManagerA");
 210     GNCreateService =
 211         (TCreateService) GetProcAddress (hAdvapi, "CreateServiceA");
 212     GNCloseServiceHandle =
 213         (TCloseServiceHandle) GetProcAddress (hAdvapi, "CloseServiceHandle");
 214     GNDeleteService =
 215         (TDeleteService) GetProcAddress (hAdvapi, "DeleteService");
 216     GNRegisterServiceCtrlHandler =
 217         (TRegisterServiceCtrlHandler) GetProcAddress (hAdvapi,
 218                                                       "RegisterServiceCtrlHandlerA");
 219     GNSetServiceStatus =
 220         (TSetServiceStatus) GetProcAddress (hAdvapi, "SetServiceStatus");
 221     GNStartServiceCtrlDispatcher =
 222         (TStartServiceCtrlDispatcher) GetProcAddress (hAdvapi,
 223                                                       "StartServiceCtrlDispatcherA");
 224     GNControlService =
 225         (TControlService) GetProcAddress (hAdvapi, "ControlService");
 226     GNOpenService = (TOpenService) GetProcAddress (hAdvapi, "OpenServiceA");
 227
 228     GNLsaOpenPolicy =
 229         (TLsaOpenPolicy) GetProcAddress (hAdvapi, "LsaOpenPolicy");
 230     GNLsaAddAccountRights =
 231         (TLsaAddAccountRights) GetProcAddress (hAdvapi, "LsaAddAccountRights");
 232     GNLsaRemoveAccountRights =
 233         (TLsaRemoveAccountRights) GetProcAddress (hAdvapi,
 234                                                   "LsaRemoveAccountRights");
 235     GNLsaClose = (TLsaClose) GetProcAddress (hAdvapi, "LsaClose");
 236     GNLookupAccountName =
 237         (TLookupAccountName) GetProcAddress (hAdvapi, "LookupAccountNameA");
 238
 239     GNGetFileSecurity =
 240         (TGetFileSecurity) GetProcAddress (hAdvapi, "GetFileSecurityA");
 241     GNInitializeSecurityDescriptor =
 242         (TInitializeSecurityDescriptor) GetProcAddress (hAdvapi,
 243                                                         "InitializeSecurityDescriptor");
 244     GNGetSecurityDescriptorDacl =
 245         (TGetSecurityDescriptorDacl) GetProcAddress (hAdvapi,
 246                                                      "GetSecurityDescriptorDacl");
 247     GNGetAclInformation =
 248         (TGetAclInformation) GetProcAddress (hAdvapi, "GetAclInformation");
 249     GNInitializeAcl =
 250         (TInitializeAcl) GetProcAddress (hAdvapi, "InitializeAcl");
 251     GNGetAce = (TGetAce) GetProcAddress (hAdvapi, "GetAce");
 252     GNEqualSid = (TEqualSid) GetProcAddress (hAdvapi, "EqualSid");
 253     GNAddAce = (TAddAce) GetProcAddress (hAdvapi, "AddAce");
 254     GNAddAccessAllowedAce =
 255         (TAddAccessAllowedAce) GetProcAddress (hAdvapi, "AddAccessAllowedAce");
 256     GNSetNamedSecurityInfo =
 257         (TSetNamedSecurityInfo) GetProcAddress (hAdvapi,
 258                                                 "SetNamedSecurityInfoA");
 259   }
 260   else
 261   {
 262     GNOpenSCManager = NULL;
 263     GNCreateService = NULL;
 264     GNCloseServiceHandle = NULL;
 265     GNDeleteService = NULL;
 266     GNRegisterServiceCtrlHandler = NULL;
 267     GNSetServiceStatus = NULL;
 268     GNStartServiceCtrlDispatcher = NULL;
 269     GNControlService = NULL;
 270     GNOpenService = NULL;
 271
 272     GNLsaOpenPolicy = NULL;
 273     GNLsaAddAccountRights = NULL;
 274     GNLsaRemoveAccountRights = NULL;
 275     GNLsaClose = NULL;
 276     GNLookupAccountName = NULL;
 277
 278     GNGetFileSecurity = NULL;
 279     GNInitializeSecurityDescriptor = NULL;
 280     GNGetSecurityDescriptorDacl = NULL;
 281     GNGetAclInformation = NULL;
 282     GNInitializeAcl = NULL;
 283     GNGetAce = NULL;
 284     GNEqualSid = NULL;
 285     GNAddAce = NULL;
 286     GNAddAccessAllowedAce = NULL;
 287     GNSetNamedSecurityInfo = NULL;
 288   }
 289
 290   /* Account function */
 291   hNetapi = LoadLibrary ("netapi32.dll");
 292   if (hNetapi)
 293   {
 294     GNNetUserAdd = (TNetUserAdd) GetProcAddress (hNetapi, "NetUserAdd");
 295     GNNetUserSetInfo =
 296         (TNetUserSetInfo) GetProcAddress (hNetapi, "NetUserSetInfo");
 297   }
 298   else
 299   {
 300     GNNetUserAdd = NULL;
 301     GNNetUserSetInfo = NULL;
 302   }
 303
 304   return ret;
 305 }
 306
 307 /**
 308  * Clean up Windows environment
 309  */
 310 void
 311 GNShutdownWinEnv ()
 312 {
 313   plibc_shutdown ();
 314
 315 #ifdef W32_VEH
 316   if (GNWinVEH_handle != NULL)
 317   {
 318     RemoveVectoredExceptionHandler (GNWinVEH_handle);
 319     GNWinVEH_handle = NULL;
 320   }
 321 #endif
 322
 323   FreeLibrary (hNTDLL);
 324   FreeLibrary (hIphlpapi);
 325   FreeLibrary (hAdvapi);
 326   FreeLibrary (hNetapi);
 327
 328   CoUninitialize ();
 329 }
 330
 331 #endif /* MINGW */
 332
 333 #if !HAVE_ATOLL
 334 long long
 335 atoll (const char *nptr)
 336 {
 337   return atol (nptr);
 338 }
 339 #endif