reactos/dll/win32/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 }