648cd3d00296ac071579d1b0e1d248c6a8dbf09d
[reactos.git] / dll / cpl / inetcpl / connections.c
1 /*
2 * Copyright 2018 Piotr cabna for CodeWeavers
3 *
4 * This library is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
8 *
9 * This library is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
13 *
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with this library; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
17 *
18 */
19
20 #include <stdarg.h>
21 #include <windef.h>
22 #include <winbase.h>
23 #include <wininet.h>
24 #include <winuser.h>
25 #include <winreg.h>
26
27 #include "inetcpl.h"
28 #include "wine/debug.h"
29 #include "wine/heap.h"
30 #include "wine/unicode.h"
31
32 WINE_DEFAULT_DEBUG_CHANNEL(inetcpl);
33
34 static const WCHAR auto_config_url[] = {'A','u','t','o','C','o','n','f','i','g','U','R','L',0};
35 static const WCHAR internet_settings[] = {'S','o','f','t','w','a','r','e','\\',
36 'M','i','c','r','o','s','o','f','t','\\','W','i','n','d','o','w','s','\\',
37 'C','u','r','r','e','n','t','V','e','r','s','i','o','n','\\',
38 'I','n','t','e','r','n','e','t',' ','S','e','t','t','i','n','g','s',0};
39 static const WCHAR proxy_enable[] = {'P','r','o','x','y','E','n','a','b','l','e',0};
40 static const WCHAR proxy_server[] = {'P','r','o','x','y','S','e','r','v','e','r',0};
41 static const WCHAR connections[] = {'C','o','n','n','e','c','t','i','o','n','s',0};
42 static const WCHAR default_connection_settings[] = {'D','e','f','a','u','l','t',
43 'C','o','n','n','e','c','t','i','o','n','S','e','t','t','i','n','g','s',0};
44
45 static BOOL initdialog_done;
46
47 #define CONNECTION_SETTINGS_VERSION 0x46
48 #define CONNECTION_SETTINGS_MANUAL_PROXY 0x2
49 #define CONNECTION_SETTINGS_PAC_SCRIPT 0x4
50 #define CONNECTION_SETTINGS_WPAD 0x8
51
52 typedef struct {
53 DWORD version;
54 DWORD id;
55 DWORD flags;
56 BYTE data[1];
57 /* DWORD proxy_server_len; */
58 /* UTF8 proxy_server[proxy_server_len]; */
59 /* DWORD bypass_list_len; */
60 /* UTF8 bypass_list[bypass_list_len]; */
61 /* DWORD configuration_script_len; */
62 /* UTF8 configuration_script[configuration_script_len]; */
63 /* DWORD unk[8]; set to 0 */
64 } connection_settings;
65
66 static DWORD create_connection_settings(BOOL manual_proxy, const WCHAR *proxy_server,
67 BOOL use_wpad, BOOL use_pac_script, const WCHAR *pac_url, connection_settings **ret)
68 {
69 DWORD size = FIELD_OFFSET(connection_settings, data), pos;
70 DWORD proxy_server_len;
71 DWORD pac_url_len;
72
73 size += sizeof(DWORD);
74 if(proxy_server)
75 {
76 proxy_server_len = WideCharToMultiByte(CP_UTF8, 0, proxy_server, -1,
77 NULL, 0, NULL, NULL);
78 if(!proxy_server_len) return 0;
79 proxy_server_len--;
80 }
81 else
82 proxy_server_len = 0;
83 size += proxy_server_len;
84 if(pac_url)
85 {
86 pac_url_len = WideCharToMultiByte(CP_UTF8, 0, pac_url, -1,
87 NULL, 0, NULL, NULL);
88 if(!pac_url_len) return 0;
89 pac_url_len--;
90 }
91 else
92 pac_url_len = 0;
93 size += sizeof(DWORD)*10;
94
95 *ret = heap_alloc_zero(size);
96 if(!*ret) return 0;
97
98 (*ret)->version = CONNECTION_SETTINGS_VERSION;
99 (*ret)->flags = 1;
100 if(manual_proxy) (*ret)->flags |= CONNECTION_SETTINGS_MANUAL_PROXY;
101 if(use_pac_script) (*ret)->flags |= CONNECTION_SETTINGS_PAC_SCRIPT;
102 if(use_wpad) (*ret)->flags |= CONNECTION_SETTINGS_WPAD;
103 ((DWORD*)(*ret)->data)[0] = proxy_server_len;
104 pos = sizeof(DWORD);
105 if(proxy_server_len)
106 {
107 WideCharToMultiByte(CP_UTF8, 0, proxy_server, -1,
108 (char*)(*ret)->data+pos, proxy_server_len, NULL, NULL);
109 pos += proxy_server_len;
110 }
111 pos += sizeof(DWORD); /* skip proxy bypass list */
112 ((DWORD*)((*ret)->data+pos))[0] = pac_url_len;
113 pos += sizeof(DWORD);
114 if(pac_url_len)
115 {
116 WideCharToMultiByte(CP_UTF8, 0, pac_url, -1,
117 (char*)(*ret)->data+pos, pac_url_len, NULL, NULL);
118 pos += pac_url_len;
119 }
120 return size;
121 }
122
123 static void connections_on_initdialog(HWND hwnd)
124 {
125 DWORD type, size, enabled;
126 WCHAR address[INTERNET_MAX_URL_LENGTH], *port;
127 WCHAR pac_url[INTERNET_MAX_URL_LENGTH];
128 HKEY hkey, con;
129 LONG res;
130
131 SendMessageW(GetDlgItem(hwnd, IDC_EDIT_PAC_SCRIPT),
132 EM_LIMITTEXT, INTERNET_MAX_URL_LENGTH, 0);
133 SendMessageW(GetDlgItem(hwnd, IDC_EDIT_PROXY_SERVER),
134 EM_LIMITTEXT, INTERNET_MAX_URL_LENGTH-10, 0);
135 SendMessageW(GetDlgItem(hwnd, IDC_EDIT_PROXY_PORT), EM_LIMITTEXT, 8, 0);
136
137 res = RegOpenKeyW(HKEY_CURRENT_USER, internet_settings, &hkey);
138 if(res)
139 return;
140
141 size = sizeof(enabled);
142 res = RegQueryValueExW(hkey, proxy_enable, NULL, &type, (BYTE*)&enabled, &size);
143 if(res || type != REG_DWORD)
144 enabled = 0;
145 size = sizeof(address);
146 res = RegQueryValueExW(hkey, proxy_server, NULL, &type, (BYTE*)address, &size);
147 if(res || type != REG_SZ)
148 address[0] = 0;
149 size = sizeof(pac_url);
150 res = RegQueryValueExW(hkey, auto_config_url, NULL, &type, (BYTE*)pac_url, &size);
151 if(res || type != REG_SZ)
152 pac_url[0] = 0;
153
154 res = RegOpenKeyW(hkey, connections, &con);
155 RegCloseKey(hkey);
156 if(!res)
157 {
158 connection_settings *settings = NULL;
159 size = 0;
160
161 while((res = RegQueryValueExW(con, default_connection_settings, NULL, &type,
162 (BYTE*)settings, &size)) == ERROR_MORE_DATA || !settings)
163 {
164 connection_settings *new_settings = heap_realloc(settings, size);
165 if(!new_settings)
166 {
167 RegCloseKey(con);
168 heap_free(settings);
169 return;
170 }
171 settings = new_settings;
172 }
173 RegCloseKey(con);
174
175 if(!res && type == REG_BINARY)
176 {
177 if(settings->version != CONNECTION_SETTINGS_VERSION)
178 FIXME("unexpected structure version (%x)\n", settings->version);
179 else if(settings->flags & CONNECTION_SETTINGS_WPAD)
180 CheckDlgButton(hwnd, IDC_USE_WPAD, BST_CHECKED);
181 }
182 heap_free(settings);
183 }
184
185 TRACE("ProxyEnable = %x\n", enabled);
186 TRACE("ProxyServer = %s\n", wine_dbgstr_w(address));
187 TRACE("AutoConfigURL = %s\n", wine_dbgstr_w(auto_config_url));
188
189 if(enabled)
190 {
191 CheckDlgButton(hwnd, IDC_USE_PROXY_SERVER, BST_CHECKED);
192 EnableWindow(GetDlgItem(hwnd, IDC_EDIT_PROXY_SERVER), TRUE);
193 EnableWindow(GetDlgItem(hwnd, IDC_EDIT_PROXY_PORT), TRUE);
194 }
195
196 port = strchrW(address, ':');
197 if(port)
198 {
199 *port = 0;
200 port++;
201 }
202 SetDlgItemTextW(hwnd, IDC_EDIT_PROXY_SERVER, address);
203 if(port)
204 SetDlgItemTextW(hwnd, IDC_EDIT_PROXY_PORT, port);
205
206 if(pac_url[0])
207 {
208 CheckDlgButton(hwnd, IDC_USE_PAC_SCRIPT, BST_CHECKED);
209 EnableWindow(GetDlgItem(hwnd, IDC_EDIT_PAC_SCRIPT), TRUE);
210 SetDlgItemTextW(hwnd, IDC_EDIT_PAC_SCRIPT, pac_url);
211 }
212
213 return;
214 }
215
216 static INT_PTR connections_on_command(HWND hwnd, WPARAM wparam)
217 {
218 BOOL checked;
219
220 switch (wparam)
221 {
222 case IDC_USE_PAC_SCRIPT:
223 checked = IsDlgButtonChecked(hwnd, IDC_USE_PAC_SCRIPT);
224 EnableWindow(GetDlgItem(hwnd, IDC_EDIT_PAC_SCRIPT), checked);
225 break;
226 case IDC_USE_PROXY_SERVER:
227 checked = IsDlgButtonChecked(hwnd, IDC_USE_PROXY_SERVER);
228 EnableWindow(GetDlgItem(hwnd, IDC_EDIT_PROXY_SERVER), checked);
229 EnableWindow(GetDlgItem(hwnd, IDC_EDIT_PROXY_PORT), checked);
230 }
231
232 switch (wparam)
233 {
234 case IDC_USE_WPAD:
235 case IDC_USE_PAC_SCRIPT:
236 case IDC_USE_PROXY_SERVER:
237 case MAKEWPARAM(IDC_EDIT_PAC_SCRIPT, EN_CHANGE):
238 case MAKEWPARAM(IDC_EDIT_PROXY_SERVER, EN_CHANGE):
239 case MAKEWPARAM(IDC_EDIT_PROXY_PORT, EN_CHANGE):
240 if(initdialog_done)
241 SendMessageW(GetParent(hwnd), PSM_CHANGED, (WPARAM)hwnd, 0);
242 return TRUE;
243 }
244
245 return FALSE;
246 }
247
248 static INT_PTR connections_on_notify(HWND hwnd, WPARAM wparam, LPARAM lparam)
249 {
250 connection_settings *default_connection;
251 WCHAR proxy[INTERNET_MAX_URL_LENGTH];
252 WCHAR pac_script[INTERNET_MAX_URL_LENGTH];
253 PSHNOTIFY *psn = (PSHNOTIFY*)lparam;
254 DWORD proxy_len, port_len, pac_script_len;
255 DWORD use_proxy, use_pac_script, use_wpad, size;
256 LRESULT res;
257 HKEY hkey, con;
258
259 if(psn->hdr.code != PSN_APPLY)
260 return FALSE;
261
262 res = RegOpenKeyW(HKEY_CURRENT_USER, internet_settings, &hkey);
263 if(res)
264 return FALSE;
265
266 use_proxy = IsDlgButtonChecked(hwnd, IDC_USE_PROXY_SERVER);
267 res = RegSetValueExW(hkey, proxy_enable, 0, REG_DWORD,
268 (BYTE*)&use_proxy, sizeof(use_proxy));
269 if(res)
270 {
271 RegCloseKey(hkey);
272 return FALSE;
273 }
274 TRACE("ProxyEnable set to %x\n", use_proxy);
275
276 proxy_len = GetDlgItemTextW(hwnd, IDC_EDIT_PROXY_SERVER, proxy, ARRAY_SIZE(proxy));
277 if(proxy_len)
278 {
279 proxy[proxy_len++] = ':';
280 port_len = GetDlgItemTextW(hwnd, IDC_EDIT_PROXY_PORT, proxy+proxy_len,
281 ARRAY_SIZE(proxy)-proxy_len);
282 if(!port_len)
283 {
284 proxy[proxy_len++] = '8';
285 proxy[proxy_len++] = '0';
286 proxy[proxy_len] = 0;
287 }
288
289 res = RegSetValueExW(hkey, proxy_server, 0, REG_SZ,
290 (BYTE*)proxy, (proxy_len+port_len)*sizeof(WCHAR));
291 }
292 else
293 {
294 res = RegDeleteValueW(hkey, proxy_server);
295 if(res == ERROR_FILE_NOT_FOUND)
296 res = ERROR_SUCCESS;
297 }
298 if(res)
299 {
300 RegCloseKey(hkey);
301 return FALSE;
302 }
303 TRACE("ProxtServer set to %s\n", wine_dbgstr_w(proxy));
304
305 use_pac_script = IsDlgButtonChecked(hwnd, IDC_USE_PAC_SCRIPT);
306 pac_script_len = GetDlgItemTextW(hwnd, IDC_EDIT_PAC_SCRIPT,
307 pac_script, ARRAY_SIZE(pac_script));
308 if(!pac_script_len) use_pac_script = FALSE;
309 if(use_pac_script)
310 {
311 res = RegSetValueExW(hkey, auto_config_url, 0, REG_SZ,
312 (BYTE*)pac_script, pac_script_len*sizeof(WCHAR));
313 }
314 else
315 {
316 res = RegDeleteValueW(hkey, auto_config_url);
317 if(res == ERROR_FILE_NOT_FOUND)
318 res = ERROR_SUCCESS;
319 }
320 if(res)
321 {
322 RegCloseKey(hkey);
323 return FALSE;
324 }
325 TRACE("AutoConfigURL set to %s\n", wine_dbgstr_w(use_pac_script ? pac_script : NULL));
326
327 use_wpad = IsDlgButtonChecked(hwnd, IDC_USE_WPAD);
328
329 res = RegCreateKeyExW(hkey, connections, 0, NULL, 0, KEY_WRITE, NULL, &con, NULL);
330 RegCloseKey(hkey);
331 if(res)
332 return FALSE;
333
334 size = create_connection_settings(use_proxy, proxy, use_wpad,
335 use_pac_script, pac_script, &default_connection);
336 if(!size)
337 {
338 RegCloseKey(con);
339 return FALSE;
340 }
341
342 res = RegSetValueExW(con, default_connection_settings, 0, REG_BINARY,
343 (BYTE*)default_connection, size);
344 heap_free(default_connection);
345 RegCloseKey(con);
346 return !res;
347 }
348
349 INT_PTR CALLBACK connections_dlgproc(HWND hwnd, UINT msg, WPARAM wparam, LPARAM lparam)
350 {
351 switch (msg)
352 {
353 case WM_INITDIALOG:
354 connections_on_initdialog(hwnd);
355 initdialog_done = TRUE;
356 break;
357 case WM_COMMAND:
358 return connections_on_command(hwnd, wparam);
359 case WM_NOTIFY:
360 return connections_on_notify(hwnd, wparam, lparam);
361 }
362 return FALSE;
363 }