more send stuff, more refactoring
This commit is contained in:
parent
7399956a06
commit
0a0fb18527
@ -1,8 +1,11 @@
|
||||
#include "./tox_p2prng.hpp"
|
||||
|
||||
#include <solanaceae/contact/components.hpp>
|
||||
#include <solanaceae/tox_contacts/components.hpp>
|
||||
#include <solanaceae/toxcore/tox_interface.hpp>
|
||||
|
||||
#include <iostream>
|
||||
#include <vector>
|
||||
|
||||
// packets:
|
||||
//
|
||||
@ -25,6 +28,8 @@
|
||||
// secret_request
|
||||
// - id
|
||||
|
||||
#define TOX_PKG_ID_FRIEND 0xB1
|
||||
#define TOX_PKG_ID_GROUP 0xa6
|
||||
|
||||
void ToxP2PRNG::RngState::fillInitalStatePreamble(const ByteSpan id) {
|
||||
// id
|
||||
@ -105,6 +110,30 @@ void ToxP2PRNG::RngState::genFinalResult(void) {
|
||||
// we done
|
||||
}
|
||||
|
||||
P2PRNG::State ToxP2PRNG::RngState::getState(void) const {
|
||||
//DONE, // rng is done, event contains full rng
|
||||
if (!final_result.empty()) {
|
||||
return P2PRNG::DONE;
|
||||
}
|
||||
|
||||
//SECRET, // got a secret (phase start will be denoted by own, secrets received before we have all hmacs get queued up)
|
||||
if (hmacs.size() == contacts.size()) {
|
||||
return P2PRNG::SECRET;
|
||||
}
|
||||
|
||||
//HMAC, // got a hmac (phase start will be denoted by the initiators hmac)
|
||||
if (!hmacs.empty()) {
|
||||
return P2PRNG::HMAC;
|
||||
}
|
||||
|
||||
//INIT, // inital params (incoming or outgoing?)
|
||||
if (!inital_state.empty() && !inital_state_preamble.empty()) {
|
||||
return P2PRNG::INIT;
|
||||
}
|
||||
|
||||
return P2PRNG::UNKNOWN;
|
||||
}
|
||||
|
||||
ToxP2PRNG::ToxP2PRNG(
|
||||
ToxI& t,
|
||||
ToxEventProviderI& tep,
|
||||
@ -137,32 +166,31 @@ ByteSpan ToxP2PRNG::getResult(const ByteSpan id) {
|
||||
bool ToxP2PRNG::handlePacket(
|
||||
Contact3Handle c,
|
||||
PKG pkg_type,
|
||||
const uint8_t* data,
|
||||
const size_t data_size
|
||||
ByteSpan data
|
||||
) {
|
||||
if (data_size < 1) {
|
||||
if (data.size < 1) {
|
||||
return false; // waht
|
||||
}
|
||||
|
||||
// rn all are prefixed with ID, so here we go
|
||||
if (data_size < 32) {
|
||||
if (data.size < 32) {
|
||||
std::cerr << "TP2PRNG error: packet without id\n";
|
||||
return false;
|
||||
}
|
||||
|
||||
ByteSpan id{data, 32};
|
||||
ByteSpan id{data.ptr, 32};
|
||||
|
||||
switch (pkg_type) {
|
||||
case PKG::INIT_WITH_HMAC:
|
||||
return handle_init_with_hmac(c, id, data+32, data_size-(32));
|
||||
return handle_init_with_hmac(c, id, {data.ptr+32, data.size-(32)});
|
||||
case PKG::HMAC:
|
||||
return handle_hmac(c, id, data+32, data_size-(32));
|
||||
return handle_hmac(c, id, {data.ptr+32, data.size-(32)});
|
||||
case PKG::HMAC_REQUEST:
|
||||
return handle_hmac_request(c, id, data+32, data_size-(32));
|
||||
return handle_hmac_request(c, id, {data.ptr+32, data.size-(32)});
|
||||
case PKG::SECRET:
|
||||
return handle_secret(c, id, data+32, data_size-(32));
|
||||
return handle_secret(c, id, {data.ptr+32, data.size-(32)});
|
||||
case PKG::SECRET_REQUEST:
|
||||
return handle_secret_request(c, id, data+32, data_size-(32));
|
||||
return handle_secret_request(c, id, {data.ptr+32, data.size-(32)});
|
||||
default:
|
||||
return false;
|
||||
}
|
||||
@ -172,15 +200,14 @@ bool ToxP2PRNG::handlePacket(
|
||||
|
||||
bool ToxP2PRNG::handleFriendPacket(
|
||||
const uint32_t friend_number,
|
||||
const uint8_t* data,
|
||||
const size_t data_size
|
||||
ByteSpan data
|
||||
) {
|
||||
// packet id + packet id + id
|
||||
if (data_size < 1+1+32) {
|
||||
if (data.size < 1+1+32) {
|
||||
return false;
|
||||
}
|
||||
|
||||
if (data[0] != 0xB1) { // 177
|
||||
if (data[0] != TOX_PKG_ID_FRIEND) {
|
||||
return false;
|
||||
}
|
||||
|
||||
@ -191,22 +218,21 @@ bool ToxP2PRNG::handleFriendPacket(
|
||||
return false;
|
||||
}
|
||||
|
||||
return handlePacket(c, tpr_pkg_type, data+2, data_size-2);
|
||||
return handlePacket(c, tpr_pkg_type, {data.ptr+2, data.size-2});
|
||||
}
|
||||
|
||||
bool ToxP2PRNG::handleGroupPacket(
|
||||
const uint32_t group_number,
|
||||
const uint32_t peer_number,
|
||||
const uint8_t* data,
|
||||
const size_t data_size,
|
||||
ByteSpan data,
|
||||
const bool /*_private*/
|
||||
) {
|
||||
// packet id + packet id + id
|
||||
if (data_size < 1+1+32) {
|
||||
if (data.size < 1+1+32) {
|
||||
return false;
|
||||
}
|
||||
|
||||
if (data[0] != (0x80 | 38u)) { // 0xa6 / 166
|
||||
if (data[0] != TOX_PKG_ID_GROUP) {
|
||||
return false;
|
||||
}
|
||||
|
||||
@ -217,50 +243,185 @@ bool ToxP2PRNG::handleGroupPacket(
|
||||
return false;
|
||||
}
|
||||
|
||||
return handlePacket(c, tpr_pkg_type, data+2, data_size-2);
|
||||
return handlePacket(c, tpr_pkg_type, {data.ptr+2, data.size-2});
|
||||
}
|
||||
|
||||
bool ToxP2PRNG::handle_init_with_hmac(Contact3Handle c, ByteSpan id, const uint8_t* data, size_t data_size) {
|
||||
bool ToxP2PRNG::handle_init_with_hmac(Contact3Handle c, ByteSpan id, ByteSpan data) {
|
||||
std::cerr << "TP2PRNG: got packet INIT_WITH_HMAC\n";
|
||||
size_t curser = 0;
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
bool ToxP2PRNG::handle_hmac(Contact3Handle c, ByteSpan id, const uint8_t* data, size_t data_size) {
|
||||
bool ToxP2PRNG::handle_hmac(Contact3Handle c, ByteSpan id, ByteSpan data) {
|
||||
std::cerr << "TP2PRNG: got packet HMAC\n";
|
||||
size_t curser = 0;
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
bool ToxP2PRNG::handle_hmac_request(Contact3Handle c, ByteSpan id, const uint8_t* data, size_t data_size) {
|
||||
bool ToxP2PRNG::handle_hmac_request(Contact3Handle c, ByteSpan id, ByteSpan data) {
|
||||
std::cerr << "TP2PRNG: got packet HMAC_REQUEST\n";
|
||||
|
||||
if (data_size > 0) {
|
||||
if (!data.empty()) {
|
||||
std::cerr << "TP2PRNG warning: HMAC_REQUEST pkg has extra data!\n";
|
||||
}
|
||||
|
||||
const auto* rng_state = getRngSate(c, id);
|
||||
if (rng_state == nullptr) {
|
||||
return false;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
bool ToxP2PRNG::handle_secret(Contact3Handle c, ByteSpan id, const uint8_t* data, size_t data_size) {
|
||||
bool ToxP2PRNG::handle_secret(Contact3Handle c, ByteSpan id, ByteSpan data) {
|
||||
std::cerr << "TP2PRNG: got packet SECRET\n";
|
||||
size_t curser = 0;
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
bool ToxP2PRNG::handle_secret_request(Contact3Handle c, ByteSpan id, const uint8_t* data, size_t data_size) {
|
||||
bool ToxP2PRNG::handle_secret_request(Contact3Handle c, ByteSpan id, ByteSpan data) {
|
||||
std::cerr << "TP2PRNG: got packet SECRET_REQUEST\n";
|
||||
|
||||
if (data_size > 0) {
|
||||
if (!data.empty()) {
|
||||
std::cerr << "TP2PRNG warning: SECRET_REQUEST pkg has extra data!\n";
|
||||
}
|
||||
|
||||
const auto* rng_state = getRngSate(c, id);
|
||||
if (rng_state == nullptr) {
|
||||
return false;
|
||||
}
|
||||
|
||||
// can only request secret if secret or done state
|
||||
if (const auto current_state = rng_state->getState(); !(current_state == P2PRNG::SECRET || current_state == P2PRNG::DONE)) {
|
||||
return false;
|
||||
}
|
||||
|
||||
// find self contact
|
||||
// TODO: accel
|
||||
// TODO: do TagSelfStrong over contacts instead?? probably. maybe additionally
|
||||
Contact3Handle self;
|
||||
if (c.all_of<Contact::Components::Self>()) {
|
||||
self = Contact3Handle{*c.registry(), c.get<Contact::Components::Self>().self};
|
||||
} else if (c.all_of<Contact::Components::Parent>()) {
|
||||
Contact3Handle parent = {*c.registry(), c.get<Contact::Components::Parent>().parent};
|
||||
if (static_cast<bool>(parent) && parent.all_of<Contact::Components::Self>()) {
|
||||
self = Contact3Handle{*c.registry(), parent.get<Contact::Components::Self>().self};
|
||||
}
|
||||
}
|
||||
|
||||
if (!static_cast<bool>(self)) {
|
||||
std::cerr << "TP2PRNG error: failed to look up self\n";
|
||||
return false;
|
||||
}
|
||||
|
||||
auto self_secret_it = rng_state->secrets.find(self);
|
||||
if (self_secret_it == rng_state->secrets.cend()) {
|
||||
// hmmmmmmmmmm this bad
|
||||
std::cerr << "hmmmmmmmmmm this bad\n";
|
||||
return false;
|
||||
}
|
||||
|
||||
// TODO: queue these instead
|
||||
// SEND secret to c
|
||||
send_secret(c, id, ByteSpan{self_secret_it->second});
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
static std::tuple<std::vector<uint8_t>, const Contact::Components::ToxFriendEphemeral*, const Contact::Components::ToxGroupPeerEphemeral*> prepSendPkgWithID(Contact3Handle c, ToxP2PRNG::PKG pkg_type, ByteSpan id) {
|
||||
std::vector<uint8_t> pkg;
|
||||
|
||||
// determine friend or group (meh)
|
||||
const auto* tfe = c.try_get<Contact::Components::ToxFriendEphemeral>();
|
||||
const auto* tgpe = c.try_get<Contact::Components::ToxGroupPeerEphemeral>();
|
||||
if (tfe != nullptr) {
|
||||
pkg.push_back(TOX_PKG_ID_FRIEND);
|
||||
} else if (tgpe != nullptr) {
|
||||
pkg.push_back(TOX_PKG_ID_GROUP);
|
||||
} else {
|
||||
return {};
|
||||
}
|
||||
pkg.push_back(static_cast<uint8_t>(pkg_type));
|
||||
|
||||
// pack packet
|
||||
// - id
|
||||
pkg.insert(pkg.cend(), id.cbegin(), id.cend());
|
||||
|
||||
return {pkg, tfe, tgpe};
|
||||
}
|
||||
|
||||
static bool sendToxPrivatePacket(
|
||||
ToxI& t,
|
||||
const Contact::Components::ToxFriendEphemeral* tfe,
|
||||
const Contact::Components::ToxGroupPeerEphemeral* tgpe,
|
||||
const std::vector<uint8_t>& pkg
|
||||
) {
|
||||
// send to friend or group peer
|
||||
if (tfe != nullptr) {
|
||||
return
|
||||
t.toxFriendSendLosslessPacket(
|
||||
tfe->friend_number,
|
||||
pkg
|
||||
) == TOX_ERR_FRIEND_CUSTOM_PACKET_OK
|
||||
;
|
||||
} else if (tgpe != nullptr) {
|
||||
return
|
||||
t.toxGroupSendCustomPrivatePacket(
|
||||
tgpe->group_number, tgpe->peer_number,
|
||||
true,
|
||||
pkg
|
||||
) == TOX_ERR_GROUP_SEND_CUSTOM_PRIVATE_PACKET_OK
|
||||
;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
bool ToxP2PRNG::send_hmac(Contact3Handle c, ByteSpan id, const ByteSpan hmac) {
|
||||
auto [pkg, tfe, tgpe] = prepSendPkgWithID(c, PKG::HMAC, id);
|
||||
if (pkg.empty()) {
|
||||
return false;
|
||||
}
|
||||
|
||||
// - hmac
|
||||
pkg.insert(pkg.cend(), hmac.cbegin(), hmac.cend());
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
bool ToxP2PRNG::send_hmac_request(Contact3Handle c, ByteSpan id) {
|
||||
auto [pkg, tfe, tgpe] = prepSendPkgWithID(c, PKG::HMAC_REQUEST, id);
|
||||
if (pkg.empty()) {
|
||||
return false;
|
||||
}
|
||||
|
||||
return sendToxPrivatePacket(_t, tfe, tgpe, pkg);
|
||||
}
|
||||
|
||||
bool ToxP2PRNG::send_secret(Contact3Handle c, ByteSpan id, const ByteSpan secret) {
|
||||
auto [pkg, tfe, tgpe] = prepSendPkgWithID(c, PKG::SECRET, id);
|
||||
if (pkg.empty()) {
|
||||
return false;
|
||||
}
|
||||
|
||||
// - secret (msg+k)
|
||||
pkg.insert(pkg.cend(), secret.cbegin(), secret.cend());
|
||||
|
||||
return sendToxPrivatePacket(_t, tfe, tgpe, pkg);
|
||||
}
|
||||
|
||||
bool ToxP2PRNG::send_secret_request(Contact3Handle c, ByteSpan id) {
|
||||
auto [pkg, tfe, tgpe] = prepSendPkgWithID(c, PKG::SECRET_REQUEST, id);
|
||||
if (pkg.empty()) {
|
||||
return false;
|
||||
}
|
||||
|
||||
return sendToxPrivatePacket(_t, tfe, tgpe, pkg);
|
||||
}
|
||||
|
||||
ToxP2PRNG::RngState* ToxP2PRNG::getRngSate(Contact3Handle c, ByteSpan id_bytes) {
|
||||
if (id_bytes.size != ID{}.size()) {
|
||||
return nullptr;
|
||||
@ -293,7 +454,7 @@ bool ToxP2PRNG::onToxEvent(const Tox_Event_Friend_Lossless_Packet* e) {
|
||||
const uint8_t* data = tox_event_friend_lossless_packet_get_data(e);
|
||||
const auto data_length = tox_event_friend_lossless_packet_get_data_length(e);
|
||||
|
||||
return handleFriendPacket(friend_number, data, data_length);
|
||||
return handleFriendPacket(friend_number, {data, data_length});
|
||||
}
|
||||
|
||||
bool ToxP2PRNG::onToxEvent(const Tox_Event_Group_Custom_Packet* e) {
|
||||
@ -302,7 +463,7 @@ bool ToxP2PRNG::onToxEvent(const Tox_Event_Group_Custom_Packet* e) {
|
||||
const uint8_t* data = tox_event_group_custom_packet_get_data(e);
|
||||
const auto data_length = tox_event_group_custom_packet_get_data_length(e);
|
||||
|
||||
return handleGroupPacket(group_number, peer_number, data, data_length, false);
|
||||
return handleGroupPacket(group_number, peer_number, {data, data_length}, false);
|
||||
}
|
||||
|
||||
bool ToxP2PRNG::onToxEvent(const Tox_Event_Group_Custom_Private_Packet* e) {
|
||||
@ -311,6 +472,6 @@ bool ToxP2PRNG::onToxEvent(const Tox_Event_Group_Custom_Private_Packet* e) {
|
||||
const uint8_t* data = tox_event_group_custom_private_packet_get_data(e);
|
||||
const auto data_length = tox_event_group_custom_private_packet_get_data_length(e);
|
||||
|
||||
return handleGroupPacket(group_number, peer_number, data, data_length, true);
|
||||
return handleGroupPacket(group_number, peer_number, {data, data_length}, true);
|
||||
}
|
||||
|
||||
|
||||
@ -19,6 +19,7 @@ class ToxP2PRNG : public P2PRNGI, public ToxEventI {
|
||||
ToxEventProviderI& _tep;
|
||||
ToxContactModel2& _tcm;
|
||||
|
||||
public:
|
||||
enum class PKG : uint8_t {
|
||||
INVALID = 0u,
|
||||
|
||||
@ -32,6 +33,7 @@ class ToxP2PRNG : public P2PRNGI, public ToxEventI {
|
||||
using ID = std::array<uint8_t, 32>;
|
||||
struct IDHash {size_t operator()(const ID& a) const {return (a[0] | a[1] << 1*8 | a[2] << 1*16 | a[3] << 1*24) ^ (a[31] | a[30] << 1*8);}};
|
||||
|
||||
private:
|
||||
struct RngState {
|
||||
// all contacts participating, including self
|
||||
std::vector<Contact3Handle> contacts;
|
||||
@ -53,6 +55,8 @@ class ToxP2PRNG : public P2PRNGI, public ToxEventI {
|
||||
void genFinalResult(void);
|
||||
|
||||
std::vector<uint8_t> final_result; // cached
|
||||
|
||||
P2PRNG::State getState(void) const;
|
||||
};
|
||||
entt::dense_map<ID, RngState, IDHash> _global_map;
|
||||
|
||||
@ -75,28 +79,31 @@ class ToxP2PRNG : public P2PRNGI, public ToxEventI {
|
||||
bool handlePacket(
|
||||
Contact3Handle c,
|
||||
PKG pkg_type,
|
||||
const uint8_t* data,
|
||||
const size_t data_size
|
||||
ByteSpan data
|
||||
);
|
||||
bool handleFriendPacket(
|
||||
const uint32_t friend_number,
|
||||
const uint8_t* data,
|
||||
const size_t data_size
|
||||
ByteSpan data
|
||||
);
|
||||
|
||||
bool handleGroupPacket(
|
||||
const uint32_t group_number,
|
||||
const uint32_t peer_number,
|
||||
const uint8_t* data,
|
||||
const size_t data_size,
|
||||
ByteSpan data,
|
||||
const bool _private
|
||||
);
|
||||
|
||||
bool handle_init_with_hmac(Contact3Handle c, ByteSpan id, const uint8_t* data, size_t data_size);
|
||||
bool handle_hmac(Contact3Handle c, ByteSpan id, const uint8_t* data, size_t data_size);
|
||||
bool handle_hmac_request(Contact3Handle c, ByteSpan id, const uint8_t* data, size_t data_size);
|
||||
bool handle_secret(Contact3Handle c, ByteSpan id, const uint8_t* data, size_t data_size);
|
||||
bool handle_secret_request(Contact3Handle c, ByteSpan id, const uint8_t* data, size_t data_size);
|
||||
bool handle_init_with_hmac(Contact3Handle c, ByteSpan id, ByteSpan data);
|
||||
bool handle_hmac(Contact3Handle c, ByteSpan id, ByteSpan data);
|
||||
bool handle_hmac_request(Contact3Handle c, ByteSpan id, ByteSpan data);
|
||||
bool handle_secret(Contact3Handle c, ByteSpan id, ByteSpan data);
|
||||
bool handle_secret_request(Contact3Handle c, ByteSpan id, ByteSpan data);
|
||||
|
||||
//bool send_init_with_hmac(Contact3Handle c, ByteSpan id, const uint8_t* data, size_t data_size);
|
||||
bool send_hmac(Contact3Handle c, ByteSpan id, const ByteSpan hmac);
|
||||
bool send_hmac_request(Contact3Handle c, ByteSpan id);
|
||||
bool send_secret(Contact3Handle c, ByteSpan id, const ByteSpan secret);
|
||||
bool send_secret_request(Contact3Handle c, ByteSpan id);
|
||||
|
||||
RngState* getRngSate(Contact3Handle c, ByteSpan id);
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user