1/**
 2 * This file has no copyright assigned and is placed in the Public Domain.
 3 * This file is part of the mingw-w64 runtime package.
 4 * No warranty is given; refer to the file DISCLAIMER.PD within this package.
 5 */
 6
 7#ifndef SPHelper_h
 8#define SPHelper_h
 9
10#include <winapifamily.h>
11
12#if WINAPI_FAMILY_PARTITION(WINAPI_PARTITION_DESKTOP)
13
14#include <malloc.h>
15#include <sapi.h>
16/* #include <sapiddk.h> */
17#include <sperror.h>
18#include <limits.h>
19#include <mmsystem.h>
20#include <comcat.h>
21#include <mmreg.h>
22/* #include <atlbase.h> */
23#include <wchar.h>
24#include <tchar.h>
25#include <strsafe.h>
26#include <intsafe.h>
27
28inline HRESULT SpGetCategoryFromId(const WCHAR *category_id, ISpObjectTokenCategory **ret, BOOL fCreateIfNotExist = FALSE) {
29    ISpObjectTokenCategory *obj_token_cat;
30    HRESULT hres;
31
32    hres = ::CoCreateInstance(CLSID_SpObjectTokenCategory, NULL, CLSCTX_ALL, __uuidof(ISpObjectTokenCategory),
33            (void**)&obj_token_cat);
34    if(FAILED(hres))
35        return hres;
36
37    hres = obj_token_cat->SetId(category_id, fCreateIfNotExist);
38    if(FAILED(hres)) {
39        obj_token_cat->Release();
40        return hres;
41    }
42
43    *ret = obj_token_cat;
44    return S_OK;
45}
46
47inline HRESULT SpEnumTokens(const WCHAR *category_id, const WCHAR *req_attrs, const WCHAR *opt_attrs, IEnumSpObjectTokens **ret) {
48    ISpObjectTokenCategory *category;
49    HRESULT hres;
50
51    hres = SpGetCategoryFromId(category_id, &category);
52    if(SUCCEEDED(hres)) {
53        hres = category->EnumTokens(req_attrs, opt_attrs, ret);
54        category->Release();
55    }
56
57    return hres;
58}
59
60/* str must be at least 9 chars (8 for 32-bit integer in hex + one for '\0').  */
61inline void SpHexFromUlong(WCHAR *str, ULONG ul) {
62    ::_ultow(ul, str, 16);
63}
64
65inline HRESULT SpGetDescription(ISpObjectToken *obj_token, WCHAR **description, LANGID language = GetUserDefaultUILanguage()) {
66    WCHAR lang_id[9];
67    HRESULT hres;
68
69    SpHexFromUlong(lang_id, language);
70    hres = obj_token->GetStringValue(lang_id, description);
71    if(hres == SPERR_NOT_FOUND)
72        hres = obj_token->GetStringValue(NULL, description);
73
74    return hres;
75}
76
77#endif
78#endif