mirror of
https://github.com/aaru-dps/fstester.git
synced 2025-12-16 19:24:39 +00:00
Split win32 code.
This commit is contained in:
221
setter/src/win32/links.c
Normal file
221
setter/src/win32/links.c
Normal file
@@ -0,0 +1,221 @@
|
||||
/****************************************************************************
|
||||
Aaru Data Preservation Suite
|
||||
-----------------------------------------------------------------------------
|
||||
|
||||
Filename : win32.c
|
||||
Author(s) : Natalia Portillo
|
||||
|
||||
--[ Description ] -----------------------------------------------------------
|
||||
|
||||
Contains 32-bit and 64-bit Windows code
|
||||
|
||||
--[ License ] ---------------------------------------------------------------
|
||||
This program is free software: you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as
|
||||
published by the Free Software Foundation, either version 3 of the
|
||||
License, or (at your option) any later version.
|
||||
|
||||
This program 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 General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
-----------------------------------------------------------------------------
|
||||
Copyright (C) 2011-2021 Natalia Portillo
|
||||
*****************************************************************************/
|
||||
|
||||
// ReSharper disable CppJoinDeclarationAndAssignment
|
||||
// ReSharper disable CppDeprecatedEntity
|
||||
#if defined(__WINDOWS__) || defined(__TOS_WIN__) || defined(__WIN32__) || defined(_WIN64) || defined(_WIN32) || \
|
||||
defined(__NT__)
|
||||
|
||||
#define _CRT_SECURE_NO_WARNINGS 1
|
||||
|
||||
#include <windows.h>
|
||||
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
|
||||
#include "win32.h"
|
||||
|
||||
#include "links.h"
|
||||
|
||||
#include "../include/consts.h"
|
||||
#include "../include/defs.h"
|
||||
|
||||
static DWORD dwMaxNameSize = MAX_PATH + 1;
|
||||
static DWORD dwFilePermissions = GENERIC_READ | GENERIC_WRITE;
|
||||
static DWORD oldVersion;
|
||||
static HINSTANCE kernel32;
|
||||
|
||||
void Links(const char* path)
|
||||
{
|
||||
BOOL ret;
|
||||
DWORD error;
|
||||
WIN_OSVERSIONINFO verInfo;
|
||||
void* func;
|
||||
DWORD dwNumberOfBytesWritten;
|
||||
DWORD rc, wRc, cRc, lRc;
|
||||
char message[300];
|
||||
HANDLE h;
|
||||
LPSTR lpRootPathName;
|
||||
size_t pathSize = strlen(path);
|
||||
|
||||
if(WinGetVersionExA)
|
||||
{
|
||||
verInfo.dwOSVersionInfoSize = sizeof(WIN_OSVERSIONINFO);
|
||||
ret = WinGetVersionExA(&verInfo);
|
||||
|
||||
if(!ret)
|
||||
{
|
||||
error = GetLastError();
|
||||
printf("Error %lu querying Windows version.\n", error);
|
||||
return;
|
||||
}
|
||||
}
|
||||
else if(oldVersion == 0)
|
||||
verInfo.dwPlatformId = VER_PLATFORM_WIN32_NT;
|
||||
|
||||
if(verInfo.dwPlatformId != VER_PLATFORM_WIN32_NT)
|
||||
{
|
||||
// Not supported on Windows 9x
|
||||
return;
|
||||
}
|
||||
|
||||
lpRootPathName = malloc(dwMaxNameSize);
|
||||
|
||||
if(!lpRootPathName)
|
||||
{
|
||||
printf("Could not allocate memory.\n");
|
||||
return;
|
||||
}
|
||||
|
||||
memset(lpRootPathName, 0x00, MAX_PATH);
|
||||
strncpy(lpRootPathName, path, MAX_PATH);
|
||||
|
||||
if(path[pathSize - 1] != '\\') { lpRootPathName[pathSize] = '\\'; }
|
||||
|
||||
ret = SetCurrentDirectoryA(lpRootPathName);
|
||||
|
||||
if(!ret)
|
||||
{
|
||||
error = GetLastError();
|
||||
printf("Error %lu changing to specified path.\n", error);
|
||||
return;
|
||||
}
|
||||
|
||||
ret = CreateDirectoryA("LINKS", NULL);
|
||||
|
||||
if(!ret)
|
||||
{
|
||||
error = GetLastError();
|
||||
printf("Error %lu creating working directory.\n", error);
|
||||
return;
|
||||
}
|
||||
|
||||
ret = SetCurrentDirectoryA("LINKS");
|
||||
|
||||
if(!ret)
|
||||
{
|
||||
error = GetLastError();
|
||||
printf("Error %lu changing to working directory.\n", error);
|
||||
return;
|
||||
}
|
||||
|
||||
kernel32 = LoadLibraryA("kernel32.dll");
|
||||
|
||||
if(!kernel32)
|
||||
{
|
||||
error = GetLastError();
|
||||
printf("Error %lu loading KERNEL32.DLL.\n", error);
|
||||
return;
|
||||
}
|
||||
|
||||
func = GetProcAddress(kernel32, "CreateSymbolicLinkA");
|
||||
|
||||
if(!func)
|
||||
{
|
||||
error = GetLastError();
|
||||
printf("Error %lu finding CreateSymbolicLinkA.\n", error);
|
||||
}
|
||||
else
|
||||
{
|
||||
WinNtCreateSymbolicLinkA = func;
|
||||
printf("Creating symbolic links.\n");
|
||||
|
||||
h = CreateFileA("TARGET",
|
||||
dwFilePermissions,
|
||||
FILE_SHARE_READ | FILE_SHARE_WRITE,
|
||||
NULL,
|
||||
CREATE_ALWAYS,
|
||||
FILE_ATTRIBUTE_NORMAL,
|
||||
NULL);
|
||||
rc = 0;
|
||||
wRc = 0;
|
||||
cRc = 0;
|
||||
lRc = 0;
|
||||
if(h == INVALID_HANDLE_VALUE) rc = GetLastError();
|
||||
else
|
||||
{
|
||||
memset(message, 0, 300);
|
||||
sprintf(message, "This file is the target of a symbolic link.\n");
|
||||
ret = WriteFile(h, message, strlen(message), &dwNumberOfBytesWritten, NULL);
|
||||
if(!ret) wRc = GetLastError();
|
||||
|
||||
ret = CloseHandle(h);
|
||||
if(!ret) cRc = GetLastError();
|
||||
|
||||
ret = WinNtCreateSymbolicLinkA("SYMLINK", "TARGET", 0);
|
||||
if(!ret) lRc = GetLastError();
|
||||
}
|
||||
|
||||
printf("\tSymbolic link, rc = 0x%08lx, wRc = %lu, cRc = %lu, lRc = %lu\n", rc, wRc, cRc, lRc);
|
||||
}
|
||||
|
||||
func = GetProcAddress(kernel32, "CreateHardLinkA");
|
||||
|
||||
if(!func)
|
||||
{
|
||||
error = GetLastError();
|
||||
printf("Error %lu finding CreateHardLinkA.\n", error);
|
||||
}
|
||||
else
|
||||
{
|
||||
WinNtCreateHardLinkA = func;
|
||||
printf("Creating hard links.\n");
|
||||
|
||||
h = CreateFileA("HARDTRGT",
|
||||
dwFilePermissions,
|
||||
FILE_SHARE_READ | FILE_SHARE_WRITE,
|
||||
NULL,
|
||||
CREATE_ALWAYS,
|
||||
FILE_ATTRIBUTE_NORMAL,
|
||||
NULL);
|
||||
rc = 0;
|
||||
wRc = 0;
|
||||
cRc = 0;
|
||||
lRc = 0;
|
||||
if(h == INVALID_HANDLE_VALUE) rc = GetLastError();
|
||||
else
|
||||
{
|
||||
memset(message, 0, 300);
|
||||
sprintf(message, "This file is part of a hard link.\n");
|
||||
ret = WriteFile(h, message, strlen(message), &dwNumberOfBytesWritten, NULL);
|
||||
if(!ret) wRc = GetLastError();
|
||||
|
||||
ret = CloseHandle(h);
|
||||
if(!ret) cRc = GetLastError();
|
||||
|
||||
ret = WinNtCreateHardLinkA("HARDLINK", "HARDTRGT", NULL);
|
||||
if(!ret) lRc = GetLastError();
|
||||
}
|
||||
|
||||
printf("\tHard link, rc = 0x%08lx, wRc = %lu, cRc = %lu, lRc = %lu\n", rc, wRc, cRc, lRc);
|
||||
}
|
||||
}
|
||||
|
||||
#endif
|
||||
Reference in New Issue
Block a user