move source files to src/

1 files changed, 68 insertions, 0 deletions

diff --git a/src/sid.hpp b/src/sid.hpp
new file mode 100644
index 0000000..15ce73b
--- /dev/null
+++ b/src/sid.hpp
@@ -0,0 +1,68 @@
+#pragma once
+
+#include "error.hpp"
+
+#include <Windows.h>
+#include <sddl.h>
+
+#include <array>
+#include <memory>
+#include <string>
+
+constexpr DWORD max_sid_size = SECURITY_MAX_SID_SIZE;
+typedef std::array<unsigned char, max_sid_size> SidBuffer;
+
+namespace sid
+{
+    SidBuffer well_known(WELL_KNOWN_SID_TYPE type)
+    {
+        SidBuffer buffer;
+        DWORD cb = static_cast<DWORD>(buffer.size());
+
+        if (!CreateWellKnownSid(type, NULL, buffer.data(), &cb))
+            error::raise("CreateWellKnownSid");
+
+        return buffer;
+    }
+
+    SidBuffer builtin_administrators()
+    {
+        /*
+        void* sid = nullptr;
+        SID_IDENTIFIER_AUTHORITY authority = SECURITY_NT_AUTHORITY;
+
+        if (!AllocateAndInitializeSid(
+            &authority,
+            2,
+            SECURITY_BUILTIN_DOMAIN_RID,
+            DOMAIN_ALIAS_RID_ADMINS,
+            0, 0, 0, 0, 0, 0,
+            &sid))
+        {
+            error::raise("AllocateAndInitializeSid");
+        }
+
+        return std::unique_ptr<void, FreeSid>{sid};
+        */
+
+        return well_known(WinBuiltinAdministratorsSid);
+    }
+
+    struct DeleteSidString
+    {
+        void operator()(wchar_t* s) const
+        {
+            LocalFree(s);
+        }
+    };
+
+    std::wstring to_string(const SidBuffer& sid)
+    {
+        wchar_t* s = nullptr;
+
+        if (!ConvertSidToStringSidW(const_cast<unsigned char*>(sid.data()), &s))
+            error::raise("ConvertSidToStringSidW");
+
+        return std::unique_ptr<wchar_t, DeleteSidString>{s}.get();
+    }
+}