src/msw/utils.cpp

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        msw/utils.cpp
   3 // Purpose:     Various utilities
   4 // Author:      Julian Smart
   5 // Modified by:
   6 // Created:     04/01/98
   7 // RCS-ID:      $Id$
   8 // Copyright:   (c) Julian Smart
   9 // Licence:     wxWindows licence
  10 /////////////////////////////////////////////////////////////////////////////
  11
  12 // ============================================================================
  13 // declarations
  14 // ============================================================================
  15
  16 // ----------------------------------------------------------------------------
  17 // headers
  18 // ----------------------------------------------------------------------------
  19
  20 // For compilers that support precompilation, includes "wx.h".
  21 #include "wx/wxprec.h"
  22
  23 #ifdef __BORLANDC__
  24     #pragma hdrstop
  25 #endif
  26
  27 #ifndef WX_PRECOMP
  28     #include "wx/utils.h"
  29     #include "wx/app.h"
  30     #include "wx/intl.h"
  31     #include "wx/log.h"
  32 #endif  //WX_PRECOMP
  33
  34 #include "wx/apptrait.h"
  35 #include "wx/dynload.h"
  36
  37 #include "wx/msw/private.h"     // includes <windows.h>
  38 #include "wx/msw/missing.h"     // CHARSET_HANGUL
  39
  40 #if defined(__GNUWIN32_OLD__) || defined(__WXWINCE__) \
  41     || defined(__CYGWIN32__)
  42     // apparently we need to include winsock.h to get WSADATA and other stuff
  43     // used in wxGetFullHostName() with the old mingw32 versions
  44     #include <winsock.h>
  45 #endif
  46
  47 #include "wx/timer.h"
  48
  49 #if !defined(__GNUWIN32__) && !defined(__SALFORDC__) && !defined(__WXMICROWIN__) && !defined(__WXWINCE__)
  50     #include <direct.h>
  51
  52     #ifndef __MWERKS__
  53         #include <dos.h>
  54     #endif
  55 #endif  //GNUWIN32
  56
  57 #if defined(__CYGWIN__)
  58     #include <sys/unistd.h>
  59     #include <sys/stat.h>
  60     #include <sys/cygwin.h> // for cygwin_conv_to_full_win32_path()
  61 #endif  //GNUWIN32
  62
  63 #ifdef __BORLANDC__ // Please someone tell me which version of Borland needs
  64                     // this (3.1 I believe) and how to test for it.
  65                     // If this works for Borland 4.0 as well, then no worries.
  66     #include <dir.h>
  67 #endif
  68
  69 // VZ: there is some code using NetXXX() functions to get the full user name:
  70 //     I don't think it's a good idea because they don't work under Win95 and
  71 //     seem to return the same as wxGetUserId() under NT. If you really want
  72 //     to use them, just #define USE_NET_API
  73 #undef USE_NET_API
  74
  75 #ifdef USE_NET_API
  76     #include <lm.h>
  77 #endif // USE_NET_API
  78
  79 #if defined(__WIN32__) && !defined(__WXMICROWIN__) && !defined(__WXWINCE__)
  80     #ifndef __UNIX__
  81         #include <io.h>
  82     #endif
  83
  84     #ifndef __GNUWIN32__
  85         #include <shellapi.h>
  86     #endif
  87 #endif
  88
  89 #ifndef __WATCOMC__
  90     #if !(defined(_MSC_VER) && (_MSC_VER > 800))
  91         #include <errno.h>
  92     #endif
  93 #endif
  94
  95 // ----------------------------------------------------------------------------
  96 // constants
  97 // ----------------------------------------------------------------------------
  98
  99 // In the WIN.INI file
 100 static const wxChar WX_SECTION[] = wxT("wxWindows");
 101 static const wxChar eUSERNAME[]  = wxT("UserName");
 102
 103 // these are only used under Win16
 104 #if !defined(__WIN32__) && !defined(__WXMICROWIN__)
 105 static const wxChar eHOSTNAME[]  = wxT("HostName");
 106 static const wxChar eUSERID[]    = wxT("UserId");
 107 #endif // !Win32
 108
 109 // ============================================================================
 110 // implementation
 111 // ============================================================================
 112
 113 // ----------------------------------------------------------------------------
 114 // get host name and related
 115 // ----------------------------------------------------------------------------
 116
 117 // Get hostname only (without domain name)
 118 bool wxGetHostName(wxChar *buf, int maxSize)
 119 {
 120 #if defined(__WXWINCE__)
 121     return FALSE;
 122 #elif defined(__WIN32__) && !defined(__WXMICROWIN__)
 123     DWORD nSize = maxSize;
 124     if ( !::GetComputerName(buf, &nSize) )
 125     {
 126         wxLogLastError(wxT("GetComputerName"));
 127
 128         return FALSE;
 129     }
 130
 131     return TRUE;
 132 #else
 133     wxChar *sysname;
 134     const wxChar *default_host = wxT("noname");
 135
 136     if ((sysname = wxGetenv(wxT("SYSTEM_NAME"))) == NULL) {
 137         GetProfileString(WX_SECTION, eHOSTNAME, default_host, buf, maxSize - 1);
 138     } else
 139         wxStrncpy(buf, sysname, maxSize - 1);
 140     buf[maxSize] = wxT('\0');
 141     return *buf ? TRUE : FALSE;
 142 #endif
 143 }
 144
 145 // get full hostname (with domain name if possible)
 146 bool wxGetFullHostName(wxChar *buf, int maxSize)
 147 {
 148 #if !defined( __WXMICROWIN__) && wxUSE_DYNAMIC_LOADER
 149     // TODO should use GetComputerNameEx() when available
 150
 151     // we don't want to always link with Winsock DLL as we might not use it at
 152     // all, so load it dynamically here if needed
 153     wxDynamicLibrary dllWinsock(_T("ws2_32.dll"), wxDL_VERBATIM);
 154     if ( dllWinsock.IsLoaded() )
 155     {
 156         typedef int (PASCAL *WSAStartup_t)(WORD, WSADATA *);
 157         typedef int (PASCAL *gethostname_t)(char *, int);
 158         typedef hostent* (PASCAL *gethostbyname_t)(const char *);
 159         typedef hostent* (PASCAL *gethostbyaddr_t)(const char *, int , int);
 160         typedef int (PASCAL *WSACleanup_t)(void);
 161
 162         #define LOAD_WINSOCK_FUNC(func)                                       \
 163             func ## _t                                                        \
 164                 pfn ## func = (func ## _t)dllWinsock.GetSymbol(_T(#func))
 165
 166         LOAD_WINSOCK_FUNC(WSAStartup);
 167
 168         WSADATA wsa;
 169         if ( pfnWSAStartup && pfnWSAStartup(MAKEWORD(1, 1), &wsa) == 0 )
 170         {
 171             LOAD_WINSOCK_FUNC(gethostname);
 172
 173             wxString host;
 174             if ( pfngethostname )
 175             {
 176                 char bufA[256];
 177                 if ( pfngethostname(bufA, WXSIZEOF(bufA)) == 0 )
 178                 {
 179                     // gethostname() won't usually include the DNS domain name,
 180                     // for this we need to work a bit more
 181                     if ( !strchr(bufA, '.') )
 182                     {
 183                         LOAD_WINSOCK_FUNC(gethostbyname);
 184
 185                         struct hostent *pHostEnt = pfngethostbyname
 186                                                     ? pfngethostbyname(bufA)
 187                                                     : NULL;
 188
 189                         if ( pHostEnt )
 190                         {
 191                             // Windows will use DNS internally now
 192                             LOAD_WINSOCK_FUNC(gethostbyaddr);
 193
 194                             pHostEnt = pfngethostbyaddr
 195                                         ? pfngethostbyaddr(pHostEnt->h_addr,
 196                                                            4, AF_INET)
 197                                         : NULL;
 198                         }
 199
 200                         if ( pHostEnt )
 201                         {
 202                             host = wxString::FromAscii(pHostEnt->h_name);
 203                         }
 204                     }
 205                 }
 206             }
 207
 208             LOAD_WINSOCK_FUNC(WSACleanup);
 209             if ( pfnWSACleanup )
 210                 pfnWSACleanup();
 211
 212
 213             if ( !host.empty() )
 214             {
 215                 wxStrncpy(buf, host, maxSize);
 216
 217                 return TRUE;
 218             }
 219         }
 220     }
 221 #endif // !__WXMICROWIN__
 222
 223     return wxGetHostName(buf, maxSize);
 224 }
 225
 226 // Get user ID e.g. jacs
 227 bool wxGetUserId(wxChar *buf, int maxSize)
 228 {
 229 #if defined(__WXWINCE__)
 230     return FALSE;
 231 #elif defined(__WIN32__) && !defined(__win32s__) && !defined(__WXMICROWIN__)
 232     DWORD nSize = maxSize;
 233     if ( ::GetUserName(buf, &nSize) == 0 )
 234     {
 235         // actually, it does happen on Win9x if the user didn't log on
 236         DWORD res = ::GetEnvironmentVariable(wxT("username"), buf, maxSize);
 237         if ( res == 0 )
 238         {
 239             // not found
 240             return FALSE;
 241         }
 242     }
 243
 244     return TRUE;
 245 #else   // Win16 or Win32s
 246     wxChar *user;
 247     const wxChar *default_id = wxT("anonymous");
 248
 249     // Can't assume we have NIS (PC-NFS) or some other ID daemon
 250     // So we ...
 251     if (  (user = wxGetenv(wxT("USER"))) == NULL &&
 252             (user = wxGetenv(wxT("LOGNAME"))) == NULL )
 253     {
 254         // Use wxWindows configuration data (comming soon)
 255         GetProfileString(WX_SECTION, eUSERID, default_id, buf, maxSize - 1);
 256     }
 257     else
 258     {
 259         wxStrncpy(buf, user, maxSize - 1);
 260     }
 261
 262     return *buf ? TRUE : FALSE;
 263 #endif
 264 }
 265
 266 // Get user name e.g. Julian Smart
 267 bool wxGetUserName(wxChar *buf, int maxSize)
 268 {
 269 #if defined(__WXWINCE__)
 270     return FALSE;
 271 #elif defined(USE_NET_API)
 272     CHAR szUserName[256];
 273     if ( !wxGetUserId(szUserName, WXSIZEOF(szUserName)) )
 274         return FALSE;
 275
 276     // TODO how to get the domain name?
 277     CHAR *szDomain = "";
 278
 279     // the code is based on the MSDN example (also see KB article Q119670)
 280     WCHAR wszUserName[256];          // Unicode user name
 281     WCHAR wszDomain[256];
 282     LPBYTE ComputerName;
 283
 284     USER_INFO_2 *ui2;         // User structure
 285
 286     // Convert ANSI user name and domain to Unicode
 287     MultiByteToWideChar( CP_ACP, 0, szUserName, strlen(szUserName)+1,
 288             wszUserName, WXSIZEOF(wszUserName) );
 289     MultiByteToWideChar( CP_ACP, 0, szDomain, strlen(szDomain)+1,
 290             wszDomain, WXSIZEOF(wszDomain) );
 291
 292     // Get the computer name of a DC for the domain.
 293     if ( NetGetDCName( NULL, wszDomain, &ComputerName ) != NERR_Success )
 294     {
 295         wxLogError(wxT("Can not find domain controller"));
 296
 297         goto error;
 298     }
 299
 300     // Look up the user on the DC
 301     NET_API_STATUS status = NetUserGetInfo( (LPWSTR)ComputerName,
 302             (LPWSTR)&wszUserName,
 303             2, // level - we want USER_INFO_2
 304             (LPBYTE *) &ui2 );
 305     switch ( status )
 306     {
 307         case NERR_Success:
 308             // ok
 309             break;
 310
 311         case NERR_InvalidComputer:
 312             wxLogError(wxT("Invalid domain controller name."));
 313
 314             goto error;
 315
 316         case NERR_UserNotFound:
 317             wxLogError(wxT("Invalid user name '%s'."), szUserName);
 318
 319             goto error;
 320
 321         default:
 322             wxLogSysError(wxT("Can't get information about user"));
 323
 324             goto error;
 325     }
 326
 327     // Convert the Unicode full name to ANSI
 328     WideCharToMultiByte( CP_ACP, 0, ui2->usri2_full_name, -1,
 329             buf, maxSize, NULL, NULL );
 330
 331     return TRUE;
 332
 333 error:
 334     wxLogError(wxT("Couldn't look up full user name."));
 335
 336     return FALSE;
 337 #else  // !USE_NET_API
 338     // Could use NIS, MS-Mail or other site specific programs
 339     // Use wxWindows configuration data
 340     bool ok = GetProfileString(WX_SECTION, eUSERNAME, wxEmptyString, buf, maxSize - 1) != 0;
 341     if ( !ok )
 342     {
 343         ok = wxGetUserId(buf, maxSize);
 344     }
 345
 346     if ( !ok )
 347     {
 348         wxStrncpy(buf, wxT("Unknown User"), maxSize);
 349     }
 350 #endif // Win32/16
 351
 352     return TRUE;
 353 }
 354
 355 const wxChar* wxGetHomeDir(wxString *pstr)
 356 {
 357   wxString& strDir = *pstr;
 358
 359 #if defined(__UNIX__)
 360     const wxChar *szHome = wxGetenv("HOME");
 361     if ( szHome == NULL ) {
 362       // we're homeless...
 363       wxLogWarning(_("can't find user's HOME, using current directory."));
 364       strDir = wxT(".");
 365     }
 366     else
 367        strDir = szHome;
 368
 369     // add a trailing slash if needed
 370     if ( strDir.Last() != wxT('/') )
 371       strDir << wxT('/');
 372
 373     #ifdef __CYGWIN__
 374       // Cygwin returns unix type path but that does not work well
 375       static wxChar windowsPath[MAX_PATH];
 376       cygwin_conv_to_full_win32_path(strDir, windowsPath);
 377       strDir = windowsPath;
 378     #endif
 379 #elif defined(__WXWINCE__)
 380       // Nothing
 381 #else
 382     #ifdef  __WIN32__
 383       strDir.clear();
 384
 385       // If we have a valid HOME directory, as is used on many machines that
 386       // have unix utilities on them, we should use that.
 387       const wxChar *szHome = wxGetenv(wxT("HOME"));
 388
 389       if ( szHome != NULL )
 390       {
 391         strDir = szHome;
 392       }
 393       else // no HOME, try HOMEDRIVE/PATH
 394       {
 395           szHome = wxGetenv(wxT("HOMEDRIVE"));
 396           if ( szHome != NULL )
 397             strDir << szHome;
 398           szHome = wxGetenv(wxT("HOMEPATH"));
 399
 400           if ( szHome != NULL )
 401           {
 402             strDir << szHome;
 403
 404             // the idea is that under NT these variables have default values
 405             // of "%systemdrive%:" and "\\". As we don't want to create our
 406             // config files in the root directory of the system drive, we will
 407             // create it in our program's dir. However, if the user took care
 408             // to set HOMEPATH to something other than "\\", we suppose that he
 409             // knows what he is doing and use the supplied value.
 410             if ( wxStrcmp(szHome, wxT("\\")) == 0 )
 411               strDir.clear();
 412           }
 413       }
 414
 415       if ( strDir.empty() )
 416       {
 417           // If we have a valid USERPROFILE directory, as is the case in
 418           // Windows NT, 2000 and XP, we should use that as our home directory.
 419           szHome = wxGetenv(wxT("USERPROFILE"));
 420
 421           if ( szHome != NULL )
 422             strDir = szHome;
 423       }
 424
 425       if ( !strDir.empty() )
 426       {
 427           return strDir.c_str();
 428       }
 429       //else: fall back to the prograrm directory
 430     #else   // Win16
 431       // Win16 has no idea about home, so use the executable directory instead
 432     #endif  // WIN16/32
 433
 434     // 260 was taken from windef.h
 435     #ifndef MAX_PATH
 436       #define MAX_PATH  260
 437     #endif
 438
 439     wxString strPath;
 440     ::GetModuleFileName(::GetModuleHandle(NULL),
 441                         wxStringBuffer(strPath, MAX_PATH), MAX_PATH);
 442
 443     // extract the dir name
 444     wxSplitPath(strPath, &strDir, NULL, NULL);
 445
 446 #endif  // UNIX/Win
 447
 448   return strDir.c_str();
 449 }
 450
 451 wxChar *wxGetUserHome(const wxString& WXUNUSED(user))
 452 {
 453     // VZ: the old code here never worked for user != "" anyhow! Moreover, it
 454     //     returned sometimes a malloc()'d pointer, sometimes a pointer to a
 455     //     static buffer and sometimes I don't even know what.
 456     static wxString s_home;
 457
 458     return (wxChar *)wxGetHomeDir(&s_home);
 459 }
 460
 461 bool wxDirExists(const wxString& dir)
 462 {
 463 #ifdef __WXMICROWIN__
 464     return wxPathExist(dir);
 465 #elif defined(__WIN32__)
 466     DWORD attribs = GetFileAttributes(dir);
 467     return ((attribs != (DWORD)-1) && (attribs & FILE_ATTRIBUTE_DIRECTORY));
 468 #else // Win16
 469     #ifdef __BORLANDC__
 470         struct ffblk fileInfo;
 471     #else
 472         struct find_t fileInfo;
 473     #endif
 474     // In Borland findfirst has a different argument
 475     // ordering from _dos_findfirst. But _dos_findfirst
 476     // _should_ be ok in both MS and Borland... why not?
 477     #ifdef __BORLANDC__
 478         return (findfirst(dir, &fileInfo, _A_SUBDIR) == 0 &&
 479                (fileInfo.ff_attrib & _A_SUBDIR) != 0);
 480     #else
 481         return (_dos_findfirst(dir, _A_SUBDIR, &fileInfo) == 0) &&
 482                ((fileInfo.attrib & _A_SUBDIR) != 0);
 483     #endif
 484 #endif // Win32/16
 485 }
 486
 487 bool wxGetDiskSpace(const wxString& path, wxLongLong *pTotal, wxLongLong *pFree)
 488 {
 489 #ifdef __WXWINCE__
 490     return FALSE;
 491 #else
 492     if ( path.empty() )
 493         return FALSE;
 494
 495 // old w32api don't have ULARGE_INTEGER
 496 #if defined(__WIN32__) && \
 497     (!defined(__GNUWIN32__) || wxCHECK_W32API_VERSION( 0, 3 ))
 498     // GetDiskFreeSpaceEx() is not available under original Win95, check for
 499     // it
 500     typedef BOOL (WINAPI *GetDiskFreeSpaceEx_t)(LPCTSTR,
 501                                                 PULARGE_INTEGER,
 502                                                 PULARGE_INTEGER,
 503                                                 PULARGE_INTEGER);
 504
 505     GetDiskFreeSpaceEx_t
 506         pGetDiskFreeSpaceEx = (GetDiskFreeSpaceEx_t)::GetProcAddress
 507                               (
 508                                 ::GetModuleHandle(_T("kernel32.dll")),
 509 #if wxUSE_UNICODE
 510                                 "GetDiskFreeSpaceExW"
 511 #else
 512                                 "GetDiskFreeSpaceExA"
 513 #endif
 514                               );
 515
 516     if ( pGetDiskFreeSpaceEx )
 517     {
 518         ULARGE_INTEGER bytesFree, bytesTotal;
 519
 520         // may pass the path as is, GetDiskFreeSpaceEx() is smart enough
 521         if ( !pGetDiskFreeSpaceEx(path,
 522                                   &bytesFree,
 523                                   &bytesTotal,
 524                                   NULL) )
 525         {
 526             wxLogLastError(_T("GetDiskFreeSpaceEx"));
 527
 528             return FALSE;
 529         }
 530
 531         // ULARGE_INTEGER is a union of a 64 bit value and a struct containing
 532         // two 32 bit fields which may be or may be not named - try to make it
 533         // compile in all cases
 534 #if defined(__BORLANDC__) && !defined(_ANONYMOUS_STRUCT)
 535         #define UL(ul) ul.u
 536 #else // anon union
 537         #define UL(ul) ul
 538 #endif
 539         if ( pTotal )
 540         {
 541             *pTotal = wxLongLong(UL(bytesTotal).HighPart, UL(bytesTotal).LowPart);
 542         }
 543
 544         if ( pFree )
 545         {
 546             *pFree = wxLongLong(UL(bytesFree).HighPart, UL(bytesFree).LowPart);
 547         }
 548     }
 549     else
 550 #endif // Win32
 551     {
 552         // there's a problem with drives larger than 2GB, GetDiskFreeSpaceEx()
 553         // should be used instead - but if it's not available, fall back on
 554         // GetDiskFreeSpace() nevertheless...
 555
 556         DWORD lSectorsPerCluster,
 557               lBytesPerSector,
 558               lNumberOfFreeClusters,
 559               lTotalNumberOfClusters;
 560
 561         // FIXME: this is wrong, we should extract the root drive from path
 562         //        instead, but this is the job for wxFileName...
 563         if ( !::GetDiskFreeSpace(path,
 564                                  &lSectorsPerCluster,
 565                                  &lBytesPerSector,
 566                                  &lNumberOfFreeClusters,
 567                                  &lTotalNumberOfClusters) )
 568         {
 569             wxLogLastError(_T("GetDiskFreeSpace"));
 570
 571             return FALSE;
 572         }
 573
 574         wxLongLong lBytesPerCluster = lSectorsPerCluster;
 575         lBytesPerCluster *= lBytesPerSector;
 576
 577         if ( pTotal )
 578         {
 579             *pTotal = lBytesPerCluster;
 580             *pTotal *= lTotalNumberOfClusters;
 581         }
 582
 583         if ( pFree )
 584         {
 585             *pFree = lBytesPerCluster;
 586             *pFree *= lNumberOfFreeClusters;
 587         }
 588     }
 589
 590     return TRUE;
 591 #endif
 592     // __WXWINCE__
 593 }
 594
 595 // ----------------------------------------------------------------------------
 596 // env vars
 597 // ----------------------------------------------------------------------------
 598
 599 bool wxGetEnv(const wxString& var, wxString *value)
 600 {
 601 #ifdef __WXWINCE__
 602     return FALSE;
 603 #elif defined(__WIN16__)
 604     const wxChar* ret = wxGetenv(var);
 605     if ( !ret )
 606         return FALSE;
 607
 608     if ( value )
 609     {
 610         *value = ret;
 611     }
 612
 613     return TRUE;
 614 #else // Win32
 615     // first get the size of the buffer
 616     DWORD dwRet = ::GetEnvironmentVariable(var, NULL, 0);
 617     if ( !dwRet )
 618     {
 619         // this means that there is no such variable
 620         return FALSE;
 621     }
 622
 623     if ( value )
 624     {
 625         (void)::GetEnvironmentVariable(var, wxStringBuffer(*value, dwRet),
 626                                        dwRet);
 627     }
 628
 629     return TRUE;
 630 #endif // Win16/32
 631 }
 632
 633 bool wxSetEnv(const wxString& var, const wxChar *value)
 634 {
 635     // some compilers have putenv() or _putenv() or _wputenv() but it's better
 636     // to always use Win32 function directly instead of dealing with them
 637 #if defined(__WIN32__) && !defined(__WXWINCE__)
 638     if ( !::SetEnvironmentVariable(var, value) )
 639     {
 640         wxLogLastError(_T("SetEnvironmentVariable"));
 641
 642         return FALSE;
 643     }
 644
 645     return TRUE;
 646 #else // no way to set env vars
 647     return FALSE;
 648 #endif
 649 }
 650
 651 // ----------------------------------------------------------------------------
 652 // process management
 653 // ----------------------------------------------------------------------------
 654
 655 // structure used to pass parameters from wxKill() to wxEnumFindByPidProc()
 656 struct wxFindByPidParams
 657 {
 658     wxFindByPidParams() { hwnd = 0; pid = 0; }
 659
 660     // the HWND used to return the result
 661     HWND hwnd;
 662
 663     // the PID we're looking from
 664     DWORD pid;
 665
 666     DECLARE_NO_COPY_CLASS(wxFindByPidParams)
 667 };
 668
 669 // wxKill helper: EnumWindows() callback which is used to find the first (top
 670 // level) window belonging to the given process
 671 BOOL CALLBACK wxEnumFindByPidProc(HWND hwnd, LPARAM lParam)
 672 {
 673     DWORD pid;
 674     (void)::GetWindowThreadProcessId(hwnd, &pid);
 675
 676     wxFindByPidParams *params = (wxFindByPidParams *)lParam;
 677     if ( pid == params->pid )
 678     {
 679         // remember the window we found
 680         params->hwnd = hwnd;
 681
 682         // return FALSE to stop the enumeration
 683         return FALSE;
 684     }
 685
 686     // continue enumeration
 687     return TRUE;
 688 }
 689
 690 int wxKill(long pid, wxSignal sig, wxKillError *krc)
 691 {
 692     // get the process handle to operate on
 693     HANDLE hProcess = ::OpenProcess(SYNCHRONIZE |
 694                                     PROCESS_TERMINATE |
 695                                     PROCESS_QUERY_INFORMATION,
 696                                     FALSE, // not inheritable
 697                                     (DWORD)pid);
 698     if ( hProcess == NULL )
 699     {
 700         if ( krc )
 701         {
 702             if ( ::GetLastError() == ERROR_ACCESS_DENIED )
 703             {
 704                 *krc = wxKILL_ACCESS_DENIED;
 705             }
 706             else
 707             {
 708                 *krc = wxKILL_NO_PROCESS;
 709             }
 710         }
 711
 712         return -1;
 713     }
 714
 715     bool ok = TRUE;
 716     switch ( sig )
 717     {
 718         case wxSIGKILL:
 719             // kill the process forcefully returning -1 as error code
 720             if ( !::TerminateProcess(hProcess, (UINT)-1) )
 721             {
 722                 wxLogSysError(_("Failed to kill process %d"), pid);
 723
 724                 if ( krc )
 725                 {
 726                     // this is not supposed to happen if we could open the
 727                     // process
 728                     *krc = wxKILL_ERROR;
 729                 }
 730
 731                 ok = FALSE;
 732             }
 733             break;
 734
 735         case wxSIGNONE:
 736             // do nothing, we just want to test for process existence
 737             break;
 738
 739         default:
 740             // any other signal means "terminate"
 741             {
 742                 wxFindByPidParams params;
 743                 params.pid = (DWORD)pid;
 744
 745                 // EnumWindows() has nice semantics: it returns 0 if it found
 746                 // something or if an error occured and non zero if it
 747                 // enumerated all the window
 748                 if ( !::EnumWindows(wxEnumFindByPidProc, (LPARAM)&params) )
 749                 {
 750                     // did we find any window?
 751                     if ( params.hwnd )
 752                     {
 753                         // tell the app to close
 754                         //
 755                         // NB: this is the harshest way, the app won't have
 756                         //     opportunity to save any files, for example, but
 757                         //     this is probably what we want here. If not we
 758                         //     can also use SendMesageTimeout(WM_CLOSE)
 759                         if ( !::PostMessage(params.hwnd, WM_QUIT, 0, 0) )
 760                         {
 761                             wxLogLastError(_T("PostMessage(WM_QUIT)"));
 762                         }
 763                     }
 764                     else // it was an error then
 765                     {
 766                         wxLogLastError(_T("EnumWindows"));
 767
 768                         ok = FALSE;
 769                     }
 770                 }
 771                 else // no windows for this PID
 772                 {
 773                     if ( krc )
 774                     {
 775                         *krc = wxKILL_ERROR;
 776                     }
 777
 778                     ok = FALSE;
 779                 }
 780             }
 781     }
 782
 783     // the return code
 784     DWORD rc;
 785
 786     if ( ok )
 787     {
 788         // as we wait for a short time, we can use just WaitForSingleObject()
 789         // and not MsgWaitForMultipleObjects()
 790         switch ( ::WaitForSingleObject(hProcess, 500 /* msec */) )
 791         {
 792             case WAIT_OBJECT_0:
 793                 // process terminated
 794                 if ( !::GetExitCodeProcess(hProcess, &rc) )
 795                 {
 796                     wxLogLastError(_T("GetExitCodeProcess"));
 797                 }
 798                 break;
 799
 800             default:
 801                 wxFAIL_MSG( _T("unexpected WaitForSingleObject() return") );
 802                 // fall through
 803
 804             case WAIT_FAILED:
 805                 wxLogLastError(_T("WaitForSingleObject"));
 806                 // fall through
 807
 808             case WAIT_TIMEOUT:
 809                 if ( krc )
 810                 {
 811                     *krc = wxKILL_ERROR;
 812                 }
 813
 814                 rc = STILL_ACTIVE;
 815                 break;
 816         }
 817     }
 818     else // !ok
 819     {
 820         // just to suppress the warnings about uninitialized variable
 821         rc = 0;
 822     }
 823
 824     ::CloseHandle(hProcess);
 825
 826     // the return code is the same as from Unix kill(): 0 if killed
 827     // successfully or -1 on error
 828     //
 829     // be careful to interpret rc correctly: for wxSIGNONE we return success if
 830     // the process exists, for all the other sig values -- if it doesn't
 831     if ( ok &&
 832             ((sig == wxSIGNONE) == (rc == STILL_ACTIVE)) )
 833     {
 834         if ( krc )
 835         {
 836             *krc = wxKILL_OK;
 837         }
 838
 839         return 0;
 840     }
 841
 842     // error
 843     return -1;
 844 }
 845
 846 // Execute a program in an Interactive Shell
 847 bool wxShell(const wxString& command)
 848 {
 849 #ifdef __WXWINCE__
 850     return FALSE;
 851 #else
 852     wxChar *shell = wxGetenv(wxT("COMSPEC"));
 853     if ( !shell )
 854         shell = (wxChar*) wxT("\\COMMAND.COM");
 855
 856     wxString cmd;
 857     if ( !command )
 858     {
 859         // just the shell
 860         cmd = shell;
 861     }
 862     else
 863     {
 864         // pass the command to execute to the command processor
 865         cmd.Printf(wxT("%s /c %s"), shell, command.c_str());
 866     }
 867
 868     return wxExecute(cmd, wxEXEC_SYNC) == 0;
 869 #endif
 870 }
 871
 872 // Shutdown or reboot the PC
 873 bool wxShutdown(wxShutdownFlags wFlags)
 874 {
 875 #ifdef __WXWINCE__
 876     return FALSE;
 877 #elif defined(__WIN32__)
 878     bool bOK = TRUE;
 879
 880     if ( wxGetOsVersion(NULL, NULL) == wxWINDOWS_NT ) // if is NT or 2K
 881     {
 882         // Get a token for this process.
 883         HANDLE hToken;
 884         bOK = ::OpenProcessToken(GetCurrentProcess(),
 885                                  TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY,
 886                                  &hToken) != 0;
 887         if ( bOK )
 888         {
 889             TOKEN_PRIVILEGES tkp;
 890
 891             // Get the LUID for the shutdown privilege.
 892             ::LookupPrivilegeValue(NULL, SE_SHUTDOWN_NAME,
 893                                    &tkp.Privileges[0].Luid);
 894
 895             tkp.PrivilegeCount = 1;  // one privilege to set
 896             tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
 897
 898             // Get the shutdown privilege for this process.
 899             ::AdjustTokenPrivileges(hToken, FALSE, &tkp, 0,
 900                                     (PTOKEN_PRIVILEGES)NULL, 0);
 901
 902             // Cannot test the return value of AdjustTokenPrivileges.
 903             bOK = ::GetLastError() == ERROR_SUCCESS;
 904         }
 905     }
 906
 907     if ( bOK )
 908     {
 909         UINT flags = EWX_SHUTDOWN | EWX_FORCE;
 910         switch ( wFlags )
 911         {
 912             case wxSHUTDOWN_POWEROFF:
 913                 flags |= EWX_POWEROFF;
 914                 break;
 915
 916             case wxSHUTDOWN_REBOOT:
 917                 flags |= EWX_REBOOT;
 918                 break;
 919
 920             default:
 921                 wxFAIL_MSG( _T("unknown wxShutdown() flag") );
 922                 return FALSE;
 923         }
 924
 925         bOK = ::ExitWindowsEx(EWX_SHUTDOWN | EWX_FORCE | EWX_REBOOT, 0) != 0;
 926     }
 927
 928     return bOK;
 929 #else // Win16
 930     return FALSE;
 931 #endif // Win32/16
 932 }
 933
 934 // ----------------------------------------------------------------------------
 935 // misc
 936 // ----------------------------------------------------------------------------
 937
 938 // Get free memory in bytes, or -1 if cannot determine amount (e.g. on UNIX)
 939 long wxGetFreeMemory()
 940 {
 941 #if defined(__WIN32__) && !defined(__BORLANDC__)
 942     MEMORYSTATUS memStatus;
 943     memStatus.dwLength = sizeof(MEMORYSTATUS);
 944     GlobalMemoryStatus(&memStatus);
 945     return memStatus.dwAvailPhys;
 946 #else
 947     return (long)GetFreeSpace(0);
 948 #endif
 949 }
 950
 951 unsigned long wxGetProcessId()
 952 {
 953 #ifdef __WIN32__
 954     return ::GetCurrentProcessId();
 955 #else
 956     return 0;
 957 #endif
 958 }
 959
 960 // Emit a beeeeeep
 961 void wxBell()
 962 {
 963     ::MessageBeep((UINT)-1);        // default sound
 964 }
 965
 966 wxString wxGetOsDescription()
 967 {
 968 #ifdef __WIN32__
 969     wxString str;
 970
 971     OSVERSIONINFO info;
 972     wxZeroMemory(info);
 973
 974     info.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);
 975     if ( ::GetVersionEx(&info) )
 976     {
 977         switch ( info.dwPlatformId )
 978         {
 979             case VER_PLATFORM_WIN32s:
 980                 str = _("Win32s on Windows 3.1");
 981                 break;
 982
 983             case VER_PLATFORM_WIN32_WINDOWS:
 984                 str.Printf(_("Windows 9%c"),
 985                            info.dwMinorVersion == 0 ? _T('5') : _T('8'));
 986                 if ( !wxIsEmpty(info.szCSDVersion) )
 987                 {
 988                     str << _T(" (") << info.szCSDVersion << _T(')');
 989                 }
 990                 break;
 991
 992             case VER_PLATFORM_WIN32_NT:
 993                 str.Printf(_T("Windows NT %lu.%lu (build %lu"),
 994                            info.dwMajorVersion,
 995                            info.dwMinorVersion,
 996                            info.dwBuildNumber);
 997                 if ( !wxIsEmpty(info.szCSDVersion) )
 998                 {
 999                     str << _T(", ") << info.szCSDVersion;
1000                 }
1001                 str << _T(')');
1002                 break;
1003         }
1004     }
1005     else
1006     {
1007         wxFAIL_MSG( _T("GetVersionEx() failed") ); // should never happen
1008     }
1009
1010     return str;
1011 #else // Win16
1012     return _("Windows 3.1");
1013 #endif // Win32/16
1014 }
1015
1016 wxToolkitInfo& wxAppTraits::GetToolkitInfo()
1017 {
1018     // cache the version info, it's not going to change
1019     //
1020     // NB: this is MT-safe, we may use these static vars from different threads
1021     //     but as they always have the same value it doesn't matter
1022     static int s_ver = -1,
1023                s_major = -1,
1024                s_minor = -1;
1025
1026     if ( s_ver == -1 )
1027     {
1028         OSVERSIONINFO info;
1029         wxZeroMemory(info);
1030
1031         s_ver = wxWINDOWS;
1032         info.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);
1033         if ( ::GetVersionEx(&info) )
1034         {
1035             s_major = info.dwMajorVersion;
1036             s_minor = info.dwMinorVersion;
1037
1038             switch ( info.dwPlatformId )
1039             {
1040                 case VER_PLATFORM_WIN32s:
1041                     s_ver = wxWIN32S;
1042                     break;
1043
1044                 case VER_PLATFORM_WIN32_WINDOWS:
1045                     s_ver = wxWIN95;
1046                     break;
1047
1048                 case VER_PLATFORM_WIN32_NT:
1049                     s_ver = wxWINDOWS_NT;
1050                     break;
1051 #ifdef __WXWINCE__
1052                 case VER_PLATFORM_WIN32_CE:
1053                     s_ver = wxWINDOWS_CE;
1054                     break;
1055 #endif
1056             }
1057         }
1058     }
1059
1060     static wxToolkitInfo info;
1061     info.versionMajor = s_major;
1062     info.versionMinor = s_minor;
1063     info.os = s_ver;
1064     info.name = _T("wxBase");
1065     return info;
1066 }
1067
1068 // ----------------------------------------------------------------------------
1069 // sleep functions
1070 // ----------------------------------------------------------------------------
1071
1072 void wxUsleep(unsigned long milliseconds)
1073 {
1074     ::Sleep(milliseconds);
1075 }
1076
1077 void wxSleep(int nSecs)
1078 {
1079     wxUsleep(1000*nSecs);
1080 }
1081
1082 // ----------------------------------------------------------------------------
1083 // font encoding <-> Win32 codepage conversion functions
1084 // ----------------------------------------------------------------------------
1085
1086 extern WXDLLIMPEXP_BASE long wxEncodingToCharset(wxFontEncoding encoding)
1087 {
1088     switch ( encoding )
1089     {
1090         // although this function is supposed to return an exact match, do do
1091         // some mappings here for the most common case of "standard" encoding
1092         case wxFONTENCODING_SYSTEM:
1093             return DEFAULT_CHARSET;
1094
1095         case wxFONTENCODING_ISO8859_1:
1096         case wxFONTENCODING_ISO8859_15:
1097         case wxFONTENCODING_CP1252:
1098             return ANSI_CHARSET;
1099
1100 #if !defined(__WXMICROWIN__)
1101         // The following four fonts are multi-byte charsets
1102         case wxFONTENCODING_CP932:
1103             return SHIFTJIS_CHARSET;
1104
1105         case wxFONTENCODING_CP936:
1106             return GB2312_CHARSET;
1107
1108         case wxFONTENCODING_CP949:
1109             return HANGUL_CHARSET;
1110
1111         case wxFONTENCODING_CP950:
1112             return CHINESEBIG5_CHARSET;
1113
1114         // The rest are single byte encodings
1115         case wxFONTENCODING_CP1250:
1116             return EASTEUROPE_CHARSET;
1117
1118         case wxFONTENCODING_CP1251:
1119             return RUSSIAN_CHARSET;
1120
1121         case wxFONTENCODING_CP1253:
1122             return GREEK_CHARSET;
1123
1124         case wxFONTENCODING_CP1254:
1125             return TURKISH_CHARSET;
1126
1127         case wxFONTENCODING_CP1255:
1128             return HEBREW_CHARSET;
1129
1130         case wxFONTENCODING_CP1256:
1131             return ARABIC_CHARSET;
1132
1133         case wxFONTENCODING_CP1257:
1134             return BALTIC_CHARSET;
1135
1136         case wxFONTENCODING_CP874:
1137             return THAI_CHARSET;
1138 #endif // !__WXMICROWIN__
1139
1140         case wxFONTENCODING_CP437:
1141             return OEM_CHARSET;
1142
1143         default:
1144             // no way to translate this encoding into a Windows charset
1145             return -1;
1146     }
1147 }
1148
1149 // we have 2 versions of wxCharsetToCodepage(): the old one which directly
1150 // looks up the vlaues in the registry and the new one which is more
1151 // politically correct and has more chances to work on other Windows versions
1152 // as well but the old version is still needed for !wxUSE_FONTMAP case
1153 #if wxUSE_FONTMAP
1154
1155 #include "wx/fontmap.h"
1156
1157 extern WXDLLIMPEXP_BASE long wxEncodingToCodepage(wxFontEncoding encoding)
1158 {
1159     // translate encoding into the Windows CHARSET
1160     long charset = wxEncodingToCharset(encoding);
1161     if ( charset == -1 )
1162         return -1;
1163
1164     // translate CHARSET to code page
1165     CHARSETINFO csetInfo;
1166     if ( !::TranslateCharsetInfo((DWORD *)(DWORD)charset,
1167                                  &csetInfo,
1168                                  TCI_SRCCHARSET) )
1169     {
1170         wxLogLastError(_T("TranslateCharsetInfo(TCI_SRCCHARSET)"));
1171
1172         return -1;
1173     }
1174
1175     return csetInfo.ciACP;
1176 }
1177
1178 extern long wxCharsetToCodepage(const wxChar *name)
1179 {
1180     // first get the font encoding for this charset
1181     if ( !name )
1182         return -1;
1183
1184     wxFontEncoding enc = wxFontMapper::Get()->CharsetToEncoding(name, FALSE);
1185     if ( enc == wxFONTENCODING_SYSTEM )
1186         return -1;
1187
1188     // the use the helper function
1189     return wxEncodingToCodepage(enc);
1190 }
1191
1192 #else // !wxUSE_FONTMAP
1193
1194 #include "wx/msw/registry.h"
1195
1196 // this should work if Internet Exploiter is installed
1197 extern long wxCharsetToCodepage(const wxChar *name)
1198 {
1199     if (!name)
1200         return GetACP();
1201
1202     long CP = -1;
1203
1204     wxString path(wxT("MIME\\Database\\Charset\\"));
1205     wxString cn(name);
1206
1207     // follow the alias loop
1208     for ( ;; )
1209     {
1210         wxRegKey key(wxRegKey::HKCR, path + cn);
1211
1212         if (!key.Exists())
1213             break;
1214
1215         // two cases: either there's an AliasForCharset string,
1216         // or there are Codepage and InternetEncoding dwords.
1217         // The InternetEncoding gives us the actual encoding,
1218         // the Codepage just says which Windows character set to
1219         // use when displaying the data.
1220         if (key.HasValue(wxT("InternetEncoding")) &&
1221             key.QueryValue(wxT("InternetEncoding"), &CP))
1222             break;
1223
1224         // no encoding, see if it's an alias
1225         if (!key.HasValue(wxT("AliasForCharset")) ||
1226             !key.QueryValue(wxT("AliasForCharset"), cn))
1227             break;
1228     }
1229
1230     return CP;
1231 }
1232
1233 #endif // wxUSE_FONTMAP/!wxUSE_FONTMAP
1234