advapi32/tests: Fix another test failure.
[wine] / dlls / wininet / utility.c
1 /*
2  * Wininet - Utility functions
3  *
4  * Copyright 1999 Corel Corporation
5  * Copyright 2002 CodeWeavers Inc.
6  *
7  * Ulrich Czekalla
8  * Aric Stewart
9  *
10  * This library is free software; you can redistribute it and/or
11  * modify it under the terms of the GNU Lesser General Public
12  * License as published by the Free Software Foundation; either
13  * version 2.1 of the License, or (at your option) any later version.
14  *
15  * This library is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
18  * Lesser General Public License for more details.
19  *
20  * You should have received a copy of the GNU Lesser General Public
21  * License along with this library; if not, write to the Free Software
22  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
23  */
24
25 #include "config.h"
26 #include "wine/port.h"
27
28 #if defined(__MINGW32__) || defined (_MSC_VER)
29 #include <ws2tcpip.h>
30 #endif
31
32 #include <stdarg.h>
33 #include <stdlib.h>
34 #include <string.h>
35 #include <time.h>
36
37 #include "windef.h"
38 #include "winbase.h"
39 #include "wininet.h"
40 #include "winnls.h"
41
42 #include "wine/debug.h"
43 #include "internet.h"
44
45 WINE_DEFAULT_DEBUG_CHANNEL(wininet);
46
47 #ifndef HAVE_GETADDRINFO
48
49 /* critical section to protect non-reentrant gethostbyname() */
50 static CRITICAL_SECTION cs_gethostbyname;
51 static CRITICAL_SECTION_DEBUG critsect_debug =
52 {
53     0, 0, &cs_gethostbyname,
54     { &critsect_debug.ProcessLocksList, &critsect_debug.ProcessLocksList },
55       0, 0, { (DWORD_PTR)(__FILE__ ": cs_gethostbyname") }
56 };
57 static CRITICAL_SECTION cs_gethostbyname = { &critsect_debug, -1, 0, 0, 0, 0 };
58
59 #endif
60
61 #define TIME_STRING_LEN  30
62
63 time_t ConvertTimeString(LPCWSTR asctime)
64 {
65     WCHAR tmpChar[TIME_STRING_LEN];
66     WCHAR *tmpChar2;
67     struct tm t;
68     int timelen = strlenW(asctime);
69
70     if(!timelen)
71         return 0;
72
73     /* FIXME: the atoiWs below rely on that tmpChar is \0 padded */
74     memset( tmpChar, 0, sizeof(tmpChar) );
75     lstrcpynW(tmpChar, asctime, TIME_STRING_LEN);
76
77     /* Assert that the string is the expected length */
78     if (strlenW(asctime) >= TIME_STRING_LEN) FIXME("\n");
79
80     /* Convert a time such as 'Mon, 15 Nov 1999 16:09:35 GMT' into a SYSTEMTIME structure
81      * We assume the time is in this format
82      * and divide it into easy to swallow chunks
83      */
84     tmpChar[3]='\0';
85     tmpChar[7]='\0';
86     tmpChar[11]='\0';
87     tmpChar[16]='\0';
88     tmpChar[19]='\0';
89     tmpChar[22]='\0';
90     tmpChar[25]='\0';
91
92     memset( &t, 0, sizeof(t) );
93     t.tm_year = atoiW(tmpChar+12) - 1900;
94     t.tm_mday = atoiW(tmpChar+5);
95     t.tm_hour = atoiW(tmpChar+17);
96     t.tm_min = atoiW(tmpChar+20);
97     t.tm_sec = atoiW(tmpChar+23);
98
99     /* and month */
100     tmpChar2 = tmpChar + 8;
101     switch(tmpChar2[2])
102     {
103         case 'n':
104             if(tmpChar2[1]=='a')
105                 t.tm_mon = 0;
106             else
107                 t.tm_mon = 5;
108             break;
109         case 'b':
110             t.tm_mon = 1;
111             break;
112         case 'r':
113             if(tmpChar2[1]=='a')
114                 t.tm_mon = 2;
115             else
116                 t.tm_mon = 3;
117             break;
118         case 'y':
119             t.tm_mon = 4;
120             break;
121         case 'l':
122             t.tm_mon = 6;
123             break;
124         case 'g':
125             t.tm_mon = 7;
126             break;
127         case 'p':
128             t.tm_mon = 8;
129             break;
130         case 't':
131             t.tm_mon = 9;
132             break;
133         case 'v':
134             t.tm_mon = 10;
135             break;
136         case 'c':
137             t.tm_mon = 11;
138             break;
139         default:
140             FIXME("\n");
141     }
142
143     return mktime(&t);
144 }
145
146
147 BOOL GetAddress(LPCWSTR lpszServerName, INTERNET_PORT nServerPort,
148         struct sockaddr *psa, socklen_t *sa_len)
149 {
150     WCHAR *found;
151     char *name;
152     int len, sz;
153 #ifdef HAVE_GETADDRINFO
154     struct addrinfo *res, hints;
155     int ret;
156 #else
157     struct hostent *phe;
158     struct sockaddr_in *sin = (struct sockaddr_in *)psa;
159 #endif
160
161     TRACE("%s\n", debugstr_w(lpszServerName));
162
163     /* Validate server name first
164      * Check if there is sth. like
165      * pinger.macromedia.com:80
166      * if yes, eliminate the :80....
167      */
168     found = strchrW(lpszServerName, ':');
169     if (found)
170         len = found - lpszServerName;
171     else
172         len = strlenW(lpszServerName);
173
174     sz = WideCharToMultiByte( CP_UNIXCP, 0, lpszServerName, len, NULL, 0, NULL, NULL );
175     if (!(name = heap_alloc(sz + 1))) return FALSE;
176     WideCharToMultiByte( CP_UNIXCP, 0, lpszServerName, len, name, sz, NULL, NULL );
177     name[sz] = 0;
178
179 #ifdef HAVE_GETADDRINFO
180     memset( &hints, 0, sizeof(struct addrinfo) );
181     /* Prefer IPv4 to IPv6 addresses, since some servers do not listen on
182      * their IPv6 addresses even though they have IPv6 addresses in the DNS.
183      */
184     hints.ai_family = AF_INET;
185
186     ret = getaddrinfo( name, NULL, &hints, &res );
187     if (ret != 0)
188     {
189         TRACE("failed to get IPv4 address of %s (%s), retrying with IPv6\n", debugstr_w(lpszServerName), gai_strerror(ret));
190         hints.ai_family = AF_INET6;
191         ret = getaddrinfo( name, NULL, &hints, &res );
192     }
193     heap_free( name );
194     if (ret != 0)
195     {
196         TRACE("failed to get address of %s (%s)\n", debugstr_w(lpszServerName), gai_strerror(ret));
197         return FALSE;
198     }
199     if (*sa_len < res->ai_addrlen)
200     {
201         WARN("address too small\n");
202         freeaddrinfo( res );
203         return FALSE;
204     }
205     *sa_len = res->ai_addrlen;
206     memcpy( psa, res->ai_addr, res->ai_addrlen );
207     /* Copy port */
208     switch (res->ai_family)
209     {
210     case AF_INET:
211         ((struct sockaddr_in *)psa)->sin_port = htons(nServerPort);
212         break;
213     case AF_INET6:
214         ((struct sockaddr_in6 *)psa)->sin6_port = htons(nServerPort);
215         break;
216     }
217
218     freeaddrinfo( res );
219 #else
220     EnterCriticalSection( &cs_gethostbyname );
221     phe = gethostbyname(name);
222     heap_free( name );
223
224     if (NULL == phe)
225     {
226         TRACE("failed to get address of %s (%d)\n", debugstr_w(lpszServerName), h_errno);
227         LeaveCriticalSection( &cs_gethostbyname );
228         return FALSE;
229     }
230     if (*sa_len < sizeof(struct sockaddr_in))
231     {
232         WARN("address too small\n");
233         LeaveCriticalSection( &cs_gethostbyname );
234         return FALSE;
235     }
236     *sa_len = sizeof(struct sockaddr_in);
237     memset(sin,0,sizeof(struct sockaddr_in));
238     memcpy((char *)&sin->sin_addr, phe->h_addr, phe->h_length);
239     sin->sin_family = phe->h_addrtype;
240     sin->sin_port = htons(nServerPort);
241
242     LeaveCriticalSection( &cs_gethostbyname );
243 #endif
244     return TRUE;
245 }
246
247 /*
248  * Helper function for sending async Callbacks
249  */
250
251 static const char *get_callback_name(DWORD dwInternetStatus) {
252     static const wininet_flag_info internet_status[] = {
253 #define FE(x) { x, #x }
254         FE(INTERNET_STATUS_RESOLVING_NAME),
255         FE(INTERNET_STATUS_NAME_RESOLVED),
256         FE(INTERNET_STATUS_CONNECTING_TO_SERVER),
257         FE(INTERNET_STATUS_CONNECTED_TO_SERVER),
258         FE(INTERNET_STATUS_SENDING_REQUEST),
259         FE(INTERNET_STATUS_REQUEST_SENT),
260         FE(INTERNET_STATUS_RECEIVING_RESPONSE),
261         FE(INTERNET_STATUS_RESPONSE_RECEIVED),
262         FE(INTERNET_STATUS_CTL_RESPONSE_RECEIVED),
263         FE(INTERNET_STATUS_PREFETCH),
264         FE(INTERNET_STATUS_CLOSING_CONNECTION),
265         FE(INTERNET_STATUS_CONNECTION_CLOSED),
266         FE(INTERNET_STATUS_HANDLE_CREATED),
267         FE(INTERNET_STATUS_HANDLE_CLOSING),
268         FE(INTERNET_STATUS_REQUEST_COMPLETE),
269         FE(INTERNET_STATUS_REDIRECT),
270         FE(INTERNET_STATUS_INTERMEDIATE_RESPONSE),
271         FE(INTERNET_STATUS_USER_INPUT_REQUIRED),
272         FE(INTERNET_STATUS_STATE_CHANGE),
273         FE(INTERNET_STATUS_COOKIE_SENT),
274         FE(INTERNET_STATUS_COOKIE_RECEIVED),
275         FE(INTERNET_STATUS_PRIVACY_IMPACTED),
276         FE(INTERNET_STATUS_P3P_HEADER),
277         FE(INTERNET_STATUS_P3P_POLICYREF),
278         FE(INTERNET_STATUS_COOKIE_HISTORY)
279 #undef FE
280     };
281     DWORD i;
282
283     for (i = 0; i < (sizeof(internet_status) / sizeof(internet_status[0])); i++) {
284         if (internet_status[i].val == dwInternetStatus) return internet_status[i].name;
285     }
286     return "Unknown";
287 }
288
289 static const char *debugstr_status_info(DWORD status, void *info)
290 {
291     switch(status) {
292     case INTERNET_STATUS_REQUEST_COMPLETE: {
293         INTERNET_ASYNC_RESULT *iar = info;
294         return wine_dbg_sprintf("{%s, %d}", wine_dbgstr_longlong(iar->dwResult), iar->dwError);
295     }
296     default:
297         return wine_dbg_sprintf("%p", info);
298     }
299 }
300
301 VOID INTERNET_SendCallback(object_header_t *hdr, DWORD_PTR dwContext,
302                            DWORD dwInternetStatus, LPVOID lpvStatusInfo,
303                            DWORD dwStatusInfoLength)
304 {
305     LPVOID lpvNewInfo = NULL;
306
307     if( !hdr->lpfnStatusCB )
308         return;
309
310     /* the IE5 version of wininet does not
311        send callbacks if dwContext is zero */
312     if( !dwContext )
313         return;
314
315     lpvNewInfo = lpvStatusInfo;
316     if(hdr->dwInternalFlags & INET_CALLBACKW) {
317         switch(dwInternetStatus) {
318         case INTERNET_STATUS_NAME_RESOLVED:
319         case INTERNET_STATUS_CONNECTING_TO_SERVER:
320         case INTERNET_STATUS_CONNECTED_TO_SERVER:
321             lpvNewInfo = heap_strdupAtoW(lpvStatusInfo);
322             dwStatusInfoLength *= sizeof(WCHAR);
323             break;
324         case INTERNET_STATUS_RESOLVING_NAME:
325         case INTERNET_STATUS_REDIRECT:
326             lpvNewInfo = heap_strdupW(lpvStatusInfo);
327             break;
328         }
329     }else {
330         switch(dwInternetStatus)
331         {
332         case INTERNET_STATUS_NAME_RESOLVED:
333         case INTERNET_STATUS_CONNECTING_TO_SERVER:
334         case INTERNET_STATUS_CONNECTED_TO_SERVER:
335             lpvNewInfo = heap_alloc(strlen(lpvStatusInfo) + 1);
336             if (lpvNewInfo) strcpy(lpvNewInfo, lpvStatusInfo);
337             break;
338         case INTERNET_STATUS_RESOLVING_NAME:
339         case INTERNET_STATUS_REDIRECT:
340             lpvNewInfo = heap_strdupWtoA(lpvStatusInfo);
341             dwStatusInfoLength /= sizeof(WCHAR);
342             break;
343         }
344     }
345     
346     TRACE(" callback(%p) (%p (%p), %08lx, %d (%s), %s, %d)\n",
347           hdr->lpfnStatusCB, hdr->hInternet, hdr, dwContext, dwInternetStatus, get_callback_name(dwInternetStatus),
348           debugstr_status_info(dwInternetStatus, lpvNewInfo), dwStatusInfoLength);
349     
350     hdr->lpfnStatusCB(hdr->hInternet, dwContext, dwInternetStatus,
351                       lpvNewInfo, dwStatusInfoLength);
352
353     TRACE(" end callback().\n");
354
355     if(lpvNewInfo != lpvStatusInfo)
356         heap_free(lpvNewInfo);
357 }
358
359 static void SendAsyncCallbackProc(WORKREQUEST *workRequest)
360 {
361     struct WORKREQ_SENDCALLBACK const *req = &workRequest->u.SendCallback;
362
363     TRACE("%p\n", workRequest->hdr);
364
365     INTERNET_SendCallback(workRequest->hdr,
366                           req->dwContext, req->dwInternetStatus, req->lpvStatusInfo,
367                           req->dwStatusInfoLength);
368
369     /* And frees the copy of the status info */
370     heap_free(req->lpvStatusInfo);
371 }
372
373 void SendAsyncCallback(object_header_t *hdr, DWORD_PTR dwContext,
374                        DWORD dwInternetStatus, LPVOID lpvStatusInfo,
375                        DWORD dwStatusInfoLength)
376 {
377     TRACE("(%p, %08lx, %d (%s), %p, %d): %sasync call with callback %p\n",
378           hdr, dwContext, dwInternetStatus, get_callback_name(dwInternetStatus),
379           lpvStatusInfo, dwStatusInfoLength,
380           hdr->dwFlags & INTERNET_FLAG_ASYNC ? "" : "non ",
381           hdr->lpfnStatusCB);
382     
383     if (!(hdr->lpfnStatusCB))
384         return;
385     
386     if (hdr->dwFlags & INTERNET_FLAG_ASYNC)
387     {
388         WORKREQUEST workRequest;
389         struct WORKREQ_SENDCALLBACK *req;
390         void *lpvStatusInfo_copy = lpvStatusInfo;
391
392         if (lpvStatusInfo)
393         {
394             lpvStatusInfo_copy = heap_alloc(dwStatusInfoLength);
395             memcpy(lpvStatusInfo_copy, lpvStatusInfo, dwStatusInfoLength);
396         }
397
398         workRequest.asyncproc = SendAsyncCallbackProc;
399         workRequest.hdr = WININET_AddRef( hdr );
400         req = &workRequest.u.SendCallback;
401         req->dwContext = dwContext;
402         req->dwInternetStatus = dwInternetStatus;
403         req->lpvStatusInfo = lpvStatusInfo_copy;
404         req->dwStatusInfoLength = dwStatusInfoLength;
405         
406         INTERNET_AsyncCall(&workRequest);
407     }
408     else
409         INTERNET_SendCallback(hdr, dwContext, dwInternetStatus,
410                               lpvStatusInfo, dwStatusInfoLength);
411 }