1 /*
2 * Services.exe - some utility functions
3 *
4 * Copyright 2007 Google (Mikolaj Zalewski)
5 *
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * This library 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 GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19 */
20
21 #define WIN32_LEAN_AND_MEAN
22
23 #include <stdarg.h>
24 #include <windows.h>
25
26 #include "wine/unicode.h"
27 #include "wine/debug.h"
28 #include "services.h"
29
30 WINE_DEFAULT_DEBUG_CHANNEL(service);
31
32 LPWSTR strdupW(LPCWSTR str)
33 {
34 int len;
35 WCHAR *buf;
36
37 if (str == NULL)
38 return NULL;
39 len = strlenW(str);
40 buf = HeapAlloc(GetProcessHeap(), 0, sizeof(WCHAR)*(len+1));
41 if (buf == NULL)
42 return NULL;
43 strcpyW(buf, str);
44 return buf;
45 }
46
47 BOOL check_multisz(LPCWSTR lpMultiSz, DWORD cbSize)
48 {
49 if (cbSize == 0 || (cbSize == sizeof(WCHAR) && lpMultiSz[0] == 0))
50 return TRUE;
51 if ((cbSize % sizeof(WCHAR)) != 0 || cbSize < 2*sizeof(WCHAR))
52 return FALSE;
53 if (lpMultiSz[cbSize/2 - 1] || lpMultiSz[cbSize/2 - 2])
54 return FALSE;
55 return TRUE;
56 }
57
58 DWORD load_reg_string(HKEY hKey, LPCWSTR szValue, BOOL bExpand, LPWSTR *output)
59 {
60 DWORD size, type;
61 LPWSTR buf = NULL;
62 DWORD err;
63
64 *output = NULL;
65 if ((err = RegQueryValueExW(hKey, szValue, 0, &type, NULL, &size)) != 0)
66 {
67 if (err == ERROR_FILE_NOT_FOUND)
68 return ERROR_SUCCESS;
69 goto failed;
70 }
71 if (!(type == REG_SZ || (type == REG_EXPAND_SZ && bExpand)))
72 {
73 err = ERROR_INVALID_DATATYPE;
74 goto failed;
75 }
76 buf = HeapAlloc(GetProcessHeap(), 0, size + sizeof(WCHAR));
77 if ((err = RegQueryValueExW(hKey, szValue, 0, &type, (LPBYTE)buf, &size)) != 0)
78 goto failed;
79 buf[size/sizeof(WCHAR)] = 0;
80
81 if (type == REG_EXPAND_SZ)
82 {
83 LPWSTR str;
84 DWORD size = ExpandEnvironmentStringsW(buf, NULL, 0);
85 if (size == 0)
86 {
87 err = GetLastError();
88 goto failed;
89 }
90 str = HeapAlloc(GetProcessHeap(), 0, size * sizeof(WCHAR));
91 ExpandEnvironmentStringsW(buf, str, size);
92 HeapFree(GetProcessHeap(), 0, buf);
93 *output = str;
94 }
95 else
96 *output = buf;
97 return ERROR_SUCCESS;
98
99 failed:
100 WINE_ERR("Error %d while reading value %s\n", err, wine_dbgstr_w(szValue));
101 HeapFree(GetProcessHeap(), 0, buf);
102 return err;
103 }
104
105 DWORD load_reg_multisz(HKEY hKey, LPCWSTR szValue, BOOL bAllowSingle, LPWSTR *output)
106 {
107 DWORD size, type;
108 LPWSTR buf = NULL;
109 DWORD err;
110
111 *output = NULL;
112 if ((err = RegQueryValueExW(hKey, szValue, 0, &type, NULL, &size)) != 0)
113 {
114 if (err == ERROR_FILE_NOT_FOUND)
115 {
116 *output = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(WCHAR));
117 return ERROR_SUCCESS;
118 }
119 goto failed;
120 }
121 if (!((type == REG_MULTI_SZ) || ((type == REG_SZ) && bAllowSingle)))
122 {
123 err = ERROR_INVALID_DATATYPE;
124 goto failed;
125 }
126 buf = HeapAlloc(GetProcessHeap(), 0, size + 2*sizeof(WCHAR));
127 if ((err = RegQueryValueExW(hKey, szValue, 0, &type, (LPBYTE)buf, &size)) != 0)
128 goto failed;
129 buf[size/sizeof(WCHAR)] = 0;
130 buf[size/sizeof(WCHAR) + 1] = 0;
131 *output = buf;
132 return ERROR_SUCCESS;
133
134 failed:
135 WINE_ERR("Error %d while reading value %s\n", err, wine_dbgstr_w(szValue));
136 HeapFree(GetProcessHeap(), 0, buf);
137 return err;
138 }
139
140 DWORD load_reg_dword(HKEY hKey, LPCWSTR szValue, DWORD *output)
141 {
142 DWORD size, type;
143 DWORD err;
144
145 *output = 0;
146 size = sizeof(DWORD);
147 if ((err = RegQueryValueExW(hKey, szValue, 0, &type, (LPBYTE)output, &size)) != 0)
148 {
149 if (err == ERROR_FILE_NOT_FOUND)
150 return ERROR_SUCCESS;
151 goto failed;
152 }
153 if (type != REG_DWORD || size != sizeof(DWORD))
154 {
155 err = ERROR_INVALID_DATATYPE;
156 goto failed;
157 }
158 return ERROR_SUCCESS;
159
160 failed:
161 WINE_ERR("Error %d while reading value %s\n", err, wine_dbgstr_w(szValue));
162 return err;
163 }
164
This page was automatically generated by the
LXR engine.
Visit the LXR main site for more
information.