programs/fsutil/tests/fsutil.c

   1 /*
   2  * Copyright 2020 Arkadiusz Hiler for CodeWeavers
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  17  */
  18
  19 #include <windows.h>
  20 #include <stdio.h>
  21
  22 #include "wine/test.h"
  23
  24 static BOOL is_process_elevated(void)
  25 {
  26     HANDLE token;
  27     if (OpenProcessToken( GetCurrentProcess(), TOKEN_QUERY, &token ))
  28     {
  29         TOKEN_ELEVATION_TYPE type;
  30         DWORD size;
  31         BOOL ret;
  32
  33         ret = GetTokenInformation( token, TokenElevationType, &type, sizeof(type), &size );
  34         CloseHandle( token );
  35         return (ret && type == TokenElevationTypeFull);
  36     }
  37     return FALSE;
  38 }
  39
  40 static DWORD runcmd(const char* cmd)
  41 {
  42     STARTUPINFOA si = { sizeof(STARTUPINFOA) };
  43     PROCESS_INFORMATION pi;
  44     char* wcmd;
  45     DWORD rc;
  46
  47     /* Create a writable copy for CreateProcessA() */
  48     wcmd = HeapAlloc(GetProcessHeap(), 0, strlen(cmd) + 1);
  49     strcpy(wcmd, cmd);
  50
  51     rc = CreateProcessA(NULL, wcmd, NULL, NULL, FALSE, 0, NULL, NULL, &si, &pi);
  52     HeapFree(GetProcessHeap(), 0, wcmd);
  53
  54     if (!rc)
  55         return 260;
  56
  57     rc = WaitForSingleObject(pi.hProcess, 5000);
  58
  59     if (rc == WAIT_OBJECT_0)
  60         GetExitCodeProcess(pi.hProcess, &rc);
  61     else
  62         TerminateProcess(pi.hProcess, 1);
  63
  64     CloseHandle(pi.hThread);
  65     CloseHandle(pi.hProcess);
  66
  67     return rc;
  68 }
  69
  70 static void test_hardlink(void)
  71 {
  72     DWORD rc;
  73     BOOL boolrc;
  74     HANDLE hfile;
  75     BY_HANDLE_FILE_INFORMATION info;
  76
  77     hfile = CreateFileA("file", GENERIC_WRITE, 0, NULL, CREATE_ALWAYS,
  78                         FILE_ATTRIBUTE_NORMAL, NULL);
  79     ok(hfile != INVALID_HANDLE_VALUE, "failed to create a file\n");
  80     CloseHandle(hfile);
  81
  82     rc = runcmd("fsutil");
  83     if (rc == 1 && !is_process_elevated())
  84     {
  85         win_skip("Cannot run fsutil without elevated privileges on Windows <= 7\n");
  86         return;
  87     }
  88     ok(rc == 0, "failed to run fsutil\n");
  89
  90     rc = runcmd("fsutil hardlink");
  91     ok(rc == 0, "failed to run fsutil hardlink\n");
  92
  93     rc = runcmd("fsutil hardlink create link file");
  94     ok(rc == 0, "failed to create a hardlink\n");
  95
  96     hfile = CreateFileA("link", GENERIC_READ, 0, NULL, OPEN_EXISTING,
  97                         FILE_ATTRIBUTE_NORMAL, NULL);
  98     ok(hfile != INVALID_HANDLE_VALUE, "failed to open the hardlink\n");
  99     boolrc = GetFileInformationByHandle(hfile, &info);
 100     ok(boolrc, "failed to get info about hardlink, error %#x\n", GetLastError());
 101     CloseHandle(hfile);
 102
 103     ok(info.nNumberOfLinks == 2, "our link is not a hardlink\n");
 104
 105     rc = runcmd("fsutil hardlink create link file");
 106     ok(rc == 1, "fsutil didn't complain about already existing destination\n");
 107
 108     rc = runcmd("fsutil hardlink create newlink nonexistingfile");
 109     ok(rc == 1, "fsutil didn't complain about nonexisting source file\n");
 110
 111     boolrc = DeleteFileA("link");
 112     ok(boolrc, "failed to delete the hardlink, error %#x\n", GetLastError());
 113     boolrc = DeleteFileA("file");
 114     ok(boolrc, "failed to delete the file, error %#x\n", GetLastError());
 115 }
 116
 117 START_TEST(fsutil)
 118 {
 119     char tmpdir[MAX_PATH];
 120
 121     GetTempPathA(MAX_PATH, tmpdir);
 122     SetCurrentDirectoryA(tmpdir);
 123
 124     test_hardlink();
 125 }