]>
Commit | Line | Data |
---|---|---|
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)¶ms) ) | |
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 |