Loading...
Searching...
No Matches
OmemoLibWrappers.h
// SPDX-FileCopyrightText: 2022 Linus Jahn <lnj@kaidan.im>
//
// SPDX-License-Identifier: LGPL-2.1-or-later
#ifndef OMEMOLIBWRAPPERS_H
#define OMEMOLIBWRAPPERS_H
#include <key_helper.h>
#include <session_builder.h>
#include <session_cipher.h>
#include <signal_protocol.h>
#include <QByteArray>
// Wraps various types of the OMEMO library.
template<typename T, void(destruct)(T *)>
class OmemoLibPtr
{
T *m_ptr = nullptr;
public:
OmemoLibPtr(T *ptr = nullptr) : m_ptr(ptr) { }
OmemoLibPtr(const OmemoLibPtr &) = delete;
~OmemoLibPtr()
{
if (m_ptr) {
destruct(m_ptr);
}
}
OmemoLibPtr &operator=(const OmemoLibPtr &) = delete;
OmemoLibPtr<T, destruct> &operator=(T *ptr)
{
reset(ptr);
return *this;
}
operator bool() const { return m_ptr != nullptr; }
T *operator->() const { return m_ptr; }
T *get() const { return m_ptr; }
T **ptrRef() { return &m_ptr; }
void reset(T *ptr)
{
if (m_ptr) {
destruct(m_ptr);
}
m_ptr = ptr;
}
};
template<typename T>
void omemoLibUnrefHelper(T *ptr)
{
SIGNAL_UNREF(ptr);
}
template<typename T>
using RefCountedPtr = OmemoLibPtr<T, omemoLibUnrefHelper<T>>;
static QByteArray omemoLibBufferToByteArray(signal_buffer *buffer)
{
return QByteArray(reinterpret_cast<const char *>(signal_buffer_data(buffer)), signal_buffer_len(buffer));
}
static signal_buffer *omemoLibBufferFromByteArray(const QByteArray &bytes)
{
return signal_buffer_create(reinterpret_cast<const unsigned char *>(bytes.constData()), bytes.size());
}
template<void(destruct)(signal_buffer *)>
class BufferPtrBase : public OmemoLibPtr<signal_buffer, destruct>
{
public:
BufferPtrBase(signal_buffer *ptr)
: OmemoLibPtr<signal_buffer, destruct>(ptr)
{
}
QByteArray toByteArray() const
{
return omemoLibBufferToByteArray(this->get());
}
};
class BufferSecurePtr : public BufferPtrBase<signal_buffer_bzero_free>
{
public:
BufferSecurePtr() : BufferPtrBase<signal_buffer_bzero_free>(nullptr) { }
BufferSecurePtr(signal_buffer *ptr)
: BufferPtrBase<signal_buffer_bzero_free>(ptr)
{
}
static BufferSecurePtr fromByteArray(const QByteArray &bytes)
{
return { omemoLibBufferFromByteArray(bytes) };
}
};
class BufferPtr : public BufferPtrBase<signal_buffer_free>
{
public:
BufferPtr() : BufferPtrBase<signal_buffer_free>(nullptr) { }
BufferPtr(signal_buffer *ptr)
: BufferPtrBase<signal_buffer_free>(ptr)
{
}
static BufferPtr fromByteArray(const QByteArray &bytes)
{
return { omemoLibBufferFromByteArray(bytes) };
}
};
using KeyListNodePtr = OmemoLibPtr<signal_protocol_key_helper_pre_key_list_node, signal_protocol_key_helper_key_list_free>;
using SessionCipherPtr = OmemoLibPtr<session_cipher, session_cipher_free>;
using SessionBuilderPtr = OmemoLibPtr<session_builder, session_builder_free>;
using OmemoContextPtr = OmemoLibPtr<signal_context, signal_context_destroy>;
using StoreContextPtr = OmemoLibPtr<signal_protocol_store_context, signal_protocol_store_context_destroy>;
#endif // OMEMOLIBWRAPPERS_H