Everything needed to run simple Rust programs (#40)

* Everything needed to run simple Rust programs

* Add IsDBCSLeadByte implementation

* Address PR comments

.clang-format

@@ -0,0 +1,6 @@
+---
+BasedOnStyle: LLVM
+IndentWidth: 4
+TabWidth: 4
+UseTab: Always
+ColumnLimit: 120

.clang-tidy

@@ -0,0 +1,146 @@
+---
+Checks: '-*,
+bugprone-argument-comment,
+bugprone-assert-side-effect,
+bugprone-bad-signal-to-kill-thread,
+bugprone-branch-clone,
+bugprone-copy-constructor-init,
+bugprone-dangling-handle,
+bugprone-dynamic-static-initializers,
+bugprone-fold-init-type,
+bugprone-forward-declaration-namespace,
+bugprone-forwarding-reference-overload,
+bugprone-inaccurate-erase,
+bugprone-incorrect-roundings,
+bugprone-integer-division,
+bugprone-lambda-function-name,
+bugprone-macro-parentheses,
+bugprone-macro-repeated-side-effects,
+bugprone-misplaced-operator-in-strlen-in-alloc,
+bugprone-misplaced-pointer-arithmetic-in-alloc,
+bugprone-misplaced-widening-cast,
+bugprone-move-forwarding-reference,
+bugprone-multiple-statement-macro,
+bugprone-no-escape,
+bugprone-not-null-terminated-result,
+bugprone-parent-virtual-call,
+bugprone-posix-return,
+bugprone-sizeof-container,
+bugprone-sizeof-expression,
+bugprone-spuriously-wake-up-functions,
+bugprone-string-constructor,
+bugprone-string-integer-assignment,
+bugprone-string-literal-with-embedded-nul,
+bugprone-suspicious-enum-usage,
+bugprone-suspicious-include,
+bugprone-suspicious-memset-usage,
+bugprone-suspicious-missing-comma,
+bugprone-suspicious-semicolon,
+bugprone-suspicious-string-compare,
+bugprone-suspicious-memory-comparison,
+bugprone-suspicious-realloc-usage,
+bugprone-swapped-arguments,
+bugprone-terminating-continue,
+bugprone-throw-keyword-missing,
+bugprone-too-small-loop-variable,
+bugprone-undefined-memory-manipulation,
+bugprone-undelegated-constructor,
+bugprone-unhandled-self-assignment,
+bugprone-unused-raii,
+bugprone-unused-return-value,
+bugprone-use-after-move,
+bugprone-virtual-near-miss,
+cert-dcl21-cpp,
+cert-dcl58-cpp,
+cert-err34-c,
+cert-err52-cpp,
+cert-err60-cpp,
+cert-flp30-c,
+cert-msc50-cpp,
+cert-msc51-cpp,
+cert-str34-c,
+cppcoreguidelines-interfaces-global-init,
+cppcoreguidelines-narrowing-conversions,
+cppcoreguidelines-pro-type-member-init,
+cppcoreguidelines-pro-type-static-cast-downcast,
+cppcoreguidelines-slicing,
+google-default-arguments,
+google-explicit-constructor,
+google-runtime-operator,
+hicpp-exception-baseclass,
+hicpp-multiway-paths-covered,
+misc-misplaced-const,
+misc-new-delete-overloads,
+misc-no-recursion,
+misc-non-copyable-objects,
+misc-throw-by-value-catch-by-reference,
+misc-unconventional-assign-operator,
+misc-uniqueptr-reset-release,
+modernize-avoid-bind,
+modernize-concat-nested-namespaces,
+modernize-deprecated-headers,
+modernize-deprecated-ios-base-aliases,
+modernize-loop-convert,
+modernize-make-shared,
+modernize-make-unique,
+modernize-pass-by-value,
+modernize-raw-string-literal,
+modernize-redundant-void-arg,
+modernize-replace-auto-ptr,
+modernize-replace-disallow-copy-and-assign-macro,
+modernize-replace-random-shuffle,
+modernize-return-braced-init-list,
+modernize-shrink-to-fit,
+modernize-unary-static-assert,
+modernize-use-auto,
+modernize-use-bool-literals,
+modernize-use-emplace,
+modernize-use-equals-default,
+modernize-use-equals-delete,
+modernize-use-nodiscard,
+modernize-use-noexcept,
+modernize-use-nullptr,
+modernize-use-override,
+modernize-use-transparent-functors,
+modernize-use-uncaught-exceptions,
+mpi-buffer-deref,
+mpi-type-mismatch,
+openmp-use-default-none,
+performance-faster-string-find,
+performance-for-range-copy,
+performance-implicit-conversion-in-loop,
+performance-inefficient-algorithm,
+performance-inefficient-string-concatenation,
+performance-inefficient-vector-operation,
+performance-move-const-arg,
+performance-move-constructor-init,
+performance-no-automatic-move,
+performance-noexcept-move-constructor,
+performance-trivially-destructible,
+performance-type-promotion-in-math-fn,
+performance-unnecessary-copy-initialization,
+performance-unnecessary-value-param,
+portability-simd-intrinsics,
+readability-avoid-const-params-in-decls,
+readability-const-return-type,
+readability-container-size-empty,
+readability-convert-member-functions-to-static,
+readability-delete-null-pointer,
+readability-deleted-default,
+readability-inconsistent-declaration-parameter-name,
+readability-make-member-function-const,
+readability-misleading-indentation,
+readability-misplaced-array-index,
+readability-non-const-parameter,
+readability-redundant-control-flow,
+readability-redundant-declaration,
+readability-redundant-function-ptr-dereference,
+readability-redundant-smartptr-get,
+readability-redundant-string-cstr,
+readability-redundant-string-init,
+readability-simplify-subscript-expr,
+readability-static-accessed-through-instance,
+readability-static-definition-in-anonymous-namespace,
+readability-string-compare,
+readability-uniqueptr-delete-release,
+readability-use-anyofallof'

CMakeLists.txt

@@ -16,10 +16,14 @@ set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -m32 -Wall")
 include_directories(.)
 add_executable(wibo
     dll/advapi32.cpp
+    dll/bcrypt.cpp
+    dll/crt.cpp
     dll/kernel32.cpp
     dll/lmgr.cpp
+    dll/ntdll.cpp
     dll/ole32.cpp
     dll/user32.cpp
+    dll/vcruntime.cpp
     dll/version.cpp
     files.cpp
     handles.cpp

common.h

@@ -1,36 +1,95 @@
-#include <stdint.h>
-#include <stdio.h>
-#include <stdlib.h>
-#include <string.h>
+#include <cassert>
+#include <cstdint>
+#include <cstdio>
+#include <cstdlib>
+#include <cstring>
+#include <string>
 #include <sys/stat.h>
 #include <unistd.h>
-#include <assert.h>
 
 // On Windows, the incoming stack is aligned to a 4 byte boundary.
 // force_align_arg_pointer will realign the stack to match GCC's 16 byte alignment.
 #define WIN_ENTRY __attribute__((force_align_arg_pointer, callee_pop_aggregate_return(0)))
 #define WIN_FUNC WIN_ENTRY __attribute__((stdcall))
 #define DEBUG_LOG(...) wibo::debug_log(__VA_ARGS__)
 
-namespace user32 {
-	int WIN_FUNC MessageBoxA(void *hwnd, const char *lpText, const char *lpCaption, unsigned int uType);
-}
+typedef void *HANDLE;
+typedef void *HMODULE;
+typedef void *PVOID;
+typedef void *LPVOID;
+typedef void *FARPROC;
+typedef uint32_t DWORD;
+typedef DWORD *PDWORD;
+typedef DWORD *LPDWORD;
+typedef int32_t LONG;
+typedef LONG *PLONG;
+typedef uint32_t ULONG;
+typedef ULONG *PULONG;
+typedef int64_t LARGE_INTEGER;
+typedef LARGE_INTEGER *PLARGE_INTEGER;
+typedef uintptr_t ULONG_PTR;
+typedef char *LPSTR;
+typedef const char *LPCSTR;
+typedef uint16_t *LPWSTR;
+typedef const uint16_t *LPCWSTR;
+typedef int BOOL;
+typedef BOOL *PBOOL;
+typedef unsigned char UCHAR;
+typedef UCHAR *PUCHAR;
+typedef size_t SIZE_T;
+typedef SIZE_T *PSIZE_T;
+typedef unsigned char BYTE;
+
+#define TRUE 1
+#define FALSE 0
+
+#define ERROR_SUCCESS 0
+#define ERROR_FILE_NOT_FOUND 2
+#define ERROR_PATH_NOT_FOUND 3
+#define ERROR_ACCESS_DENIED 5
+#define ERROR_INVALID_HANDLE 6
+#define ERROR_READ_FAULT 30
+#define ERROR_HANDLE_EOF 38
+#define ERROR_NOT_SUPPORTED 50
+#define ERROR_INVALID_PARAMETER 87
+#define ERROR_NEGATIVE_SEEK 131
+#define ERROR_ALREADY_EXISTS 183
+
+#define INVALID_SET_FILE_POINTER ((DWORD)-1)
+#define INVALID_HANDLE_VALUE ((HANDLE)-1)
+
+typedef int NTSTATUS;
+#define STATUS_SUCCESS ((NTSTATUS)0x00000000)
+#define STATUS_INVALID_HANDLE ((NTSTATUS)0xC0000008)
+#define STATUS_END_OF_FILE ((NTSTATUS)0xC0000011)
+#define STATUS_NOT_SUPPORTED ((NTSTATUS)0xC00000BB)
+#define STATUS_UNEXPECTED_IO_ERROR ((NTSTATUS)0xC00000E9)
+
+typedef int HRESULT;
+#define S_OK ((HRESULT)0x00000000)
 
 namespace wibo {
 	extern uint32_t lastError;
+	extern char **argv;
+	extern int argc;
 	extern char *commandLine;
 	extern bool debugEnabled;
 
 	void debug_log(const char *fmt, ...);
 
-	void *resolveVersion(const char *name);
-	void *resolveKernel32(const char *name);
-	void *resolveUser32(const char *name);
-	void *resolveOle32(const char *name);
-	void *resolveAdvApi32(const char *name);
-	void *resolveLmgr(uint16_t ordinal);
-	void *resolveFuncByName(const char *dllName, const char *funcName);
-	void *resolveFuncByOrdinal(const char *dllName, uint16_t ordinal);
+	using ResolveByName = void *(*)(const char *);
+	using ResolveByOrdinal = void *(*)(uint16_t);
+	struct Module {
+		const char** names;
+		ResolveByName byName;
+		ResolveByOrdinal byOrdinal;
+	};
+	extern const Module *modules[];
+
+	HMODULE loadModule(const char *name);
+	void freeModule(HMODULE module);
+	void *resolveFuncByName(HMODULE module, const char *funcName);
+	void *resolveFuncByOrdinal(HMODULE module, uint16_t ordinal);
 
 	struct Executable {
 		Executable();

dll/advapi32.cpp

@@ -7,8 +7,17 @@ namespace advapi32 {
 	}
 }
 
-void *wibo::resolveAdvApi32(const char *name) {
+static void *resolveByName(const char *name) {
 	if (strcmp(name, "RegOpenKeyExA") == 0) return (void *) advapi32::RegOpenKeyExA;
-	return 0;
+	return nullptr;
 }
 
+wibo::Module lib_advapi32 = {
+	(const char *[]){
+		"advapi32",
+		"advapi32.dll",
+		nullptr,
+	},
+	resolveByName,
+	nullptr,
+};

dll/bcrypt.cpp

@@ -0,0 +1,38 @@
+#include "common.h"
+
+#include <algorithm>
+#include <climits>
+#include <random>
+
+typedef PVOID BCRYPT_ALG_HANDLE;
+
+namespace bcrypt {
+
+using random_bytes_engine = std::independent_bits_engine<std::default_random_engine, CHAR_BIT, unsigned char>;
+
+NTSTATUS WIN_FUNC BCryptGenRandom(BCRYPT_ALG_HANDLE hAlgorithm, PUCHAR pbBuffer, ULONG cbBuffer, ULONG dwFlags) {
+	DEBUG_LOG("BCryptGenRandom(%p, %p, %lu, %lu)\n", hAlgorithm, pbBuffer, cbBuffer, dwFlags);
+	assert(hAlgorithm == nullptr);
+	assert(dwFlags == 0 || dwFlags == 2 /* BCRYPT_USE_SYSTEM_PREFERRED_RNG */);
+	random_bytes_engine rbe;
+	std::generate(pbBuffer, pbBuffer + cbBuffer, std::ref(rbe));
+	return STATUS_SUCCESS;
+}
+
+} // namespace bcrypt
+
+static void *resolveByName(const char *name) {
+	if (strcmp(name, "BCryptGenRandom") == 0)
+		return (void *)bcrypt::BCryptGenRandom;
+	return nullptr;
+}
+
+wibo::Module lib_bcrypt = {
+	(const char *[]){
+		"bcrypt",
+		"bcrypt.dll",
+		nullptr,
+	},
+	resolveByName,
+	nullptr,
+};