ole32/tests: Fix some test failures on Win9x/WinME.
[wine] / dlls / wininet / cookie.c
1 /*
2  * Wininet - cookie handling stuff
3  *
4  * Copyright 2002 TransGaming Technologies Inc.
5  *
6  * David Hammerton
7  *
8  * This library is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * This library is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with this library; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
21  */
22
23 #include "config.h"
24 #include "wine/port.h"
25
26 #if defined(__MINGW32__) || defined (_MSC_VER)
27 #include <ws2tcpip.h>
28 #endif
29
30 #include <stdarg.h>
31 #include <stdio.h>
32 #include <stdlib.h>
33 #include <string.h>
34 #ifdef HAVE_UNISTD_H
35 # include <unistd.h>
36 #endif
37
38 #include "windef.h"
39 #include "winbase.h"
40 #include "wininet.h"
41 #include "winerror.h"
42
43 #include "wine/debug.h"
44 #include "internet.h"
45
46 #define RESPONSE_TIMEOUT        30            /* FROM internet.c */
47
48
49 WINE_DEFAULT_DEBUG_CHANNEL(wininet);
50
51 /* FIXME
52  *     Cookies are currently memory only.
53  *     Cookies are NOT THREAD SAFE
54  *     Cookies could use A LOT OF MEMORY. We need some kind of memory management here!
55  */
56
57 typedef struct _cookie_domain cookie_domain;
58 typedef struct _cookie cookie;
59
60 struct _cookie
61 {
62     struct list entry;
63
64     struct _cookie_domain *parent;
65
66     LPWSTR lpCookieName;
67     LPWSTR lpCookieData;
68     FILETIME expiry;
69 };
70
71 struct _cookie_domain
72 {
73     struct list entry;
74
75     LPWSTR lpCookieDomain;
76     LPWSTR lpCookiePath;
77     struct list cookie_list;
78 };
79
80 static struct list domain_list = LIST_INIT(domain_list);
81
82 static cookie *COOKIE_addCookie(cookie_domain *domain, LPCWSTR name, LPCWSTR data, FILETIME expiry);
83 static cookie *COOKIE_findCookie(cookie_domain *domain, LPCWSTR lpszCookieName);
84 static void COOKIE_deleteCookie(cookie *deadCookie, BOOL deleteDomain);
85 static cookie_domain *COOKIE_addDomain(LPCWSTR domain, LPCWSTR path);
86 static void COOKIE_deleteDomain(cookie_domain *deadDomain);
87
88
89 /* adds a cookie to the domain */
90 static cookie *COOKIE_addCookie(cookie_domain *domain, LPCWSTR name, LPCWSTR data, FILETIME expiry)
91 {
92     cookie *newCookie = HeapAlloc(GetProcessHeap(), 0, sizeof(cookie));
93
94     list_init(&newCookie->entry);
95     newCookie->lpCookieName = NULL;
96     newCookie->lpCookieData = NULL;
97     newCookie->expiry = expiry;
98
99     if (name)
100     {
101         newCookie->lpCookieName = HeapAlloc(GetProcessHeap(), 0, (strlenW(name) + 1)*sizeof(WCHAR));
102         lstrcpyW(newCookie->lpCookieName, name);
103     }
104     if (data)
105     {
106         newCookie->lpCookieData = HeapAlloc(GetProcessHeap(), 0, (strlenW(data) + 1)*sizeof(WCHAR));
107         lstrcpyW(newCookie->lpCookieData, data);
108     }
109
110     TRACE("added cookie %p (data is %s)\n", newCookie, debugstr_w(data) );
111
112     list_add_tail(&domain->cookie_list, &newCookie->entry);
113     newCookie->parent = domain;
114     return newCookie;
115 }
116
117
118 /* finds a cookie in the domain matching the cookie name */
119 static cookie *COOKIE_findCookie(cookie_domain *domain, LPCWSTR lpszCookieName)
120 {
121     struct list * cursor;
122     TRACE("(%p, %s)\n", domain, debugstr_w(lpszCookieName));
123
124     LIST_FOR_EACH(cursor, &domain->cookie_list)
125     {
126         cookie *searchCookie = LIST_ENTRY(cursor, cookie, entry);
127         BOOL candidate = TRUE;
128         if (candidate && lpszCookieName)
129         {
130             if (candidate && !searchCookie->lpCookieName)
131                 candidate = FALSE;
132             if (candidate && strcmpW(lpszCookieName, searchCookie->lpCookieName) != 0)
133                 candidate = FALSE;
134         }
135         if (candidate)
136             return searchCookie;
137     }
138     return NULL;
139 }
140
141 /* removes a cookie from the list, if its the last cookie we also remove the domain */
142 static void COOKIE_deleteCookie(cookie *deadCookie, BOOL deleteDomain)
143 {
144     HeapFree(GetProcessHeap(), 0, deadCookie->lpCookieName);
145     HeapFree(GetProcessHeap(), 0, deadCookie->lpCookieData);
146     list_remove(&deadCookie->entry);
147
148     /* special case: last cookie, lets remove the domain to save memory */
149     if (list_empty(&deadCookie->parent->cookie_list) && deleteDomain)
150         COOKIE_deleteDomain(deadCookie->parent);
151     HeapFree(GetProcessHeap(), 0, deadCookie);
152 }
153
154 /* allocates a domain and adds it to the end */
155 static cookie_domain *COOKIE_addDomain(LPCWSTR domain, LPCWSTR path)
156 {
157     cookie_domain *newDomain = HeapAlloc(GetProcessHeap(), 0, sizeof(cookie_domain));
158
159     list_init(&newDomain->entry);
160     list_init(&newDomain->cookie_list);
161     newDomain->lpCookieDomain = NULL;
162     newDomain->lpCookiePath = NULL;
163
164     if (domain)
165     {
166         newDomain->lpCookieDomain = HeapAlloc(GetProcessHeap(), 0, (strlenW(domain) + 1)*sizeof(WCHAR));
167         strcpyW(newDomain->lpCookieDomain, domain);
168     }
169     if (path)
170     {
171         newDomain->lpCookiePath = HeapAlloc(GetProcessHeap(), 0, (strlenW(path) + 1)*sizeof(WCHAR));
172         lstrcpyW(newDomain->lpCookiePath, path);
173     }
174
175     list_add_tail(&domain_list, &newDomain->entry);
176
177     TRACE("Adding domain: %p\n", newDomain);
178     return newDomain;
179 }
180
181 static BOOL COOKIE_crackUrlSimple(LPCWSTR lpszUrl, LPWSTR hostName, int hostNameLen, LPWSTR path, int pathLen)
182 {
183     URL_COMPONENTSW UrlComponents;
184     BOOL rc;
185
186     UrlComponents.lpszExtraInfo = NULL;
187     UrlComponents.lpszPassword = NULL;
188     UrlComponents.lpszScheme = NULL;
189     UrlComponents.lpszUrlPath = path;
190     UrlComponents.lpszUserName = NULL;
191     UrlComponents.lpszHostName = hostName;
192     UrlComponents.dwExtraInfoLength = 0;
193     UrlComponents.dwPasswordLength = 0;
194     UrlComponents.dwSchemeLength = 0;
195     UrlComponents.dwUserNameLength = 0;
196     UrlComponents.dwHostNameLength = hostNameLen;
197     UrlComponents.dwUrlPathLength = pathLen;
198
199     rc = InternetCrackUrlW(lpszUrl, 0, 0, &UrlComponents);
200
201     /* discard the webpage off the end of the path */
202     if (pathLen > 0 && path[pathLen-1] != '/')
203     {
204         LPWSTR ptr;
205         ptr = strrchrW(path,'/');
206         if (ptr)
207             *(++ptr) = 0;
208         else
209         {
210             path[0] = '/';
211             path[1] = 0;
212         }
213     }
214     return rc;
215 }
216
217 /* match a domain. domain must match if the domain is not NULL. path must match if the path is not NULL */
218 static BOOL COOKIE_matchDomain(LPCWSTR lpszCookieDomain, LPCWSTR lpszCookiePath,
219                                cookie_domain *searchDomain, BOOL allow_partial)
220 {
221     TRACE("searching on domain %p\n", searchDomain);
222         if (lpszCookieDomain)
223         {
224             if (!searchDomain->lpCookieDomain)
225             return FALSE;
226
227             TRACE("comparing domain %s with %s\n", 
228             debugstr_w(lpszCookieDomain), 
229             debugstr_w(searchDomain->lpCookieDomain));
230
231         if (allow_partial && !strstrW(lpszCookieDomain, searchDomain->lpCookieDomain))
232             return FALSE;
233         else if (!allow_partial && lstrcmpW(lpszCookieDomain, searchDomain->lpCookieDomain) != 0)
234             return FALSE;
235         }
236     if (lpszCookiePath)
237     {
238         INT len;
239         TRACE("comparing paths: %s with %s\n", debugstr_w(lpszCookiePath), debugstr_w(searchDomain->lpCookiePath));
240         /* paths match at the beginning.  so a path of  /foo would match
241          * /foobar and /foo/bar
242          */
243         if (!searchDomain->lpCookiePath)
244             return FALSE;
245         if (allow_partial)
246         {
247             len = lstrlenW(searchDomain->lpCookiePath);
248             if (strncmpiW(searchDomain->lpCookiePath, lpszCookiePath, len)!=0)
249                 return FALSE;
250         }
251         else if (strcmpW(lpszCookiePath, searchDomain->lpCookiePath))
252             return FALSE;
253
254         }
255         return TRUE;
256 }
257
258 /* remove a domain from the list and delete it */
259 static void COOKIE_deleteDomain(cookie_domain *deadDomain)
260 {
261     struct list * cursor;
262     while ((cursor = list_tail(&deadDomain->cookie_list)))
263     {
264         COOKIE_deleteCookie(LIST_ENTRY(cursor, cookie, entry), FALSE);
265         list_remove(cursor);
266     }
267
268     HeapFree(GetProcessHeap(), 0, deadDomain->lpCookieDomain);
269     HeapFree(GetProcessHeap(), 0, deadDomain->lpCookiePath);
270
271     list_remove(&deadDomain->entry);
272
273     HeapFree(GetProcessHeap(), 0, deadDomain);
274 }
275
276 /***********************************************************************
277  *           InternetGetCookieW (WININET.@)
278  *
279  * Retrieve cookie from the specified url
280  *
281  *  It should be noted that on windows the lpszCookieName parameter is "not implemented".
282  *    So it won't be implemented here.
283  *
284  * RETURNS
285  *    TRUE  on success
286  *    FALSE on failure
287  *
288  */
289 BOOL WINAPI InternetGetCookieW(LPCWSTR lpszUrl, LPCWSTR lpszCookieName,
290     LPWSTR lpCookieData, LPDWORD lpdwSize)
291 {
292     BOOL ret;
293     struct list * cursor;
294     unsigned int cnt = 0, domain_count = 0, cookie_count = 0;
295     WCHAR hostName[2048], path[2048];
296     FILETIME tm;
297
298     TRACE("(%s, %s, %p, %p)\n", debugstr_w(lpszUrl),debugstr_w(lpszCookieName),
299           lpCookieData, lpdwSize);
300
301     if (!lpszUrl)
302     {
303         SetLastError(ERROR_INVALID_PARAMETER);
304         return FALSE;
305     }
306
307     hostName[0] = 0;
308     ret = COOKIE_crackUrlSimple(lpszUrl, hostName, sizeof(hostName)/sizeof(hostName[0]), path, sizeof(path)/sizeof(path[0]));
309     if (!ret || !hostName[0]) return FALSE;
310
311     GetSystemTimeAsFileTime(&tm);
312
313     LIST_FOR_EACH(cursor, &domain_list)
314     {
315         cookie_domain *cookiesDomain = LIST_ENTRY(cursor, cookie_domain, entry);
316         if (COOKIE_matchDomain(hostName, path, cookiesDomain, TRUE))
317         {
318             struct list * cursor;
319             domain_count++;
320             TRACE("found domain %p\n", cookiesDomain);
321     
322             LIST_FOR_EACH(cursor, &cookiesDomain->cookie_list)
323             {
324                 cookie *thisCookie = LIST_ENTRY(cursor, cookie, entry);
325                 /* check for expiry */
326                 if ((thisCookie->expiry.dwLowDateTime != 0 || thisCookie->expiry.dwHighDateTime != 0) && CompareFileTime(&tm,&thisCookie->expiry)  > 0)
327                 {
328                     TRACE("Found expired cookie. deleting\n");
329                     COOKIE_deleteCookie(thisCookie, FALSE);
330                     continue;
331                 }
332
333                 if (lpCookieData == NULL) /* return the size of the buffer required to lpdwSize */
334                 {
335                     unsigned int len;
336
337                     if (cookie_count) cnt += 2; /* '; ' */
338                     cnt += strlenW(thisCookie->lpCookieName);
339                     if ((len = strlenW(thisCookie->lpCookieData)))
340                     {
341                         cnt += 1; /* = */
342                         cnt += len;
343                     }
344                 }
345                 else
346                 {
347                     static const WCHAR szsc[] = { ';',' ',0 };
348                     static const WCHAR szname[] = { '%','s',0 };
349                     static const WCHAR szdata[] = { '=','%','s',0 };
350
351                     if (cookie_count) cnt += snprintfW(lpCookieData + cnt, *lpdwSize - cnt, szsc);
352                     cnt += snprintfW(lpCookieData + cnt, *lpdwSize - cnt, szname, thisCookie->lpCookieName);
353
354                     if (thisCookie->lpCookieData[0])
355                         cnt += snprintfW(lpCookieData + cnt, *lpdwSize - cnt, szdata, thisCookie->lpCookieData);
356
357                     TRACE("Cookie: %s\n", debugstr_w(lpCookieData));
358                 }
359                 cookie_count++;
360             }
361         }
362     }
363
364     if (!domain_count)
365     {
366         TRACE("no cookies found for %s\n", debugstr_w(hostName));
367         SetLastError(ERROR_NO_MORE_ITEMS);
368         return FALSE;
369     }
370
371     if (lpCookieData == NULL)
372     {
373         *lpdwSize = (cnt + 1) * sizeof(WCHAR);
374         TRACE("returning %u\n", *lpdwSize);
375         return TRUE;
376     }
377
378     *lpdwSize = cnt + 1;
379
380     TRACE("Returning %u (from %u domains): %s\n", cnt, domain_count,
381            debugstr_w(lpCookieData));
382
383     return (cnt ? TRUE : FALSE);
384 }
385
386
387 /***********************************************************************
388  *           InternetGetCookieA (WININET.@)
389  *
390  * Retrieve cookie from the specified url
391  *
392  * RETURNS
393  *    TRUE  on success
394  *    FALSE on failure
395  *
396  */
397 BOOL WINAPI InternetGetCookieA(LPCSTR lpszUrl, LPCSTR lpszCookieName,
398     LPSTR lpCookieData, LPDWORD lpdwSize)
399 {
400     DWORD len;
401     LPWSTR szCookieData = NULL, szUrl = NULL, szCookieName = NULL;
402     BOOL r;
403
404     TRACE("(%s,%s,%p)\n", debugstr_a(lpszUrl), debugstr_a(lpszCookieName),
405         lpCookieData);
406
407     if( lpszUrl )
408     {
409         len = MultiByteToWideChar( CP_ACP, 0, lpszUrl, -1, NULL, 0 );
410         szUrl = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) );
411         MultiByteToWideChar( CP_ACP, 0, lpszUrl, -1, szUrl, len );
412     }
413
414     if( lpszCookieName )
415     {
416         len = MultiByteToWideChar( CP_ACP, 0, lpszCookieName, -1, NULL, 0 );
417         szCookieName = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) );
418         MultiByteToWideChar( CP_ACP, 0, lpszCookieName, -1, szCookieName, len );
419     }
420
421     r = InternetGetCookieW( szUrl, szCookieName, NULL, &len );
422     if( r )
423     {
424         szCookieData = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) );
425         if( !szCookieData )
426         {
427             r = FALSE;
428         }
429         else
430         {
431             r = InternetGetCookieW( szUrl, szCookieName, szCookieData, &len );
432
433             *lpdwSize = WideCharToMultiByte( CP_ACP, 0, szCookieData, len,
434                                     lpCookieData, *lpdwSize, NULL, NULL );
435         }
436     }
437
438     HeapFree( GetProcessHeap(), 0, szCookieData );
439     HeapFree( GetProcessHeap(), 0, szCookieName );
440     HeapFree( GetProcessHeap(), 0, szUrl );
441
442     return r;
443 }
444
445 static BOOL set_cookie(LPCWSTR domain, LPCWSTR path, LPCWSTR cookie_name, LPCWSTR cookie_data)
446 {
447     cookie_domain *thisCookieDomain = NULL;
448     cookie *thisCookie;
449     struct list *cursor;
450     LPWSTR data, value;
451     WCHAR *ptr;
452     FILETIME expiry;
453     BOOL expired = FALSE;
454
455     value = data = HeapAlloc(GetProcessHeap(), 0, (strlenW(cookie_data) + 1) * sizeof(WCHAR));
456     if (data == NULL)
457     {
458         ERR("could not allocate %zu bytes for the cookie data buffer\n", (strlenW(cookie_data) + 1) * sizeof(WCHAR));
459         return FALSE;
460     }
461
462     strcpyW(data,cookie_data);
463     memset(&expiry,0,sizeof(expiry));
464
465     /* lots of information can be parsed out of the cookie value */
466
467     ptr = data;
468     for (;;)
469     {
470         static const WCHAR szDomain[] = {'d','o','m','a','i','n','=',0};
471         static const WCHAR szPath[] = {'p','a','t','h','=',0};
472         static const WCHAR szExpires[] = {'e','x','p','i','r','e','s','=',0};
473         static const WCHAR szSecure[] = {'s','e','c','u','r','e',0};
474         static const WCHAR szHttpOnly[] = {'h','t','t','p','o','n','l','y',0};
475
476         if (!(ptr = strchrW(ptr,';'))) break;
477         *ptr++ = 0;
478
479         if (value != data)
480             HeapFree(GetProcessHeap(), 0, value);
481         value = HeapAlloc(GetProcessHeap(), 0, (ptr - data) * sizeof(WCHAR));
482         if (value == NULL)
483         {
484             HeapFree(GetProcessHeap(), 0, data);
485             ERR("could not allocate %zu bytes for the cookie value buffer\n", (ptr - data) * sizeof(WCHAR));
486             return FALSE;
487         }
488         strcpyW(value, data);
489
490         while (*ptr == ' ') ptr++; /* whitespace */
491
492         if (strncmpiW(ptr, szDomain, 7) == 0)
493         {
494             ptr+=strlenW(szDomain);
495             domain = ptr;
496             TRACE("Parsing new domain %s\n",debugstr_w(domain));
497         }
498         else if (strncmpiW(ptr, szPath, 5) == 0)
499         {
500             ptr+=strlenW(szPath);
501             path = ptr;
502             TRACE("Parsing new path %s\n",debugstr_w(path));
503         }
504         else if (strncmpiW(ptr, szExpires, 8) == 0)
505         {
506             FILETIME ft;
507             SYSTEMTIME st;
508             FIXME("persistent cookies not handled (%s)\n",debugstr_w(ptr));
509             ptr+=strlenW(szExpires);
510             if (InternetTimeToSystemTimeW(ptr, &st, 0))
511             {
512                 SystemTimeToFileTime(&st, &expiry);
513                 GetSystemTimeAsFileTime(&ft);
514
515                 if (CompareFileTime(&ft,&expiry) > 0)
516                 {
517                     TRACE("Cookie already expired.\n");
518                     expired = TRUE;
519                 }
520             }
521         }
522         else if (strncmpiW(ptr, szSecure, 6) == 0)
523         {
524             FIXME("secure not handled (%s)\n",debugstr_w(ptr));
525             ptr += strlenW(szSecure);
526         }
527         else if (strncmpiW(ptr, szHttpOnly, 8) == 0)
528         {
529             FIXME("httponly not handled (%s)\n",debugstr_w(ptr));
530             ptr += strlenW(szHttpOnly);
531         }
532         else if (*ptr)
533         {
534             FIXME("Unknown additional option %s\n",debugstr_w(ptr));
535             break;
536         }
537     }
538
539     LIST_FOR_EACH(cursor, &domain_list)
540     {
541         thisCookieDomain = LIST_ENTRY(cursor, cookie_domain, entry);
542         if (COOKIE_matchDomain(domain, path, thisCookieDomain, FALSE))
543             break;
544         thisCookieDomain = NULL;
545     }
546
547     if (!thisCookieDomain)
548     {
549         if (!expired)
550             thisCookieDomain = COOKIE_addDomain(domain, path);
551         else
552         {
553             HeapFree(GetProcessHeap(),0,data);
554             if (value != data) HeapFree(GetProcessHeap(), 0, value);
555             return TRUE;
556         }
557     }
558
559     if ((thisCookie = COOKIE_findCookie(thisCookieDomain, cookie_name)))
560         COOKIE_deleteCookie(thisCookie, FALSE);
561
562     TRACE("setting cookie %s=%s for domain %s path %s\n", debugstr_w(cookie_name),
563           debugstr_w(value), debugstr_w(thisCookieDomain->lpCookieDomain),debugstr_w(thisCookieDomain->lpCookiePath));
564
565     if (!expired && !COOKIE_addCookie(thisCookieDomain, cookie_name, value, expiry))
566     {
567         HeapFree(GetProcessHeap(),0,data);
568         if (value != data) HeapFree(GetProcessHeap(), 0, value);
569         return FALSE;
570     }
571
572     HeapFree(GetProcessHeap(),0,data);
573     if (value != data) HeapFree(GetProcessHeap(), 0, value);
574     return TRUE;
575 }
576
577 /***********************************************************************
578  *           InternetSetCookieW (WININET.@)
579  *
580  * Sets cookie for the specified url
581  *
582  * RETURNS
583  *    TRUE  on success
584  *    FALSE on failure
585  *
586  */
587 BOOL WINAPI InternetSetCookieW(LPCWSTR lpszUrl, LPCWSTR lpszCookieName,
588     LPCWSTR lpCookieData)
589 {
590     BOOL ret;
591     WCHAR hostName[2048], path[2048];
592
593     TRACE("(%s,%s,%s)\n", debugstr_w(lpszUrl),
594         debugstr_w(lpszCookieName), debugstr_w(lpCookieData));
595
596     if (!lpszUrl || !lpCookieData)
597     {
598         SetLastError(ERROR_INVALID_PARAMETER);
599         return FALSE;
600     }
601
602     hostName[0] = path[0] = 0;
603     ret = COOKIE_crackUrlSimple(lpszUrl, hostName, sizeof(hostName)/sizeof(hostName[0]), path, sizeof(path)/sizeof(path[0]));
604     if (!ret || !hostName[0]) return FALSE;
605
606     if (!lpszCookieName)
607     {
608         unsigned int len;
609         WCHAR *cookie, *data;
610
611         len = strlenW(lpCookieData);
612         if (!(cookie = HeapAlloc(GetProcessHeap(), 0, (len + 1) * sizeof(WCHAR))))
613         {
614             SetLastError(ERROR_OUTOFMEMORY);
615             return FALSE;
616         }
617         strcpyW(cookie, lpCookieData);
618
619         /* some apps (or is it us??) try to add a cookie with no cookie name, but
620          * the cookie data in the form of name[=data].
621          */
622         if (!(data = strchrW(cookie, '='))) data = cookie + len;
623         else *data++ = 0;
624
625         ret = set_cookie(hostName, path, cookie, data);
626
627         HeapFree(GetProcessHeap(), 0, cookie);
628         return ret;
629     }
630     return set_cookie(hostName, path, lpszCookieName, lpCookieData);
631 }
632
633
634 /***********************************************************************
635  *           InternetSetCookieA (WININET.@)
636  *
637  * Sets cookie for the specified url
638  *
639  * RETURNS
640  *    TRUE  on success
641  *    FALSE on failure
642  *
643  */
644 BOOL WINAPI InternetSetCookieA(LPCSTR lpszUrl, LPCSTR lpszCookieName,
645     LPCSTR lpCookieData)
646 {
647     DWORD len;
648     LPWSTR szCookieData = NULL, szUrl = NULL, szCookieName = NULL;
649     BOOL r;
650
651     TRACE("(%s,%s,%s)\n", debugstr_a(lpszUrl),
652         debugstr_a(lpszCookieName), debugstr_a(lpCookieData));
653
654     if( lpszUrl )
655     {
656         len = MultiByteToWideChar( CP_ACP, 0, lpszUrl, -1, NULL, 0 );
657         szUrl = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) );
658         MultiByteToWideChar( CP_ACP, 0, lpszUrl, -1, szUrl, len );
659     }
660
661     if( lpszCookieName )
662     {
663         len = MultiByteToWideChar( CP_ACP, 0, lpszCookieName, -1, NULL, 0 );
664         szCookieName = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) );
665         MultiByteToWideChar( CP_ACP, 0, lpszCookieName, -1, szCookieName, len );
666     }
667
668     if( lpCookieData )
669     {
670         len = MultiByteToWideChar( CP_ACP, 0, lpCookieData, -1, NULL, 0 );
671         szCookieData = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) );
672         MultiByteToWideChar( CP_ACP, 0, lpCookieData, -1, szCookieData, len );
673     }
674
675     r = InternetSetCookieW( szUrl, szCookieName, szCookieData );
676
677     HeapFree( GetProcessHeap(), 0, szCookieData );
678     HeapFree( GetProcessHeap(), 0, szCookieName );
679     HeapFree( GetProcessHeap(), 0, szUrl );
680
681     return r;
682 }
683
684 /***********************************************************************
685  *           InternetSetCookieExA (WININET.@)
686  *
687  * See InternetSetCookieExW.
688  */
689 DWORD WINAPI InternetSetCookieExA( LPCSTR lpszURL, LPCSTR lpszCookieName, LPCSTR lpszCookieData,
690                                    DWORD dwFlags, DWORD_PTR dwReserved)
691 {
692     TRACE("(%s, %s, %s, 0x%08x, 0x%08lx)\n",
693           debugstr_a(lpszURL), debugstr_a(lpszCookieName), debugstr_a(lpszCookieData),
694           dwFlags, dwReserved);
695
696     if (dwFlags) FIXME("flags 0x%08x not supported\n", dwFlags);
697     return InternetSetCookieA(lpszURL, lpszCookieName, lpszCookieData);
698 }
699
700 /***********************************************************************
701  *           InternetSetCookieExW (WININET.@)
702  *
703  * Sets a cookie for the specified URL.
704  *
705  * RETURNS
706  *    TRUE  on success
707  *    FALSE on failure
708  *
709  */
710 DWORD WINAPI InternetSetCookieExW( LPCWSTR lpszURL, LPCWSTR lpszCookieName, LPCWSTR lpszCookieData,
711                                    DWORD dwFlags, DWORD_PTR dwReserved)
712 {
713     TRACE("(%s, %s, %s, 0x%08x, 0x%08lx)\n",
714           debugstr_w(lpszURL), debugstr_w(lpszCookieName), debugstr_w(lpszCookieData),
715           dwFlags, dwReserved);
716
717     if (dwFlags) FIXME("flags 0x%08x not supported\n", dwFlags);
718     return InternetSetCookieW(lpszURL, lpszCookieName, lpszCookieData);
719 }
720
721 /***********************************************************************
722  *           InternetGetCookieExA (WININET.@)
723  *
724  * See InternetGetCookieExW.
725  */
726 BOOL WINAPI InternetGetCookieExA( LPCSTR pchURL, LPCSTR pchCookieName, LPSTR pchCookieData,
727                                   LPDWORD pcchCookieData, DWORD dwFlags, LPVOID lpReserved)
728 {
729     TRACE("(%s, %s, %s, %p, 0x%08x, %p)\n",
730           debugstr_a(pchURL), debugstr_a(pchCookieName), debugstr_a(pchCookieData),
731           pcchCookieData, dwFlags, lpReserved);
732
733     if (dwFlags) FIXME("flags 0x%08x not supported\n", dwFlags);
734     return InternetGetCookieA(pchURL, pchCookieName, pchCookieData, pcchCookieData);
735 }
736
737 /***********************************************************************
738  *           InternetGetCookieExW (WININET.@)
739  *
740  * Retrieve cookie for the specified URL.
741  *
742  * RETURNS
743  *    TRUE  on success
744  *    FALSE on failure
745  *
746  */
747 BOOL WINAPI InternetGetCookieExW( LPCWSTR pchURL, LPCWSTR pchCookieName, LPWSTR pchCookieData,
748                                   LPDWORD pcchCookieData, DWORD dwFlags, LPVOID lpReserved)
749 {
750     TRACE("(%s, %s, %s, %p, 0x%08x, %p)\n",
751           debugstr_w(pchURL), debugstr_w(pchCookieName), debugstr_w(pchCookieData),
752           pcchCookieData, dwFlags, lpReserved);
753
754     if (dwFlags) FIXME("flags 0x%08x not supported\n", dwFlags);
755     return InternetGetCookieW(pchURL, pchCookieName, pchCookieData, pcchCookieData);
756 }
757
758 /***********************************************************************
759  *           InternetClearAllPerSiteCookieDecisions (WININET.@)
760  *
761  * Clears all per-site decisions about cookies.
762  *
763  * RETURNS
764  *    TRUE  on success
765  *    FALSE on failure
766  *
767  */
768 BOOL WINAPI InternetClearAllPerSiteCookieDecisions( VOID )
769 {
770     FIXME("stub\n");
771     return TRUE;
772 }
773
774 /***********************************************************************
775  *           InternetEnumPerSiteCookieDecisionA (WININET.@)
776  *
777  * See InternetEnumPerSiteCookieDecisionW.
778  */
779 BOOL WINAPI InternetEnumPerSiteCookieDecisionA( LPSTR pszSiteName, ULONG *pcSiteNameSize,
780                                                 ULONG *pdwDecision, ULONG dwIndex )
781 {
782     FIXME("(%s, %p, %p, 0x%08x) stub\n",
783           debugstr_a(pszSiteName), pcSiteNameSize, pdwDecision, dwIndex);
784     return FALSE;
785 }
786
787 /***********************************************************************
788  *           InternetEnumPerSiteCookieDecisionW (WININET.@)
789  *
790  * Enumerates all per-site decisions about cookies.
791  *
792  * RETURNS
793  *    TRUE  on success
794  *    FALSE on failure
795  *
796  */
797 BOOL WINAPI InternetEnumPerSiteCookieDecisionW( LPWSTR pszSiteName, ULONG *pcSiteNameSize,
798                                                 ULONG *pdwDecision, ULONG dwIndex )
799 {
800     FIXME("(%s, %p, %p, 0x%08x) stub\n",
801           debugstr_w(pszSiteName), pcSiteNameSize, pdwDecision, dwIndex);
802     return FALSE;
803 }
804
805 /***********************************************************************
806  *           InternetGetPerSiteCookieDecisionA (WININET.@)
807  */
808 BOOL WINAPI InternetGetPerSiteCookieDecisionA( LPCSTR pwchHostName, ULONG *pResult )
809 {
810     FIXME("(%s, %p) stub\n", debugstr_a(pwchHostName), pResult);
811     return FALSE;
812 }
813
814 /***********************************************************************
815  *           InternetGetPerSiteCookieDecisionW (WININET.@)
816  */
817 BOOL WINAPI InternetGetPerSiteCookieDecisionW( LPCWSTR pwchHostName, ULONG *pResult )
818 {
819     FIXME("(%s, %p) stub\n", debugstr_w(pwchHostName), pResult);
820     return FALSE;
821 }
822
823 /***********************************************************************
824  *           InternetSetPerSiteCookieDecisionA (WININET.@)
825  */
826 BOOL WINAPI InternetSetPerSiteCookieDecisionA( LPCSTR pchHostName, DWORD dwDecision )
827 {
828     FIXME("(%s, 0x%08x) stub\n", debugstr_a(pchHostName), dwDecision);
829     return FALSE;
830 }
831
832 /***********************************************************************
833  *           InternetSetPerSiteCookieDecisionW (WININET.@)
834  */
835 BOOL WINAPI InternetSetPerSiteCookieDecisionW( LPCWSTR pchHostName, DWORD dwDecision )
836 {
837     FIXME("(%s, 0x%08x) stub\n", debugstr_w(pchHostName), dwDecision);
838     return FALSE;
839 }
840
841 /***********************************************************************
842  *           IsDomainLegalCookieDomainW (WININET.@)
843  */
844 BOOL WINAPI IsDomainLegalCookieDomainW( LPCWSTR s1, LPCWSTR s2 )
845 {
846     const WCHAR *p;
847
848     FIXME("(%s, %s)\n", debugstr_w(s1), debugstr_w(s2));
849
850     if (!s1 || !s2)
851     {
852         SetLastError(ERROR_INVALID_PARAMETER);
853         return FALSE;
854     }
855     if (s1[0] == '.' || !s1[0] || s2[0] == '.' || !s2[0])
856     {
857         SetLastError(ERROR_INVALID_NAME);
858         return FALSE;
859     }
860     if (!(p = strchrW(s2, '.'))) return FALSE;
861     if (strchrW(p + 1, '.') && !strcmpW(p + 1, s1)) return TRUE;
862     else if (!strcmpW(s1, s2)) return TRUE;
863     return FALSE;
864 }