| /* |
| * Wininet - cookie handling stuff |
| * |
| * Copyright 2002 TransGaming Technologies Inc. |
| * |
| * David Hammerton |
| * |
| * This library is free software; you can redistribute it and/or |
| * modify it under the terms of the GNU Lesser General Public |
| * License as published by the Free Software Foundation; either |
| * version 2.1 of the License, or (at your option) any later version. |
| * |
| * This library is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| * Lesser General Public License for more details. |
| * |
| * You should have received a copy of the GNU Lesser General Public |
| * License along with this library; if not, write to the Free Software |
| * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA |
| */ |
| |
| #include "config.h" |
| #include "wine/port.h" |
| |
| #if defined(__MINGW32__) || defined (_MSC_VER) |
| #include <ws2tcpip.h> |
| #endif |
| |
| #include <stdarg.h> |
| #include <stdio.h> |
| #include <stdlib.h> |
| #include <string.h> |
| #ifdef HAVE_UNISTD_H |
| # include <unistd.h> |
| #endif |
| |
| #include "windef.h" |
| #include "winbase.h" |
| #include "wininet.h" |
| #include "winerror.h" |
| |
| #include "wine/debug.h" |
| #include "internet.h" |
| |
| #define RESPONSE_TIMEOUT 30 /* FROM internet.c */ |
| |
| |
| WINE_DEFAULT_DEBUG_CHANNEL(wininet); |
| |
| /* FIXME |
| * Cookies could use A LOT OF MEMORY. We need some kind of memory management here! |
| */ |
| |
| typedef struct _cookie_domain cookie_domain; |
| typedef struct _cookie cookie; |
| |
| struct _cookie |
| { |
| struct list entry; |
| |
| struct _cookie_domain *parent; |
| |
| LPWSTR lpCookieName; |
| LPWSTR lpCookieData; |
| DWORD flags; |
| FILETIME expiry; |
| FILETIME create; |
| }; |
| |
| struct _cookie_domain |
| { |
| struct list entry; |
| |
| LPWSTR lpCookieDomain; |
| LPWSTR lpCookiePath; |
| struct list cookie_list; |
| }; |
| |
| static CRITICAL_SECTION cookie_cs; |
| static CRITICAL_SECTION_DEBUG cookie_cs_debug = |
| { |
| 0, 0, &cookie_cs, |
| { &cookie_cs_debug.ProcessLocksList, &cookie_cs_debug.ProcessLocksList }, |
| 0, 0, { (DWORD_PTR)(__FILE__ ": cookie_cs") } |
| }; |
| static CRITICAL_SECTION cookie_cs = { &cookie_cs_debug, -1, 0, 0, 0, 0 }; |
| static struct list domain_list = LIST_INIT(domain_list); |
| |
| static cookie *COOKIE_addCookie(cookie_domain *domain, LPCWSTR name, LPCWSTR data, |
| FILETIME expiry, FILETIME create, DWORD flags); |
| static cookie *COOKIE_findCookie(cookie_domain *domain, LPCWSTR lpszCookieName); |
| static void COOKIE_deleteCookie(cookie *deadCookie, BOOL deleteDomain); |
| static cookie_domain *COOKIE_addDomain(LPCWSTR domain, LPCWSTR path); |
| static void COOKIE_deleteDomain(cookie_domain *deadDomain); |
| static BOOL COOKIE_matchDomain(LPCWSTR lpszCookieDomain, LPCWSTR lpszCookiePath, |
| cookie_domain *searchDomain, BOOL allow_partial); |
| |
| static BOOL create_cookie_url(LPCWSTR domain, LPCWSTR path, WCHAR *buf, DWORD buf_len) |
| { |
| static const WCHAR cookie_prefix[] = {'C','o','o','k','i','e',':'}; |
| |
| WCHAR *p; |
| DWORD len; |
| |
| if(buf_len < sizeof(cookie_prefix)/sizeof(WCHAR)) |
| return FALSE; |
| memcpy(buf, cookie_prefix, sizeof(cookie_prefix)); |
| buf += sizeof(cookie_prefix)/sizeof(WCHAR); |
| buf_len -= sizeof(cookie_prefix)/sizeof(WCHAR); |
| p = buf; |
| |
| len = buf_len; |
| if(!GetUserNameW(buf, &len)) |
| return FALSE; |
| buf += len-1; |
| buf_len -= len-1; |
| |
| if(!buf_len) |
| return FALSE; |
| *(buf++) = '@'; |
| buf_len--; |
| |
| len = strlenW(domain); |
| if(len >= buf_len) |
| return FALSE; |
| memcpy(buf, domain, len*sizeof(WCHAR)); |
| buf += len; |
| buf_len -= len; |
| |
| len = strlenW(path); |
| if(len >= buf_len) |
| return FALSE; |
| memcpy(buf, path, len*sizeof(WCHAR)); |
| buf += len; |
| |
| *buf = 0; |
| |
| for(; *p; p++) |
| *p = tolowerW(*p); |
| return TRUE; |
| } |
| |
| static BOOL load_persistent_cookie(LPCWSTR domain, LPCWSTR path) |
| { |
| INTERNET_CACHE_ENTRY_INFOW *info; |
| cookie_domain *domain_container = NULL; |
| cookie *old_cookie; |
| struct list *iter; |
| WCHAR cookie_url[MAX_PATH]; |
| HANDLE cookie; |
| char *str = NULL, *pbeg, *pend; |
| DWORD size, flags; |
| WCHAR *name, *data; |
| FILETIME expiry, create, time; |
| |
| if (!create_cookie_url(domain, path, cookie_url, sizeof(cookie_url)/sizeof(cookie_url[0]))) |
| return FALSE; |
| |
| size = 0; |
| RetrieveUrlCacheEntryStreamW(cookie_url, NULL, &size, FALSE, 0); |
| if(GetLastError() != ERROR_INSUFFICIENT_BUFFER) |
| return TRUE; |
| info = heap_alloc(size); |
| if(!info) |
| return FALSE; |
| cookie = RetrieveUrlCacheEntryStreamW(cookie_url, info, &size, FALSE, 0); |
| size = info->dwSizeLow; |
| heap_free(info); |
| if(!cookie) |
| return FALSE; |
| |
| if(!(str = heap_alloc(size+1)) || !ReadUrlCacheEntryStream(cookie, 0, str, &size, 0)) { |
| UnlockUrlCacheEntryStream(cookie, 0); |
| heap_free(str); |
| return FALSE; |
| } |
| str[size] = 0; |
| UnlockUrlCacheEntryStream(cookie, 0); |
| |
| LIST_FOR_EACH(iter, &domain_list) |
| { |
| domain_container = LIST_ENTRY(iter, cookie_domain, entry); |
| if(COOKIE_matchDomain(domain, path, domain_container, FALSE)) |
| break; |
| domain_container = NULL; |
| } |
| if(!domain_container) |
| domain_container = COOKIE_addDomain(domain, path); |
| if(!domain_container) { |
| heap_free(str); |
| return FALSE; |
| } |
| |
| GetSystemTimeAsFileTime(&time); |
| for(pbeg=str; pbeg && *pbeg; name=data=NULL) { |
| pend = strchr(pbeg, '\n'); |
| if(!pend) |
| break; |
| *pend = 0; |
| name = heap_strdupAtoW(pbeg); |
| |
| pbeg = pend+1; |
| pend = strchr(pbeg, '\n'); |
| if(!pend) |
| break; |
| *pend = 0; |
| data = heap_strdupAtoW(pbeg); |
| |
| pbeg = pend+1; |
| pbeg = strchr(pend+1, '\n'); |
| if(!pbeg) |
| break; |
| sscanf(pbeg, "%u %u %u %u %u", &flags, &expiry.dwLowDateTime, &expiry.dwHighDateTime, |
| &create.dwLowDateTime, &create.dwHighDateTime); |
| |
| /* skip "*\n" */ |
| pbeg = strchr(pbeg, '*'); |
| if(pbeg) { |
| pbeg++; |
| if(*pbeg) |
| pbeg++; |
| } |
| |
| if(!name || !data) |
| break; |
| |
| if(CompareFileTime(&time, &expiry) <= 0) { |
| if((old_cookie = COOKIE_findCookie(domain_container, name))) |
| COOKIE_deleteCookie(old_cookie, FALSE); |
| COOKIE_addCookie(domain_container, name, data, expiry, create, flags); |
| } |
| heap_free(name); |
| heap_free(data); |
| } |
| heap_free(name); |
| heap_free(data); |
| |
| return TRUE; |
| } |
| |
| static BOOL save_persistent_cookie(cookie_domain *domain) |
| { |
| static const WCHAR txtW[] = {'t','x','t',0}; |
| |
| WCHAR cookie_url[MAX_PATH], cookie_file[MAX_PATH]; |
| HANDLE cookie_handle; |
| cookie *cookie_container = NULL, *cookie_iter; |
| BOOL do_save = FALSE; |
| char buf[64], *dyn_buf; |
| FILETIME time; |
| |
| if (!create_cookie_url(domain->lpCookieDomain, domain->lpCookiePath, cookie_url, sizeof(cookie_url)/sizeof(cookie_url[0]))) |
| return FALSE; |
| |
| /* check if there's anything to save */ |
| GetSystemTimeAsFileTime(&time); |
| LIST_FOR_EACH_ENTRY_SAFE(cookie_container, cookie_iter, &domain->cookie_list, cookie, entry) |
| { |
| if((cookie_container->expiry.dwLowDateTime || cookie_container->expiry.dwHighDateTime) |
| && CompareFileTime(&time, &cookie_container->expiry) > 0) { |
| COOKIE_deleteCookie(cookie_container, FALSE); |
| continue; |
| } |
| |
| if(!(cookie_container->flags & INTERNET_COOKIE_IS_SESSION)) { |
| do_save = TRUE; |
| break; |
| } |
| } |
| if(!do_save) { |
| DeleteUrlCacheEntryW(cookie_url); |
| return TRUE; |
| } |
| |
| if(!CreateUrlCacheEntryW(cookie_url, 0, txtW, cookie_file, 0)) |
| return FALSE; |
| cookie_handle = CreateFileW(cookie_file, GENERIC_WRITE, 0, NULL, OPEN_EXISTING, 0, NULL); |
| if(cookie_handle == INVALID_HANDLE_VALUE) { |
| DeleteFileW(cookie_file); |
| return FALSE; |
| } |
| |
| LIST_FOR_EACH_ENTRY(cookie_container, &domain->cookie_list, cookie, entry) |
| { |
| if(cookie_container->flags & INTERNET_COOKIE_IS_SESSION) |
| continue; |
| |
| dyn_buf = heap_strdupWtoA(cookie_container->lpCookieName); |
| if(!dyn_buf || !WriteFile(cookie_handle, dyn_buf, strlen(dyn_buf), NULL, NULL)) { |
| heap_free(dyn_buf); |
| do_save = FALSE; |
| break; |
| } |
| heap_free(dyn_buf); |
| if(!WriteFile(cookie_handle, "\n", 1, NULL, NULL)) { |
| do_save = FALSE; |
| break; |
| } |
| |
| dyn_buf = heap_strdupWtoA(cookie_container->lpCookieData); |
| if(!dyn_buf || !WriteFile(cookie_handle, dyn_buf, strlen(dyn_buf), NULL, NULL)) { |
| heap_free(dyn_buf); |
| do_save = FALSE; |
| break; |
| } |
| heap_free(dyn_buf); |
| if(!WriteFile(cookie_handle, "\n", 1, NULL, NULL)) { |
| do_save = FALSE; |
| break; |
| } |
| |
| dyn_buf = heap_strdupWtoA(domain->lpCookieDomain); |
| if(!dyn_buf || !WriteFile(cookie_handle, dyn_buf, strlen(dyn_buf), NULL, NULL)) { |
| heap_free(dyn_buf); |
| do_save = FALSE; |
| break; |
| } |
| heap_free(dyn_buf); |
| |
| dyn_buf = heap_strdupWtoA(domain->lpCookiePath); |
| if(!dyn_buf || !WriteFile(cookie_handle, dyn_buf, strlen(dyn_buf), NULL, NULL)) { |
| heap_free(dyn_buf); |
| do_save = FALSE; |
| break; |
| } |
| heap_free(dyn_buf); |
| |
| sprintf(buf, "\n%u\n%u\n%u\n%u\n%u\n*\n", cookie_container->flags, |
| cookie_container->expiry.dwLowDateTime, cookie_container->expiry.dwHighDateTime, |
| cookie_container->create.dwLowDateTime, cookie_container->create.dwHighDateTime); |
| if(!WriteFile(cookie_handle, buf, strlen(buf), NULL, NULL)) { |
| do_save = FALSE; |
| break; |
| } |
| } |
| |
| CloseHandle(cookie_handle); |
| if(!do_save) { |
| ERR("error saving cookie file\n"); |
| DeleteFileW(cookie_file); |
| return FALSE; |
| } |
| |
| memset(&time, 0, sizeof(time)); |
| return CommitUrlCacheEntryW(cookie_url, cookie_file, time, time, 0, NULL, 0, txtW, 0); |
| } |
| |
| /* adds a cookie to the domain */ |
| static cookie *COOKIE_addCookie(cookie_domain *domain, LPCWSTR name, LPCWSTR data, |
| FILETIME expiry, FILETIME create, DWORD flags) |
| { |
| cookie *newCookie = heap_alloc(sizeof(cookie)); |
| if (!newCookie) |
| return NULL; |
| |
| newCookie->lpCookieName = heap_strdupW(name); |
| newCookie->lpCookieData = heap_strdupW(data); |
| |
| if (!newCookie->lpCookieName || !newCookie->lpCookieData) |
| { |
| heap_free(newCookie->lpCookieName); |
| heap_free(newCookie->lpCookieData); |
| heap_free(newCookie); |
| |
| return NULL; |
| } |
| |
| newCookie->flags = flags; |
| newCookie->expiry = expiry; |
| newCookie->create = create; |
| |
| TRACE("added cookie %p (data is %s)\n", newCookie, debugstr_w(data) ); |
| |
| list_add_tail(&domain->cookie_list, &newCookie->entry); |
| newCookie->parent = domain; |
| return newCookie; |
| } |
| |
| |
| /* finds a cookie in the domain matching the cookie name */ |
| static cookie *COOKIE_findCookie(cookie_domain *domain, LPCWSTR lpszCookieName) |
| { |
| struct list * cursor; |
| TRACE("(%p, %s)\n", domain, debugstr_w(lpszCookieName)); |
| |
| LIST_FOR_EACH(cursor, &domain->cookie_list) |
| { |
| cookie *searchCookie = LIST_ENTRY(cursor, cookie, entry); |
| BOOL candidate = TRUE; |
| if (candidate && lpszCookieName) |
| { |
| if (candidate && !searchCookie->lpCookieName) |
| candidate = FALSE; |
| if (candidate && strcmpW(lpszCookieName, searchCookie->lpCookieName) != 0) |
| candidate = FALSE; |
| } |
| if (candidate) |
| return searchCookie; |
| } |
| return NULL; |
| } |
| |
| /* removes a cookie from the list, if its the last cookie we also remove the domain */ |
| static void COOKIE_deleteCookie(cookie *deadCookie, BOOL deleteDomain) |
| { |
| heap_free(deadCookie->lpCookieName); |
| heap_free(deadCookie->lpCookieData); |
| list_remove(&deadCookie->entry); |
| |
| /* special case: last cookie, lets remove the domain to save memory */ |
| if (list_empty(&deadCookie->parent->cookie_list) && deleteDomain) |
| COOKIE_deleteDomain(deadCookie->parent); |
| heap_free(deadCookie); |
| } |
| |
| /* allocates a domain and adds it to the end */ |
| static cookie_domain *COOKIE_addDomain(LPCWSTR domain, LPCWSTR path) |
| { |
| cookie_domain *newDomain = heap_alloc(sizeof(cookie_domain)); |
| |
| list_init(&newDomain->entry); |
| list_init(&newDomain->cookie_list); |
| newDomain->lpCookieDomain = heap_strdupW(domain); |
| newDomain->lpCookiePath = heap_strdupW(path); |
| |
| list_add_tail(&domain_list, &newDomain->entry); |
| |
| TRACE("Adding domain: %p\n", newDomain); |
| return newDomain; |
| } |
| |
| static BOOL COOKIE_crackUrlSimple(LPCWSTR lpszUrl, LPWSTR hostName, int hostNameLen, LPWSTR path, int pathLen) |
| { |
| URL_COMPONENTSW UrlComponents; |
| |
| UrlComponents.lpszExtraInfo = NULL; |
| UrlComponents.lpszPassword = NULL; |
| UrlComponents.lpszScheme = NULL; |
| UrlComponents.lpszUrlPath = path; |
| UrlComponents.lpszUserName = NULL; |
| UrlComponents.lpszHostName = hostName; |
| UrlComponents.dwExtraInfoLength = 0; |
| UrlComponents.dwPasswordLength = 0; |
| UrlComponents.dwSchemeLength = 0; |
| UrlComponents.dwUserNameLength = 0; |
| UrlComponents.dwHostNameLength = hostNameLen; |
| UrlComponents.dwUrlPathLength = pathLen; |
| |
| if (!InternetCrackUrlW(lpszUrl, 0, 0, &UrlComponents)) return FALSE; |
| |
| /* discard the webpage off the end of the path */ |
| if (UrlComponents.dwUrlPathLength) |
| { |
| if (path[UrlComponents.dwUrlPathLength - 1] != '/') |
| { |
| WCHAR *ptr; |
| if ((ptr = strrchrW(path, '/'))) *(++ptr) = 0; |
| else |
| { |
| path[0] = '/'; |
| path[1] = 0; |
| } |
| } |
| } |
| else if (pathLen >= 2) |
| { |
| path[0] = '/'; |
| path[1] = 0; |
| } |
| return TRUE; |
| } |
| |
| /* match a domain. domain must match if the domain is not NULL. path must match if the path is not NULL */ |
| static BOOL COOKIE_matchDomain(LPCWSTR lpszCookieDomain, LPCWSTR lpszCookiePath, |
| cookie_domain *searchDomain, BOOL allow_partial) |
| { |
| TRACE("searching on domain %p\n", searchDomain); |
| if (lpszCookieDomain) |
| { |
| if (!searchDomain->lpCookieDomain) |
| return FALSE; |
| |
| TRACE("comparing domain %s with %s\n", |
| debugstr_w(lpszCookieDomain), |
| debugstr_w(searchDomain->lpCookieDomain)); |
| |
| if (allow_partial && !strstrW(lpszCookieDomain, searchDomain->lpCookieDomain)) |
| return FALSE; |
| else if (!allow_partial && lstrcmpW(lpszCookieDomain, searchDomain->lpCookieDomain) != 0) |
| return FALSE; |
| } |
| if (lpszCookiePath) |
| { |
| INT len; |
| TRACE("comparing paths: %s with %s\n", debugstr_w(lpszCookiePath), debugstr_w(searchDomain->lpCookiePath)); |
| /* paths match at the beginning. so a path of /foo would match |
| * /foobar and /foo/bar |
| */ |
| if (!searchDomain->lpCookiePath) |
| return FALSE; |
| if (allow_partial) |
| { |
| len = lstrlenW(searchDomain->lpCookiePath); |
| if (strncmpiW(searchDomain->lpCookiePath, lpszCookiePath, len)!=0) |
| return FALSE; |
| } |
| else if (strcmpW(lpszCookiePath, searchDomain->lpCookiePath)) |
| return FALSE; |
| |
| } |
| return TRUE; |
| } |
| |
| /* remove a domain from the list and delete it */ |
| static void COOKIE_deleteDomain(cookie_domain *deadDomain) |
| { |
| struct list * cursor; |
| while ((cursor = list_tail(&deadDomain->cookie_list))) |
| { |
| COOKIE_deleteCookie(LIST_ENTRY(cursor, cookie, entry), FALSE); |
| list_remove(cursor); |
| } |
| heap_free(deadDomain->lpCookieDomain); |
| heap_free(deadDomain->lpCookiePath); |
| |
| list_remove(&deadDomain->entry); |
| |
| heap_free(deadDomain); |
| } |
| |
| BOOL get_cookie(const WCHAR *host, const WCHAR *path, WCHAR *cookie_data, DWORD *size) |
| { |
| unsigned cnt = 0, len, domain_count = 0, cookie_count = 0; |
| cookie_domain *domain; |
| FILETIME tm; |
| |
| GetSystemTimeAsFileTime(&tm); |
| |
| EnterCriticalSection(&cookie_cs); |
| |
| load_persistent_cookie(host, path); |
| |
| LIST_FOR_EACH_ENTRY(domain, &domain_list, cookie_domain, entry) { |
| struct list *cursor, *cursor2; |
| |
| if(!COOKIE_matchDomain(host, path, domain, TRUE)) |
| continue; |
| |
| domain_count++; |
| TRACE("found domain %p\n", domain); |
| |
| LIST_FOR_EACH_SAFE(cursor, cursor2, &domain->cookie_list) { |
| cookie *cookie_iter = LIST_ENTRY(cursor, cookie, entry); |
| |
| /* check for expiry */ |
| if((cookie_iter->expiry.dwLowDateTime != 0 || cookie_iter->expiry.dwHighDateTime != 0) |
| && CompareFileTime(&tm, &cookie_iter->expiry) > 0) |
| { |
| TRACE("Found expired cookie. deleting\n"); |
| COOKIE_deleteCookie(cookie_iter, FALSE); |
| continue; |
| } |
| |
| if(!cookie_data) { /* return the size of the buffer required to lpdwSize */ |
| if (cookie_count) |
| cnt += 2; /* '; ' */ |
| cnt += strlenW(cookie_iter->lpCookieName); |
| if ((len = strlenW(cookie_iter->lpCookieData))) { |
| cnt += 1; /* = */ |
| cnt += len; |
| } |
| }else { |
| static const WCHAR szsc[] = { ';',' ',0 }; |
| static const WCHAR szname[] = { '%','s',0 }; |
| static const WCHAR szdata[] = { '=','%','s',0 }; |
| |
| if (cookie_count) cnt += snprintfW(cookie_data + cnt, *size - cnt, szsc); |
| cnt += snprintfW(cookie_data + cnt, *size - cnt, szname, cookie_iter->lpCookieName); |
| |
| if (cookie_iter->lpCookieData[0]) |
| cnt += snprintfW(cookie_data + cnt, *size - cnt, szdata, cookie_iter->lpCookieData); |
| |
| TRACE("Cookie: %s\n", debugstr_w(cookie_data)); |
| } |
| cookie_count++; |
| } |
| } |
| |
| LeaveCriticalSection(&cookie_cs); |
| |
| if (!domain_count) { |
| TRACE("no cookies found for %s\n", debugstr_w(host)); |
| SetLastError(ERROR_NO_MORE_ITEMS); |
| return FALSE; |
| } |
| |
| if(!cookie_data) { |
| *size = (cnt + 1) * sizeof(WCHAR); |
| TRACE("returning %u\n", *size); |
| return TRUE; |
| } |
| |
| *size = cnt + 1; |
| |
| TRACE("Returning %u (from %u domains): %s\n", cnt, domain_count, debugstr_w(cookie_data)); |
| return cnt != 0; |
| } |
| |
| /*********************************************************************** |
| * InternetGetCookieW (WININET.@) |
| * |
| * Retrieve cookie from the specified url |
| * |
| * It should be noted that on windows the lpszCookieName parameter is "not implemented". |
| * So it won't be implemented here. |
| * |
| * RETURNS |
| * TRUE on success |
| * FALSE on failure |
| * |
| */ |
| BOOL WINAPI InternetGetCookieW(LPCWSTR lpszUrl, LPCWSTR lpszCookieName, |
| LPWSTR lpCookieData, LPDWORD lpdwSize) |
| { |
| WCHAR host[INTERNET_MAX_HOST_NAME_LENGTH], path[INTERNET_MAX_PATH_LENGTH]; |
| BOOL ret; |
| |
| TRACE("(%s, %s, %p, %p)\n", debugstr_w(lpszUrl),debugstr_w(lpszCookieName), lpCookieData, lpdwSize); |
| |
| if (!lpszUrl) |
| { |
| SetLastError(ERROR_INVALID_PARAMETER); |
| return FALSE; |
| } |
| |
| host[0] = 0; |
| ret = COOKIE_crackUrlSimple(lpszUrl, host, sizeof(host)/sizeof(host[0]), path, sizeof(path)/sizeof(path[0])); |
| if (!ret || !host[0]) { |
| SetLastError(ERROR_INVALID_PARAMETER); |
| return FALSE; |
| } |
| |
| return get_cookie(host, path, lpCookieData, lpdwSize); |
| } |
| |
| |
| /*********************************************************************** |
| * InternetGetCookieA (WININET.@) |
| * |
| * Retrieve cookie from the specified url |
| * |
| * RETURNS |
| * TRUE on success |
| * FALSE on failure |
| * |
| */ |
| BOOL WINAPI InternetGetCookieA(LPCSTR lpszUrl, LPCSTR lpszCookieName, |
| LPSTR lpCookieData, LPDWORD lpdwSize) |
| { |
| WCHAR *url, *name; |
| DWORD len; |
| BOOL r; |
| |
| TRACE("(%s %s %p %p(%u))\n", debugstr_a(lpszUrl), debugstr_a(lpszCookieName), |
| lpCookieData, lpdwSize, lpdwSize ? *lpdwSize : 0); |
| |
| url = heap_strdupAtoW(lpszUrl); |
| name = heap_strdupAtoW(lpszCookieName); |
| |
| r = InternetGetCookieW( url, name, NULL, &len ); |
| if( r ) |
| { |
| WCHAR *szCookieData; |
| |
| szCookieData = heap_alloc(len * sizeof(WCHAR)); |
| if( !szCookieData ) |
| { |
| r = FALSE; |
| } |
| else |
| { |
| r = InternetGetCookieW( url, name, szCookieData, &len ); |
| |
| *lpdwSize = WideCharToMultiByte( CP_ACP, 0, szCookieData, len, |
| lpCookieData, lpCookieData ? *lpdwSize : 0, NULL, NULL ); |
| |
| heap_free( szCookieData ); |
| } |
| } |
| heap_free( name ); |
| heap_free( url ); |
| return r; |
| } |
| |
| |
| /*********************************************************************** |
| * IsDomainLegalCookieDomainW (WININET.@) |
| */ |
| BOOL WINAPI IsDomainLegalCookieDomainW( LPCWSTR s1, LPCWSTR s2 ) |
| { |
| DWORD s1_len, s2_len; |
| |
| FIXME("(%s, %s) semi-stub\n", debugstr_w(s1), debugstr_w(s2)); |
| |
| if (!s1 || !s2) |
| { |
| SetLastError(ERROR_INVALID_PARAMETER); |
| return FALSE; |
| } |
| if (s1[0] == '.' || !s1[0] || s2[0] == '.' || !s2[0]) |
| { |
| SetLastError(ERROR_INVALID_NAME); |
| return FALSE; |
| } |
| if(!strchrW(s1, '.') || !strchrW(s2, '.')) |
| return FALSE; |
| |
| s1_len = strlenW(s1); |
| s2_len = strlenW(s2); |
| if (s1_len > s2_len) |
| return FALSE; |
| |
| if (strncmpiW(s1, s2+s2_len-s1_len, s1_len) || (s2_len>s1_len && s2[s2_len-s1_len-1]!='.')) |
| { |
| SetLastError(ERROR_INVALID_PARAMETER); |
| return FALSE; |
| } |
| |
| return TRUE; |
| } |
| |
| BOOL set_cookie(LPCWSTR domain, LPCWSTR path, LPCWSTR cookie_name, LPCWSTR cookie_data) |
| { |
| cookie_domain *thisCookieDomain = NULL; |
| cookie *thisCookie; |
| struct list *cursor; |
| LPWSTR data, value; |
| WCHAR *ptr; |
| FILETIME expiry, create; |
| BOOL expired = FALSE, update_persistent = FALSE; |
| DWORD flags = 0; |
| |
| value = data = heap_strdupW(cookie_data); |
| if (!data) |
| { |
| ERR("could not allocate the cookie data buffer\n"); |
| return FALSE; |
| } |
| |
| memset(&expiry,0,sizeof(expiry)); |
| GetSystemTimeAsFileTime(&create); |
| |
| /* lots of information can be parsed out of the cookie value */ |
| |
| ptr = data; |
| for (;;) |
| { |
| static const WCHAR szDomain[] = {'d','o','m','a','i','n','=',0}; |
| static const WCHAR szPath[] = {'p','a','t','h','=',0}; |
| static const WCHAR szExpires[] = {'e','x','p','i','r','e','s','=',0}; |
| static const WCHAR szSecure[] = {'s','e','c','u','r','e',0}; |
| static const WCHAR szHttpOnly[] = {'h','t','t','p','o','n','l','y',0}; |
| |
| if (!(ptr = strchrW(ptr,';'))) break; |
| *ptr++ = 0; |
| |
| if (value != data) heap_free(value); |
| value = heap_alloc((ptr - data) * sizeof(WCHAR)); |
| if (value == NULL) |
| { |
| heap_free(data); |
| ERR("could not allocate the cookie value buffer\n"); |
| return FALSE; |
| } |
| strcpyW(value, data); |
| |
| while (*ptr == ' ') ptr++; /* whitespace */ |
| |
| if (strncmpiW(ptr, szDomain, 7) == 0) |
| { |
| WCHAR *end_ptr; |
| |
| ptr += sizeof(szDomain)/sizeof(szDomain[0])-1; |
| if(*ptr == '.') |
| ptr++; |
| end_ptr = strchrW(ptr, ';'); |
| if(end_ptr) |
| *end_ptr = 0; |
| |
| if(!IsDomainLegalCookieDomainW(ptr, domain)) |
| { |
| if(value != data) |
| heap_free(value); |
| heap_free(data); |
| return FALSE; |
| } |
| |
| if(end_ptr) |
| *end_ptr = ';'; |
| |
| domain = ptr; |
| TRACE("Parsing new domain %s\n",debugstr_w(domain)); |
| } |
| else if (strncmpiW(ptr, szPath, 5) == 0) |
| { |
| ptr+=strlenW(szPath); |
| path = ptr; |
| TRACE("Parsing new path %s\n",debugstr_w(path)); |
| } |
| else if (strncmpiW(ptr, szExpires, 8) == 0) |
| { |
| SYSTEMTIME st; |
| ptr+=strlenW(szExpires); |
| if (InternetTimeToSystemTimeW(ptr, &st, 0)) |
| { |
| SystemTimeToFileTime(&st, &expiry); |
| |
| if (CompareFileTime(&create,&expiry) > 0) |
| { |
| TRACE("Cookie already expired.\n"); |
| expired = TRUE; |
| } |
| } |
| } |
| else if (strncmpiW(ptr, szSecure, 6) == 0) |
| { |
| FIXME("secure not handled (%s)\n",debugstr_w(ptr)); |
| ptr += strlenW(szSecure); |
| } |
| else if (strncmpiW(ptr, szHttpOnly, 8) == 0) |
| { |
| FIXME("httponly not handled (%s)\n",debugstr_w(ptr)); |
| ptr += strlenW(szHttpOnly); |
| } |
| else if (*ptr) |
| { |
| FIXME("Unknown additional option %s\n",debugstr_w(ptr)); |
| break; |
| } |
| } |
| |
| EnterCriticalSection(&cookie_cs); |
| |
| load_persistent_cookie(domain, path); |
| |
| LIST_FOR_EACH(cursor, &domain_list) |
| { |
| thisCookieDomain = LIST_ENTRY(cursor, cookie_domain, entry); |
| if (COOKIE_matchDomain(domain, path, thisCookieDomain, FALSE)) |
| break; |
| thisCookieDomain = NULL; |
| } |
| |
| if (!thisCookieDomain) |
| { |
| if (!expired) |
| thisCookieDomain = COOKIE_addDomain(domain, path); |
| else |
| { |
| heap_free(data); |
| if (value != data) heap_free(value); |
| LeaveCriticalSection(&cookie_cs); |
| return TRUE; |
| } |
| } |
| |
| if(!expiry.dwLowDateTime && !expiry.dwHighDateTime) |
| flags |= INTERNET_COOKIE_IS_SESSION; |
| else |
| update_persistent = TRUE; |
| |
| if ((thisCookie = COOKIE_findCookie(thisCookieDomain, cookie_name))) |
| { |
| if (!(thisCookie->flags & INTERNET_COOKIE_IS_SESSION)) |
| update_persistent = TRUE; |
| COOKIE_deleteCookie(thisCookie, FALSE); |
| } |
| |
| TRACE("setting cookie %s=%s for domain %s path %s\n", debugstr_w(cookie_name), |
| debugstr_w(value), debugstr_w(thisCookieDomain->lpCookieDomain),debugstr_w(thisCookieDomain->lpCookiePath)); |
| |
| if (!expired && !COOKIE_addCookie(thisCookieDomain, cookie_name, value, expiry, create, flags)) |
| { |
| heap_free(data); |
| if (value != data) heap_free(value); |
| LeaveCriticalSection(&cookie_cs); |
| return FALSE; |
| } |
| heap_free(data); |
| if (value != data) heap_free(value); |
| |
| if (!update_persistent || save_persistent_cookie(thisCookieDomain)) |
| { |
| LeaveCriticalSection(&cookie_cs); |
| return TRUE; |
| } |
| LeaveCriticalSection(&cookie_cs); |
| return FALSE; |
| } |
| |
| /*********************************************************************** |
| * InternetSetCookieW (WININET.@) |
| * |
| * Sets cookie for the specified url |
| * |
| * RETURNS |
| * TRUE on success |
| * FALSE on failure |
| * |
| */ |
| BOOL WINAPI InternetSetCookieW(LPCWSTR lpszUrl, LPCWSTR lpszCookieName, |
| LPCWSTR lpCookieData) |
| { |
| BOOL ret; |
| WCHAR hostName[INTERNET_MAX_HOST_NAME_LENGTH], path[INTERNET_MAX_PATH_LENGTH]; |
| |
| TRACE("(%s,%s,%s)\n", debugstr_w(lpszUrl), |
| debugstr_w(lpszCookieName), debugstr_w(lpCookieData)); |
| |
| if (!lpszUrl || !lpCookieData) |
| { |
| SetLastError(ERROR_INVALID_PARAMETER); |
| return FALSE; |
| } |
| |
| hostName[0] = 0; |
| ret = COOKIE_crackUrlSimple(lpszUrl, hostName, sizeof(hostName)/sizeof(hostName[0]), path, sizeof(path)/sizeof(path[0])); |
| if (!ret || !hostName[0]) return FALSE; |
| |
| if (!lpszCookieName) |
| { |
| WCHAR *cookie, *data; |
| |
| cookie = heap_strdupW(lpCookieData); |
| if (!cookie) |
| { |
| SetLastError(ERROR_OUTOFMEMORY); |
| return FALSE; |
| } |
| |
| /* some apps (or is it us??) try to add a cookie with no cookie name, but |
| * the cookie data in the form of name[=data]. |
| */ |
| if (!(data = strchrW(cookie, '='))) data = cookie + strlenW(cookie); |
| else *data++ = 0; |
| |
| ret = set_cookie(hostName, path, cookie, data); |
| |
| heap_free(cookie); |
| return ret; |
| } |
| return set_cookie(hostName, path, lpszCookieName, lpCookieData); |
| } |
| |
| |
| /*********************************************************************** |
| * InternetSetCookieA (WININET.@) |
| * |
| * Sets cookie for the specified url |
| * |
| * RETURNS |
| * TRUE on success |
| * FALSE on failure |
| * |
| */ |
| BOOL WINAPI InternetSetCookieA(LPCSTR lpszUrl, LPCSTR lpszCookieName, |
| LPCSTR lpCookieData) |
| { |
| LPWSTR data, url, name; |
| BOOL r; |
| |
| TRACE("(%s,%s,%s)\n", debugstr_a(lpszUrl), |
| debugstr_a(lpszCookieName), debugstr_a(lpCookieData)); |
| |
| url = heap_strdupAtoW(lpszUrl); |
| name = heap_strdupAtoW(lpszCookieName); |
| data = heap_strdupAtoW(lpCookieData); |
| |
| r = InternetSetCookieW( url, name, data ); |
| |
| heap_free( data ); |
| heap_free( name ); |
| heap_free( url ); |
| return r; |
| } |
| |
| /*********************************************************************** |
| * InternetSetCookieExA (WININET.@) |
| * |
| * See InternetSetCookieExW. |
| */ |
| DWORD WINAPI InternetSetCookieExA( LPCSTR lpszURL, LPCSTR lpszCookieName, LPCSTR lpszCookieData, |
| DWORD dwFlags, DWORD_PTR dwReserved) |
| { |
| TRACE("(%s, %s, %s, 0x%08x, 0x%08lx)\n", |
| debugstr_a(lpszURL), debugstr_a(lpszCookieName), debugstr_a(lpszCookieData), |
| dwFlags, dwReserved); |
| |
| if (dwFlags) FIXME("flags 0x%08x not supported\n", dwFlags); |
| return InternetSetCookieA(lpszURL, lpszCookieName, lpszCookieData); |
| } |
| |
| /*********************************************************************** |
| * InternetSetCookieExW (WININET.@) |
| * |
| * Sets a cookie for the specified URL. |
| * |
| * RETURNS |
| * TRUE on success |
| * FALSE on failure |
| * |
| */ |
| DWORD WINAPI InternetSetCookieExW( LPCWSTR lpszURL, LPCWSTR lpszCookieName, LPCWSTR lpszCookieData, |
| DWORD dwFlags, DWORD_PTR dwReserved) |
| { |
| TRACE("(%s, %s, %s, 0x%08x, 0x%08lx)\n", |
| debugstr_w(lpszURL), debugstr_w(lpszCookieName), debugstr_w(lpszCookieData), |
| dwFlags, dwReserved); |
| |
| if (dwFlags) FIXME("flags 0x%08x not supported\n", dwFlags); |
| return InternetSetCookieW(lpszURL, lpszCookieName, lpszCookieData); |
| } |
| |
| /*********************************************************************** |
| * InternetGetCookieExA (WININET.@) |
| * |
| * See InternetGetCookieExW. |
| */ |
| BOOL WINAPI InternetGetCookieExA( LPCSTR pchURL, LPCSTR pchCookieName, LPSTR pchCookieData, |
| LPDWORD pcchCookieData, DWORD dwFlags, LPVOID lpReserved) |
| { |
| TRACE("(%s, %s, %s, %p, 0x%08x, %p)\n", |
| debugstr_a(pchURL), debugstr_a(pchCookieName), debugstr_a(pchCookieData), |
| pcchCookieData, dwFlags, lpReserved); |
| |
| if (dwFlags) FIXME("flags 0x%08x not supported\n", dwFlags); |
| return InternetGetCookieA(pchURL, pchCookieName, pchCookieData, pcchCookieData); |
| } |
| |
| /*********************************************************************** |
| * InternetGetCookieExW (WININET.@) |
| * |
| * Retrieve cookie for the specified URL. |
| * |
| * RETURNS |
| * TRUE on success |
| * FALSE on failure |
| * |
| */ |
| BOOL WINAPI InternetGetCookieExW( LPCWSTR pchURL, LPCWSTR pchCookieName, LPWSTR pchCookieData, |
| LPDWORD pcchCookieData, DWORD dwFlags, LPVOID lpReserved) |
| { |
| TRACE("(%s, %s, %s, %p, 0x%08x, %p)\n", |
| debugstr_w(pchURL), debugstr_w(pchCookieName), debugstr_w(pchCookieData), |
| pcchCookieData, dwFlags, lpReserved); |
| |
| if (dwFlags) FIXME("flags 0x%08x not supported\n", dwFlags); |
| return InternetGetCookieW(pchURL, pchCookieName, pchCookieData, pcchCookieData); |
| } |
| |
| /*********************************************************************** |
| * InternetClearAllPerSiteCookieDecisions (WININET.@) |
| * |
| * Clears all per-site decisions about cookies. |
| * |
| * RETURNS |
| * TRUE on success |
| * FALSE on failure |
| * |
| */ |
| BOOL WINAPI InternetClearAllPerSiteCookieDecisions( VOID ) |
| { |
| FIXME("stub\n"); |
| return TRUE; |
| } |
| |
| /*********************************************************************** |
| * InternetEnumPerSiteCookieDecisionA (WININET.@) |
| * |
| * See InternetEnumPerSiteCookieDecisionW. |
| */ |
| BOOL WINAPI InternetEnumPerSiteCookieDecisionA( LPSTR pszSiteName, ULONG *pcSiteNameSize, |
| ULONG *pdwDecision, ULONG dwIndex ) |
| { |
| FIXME("(%s, %p, %p, 0x%08x) stub\n", |
| debugstr_a(pszSiteName), pcSiteNameSize, pdwDecision, dwIndex); |
| return FALSE; |
| } |
| |
| /*********************************************************************** |
| * InternetEnumPerSiteCookieDecisionW (WININET.@) |
| * |
| * Enumerates all per-site decisions about cookies. |
| * |
| * RETURNS |
| * TRUE on success |
| * FALSE on failure |
| * |
| */ |
| BOOL WINAPI InternetEnumPerSiteCookieDecisionW( LPWSTR pszSiteName, ULONG *pcSiteNameSize, |
| ULONG *pdwDecision, ULONG dwIndex ) |
| { |
| FIXME("(%s, %p, %p, 0x%08x) stub\n", |
| debugstr_w(pszSiteName), pcSiteNameSize, pdwDecision, dwIndex); |
| return FALSE; |
| } |
| |
| /*********************************************************************** |
| * InternetGetPerSiteCookieDecisionA (WININET.@) |
| */ |
| BOOL WINAPI InternetGetPerSiteCookieDecisionA( LPCSTR pwchHostName, ULONG *pResult ) |
| { |
| FIXME("(%s, %p) stub\n", debugstr_a(pwchHostName), pResult); |
| return FALSE; |
| } |
| |
| /*********************************************************************** |
| * InternetGetPerSiteCookieDecisionW (WININET.@) |
| */ |
| BOOL WINAPI InternetGetPerSiteCookieDecisionW( LPCWSTR pwchHostName, ULONG *pResult ) |
| { |
| FIXME("(%s, %p) stub\n", debugstr_w(pwchHostName), pResult); |
| return FALSE; |
| } |
| |
| /*********************************************************************** |
| * InternetSetPerSiteCookieDecisionA (WININET.@) |
| */ |
| BOOL WINAPI InternetSetPerSiteCookieDecisionA( LPCSTR pchHostName, DWORD dwDecision ) |
| { |
| FIXME("(%s, 0x%08x) stub\n", debugstr_a(pchHostName), dwDecision); |
| return FALSE; |
| } |
| |
| /*********************************************************************** |
| * InternetSetPerSiteCookieDecisionW (WININET.@) |
| */ |
| BOOL WINAPI InternetSetPerSiteCookieDecisionW( LPCWSTR pchHostName, DWORD dwDecision ) |
| { |
| FIXME("(%s, 0x%08x) stub\n", debugstr_w(pchHostName), dwDecision); |
| return FALSE; |
| } |
| |
| void free_cookie(void) |
| { |
| DeleteCriticalSection(&cookie_cs); |
| } |
