xpcom/tests/windows/TestWinFileAttribs.cpp

   1 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* ***** BEGIN LICENSE BLOCK *****
   3  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
   4  *
   5  * The contents of this file are subject to the Mozilla Public License Version
   6  * 1.1 (the "License"); you may not use this file except in compliance with
   7  * the License. You may obtain a copy of the License at
   8  * http://www.mozilla.org/MPL/
   9  *
  10  * Software distributed under the License is distributed on an "AS IS" basis,
  11  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  12  * for the specific language governing rights and limitations under the
  13  * License.
  14  *
  15  * The Original Code is mozilla.org code.
  16  *
  17  * The Initial Developer of the Original Code is
  18  * Mozilla Foundation.
  19  * Portions created by the Initial Developer are Copyright (C) 2009
  20  * the Initial Developer. All Rights Reserved.
  21  *
  22  * Contributor(s):
  23  *   Jim Mathies <jmathies@mozilla.com>.
  24  *
  25  * Alternatively, the contents of this file may be used under the terms of
  26  * either the GNU General Public License Version 2 or later (the "GPL"), or
  27  * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  28  * in which case the provisions of the GPL or the LGPL are applicable instead
  29  * of those above. If you wish to allow use of your version of this file only
  30  * under the terms of either the GPL or the LGPL, and not to allow others to
  31  * use your version of this file under the terms of the MPL, indicate your
  32  * decision by deleting the provisions above and replace them with the notice
  33  * and other provisions required by the GPL or the LGPL. If you do not delete
  34  * the provisions above, a recipient may use your version of this file under
  35  * the terms of any one of the MPL, the GPL or the LGPL.
  36  *
  37  * ***** END LICENSE BLOCK ***** */
  38
  39 /*
  40  * Test:
  41  */
  42
  43 #include "../TestHarness.h"
  44 #include "nsEmbedString.h"
  45 #include "nsILocalFileWin.h"
  46 #include <windows.h>
  47
  48 #define BUFFSIZE 512
  49
  50 BOOL IsXPOrGreater()
  51 {
  52   OSVERSIONINFO osvi;
  53
  54   ZeroMemory(&osvi, sizeof(OSVERSIONINFO));
  55   osvi.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);
  56
  57   GetVersionEx(&osvi);
  58
  59   return ( (osvi.dwMajorVersion > 5) ||
  60      ( (osvi.dwMajorVersion == 5) && (osvi.dwMinorVersion >= 1) ));
  61 }
  62
  63 nsresult TestWinAttribs()
  64 {
  65     if (!IsXPOrGreater())
  66       return NS_OK;
  67
  68     printf("Is XP or greater, running tests...\n");
  69
  70     nsresult rv;
  71
  72     // File variables
  73     HANDLE hIndexed;
  74     nsCOMPtr<nsILocalFile> localFile;
  75     WCHAR filePath[MAX_PATH];
  76
  77     // Create and open temporary file
  78     hIndexed = CreateFileW(L".\\indexbit.txt",
  79                             GENERIC_READ | GENERIC_WRITE,
  80                             0,
  81                             NULL,
  82                             CREATE_ALWAYS,
  83                             FILE_ATTRIBUTE_NORMAL, //FILE_ATTRIBUTE_NOT_CONTENT_INDEXED, not supported by cf
  84                             NULL);
  85
  86     if(hIndexed == INVALID_HANDLE_VALUE)
  87     {
  88         fail("Test Win Attribs: Creating Test File");
  89         return NS_ERROR_FAILURE;
  90     }
  91
  92     CloseHandle(hIndexed);
  93
  94     GetFullPathNameW((LPCWSTR)L".\\indexbit.txt",
  95                         MAX_PATH, filePath, NULL);
  96
  97     //wprintf(filePath);
  98     //wprintf(L"\n");
  99
 100     rv = NS_NewLocalFile(nsEmbedString(filePath), PR_FALSE,
 101                          getter_AddRefs(localFile));
 102     if (NS_FAILED(rv))
 103     {
 104         fail("Test Win Attribs: Opening Test File");
 105         DeleteFileW(filePath);
 106         return rv;
 107     }
 108
 109     nsCOMPtr<nsILocalFileWin> localFileWin(do_QueryInterface(localFile));
 110
 111     DWORD dwAttrs = GetFileAttributesW(filePath);
 112     if (dwAttrs == INVALID_FILE_ATTRIBUTES)
 113     {
 114         fail("Test Win Attribs: GetFileAttributesW - couldn't find our temp file.");
 115         DeleteFileW(filePath);
 116         return NS_ERROR_FAILURE;
 117     }
 118
 119     dwAttrs |= FILE_ATTRIBUTE_NOT_CONTENT_INDEXED;
 120     SetFileAttributesW(filePath, dwAttrs);
 121
 122     PRUint32 attribs = 0;
 123     rv = localFileWin->GetFileAttributesWin(&attribs);
 124
 125     if (NS_FAILED(rv))
 126     {
 127         fail("Test Win Attribs: GetFileAttributesWin failed to GET attributes. (1)");
 128         DeleteFileW(filePath);
 129         return NS_ERROR_FAILURE;
 130     }
 131
 132     if (attribs & nsILocalFileWin::WFA_SEARCH_INDEXED)
 133     {
 134         fail("Test Win Attribs: GetFileAttributesWin attributed did not match. (2)");
 135         DeleteFileW(filePath);
 136         return NS_ERROR_FAILURE;
 137     }
 138
 139     dwAttrs &= ~FILE_ATTRIBUTE_NOT_CONTENT_INDEXED;
 140     SetFileAttributesW(filePath, dwAttrs);
 141
 142     rv = localFileWin->GetFileAttributesWin(&attribs);
 143
 144     if (NS_FAILED(rv))
 145     {
 146         fail("Test Win Attribs: GetFileAttributesWin failed to GET attributes. (3)");
 147         DeleteFileW(filePath);
 148         return NS_ERROR_FAILURE;
 149     }
 150
 151     if (!(attribs & nsILocalFileWin::WFA_SEARCH_INDEXED))
 152     {
 153         fail("Test Win Attribs: GetFileAttributesWin attributed did not match. (4)");
 154         DeleteFileW(filePath);
 155         return NS_ERROR_FAILURE;
 156     }
 157
 158     dwAttrs &= ~FILE_ATTRIBUTE_NOT_CONTENT_INDEXED;
 159     SetFileAttributesW(filePath, dwAttrs);
 160
 161     attribs = nsILocalFileWin::WFA_SEARCH_INDEXED;
 162     rv = localFileWin->SetFileAttributesWin(attribs);
 163
 164     dwAttrs = GetFileAttributesW(filePath);
 165
 166     if (NS_FAILED(rv))
 167     {
 168         fail("Test Win Attribs: GetFileAttributesWin failed to SET attributes. (5)");
 169         DeleteFileW(filePath);
 170         return NS_ERROR_FAILURE;
 171     }
 172
 173     if (dwAttrs & FILE_ATTRIBUTE_NOT_CONTENT_INDEXED)
 174     {
 175         fail("Test Win Attribs: SetFileAttributesWin attributed did not match. (6)");
 176         DeleteFileW(filePath);
 177         return NS_ERROR_FAILURE;
 178     }
 179
 180     dwAttrs |= FILE_ATTRIBUTE_NOT_CONTENT_INDEXED;
 181     SetFileAttributesW(filePath, dwAttrs);
 182
 183     attribs = 0;
 184     rv = localFileWin->SetFileAttributesWin(attribs);
 185
 186     dwAttrs = GetFileAttributesW(filePath);
 187
 188     if (NS_FAILED(rv))
 189     {
 190         fail("Test Win Attribs: GetFileAttributesWin failed to SET attributes. (7)");
 191         DeleteFileW(filePath);
 192         return NS_ERROR_FAILURE;
 193     }
 194
 195     if (!(dwAttrs & FILE_ATTRIBUTE_NOT_CONTENT_INDEXED))
 196     {
 197         fail("Test Win Attribs: SetFileAttributesWin attributed did not match. (8)");
 198         DeleteFileW(filePath);
 199         return NS_ERROR_FAILURE;
 200     }
 201
 202     DeleteFileW(filePath);
 203
 204     passed("Test Win Attribs: passed tests.");
 205
 206     return NS_OK;
 207 }
 208
 209 int main(int argc, char** argv)
 210 {
 211     ScopedXPCOM xpcom("WinFileAttributes");
 212     if (xpcom.failed())
 213         return 1;
 214
 215     int rv = 0;
 216
 217     if(NS_FAILED(TestWinAttribs()))
 218         rv = 1;
 219
 220     return rv;
 221
 222 }
 223