/[winpt]/trunk/Src/wptW32API.cpp
ViewVC logotype

Contents of /trunk/Src/wptW32API.cpp

Parent Directory Parent Directory | Revision Log Revision Log


Revision 376 - (show annotations)
Fri Dec 9 12:25:36 2011 UTC (13 years, 2 months ago) by twoaday
File size: 16322 byte(s)


1 /* wptW32API.cpp - Common W32 API functions
2 * Copyright (C) 2001-2003, 2005-2006, 2009 Timo Schulz
3 *
4 * This file is part of WinPT.
5 *
6 * WinPT is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
10 *
11 * WinPT is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
15 */
16 #ifdef HAVE_CONFIG_H
17 #include <config.h>
18 #endif
19
20 #include <windows.h>
21 #include <stdio.h>
22 #include <sys/types.h>
23 #include <sys/stat.h>
24 #include <shellapi.h>
25 #include <shlobj.h>
26 #include <commctrl.h>
27 #include <time.h>
28
29 #include "wptNLS.h"
30 #include "wptW32API.h"
31 #include "wptErrors.h"
32 #include "wptVersion.h"
33 #include "wptUTF8.h"
34 #include "wptTypes.h"
35 #include "StringBuffer.h"
36
37
38 extern "C" void _SHFree (void *p);
39
40
41 /* Insert a new item into the menu @hm at position @pos. */
42 void
43 insert_menu_item (HMENU hm, int pos, UINT m_id, const char *text)
44 {
45 MENUITEMINFO mi;
46
47 memset (&mi, 0, sizeof mi);
48 mi.cbSize = sizeof mi;
49 mi.fType = MF_STRING;
50 mi.dwTypeData = (char *)text;
51 mi.cch = strlen (text);
52 mi.wID = m_id;
53 mi.fMask = MIIM_ID|MIIM_DATA| MIIM_TYPE;
54 InsertMenuItem (hm, pos, FALSE, &mi);
55 }
56
57
58 static void
59 set_menu_text_ext (HMENU menu, int by_pos, int m_uid, const char *text)
60 {
61 MENUITEMINFO mii;
62
63 memset (&mii, 0, sizeof mii);
64 mii.cbSize = sizeof mii;
65 mii.fMask = MIIM_TYPE;
66 mii.fType = MFT_STRING;
67 mii.dwTypeData = (char *) text;
68 SetMenuItemInfo (menu, m_uid, by_pos? TRUE : FALSE, &mii);
69 }
70
71
72 /* Set the text of a menu item @m_uid to @text. */
73 void
74 set_menu_text (HMENU menu, UINT m_uid, const char *text)
75 {
76 set_menu_text_ext (menu, 0, m_uid, text);
77 }
78
79
80 /* Set the text of a menu item with the position @pos to @text. */
81 void
82 set_menu_text_bypos (HMENU menu, UINT pos, const char *text)
83 {
84 set_menu_text_ext (menu, 1, pos, text);
85 }
86
87
88 /* Set the state of a menu item @m_uid to @state. */
89 void
90 set_menu_state (HMENU menu, UINT m_uid, UINT state)
91 {
92 MENUITEMINFO mii;
93
94 memset (&mii, 0, sizeof (mii));
95 mii.cbSize = sizeof (mii);
96 mii.fMask = MIIM_STATE;
97 mii.fState = state;
98 SetMenuItemInfo (menu, m_uid, FALSE, &mii);
99 }
100
101
102 /* Retrieve the state of the menu item @m_uid and return it. */
103 UINT
104 get_menu_state (HMENU menu, UINT m_uid)
105 {
106 MENUITEMINFO mii;
107
108 memset (&mii, 0, sizeof (mii));
109 mii.cbSize = sizeof (mii);
110 mii.fMask = MIIM_STATE;
111 GetMenuItemInfo (menu, m_uid, FALSE, &mii);
112 return mii.fState;
113 }
114
115
116 enum {
117 CDLG_FILE_OPEN = 0,
118 CDLG_FILE_SAVE = 1
119 };
120
121 /* Use the common dialog to request a file from the user.
122 id can be either FILE_OPEN or FILE_SAVE.
123 The return value is the file name or NULL if cancel was chosen. */
124 const char *
125 get_filename_dlg (HWND hwnd, int id, const char *title,
126 const char *filter, const char *name)
127 {
128 static char file[2*MAX_PATH+1] = "";
129 char _filter[512];
130 char workdir[MAX_PATH+1];
131 char *sel_file = NULL;
132 OPENFILENAME open;
133
134 if (name && strlen (name) < (DIM (file)-1))
135 strcpy (file, name);
136 else
137 memset (file, 0, sizeof (file));
138 if (!filter) {
139 strcpy (_filter, _("All Files (*.*)"));
140 memcpy (_filter+strlen (_filter)+1, "*.*\0\0", 5);
141 filter = _filter;
142 }
143
144 GetCurrentDirectory (MAX_PATH, workdir);
145 memset (&open, 0, sizeof (open));
146 open.lStructSize = sizeof (OPENFILENAME);
147 open.hInstance = glob_hinst;
148 open.lpstrTitle = title;
149 open.lpstrFilter = filter;
150 open.hwndOwner = hwnd;
151 open.lpstrFile = file;
152 open.nMaxFile = DIM (file) - 1;
153 if (id == CDLG_FILE_OPEN)
154 open.Flags = OFN_FILEMUSTEXIST|OFN_PATHMUSTEXIST;
155 else
156 open.Flags = OFN_OVERWRITEPROMPT;
157
158 if (id == CDLG_FILE_OPEN && GetOpenFileName (&open))
159 sel_file = open.lpstrFile;
160 else if (id == CDLG_FILE_SAVE && GetSaveFileName (&open))
161 sel_file = open.lpstrFile;
162 SetCurrentDirectory (workdir);
163
164 return sel_file;
165 }
166
167 const char*
168 get_filesave_dlg (HWND hwnd, const char *title,
169 const char *filter, const char *name)
170 {
171 return get_filename_dlg (hwnd, CDLG_FILE_SAVE, title, filter, name);
172 }
173
174 const char*
175 get_fileopen_dlg (HWND hwnd, const char *title, const char *filter,
176 const char *name)
177 {
178 return get_filename_dlg (hwnd, CDLG_FILE_OPEN, title, filter, name);
179 }
180
181
182 /* Use the common dialog to allow the user to select a folder.
183 The return value is either the folder path or NULL if cancel was chosen. */
184 const char*
185 get_folder_dlg (HWND hwnd, const char *title, const char *name)
186 {
187 static char folder[2*MAX_PATH+1] = "";
188 BROWSEINFO bi;
189 ITEMIDLIST *il;
190
191 memset (&bi, 0, sizeof (bi));
192 bi.hwndOwner = hwnd;
193 if (title)
194 bi.lpszTitle = title;
195 if (name && strlen (name) < MAX_PATH-1)
196 strcpy (folder, name);
197 else
198 memset (folder, 0, sizeof (folder));
199 il = SHBrowseForFolder (&bi);
200 if (il != NULL) {
201 SHGetPathFromIDList (il, folder);
202 _SHFree (il);
203 return folder;
204 }
205 return NULL;
206 }
207
208
209 /* Return the clipboard contents as a string or NULL
210 if the clipboard does not contain text. */
211 char*
212 get_clip_text (HWND hwnd)
213 {
214 HANDLE clipmem;
215 char *cliptxt, *p;
216 int len;
217
218 if (OpenClipboard (hwnd) == FALSE)
219 return NULL;
220 clipmem = GetClipboardData (CF_TEXT);
221 if (clipmem == NULL) {
222 p = NULL;
223 goto leave;
224 }
225 cliptxt = (char *) GlobalLock (clipmem);
226 if (cliptxt == NULL) {
227 p = NULL;
228 goto leave;
229 }
230
231 len = strlen (cliptxt);
232 p = new char[len + 1];
233 if (!p)
234 BUG (NULL);
235 memcpy (p, cliptxt, len);
236 p[len] = '\0';
237 GlobalUnlock (clipmem);
238
239 leave:
240 CloseClipboard ();
241 return p;
242 }
243
244
245 /* Set @text as the new clipboard content. */
246 int
247 set_clip_text (HWND hwnd, const char *text, int nbytes)
248 {
249 HANDLE clipmem;
250 int rc = 0;
251 char *p;
252
253 if (OpenClipboard (hwnd) == FALSE)
254 return WPTERR_CLIP_OPEN;
255 EmptyClipboard ();
256
257 clipmem = GlobalAlloc (GHND, nbytes + 1);
258 if (clipmem == NULL)
259 BUG (NULL);
260 p = (char *) GlobalLock (clipmem);
261 if (p == NULL) {
262 CloseClipboard ();
263 GlobalFree (clipmem);
264 return WPTERR_GENERAL;
265 }
266 memcpy (p, text, nbytes);
267 p[nbytes] = '\0';
268
269 SetClipboardData (CF_TEXT, clipmem);
270 GlobalUnlock (clipmem);
271 CloseClipboard ();
272 GlobalFree (clipmem);
273
274 return rc;
275 }
276
277
278 /* Append or prepend some text to the clipboard contents.
279 If as_footer = 1, append the text otherwise prepend. */
280 int
281 set_clip_text2 (HWND hwnd, const char *text, int nbytes, int as_footer)
282 {
283 const char *fmt;
284 char *p, *new_text;
285
286 p = get_clip_text (hwnd);
287 if (!p)
288 return WPTERR_CLIP_GET;
289 if (as_footer == 0)
290 fmt = "%s\r\n%s\r\n\r\n";
291 else
292 fmt = "%s\n%s\n\n";
293 new_text = new char [strlen (p)+strlen (text)+strlen (fmt)+2];
294 if (!new_text)
295 BUG (0);
296 if (as_footer == 0)
297 sprintf (new_text, fmt, text, p);
298 else
299 sprintf (new_text, fmt, p, text);
300 set_clip_text (hwnd, new_text, strlen (new_text)+1);
301 free_if_alloc (p);
302 free_if_alloc (new_text);
303 return 0;
304 }
305
306
307 /* Make a file name out of the path @path, the file @file and
308 an extension. @ext.
309 Return value: the full file name on success. */
310 char*
311 make_filename (const char *path, const char *file, const char *ext)
312 {
313 char *p;
314 size_t size = 0;
315
316 if (path && *path)
317 size += strlen (path);
318 if (file && *file)
319 size += strlen (file);
320 if (ext && *ext)
321 size += strlen (ext);
322 p = new char[size + 4];
323 if (!p)
324 BUG (0);
325 memset (p, 0, size);
326 if (path) {
327 strcat (p, path);
328 if (path[strlen (path) -1] != '\\')
329 strcat (p, "\\");
330 }
331 if (file != NULL)
332 strcat (p, file);
333 if (ext != NULL) {
334 strcat (p, ".");
335 strcat (p, ext);
336 }
337 return p;
338 }
339
340
341 /* Generates a file name from a special dirctory. */
342 char*
343 make_special_filename (int folder, const char *file, const char *ext)
344 {
345 BOOL ec;
346 char path[MAX_PATH+1], *p;
347 size_t n=0;
348
349 /* MSDN: buf must be at least MAX_PATH=256 bytes */
350 ec = SHGetSpecialFolderPath (HWND_DESKTOP, path, folder, TRUE);
351 if (ec != 1) {
352 log_debug ("SHGetSpecialFolderPath() failed\r\n", (int)GetLastError ());
353 return NULL;
354 }
355
356 n = strlen (path)+1;
357 if (file)
358 n += strlen (file)+1;
359 if (ext)
360 n += strlen (ext)+1;
361 p = new char[n+2];
362 if (!p)
363 BUG (0);
364 memset (p, 0, n+2);
365 strcpy (p, path);
366 if (file) {
367 strcat (p, "\\");
368 strcat (p, file);
369 }
370 if (ext)
371 strcat (p, ext);
372 return p;
373 }
374
375
376 /* return 0 if the file @fname exists, otherwise >0. */
377 int
378 file_exist_check (const char *fname)
379 {
380 struct stat st;
381 if (stat (fname, &st) == -1)
382 return WPTERR_FILE_EXIST;
383 return 0;
384 }
385
386
387 /* Check if the current folder exists.
388 Return 0 for success. */
389 int
390 dir_exist_check (const char *dir)
391 {
392 struct stat statbuf;
393
394 if (stat (dir, &statbuf) == -1)
395 return WPTERR_GENERAL;
396 if (statbuf.st_mode & _S_IFDIR)
397 return 0;
398 return WPTERR_GENERAL;
399 }
400
401
402 /* Return the file size of the given file @fname. */
403 DWORD
404 get_file_size (const char *fname)
405 {
406 DWORD fsize;
407 HANDLE fh;
408
409 fh = CreateFile (fname, GENERIC_READ, FILE_SHARE_READ,
410 NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
411 if (fh == INVALID_HANDLE_VALUE)
412 return 0;
413 fsize = GetFileSize (fh, NULL);
414 if (fsize == 0xFFFFFFFF)
415 fsize = 0;
416 CloseHandle (fh);
417 return fsize;
418 }
419
420
421 /* Start a dialog with the exception that before it is checked that the
422 dialog is not already openened. */
423 int
424 dialog_box_param (HINSTANCE hinst, LPCTSTR name, HWND parent, DLGPROC fnc,
425 LPARAM param, LPCTSTR title, int title_id)
426 {
427 if (FindWindowEx (GetDesktopWindow (), NULL, NULL, title))
428 return -1;
429 return DialogBoxParam (hinst, name, parent, fnc, param);
430 }
431
432
433 /* Wrapper for message box which forces the message box into the
434 foreground and it is displayed always on top. */
435 int
436 msg_box (HWND hwnd, const char *text, const char *title, int mode)
437 {
438 mode |= MB_SETFOREGROUND;
439 mode |= MB_TASKMODAL;
440 mode |= MB_TOPMOST;
441 return MessageBox(hwnd, text, title, mode);
442 }
443
444
445 /* Center the hwndChild relative to parent.
446 The style param allows to specificy additional styles (like topmost). */
447 void
448 center_window2 (HWND hwndChild, HWND parent, HWND style)
449 {
450 HWND hwndParent;
451 RECT rChild, rParent;
452 HDC hdc;
453 int wChild, hChild, wParent, hParent;
454 int wScreen, hScreen, xNew, yNew;
455 int flags = SWP_NOSIZE | SWP_NOZORDER;
456
457 hwndParent = parent;
458 if (hwndParent == NULL)
459 hwndParent = GetDesktopWindow ();
460 GetWindowRect (hwndChild, &rChild);
461 wChild = rChild.right - rChild.left;
462 hChild = rChild.bottom - rChild.top;
463
464 GetWindowRect (hwndParent, &rParent);
465 wParent = rParent.right - rParent.left;
466 hParent = rParent.bottom - rParent.top;
467
468 hdc = GetDC (hwndChild);
469 wScreen = GetDeviceCaps (hdc, HORZRES);
470 hScreen = GetDeviceCaps (hdc, VERTRES);
471 ReleaseDC (hwndChild, hdc);
472 xNew = rParent.left + ((wParent - wChild) /2);
473 if (xNew < 0)
474 xNew = 0;
475 else if ((xNew+wChild) > wScreen)
476 xNew = wScreen - wChild;
477 yNew = rParent.top + ((hParent - hChild) /2);
478 if (yNew < 0)
479 yNew = 0;
480 else if ((yNew+hChild) > hScreen)
481 yNew = hScreen - hChild;
482 if (style == HWND_TOPMOST || style == HWND_NOTOPMOST)
483 flags = SWP_NOMOVE | SWP_NOSIZE;
484 SetWindowPos (hwndChild, style? style : NULL, xNew, yNew, 0, 0, flags);
485 }
486
487
488 /* Center the given hwndChild window with no special style. */
489 void
490 center_window (HWND child, HWND parent)
491 {
492 center_window2 (child, parent, NULL);
493 }
494
495
496
497
498 /* Return date in a format which complies with the
499 locale user settings. */
500 const char*
501 get_locale_date (long tm_t, char *buf, DWORD buflen)
502 {
503 SYSTEMTIME st;
504 struct tm *ptm;
505
506 ptm = localtime (&tm_t);
507 memset (&st, 0, sizeof (st));
508 st.wYear = (WORD)ptm->tm_year+1900;
509 st.wMonth = (WORD)ptm->tm_mon+1;
510 st.wDay = (WORD)ptm->tm_mday;
511 if (!GetDateFormat (LOCALE_USER_DEFAULT, DATE_SHORTDATE, &st,
512 NULL, buf, buflen))
513 return NULL;
514 return buf;
515 }
516
517 /* Return the time in a format which complies with the locale user
518 settings. */
519 const char*
520 get_locale_time (long tm_t, char *buf, DWORD buflen)
521 {
522 SYSTEMTIME st;
523 struct tm *ptm;
524
525 ptm = localtime (&tm_t);
526 memset (&st, 0, sizeof (st));
527 st.wMinute = (WORD)ptm->tm_min;
528 st.wHour = (WORD)ptm->tm_hour;
529 st.wSecond = (WORD)ptm->tm_sec;
530
531 if (!GetTimeFormat (LOCALE_USER_DEFAULT, TIME_FORCE24HOURFORMAT, &st,
532 NULL, buf, buflen))
533 return NULL;
534 return buf;
535 }
536
537
538 /* Return a combination of date+time which complies to the
539 local user settings. */
540 const char*
541 get_locale_timedate (long tm_t, char *buf, DWORD buflen)
542 {
543 DWORD nleft = buflen;
544
545 if (!get_locale_date (tm_t, buf, buflen))
546 return NULL;
547 nleft -= strlen (buf)+1;
548 if (nleft < 1)
549 return NULL;
550 strcat (buf, " ");
551 nleft--;
552 if (!get_locale_time (tm_t, buf+strlen (buf), nleft))
553 return NULL;
554 return buf;
555 }
556
557
558 /* Generate a temporary file name by using the users
559 temp path and optionally a name @name provided by the caller.
560 Return value: 0 on success. */
561 int
562 get_temp_name (char *buf, DWORD buflen, const char *name)
563 {
564 char tmp[32];
565
566
567 if (!name) {
568 _snprintf (tmp, DIM (tmp)-1, "%08lX", GetTickCount ());
569 name = tmp;
570 }
571
572 if (!GetTempPath (buflen - strlen (name) -2, buf)) {
573 log_debug ("GetTempPath() failed ec=%d\n", (int)GetLastError ());
574 return -1;
575 }
576
577 strcat (buf, name);
578 return 0;
579 }
580
581
582 void
583 ListBox_AddString_utf8 (HWND lb, const char *txt)
584 {
585 char *utf8_txt;
586
587 utf8_txt = utf8_to_native (txt);
588 SendMessage (lb, LB_ADDSTRING, 0, (LPARAM)(LPCSTR)(utf8_txt));
589 safe_free (utf8_txt);
590 }
591
592
593 void
594 ComboBox_AddString_utf8 (HWND cb, const char *txt)
595 {
596 char *utf8_txt;
597
598 utf8_txt = utf8_to_native (txt);
599 SendMessage ((cb), CB_ADDSTRING, 0, (LPARAM)(LPCSTR)(utf8_txt));
600 safe_free (utf8_txt);
601 }
602
603
604 /* GetDlgItemText replacement with UTF8 support. */
605 int
606 GetDlgItemText_utf8 (HWND dlg, int id, char **r_txt)
607 {
608 int len = GetWindowTextLength (GetDlgItem (dlg, id));
609 char *txt;
610
611 *r_txt = NULL;
612 if (len < 1)
613 return 0;
614 txt = new char[len+2];
615 if (!txt)
616 BUG (NULL);
617 GetDlgItemText (dlg, id, txt, len+1);
618 *r_txt = native_to_utf8 (txt);
619 free_if_alloc (txt);
620 return len;
621 }
622
623 int
624 html_help_show (int map_id)
625 {
626 STARTUPINFO si;
627 PROCESS_INFORMATION pi;
628 SECURITY_ATTRIBUTES sec_attr;
629 char path[2*MAX_PATH+1];
630
631 memset (&sec_attr, 0, sizeof (sec_attr));
632 memset (&si, 0, sizeof (si));
633 si.cb = sizeof (STARTUPINFO);
634 memset (&pi, 0, sizeof (pi));
635 //hh.exe -mapid 40 "winpt.$lang.chm"
636
637 _snprintf (path, 2*MAX_PATH, "hh.exe -mapid %d winpt.%s.chm", map_id, gettext_get_langid ());
638 CreateProcess (NULL, path, &sec_attr, &sec_attr, FALSE, 0, NULL, NULL, &si, &pi);
639 CloseHandle (pi.hProcess);
640 CloseHandle (pi.hThread);
641 return 0;
642 }
643
644
645 /* Return TRUE if the current user has admin privileges. */
646 BOOL
647 user_is_admin (void)
648 {
649 SID_IDENTIFIER_AUTHORITY SystemSidAuthority = {SECURITY_NT_AUTHORITY};
650 HANDLE hd;
651 TOKEN_GROUPS *ptg = NULL;
652 DWORD ngtoken;
653 DWORD i;
654 BOOL admin = FALSE;
655 PSID psid = 0;
656
657 if (!OpenThreadToken (GetCurrentThread (), TOKEN_QUERY, FALSE, &hd) &&
658 !OpenProcessToken (GetCurrentProcess (), TOKEN_QUERY, &hd))
659 return FALSE;
660
661 if (!GetTokenInformation (hd, TokenGroups, NULL, 0, &ngtoken) &&
662 GetLastError() == ERROR_INSUFFICIENT_BUFFER) {
663 ptg = (TOKEN_GROUPS*)GlobalAlloc (GPTR, ngtoken);
664 if (!ptg)
665 return FALSE;
666
667 if (!GetTokenInformation (hd, TokenGroups, ptg, ngtoken, &ngtoken)) {
668 GlobalFree (ptg);
669 return FALSE;
670 }
671 AllocateAndInitializeSid (&SystemSidAuthority,
672 2, SECURITY_BUILTIN_DOMAIN_RID,
673 DOMAIN_ALIAS_RID_ADMINS,
674 0, 0, 0, 0, 0, 0,
675 &psid);
676 for (i = 0; i < ptg->GroupCount; i++) {
677 if (EqualSid (ptg->Groups[i].Sid, psid)) {
678 admin = TRUE;
679 break;
680 }
681 }
682 FreeSid (psid);
683 GlobalFree (ptg);
684 }
685
686 CloseHandle (hd);
687 return admin;
688 }

Properties

Name Value
svn:eol-style native

[email protected]
ViewVC Help
Powered by ViewVC 1.1.26