mirror of
https://github.com/meshtastic/firmware.git
synced 2025-02-04 19:59:55 +00:00
76 lines
2.9 KiB
C++
76 lines
2.9 KiB
C++
#pragma once
|
|
#include "SinglePortPlugin.h"
|
|
|
|
/**
|
|
* A base class for mesh plugins that assume that they are sending/receiving one particular protobuf based
|
|
* payload. Using one particular app ID.
|
|
*
|
|
* If you are using protobufs to encode your packets (recommended) you can use this as a baseclass for your plugin
|
|
* and avoid a bunch of boilerplate code.
|
|
*/
|
|
template <class T> class ProtobufPlugin : protected SinglePortPlugin
|
|
{
|
|
const pb_msgdesc_t *fields;
|
|
|
|
public:
|
|
/** Constructor
|
|
* name is for debugging output
|
|
*/
|
|
ProtobufPlugin(const char *_name, PortNum _ourPortNum, const pb_msgdesc_t *_fields)
|
|
: SinglePortPlugin(_name, _ourPortNum), fields(_fields)
|
|
{
|
|
}
|
|
|
|
protected:
|
|
/**
|
|
* Handle a received message, the data field in the message is already decoded and is provided
|
|
*
|
|
* In general decoded will always be !NULL. But in some special applications (where you have handling packets
|
|
* for multiple port numbers, decoding will ONLY be attempted for packets where the portnum matches our expected ourPortNum.
|
|
*/
|
|
virtual bool handleReceivedProtobuf(const MeshPacket &mp, const T *decoded) = 0;
|
|
|
|
/**
|
|
* Return a mesh packet which has been preinited with a particular protobuf data payload and port number.
|
|
* You can then send this packet (after customizing any of the payload fields you might need) with
|
|
* service.sendToMesh()
|
|
*/
|
|
MeshPacket *allocDataProtobuf(const T &payload)
|
|
{
|
|
// Update our local node info with our position (even if we don't decide to update anyone else)
|
|
MeshPacket *p = allocDataPacket();
|
|
|
|
p->decoded.payload.size =
|
|
pb_encode_to_bytes(p->decoded.payload.bytes, sizeof(p->decoded.payload.bytes), fields, &payload);
|
|
// DEBUG_MSG("did encode\n");
|
|
return p;
|
|
}
|
|
|
|
private:
|
|
/** Called to handle a particular incoming message
|
|
|
|
@return true if you've guaranteed you've handled this message and no other handlers should be considered for it
|
|
*/
|
|
virtual bool handleReceived(const MeshPacket &mp)
|
|
{
|
|
// FIXME - we currently update position data in the DB only if the message was a broadcast or destined to us
|
|
// it would be better to update even if the message was destined to others.
|
|
|
|
auto &p = mp.decoded;
|
|
DEBUG_MSG("Received %s from=0x%0x, id=0x%x, portnum=%d, payloadlen=%d\n", name, mp.from, mp.id, p.portnum,
|
|
p.payload.size);
|
|
|
|
T scratch;
|
|
T *decoded = NULL;
|
|
if (mp.which_payloadVariant == MeshPacket_decoded_tag && mp.decoded.portnum == ourPortNum) {
|
|
memset(&scratch, 0, sizeof(scratch));
|
|
if (pb_decode_from_bytes(p.payload.bytes, p.payload.size, fields, &scratch))
|
|
decoded = &scratch;
|
|
else
|
|
DEBUG_MSG("Error decoding protobuf plugin!\n");
|
|
}
|
|
|
|
return handleReceivedProtobuf(mp, decoded);
|
|
}
|
|
};
|