TestSetFileAttributes.c

 
#include <stdio.h>
#include <winpr/crt.h>
#include <winpr/file.h>
#include <winpr/path.h>
#include <winpr/handle.h>
#include <winpr/windows.h>
#include <winpr/sysinfo.h>
 
static const DWORD allflags[] = {
  0,
  FILE_ATTRIBUTE_READONLY,
  FILE_ATTRIBUTE_HIDDEN,
  FILE_ATTRIBUTE_SYSTEM,
  FILE_ATTRIBUTE_DIRECTORY,
  FILE_ATTRIBUTE_ARCHIVE,
  FILE_ATTRIBUTE_DEVICE,
  FILE_ATTRIBUTE_NORMAL,
  FILE_ATTRIBUTE_TEMPORARY,
  FILE_ATTRIBUTE_SPARSE_FILE,
  FILE_ATTRIBUTE_REPARSE_POINT,
  FILE_ATTRIBUTE_COMPRESSED,
  FILE_ATTRIBUTE_OFFLINE,
  FILE_ATTRIBUTE_NOT_CONTENT_INDEXED,
  FILE_ATTRIBUTE_ENCRYPTED,
  FILE_ATTRIBUTE_VIRTUAL,
  FILE_ATTRIBUTE_READONLY | FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_SYSTEM,
  FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_ARCHIVE | FILE_ATTRIBUTE_DEVICE |
      FILE_ATTRIBUTE_NORMAL,
  FILE_ATTRIBUTE_TEMPORARY | FILE_ATTRIBUTE_SPARSE_FILE | FILE_ATTRIBUTE_REPARSE_POINT |
      FILE_ATTRIBUTE_COMPRESSED | FILE_ATTRIBUTE_OFFLINE,
  FILE_ATTRIBUTE_NOT_CONTENT_INDEXED | FILE_ATTRIBUTE_ENCRYPTED | FILE_ATTRIBUTE_VIRTUAL
};
 
static BOOL test_SetFileAttributesA(void)
{
  BOOL rc = FALSE;
  HANDLE handle = NULL;
  const DWORD flags[] = { 0, FILE_ATTRIBUTE_READONLY };
  char* name = GetKnownSubPath(KNOWN_PATH_TEMP, "afsklhjwe4oq5iu432oijrlkejadlkhjaklhfdkahfd");
  if (!name)
    goto fail;
 
  for (size_t x = 0; x < ARRAYSIZE(allflags); x++)
  {
    const DWORD flag = allflags[x];
    const BOOL brc = SetFileAttributesA(NULL, flag);
    if (brc)
      goto fail;
 
    const BOOL crc = SetFileAttributesA(name, flag);
    if (crc)
      goto fail;
  }
 
  handle = CreateFileA(name, GENERIC_READ | GENERIC_WRITE, 0, NULL, CREATE_NEW,
                       FILE_ATTRIBUTE_NORMAL, NULL);
  if (handle == INVALID_HANDLE_VALUE)
    goto fail;
  (void)CloseHandle(handle);
 
  for (size_t x = 0; x < ARRAYSIZE(flags); x++)
  {
    DWORD attr = 0;
    const DWORD flag = flags[x];
    const BOOL brc = SetFileAttributesA(name, flag);
    if (!brc)
      goto fail;
 
    attr = GetFileAttributesA(name);
    if (flag != 0)
    {
      if ((attr & flag) == 0)
        goto fail;
    }
  }
 
  rc = TRUE;
 
fail:
  DeleteFileA(name);
  free(name);
  return rc;
}
 
static BOOL test_SetFileAttributesW(void)
{
  BOOL rc = FALSE;
  WCHAR* name = NULL;
  HANDLE handle = NULL;
  const DWORD flags[] = { 0, FILE_ATTRIBUTE_READONLY };
  char* base = GetKnownSubPath(KNOWN_PATH_TEMP, "afsklhjwe4oq5iu432oijrlkejadlkhjaklhfdkahfd");
  if (!base)
    goto fail;
 
  name = ConvertUtf8ToWCharAlloc(base, NULL);
  if (!name)
    goto fail;
 
  for (size_t x = 0; x < ARRAYSIZE(allflags); x++)
  {
    const DWORD flag = allflags[x];
    const BOOL brc = SetFileAttributesW(NULL, flag);
    if (brc)
      goto fail;
 
    const BOOL crc = SetFileAttributesW(name, flag);
    if (crc)
      goto fail;
  }
 
  handle = CreateFileW(name, GENERIC_READ | GENERIC_WRITE, 0, NULL, CREATE_NEW,
                       FILE_ATTRIBUTE_NORMAL, NULL);
  if (handle == INVALID_HANDLE_VALUE)
    goto fail;
  (void)CloseHandle(handle);
 
  for (size_t x = 0; x < ARRAYSIZE(flags); x++)
  {
    DWORD attr = 0;
    const DWORD flag = flags[x];
    const BOOL brc = SetFileAttributesW(name, flag);
    if (!brc)
      goto fail;
 
    attr = GetFileAttributesW(name);
    if (flag != 0)
    {
      if ((attr & flag) == 0)
        goto fail;
    }
  }
 
  rc = TRUE;
fail:
  DeleteFileW(name);
  free(name);
  free(base);
  return rc;
}
 
int TestSetFileAttributes(int argc, char* argv[])
{
  WINPR_UNUSED(argc);
  WINPR_UNUSED(argv);
 
  if (!test_SetFileAttributesA())
    return -1;
  if (!test_SetFileAttributesW())
    return -1;
  return 0;
}