wibo/dll/kernel32/internal.h

#pragma once

#include "common.h"
#include "handles.h"
#include "mimalloc.h"
#include "types.h"

#include <condition_variable>
#include <pthread.h>

namespace kernel32 {

struct FsObject : ObjectBase {
	std::mutex m;
	int fd = -1;
	std::filesystem::path canonicalPath;
	uint32_t shareAccess = FILE_SHARE_READ | FILE_SHARE_WRITE;
	bool deletePending = false;
	bool closeOnDestroy = true;

	~FsObject() override;
	[[nodiscard]] bool valid() const { return fd >= 0; }

  protected:
	explicit FsObject(ObjectType type, int fd) : ObjectBase(type), fd(fd) {}
};

struct FileObject : FsObject {
	off_t filePos = 0;
	bool overlapped = false;
	bool appendOnly = false;
	bool isPipe = false;
	// Used to notify overlapped operations without an event handle
	std::condition_variable overlappedCv;

	explicit FileObject(int fd) : FileObject(ObjectType::File, fd) {}
	FileObject(ObjectType type, int fd) : FsObject(type, fd) {
		flags |= Of_File;
		if (fd >= 0) {
			off_t pos = lseek(fd, 0, SEEK_CUR);
			if (pos == -1 && errno == ESPIPE) {
				isPipe = true;
			} else if (pos >= 0) {
				filePos = pos;
			}
		}
	}

	~FileObject() override = default;
};

struct DirectoryObject final : FsObject {
	static constexpr ObjectType kType = ObjectType::Directory;

	uint64_t enumCookie = 0;

	explicit DirectoryObject(int dirfd) : FsObject(kType, dirfd) {}
};

struct ProcessObject final : WaitableObject {
	static constexpr ObjectType kType = ObjectType::Process;

	pid_t pid;
	int pidfd;
	DWORD exitCode = STILL_ACTIVE;
	bool forcedExitCode = false;

	explicit ProcessObject(pid_t pid, int pidfd) : WaitableObject(kType), pid(pid), pidfd(pidfd) {}

	~ProcessObject() override {
		if (pidfd != -1) {
			close(pidfd);
			pidfd = -1;
		}
	}
};

struct ThreadObject final : WaitableObject {
	static constexpr ObjectType kType = ObjectType::Thread;

	pthread_t thread;
	DWORD exitCode = STILL_ACTIVE;
	unsigned int suspendCount = 0;
	TEB *tib = nullptr;

	explicit ThreadObject(pthread_t thread) : WaitableObject(kType), thread(thread) {}

	~ThreadObject() override {
		// Threads are detached at creation; we can safely drop
		if (tib) {
			wibo::destroyTib(tib);
			tib = nullptr;
		}
	}
};

struct MutexObject final : WaitableObject {
	static constexpr ObjectType kType = ObjectType::Mutex;

	bool ownerValid = false;
	pthread_t owner{};
	unsigned int recursionCount = 0;
	bool abandoned = false; // Owner exited without releasing

	MutexObject() : WaitableObject(kType) { signaled = true; }
};

struct EventObject final : WaitableObject {
	static constexpr ObjectType kType = ObjectType::Event;

	bool manualReset = false;

	explicit EventObject(bool manual) : WaitableObject(kType), manualReset(manual) {}

	void set() {
		bool resetAll = false;
		{
			std::lock_guard lk(m);
			signaled = true;
			resetAll = manualReset;
		}
		if (resetAll) {
			cv.notify_all();
		} else {
			cv.notify_one();
		}
		notifyWaiters(false);
	}

	void reset() {
		std::lock_guard lk(m);
		signaled = false;
	}
};

struct SemaphoreObject final : WaitableObject {
	static constexpr ObjectType kType = ObjectType::Semaphore;

	LONG count = 0;
	LONG maxCount = 0;

	SemaphoreObject(LONG initial, LONG maximum) : WaitableObject(kType), count(initial), maxCount(maximum) {}
};

struct HeapObject : public ObjectBase {
	static constexpr ObjectType kType = ObjectType::Heap;

	mi_heap_t *heap;
	const pthread_t owner;
	DWORD createFlags = 0;
	SIZE_T initialSize = 0;
	SIZE_T maximumSize = 0;
	DWORD compatibility = 0;
	bool isProcessHeap = false;

	explicit HeapObject(mi_heap_t *heap) : ObjectBase(kType), heap(heap), owner(pthread_self()) {}
	~HeapObject() override;

	[[nodiscard]] inline bool isOwner() const { return pthread_equal(owner, pthread_self()); }
	[[nodiscard]] inline bool canAccess() const { return isProcessHeap || (isOwner() && heap != nullptr); }
};

inline constexpr uintptr_t kPseudoCurrentProcessHandleValue = static_cast<uintptr_t>(-1);
inline constexpr uintptr_t kPseudoCurrentThreadHandleValue = static_cast<uintptr_t>(-2);

inline bool isPseudoCurrentProcessHandle(HANDLE h) {
	uintptr_t rawHandle = reinterpret_cast<uintptr_t>(h);
	return rawHandle == kPseudoCurrentProcessHandleValue;
}

inline bool isPseudoCurrentThreadHandle(HANDLE h) {
	uintptr_t rawHandle = reinterpret_cast<uintptr_t>(h);
	return rawHandle == kPseudoCurrentThreadHandleValue;
}

void tryMarkExecutable(void *mem);
void setLastErrorFromErrno();
[[noreturn]] void exitInternal(DWORD exitCode);

DWORD getLastError();
void setLastError(DWORD error);

} // namespace kernel32

namespace detail {

template <> constexpr bool typeMatches<kernel32::FileObject>(const ObjectBase *o) noexcept {
	return o && (o->flags & Of_File);
}

} // namespace detail