141 lines
3.0 KiB
C++
141 lines
3.0 KiB
C++
#pragma once
|
|
|
|
#define WIN32_LEAN_AND_MEAN
|
|
#include <windows.h>
|
|
#include <Wincrypt.h>
|
|
|
|
#include <jansson.h>
|
|
|
|
#include <cstdint>
|
|
#include <string>
|
|
|
|
/* ------------------------------------------------------------------------ */
|
|
|
|
template<typename T, void freefunc(T)> class CustomHandle {
|
|
T handle;
|
|
|
|
public:
|
|
inline CustomHandle() : handle(0) {}
|
|
inline CustomHandle(T in) : handle(in) {}
|
|
inline ~CustomHandle()
|
|
{
|
|
if (handle)
|
|
freefunc(handle);
|
|
}
|
|
|
|
inline T *operator&() { return &handle; }
|
|
inline operator T() const { return handle; }
|
|
inline T get() const { return handle; }
|
|
|
|
inline CustomHandle<T, freefunc> &operator=(T in)
|
|
{
|
|
if (handle)
|
|
freefunc(handle);
|
|
handle = in;
|
|
return *this;
|
|
}
|
|
|
|
inline bool operator!() const { return !handle; }
|
|
};
|
|
|
|
void FreeProvider(HCRYPTPROV prov);
|
|
void FreeHash(HCRYPTHASH hash);
|
|
void FreeKey(HCRYPTKEY key);
|
|
|
|
using CryptProvider = CustomHandle<HCRYPTPROV, FreeProvider>;
|
|
using CryptHash = CustomHandle<HCRYPTHASH, FreeHash>;
|
|
using CryptKey = CustomHandle<HCRYPTKEY, FreeKey>;
|
|
|
|
/* ------------------------------------------------------------------------ */
|
|
|
|
template<typename T> class LocalPtr {
|
|
T *ptr = nullptr;
|
|
|
|
public:
|
|
inline ~LocalPtr()
|
|
{
|
|
if (ptr)
|
|
LocalFree(ptr);
|
|
}
|
|
|
|
inline T **operator&() { return &ptr; }
|
|
inline operator T() const { return ptr; }
|
|
inline T *get() const { return ptr; }
|
|
|
|
inline bool operator!() const { return !ptr; }
|
|
|
|
inline T *operator->() { return ptr; }
|
|
};
|
|
|
|
/* ------------------------------------------------------------------------ */
|
|
|
|
class Json {
|
|
json_t *json;
|
|
|
|
public:
|
|
inline Json() : json(nullptr) {}
|
|
explicit inline Json(json_t *json_) : json(json_) {}
|
|
inline Json(const Json &from) : json(json_incref(from.json)) {}
|
|
inline Json(Json &&from) : json(from.json) { from.json = nullptr; }
|
|
|
|
inline ~Json()
|
|
{
|
|
if (json)
|
|
json_decref(json);
|
|
}
|
|
|
|
inline Json &operator=(json_t *json_)
|
|
{
|
|
if (json)
|
|
json_decref(json);
|
|
json = json_;
|
|
return *this;
|
|
}
|
|
inline Json &operator=(const Json &from)
|
|
{
|
|
if (json)
|
|
json_decref(json);
|
|
json = json_incref(from.json);
|
|
return *this;
|
|
}
|
|
inline Json &operator=(Json &&from)
|
|
{
|
|
if (json)
|
|
json_decref(json);
|
|
json = from.json;
|
|
from.json = nullptr;
|
|
return *this;
|
|
}
|
|
|
|
inline operator json_t *() const { return json; }
|
|
|
|
inline bool operator!() const { return !json; }
|
|
|
|
inline const char *GetString(const char *name,
|
|
const char *def = nullptr) const
|
|
{
|
|
json_t *obj(json_object_get(json, name));
|
|
if (!obj)
|
|
return def;
|
|
return json_string_value(obj);
|
|
}
|
|
inline int64_t GetInt(const char *name, int def = 0) const
|
|
{
|
|
json_t *obj(json_object_get(json, name));
|
|
if (!obj)
|
|
return def;
|
|
return json_integer_value(obj);
|
|
}
|
|
inline json_t *GetObject(const char *name) const
|
|
{
|
|
return json_object_get(json, name);
|
|
}
|
|
|
|
inline json_t *get() const { return json; }
|
|
};
|
|
|
|
/* ------------------------------------------------------------------------ */
|
|
|
|
std::string vstrprintf(const char *format, va_list args);
|
|
std::string strprintf(const char *format, ...);
|