From 512a25658dd6b737a16b91fb7eba43deef004d87 Mon Sep 17 00:00:00 2001 From: Egor Tensin Date: Sun, 18 Oct 2020 04:05:48 +0300 Subject: add Sid class It's another newcomer straight from privilege-check. --- src/cmd_line.cpp | 5 +---- src/handle.cpp | 2 +- src/sid.cpp | 48 ++++++++++++++++++++++++++++++++++++++++++++++++ 3 files changed, 50 insertions(+), 5 deletions(-) create mode 100644 src/sid.cpp (limited to 'src') diff --git a/src/cmd_line.cpp b/src/cmd_line.cpp index eaf505f..d3dc781 100644 --- a/src/cmd_line.cpp +++ b/src/cmd_line.cpp @@ -6,6 +6,7 @@ #include #include #include +#include #include #include @@ -34,10 +35,6 @@ std::vector narrow_all(int argc, wchar_t** argv) { return utf; } -struct LocalDelete { - void operator()(wchar_t* argv[]) const { ::LocalFree(argv); } -}; - CommandLine do_parse(std::wstring src) { boost::trim(src); if (src.empty()) { diff --git a/src/handle.cpp b/src/handle.cpp index 1147cf6..c94cb16 100644 --- a/src/handle.cpp +++ b/src/handle.cpp @@ -6,7 +6,7 @@ #include #include #include -#include +#include #include diff --git a/src/sid.cpp b/src/sid.cpp new file mode 100644 index 0000000..23be61a --- /dev/null +++ b/src/sid.cpp @@ -0,0 +1,48 @@ +// Copyright (c) 2020 Egor Tensin +// This file is part of the "winapi-common" project. +// For details, see https://github.com/egor-tensin/winapi-common. +// Distributed under the MIT License. + +#include +#include +#include +#include +#include + +// clang-format off +#include +#include +// clang-format on + +#include +#include + +namespace winapi { + +Sid Sid::well_known(WELL_KNOWN_SID_TYPE type) { + Buffer buffer; + buffer.resize(MAX_SID_SIZE); + + DWORD cb = static_cast(buffer.size()); + + if (!::CreateWellKnownSid(type, NULL, buffer.data(), &cb)) + throw error::windows(GetLastError(), "CreateWellKnownSid"); + + buffer.resize(cb); + return Sid{buffer}; +} + +Sid Sid::builtin_administrators() { + return well_known(WinBuiltinAdministratorsSid); +} + +std::string Sid::to_string() const { + wchar_t* s = nullptr; + + if (!::ConvertSidToStringSidW(const_cast(&get_impl()), &s)) + throw error::windows(GetLastError(), "ConvertSidToStringSidW"); + + return narrow(std::unique_ptr{s}.get()); +} + +} // namespace winapi -- cgit v1.2.3