dlls/wininet/tests/urlcache.c

   1 /*
   2  * URL Cache Tests
   3  *
   4  * Copyright 2008 Robert Shearman for CodeWeavers
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include <stdarg.h>
  22 #include <stdio.h>
  23 #include <stdlib.h>
  24
  25 #include "windef.h"
  26 #include "winbase.h"
  27 #include "wininet.h"
  28
  29 #include "wine/test.h"
  30
  31 #define TEST_URL    "http://urlcachetest.winehq.org/index.html"
  32
  33 static BOOL (WINAPI *pDeleteUrlCacheEntryA)(LPCSTR);
  34 static BOOL (WINAPI *pUnlockUrlCacheEntryFileA)(LPCSTR,DWORD);
  35
  36 static char filenameA[MAX_PATH + 1];
  37
  38 static void check_cache_entry_infoA(const char *returnedfrom, LPINTERNET_CACHE_ENTRY_INFO lpCacheEntryInfo)
  39 {
  40     ok(lpCacheEntryInfo->dwStructSize == sizeof(*lpCacheEntryInfo), "%s: dwStructSize was %d\n", returnedfrom, lpCacheEntryInfo->dwStructSize);
  41     ok(!strcmp(lpCacheEntryInfo->lpszSourceUrlName, TEST_URL), "%s: lpszSourceUrlName should be %s instead of %s\n", returnedfrom, TEST_URL, lpCacheEntryInfo->lpszSourceUrlName);
  42     ok(!strcmp(lpCacheEntryInfo->lpszLocalFileName, filenameA), "%s: lpszLocalFileName should be %s instead of %s\n", returnedfrom, filenameA, lpCacheEntryInfo->lpszLocalFileName);
  43     ok(!strcmp(lpCacheEntryInfo->lpszFileExtension, "html"), "%s: lpszFileExtension should be html instead of %s\n", returnedfrom, lpCacheEntryInfo->lpszFileExtension);
  44 }
  45
  46 static void test_find_url_cache_entriesA(void)
  47 {
  48     BOOL ret;
  49     HANDLE hEnumHandle;
  50     BOOL found = FALSE;
  51     DWORD cbCacheEntryInfo;
  52     DWORD cbCacheEntryInfoSaved;
  53     LPINTERNET_CACHE_ENTRY_INFO lpCacheEntryInfo;
  54
  55     cbCacheEntryInfo = 0;
  56     hEnumHandle = FindFirstUrlCacheEntry(NULL, NULL, &cbCacheEntryInfo);
  57     ok(!hEnumHandle, "FindFirstUrlCacheEntry should have failed\n");
  58     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER, "FindFirstUrlCacheEntry should have set last error to ERROR_INSUFFICIENT_BUFFER instead of %d\n", GetLastError());
  59     lpCacheEntryInfo = HeapAlloc(GetProcessHeap(), 0, cbCacheEntryInfo * sizeof(char));
  60     cbCacheEntryInfoSaved = cbCacheEntryInfo;
  61     hEnumHandle = FindFirstUrlCacheEntry(NULL, lpCacheEntryInfo, &cbCacheEntryInfo);
  62     ok(hEnumHandle != NULL, "FindFirstUrlCacheEntry failed with error %d\n", GetLastError());
  63     while (TRUE)
  64     {
  65         if (!strcmp(lpCacheEntryInfo->lpszSourceUrlName, TEST_URL))
  66         {
  67             found = TRUE;
  68             break;
  69         }
  70         cbCacheEntryInfo = cbCacheEntryInfoSaved;
  71         ret = FindNextUrlCacheEntry(hEnumHandle, lpCacheEntryInfo, &cbCacheEntryInfo);
  72         if (!ret)
  73         {
  74             if (GetLastError() == ERROR_INSUFFICIENT_BUFFER)
  75             {
  76                 lpCacheEntryInfo = HeapReAlloc(GetProcessHeap(), 0, lpCacheEntryInfo, cbCacheEntryInfo);
  77                 cbCacheEntryInfoSaved = cbCacheEntryInfo;
  78                 ret = FindNextUrlCacheEntry(hEnumHandle, lpCacheEntryInfo, &cbCacheEntryInfo);
  79             }
  80         }
  81         ok(ret, "FindNextUrlCacheEntry failed with error %d\n", GetLastError());
  82         if (!ret)
  83             break;
  84     }
  85     ok(found, "committed url cache entry not found during enumeration\n");
  86
  87     ret = FindCloseUrlCache(hEnumHandle);
  88     ok(ret, "FindCloseUrlCache failed with error %d\n", GetLastError());
  89 }
  90
  91 static void test_GetUrlCacheEntryInfoExA(void)
  92 {
  93     BOOL ret;
  94     DWORD cbCacheEntryInfo;
  95     LPINTERNET_CACHE_ENTRY_INFO lpCacheEntryInfo;
  96
  97     ret = GetUrlCacheEntryInfoEx(NULL, NULL, NULL, NULL, NULL, NULL, 0);
  98     ok(!ret, "GetUrlCacheEntryInfoEx with NULL URL and NULL args should have failed\n");
  99     ok(GetLastError() == ERROR_INVALID_PARAMETER, "GetUrlCacheEntryInfoEx with NULL URL and NULL args should have set last error to ERROR_INVALID_PARAMETER instead of %d\n", GetLastError());
 100
 101     ret = GetUrlCacheEntryInfoEx(TEST_URL, NULL, NULL, NULL, NULL, NULL, 0);
 102     ok(ret, "GetUrlCacheEntryInfoEx with NULL args failed with error %d\n", GetLastError());
 103
 104     cbCacheEntryInfo = 0;
 105     ret = GetUrlCacheEntryInfoEx(TEST_URL, NULL, &cbCacheEntryInfo, NULL, NULL, NULL, 0);
 106     ok(!ret, "GetUrlCacheEntryInfoEx with zero-length buffer should fail\n");
 107     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER, "GetUrlCacheEntryInfoEx should have set last error to ERROR_INSUFFICIENT_BUFFER instead of %d\n", GetLastError());
 108
 109     lpCacheEntryInfo = HeapAlloc(GetProcessHeap(), 0, cbCacheEntryInfo);
 110     ret = GetUrlCacheEntryInfoEx(TEST_URL, lpCacheEntryInfo, &cbCacheEntryInfo, NULL, NULL, NULL, 0);
 111     ok(ret, "GetUrlCacheEntryInfoEx failed with error %d\n", GetLastError());
 112
 113     check_cache_entry_infoA("GetUrlCacheEntryInfoEx", lpCacheEntryInfo);
 114
 115     HeapFree(GetProcessHeap(), 0, lpCacheEntryInfo);
 116 }
 117
 118 static void test_urlcacheA(void)
 119 {
 120     BOOL ret;
 121     HANDLE hFile;
 122     DWORD written;
 123     BYTE zero_byte = 0;
 124     LPINTERNET_CACHE_ENTRY_INFO lpCacheEntryInfo;
 125     DWORD cbCacheEntryInfo;
 126     static const FILETIME filetime_zero;
 127
 128     ret = CreateUrlCacheEntry(TEST_URL, 0, "html", filenameA, 0);
 129     ok(ret, "CreateUrlCacheEntry failed with error %d\n", GetLastError());
 130
 131     hFile = CreateFileA(filenameA, GENERIC_WRITE, FILE_SHARE_READ|FILE_SHARE_WRITE,
 132                         NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
 133     ok(hFile != INVALID_HANDLE_VALUE, "CreateFileA failed with error %d\n", GetLastError());
 134
 135     ret = WriteFile(hFile, &zero_byte, sizeof(zero_byte), &written, NULL);
 136     ok(ret, "WriteFile failed with error %d\n", GetLastError());
 137
 138     CloseHandle(hFile);
 139
 140     ret = CommitUrlCacheEntry(TEST_URL, filenameA, filetime_zero, filetime_zero, NORMAL_CACHE_ENTRY, NULL, 0, "html", NULL);
 141     ok(ret, "CommitUrlCacheEntry failed with error %d\n", GetLastError());
 142
 143     cbCacheEntryInfo = 0;
 144     ret = RetrieveUrlCacheEntryFile(TEST_URL, NULL, &cbCacheEntryInfo, 0);
 145     ok(!ret, "RetrieveUrlCacheEntryFile should have failed\n");
 146     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER, "RetrieveUrlCacheEntryFile should have set last error to ERROR_INSUFFICIENT_BUFFER instead of %d\n", GetLastError());
 147
 148     lpCacheEntryInfo = HeapAlloc(GetProcessHeap(), 0, cbCacheEntryInfo);
 149     ret = RetrieveUrlCacheEntryFile(TEST_URL, lpCacheEntryInfo, &cbCacheEntryInfo, 0);
 150     ok(ret, "RetrieveUrlCacheEntryFile failed with error %d\n", GetLastError());
 151
 152     check_cache_entry_infoA("RetrieveUrlCacheEntryFile", lpCacheEntryInfo);
 153
 154     HeapFree(GetProcessHeap(), 0, lpCacheEntryInfo);
 155
 156     if (pUnlockUrlCacheEntryFileA)
 157     {
 158         ret = pUnlockUrlCacheEntryFileA(TEST_URL, 0);
 159         ok(ret, "UnlockUrlCacheEntryFileA failed with error %d\n", GetLastError());
 160     }
 161
 162     /* test Find*UrlCacheEntry functions */
 163     test_find_url_cache_entriesA();
 164
 165     test_GetUrlCacheEntryInfoExA();
 166
 167     if (pDeleteUrlCacheEntryA)
 168     {
 169         ret = pDeleteUrlCacheEntryA(TEST_URL);
 170         ok(ret, "DeleteUrlCacheEntryA failed with error %d\n", GetLastError());
 171     }
 172
 173     ret = DeleteFile(filenameA);
 174     todo_wine
 175     ok(!ret && GetLastError() == ERROR_FILE_NOT_FOUND, "local file should no longer exist\n");
 176 }
 177
 178 static void test_FindCloseUrlCache(void)
 179 {
 180     BOOL r;
 181     DWORD err;
 182     r = FindCloseUrlCache(NULL);
 183     err = GetLastError();
 184     ok(0 == r, "expected 0, got %d\n", r);
 185     ok(ERROR_INVALID_HANDLE == err, "expected %d, got %d\n", ERROR_INVALID_HANDLE, err);
 186 }
 187
 188 START_TEST(urlcache)
 189 {
 190     HMODULE hdll;
 191     hdll = GetModuleHandleA("wininet.dll");
 192     pDeleteUrlCacheEntryA = (void*)GetProcAddress(hdll, "DeleteUrlCacheEntryA");
 193     pUnlockUrlCacheEntryFileA = (void*)GetProcAddress(hdll, "UnlockUrlCacheEntryFileA");
 194     test_urlcacheA();
 195     test_FindCloseUrlCache();
 196 }