mitm-hmac.h

#pragma once

#include "veins/modules/application/ieee80211p/DemoBaseApplLayer.h"
#include "veins/modules/application/traci/TraCIDemo11pMessage_m.h"

namespace veins {

class MITMAttackApp : public DemoBaseApplLayer {
public:
    MITMAttackApp() : sendMessageEvt(nullptr),
                      isAttacker(false),
                      hasAttacked(false), hasReceivedWarning(false),
                      hasSentMessage(false), nodeId(-1), senderNode(-1) {}
    
    virtual ~MITMAttackApp() {
        cancelAndDelete(sendMessageEvt);
    }

protected:
    virtual void initialize(int stage) override;
    virtual void handleSelfMsg(cMessage* msg) override;
    virtual void onWSM(BaseFrame1609_4* wsm) override;
    virtual void finish() override;

private:
    // Function to send a normal message with HMAC
    void sendNormalMessage();

    // Helper functions for HMAC generation and verification
    std::string generateHMAC(const std::string& message, const std::string& key);
    bool verifyHMAC(const std::string& message, const std::string& receivedHMAC, const std::string& key);

    // Simulation variables
    cMessage* sendMessageEvt;

    bool isAttacker;
    bool hasAttacked;
    bool hasReceivedWarning;
    bool hasSentMessage;
    
    int nodeId;
    int senderNode;
    double attackTime;

    // Shared secret key (optional: move to a secure location)
    const std::string sharedSecretKey = "shared_secret_key";
};

}