nativelibc/rmdir.c

/*	A part of the Native C Library for Windows NT
	Copyright 2007-2015 PC GO Ld.
	Copyright 2015-2017 Rivoreo

	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 2 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.
*/

#include <windows.h>
#include <nt.h>
#include <errno.h>
#include <string.h>
#include <pathname.h>
#include <ntstatus.h>

//#include <stdio.h>

int rmdir(const char *pathname) {
	if(!pathname) {
		errno = EFAULT;
		return -1;
	}
	if(!*pathname) {
		errno = ENOENT;
		return -1;
	}

	if(*pathname != '/') {
		if(strcmp(pathname, ".") == 0 || strcmp(pathname, "./") == 0) {
			errno = EINVAL;
			return -1;
		}
		errno = ENOSYS;
		return -1;
	}

	if(!pathname[1]) {
		// Want to remove / ??
		errno = EBUSY;
		return -1;
	}

	UNICODE_STRING ntpathname;
	if(!RtlCreateUnicodeStringFromAsciiz(&ntpathname, pathname)) {
		errno = ENOMEM;
		return -1;
	}
	if(ntpathname.Length > sizeof(wchar_t) && ntpathname.Buffer[ntpathname.Length / sizeof(wchar_t) - 1] == L'/') {
		ntpathname.Buffer[ntpathname.Length / sizeof(wchar_t) - 1] = 0;
		ntpathname.Length -= sizeof(wchar_t);
		ntpathname.MaximumLength -= sizeof(wchar_t);
	}
	PATHNAME_UNIX2NT_UTF16_STRUCT(ntpathname);
	void *handle;
	OBJECT_ATTRIBUTES object_attrib = { sizeof(OBJECT_ATTRIBUTES), NULL, &ntpathname, 0, NULL, NULL };

	long int status = NtOpenSymbolicLinkObject(&handle, 0, &object_attrib);
	//printf("nativelibc debug: rmdir: check is sym link: status = 0x%lx\n", status);
	if(status != STATUS_OBJECT_TYPE_MISMATCH) {
		// The rmdir will failed on a symbolic link even the pathname contents a '/' at end
		RtlFreeUnicodeString(&ntpathname);
/*
		if(status < 0) __set_errno_from_ntstatus(status);
		else {
			NtClose(handle);
			errno = ENOTDIR;
		}
*/
		if(status >= 0 || status == STATUS_ACCESS_DENIED) {
			if(status >= 0) NtClose(handle);
			errno = ENOTDIR;
		} else __set_errno_from_ntstatus(status);
		return -1;
	}

	status = NtOpenDirectoryObject(&handle, DIRECTORY_QUERY | DELETE, &object_attrib);
	//printf("nativelibc debug: rmdir: check is dir object: status = 0x%lx\n", status);
	if(status != STATUS_OBJECT_TYPE_MISMATCH) {
		unsigned long int context = 0;

		RtlFreeUnicodeString(&ntpathname);
		if(status < 0) {
			__set_errno_from_ntstatus(status);
			return -1;
		}

		status = NtQueryDirectoryObject(handle, NULL, 0, 0, 1, &context, NULL);
		//printf("nativelibc debug: rmdir: check dir object contents: status = 0x%lx\n", status);
		if(status != STATUS_NO_MORE_ENTRIES) {
			NtClose(handle);
			if(status < 0) __set_errno_from_ntstatus(status);
			else errno = ENOTEMPTY;
			return -1;
		}

		OBJECT_BASIC_INFORMATION obi;
		status = NtQueryObject(handle, ObjectBasicInformation, &obi, sizeof obi, NULL);
		//printf("nativelibc debug: rmdir: check other opens: NtQueryObject status = 0x%lx\n", status);
		if(status < 0) {
			NtClose(handle);
			__set_errno_from_ntstatus(status);
			return -1;
		}
		if(obi.HandleCount > 1) {
			NtClose(handle);
			errno = EBUSY;
			return -1;
		}

		status = NtMakeTemporaryObject(handle);
		//printf("nativelibc debug: rmdir: remove dir object: status = 0x%lx\n", status);
		NtClose(handle);
		return __set_errno_from_ntstatus(status) ? -1 : 0;
	}

	IO_STATUS_BLOCK io_status;
	unsigned long int file_share = FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE;
	unsigned long int options = FILE_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT | FILE_OPEN_FOR_BACKUP_INTENT;
	status = NtOpenFile(&handle, FILE_LIST_DIRECTORY | DELETE | SYNCHRONIZE, &object_attrib, &io_status, file_share, options);
	//printf("nativelibc debug: rmdir: check is dir file: status = 0x%lx\n", status);
	if(status != STATUS_OBJECT_TYPE_MISMATCH) {
		RtlFreeUnicodeString(&ntpathname);

		if(status < 0) {
			__set_errno_from_ntstatus(status);
			return -1;
		}

		size_t fni_size = sizeof(FILE_NAMES_INFORMATION) + 2 * sizeof(wchar_t);
		FILE_NAMES_INFORMATION *fni = malloc(fni_size);
		if(!fni) {
			NtClose(handle);
			errno = ENOMEM;
			return -1;
		}
		while(1) {
			status = NtQueryDirectoryFile(handle, NULL, NULL, NULL, &io_status, fni, fni_size, FileNamesInformation, 0, NULL, 0);
			//printf("nativelibc debug: rmdir: check dir file contents: status = 0x%lx\n", status);
			if(status == STATUS_NO_MORE_FILES) break;
			if(status == STATUS_BUFFER_OVERFLOW) goto enotempty;
			if(status < 0) {
				free(fni);
				NtClose(handle);
				__set_errno_from_ntstatus(status);
				return -1;
			}
			if(fni->FileNameLength > 2 * sizeof(wchar_t) || fni->FileName[0] != L'.' ||
			(fni->FileNameLength > sizeof(wchar_t) && fni->FileName[1] != L'.')) {
enotempty:
				free(fni);
				NtClose(handle);
				errno = ENOTEMPTY;
				return -1;
			}
		}
		free(fni);

		FILE_DISPOSITION_INFORMATION fdi;
		fdi.DeleteFile = 1;
		status = NtSetInformationFile(handle, &io_status, &fdi, sizeof fdi, FileDispositionInformation);
		//printf("nativelibc debug: rmdir: remove dir file: status = 0x%lx\n", status);
		if(__set_errno_from_ntstatus(NtClose(handle))) return -1;
		return __set_errno_from_ntstatus(status) ? -1 : 0;
	}

	// The pathname exists but is not a directory
	RtlFreeUnicodeString(&ntpathname);
	errno = ENOTDIR;
	return -1;
}