WinStd
Windows Win32 API using Standard C++
Loading...
Searching...
No Matches
Sec.h
1/*
2 SPDX-License-Identifier: MIT
3 Copyright © 1991-2023 Amebis
4 Copyright © 2016 GÉANT
5*/
6
8
9#pragma once
10
11#include "Common.h"
12#include <Security.h>
13#include <string>
14
17
18#if defined(SECURITY_WIN32) || defined(SECURITY_KERNEL)
19
21template<class _Traits, class _Ax>
22static BOOLEAN GetUserNameExA(_In_ EXTENDED_NAME_FORMAT NameFormat, _Inout_ std::basic_string<char, _Traits, _Ax> &sName)
23{
24 assert(0); // TODO: Test this code.
25
26 char szStackBuffer[WINSTD_STACK_BUFFER_BYTES/sizeof(char)];
27 ULONG ulSize = _countof(szStackBuffer);
28
29 // Try with stack buffer first.
30 if (::GetUserNameExA(NameFormat, szStackBuffer, &ulSize)) {
31 // Copy from stack.
32 sName.assign(szStackBuffer, ulSize);
33 return TRUE;
34 }
35 if (::GetLastError() == ERROR_MORE_DATA) {
36 // Allocate buffer on heap and retry.
37 std::unique_ptr<char[]> szBuffer(new char[ulSize]);
38 if (::GetUserNameExA(NameFormat, szBuffer.get(), &ulSize)) {
39 sName.assign(szBuffer.get(), ulSize);
40 return TRUE;
41 }
42 }
43 return FALSE;
44}
45
51template<class _Traits, class _Ax>
52static BOOLEAN GetUserNameExW(_In_ EXTENDED_NAME_FORMAT NameFormat, _Inout_ std::basic_string<wchar_t, _Traits, _Ax> &sName)
53{
54 assert(0); // TODO: Test this code.
55
56 wchar_t szStackBuffer[WINSTD_STACK_BUFFER_BYTES/sizeof(wchar_t)];
57 ULONG ulSize = _countof(szStackBuffer);
58
59 // Try with stack buffer first.
60 if (::GetUserNameExW(NameFormat, szStackBuffer, &ulSize)) {
61 // Copy from stack.
62 sName.assign(szStackBuffer, ulSize);
63 return TRUE;
64 }
65 if (::GetLastError() == ERROR_MORE_DATA) {
66 // Allocate buffer on heap and retry.
67 std::unique_ptr<wchar_t[]> szBuffer(new wchar_t[ulSize]);
68 if (::GetUserNameExW(NameFormat, szBuffer.get(), &ulSize)) {
69 sName.assign(szBuffer.get(), ulSize);
70 return TRUE;
71 }
72 }
73 return FALSE;
74}
75
76#endif
77
79
80namespace winstd
81{
84
88 class sec_credentials : public handle<PCredHandle, NULL>
89 {
91
92 public:
97 {
98 m_expires.QuadPart = -1;
99 }
100
111
118 m_expires(std::move(h.m_expires)),
119 handle<PCredHandle, NULL>(std::move(h))
120 {}
121
128 {
129 if (m_h != invalid)
131 }
132
139 {
140 if (this != std::addressof(h)) {
141 *(handle<handle_type, NULL>*)this = std::move(h);
142 m_expires = std::move(h.m_expires);
143 }
144 return *this;
145 }
146
175
176 protected:
182 void free_internal() noexcept override
183 {
185 delete m_h;
186 }
187
188 public:
190 };
191
195 class sec_context : public handle<PCtxtHandle, NULL>
196 {
197 public:
202 m_attrib(0),
204 {
205 m_expires.QuadPart = -1;
206 }
207
214 m_attrib (std::move(h.m_attrib )),
215 m_expires(std::move(h.m_expires)),
216 handle<PCtxtHandle, NULL>(std::move(h))
217 {}
218
224 virtual ~sec_context()
225 {
226 if (m_h != invalid)
228 }
229
236 {
237 if (this != std::addressof(h)) {
238 *(handle<handle_type, NULL>*)this = std::move(h);
239 m_attrib = std::move(h.m_attrib);
240 m_expires = std::move(h.m_expires);
241 }
242 return *this;
243 }
244
276
296
297 protected:
303 void free_internal() noexcept override
304 {
306 delete m_h;
307 }
308
309 public:
312 };
313
317 class sec_buffer_desc : public SecBufferDesc
318 {
319 public:
329
336 {
337 for (ULONG i = 0; i < cBuffers; i++) {
338 if (pBuffers[i].pvBuffer)
340 }
341 }
342 };
343
345
348
354 class sec_runtime_error : public num_runtime_error<SECURITY_STATUS>
355 {
356 public:
365
374
382 };
383
385}
Base abstract template class to support generic object handle keeping.
Definition Common.h:1020
handle_type m_h
Object handle.
Definition Common.h:1272
void attach(handle_type h) noexcept
Sets a new object handle for the class.
Definition Common.h:1235
Numerical runtime error.
Definition Common.h:1477
Helper class for returning pointers to std::unique_ptr.
Definition Common.h:859
SecBufferDesc wrapper class.
Definition Sec.h:318
virtual ~sec_buffer_desc()
Frees the security buffer descriptor.
Definition Sec.h:335
sec_buffer_desc(PSecBuffer buf, ULONG count, ULONG version=SECBUFFER_VERSION)
Initializes security buffer descriptor.
Definition Sec.h:323
PCtxtHandle wrapper class.
Definition Sec.h:196
sec_context(sec_context &&h) noexcept
Move constructor.
Definition Sec.h:213
SECURITY_STATUS process(PCredHandle phCredential, LPCTSTR pszTargetName, ULONG fContextReq, ULONG TargetDataRep, PSecBufferDesc pInput, PSecBufferDesc pOutput)
Continue security context.
Definition Sec.h:286
virtual ~sec_context()
Frees the security context.
Definition Sec.h:224
sec_context()
Initializes a new class instance with the object handle set to NULL.
Definition Sec.h:201
SECURITY_STATUS initialize(PCredHandle phCredential, LPCTSTR pszTargetName, ULONG fContextReq, ULONG TargetDataRep, PSecBufferDesc pInput, PSecBufferDesc pOutput)
Initializes security context.
Definition Sec.h:254
ULONG m_attrib
Context attributes.
Definition Sec.h:310
TimeStamp m_expires
Context expiration time.
Definition Sec.h:311
sec_context & operator=(sec_context &&h) noexcept
Move assignment.
Definition Sec.h:235
void free_internal() noexcept override
Frees the security context.
Definition Sec.h:303
PCredHandle wrapper class.
Definition Sec.h:89
sec_credentials()
Initializes a new class instance with the object handle set to NULL.
Definition Sec.h:96
void free_internal() noexcept override
Frees the security credentials.
Definition Sec.h:182
TimeStamp m_expires
Credentials expiration time.
Definition Sec.h:189
sec_credentials(sec_credentials &&h) noexcept
Move constructor.
Definition Sec.h:117
virtual ~sec_credentials()
Frees the security credentials.
Definition Sec.h:127
sec_credentials(handle_type h, const TimeStamp expires)
Initializes a new class with an already available object handle.
Definition Sec.h:107
SECURITY_STATUS acquire(LPTSTR pszPrincipal, LPTSTR pszPackage, unsigned long fCredentialUse, void *pvLogonId, void *pAuthData, SEC_GET_KEY_FN pGetKeyFn=NULL, void *pvGetKeyArgument=NULL)
Acquires the security credentials.
Definition Sec.h:156
sec_credentials & operator=(sec_credentials &&h) noexcept
Move assignment.
Definition Sec.h:138
Security runtime error.
Definition Sec.h:355
sec_runtime_error(error_type num, const char *msg=nullptr)
Constructs an exception.
Definition Sec.h:372
sec_runtime_error(const sec_runtime_error &other)
Copies an exception.
Definition Sec.h:380
sec_runtime_error(error_type num, const std::string &msg)
Constructs an exception.
Definition Sec.h:363
#define WINSTD_NONCOPYABLE(C)
Declares a class as non-copyable.
Definition Common.h:67
#define WINSTD_STACK_BUFFER_BYTES
Size of the stack buffer in bytes used for initial system function call.
Definition Common.h:94
static const PCredHandle invalid
Invalid handle value.
Definition Common.h:1030