SNMP.h

Go to the documentation of this file.

#ifndef SNMP_H_
#define SNMP_H_
 
#include "SNMPMessage.h"
 
#include <Udp.h>
 
namespace SNMP {
 
struct Port {
    static constexpr uint16_t SNMP = 161; 
    static constexpr uint16_t Trap = 162; 
};
 
class SNMP {
    using Event = void (*)(const Message*, const IPAddress, const uint16_t);
 
public:
    bool begin(UDP& udp) {
        _udp = &udp;
        return _udp->begin(_port);
    }
 
    void loop() {
#if SNMP_STREAM
        if (_udp->parsePacket()) {
            Message *message = new Message();
            message->parse(*_udp);
            _onMessage(message, _udp->remoteIP(), _udp->remotePort());
            delete message;
        }
#else
        if (_udp->parsePacket()) {
            uint32_t length = _udp->available();
            uint8_t *buffer = static_cast<uint8_t*>(malloc(length));
            if (buffer) {
                _udp->read(buffer, length);
                Message *message = new Message();
                message->parse(buffer);
                free(buffer);
                _onMessage(message, _udp->remoteIP(), _udp->remotePort());
                delete message;
            }
        }
#endif
    }
 
    bool send(Message *message, const IPAddress ip, const uint16_t port) {
#if SNMP_STREAM
        _udp->beginPacket(ip, port);
        message->build(*_udp);
        return _udp->endPacket();
#else
        uint32_t length = message->getSize(true);
        uint8_t *buffer = static_cast<uint8_t*>(malloc(length));
        message->build(buffer);
        _udp->beginPacket(ip, port);
        _udp->write(buffer, length);
        free(buffer);
        return _udp->endPacket();
#endif
    }
 
    void onMessage(Event event) {
        _onMessage = event;
    }
 
private:
    SNMP(const uint16_t port) {
        _port = port;
    }
 
    uint16_t _port = Port::SNMP;
    UDP *_udp = nullptr;
    Event _onMessage = nullptr;
 
    friend class Agent;
    friend class Manager;
};
 
class Agent: public SNMP {
public:
    Agent() :
            SNMP(Port::SNMP) {
    }
};
 
class Manager: public SNMP {
public:
    Manager() :
            SNMP(Port::Trap) {
    }
};
 
} // namespace SNMP
 
#endif /* SNMP_H_ */