msi: Validate the szProduct parameter of MsiConfigureProductEx.
[wine/gsoc_dplay.git] / programs / services / utils.c
blob89eb5007822b753041b520c4bb287933eb8ab48d
1 /*
2 * Services.exe - some utility functions
4 * Copyright 2007 Google (Mikolaj Zalewski)
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.
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.
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
21 #define WIN32_LEAN_AND_MEAN
23 #include <stdarg.h>
24 #include <windows.h>
26 #include "wine/unicode.h"
27 #include "wine/debug.h"
28 #include "services.h"
30 WINE_DEFAULT_DEBUG_CHANNEL(service);
32 LPWSTR strdupW(LPCWSTR str)
34 int len;
35 WCHAR *buf;
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;
47 BOOL check_multisz(LPCWSTR lpMultiSz, DWORD cbSize)
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;
58 DWORD load_reg_string(HKEY hKey, LPCWSTR szValue, BOOL bExpand, LPWSTR *output)
60 DWORD size, type;
61 LPWSTR buf = NULL;
62 DWORD err;
64 *output = NULL;
65 if ((err = RegQueryValueExW(hKey, szValue, 0, &type, NULL, &size)) != 0)
67 if (err == ERROR_FILE_NOT_FOUND)
68 return ERROR_SUCCESS;
69 goto failed;
71 if (!(type == REG_SZ || (type == REG_EXPAND_SZ && bExpand)))
73 err = ERROR_INVALID_DATATYPE;
74 goto failed;
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;
81 if (type == REG_EXPAND_SZ)
83 LPWSTR str;
84 DWORD size = ExpandEnvironmentStringsW(buf, NULL, 0);
85 if (size == 0)
87 err = GetLastError();
88 goto failed;
90 str = HeapAlloc(GetProcessHeap(), 0, size * sizeof(WCHAR));
91 ExpandEnvironmentStringsW(buf, str, size);
92 HeapFree(GetProcessHeap(), 0, buf);
93 *output = str;
95 else
96 *output = buf;
97 return ERROR_SUCCESS;
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;
105 DWORD load_reg_multisz(HKEY hKey, LPCWSTR szValue, LPWSTR *output)
107 DWORD size, type;
108 LPWSTR buf = NULL;
109 DWORD err;
111 *output = NULL;
112 if ((err = RegQueryValueExW(hKey, szValue, 0, &type, NULL, &size)) != 0)
114 if (err == ERROR_FILE_NOT_FOUND)
116 *output = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(WCHAR));
117 return ERROR_SUCCESS;
119 goto failed;
121 if (type != REG_MULTI_SZ)
123 err = ERROR_INVALID_DATATYPE;
124 goto failed;
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;
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;
140 DWORD load_reg_dword(HKEY hKey, LPCWSTR szValue, DWORD *output)
142 DWORD size, type;
143 DWORD err;
145 *output = 0;
146 size = sizeof(DWORD);
147 if ((err = RegQueryValueExW(hKey, szValue, 0, &type, (LPBYTE)output, &size)) != 0)
149 if (err == ERROR_FILE_NOT_FOUND)
150 return ERROR_SUCCESS;
151 goto failed;
153 if (type != REG_DWORD || size != sizeof(DWORD))
155 err = ERROR_INVALID_DATATYPE;
156 goto failed;
158 return ERROR_SUCCESS;
160 failed:
161 WINE_ERR("Error %d while reading value %s\n", err, wine_dbgstr_w(szValue));
162 return err;