469 lines
12 KiB
C++
469 lines
12 KiB
C++
#include "./ngc_hs1.h"
|
|
|
|
#include "ngc_ext_common.hpp"
|
|
|
|
#include <cstdint>
|
|
#include <cassert>
|
|
#include <new>
|
|
#include <map>
|
|
#include <list>
|
|
#include <set>
|
|
#include <optional>
|
|
#include <algorithm>
|
|
|
|
struct NGC_HS1 {
|
|
NGC_HS1_options options;
|
|
|
|
// callbacks
|
|
//tox_group_message_cb* client_tox_msg_callback;
|
|
|
|
|
|
// key - key - key - value store
|
|
// group pubkey - peer pubkey - msg_id - message(type + text)
|
|
struct Message {
|
|
uint32_t msg_id{};
|
|
Tox_Message_Type type{};
|
|
std::string text{};
|
|
};
|
|
|
|
struct Peer {
|
|
std::optional<uint32_t> id;
|
|
std::map<uint32_t, Message> dict;
|
|
std::list<uint32_t> order; // ordered list of message ids
|
|
|
|
// msg_ids we have only heard of, with peer_number of who we heard it from
|
|
std::map<uint32_t, std::set<uint32_t>> heard_of;
|
|
|
|
// dont start immediatly
|
|
float time_since_last_request_sent {0.f};
|
|
|
|
void append(uint32_t msg_id, Tox_Message_Type type, const std::string& text);
|
|
|
|
// returns if new (from that peer)
|
|
bool hear(uint32_t msg_id, uint32_t peer_number);
|
|
};
|
|
|
|
struct Group {
|
|
std::map<_PeerKey, Peer> peers;
|
|
};
|
|
|
|
std::map<_GroupKey, Group> history;
|
|
};
|
|
|
|
void NGC_HS1::Peer::append(uint32_t msg_id, Tox_Message_Type type, const std::string& text) {
|
|
order.push_back(msg_id);
|
|
|
|
// overwrites
|
|
auto& new_msg = dict[msg_id];
|
|
new_msg.msg_id = msg_id;
|
|
new_msg.type = type;
|
|
new_msg.text = text;
|
|
|
|
if (heard_of.count(msg_id)) {
|
|
// we got history before we got the message
|
|
heard_of.erase(msg_id);
|
|
}
|
|
|
|
fprintf(stderr, "######## last msgs ########\n");
|
|
auto rit = order.crbegin();
|
|
for (size_t i = 0; i < 10 && rit != order.crend(); i++, rit++) {
|
|
fprintf(stderr, " %08X - %s\n", *rit, dict.at(*rit).text.c_str());
|
|
}
|
|
}
|
|
|
|
bool NGC_HS1::Peer::hear(uint32_t msg_id, uint32_t peer_number) {
|
|
if (dict.count(msg_id)) {
|
|
// we know
|
|
return false;
|
|
}
|
|
|
|
if (heard_of.count(msg_id) && heard_of.at(msg_id).count(peer_number)) {
|
|
// we heard it from that peer before
|
|
return false;
|
|
}
|
|
|
|
heard_of[msg_id].emplace(peer_number);
|
|
|
|
return true;
|
|
}
|
|
|
|
NGC_HS1* NGC_HS1_new(const struct NGC_HS1_options* options) {
|
|
NGC_HS1* context = new NGC_HS1;
|
|
|
|
context->options = *options;
|
|
|
|
return context;
|
|
}
|
|
|
|
void NGC_HS1_kill(NGC_HS1* ngc_hs1_ctx) {
|
|
delete ngc_hs1_ctx;
|
|
}
|
|
|
|
static void _iterate_group(Tox *tox, NGC_HS1* ngc_hs1_ctx, uint32_t group_number, float time_delta) {
|
|
//fprintf(stderr, "g:%u\n", g_i);
|
|
_GroupKey g_id{};
|
|
{ // TODO: error
|
|
tox_group_get_chat_id(tox, group_number, g_id.data.data(), nullptr);
|
|
}
|
|
|
|
if (ngc_hs1_ctx->history.count(g_id) == 0) {
|
|
fprintf(stderr, "adding new group: %u %X%X%X%X\n",
|
|
group_number,
|
|
g_id.data.data()[0],
|
|
g_id.data.data()[1],
|
|
g_id.data.data()[2],
|
|
g_id.data.data()[3]
|
|
);
|
|
ngc_hs1_ctx->history[g_id];
|
|
} else {
|
|
auto& group = ngc_hs1_ctx->history[g_id];
|
|
|
|
// for each peer
|
|
for (auto& [key, peer] : group.peers) {
|
|
//fprintf(stderr, " p: %X%X%X%X\n", key.data.data()[0], key.data.data()[1], key.data.data()[2], key.data.data()[3]);
|
|
peer.time_since_last_request_sent += time_delta;
|
|
if (peer.time_since_last_request_sent > ngc_hs1_ctx->options.query_interval_per_peer) {
|
|
peer.time_since_last_request_sent = 0.f;
|
|
|
|
fprintf(stderr, "requesting ids for %X%X%X%X\n", key.data.data()[0], key.data.data()[1], key.data.data()[2], key.data.data()[3]);
|
|
|
|
// TODO: other way around?
|
|
// ask everyone if they have newer stuff for this peer
|
|
|
|
// - 1 byte packet id
|
|
// - peer_key bytes (peer key we want to know ids for)
|
|
// - 1 byte (uint8_t count ids, atleast 1)
|
|
std::array<uint8_t, 1+TOX_GROUP_PEER_PUBLIC_KEY_SIZE+1> pkg;
|
|
pkg[0] = HS1_REQUEST_LAST_IDS;
|
|
std::copy(key.data.begin(), key.data.end(), pkg.begin()+1);
|
|
pkg[1+TOX_GROUP_PEER_PUBLIC_KEY_SIZE] = ngc_hs1_ctx->options.last_msg_ids_count; // request last (up to) 5 msg_ids
|
|
|
|
tox_group_send_custom_packet(tox, group_number, true, pkg.data(), pkg.size(), nullptr);
|
|
}
|
|
}
|
|
}
|
|
assert(ngc_hs1_ctx->history.size() != 0);
|
|
|
|
assert(ngc_hs1_ctx->history.count(g_id));
|
|
}
|
|
|
|
void NGC_HS1_iterate(Tox *tox, NGC_HS1* ngc_hs1_ctx/*, void *user_data*/) {
|
|
assert(ngc_hs1_ctx);
|
|
|
|
//fprintf(stderr, "groups: %u\n", ngc_hs1_ctx->history.size());
|
|
|
|
uint32_t group_count = tox_group_get_number_groups(tox);
|
|
// this can loop endless if toxcore misbehaves
|
|
for (uint32_t g_i = 0, g_c_done = 0; g_c_done < group_count; g_i++) {
|
|
Tox_Err_Group_Is_Connected g_err;
|
|
if (tox_group_is_connected(tox, g_i, &g_err)) {
|
|
// valid and connected here
|
|
// TODO: delta time, or other timers
|
|
_iterate_group(tox, ngc_hs1_ctx, g_i, 0.02f);
|
|
g_c_done++;
|
|
} else if (g_err != TOX_ERR_GROUP_IS_CONNECTED_GROUP_NOT_FOUND) {
|
|
g_c_done++;
|
|
} // else do nothing
|
|
|
|
// safety
|
|
if (g_i > group_count + 1000) {
|
|
fprintf(stderr, "WAY PAST GOUPS in iterate\n");
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
void NGC_HS1_peer_online(Tox* tox, NGC_HS1* ngc_hs1_ctx, uint32_t group_number, uint32_t peer_number, bool online) {
|
|
// get group id
|
|
_GroupKey g_id{};
|
|
{ // TODO: error
|
|
tox_group_get_chat_id(tox, group_number, g_id.data.data(), nullptr);
|
|
}
|
|
|
|
auto& group = ngc_hs1_ctx->history[g_id];
|
|
|
|
if (online) {
|
|
// get peer id
|
|
_PeerKey p_id{};
|
|
{ // TODO: error
|
|
tox_group_peer_get_public_key(tox, group_number, peer_number, p_id.data.data(), nullptr);
|
|
}
|
|
|
|
auto& peer = group.peers[p_id];
|
|
peer.id = peer_number;
|
|
} else { // offline
|
|
// search
|
|
for (auto& [key, peer] : group.peers) {
|
|
if (peer.id.has_value() && peer.id.value() == peer_number) {
|
|
peer.id = {}; // reset
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
bool NGC_HS1_shim_group_send_message(
|
|
const Tox *tox,
|
|
NGC_HS1* ngc_hs1_ctx,
|
|
|
|
uint32_t group_number,
|
|
|
|
Tox_Message_Type type, const uint8_t *message, size_t length,
|
|
|
|
uint32_t *message_id,
|
|
Tox_Err_Group_Send_Message *error
|
|
) {
|
|
uint32_t* msg_id_ptr = message_id;
|
|
uint32_t msg_id_placeholder = 0;
|
|
if (msg_id_ptr == nullptr) {
|
|
msg_id_ptr = &msg_id_placeholder;
|
|
}
|
|
|
|
bool ret = tox_group_send_message(tox, group_number, type, message, length, msg_id_ptr, error);
|
|
|
|
NGC_HS1_record_own_message(tox, ngc_hs1_ctx, group_number, type, message, length, *msg_id_ptr);
|
|
|
|
return ret;
|
|
}
|
|
|
|
|
|
// record own msg
|
|
void NGC_HS1_record_own_message(
|
|
const Tox *tox,
|
|
NGC_HS1* ngc_hs1_ctx,
|
|
|
|
uint32_t group_number,
|
|
|
|
Tox_Message_Type type, const uint8_t *message, size_t length, uint32_t message_id
|
|
) {
|
|
fprintf(stderr, "record_own_message %08X\n", message_id);
|
|
// get group id
|
|
_GroupKey g_id{};
|
|
{ // TODO: error
|
|
tox_group_get_chat_id(tox, group_number, g_id.data.data(), nullptr);
|
|
}
|
|
|
|
// get peer id
|
|
_PeerKey p_id{};
|
|
{ // TODO: error
|
|
tox_group_self_get_public_key(tox, group_number, p_id.data.data(), nullptr);
|
|
}
|
|
|
|
ngc_hs1_ctx->history[g_id].peers[p_id].append(message_id, type, std::string{message, message+length});
|
|
assert(ngc_hs1_ctx->history.size() != 0);
|
|
assert(ngc_hs1_ctx->history.count(g_id));
|
|
}
|
|
|
|
// record others msg
|
|
void NGC_HS1_record_message(
|
|
const Tox *tox,
|
|
NGC_HS1* ngc_hs1_ctx,
|
|
|
|
uint32_t group_number,
|
|
uint32_t peer_number,
|
|
|
|
Tox_Message_Type type, const uint8_t *message, size_t length, uint32_t message_id
|
|
) {
|
|
if (!ngc_hs1_ctx->options.record_others) {
|
|
return;
|
|
}
|
|
|
|
fprintf(stderr, "record_message %08X\n", message_id);
|
|
// get group id
|
|
_GroupKey g_id{};
|
|
{ // TODO: error
|
|
tox_group_get_chat_id(tox, group_number, g_id.data.data(), nullptr);
|
|
}
|
|
|
|
// get peer id
|
|
_PeerKey p_id{};
|
|
{ // TODO: error
|
|
tox_group_peer_get_public_key(tox, group_number, peer_number, p_id.data.data(), nullptr);
|
|
}
|
|
|
|
ngc_hs1_ctx->history[g_id].peers[p_id].append(message_id, type, std::string{message, message+length});
|
|
}
|
|
|
|
static void _handle_HS1_REQUEST_LAST_IDS(
|
|
Tox* tox,
|
|
NGC_HS1* ngc_hs1_ctx,
|
|
|
|
uint32_t group_number,
|
|
uint32_t peer_number,
|
|
|
|
const uint8_t *data,
|
|
size_t length
|
|
);
|
|
|
|
static void _handle_HS1_RESPONSE_LAST_IDS(
|
|
Tox* tox,
|
|
NGC_HS1* ngc_hs1_ctx,
|
|
|
|
uint32_t group_number,
|
|
uint32_t peer_number,
|
|
|
|
const uint8_t *data,
|
|
size_t length
|
|
);
|
|
|
|
#define _HS1_HAVE(x, error) if ((length - curser) < (x)) { error; }
|
|
void NGC_HS1_handle_group_custom_packet(
|
|
Tox* tox,
|
|
NGC_HS1* ngc_hs1_ctx,
|
|
|
|
uint32_t group_number,
|
|
uint32_t peer_number,
|
|
|
|
const uint8_t *data,
|
|
size_t length
|
|
//void *user_data
|
|
) {
|
|
size_t curser = 0;
|
|
|
|
_HS1_HAVE(1, return)
|
|
|
|
_PacketType pkg_type = static_cast<_PacketType>(*(data + curser));
|
|
curser++;
|
|
|
|
fprintf(stderr, "custom_packet [%s] %u\n", _pkgid2str(pkg_type), length);
|
|
|
|
switch (pkg_type) {
|
|
case INVALID:
|
|
break;
|
|
case HS1_REQUEST_LAST_IDS:
|
|
_handle_HS1_REQUEST_LAST_IDS(tox, ngc_hs1_ctx, group_number, peer_number, data+curser, length-curser);
|
|
break;
|
|
case HS1_RESPONSE_LAST_IDS:
|
|
_handle_HS1_RESPONSE_LAST_IDS(tox, ngc_hs1_ctx, group_number, peer_number, data+curser, length-curser);
|
|
break;
|
|
}
|
|
|
|
}
|
|
|
|
static void _handle_HS1_REQUEST_LAST_IDS(
|
|
Tox* tox,
|
|
NGC_HS1* ngc_hs1_ctx,
|
|
|
|
uint32_t group_number,
|
|
uint32_t peer_number,
|
|
|
|
const uint8_t *data,
|
|
size_t length
|
|
) {
|
|
size_t curser = 0;
|
|
|
|
_PeerKey p_key;
|
|
_HS1_HAVE(p_key.data.size(), fprintf(stderr, "packet too small, missing pkey\n"); return)
|
|
|
|
std::copy(data+curser, data+curser+p_key.data.size(), p_key.data.begin());
|
|
curser += p_key.data.size();
|
|
|
|
_HS1_HAVE(1, fprintf(stderr, "packet too small, missing count\n"); return)
|
|
uint8_t last_msg_id_count = data[curser++];
|
|
|
|
fprintf(stderr, "got request for last %u ids\n", last_msg_id_count);
|
|
|
|
// get group id
|
|
_GroupKey g_id{};
|
|
{ // TODO: error
|
|
tox_group_get_chat_id(tox, group_number, g_id.data.data(), nullptr);
|
|
}
|
|
|
|
auto& group = ngc_hs1_ctx->history[g_id];
|
|
|
|
std::vector<uint32_t> message_ids{};
|
|
|
|
if (!group.peers.empty() && group.peers.count(p_key)) {
|
|
const auto& peer = group.peers.at(p_key);
|
|
auto rit = peer.order.crbegin();
|
|
for (size_t c = 0; c < last_msg_id_count && rit != peer.order.crend(); c++, rit++) {
|
|
message_ids.push_back(*rit);
|
|
}
|
|
}
|
|
|
|
// - 1 byte packet id
|
|
// respond to a request with 0 or more message ids, sorted by newest first
|
|
// - peer_key bytes (the msg_ids are from)
|
|
// - 1 byte (uint8_t count ids, can be 0)
|
|
// - array [
|
|
// - msg_id bytes (the message id
|
|
// - ]
|
|
//std::array<uint8_t, 1+TOX_GROUP_PEER_PUBLIC_KEY_SIZE+1+> pkg;
|
|
std::vector<uint8_t> pkg;
|
|
pkg.resize(1+TOX_GROUP_PEER_PUBLIC_KEY_SIZE+1+sizeof(uint32_t)*message_ids.size());
|
|
|
|
size_t packing_curser = 0;
|
|
|
|
pkg[packing_curser++] = HS1_RESPONSE_LAST_IDS;
|
|
|
|
std::copy(p_key.data.begin(), p_key.data.end(), pkg.begin()+packing_curser);
|
|
packing_curser += p_key.data.size();
|
|
|
|
pkg[packing_curser++] = message_ids.size();
|
|
|
|
for (size_t i = 0; i < message_ids.size(); i++) {
|
|
const uint8_t* tmp_ptr = reinterpret_cast<uint8_t*>(message_ids.data()+i);
|
|
// HACK: little endian
|
|
//std::copy(tmp_ptr, tmp_ptr+sizeof(uint32_t), pkg.begin()+1+TOX_GROUP_PEER_PUBLIC_KEY_SIZE+1+i*sizeof(uint32_t));
|
|
std::copy(tmp_ptr, tmp_ptr+sizeof(uint32_t), pkg.begin()+packing_curser);
|
|
packing_curser += sizeof(uint32_t);
|
|
}
|
|
|
|
tox_group_send_custom_private_packet(tox, group_number, peer_number, true, pkg.data(), pkg.size(), nullptr);
|
|
}
|
|
|
|
static void _handle_HS1_RESPONSE_LAST_IDS(
|
|
Tox* tox,
|
|
NGC_HS1* ngc_hs1_ctx,
|
|
|
|
uint32_t group_number,
|
|
uint32_t peer_number,
|
|
|
|
const uint8_t *data,
|
|
size_t length
|
|
) {
|
|
size_t curser = 0;
|
|
|
|
_PeerKey p_key;
|
|
_HS1_HAVE(p_key.data.size(), fprintf(stderr, "packet too small, missing pkey\n"); return)
|
|
|
|
std::copy(data+curser, data+curser+p_key.data.size(), p_key.data.begin());
|
|
curser += p_key.data.size();
|
|
|
|
// TODO: did we ask?
|
|
|
|
_HS1_HAVE(1, fprintf(stderr, "packet too small, missing count\n"); return)
|
|
uint8_t last_msg_id_count = data[curser++];
|
|
|
|
fprintf(stderr, "got response with last %u ids:\n", last_msg_id_count);
|
|
|
|
if (last_msg_id_count == 0) {
|
|
return;
|
|
}
|
|
|
|
// get group id
|
|
_GroupKey g_id{};
|
|
{ // TODO: error
|
|
tox_group_get_chat_id(tox, group_number, g_id.data.data(), nullptr);
|
|
}
|
|
|
|
// get peer
|
|
auto& peer = ngc_hs1_ctx->history[g_id].peers[p_key];
|
|
|
|
for (size_t i = 0; i < last_msg_id_count && curser+sizeof(uint32_t) <= length; i++) {
|
|
uint32_t msg_id;
|
|
// HACK: little endian
|
|
std::copy(data+curser, data+curser+sizeof(uint32_t), reinterpret_cast<uint8_t*>(&msg_id));
|
|
curser += sizeof(uint32_t);
|
|
fprintf(stderr, " %08X", msg_id);
|
|
|
|
if (peer.hear(msg_id, peer_number)) {
|
|
fprintf(stderr, " - NEW");
|
|
}
|
|
|
|
fprintf(stderr, "\n");
|
|
}
|
|
}
|
|
|
|
#undef _HS1_HAVE
|