306 lines
6.3 KiB
C++
306 lines
6.3 KiB
C++
/*
|
|
llrp.cpp
|
|
|
|
Copyright (c) 2021 Kevin Matz (kevin.matz@gmail.com)
|
|
|
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
of this software and associated documentation files (the "Software"), to deal
|
|
in the Software without restriction, including without limitation the rights
|
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
copies of the Software, and to permit persons to whom the Software is
|
|
furnished to do so, subject to the following conditions:
|
|
|
|
The above copyright notice and this permission notice shall be included in all
|
|
copies or substantial portions of the Software.
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
SOFTWARE.
|
|
*/
|
|
|
|
#include "llrp.h"
|
|
#include "rdmnet.h"
|
|
|
|
namespace RDMnet {
|
|
namespace LLRP {
|
|
|
|
/**
|
|
* @brief llrp_data::iStream
|
|
*/
|
|
void llrp_data::iStream(ACN::PDU::Stream stream)
|
|
{
|
|
/// Destination CID - The Destination CID indicates the CID of the intended
|
|
/// recipient of this PDU.
|
|
uint8_t * buffer = new uint8_t[UUID_LENGTH];
|
|
stream->read(buffer, UUID_LENGTH);
|
|
if (stream->gcount() != UUID_LENGTH)
|
|
stream->setstate(std::ios_base::failbit);
|
|
destination = UUID::uuid(buffer);
|
|
delete[] buffer;
|
|
|
|
/// Transaction Number - The Transaction Number allows an LLRP Manager to
|
|
/// associate reply messages with requests.
|
|
*stream >> transaction;
|
|
}
|
|
|
|
|
|
/**
|
|
* @brief llrp_data::oStream
|
|
* @param stream
|
|
*/
|
|
void llrp_data::oStream(ACN::PDU::Stream stream) const
|
|
{
|
|
stream->write(destination.bytes(), UUID_LENGTH);
|
|
*stream << transaction;
|
|
if (child)
|
|
child->oStream(stream);
|
|
}
|
|
|
|
|
|
/**
|
|
* @brief LLRP::Pdu::Pdu
|
|
*/
|
|
Pdu::Pdu()
|
|
: RDMnet::Pdu(4)
|
|
{
|
|
}
|
|
|
|
|
|
/**
|
|
* @brief LLRP::Pdu::iStream
|
|
* @param stream
|
|
*/
|
|
void Pdu::iStream(ACN::PDU::Stream stream)
|
|
{
|
|
RDMnet::Pdu::iStream(stream); // do base class first
|
|
if (stream->fail()) return;
|
|
if (!stream_->good()) return;
|
|
|
|
/// has no header
|
|
|
|
auto payload = new llrp_data();
|
|
payload->iStream(stream_);
|
|
data_ = payload;
|
|
|
|
switch (vector_) {
|
|
case VECTOR_LLRP_PROBE_REQUEST:
|
|
{
|
|
auto block = ACN::PDU::Block<ProbeRequest::Pdu>();
|
|
block.iStream(stream_);
|
|
if (!block.pdu->empty())
|
|
payload->child = block.pdu->front();
|
|
}
|
|
break;
|
|
case VECTOR_LLRP_PROBE_REPLY:
|
|
{
|
|
auto block = ACN::PDU::Block<ProbeReply::Pdu>();
|
|
block.iStream(stream_);
|
|
if (!block.pdu->empty())
|
|
payload->child = block.pdu->front();
|
|
}
|
|
break;
|
|
case VECTOR_LLRP_RDM_CMD:
|
|
{
|
|
auto block = ACN::PDU::Block<RdmCmd::Pdu>();
|
|
block.iStream(stream_);
|
|
if (!block.pdu->empty())
|
|
payload->child = block.pdu->front();
|
|
}
|
|
break;
|
|
default:
|
|
stream_->setstate(std::ios_base::failbit);
|
|
return;
|
|
}
|
|
|
|
if (!payload->child)
|
|
stream_->setstate(std::ios_base::failbit);
|
|
}
|
|
|
|
|
|
namespace ProbeRequest {
|
|
|
|
/**
|
|
* @brief request_data::iStream
|
|
* @param stream
|
|
*/
|
|
void request_data::iStream(ACN::PDU::Stream stream)
|
|
{
|
|
auto readUID = [stream] (RDM::UID& uid) {
|
|
*stream >> uid.device;
|
|
*stream >> uid.manufacturer;
|
|
};
|
|
|
|
readUID(lower);
|
|
readUID(upper);
|
|
*stream >> filter.value;
|
|
while (stream->good()) {
|
|
known.emplace_back(RDM::UID());
|
|
readUID(known.back());
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* @brief request_data::oStream
|
|
* @param stream
|
|
*/
|
|
void request_data::oStream(ACN::PDU::Stream stream) const
|
|
{
|
|
auto writeUID = [stream] (const RDM::UID& uid) {
|
|
*stream << uid.device;
|
|
*stream << uid.manufacturer;
|
|
};
|
|
|
|
writeUID(lower);
|
|
writeUID(upper);
|
|
*stream << filter.value;
|
|
for ( auto& id : known )
|
|
writeUID(id);
|
|
}
|
|
|
|
|
|
/**
|
|
* @brief LLRP::ProbeRequest::Pdu::Pdu
|
|
*/
|
|
Pdu::Pdu()
|
|
: RDMnet::Pdu(1)
|
|
{
|
|
}
|
|
|
|
|
|
/**
|
|
* @brief LLRP::ProbeRequest::Pdu::iStream
|
|
* @param stream
|
|
*/
|
|
void Pdu::iStream(ACN::PDU::Stream stream)
|
|
{
|
|
RDMnet::Pdu::iStream(stream); // do base class first
|
|
if (stream->fail()) return;
|
|
if (!stream_->good()) return;
|
|
|
|
/// LLRP::PDU has no header
|
|
|
|
data_ = new request_data();
|
|
data_->iStream(stream_);
|
|
}
|
|
|
|
} // namespace ProbeRequest
|
|
|
|
|
|
namespace ProbeReply {
|
|
|
|
/**
|
|
* @brief reply_data::iStream
|
|
*/
|
|
void reply_data::iStream(ACN::PDU::Stream stream)
|
|
{
|
|
*stream >> id.device;
|
|
*stream >> id.manufacturer;
|
|
stream->read(address, sizeof(address));
|
|
if (stream->gcount() != sizeof(address))
|
|
stream->setstate(std::ios_base::failbit);
|
|
*stream >> type;
|
|
}
|
|
|
|
|
|
/**
|
|
* @brief reply_data::oStream
|
|
*/
|
|
void reply_data::oStream(ACN::PDU::Stream stream) const
|
|
{
|
|
*stream << id.device;
|
|
*stream << id.manufacturer;
|
|
stream->write(address, sizeof(address));
|
|
*stream << type;
|
|
}
|
|
|
|
/**
|
|
* @brief LLRP::ProbeReply::Pdu::Pdu
|
|
*/
|
|
Pdu::Pdu()
|
|
: RDMnet::Pdu(1)
|
|
{
|
|
}
|
|
|
|
|
|
/**
|
|
* @brief LLRP::ProbeReply::Pdu::iStream
|
|
* @param stream
|
|
*/
|
|
void Pdu::iStream(ACN::PDU::Stream stream)
|
|
{
|
|
RDMnet::Pdu::iStream(stream); // do base class first
|
|
if (stream->fail()) return;
|
|
if (!stream_->good()) return;
|
|
|
|
/// has no header
|
|
|
|
data_ = new reply_data();
|
|
data_->iStream(stream_);
|
|
}
|
|
|
|
} // namespace ProbeReply
|
|
|
|
|
|
namespace RdmCmd {
|
|
|
|
/**
|
|
* @brief rdm_data::iStream
|
|
*/
|
|
void rdm_data::iStream(ACN::PDU::Stream stream)
|
|
{
|
|
auto buffer = std::vector<uint8_t>();
|
|
buffer.push_back(RDM::SC_RDM);
|
|
while (stream->good())
|
|
buffer.push_back(stream->readType<uint8_t>());
|
|
message.read(buffer);
|
|
}
|
|
|
|
|
|
/**
|
|
* @brief rdm_data::oStream
|
|
*/
|
|
void rdm_data::oStream(ACN::PDU::Stream stream) const
|
|
{
|
|
auto buffer = std::vector<uint8_t>();
|
|
message.write(buffer);
|
|
for (size_t i = 1; i < buffer.size(); i++ )
|
|
*stream << buffer.at(i);
|
|
}
|
|
|
|
|
|
/**
|
|
* @brief LLRP::RdmCmd::Pdu::Pdu
|
|
*/
|
|
Pdu::Pdu()
|
|
: RDMnet::Pdu(1)
|
|
{
|
|
}
|
|
|
|
|
|
/**
|
|
* @brief LLRP::ProbeReply::Pdu::iStream
|
|
* @param stream
|
|
*/
|
|
void Pdu::iStream(ACN::PDU::Stream stream)
|
|
{
|
|
RDMnet::Pdu::iStream(stream); // do base class first
|
|
if (stream->fail()) return;
|
|
if (!stream_->good()) return;
|
|
|
|
/// has no header
|
|
|
|
data_ = new rdm_data();
|
|
data_->iStream(stream_);
|
|
}
|
|
|
|
} // namespace RdmCmd
|
|
|
|
|
|
} // namespace LLRP
|
|
} // namespace RDMnet
|