version-0.73 (was Re: Change 31920: Don't use ~0 as a version
[p5sagit/p5-mst-13.2.git] / ext / Win32 / Win32.xs
CommitLineData
34f7f30d 1#include <wctype.h>
b4ad57f4 2#include <windows.h>
34f7f30d 3#include <shlobj.h>
b4ad57f4 4
34f7f30d 5#define PERL_NO_GET_CONTEXT
b4ad57f4 6#include "EXTERN.h"
7#include "perl.h"
8#include "XSUB.h"
9
34f7f30d 10#ifndef countof
11# define countof(array) (sizeof (array) / sizeof (*(array)))
12#endif
13
b4ad57f4 14#define SE_SHUTDOWN_NAMEA "SeShutdownPrivilege"
b4ad57f4 15
34f7f30d 16#ifndef WC_NO_BEST_FIT_CHARS
17# define WC_NO_BEST_FIT_CHARS 0x00000400
18#endif
19
20#define GETPROC(fn) pfn##fn = (PFN##fn)GetProcAddress(module, #fn)
21
22typedef BOOL (WINAPI *PFNSHGetSpecialFolderPathA)(HWND, char*, int, BOOL);
23typedef BOOL (WINAPI *PFNSHGetSpecialFolderPathW)(HWND, WCHAR*, int, BOOL);
24typedef HRESULT (WINAPI *PFNSHGetFolderPathA)(HWND, int, HANDLE, DWORD, LPTSTR);
25typedef HRESULT (WINAPI *PFNSHGetFolderPathW)(HWND, int, HANDLE, DWORD, LPWSTR);
26typedef BOOL (WINAPI *PFNCreateEnvironmentBlock)(void**, HANDLE, BOOL);
27typedef BOOL (WINAPI *PFNDestroyEnvironmentBlock)(void*);
5d477a6d 28typedef int (__stdcall *PFNDllRegisterServer)(void);
29typedef int (__stdcall *PFNDllUnregisterServer)(void);
34f7f30d 30typedef DWORD (__stdcall *PFNNetApiBufferFree)(void*);
31typedef DWORD (__stdcall *PFNNetWkstaGetInfo)(LPWSTR, DWORD, void*);
32
33typedef BOOL (__stdcall *PFNOpenProcessToken)(HANDLE, DWORD, HANDLE*);
34typedef BOOL (__stdcall *PFNOpenThreadToken)(HANDLE, DWORD, BOOL, HANDLE*);
35typedef BOOL (__stdcall *PFNGetTokenInformation)(HANDLE, TOKEN_INFORMATION_CLASS, void*, DWORD, DWORD*);
36typedef BOOL (__stdcall *PFNAllocateAndInitializeSid)(PSID_IDENTIFIER_AUTHORITY, BYTE, DWORD, DWORD,
37 DWORD, DWORD, DWORD, DWORD, DWORD, DWORD, PSID*);
38typedef BOOL (__stdcall *PFNEqualSid)(PSID, PSID);
39typedef void* (__stdcall *PFNFreeSid)(PSID);
40typedef BOOL (__stdcall *PFNIsUserAnAdmin)();
41
9c6a461f 42#ifndef CSIDL_MYMUSIC
43# define CSIDL_MYMUSIC 0x000D
b4ad57f4 44#endif
9c6a461f 45#ifndef CSIDL_MYVIDEO
46# define CSIDL_MYVIDEO 0x000E
47#endif
48#ifndef CSIDL_LOCAL_APPDATA
34f7f30d 49# define CSIDL_LOCAL_APPDATA 0x001C
9c6a461f 50#endif
51#ifndef CSIDL_COMMON_FAVORITES
52# define CSIDL_COMMON_FAVORITES 0x001F
53#endif
54#ifndef CSIDL_INTERNET_CACHE
55# define CSIDL_INTERNET_CACHE 0x0020
56#endif
57#ifndef CSIDL_COOKIES
58# define CSIDL_COOKIES 0x0021
59#endif
60#ifndef CSIDL_HISTORY
61# define CSIDL_HISTORY 0x0022
62#endif
63#ifndef CSIDL_COMMON_APPDATA
64# define CSIDL_COMMON_APPDATA 0x0023
65#endif
66#ifndef CSIDL_WINDOWS
67# define CSIDL_WINDOWS 0x0024
68#endif
69#ifndef CSIDL_PROGRAM_FILES
70# define CSIDL_PROGRAM_FILES 0x0026
71#endif
72#ifndef CSIDL_MYPICTURES
34f7f30d 73# define CSIDL_MYPICTURES 0x0027
9c6a461f 74#endif
75#ifndef CSIDL_PROFILE
34f7f30d 76# define CSIDL_PROFILE 0x0028
9c6a461f 77#endif
78#ifndef CSIDL_PROGRAM_FILES_COMMON
34f7f30d 79# define CSIDL_PROGRAM_FILES_COMMON 0x002B
34f7f30d 80#endif
9c6a461f 81#ifndef CSIDL_COMMON_TEMPLATES
82# define CSIDL_COMMON_TEMPLATES 0x002D
83#endif
84#ifndef CSIDL_COMMON_DOCUMENTS
85# define CSIDL_COMMON_DOCUMENTS 0x002E
86#endif
87#ifndef CSIDL_COMMON_ADMINTOOLS
88# define CSIDL_COMMON_ADMINTOOLS 0x002F
89#endif
90#ifndef CSIDL_ADMINTOOLS
91# define CSIDL_ADMINTOOLS 0x0030
34f7f30d 92#endif
34f7f30d 93#ifndef CSIDL_COMMON_MUSIC
94# define CSIDL_COMMON_MUSIC 0x0035
95#endif
34f7f30d 96#ifndef CSIDL_COMMON_PICTURES
97# define CSIDL_COMMON_PICTURES 0x0036
98#endif
34f7f30d 99#ifndef CSIDL_COMMON_VIDEO
100# define CSIDL_COMMON_VIDEO 0x0037
101#endif
9c6a461f 102#ifndef CSIDL_CDBURN_AREA
103# define CSIDL_CDBURN_AREA 0x003B
34f7f30d 104#endif
9c6a461f 105#ifndef CSIDL_FLAG_CREATE
106# define CSIDL_FLAG_CREATE 0x8000
34f7f30d 107#endif
108
109/* Use explicit struct definition because wSuiteMask and
110 * wProductType are not defined in the VC++ 6.0 headers.
111 * WORD type has been replaced by unsigned short because
112 * WORD is already used by Perl itself.
113 */
114struct {
115 DWORD dwOSVersionInfoSize;
116 DWORD dwMajorVersion;
117 DWORD dwMinorVersion;
118 DWORD dwBuildNumber;
119 DWORD dwPlatformId;
120 CHAR szCSDVersion[128];
121 unsigned short wServicePackMajor;
122 unsigned short wServicePackMinor;
123 unsigned short wSuiteMask;
124 BYTE wProductType;
125 BYTE wReserved;
126} g_osver = {0, 0, 0, 0, 0, "", 0, 0, 0, 0, 0};
127BOOL g_osver_ex = TRUE;
cf8b4e93 128
129#define ONE_K_BUFSIZE 1024
130
131int
132IsWin95(void)
133{
134 return (g_osver.dwPlatformId == VER_PLATFORM_WIN32_WINDOWS);
135}
136
137int
138IsWinNT(void)
139{
140 return (g_osver.dwPlatformId == VER_PLATFORM_WIN32_NT);
141}
142
34f7f30d 143int
144IsWin2000(void)
145{
146 return (g_osver.dwMajorVersion > 4);
147}
5bd7e777 148
34f7f30d 149/* Convert SV to wide character string. The return value must be
150 * freed using Safefree().
151 */
152WCHAR*
153sv_to_wstr(pTHX_ SV *sv)
5bd7e777 154{
34f7f30d 155 DWORD wlen;
156 WCHAR *wstr;
157 STRLEN len;
158 char *str = SvPV(sv, len);
159 UINT cp = SvUTF8(sv) ? CP_UTF8 : CP_ACP;
160
161 wlen = MultiByteToWideChar(cp, 0, str, len+1, NULL, 0);
162 New(0, wstr, wlen, WCHAR);
163 MultiByteToWideChar(cp, 0, str, len+1, wstr, wlen);
164
165 return wstr;
166}
167
168/* Convert wide character string to mortal SV. Use UTF8 encoding
169 * if the string cannot be represented in the system codepage.
170 */
171SV *
172wstr_to_sv(pTHX_ WCHAR *wstr)
173{
174 size_t wlen = wcslen(wstr)+1;
175 BOOL use_default = FALSE;
176 int len = WideCharToMultiByte(CP_ACP, WC_NO_BEST_FIT_CHARS, wstr, wlen, NULL, 0, NULL, NULL);
177 SV *sv = sv_2mortal(newSV(len));
178
179 len = WideCharToMultiByte(CP_ACP, WC_NO_BEST_FIT_CHARS, wstr, wlen, SvPVX(sv), len, NULL, &use_default);
180 if (use_default) {
181 len = WideCharToMultiByte(CP_UTF8, 0, wstr, wlen, NULL, 0, NULL, NULL);
182 sv_grow(sv, len);
183 len = WideCharToMultiByte(CP_UTF8, 0, wstr, wlen, SvPVX(sv), len, NULL, NULL);
184 SvUTF8_on(sv);
5bd7e777 185 }
34f7f30d 186 /* Shouldn't really ever fail since we ask for the required length first, but who knows... */
187 if (len) {
188 SvPOK_on(sv);
189 SvCUR_set(sv, len-1);
5bd7e777 190 }
34f7f30d 191 return sv;
192}
5bd7e777 193
34f7f30d 194/* Retrieve a variable from the Unicode environment in a mortal SV.
195 *
196 * Recreates the Unicode environment because a bug in earlier Perl versions
197 * overwrites it with the ANSI version, which contains replacement
198 * characters for the characters not in the ANSI codepage.
199 */
200SV*
201get_unicode_env(pTHX_ WCHAR *name)
202{
203 SV *sv = NULL;
204 void *env;
205 HANDLE token;
206 HMODULE module;
207 PFNOpenProcessToken pfnOpenProcessToken;
5bd7e777 208
34f7f30d 209 /* Get security token for the current process owner */
210 module = LoadLibrary("advapi32.dll");
211 if (!module)
212 return NULL;
213
214 GETPROC(OpenProcessToken);
215
216 if (pfnOpenProcessToken == NULL ||
217 !pfnOpenProcessToken(GetCurrentProcess(), TOKEN_QUERY | TOKEN_DUPLICATE, &token))
218 {
219 FreeLibrary(module);
220 return NULL;
5bd7e777 221 }
34f7f30d 222 FreeLibrary(module);
223
224 /* Create a Unicode environment block for this process */
225 module = LoadLibrary("userenv.dll");
226 if (module) {
227 PFNCreateEnvironmentBlock pfnCreateEnvironmentBlock;
228 PFNDestroyEnvironmentBlock pfnDestroyEnvironmentBlock;
229
230 GETPROC(CreateEnvironmentBlock);
231 GETPROC(DestroyEnvironmentBlock);
232
233 if (pfnCreateEnvironmentBlock && pfnDestroyEnvironmentBlock &&
234 pfnCreateEnvironmentBlock(&env, token, FALSE))
235 {
236 size_t name_len = wcslen(name);
237 WCHAR *entry = env;
238 while (*entry) {
239 size_t i;
240 size_t entry_len = wcslen(entry);
241 BOOL equal = (entry_len > name_len) && (entry[name_len] == '=');
242
243 for (i=0; equal && i < name_len; ++i)
244 equal = (towupper(entry[i]) == towupper(name[i]));
245
246 if (equal) {
247 sv = wstr_to_sv(aTHX_ entry+name_len+1);
248 break;
249 }
250 entry += entry_len+1;
251 }
252 pfnDestroyEnvironmentBlock(env);
253 }
254 FreeLibrary(module);
255 }
256 CloseHandle(token);
257 return sv;
5bd7e777 258}
259
34f7f30d 260/* Define both an ANSI and a Wide version of win32_longpath */
261
262#define CHAR_T char
263#define WIN32_FIND_DATA_T WIN32_FIND_DATAA
264#define FN_FINDFIRSTFILE FindFirstFileA
265#define FN_STRLEN strlen
266#define FN_STRCPY strcpy
267#define LONGPATH my_longpathA
268#include "longpath.inc"
269
270#define CHAR_T WCHAR
271#define WIN32_FIND_DATA_T WIN32_FIND_DATAW
272#define FN_FINDFIRSTFILE FindFirstFileW
273#define FN_STRLEN wcslen
274#define FN_STRCPY wcscpy
275#define LONGPATH my_longpathW
276#include "longpath.inc"
277
278/* The my_ansipath() function takes a Unicode filename and converts it
279 * into the current Windows codepage. If some characters cannot be mapped,
280 * then it will convert the short name instead.
281 *
282 * The buffer to the ansi pathname must be freed with Safefree() when it
283 * it no longer needed.
284 *
285 * The argument to my_ansipath() must exist before this function is
286 * called; otherwise there is no way to determine the short path name.
287 *
288 * Ideas for future refinement:
289 * - Only convert those segments of the path that are not in the current
290 * codepage, but leave the other segments in their long form.
291 * - If the resulting name is longer than MAX_PATH, start converting
292 * additional path segments into short names until the full name
293 * is shorter than MAX_PATH. Shorten the filename part last!
294 */
295
296/* This is a modified version of core Perl win32/win32.c(win32_ansipath).
297 * It uses New() etc. instead of win32_malloc().
298 */
299
300char *
301my_ansipath(const WCHAR *widename)
302{
303 char *name;
304 BOOL use_default = FALSE;
305 size_t widelen = wcslen(widename)+1;
306 int len = WideCharToMultiByte(CP_ACP, WC_NO_BEST_FIT_CHARS, widename, widelen,
307 NULL, 0, NULL, NULL);
308 New(0, name, len, char);
309 WideCharToMultiByte(CP_ACP, WC_NO_BEST_FIT_CHARS, widename, widelen,
310 name, len, NULL, &use_default);
311 if (use_default) {
312 DWORD shortlen = GetShortPathNameW(widename, NULL, 0);
313 if (shortlen) {
314 WCHAR *shortname;
315 New(0, shortname, shortlen, WCHAR);
316 shortlen = GetShortPathNameW(widename, shortname, shortlen)+1;
317
318 len = WideCharToMultiByte(CP_ACP, WC_NO_BEST_FIT_CHARS, shortname, shortlen,
319 NULL, 0, NULL, NULL);
320 Renew(name, len, char);
321 WideCharToMultiByte(CP_ACP, WC_NO_BEST_FIT_CHARS, shortname, shortlen,
322 name, len, NULL, NULL);
323 Safefree(shortname);
324 }
325 }
326 return name;
327}
328
329/* Convert wide character path to ANSI path and return as mortal SV. */
330SV*
331wstr_to_ansipath(pTHX_ WCHAR *wstr)
332{
333 char *ansi = my_ansipath(wstr);
334 SV *sv = sv_2mortal(newSVpvn(ansi, strlen(ansi)));
335 Safefree(ansi);
336 return sv;
337}
338
339#ifdef __CYGWIN__
340
5bd7e777 341char*
342get_childdir(void)
343{
344 dTHX;
345 char* ptr;
5bd7e777 346
34f7f30d 347 if (IsWin2000()) {
348 WCHAR filename[MAX_PATH+1];
349 GetCurrentDirectoryW(MAX_PATH+1, filename);
350 ptr = my_ansipath(filename);
351 }
352 else {
353 char filename[MAX_PATH+1];
354 GetCurrentDirectoryA(MAX_PATH+1, filename);
355 New(0, ptr, strlen(filename)+1, char);
356 strcpy(ptr, filename);
357 }
5bd7e777 358 return ptr;
359}
360
361void
34f7f30d 362free_childdir(char *d)
5bd7e777 363{
364 dTHX;
365 Safefree(d);
366}
367
368void*
369get_childenv(void)
370{
371 return NULL;
372}
373
374void
34f7f30d 375free_childenv(void *d)
5bd7e777 376{
377}
378
379# define PerlDir_mapA(dir) (dir)
380
381#endif
382
b4ad57f4 383XS(w32_ExpandEnvironmentStrings)
384{
385 dXSARGS;
b4ad57f4 386
387 if (items != 1)
388 croak("usage: Win32::ExpandEnvironmentStrings($String);\n");
389
34f7f30d 390 if (IsWin2000()) {
391 WCHAR value[31*1024];
392 WCHAR *source = sv_to_wstr(aTHX_ ST(0));
393 ExpandEnvironmentStringsW(source, value, countof(value)-1);
394 ST(0) = wstr_to_sv(aTHX_ value);
395 Safefree(source);
396 XSRETURN(1);
397 }
398 else {
399 char value[31*1024];
400 ExpandEnvironmentStringsA(SvPV_nolen(ST(0)), value, countof(value)-2);
401 XSRETURN_PV(value);
402 }
b4ad57f4 403}
404
405XS(w32_IsAdminUser)
406{
407 dXSARGS;
34f7f30d 408 HMODULE module;
409 PFNIsUserAnAdmin pfnIsUserAnAdmin;
410 PFNOpenThreadToken pfnOpenThreadToken;
411 PFNOpenProcessToken pfnOpenProcessToken;
412 PFNGetTokenInformation pfnGetTokenInformation;
413 PFNAllocateAndInitializeSid pfnAllocateAndInitializeSid;
414 PFNEqualSid pfnEqualSid;
415 PFNFreeSid pfnFreeSid;
b4ad57f4 416 HANDLE hTok;
417 DWORD dwTokInfoLen;
418 TOKEN_GROUPS *lpTokInfo;
419 SID_IDENTIFIER_AUTHORITY NtAuth = SECURITY_NT_AUTHORITY;
420 PSID pAdminSid;
421 int iRetVal;
422 unsigned int i;
b4ad57f4 423
424 if (items)
425 croak("usage: Win32::IsAdminUser()");
426
427 /* There is no concept of "Administrator" user accounts on Win9x systems,
428 so just return true. */
34f7f30d 429 if (IsWin95())
b4ad57f4 430 XSRETURN_YES;
431
34f7f30d 432 /* Use IsUserAnAdmin() when available. On Vista this will only return TRUE
433 * if the process is running with elevated privileges and not just when the
434 * process owner is a member of the "Administrators" group.
435 */
436 module = LoadLibrary("shell32.dll");
437 if (module) {
438 GETPROC(IsUserAnAdmin);
439 if (pfnIsUserAnAdmin) {
440 EXTEND(SP, 1);
441 ST(0) = sv_2mortal(newSViv(pfnIsUserAnAdmin() ? 1 : 0));
442 FreeLibrary(module);
443 XSRETURN(1);
444 }
445 FreeLibrary(module);
446 }
447
448 module = LoadLibrary("advapi32.dll");
449 if (!module) {
b4ad57f4 450 warn("Cannot load advapi32.dll library");
451 XSRETURN_UNDEF;
452 }
453
34f7f30d 454 GETPROC(OpenThreadToken);
455 GETPROC(OpenProcessToken);
456 GETPROC(GetTokenInformation);
457 GETPROC(AllocateAndInitializeSid);
458 GETPROC(EqualSid);
459 GETPROC(FreeSid);
b4ad57f4 460
461 if (!(pfnOpenThreadToken && pfnOpenProcessToken &&
462 pfnGetTokenInformation && pfnAllocateAndInitializeSid &&
463 pfnEqualSid && pfnFreeSid))
464 {
465 warn("Cannot load functions from advapi32.dll library");
34f7f30d 466 FreeLibrary(module);
b4ad57f4 467 XSRETURN_UNDEF;
468 }
469
470 if (!pfnOpenThreadToken(GetCurrentThread(), TOKEN_QUERY, FALSE, &hTok)) {
471 if (!pfnOpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &hTok)) {
472 warn("Cannot open thread token or process token");
34f7f30d 473 FreeLibrary(module);
b4ad57f4 474 XSRETURN_UNDEF;
475 }
476 }
477
478 pfnGetTokenInformation(hTok, TokenGroups, NULL, 0, &dwTokInfoLen);
479 if (!New(1, lpTokInfo, dwTokInfoLen, TOKEN_GROUPS)) {
480 warn("Cannot allocate token information structure");
481 CloseHandle(hTok);
34f7f30d 482 FreeLibrary(module);
b4ad57f4 483 XSRETURN_UNDEF;
484 }
485
486 if (!pfnGetTokenInformation(hTok, TokenGroups, lpTokInfo, dwTokInfoLen,
487 &dwTokInfoLen))
488 {
489 warn("Cannot get token information");
490 Safefree(lpTokInfo);
491 CloseHandle(hTok);
34f7f30d 492 FreeLibrary(module);
b4ad57f4 493 XSRETURN_UNDEF;
494 }
495
496 if (!pfnAllocateAndInitializeSid(&NtAuth, 2, SECURITY_BUILTIN_DOMAIN_RID,
497 DOMAIN_ALIAS_RID_ADMINS, 0, 0, 0, 0, 0, 0, &pAdminSid))
498 {
499 warn("Cannot allocate administrators' SID");
500 Safefree(lpTokInfo);
501 CloseHandle(hTok);
34f7f30d 502 FreeLibrary(module);
b4ad57f4 503 XSRETURN_UNDEF;
504 }
505
506 iRetVal = 0;
507 for (i = 0; i < lpTokInfo->GroupCount; ++i) {
508 if (pfnEqualSid(lpTokInfo->Groups[i].Sid, pAdminSid)) {
509 iRetVal = 1;
510 break;
511 }
512 }
513
514 pfnFreeSid(pAdminSid);
515 Safefree(lpTokInfo);
516 CloseHandle(hTok);
34f7f30d 517 FreeLibrary(module);
b4ad57f4 518
519 EXTEND(SP, 1);
520 ST(0) = sv_2mortal(newSViv(iRetVal));
521 XSRETURN(1);
522}
523
524XS(w32_LookupAccountName)
525{
526 dXSARGS;
527 char SID[400];
528 DWORD SIDLen;
529 SID_NAME_USE snu;
530 char Domain[256];
531 DWORD DomLen;
b4ad57f4 532 BOOL bResult;
8c56068e 533
b4ad57f4 534 if (items != 5)
535 croak("usage: Win32::LookupAccountName($system, $account, $domain, "
536 "$sid, $sidtype);\n");
537
538 SIDLen = sizeof(SID);
539 DomLen = sizeof(Domain);
540
8c56068e 541 bResult = LookupAccountNameA(SvPV_nolen(ST(0)), /* System */
542 SvPV_nolen(ST(1)), /* Account name */
543 &SID, /* SID structure */
544 &SIDLen, /* Size of SID buffer */
545 Domain, /* Domain buffer */
546 &DomLen, /* Domain buffer size */
547 &snu); /* SID name type */
b4ad57f4 548 if (bResult) {
549 sv_setpv(ST(2), Domain);
550 sv_setpvn(ST(3), SID, SIDLen);
551 sv_setiv(ST(4), snu);
552 XSRETURN_YES;
553 }
8c56068e 554 XSRETURN_NO;
555}
b4ad57f4 556
557
558XS(w32_LookupAccountSID)
559{
560 dXSARGS;
561 PSID sid;
562 char Account[256];
563 DWORD AcctLen = sizeof(Account);
564 char Domain[256];
565 DWORD DomLen = sizeof(Domain);
566 SID_NAME_USE snu;
b4ad57f4 567 BOOL bResult;
568
569 if (items != 5)
570 croak("usage: Win32::LookupAccountSID($system, $sid, $account, $domain, $sidtype);\n");
571
8c56068e 572 sid = SvPV_nolen(ST(1));
b4ad57f4 573 if (IsValidSid(sid)) {
8c56068e 574 bResult = LookupAccountSidA(SvPV_nolen(ST(0)), /* System */
575 sid, /* SID structure */
576 Account, /* Account name buffer */
577 &AcctLen, /* name buffer length */
578 Domain, /* Domain buffer */
579 &DomLen, /* Domain buffer length */
580 &snu); /* SID name type */
b4ad57f4 581 if (bResult) {
582 sv_setpv(ST(2), Account);
583 sv_setpv(ST(3), Domain);
584 sv_setiv(ST(4), (IV)snu);
585 XSRETURN_YES;
586 }
b4ad57f4 587 }
8c56068e 588 XSRETURN_NO;
589}
b4ad57f4 590
591XS(w32_InitiateSystemShutdown)
592{
593 dXSARGS;
594 HANDLE hToken; /* handle to process token */
595 TOKEN_PRIVILEGES tkp; /* pointer to token structure */
596 BOOL bRet;
b4ad57f4 597 char *machineName, *message;
b4ad57f4 598
599 if (items != 5)
600 croak("usage: Win32::InitiateSystemShutdown($machineName, $message, "
601 "$timeOut, $forceClose, $reboot);\n");
602
8c56068e 603 machineName = SvPV_nolen(ST(0));
b4ad57f4 604
605 if (OpenProcessToken(GetCurrentProcess(),
606 TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY,
607 &hToken))
608 {
8c56068e 609 LookupPrivilegeValueA(machineName,
610 SE_SHUTDOWN_NAMEA,
611 &tkp.Privileges[0].Luid);
b4ad57f4 612
613 tkp.PrivilegeCount = 1; /* only setting one */
614 tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
615
616 /* Get shutdown privilege for this process. */
617 AdjustTokenPrivileges(hToken, FALSE, &tkp, 0,
618 (PTOKEN_PRIVILEGES)NULL, 0);
619 }
620
8c56068e 621 message = SvPV_nolen(ST(1));
622 bRet = InitiateSystemShutdownA(machineName, message,
623 SvIV(ST(2)), SvIV(ST(3)), SvIV(ST(4)));
b4ad57f4 624
625 /* Disable shutdown privilege. */
626 tkp.Privileges[0].Attributes = 0;
627 AdjustTokenPrivileges(hToken, FALSE, &tkp, 0,
628 (PTOKEN_PRIVILEGES)NULL, 0);
629 CloseHandle(hToken);
630 XSRETURN_IV(bRet);
631}
632
633XS(w32_AbortSystemShutdown)
634{
635 dXSARGS;
636 HANDLE hToken; /* handle to process token */
637 TOKEN_PRIVILEGES tkp; /* pointer to token structure */
638 BOOL bRet;
639 char *machineName;
b4ad57f4 640
641 if (items != 1)
642 croak("usage: Win32::AbortSystemShutdown($machineName);\n");
643
8c56068e 644 machineName = SvPV_nolen(ST(0));
b4ad57f4 645
646 if (OpenProcessToken(GetCurrentProcess(),
647 TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY,
648 &hToken))
649 {
8c56068e 650 LookupPrivilegeValueA(machineName,
651 SE_SHUTDOWN_NAMEA,
652 &tkp.Privileges[0].Luid);
b4ad57f4 653
654 tkp.PrivilegeCount = 1; /* only setting one */
655 tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
656
657 /* Get shutdown privilege for this process. */
658 AdjustTokenPrivileges(hToken, FALSE, &tkp, 0,
659 (PTOKEN_PRIVILEGES)NULL, 0);
660 }
661
8c56068e 662 bRet = AbortSystemShutdownA(machineName);
b4ad57f4 663
664 /* Disable shutdown privilege. */
665 tkp.Privileges[0].Attributes = 0;
666 AdjustTokenPrivileges(hToken, FALSE, &tkp, 0,
667 (PTOKEN_PRIVILEGES)NULL, 0);
668 CloseHandle(hToken);
669 XSRETURN_IV(bRet);
670}
671
672
673XS(w32_MsgBox)
674{
675 dXSARGS;
b4ad57f4 676 DWORD flags = MB_ICONEXCLAMATION;
b4ad57f4 677 I32 result;
678
679 if (items < 1 || items > 3)
680 croak("usage: Win32::MsgBox($message [, $flags [, $title]]);\n");
681
34f7f30d 682 if (items > 1)
683 flags = SvIV(ST(1));
684
685 if (IsWin2000()) {
686 WCHAR *title = NULL;
687 WCHAR *msg = sv_to_wstr(aTHX_ ST(0));
688 if (items > 2)
689 title = sv_to_wstr(aTHX_ ST(2));
690 result = MessageBoxW(GetActiveWindow(), msg, title ? title : L"Perl", flags);
691 Safefree(msg);
692 if (title)
693 Safefree(title);
694 }
695 else {
696 char *title = "Perl";
697 char *msg = SvPV_nolen(ST(0));
698 if (items > 2)
699 title = SvPV_nolen(ST(2));
700 result = MessageBoxA(GetActiveWindow(), msg, title, flags);
b4ad57f4 701 }
b4ad57f4 702 XSRETURN_IV(result);
703}
704
705XS(w32_LoadLibrary)
706{
707 dXSARGS;
b4ad57f4 708 HANDLE hHandle;
b4ad57f4 709
710 if (items != 1)
711 croak("usage: Win32::LoadLibrary($libname)\n");
8c56068e 712 hHandle = LoadLibraryA(SvPV_nolen(ST(0)));
b4ad57f4 713 XSRETURN_IV((long)hHandle);
714}
715
716XS(w32_FreeLibrary)
717{
718 dXSARGS;
8c56068e 719
b4ad57f4 720 if (items != 1)
721 croak("usage: Win32::FreeLibrary($handle)\n");
8c56068e 722 if (FreeLibrary(INT2PTR(HINSTANCE, SvIV(ST(0))))) {
b4ad57f4 723 XSRETURN_YES;
724 }
725 XSRETURN_NO;
726}
727
728XS(w32_GetProcAddress)
729{
730 dXSARGS;
8c56068e 731
b4ad57f4 732 if (items != 2)
733 croak("usage: Win32::GetProcAddress($hinstance, $procname)\n");
8c56068e 734 XSRETURN_IV(PTR2IV(GetProcAddress(INT2PTR(HINSTANCE, SvIV(ST(0))), SvPV_nolen(ST(1)))));
b4ad57f4 735}
736
737XS(w32_RegisterServer)
738{
739 dXSARGS;
740 BOOL result = FALSE;
34f7f30d 741 HMODULE module;
b4ad57f4 742
743 if (items != 1)
744 croak("usage: Win32::RegisterServer($libname)\n");
745
34f7f30d 746 module = LoadLibraryA(SvPV_nolen(ST(0)));
747 if (module) {
748 PFNDllRegisterServer pfnDllRegisterServer;
749 GETPROC(DllRegisterServer);
750 if (pfnDllRegisterServer && pfnDllRegisterServer() == 0)
b4ad57f4 751 result = TRUE;
34f7f30d 752 FreeLibrary(module);
b4ad57f4 753 }
8c56068e 754 ST(0) = boolSV(result);
755 XSRETURN(1);
b4ad57f4 756}
757
758XS(w32_UnregisterServer)
759{
760 dXSARGS;
761 BOOL result = FALSE;
34f7f30d 762 HINSTANCE module;
b4ad57f4 763
764 if (items != 1)
765 croak("usage: Win32::UnregisterServer($libname)\n");
766
34f7f30d 767 module = LoadLibraryA(SvPV_nolen(ST(0)));
768 if (module) {
769 PFNDllUnregisterServer pfnDllUnregisterServer;
770 GETPROC(DllUnregisterServer);
771 if (pfnDllUnregisterServer && pfnDllUnregisterServer() == 0)
b4ad57f4 772 result = TRUE;
34f7f30d 773 FreeLibrary(module);
b4ad57f4 774 }
8c56068e 775 ST(0) = boolSV(result);
776 XSRETURN(1);
b4ad57f4 777}
778
779/* XXX rather bogus */
780XS(w32_GetArchName)
781{
782 dXSARGS;
783 XSRETURN_PV(getenv("PROCESSOR_ARCHITECTURE"));
784}
785
786XS(w32_GetChipName)
787{
788 dXSARGS;
789 SYSTEM_INFO sysinfo;
790
791 Zero(&sysinfo,1,SYSTEM_INFO);
792 GetSystemInfo(&sysinfo);
793 /* XXX docs say dwProcessorType is deprecated on NT */
794 XSRETURN_IV(sysinfo.dwProcessorType);
795}
796
797XS(w32_GuidGen)
798{
799 dXSARGS;
800 GUID guid;
801 char szGUID[50] = {'\0'};
802 HRESULT hr = CoCreateGuid(&guid);
803
804 if (SUCCEEDED(hr)) {
805 LPOLESTR pStr = NULL;
e364e11c 806 if (SUCCEEDED(StringFromCLSID(&guid, &pStr))) {
807 WideCharToMultiByte(CP_ACP, 0, pStr, wcslen(pStr), szGUID,
808 sizeof(szGUID), NULL, NULL);
809 CoTaskMemFree(pStr);
810 XSRETURN_PV(szGUID);
811 }
b4ad57f4 812 }
e364e11c 813 XSRETURN_UNDEF;
b4ad57f4 814}
815
816XS(w32_GetFolderPath)
817{
818 dXSARGS;
819 char path[MAX_PATH+1];
34f7f30d 820 WCHAR wpath[MAX_PATH+1];
b4ad57f4 821 int folder;
822 int create = 0;
823 HMODULE module;
824
825 if (items != 1 && items != 2)
826 croak("usage: Win32::GetFolderPath($csidl [, $create])\n");
827
828 folder = SvIV(ST(0));
829 if (items == 2)
830 create = SvTRUE(ST(1)) ? CSIDL_FLAG_CREATE : 0;
831
b4ad57f4 832 module = LoadLibrary("shfolder.dll");
833 if (module) {
34f7f30d 834 PFNSHGetFolderPathA pfna;
835 if (IsWin2000()) {
836 PFNSHGetFolderPathW pfnw;
837 pfnw = (PFNSHGetFolderPathW)GetProcAddress(module, "SHGetFolderPathW");
838 if (pfnw && SUCCEEDED(pfnw(NULL, folder|create, NULL, 0, wpath))) {
839 FreeLibrary(module);
840 ST(0) = wstr_to_ansipath(aTHX_ wpath);
841 XSRETURN(1);
842 }
843 }
844 pfna = (PFNSHGetFolderPathA)GetProcAddress(module, "SHGetFolderPathA");
845 if (pfna && SUCCEEDED(pfna(NULL, folder|create, NULL, 0, path))) {
b4ad57f4 846 FreeLibrary(module);
847 XSRETURN_PV(path);
848 }
849 FreeLibrary(module);
850 }
851
852 module = LoadLibrary("shell32.dll");
853 if (module) {
34f7f30d 854 PFNSHGetSpecialFolderPathA pfna;
855 if (IsWin2000()) {
856 PFNSHGetSpecialFolderPathW pfnw;
857 pfnw = (PFNSHGetSpecialFolderPathW)GetProcAddress(module, "SHGetSpecialFolderPathW");
858 if (pfnw && pfnw(NULL, wpath, folder, !!create)) {
859 FreeLibrary(module);
860 ST(0) = wstr_to_ansipath(aTHX_ wpath);
861 XSRETURN(1);
862 }
863 }
864 pfna = (PFNSHGetSpecialFolderPathA)GetProcAddress(module, "SHGetSpecialFolderPathA");
865 if (pfna && pfna(NULL, path, folder, !!create)) {
b4ad57f4 866 FreeLibrary(module);
867 XSRETURN_PV(path);
868 }
869 FreeLibrary(module);
870 }
34f7f30d 871
872 /* SHGetFolderPathW() and SHGetSpecialFolderPathW() may fail on older
873 * Perl versions that have replaced the Unicode environment with an
874 * ANSI version. Let's go spelunking in the registry now...
875 */
876 if (IsWin2000()) {
877 SV *sv;
878 HKEY hkey;
879 HKEY root = HKEY_CURRENT_USER;
880 WCHAR *name = NULL;
881
882 switch (folder) {
883 case CSIDL_ADMINTOOLS: name = L"Administrative Tools"; break;
884 case CSIDL_APPDATA: name = L"AppData"; break;
885 case CSIDL_CDBURN_AREA: name = L"CD Burning"; break;
886 case CSIDL_COOKIES: name = L"Cookies"; break;
887 case CSIDL_DESKTOP:
888 case CSIDL_DESKTOPDIRECTORY: name = L"Desktop"; break;
889 case CSIDL_FAVORITES: name = L"Favorites"; break;
890 case CSIDL_FONTS: name = L"Fonts"; break;
891 case CSIDL_HISTORY: name = L"History"; break;
892 case CSIDL_INTERNET_CACHE: name = L"Cache"; break;
893 case CSIDL_LOCAL_APPDATA: name = L"Local AppData"; break;
894 case CSIDL_MYMUSIC: name = L"My Music"; break;
895 case CSIDL_MYPICTURES: name = L"My Pictures"; break;
896 case CSIDL_MYVIDEO: name = L"My Video"; break;
897 case CSIDL_NETHOOD: name = L"NetHood"; break;
898 case CSIDL_PERSONAL: name = L"Personal"; break;
899 case CSIDL_PRINTHOOD: name = L"PrintHood"; break;
900 case CSIDL_PROGRAMS: name = L"Programs"; break;
901 case CSIDL_RECENT: name = L"Recent"; break;
902 case CSIDL_SENDTO: name = L"SendTo"; break;
903 case CSIDL_STARTMENU: name = L"Start Menu"; break;
904 case CSIDL_STARTUP: name = L"Startup"; break;
905 case CSIDL_TEMPLATES: name = L"Templates"; break;
906 /* XXX L"Local Settings" */
907 }
908
909 if (!name) {
910 root = HKEY_LOCAL_MACHINE;
911 switch (folder) {
912 case CSIDL_COMMON_ADMINTOOLS: name = L"Common Administrative Tools"; break;
913 case CSIDL_COMMON_APPDATA: name = L"Common AppData"; break;
914 case CSIDL_COMMON_DESKTOPDIRECTORY: name = L"Common Desktop"; break;
915 case CSIDL_COMMON_DOCUMENTS: name = L"Common Documents"; break;
916 case CSIDL_COMMON_FAVORITES: name = L"Common Favorites"; break;
917 case CSIDL_COMMON_PROGRAMS: name = L"Common Programs"; break;
918 case CSIDL_COMMON_STARTMENU: name = L"Common Start Menu"; break;
919 case CSIDL_COMMON_STARTUP: name = L"Common Startup"; break;
920 case CSIDL_COMMON_TEMPLATES: name = L"Common Templates"; break;
921 case CSIDL_COMMON_MUSIC: name = L"CommonMusic"; break;
922 case CSIDL_COMMON_PICTURES: name = L"CommonPictures"; break;
923 case CSIDL_COMMON_VIDEO: name = L"CommonVideo"; break;
924 }
925 }
926 /* XXX todo
927 * case CSIDL_SYSTEM # GetSystemDirectory()
928 * case CSIDL_RESOURCES # %windir%\Resources\, For theme and other windows resources.
929 * case CSIDL_RESOURCES_LOCALIZED # %windir%\Resources\<LangID>, for theme and other windows specific resources.
930 */
931
932#define SHELL_FOLDERS "Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\Shell Folders"
933
934 if (name && RegOpenKeyEx(root, SHELL_FOLDERS, 0, KEY_QUERY_VALUE, &hkey) == ERROR_SUCCESS) {
935 WCHAR data[MAX_PATH+1];
936 DWORD cb = sizeof(data)-sizeof(WCHAR);
937 DWORD type = REG_NONE;
938 long rc = RegQueryValueExW(hkey, name, NULL, &type, (BYTE*)&data, &cb);
939 RegCloseKey(hkey);
940 if (rc == ERROR_SUCCESS && type == REG_SZ && cb > sizeof(WCHAR) && data[0]) {
941 /* Make sure the string is properly terminated */
942 data[cb/sizeof(WCHAR)] = '\0';
943 ST(0) = wstr_to_ansipath(aTHX_ data);
944 XSRETURN(1);
945 }
946 }
947
948#undef SHELL_FOLDERS
949
950 /* Unders some circumstances the registry entries seem to have a null string
951 * as their value even when the directory already exists. The environment
952 * variables do get set though, so try re-create a Unicode environment and
953 * check if they are there.
954 */
955 sv = NULL;
956 switch (folder) {
957 case CSIDL_APPDATA: sv = get_unicode_env(aTHX_ L"APPDATA"); break;
958 case CSIDL_PROFILE: sv = get_unicode_env(aTHX_ L"USERPROFILE"); break;
959 case CSIDL_PROGRAM_FILES: sv = get_unicode_env(aTHX_ L"ProgramFiles"); break;
960 case CSIDL_PROGRAM_FILES_COMMON: sv = get_unicode_env(aTHX_ L"CommonProgramFiles"); break;
961 case CSIDL_WINDOWS: sv = get_unicode_env(aTHX_ L"SystemRoot"); break;
962 }
963 if (sv) {
964 ST(0) = sv;
965 XSRETURN(1);
966 }
967 }
968
b4ad57f4 969 XSRETURN_UNDEF;
970}
971
e364e11c 972XS(w32_GetFileVersion)
973{
974 dXSARGS;
975 DWORD size;
976 DWORD handle;
977 char *filename;
978 char *data;
979
980 if (items != 1)
981 croak("usage: Win32::GetFileVersion($filename)\n");
982
983 filename = SvPV_nolen(ST(0));
984 size = GetFileVersionInfoSize(filename, &handle);
985 if (!size)
986 XSRETURN_UNDEF;
987
988 New(0, data, size, char);
989 if (!data)
990 XSRETURN_UNDEF;
991
992 if (GetFileVersionInfo(filename, handle, size, data)) {
993 VS_FIXEDFILEINFO *info;
994 UINT len;
995 if (VerQueryValue(data, "\\", (void**)&info, &len)) {
996 int dwValueMS1 = (info->dwFileVersionMS>>16);
997 int dwValueMS2 = (info->dwFileVersionMS&0xffff);
998 int dwValueLS1 = (info->dwFileVersionLS>>16);
999 int dwValueLS2 = (info->dwFileVersionLS&0xffff);
1000
1001 if (GIMME_V == G_ARRAY) {
1002 EXTEND(SP, 4);
1003 XST_mIV(0, dwValueMS1);
1004 XST_mIV(1, dwValueMS2);
1005 XST_mIV(2, dwValueLS1);
1006 XST_mIV(3, dwValueLS2);
1007 items = 4;
1008 }
1009 else {
1010 char version[50];
1011 sprintf(version, "%d.%d.%d.%d", dwValueMS1, dwValueMS2, dwValueLS1, dwValueLS2);
1012 XST_mPV(0, version);
1013 }
1014 }
1015 }
1016 else
1017 items = 0;
1018
1019 Safefree(data);
1020 XSRETURN(items);
1021}
1022
5bd7e777 1023#ifdef __CYGWIN__
cf8b4e93 1024XS(w32_SetChildShowWindow)
1025{
5bd7e777 1026 /* This function doesn't do anything useful for cygwin. In the
1027 * MSWin32 case it modifies w32_showwindow, which is used by
1028 * win32_spawnvp(). Since w32_showwindow is an internal variable
1029 * inside the thread_intern structure, the MSWin32 implementation
1030 * lives in win32/win32.c in the core Perl distribution.
1031 */
cf8b4e93 1032 dXSARGS;
5bd7e777 1033 XSRETURN_UNDEF;
cf8b4e93 1034}
5bd7e777 1035#endif
cf8b4e93 1036
cf8b4e93 1037XS(w32_GetCwd)
1038{
1039 dXSARGS;
1040 /* Make the host for current directory */
1041 char* ptr = PerlEnv_get_childdir();
1042 /*
1043 * If ptr != Nullch
1044 * then it worked, set PV valid,
1045 * else return 'undef'
1046 */
1047 if (ptr) {
1048 SV *sv = sv_newmortal();
1049 sv_setpv(sv, ptr);
1050 PerlEnv_free_childdir(ptr);
1051
1052#ifndef INCOMPLETE_TAINTS
1053 SvTAINTED_on(sv);
1054#endif
1055
1056 EXTEND(SP,1);
cf8b4e93 1057 ST(0) = sv;
1058 XSRETURN(1);
1059 }
1060 XSRETURN_UNDEF;
1061}
1062
cf8b4e93 1063XS(w32_SetCwd)
1064{
1065 dXSARGS;
1066 if (items != 1)
34f7f30d 1067 Perl_croak(aTHX_ "usage: Win32::SetCwd($cwd)");
1068
1069 if (IsWin2000() && SvUTF8(ST(0))) {
1070 WCHAR *wide = sv_to_wstr(aTHX_ ST(0));
1071 char *ansi = my_ansipath(wide);
1072 int rc = PerlDir_chdir(ansi);
1073 Safefree(wide);
1074 Safefree(ansi);
1075 if (!rc)
1076 XSRETURN_YES;
1077 }
1078 else {
1079 if (!PerlDir_chdir(SvPV_nolen(ST(0))))
1080 XSRETURN_YES;
1081 }
cf8b4e93 1082
1083 XSRETURN_NO;
1084}
1085
cf8b4e93 1086XS(w32_GetNextAvailDrive)
1087{
1088 dXSARGS;
1089 char ix = 'C';
1090 char root[] = "_:\\";
1091
1092 EXTEND(SP,1);
1093 while (ix <= 'Z') {
1094 root[0] = ix++;
1095 if (GetDriveType(root) == 1) {
1096 root[2] = '\0';
1097 XSRETURN_PV(root);
1098 }
1099 }
1100 XSRETURN_UNDEF;
1101}
1102
cf8b4e93 1103XS(w32_GetLastError)
1104{
1105 dXSARGS;
1106 EXTEND(SP,1);
1107 XSRETURN_IV(GetLastError());
1108}
1109
cf8b4e93 1110XS(w32_SetLastError)
1111{
1112 dXSARGS;
1113 if (items != 1)
1114 Perl_croak(aTHX_ "usage: Win32::SetLastError($error)");
1115 SetLastError(SvIV(ST(0)));
1116 XSRETURN_EMPTY;
1117}
1118
cf8b4e93 1119XS(w32_LoginName)
1120{
1121 dXSARGS;
cf8b4e93 1122 EXTEND(SP,1);
34f7f30d 1123 if (IsWin2000()) {
1124 WCHAR name[128];
1125 DWORD size = countof(name);
1126 if (GetUserNameW(name, &size)) {
1127 ST(0) = wstr_to_sv(aTHX_ name);
1128 XSRETURN(1);
1129 }
1130 }
1131 else {
1132 char name[128];
1133 DWORD size = countof(name);
1134 if (GetUserNameA(name, &size)) {
1135 /* size includes NULL */
1136 ST(0) = sv_2mortal(newSVpvn(name, size-1));
1137 XSRETURN(1);
1138 }
cf8b4e93 1139 }
1140 XSRETURN_UNDEF;
1141}
1142
cf8b4e93 1143XS(w32_NodeName)
1144{
1145 dXSARGS;
1146 char name[MAX_COMPUTERNAME_LENGTH+1];
1147 DWORD size = sizeof(name);
1148 EXTEND(SP,1);
1149 if (GetComputerName(name,&size)) {
1150 /* size does NOT include NULL :-( */
1151 ST(0) = sv_2mortal(newSVpvn(name,size));
1152 XSRETURN(1);
1153 }
1154 XSRETURN_UNDEF;
1155}
1156
1157
cf8b4e93 1158XS(w32_DomainName)
1159{
1160 dXSARGS;
34f7f30d 1161 HMODULE module = LoadLibrary("netapi32.dll");
1162 PFNNetApiBufferFree pfnNetApiBufferFree;
1163 PFNNetWkstaGetInfo pfnNetWkstaGetInfo;
1164
1165 if (module) {
1166 GETPROC(NetApiBufferFree);
1167 GETPROC(NetWkstaGetInfo);
cf8b4e93 1168 }
1169 EXTEND(SP,1);
34f7f30d 1170 if (module && pfnNetWkstaGetInfo && pfnNetApiBufferFree) {
cf8b4e93 1171 /* this way is more reliable, in case user has a local account. */
1172 char dname[256];
1173 DWORD dnamelen = sizeof(dname);
1174 struct {
1175 DWORD wki100_platform_id;
1176 LPWSTR wki100_computername;
1177 LPWSTR wki100_langroup;
1178 DWORD wki100_ver_major;
1179 DWORD wki100_ver_minor;
1180 } *pwi;
1181 /* NERR_Success *is* 0*/
1182 if (0 == pfnNetWkstaGetInfo(NULL, 100, &pwi)) {
1183 if (pwi->wki100_langroup && *(pwi->wki100_langroup)) {
1184 WideCharToMultiByte(CP_ACP, 0, pwi->wki100_langroup,
1185 -1, (LPSTR)dname, dnamelen, NULL, NULL);
1186 }
1187 else {
1188 WideCharToMultiByte(CP_ACP, 0, pwi->wki100_computername,
1189 -1, (LPSTR)dname, dnamelen, NULL, NULL);
1190 }
1191 pfnNetApiBufferFree(pwi);
34f7f30d 1192 FreeLibrary(module);
cf8b4e93 1193 XSRETURN_PV(dname);
1194 }
34f7f30d 1195 FreeLibrary(module);
cf8b4e93 1196 }
1197 else {
1198 /* Win95 doesn't have NetWksta*(), so do it the old way */
1199 char name[256];
1200 DWORD size = sizeof(name);
34f7f30d 1201 if (module)
1202 FreeLibrary(module);
cf8b4e93 1203 if (GetUserName(name,&size)) {
1204 char sid[ONE_K_BUFSIZE];
1205 DWORD sidlen = sizeof(sid);
1206 char dname[256];
1207 DWORD dnamelen = sizeof(dname);
1208 SID_NAME_USE snu;
1209 if (LookupAccountName(NULL, name, (PSID)&sid, &sidlen,
1210 dname, &dnamelen, &snu)) {
1211 XSRETURN_PV(dname); /* all that for this */
1212 }
1213 }
1214 }
1215 XSRETURN_UNDEF;
1216}
1217
cf8b4e93 1218XS(w32_FsType)
1219{
1220 dXSARGS;
1221 char fsname[256];
1222 DWORD flags, filecomplen;
1223 if (GetVolumeInformation(NULL, NULL, 0, NULL, &filecomplen,
1224 &flags, fsname, sizeof(fsname))) {
1225 if (GIMME_V == G_ARRAY) {
1226 XPUSHs(sv_2mortal(newSVpvn(fsname,strlen(fsname))));
1227 XPUSHs(sv_2mortal(newSViv(flags)));
1228 XPUSHs(sv_2mortal(newSViv(filecomplen)));
1229 PUTBACK;
1230 return;
1231 }
1232 EXTEND(SP,1);
1233 XSRETURN_PV(fsname);
1234 }
1235 XSRETURN_EMPTY;
1236}
1237
cf8b4e93 1238XS(w32_GetOSVersion)
1239{
1240 dXSARGS;
34f7f30d 1241
cf8b4e93 1242 if (GIMME_V == G_SCALAR) {
34f7f30d 1243 XSRETURN_IV(g_osver.dwPlatformId);
cf8b4e93 1244 }
34f7f30d 1245 XPUSHs(newSVpvn(g_osver.szCSDVersion, strlen(g_osver.szCSDVersion)));
1246
1247 XPUSHs(newSViv(g_osver.dwMajorVersion));
1248 XPUSHs(newSViv(g_osver.dwMinorVersion));
1249 XPUSHs(newSViv(g_osver.dwBuildNumber));
1250 XPUSHs(newSViv(g_osver.dwPlatformId));
1251 if (g_osver_ex) {
1252 XPUSHs(newSViv(g_osver.wServicePackMajor));
1253 XPUSHs(newSViv(g_osver.wServicePackMinor));
1254 XPUSHs(newSViv(g_osver.wSuiteMask));
1255 XPUSHs(newSViv(g_osver.wProductType));
cf8b4e93 1256 }
1257 PUTBACK;
1258}
1259
cf8b4e93 1260XS(w32_IsWinNT)
1261{
1262 dXSARGS;
1263 EXTEND(SP,1);
1264 XSRETURN_IV(IsWinNT());
1265}
1266
cf8b4e93 1267XS(w32_IsWin95)
1268{
1269 dXSARGS;
1270 EXTEND(SP,1);
1271 XSRETURN_IV(IsWin95());
1272}
1273
cf8b4e93 1274XS(w32_FormatMessage)
1275{
1276 dXSARGS;
1277 DWORD source = 0;
1278 char msgbuf[ONE_K_BUFSIZE];
1279
1280 if (items != 1)
1281 Perl_croak(aTHX_ "usage: Win32::FormatMessage($errno)");
1282
1283 if (FormatMessageA(FORMAT_MESSAGE_FROM_SYSTEM,
1284 &source, SvIV(ST(0)), 0,
1285 msgbuf, sizeof(msgbuf)-1, NULL))
1286 {
1287 XSRETURN_PV(msgbuf);
1288 }
1289
1290 XSRETURN_UNDEF;
1291}
1292
cf8b4e93 1293XS(w32_Spawn)
1294{
1295 dXSARGS;
1296 char *cmd, *args;
1297 void *env;
1298 char *dir;
1299 PROCESS_INFORMATION stProcInfo;
1300 STARTUPINFO stStartInfo;
1301 BOOL bSuccess = FALSE;
1302
1303 if (items != 3)
1304 Perl_croak(aTHX_ "usage: Win32::Spawn($cmdName, $args, $PID)");
1305
1306 cmd = SvPV_nolen(ST(0));
1307 args = SvPV_nolen(ST(1));
1308
1309 env = PerlEnv_get_childenv();
1310 dir = PerlEnv_get_childdir();
1311
1312 memset(&stStartInfo, 0, sizeof(stStartInfo)); /* Clear the block */
1313 stStartInfo.cb = sizeof(stStartInfo); /* Set the structure size */
1314 stStartInfo.dwFlags = STARTF_USESHOWWINDOW; /* Enable wShowWindow control */
1315 stStartInfo.wShowWindow = SW_SHOWMINNOACTIVE; /* Start min (normal) */
1316
1317 if (CreateProcess(
1318 cmd, /* Image path */
1319 args, /* Arguments for command line */
1320 NULL, /* Default process security */
1321 NULL, /* Default thread security */
1322 FALSE, /* Must be TRUE to use std handles */
1323 NORMAL_PRIORITY_CLASS, /* No special scheduling */
1324 env, /* Inherit our environment block */
1325 dir, /* Inherit our currrent directory */
1326 &stStartInfo, /* -> Startup info */
1327 &stProcInfo)) /* <- Process info (if OK) */
1328 {
1329 int pid = (int)stProcInfo.dwProcessId;
1330 if (IsWin95() && pid < 0)
1331 pid = -pid;
1332 sv_setiv(ST(2), pid);
1333 CloseHandle(stProcInfo.hThread);/* library source code does this. */
1334 bSuccess = TRUE;
1335 }
1336 PerlEnv_free_childenv(env);
1337 PerlEnv_free_childdir(dir);
1338 XSRETURN_IV(bSuccess);
1339}
1340
cf8b4e93 1341XS(w32_GetTickCount)
1342{
1343 dXSARGS;
1344 DWORD msec = GetTickCount();
1345 EXTEND(SP,1);
1346 if ((IV)msec > 0)
1347 XSRETURN_IV(msec);
1348 XSRETURN_NV(msec);
1349}
1350
cf8b4e93 1351XS(w32_GetShortPathName)
1352{
1353 dXSARGS;
1354 SV *shortpath;
1355 DWORD len;
1356
1357 if (items != 1)
1358 Perl_croak(aTHX_ "usage: Win32::GetShortPathName($longPathName)");
1359
34f7f30d 1360 if (IsWin2000()) {
1361 WCHAR wshort[MAX_PATH+1];
1362 WCHAR *wlong = sv_to_wstr(aTHX_ ST(0));
1363 len = GetShortPathNameW(wlong, wshort, countof(wshort));
1364 Safefree(wlong);
b3eb56ab 1365 if (len && len < sizeof(wshort)) {
34f7f30d 1366 ST(0) = wstr_to_sv(aTHX_ wshort);
1367 XSRETURN(1);
1368 }
1369 XSRETURN_UNDEF;
1370 }
1371
cf8b4e93 1372 shortpath = sv_mortalcopy(ST(0));
1373 SvUPGRADE(shortpath, SVt_PV);
1374 if (!SvPVX(shortpath) || !SvLEN(shortpath))
1375 XSRETURN_UNDEF;
1376
1377 /* src == target is allowed */
1378 do {
1379 len = GetShortPathName(SvPVX(shortpath),
1380 SvPVX(shortpath),
1381 SvLEN(shortpath));
1382 } while (len >= SvLEN(shortpath) && sv_grow(shortpath,len+1));
1383 if (len) {
1384 SvCUR_set(shortpath,len);
1385 *SvEND(shortpath) = '\0';
1386 ST(0) = shortpath;
1387 XSRETURN(1);
1388 }
1389 XSRETURN_UNDEF;
1390}
1391
cf8b4e93 1392XS(w32_GetFullPathName)
1393{
1394 dXSARGS;
34f7f30d 1395 char *fullname;
1396 char *ansi = NULL;
1397
1398/* The code below relies on the fact that PerlDir_mapX() returns an
1399 * absolute path, which is only true under PERL_IMPLICIT_SYS when
1400 * we use the virtualization code from win32/vdir.h.
1401 * Without it PerlDir_mapX() is a no-op and we need to use the same
1402 * code as we use for Cygwin.
1403 */
1404#if __CYGWIN__ || !defined(PERL_IMPLICIT_SYS)
1405 char buffer[2*MAX_PATH];
1406#endif
cf8b4e93 1407
1408 if (items != 1)
1409 Perl_croak(aTHX_ "usage: Win32::GetFullPathName($filename)");
1410
34f7f30d 1411#if __CYGWIN__ || !defined(PERL_IMPLICIT_SYS)
1412 if (IsWin2000()) {
1413 WCHAR *filename = sv_to_wstr(aTHX_ ST(0));
1414 WCHAR full[2*MAX_PATH];
1415 DWORD len = GetFullPathNameW(filename, countof(full), full, NULL);
1416 Safefree(filename);
1417 if (len == 0 || len >= countof(full))
1418 XSRETURN_EMPTY;
1419 ansi = fullname = my_ansipath(full);
1420 }
1421 else {
1422 DWORD len = GetFullPathNameA(SvPV_nolen(ST(0)), countof(buffer), buffer, NULL);
1423 if (len == 0 || len >= countof(buffer))
1424 XSRETURN_EMPTY;
1425 fullname = buffer;
1426 }
1427#else
1428 /* Don't use my_ansipath() unless the $filename argument is in Unicode.
1429 * If the relative path doesn't exist, GetShortPathName() will fail and
1430 * my_ansipath() will use the long name with replacement characters.
1431 * In that case we will be better off using PerlDir_mapA(), which
1432 * already uses the ANSI name of the current directory.
1433 *
1434 * XXX The one missing case is where we could downgrade $filename
1435 * XXX from UTF8 into the current codepage.
1436 */
1437 if (IsWin2000() && SvUTF8(ST(0))) {
1438 WCHAR *filename = sv_to_wstr(aTHX_ ST(0));
1439 WCHAR *mappedname = PerlDir_mapW(filename);
1440 Safefree(filename);
1441 ansi = fullname = my_ansipath(mappedname);
1442 }
1443 else {
1444 fullname = PerlDir_mapA(SvPV_nolen(ST(0)));
1445 }
1446# if PERL_VERSION < 8
1447 {
1448 /* PerlDir_mapX() in Perl 5.6 used to return forward slashes */
1449 char *str = fullname;
1450 while (*str) {
1451 if (*str == '/')
1452 *str = '\\';
1453 ++str;
1454 }
1455 }
1456# endif
1457#endif
cf8b4e93 1458
34f7f30d 1459 /* GetFullPathName() on Windows NT drops trailing backslash */
1460 if (g_osver.dwMajorVersion == 4 && *fullname) {
1461 STRLEN len;
1462 char *pv = SvPV(ST(0), len);
1463 char *lastchar = fullname + strlen(fullname) - 1;
1464 /* If ST(0) ends with a slash, but fullname doesn't ... */
1465 if (len && (pv[len-1] == '/' || pv[len-1] == '\\') && *lastchar != '\\') {
1466 /* fullname is the MAX_PATH+1 sized buffer returned from PerlDir_mapA()
1467 * or the 2*MAX_PATH sized local buffer in the __CYGWIN__ case.
1468 */
1469 strcpy(lastchar+1, "\\");
1470 }
cf8b4e93 1471 }
34f7f30d 1472
1473 if (GIMME_V == G_ARRAY) {
1474 char *filepart = strrchr(fullname, '\\');
1475
1476 EXTEND(SP,1);
1477 if (filepart) {
1478 XST_mPV(1, ++filepart);
1479 *filepart = '\0';
1480 }
1481 else {
1482 XST_mPVN(1, "", 0);
1483 }
1484 items = 2;
1485 }
1486 XST_mPV(0, fullname);
1487
1488 if (ansi)
1489 Safefree(ansi);
1490 XSRETURN(items);
cf8b4e93 1491}
1492
cf8b4e93 1493XS(w32_GetLongPathName)
1494{
1495 dXSARGS;
cf8b4e93 1496
1497 if (items != 1)
1498 Perl_croak(aTHX_ "usage: Win32::GetLongPathName($pathname)");
1499
34f7f30d 1500 if (IsWin2000()) {
1501 WCHAR *wstr = sv_to_wstr(aTHX_ ST(0));
1502 WCHAR wide_path[MAX_PATH+1];
1503 WCHAR *long_path;
1504
1505 wcscpy(wide_path, wstr);
1506 Safefree(wstr);
1507 long_path = my_longpathW(wide_path);
1508 if (long_path) {
1509 ST(0) = wstr_to_sv(aTHX_ long_path);
1510 XSRETURN(1);
1511 }
1512 }
1513 else {
1514 SV *path;
1515 char tmpbuf[MAX_PATH+1];
1516 char *pathstr;
1517 STRLEN len;
1518
1519 path = ST(0);
1520 pathstr = SvPV(path,len);
1521 strcpy(tmpbuf, pathstr);
1522 pathstr = my_longpathA(tmpbuf);
1523 if (pathstr) {
1524 ST(0) = sv_2mortal(newSVpvn(pathstr, strlen(pathstr)));
1525 XSRETURN(1);
1526 }
cf8b4e93 1527 }
1528 XSRETURN_EMPTY;
1529}
1530
34f7f30d 1531XS(w32_GetANSIPathName)
1532{
1533 dXSARGS;
1534 WCHAR *wide_path;
1535
1536 if (items != 1)
1537 Perl_croak(aTHX_ "usage: Win32::GetANSIPathName($pathname)");
1538
1539 wide_path = sv_to_wstr(aTHX_ ST(0));
1540 ST(0) = wstr_to_ansipath(aTHX_ wide_path);
1541 Safefree(wide_path);
1542 XSRETURN(1);
1543}
1544
cf8b4e93 1545XS(w32_Sleep)
1546{
1547 dXSARGS;
1548 if (items != 1)
1549 Perl_croak(aTHX_ "usage: Win32::Sleep($milliseconds)");
1550 Sleep(SvIV(ST(0)));
1551 XSRETURN_YES;
1552}
1553
cf8b4e93 1554XS(w32_CopyFile)
1555{
1556 dXSARGS;
1557 BOOL bResult;
1558 char szSourceFile[MAX_PATH+1];
1559
1560 if (items != 3)
1561 Perl_croak(aTHX_ "usage: Win32::CopyFile($from, $to, $overwrite)");
1562 strcpy(szSourceFile, PerlDir_mapA(SvPV_nolen(ST(0))));
1563 bResult = CopyFileA(szSourceFile, PerlDir_mapA(SvPV_nolen(ST(1))), !SvTRUE(ST(2)));
1564 if (bResult)
1565 XSRETURN_YES;
1566 XSRETURN_NO;
1567}
1568
34f7f30d 1569XS(w32_OutputDebugString)
1570{
1571 dXSARGS;
1572 if (items != 1)
1573 Perl_croak(aTHX_ "usage: Win32::OutputDebugString($string)");
1574
1575 if (SvUTF8(ST(0))) {
1576 WCHAR *str = sv_to_wstr(aTHX_ ST(0));
1577 OutputDebugStringW(str);
1578 Safefree(str);
1579 }
1580 else
1581 OutputDebugStringA(SvPV_nolen(ST(0)));
1582
1583 XSRETURN_EMPTY;
1584}
1585
1586XS(w32_GetCurrentThreadId)
1587{
1588 dXSARGS;
1589 EXTEND(SP,1);
1590 XSRETURN_IV(GetCurrentThreadId());
1591}
1592
1593XS(w32_CreateDirectory)
1594{
1595 dXSARGS;
1596 BOOL result;
1597
1598 if (items != 1)
1599 Perl_croak(aTHX_ "usage: Win32::CreateDirectory($dir)");
1600
1601 if (IsWin2000() && SvUTF8(ST(0))) {
1602 WCHAR *dir = sv_to_wstr(aTHX_ ST(0));
1603 result = CreateDirectoryW(dir, NULL);
1604 Safefree(dir);
1605 }
1606 else {
1607 result = CreateDirectoryA(SvPV_nolen(ST(0)), NULL);
1608 }
1609
1610 ST(0) = boolSV(result);
1611 XSRETURN(1);
1612}
1613
1614XS(w32_CreateFile)
b4ad57f4 1615{
1616 dXSARGS;
34f7f30d 1617 HANDLE handle;
1618
1619 if (items != 1)
1620 Perl_croak(aTHX_ "usage: Win32::CreateFile($file)");
1621
1622 if (IsWin2000() && SvUTF8(ST(0))) {
1623 WCHAR *file = sv_to_wstr(aTHX_ ST(0));
1624 handle = CreateFileW(file, GENERIC_WRITE, FILE_SHARE_WRITE,
1625 NULL, CREATE_NEW, FILE_ATTRIBUTE_NORMAL, NULL);
1626 Safefree(file);
1627 }
1628 else {
1629 handle = CreateFileA(SvPV_nolen(ST(0)), GENERIC_WRITE, FILE_SHARE_WRITE,
1630 NULL, CREATE_NEW, FILE_ATTRIBUTE_NORMAL, NULL);
1631 }
1632
1633 if (handle != INVALID_HANDLE_VALUE)
1634 CloseHandle(handle);
1635
1636 ST(0) = boolSV(handle != INVALID_HANDLE_VALUE);
1637 XSRETURN(1);
1638}
1639
1640MODULE = Win32 PACKAGE = Win32
1641
1642PROTOTYPES: DISABLE
1643
1644BOOT:
1645{
b4ad57f4 1646 char *file = __FILE__;
1647
cf8b4e93 1648 if (g_osver.dwOSVersionInfoSize == 0) {
1649 g_osver.dwOSVersionInfoSize = sizeof(g_osver);
34f7f30d 1650 if (!GetVersionExA((OSVERSIONINFOA*)&g_osver)) {
1651 g_osver_ex = FALSE;
1652 g_osver.dwOSVersionInfoSize = sizeof(OSVERSIONINFOA);
1653 GetVersionExA((OSVERSIONINFOA*)&g_osver);
1654 }
cf8b4e93 1655 }
1656
b4ad57f4 1657 newXS("Win32::LookupAccountName", w32_LookupAccountName, file);
1658 newXS("Win32::LookupAccountSID", w32_LookupAccountSID, file);
1659 newXS("Win32::InitiateSystemShutdown", w32_InitiateSystemShutdown, file);
1660 newXS("Win32::AbortSystemShutdown", w32_AbortSystemShutdown, file);
1661 newXS("Win32::ExpandEnvironmentStrings", w32_ExpandEnvironmentStrings, file);
1662 newXS("Win32::MsgBox", w32_MsgBox, file);
1663 newXS("Win32::LoadLibrary", w32_LoadLibrary, file);
1664 newXS("Win32::FreeLibrary", w32_FreeLibrary, file);
1665 newXS("Win32::GetProcAddress", w32_GetProcAddress, file);
1666 newXS("Win32::RegisterServer", w32_RegisterServer, file);
1667 newXS("Win32::UnregisterServer", w32_UnregisterServer, file);
1668 newXS("Win32::GetArchName", w32_GetArchName, file);
1669 newXS("Win32::GetChipName", w32_GetChipName, file);
1670 newXS("Win32::GuidGen", w32_GuidGen, file);
1671 newXS("Win32::GetFolderPath", w32_GetFolderPath, file);
1672 newXS("Win32::IsAdminUser", w32_IsAdminUser, file);
e364e11c 1673 newXS("Win32::GetFileVersion", w32_GetFileVersion, file);
b4ad57f4 1674
cf8b4e93 1675 newXS("Win32::GetCwd", w32_GetCwd, file);
1676 newXS("Win32::SetCwd", w32_SetCwd, file);
1677 newXS("Win32::GetNextAvailDrive", w32_GetNextAvailDrive, file);
1678 newXS("Win32::GetLastError", w32_GetLastError, file);
1679 newXS("Win32::SetLastError", w32_SetLastError, file);
1680 newXS("Win32::LoginName", w32_LoginName, file);
1681 newXS("Win32::NodeName", w32_NodeName, file);
1682 newXS("Win32::DomainName", w32_DomainName, file);
1683 newXS("Win32::FsType", w32_FsType, file);
1684 newXS("Win32::GetOSVersion", w32_GetOSVersion, file);
1685 newXS("Win32::IsWinNT", w32_IsWinNT, file);
1686 newXS("Win32::IsWin95", w32_IsWin95, file);
1687 newXS("Win32::FormatMessage", w32_FormatMessage, file);
1688 newXS("Win32::Spawn", w32_Spawn, file);
1689 newXS("Win32::GetTickCount", w32_GetTickCount, file);
1690 newXS("Win32::GetShortPathName", w32_GetShortPathName, file);
1691 newXS("Win32::GetFullPathName", w32_GetFullPathName, file);
1692 newXS("Win32::GetLongPathName", w32_GetLongPathName, file);
34f7f30d 1693 newXS("Win32::GetANSIPathName", w32_GetANSIPathName, file);
cf8b4e93 1694 newXS("Win32::CopyFile", w32_CopyFile, file);
1695 newXS("Win32::Sleep", w32_Sleep, file);
34f7f30d 1696 newXS("Win32::OutputDebugString", w32_OutputDebugString, file);
1697 newXS("Win32::GetCurrentThreadId", w32_GetCurrentThreadId, file);
1698 newXS("Win32::CreateDirectory", w32_CreateDirectory, file);
1699 newXS("Win32::CreateFile", w32_CreateFile, file);
5bd7e777 1700#ifdef __CYGWIN__
cf8b4e93 1701 newXS("Win32::SetChildShowWindow", w32_SetChildShowWindow, file);
5bd7e777 1702#endif
b4ad57f4 1703 XSRETURN_YES;
1704}