ws2_32: Use better types for some variables.
[wine] / programs / services / utils.c
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 }