wininet: Rename WININET_strdup_WtoA to heap_strdupWtoA.
[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 = HeapAlloc( GetProcessHeap(), 0, 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     HeapFree( GetProcessHeap(), 0, name );
188     if (ret != 0)
189     {
190         TRACE("failed to get IPv4 address of %s (%s), retrying with IPv6\n", debugstr_w(lpszServerName), gai_strerror(ret));
191         hints.ai_family = AF_INET6;
192         ret = getaddrinfo( name, NULL, &hints, &res );
193         if (ret != 0)
194         {
195             TRACE("failed to get address of %s (%s)\n", debugstr_w(lpszServerName), gai_strerror(ret));
196             return FALSE;
197         }
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     HeapFree( GetProcessHeap(), 0, 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 VOID INTERNET_SendCallback(object_header_t *hdr, DWORD_PTR dwContext,
290                            DWORD dwInternetStatus, LPVOID lpvStatusInfo,
291                            DWORD dwStatusInfoLength)
292 {
293     LPVOID lpvNewInfo = NULL;
294
295     if( !hdr->lpfnStatusCB )
296         return;
297
298     /* the IE5 version of wininet does not
299        send callbacks if dwContext is zero */
300     if( !dwContext )
301         return;
302
303     lpvNewInfo = lpvStatusInfo;
304     if(hdr->dwInternalFlags & INET_CALLBACKW) {
305         switch(dwInternetStatus) {
306         case INTERNET_STATUS_NAME_RESOLVED:
307         case INTERNET_STATUS_CONNECTING_TO_SERVER:
308         case INTERNET_STATUS_CONNECTED_TO_SERVER:
309             lpvNewInfo = heap_strdupAtoW(lpvStatusInfo);
310             break;
311         case INTERNET_STATUS_RESOLVING_NAME:
312         case INTERNET_STATUS_REDIRECT:
313             lpvNewInfo = heap_strdupW(lpvStatusInfo);
314             break;
315         }
316     }else {
317         switch(dwInternetStatus)
318         {
319         case INTERNET_STATUS_NAME_RESOLVED:
320         case INTERNET_STATUS_CONNECTING_TO_SERVER:
321         case INTERNET_STATUS_CONNECTED_TO_SERVER:
322             lpvNewInfo = HeapAlloc(GetProcessHeap(), 0, strlen(lpvStatusInfo) + 1);
323             if (lpvNewInfo) strcpy(lpvNewInfo, lpvStatusInfo);
324             break;
325         case INTERNET_STATUS_RESOLVING_NAME:
326         case INTERNET_STATUS_REDIRECT:
327             lpvNewInfo = heap_strdupWtoA(lpvStatusInfo);
328             break;
329         }
330     }
331     
332     TRACE(" callback(%p) (%p (%p), %08lx, %d (%s), %p, %d)\n",
333           hdr->lpfnStatusCB, hdr->hInternet, hdr, dwContext, dwInternetStatus, get_callback_name(dwInternetStatus),
334           lpvNewInfo, dwStatusInfoLength);
335     
336     hdr->lpfnStatusCB(hdr->hInternet, dwContext, dwInternetStatus,
337                       lpvNewInfo, dwStatusInfoLength);
338
339     TRACE(" end callback().\n");
340
341     if(lpvNewInfo != lpvStatusInfo)
342         HeapFree(GetProcessHeap(), 0, lpvNewInfo);
343 }
344
345 static void SendAsyncCallbackProc(WORKREQUEST *workRequest)
346 {
347     struct WORKREQ_SENDCALLBACK const *req = &workRequest->u.SendCallback;
348
349     TRACE("%p\n", workRequest->hdr);
350
351     INTERNET_SendCallback(workRequest->hdr,
352                           req->dwContext, req->dwInternetStatus, req->lpvStatusInfo,
353                           req->dwStatusInfoLength);
354
355     /* And frees the copy of the status info */
356     HeapFree(GetProcessHeap(), 0, req->lpvStatusInfo);
357 }
358
359 void SendAsyncCallback(object_header_t *hdr, DWORD_PTR dwContext,
360                        DWORD dwInternetStatus, LPVOID lpvStatusInfo,
361                        DWORD dwStatusInfoLength)
362 {
363     TRACE("(%p, %08lx, %d (%s), %p, %d): %sasync call with callback %p\n",
364           hdr, dwContext, dwInternetStatus, get_callback_name(dwInternetStatus),
365           lpvStatusInfo, dwStatusInfoLength,
366           hdr->dwFlags & INTERNET_FLAG_ASYNC ? "" : "non ",
367           hdr->lpfnStatusCB);
368     
369     if (!(hdr->lpfnStatusCB))
370         return;
371     
372     if (hdr->dwFlags & INTERNET_FLAG_ASYNC)
373     {
374         WORKREQUEST workRequest;
375         struct WORKREQ_SENDCALLBACK *req;
376         void *lpvStatusInfo_copy = lpvStatusInfo;
377
378         if (lpvStatusInfo)
379         {
380             lpvStatusInfo_copy = HeapAlloc(GetProcessHeap(), 0, dwStatusInfoLength);
381             memcpy(lpvStatusInfo_copy, lpvStatusInfo, dwStatusInfoLength);
382         }
383
384         workRequest.asyncproc = SendAsyncCallbackProc;
385         workRequest.hdr = WININET_AddRef( hdr );
386         req = &workRequest.u.SendCallback;
387         req->dwContext = dwContext;
388         req->dwInternetStatus = dwInternetStatus;
389         req->lpvStatusInfo = lpvStatusInfo_copy;
390         req->dwStatusInfoLength = dwStatusInfoLength;
391         
392         INTERNET_AsyncCall(&workRequest);
393     }
394     else
395         INTERNET_SendCallback(hdr, dwContext, dwInternetStatus,
396                               lpvStatusInfo, dwStatusInfoLength);
397 }