aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorMatteo Martincigh <matteo.martincigh@arm.com>2019-09-05 12:02:04 +0100
committerJim Flynn Arm <jim.flynn@arm.com>2019-09-18 16:24:39 +0000
commit42f9d9e8fdb15f418240a64a7b089df611a015a0 (patch)
tree7fa74bb038d1b58f45789fe533ba6a422e2bd6df
parent8355ec982eb3ff51a6a8042fe760138638ca550b (diff)
downloadarmnn-42f9d9e8fdb15f418240a64a7b089df611a015a0.tar.gz
IVGCVSW-3691 Implement SendCounterPacket.SendCounterDirectoryPacket() function
* Changed the signature of SendCounterDirectoryPacket to accept any ICounterDirectory object * Added helper methods to the SendCounterPacket class for creating the records * Created mock classes for testing * Added unit tests for both SendCounterDirectoryPacket and the helper methods * Added unit tests for the SWTrace utility functions * Added ReadUint8 utility function for getting single byte out of a buffer * Disabled extra sign-conversion warning in the conversion macro Change-Id: Ie2dddcd6824ed07b623f0cd78d9b7d05c5b70c39 Signed-off-by: Matteo Martincigh <matteo.martincigh@arm.com>
-rw-r--r--include/armnn/Conversion.hpp9
-rw-r--r--src/profiling/ISendCounterPacket.hpp4
-rw-r--r--src/profiling/ProfilingUtils.cpp7
-rw-r--r--src/profiling/ProfilingUtils.hpp2
-rw-r--r--src/profiling/SendCounterPacket.cpp670
-rw-r--r--src/profiling/SendCounterPacket.hpp23
-rw-r--r--src/profiling/test/ProfilingTests.cpp167
-rw-r--r--src/profiling/test/SendCounterPacketTests.cpp1347
-rw-r--r--src/profiling/test/SendCounterPacketTests.hpp243
9 files changed, 2460 insertions, 12 deletions
diff --git a/include/armnn/Conversion.hpp b/include/armnn/Conversion.hpp
index 37338eded7..e2bccd0372 100644
--- a/include/armnn/Conversion.hpp
+++ b/include/armnn/Conversion.hpp
@@ -8,7 +8,8 @@
#if __GNUC__
# define ARMNN_NO_CONVERSION_WARN_BEGIN \
_Pragma("GCC diagnostic push") \
- _Pragma("GCC diagnostic ignored \"-Wconversion\"")
+ _Pragma("GCC diagnostic ignored \"-Wconversion\"") \
+ _Pragma("GCC diagnostic ignored \"-Wsign-conversion\"")
# define ARMNN_NO_CONVERSION_WARN_END \
_Pragma("GCC diagnostic pop")
@@ -16,7 +17,8 @@
#elif __clang__
# define ARMNN_NO_CONVERSION_WARN_BEGIN \
_Pragma("clang diagnostic push") \
- _Pragma("clang diagnostic ignored \"-Wconversion\"")
+ _Pragma("clang diagnostic ignored \"-Wconversion\"") \
+ _Pragma("clang diagnostic ignored \"-Wsign-conversion\"")
# define ARMNN_NO_CONVERSION_WARN_END \
_Pragma("clang diagnostic pop")
@@ -24,7 +26,8 @@
#elif defined (_MSC_VER)
# define ARMNN_NO_CONVERSION_WARN_BEGIN \
__pragma(warning( push )) \
- __pragma(warning(disable : 4101))
+ __pragma(warning(disable : 4101)) \
+ __pragma(warning(disable : 4267))
# define ARMNN_NO_CONVERSION_WARN_END \
__pragma(warning( pop ))
diff --git a/src/profiling/ISendCounterPacket.hpp b/src/profiling/ISendCounterPacket.hpp
index a64e0dd0f3..d666f8bc36 100644
--- a/src/profiling/ISendCounterPacket.hpp
+++ b/src/profiling/ISendCounterPacket.hpp
@@ -5,7 +5,7 @@
#pragma once
-#include "CounterDirectory.hpp"
+#include "ICounterDirectory.hpp"
namespace armnn
{
@@ -24,7 +24,7 @@ public:
virtual void SendStreamMetaDataPacket() = 0;
/// Create and write a CounterDirectoryPacket from the parameters to the buffer.
- virtual void SendCounterDirectoryPacket(const CounterDirectory& counterDirectory) = 0;
+ virtual void SendCounterDirectoryPacket(const ICounterDirectory& counterDirectory) = 0;
/// Create and write a PeriodicCounterCapturePacket from the parameters to the buffer.
virtual void SendPeriodicCounterCapturePacket(uint64_t timestamp, const IndexValuePairsVector& values) = 0;
diff --git a/src/profiling/ProfilingUtils.cpp b/src/profiling/ProfilingUtils.cpp
index e356ed739e..7a8c678054 100644
--- a/src/profiling/ProfilingUtils.cpp
+++ b/src/profiling/ProfilingUtils.cpp
@@ -158,6 +158,13 @@ uint16_t ReadUint16(const unsigned char* buffer, unsigned int offset)
return static_cast<uint16_t>(value);
}
+uint8_t ReadUint8(const unsigned char* buffer, unsigned int offset)
+{
+ BOOST_ASSERT(buffer);
+
+ return buffer[offset];
+}
+
std::string GetSoftwareInfo()
{
return std::string("ArmNN");
diff --git a/src/profiling/ProfilingUtils.hpp b/src/profiling/ProfilingUtils.hpp
index 793f94d91b..fa96eea6e2 100644
--- a/src/profiling/ProfilingUtils.hpp
+++ b/src/profiling/ProfilingUtils.hpp
@@ -92,6 +92,8 @@ uint32_t ReadUint32(const unsigned char* buffer, unsigned int offset);
uint16_t ReadUint16(const unsigned char* buffer, unsigned int offset);
+uint8_t ReadUint8(const unsigned char* buffer, unsigned int offset);
+
std::string GetSoftwareInfo();
std::string GetSoftwareVersion();
diff --git a/src/profiling/SendCounterPacket.cpp b/src/profiling/SendCounterPacket.cpp
index f2172b3b7d..f3190eecde 100644
--- a/src/profiling/SendCounterPacket.cpp
+++ b/src/profiling/SendCounterPacket.cpp
@@ -8,6 +8,7 @@
#include "ProfilingUtils.hpp"
#include <armnn/Exceptions.hpp>
+#include <armnn/Conversion.hpp>
#include <boost/format.hpp>
#include <boost/numeric/conversion/cast.hpp>
@@ -175,9 +176,674 @@ void SendCounterPacket::SendStreamMetaDataPacket()
m_Buffer.Commit(totalSize);
}
-void SendCounterPacket::SendCounterDirectoryPacket(const CounterDirectory& counterDirectory)
+bool SendCounterPacket::CreateCategoryRecord(const CategoryPtr& category,
+ const Counters& counters,
+ CategoryRecord& categoryRecord,
+ std::string& errorMessage)
{
- throw armnn::UnimplementedException();
+ BOOST_ASSERT(category);
+
+ const std::string& categoryName = category->m_Name;
+ const std::vector<uint16_t> categoryCounters = category->m_Counters;
+ uint16_t deviceUid = category->m_DeviceUid;
+ uint16_t counterSetUid = category->m_CounterSetUid;
+
+ BOOST_ASSERT(!categoryName.empty());
+
+ // Utils
+ size_t uint32_t_size = sizeof(uint32_t);
+
+ // Category record word 0:
+ // 16:31 [16] device: the uid of a device element which identifies some hardware device that
+ // the category belongs to
+ // 0:15 [16] counter_set: the uid of a counter_set the category is associated with
+ uint32_t categoryRecordWord0 = (static_cast<uint32_t>(deviceUid) << 16) |
+ (static_cast<uint32_t>(counterSetUid));
+
+ // Category record word 1:
+ // 16:31 [16] event_count: number of events belonging to this category
+ // 0:15 [16] reserved: all zeros
+ uint32_t categoryRecordWord1 = static_cast<uint32_t>(categoryCounters.size()) << 16;
+
+ // Category record word 2:
+ // 0:31 [32] event_pointer_table_offset: offset from the beginning of the category data pool to
+ // the event_pointer_table
+ uint32_t categoryRecordWord2 = 0; // The offset is always zero here, as the event pointer table field is always
+ // the first item in the pool
+
+ // Convert the device name into a SWTrace namestring
+ std::vector<uint32_t> categoryNameBuffer;
+ if (!StringToSwTraceString<SwTraceNameCharPolicy>(categoryName, categoryNameBuffer))
+ {
+ errorMessage = boost::str(boost::format("Cannot convert the name of category \"%1%\" to an SWTrace namestring")
+ % categoryName);
+ return false;
+ }
+
+ // Process the event records
+ size_t counterCount = categoryCounters.size();
+ std::vector<EventRecord> eventRecords(counterCount);
+ std::vector<uint32_t> eventRecordOffsets(counterCount, 0);
+ size_t eventRecordsSize = 0;
+ ARMNN_NO_CONVERSION_WARN_BEGIN
+ uint32_t eventRecordsOffset = (eventRecords.size() + categoryNameBuffer.size()) * uint32_t_size;
+ ARMNN_NO_CONVERSION_WARN_END
+ for (size_t counterIndex = 0, eventRecordIndex = 0, eventRecordOffsetIndex = 0;
+ counterIndex < counterCount;
+ counterIndex++, eventRecordIndex++, eventRecordOffsetIndex++)
+ {
+ uint16_t counterUid = categoryCounters.at(counterIndex);
+ auto it = counters.find(counterUid);
+ BOOST_ASSERT(it != counters.end());
+ const CounterPtr& counter = it->second;
+
+ EventRecord& eventRecord = eventRecords.at(eventRecordIndex);
+ if (!CreateEventRecord(counter, eventRecord, errorMessage))
+ {
+ return false;
+ }
+
+ // Update the total size in words of the event records
+ eventRecordsSize += eventRecord.size();
+
+ // Add the event record offset to the event pointer table offset field
+ eventRecordOffsets[eventRecordOffsetIndex] = eventRecordsOffset;
+ ARMNN_NO_CONVERSION_WARN_BEGIN
+ eventRecordsOffset += eventRecord.size() * uint32_t_size;
+ ARMNN_NO_CONVERSION_WARN_END
+ }
+
+ // Category record word 3:
+ // 0:31 [32] name_offset (offset from the beginning of the category data pool to the name field)
+ ARMNN_NO_CONVERSION_WARN_BEGIN
+ uint32_t categoryRecordWord3 = eventRecordOffsets.size() * uint32_t_size;
+ ARMNN_NO_CONVERSION_WARN_END
+
+ // Calculate the size in words of the category record
+ size_t categoryRecordSize = 4u + // The size of the fixed part (device + counter_set + event_count + reserved +
+ // event_pointer_table_offset + name_offset)
+ eventRecordOffsets.size() + // The size of the variable part (the event pointer table +
+ categoryNameBuffer.size() + // and the category name including the null-terminator +
+ eventRecordsSize; // the event records)
+
+ // Allocate the necessary space for the category record
+ categoryRecord.resize(categoryRecordSize);
+
+ // Create the category record
+ categoryRecord[0] = categoryRecordWord0; // device + counter_set
+ categoryRecord[1] = categoryRecordWord1; // event_count + reserved
+ categoryRecord[2] = categoryRecordWord2; // event_pointer_table_offset
+ categoryRecord[3] = categoryRecordWord3; // name_offset
+ auto offset = categoryRecord.begin() + 4u;
+ std::copy(eventRecordOffsets.begin(), eventRecordOffsets.end(), offset); // event_pointer_table
+ ARMNN_NO_CONVERSION_WARN_BEGIN
+ offset += eventRecordOffsets.size();
+ ARMNN_NO_CONVERSION_WARN_END
+ std::copy(categoryNameBuffer.begin(), categoryNameBuffer.end(), offset); // name
+ ARMNN_NO_CONVERSION_WARN_BEGIN
+ offset += categoryNameBuffer.size();
+ ARMNN_NO_CONVERSION_WARN_END
+ for (const EventRecord& eventRecord : eventRecords)
+ {
+ std::copy(eventRecord.begin(), eventRecord.end(), offset); // event_record
+ ARMNN_NO_CONVERSION_WARN_BEGIN
+ offset += eventRecord.size();
+ ARMNN_NO_CONVERSION_WARN_END
+ }
+
+ return true;
+}
+
+bool SendCounterPacket::CreateDeviceRecord(const DevicePtr& device,
+ DeviceRecord& deviceRecord,
+ std::string& errorMessage)
+{
+ BOOST_ASSERT(device);
+
+ uint16_t deviceUid = device->m_Uid;
+ const std::string& deviceName = device->m_Name;
+ uint16_t deviceCores = device->m_Cores;
+
+ BOOST_ASSERT(!deviceName.empty());
+
+ // Device record word 0:
+ // 16:31 [16] uid: the unique identifier for the device
+ // 0:15 [16] cores: the number of individual streams of counters for one or more cores of some device
+ uint32_t deviceRecordWord0 = (static_cast<uint32_t>(deviceUid) << 16) |
+ (static_cast<uint32_t>(deviceCores));
+
+ // Device record word 1:
+ // 0:31 [32] name_offset: offset from the beginning of the device record pool to the name field
+ uint32_t deviceRecordWord1 = 0; // The offset is always zero here, as the name field is always
+ // the first (and only) item in the pool
+
+ // Convert the device name into a SWTrace string
+ std::vector<uint32_t> deviceNameBuffer;
+ if (!StringToSwTraceString<SwTraceCharPolicy>(deviceName, deviceNameBuffer))
+ {
+ errorMessage = boost::str(boost::format("Cannot convert the name of device %1% (\"%2%\") to an SWTrace string")
+ % deviceUid
+ % deviceName);
+ return false;
+ }
+
+ // Calculate the size in words of the device record
+ size_t deviceRecordSize = 2u + // The size of the fixed part (uid + cores + name_offset)
+ deviceNameBuffer.size(); // The size of the variable part (the device name including
+ // the null-terminator)
+
+ // Allocate the necessary space for the device record
+ deviceRecord.resize(deviceRecordSize);
+
+ // Create the device record
+ deviceRecord[0] = deviceRecordWord0; // uid + core
+ deviceRecord[1] = deviceRecordWord1; // name_offset
+ auto offset = deviceRecord.begin() + 2u;
+ std::copy(deviceNameBuffer.begin(), deviceNameBuffer.end(), offset); // name
+
+ return true;
+}
+
+bool SendCounterPacket::CreateCounterSetRecord(const CounterSetPtr& counterSet,
+ CounterSetRecord& counterSetRecord,
+ std::string& errorMessage)
+{
+ BOOST_ASSERT(counterSet);
+
+ uint16_t counterSetUid = counterSet->m_Uid;
+ const std::string& counterSetName = counterSet->m_Name;
+ uint16_t counterSetCount = counterSet->m_Count;
+
+ BOOST_ASSERT(!counterSetName.empty());
+
+ // Counter set record word 0:
+ // 16:31 [16] uid: the unique identifier for the counter_set
+ // 0:15 [16] count: the number of counters which can be active in this set at any one time
+ uint32_t counterSetRecordWord0 = (static_cast<uint32_t>(counterSetUid) << 16) |
+ (static_cast<uint32_t>(counterSetCount));
+
+ // Counter set record word 1:
+ // 0:31 [32] name_offset: offset from the beginning of the counter set pool to the name field
+ uint32_t counterSetRecordWord1 = 0; // The offset is always zero here, as the name field is always
+ // the first (and only) item in the pool
+
+ // Convert the device name into a SWTrace namestring
+ std::vector<uint32_t> counterSetNameBuffer;
+ if (!StringToSwTraceString<SwTraceNameCharPolicy>(counterSet->m_Name, counterSetNameBuffer))
+ {
+ errorMessage = boost::str(boost::format("Cannot convert the name of counter set %1% (\"%2%\") to "
+ "an SWTrace namestring")
+ % counterSetUid
+ % counterSetName);
+ return false;
+ }
+
+ // Calculate the size in words of the counter set record
+ size_t counterSetRecordSize = 2u + // The size of the fixed part (uid + cores + name_offset)
+ counterSetNameBuffer.size(); // The size of the variable part (the counter set name
+ // including the null-terminator)
+
+ // Allocate the space for the counter set record
+ counterSetRecord.resize(counterSetRecordSize);
+
+ // Create the counter set record
+ counterSetRecord[0] = counterSetRecordWord0; // uid + core
+ counterSetRecord[1] = counterSetRecordWord1; // name_offset
+ auto offset = counterSetRecord.begin() + 2u;
+ std::copy(counterSetNameBuffer.begin(), counterSetNameBuffer.end(), offset); // name
+
+ return true;
+}
+
+bool SendCounterPacket::CreateEventRecord(const CounterPtr& counter,
+ EventRecord& eventRecord,
+ std::string& errorMessage)
+{
+ BOOST_ASSERT(counter);
+
+ uint16_t counterUid = counter->m_Uid;
+ uint16_t maxCounterUid = counter->m_MaxCounterUid;
+ uint16_t deviceUid = counter->m_DeviceUid;
+ uint16_t counterSetUid = counter->m_CounterSetUid;
+ uint16_t counterClass = counter->m_Class;
+ uint16_t counterInterpolation = counter->m_Interpolation;
+ double counterMultiplier = counter->m_Multiplier;
+ const std::string& counterName = counter->m_Name;
+ const std::string& counterDescription = counter->m_Description;
+ const std::string& counterUnits = counter->m_Units;
+
+ BOOST_ASSERT(counterClass == 0 || counterClass == 1);
+ BOOST_ASSERT(counterInterpolation == 0 || counterInterpolation == 1);
+ BOOST_ASSERT(counterMultiplier);
+
+ // Utils
+ size_t uint32_t_size = sizeof(uint32_t);
+
+ // Event record word 0:
+ // 16:31 [16] max_counter_uid: if the device this event is associated with has more than one core and there
+ // is one of these counters per core this value will be set to
+ // (counter_uid + cores (from device_record)) - 1.
+ // If there is only a single core then this value will be the same as
+ // the counter_uid value
+ // 0:15 [16] count_uid: unique ID for the counter. Must be unique across all counters in all categories
+ uint32_t eventRecordWord0 = (static_cast<uint32_t>(maxCounterUid) << 16) |
+ (static_cast<uint32_t>(counterUid));
+
+ // Event record word 1:
+ // 16:31 [16] device: UID of the device this event is associated with. Set to zero if the event is NOT
+ // associated with a device
+ // 0:15 [16] counter_set: UID of the counter_set this event is associated with. Set to zero if the event
+ // is NOT associated with a counter_set
+ uint32_t eventRecordWord1 = (static_cast<uint32_t>(deviceUid) << 16) |
+ (static_cast<uint32_t>(counterSetUid));
+
+ // Event record word 2:
+ // 16:31 [16] class: type describing how to treat each data point in a stream of data points
+ // 0:15 [16] interpolation: type describing how to interpolate each data point in a stream of data points
+ uint32_t eventRecordWord2 = (static_cast<uint32_t>(counterClass) << 16) |
+ (static_cast<uint32_t>(counterInterpolation));
+
+ // Event record word 3-4:
+ // 0:63 [64] multiplier: internal data stream is represented as integer values, this allows scaling of
+ // those values as if they are fixed point numbers. Zero is not a valid value
+ uint32_t multiplier[2] = { 0u, 0u };
+ BOOST_ASSERT(sizeof(counterMultiplier) == sizeof(multiplier));
+ std::memcpy(multiplier, &counterMultiplier, sizeof(multiplier));
+ uint32_t eventRecordWord3 = multiplier[0];
+ uint32_t eventRecordWord4 = multiplier[1];
+
+ // Event record word 5:
+ // 0:31 [32] name_offset: offset from the beginning of the event record pool to the name field
+ uint32_t eventRecordWord5 = 0; // The offset is always zero here, as the name field is always
+ // the first item in the pool
+
+ // Convert the counter name into a SWTrace string
+ std::vector<uint32_t> counterNameBuffer;
+ if (!StringToSwTraceString<SwTraceCharPolicy>(counterName, counterNameBuffer))
+ {
+ errorMessage = boost::str(boost::format("Cannot convert the name of counter %1% (name: \"%2%\") "
+ "to an SWTrace string")
+ % counterUid
+ % counterName);
+ return false;
+ }
+
+ // Event record word 6:
+ // 0:31 [32] description_offset: offset from the beginning of the event record pool to the description field
+ ARMNN_NO_CONVERSION_WARN_BEGIN
+ // The size of the name buffer in bytes
+ uint32_t eventRecordWord6 = counterNameBuffer.size() * uint32_t_size;
+ ARMNN_NO_CONVERSION_WARN_END
+
+ // Convert the counter description into a SWTrace string
+ std::vector<uint32_t> counterDescriptionBuffer;
+ if (!StringToSwTraceString<SwTraceCharPolicy>(counterDescription, counterDescriptionBuffer))
+ {
+ errorMessage = boost::str(boost::format("Cannot convert the description of counter %1% (description: \"%2%\") "
+ "to an SWTrace string")
+ % counterUid
+ % counterName);
+ return false;
+ }
+
+ // Event record word 7:
+ // 0:31 [32] units_offset: (optional) offset from the beginning of the event record pool to the units field.
+ // An offset value of zero indicates this field is not provided
+ bool includeUnits = !counterUnits.empty();
+ ARMNN_NO_CONVERSION_WARN_BEGIN
+ // The size of the description buffer in bytes
+ uint32_t eventRecordWord7 = includeUnits ? eventRecordWord6 + counterDescriptionBuffer.size() * uint32_t_size : 0;
+ ARMNN_NO_CONVERSION_WARN_END
+
+ // Convert the counter units into a SWTrace namestring (optional)
+ std::vector<uint32_t> counterUnitsBuffer;
+ if (includeUnits)
+ {
+ // Convert the counter units into a SWTrace namestring
+ if (!StringToSwTraceString<SwTraceNameCharPolicy>(counterUnits, counterUnitsBuffer))
+ {
+ errorMessage = boost::str(boost::format("Cannot convert the units of counter %1% (units: \"%2%\") "
+ "to an SWTrace string")
+ % counterUid
+ % counterName);
+ return false;
+ }
+ }
+
+ // Calculate the size in words of the event record
+ size_t eventRecordSize = 8u + // The size of the fixed part (counter_uid + max_counter_uid + device +
+ // counter_set + class + interpolation +
+ // multiplier + name_offset + description_offset +
+ // units_offset)
+ counterNameBuffer.size() + // The size of the variable part (the counter name,
+ counterDescriptionBuffer.size() + // description and units including the null-terminator)
+ counterUnitsBuffer.size();
+
+ // Allocate the space for the event record
+ eventRecord.resize(eventRecordSize);
+
+ // Create the event record
+ eventRecord[0] = eventRecordWord0; // max_counter_uid + counter_uid
+ eventRecord[1] = eventRecordWord1; // device + counter_set
+ eventRecord[2] = eventRecordWord2; // class + interpolation
+ eventRecord[3] = eventRecordWord3; // multiplier
+ eventRecord[4] = eventRecordWord4; // multiplier
+ eventRecord[5] = eventRecordWord5; // name_offset
+ eventRecord[6] = eventRecordWord6; // description_offset
+ eventRecord[7] = eventRecordWord7; // units_offset
+ auto offset = eventRecord.begin() + 8u;
+ std::copy(counterNameBuffer.begin(), counterNameBuffer.end(), offset); // name
+ ARMNN_NO_CONVERSION_WARN_BEGIN
+ offset += counterNameBuffer.size();
+ ARMNN_NO_CONVERSION_WARN_END
+ std::copy(counterDescriptionBuffer.begin(), counterDescriptionBuffer.end(), offset); // description
+ if (includeUnits)
+ {
+ ARMNN_NO_CONVERSION_WARN_BEGIN
+ offset += counterDescriptionBuffer.size();
+ ARMNN_NO_CONVERSION_WARN_END
+ std::copy(counterUnitsBuffer.begin(), counterUnitsBuffer.end(), offset); // units
+ }
+
+ return true;
+}
+
+void SendCounterPacket::SendCounterDirectoryPacket(const ICounterDirectory& counterDirectory)
+{
+ // Get the amount of data that needs to be put into the packet
+ uint16_t categoryCount = counterDirectory.GetCategoryCount();
+ uint16_t deviceCount = counterDirectory.GetDeviceCount();
+ uint16_t counterSetCount = counterDirectory.GetCounterSetCount();
+
+ // Utils
+ size_t uint32_t_size = sizeof(uint32_t);
+ size_t packetHeaderSize = 2u;
+ size_t bodyHeaderSize = 6u;
+
+ // Initialize the offset for the pointer tables
+ uint32_t pointerTableOffset = 0;
+
+ // --------------
+ // Device records
+ // --------------
+
+ // Process device records
+ std::vector<DeviceRecord> deviceRecords(deviceCount);
+ const Devices& devices = counterDirectory.GetDevices();
+ std::vector<uint32_t> deviceRecordOffsets(deviceCount, 0); // device_records_pointer_table
+ size_t deviceRecordsSize = 0;
+ size_t deviceIndex = 0;
+ size_t deviceRecordOffsetIndex = 0;
+ for (auto it = devices.begin(); it != devices.end(); it++)
+ {
+ const DevicePtr& device = it->second;
+ DeviceRecord& deviceRecord = deviceRecords.at(deviceIndex);
+
+ std::string errorMessage;
+ if (!CreateDeviceRecord(device, deviceRecord, errorMessage))
+ {
+ CancelOperationAndThrow<RuntimeException>(errorMessage);
+ }
+
+ // Update the total size in words of the device records
+ deviceRecordsSize += deviceRecord.size();
+
+ // Add the device record offset to the device records pointer table offset field
+ deviceRecordOffsets[deviceRecordOffsetIndex] = pointerTableOffset;
+ ARMNN_NO_CONVERSION_WARN_BEGIN
+ pointerTableOffset += deviceRecord.size() * uint32_t_size;
+ ARMNN_NO_CONVERSION_WARN_END
+
+ deviceIndex++;
+ deviceRecordOffsetIndex++;
+ }
+
+ // -------------------
+ // Counter set records
+ // -------------------
+
+ // Process counter set records
+ std::vector<CounterSetRecord> counterSetRecords(counterSetCount);
+ const CounterSets& counterSets = counterDirectory.GetCounterSets();
+ std::vector<uint32_t> counterSetRecordOffsets(counterSetCount, 0); // counter_set_records_pointer_table
+ size_t counterSetRecordsSize = 0;
+ size_t counterSetIndex = 0;
+ size_t counterSetRecordOffsetIndex = 0;
+ for (auto it = counterSets.begin(); it != counterSets.end(); it++)
+ {
+ const CounterSetPtr& counterSet = it->second;
+ CounterSetRecord& counterSetRecord = counterSetRecords.at(counterSetIndex);
+
+ std::string errorMessage;
+ if (!CreateCounterSetRecord(counterSet, counterSetRecord, errorMessage))
+ {
+ CancelOperationAndThrow<RuntimeException>(errorMessage);
+ }
+
+ // Update the total size in words of the counter set records
+ counterSetRecordsSize += counterSetRecord.size();
+
+ // Add the counter set record offset to the counter set records pointer table offset field
+ counterSetRecordOffsets[counterSetRecordOffsetIndex] = pointerTableOffset;
+ ARMNN_NO_CONVERSION_WARN_BEGIN
+ pointerTableOffset += counterSetRecord.size() * uint32_t_size;
+ ARMNN_NO_CONVERSION_WARN_END
+
+ counterSetIndex++;
+ counterSetRecordOffsetIndex++;
+ }
+
+ // ----------------
+ // Category records
+ // ----------------
+
+ // Process category records
+ std::vector<CategoryRecord> categoryRecords(categoryCount);
+ const Categories& categories = counterDirectory.GetCategories();
+ std::vector<uint32_t> categoryRecordOffsets(categoryCount, 0); // category_records_pointer_table
+ size_t categoryRecordsSize = 0;
+ size_t categoryIndex = 0;
+ size_t categoryRecordOffsetIndex = 0;
+ for (auto it = categories.begin(); it != categories.end(); it++)
+ {
+ const CategoryPtr& category = *it;
+ CategoryRecord& categoryRecord = categoryRecords.at(categoryIndex);
+
+ std::string errorMessage;
+ if (!CreateCategoryRecord(category, counterDirectory.GetCounters(), categoryRecord, errorMessage))
+ {
+ CancelOperationAndThrow<RuntimeException>(errorMessage);
+ }
+
+ // Update the total size in words of the category records
+ categoryRecordsSize += categoryRecord.size();
+
+ // Add the category record offset to the category records pointer table offset field
+ categoryRecordOffsets[categoryRecordOffsetIndex] = pointerTableOffset;
+ ARMNN_NO_CONVERSION_WARN_BEGIN
+ pointerTableOffset += categoryRecord.size() * uint32_t_size;
+ ARMNN_NO_CONVERSION_WARN_END
+
+ categoryIndex++;
+ categoryRecordOffsetIndex++;
+ }
+
+ // Calculate the size in words of the counter directory packet
+ size_t counterDirectoryPacketSize =
+ packetHeaderSize + // The size of the packet header
+ bodyHeaderSize + // The size of the body header
+ deviceRecordOffsets.size() + // The size of the device records pointer table
+ counterSetRecordOffsets.size() + // The size of counter set pointer table
+ categoryRecordOffsets.size() + // The size of category records pointer table
+ deviceRecordsSize + // The total size of the device records
+ counterSetRecordsSize + // The total size of the counter set records
+ categoryRecordsSize; // The total size of the category records
+
+ // Allocate the necessary space for the counter directory packet
+ std::vector<uint32_t> counterDirectoryPacket(counterDirectoryPacketSize, 0);
+
+ // -------------
+ // Packet header
+ // -------------
+
+ // Packet header word 0:
+ // 26:31 [6] packet_family: control Packet Family
+ // 16:25 [10] packet_id: packet identifier
+ // 8:15 [8] reserved: all zeros
+ // 0:7 [8] reserved: all zeros
+ uint32_t packetFamily = 0;
+ uint32_t packetId = 2;
+ uint32_t packetHeaderWord0 = ((packetFamily & 0x3F) << 26) | ((packetId & 0x3FF) << 16);
+
+ // Packet header word 1:
+ // 0:31 [32] data_length: length of data, in bytes
+ ARMNN_NO_CONVERSION_WARN_BEGIN
+ uint32_t packetHeaderWord1 = counterDirectoryPacketSize * uint32_t_size;
+ ARMNN_NO_CONVERSION_WARN_END
+
+ // Create the packet header
+ uint32_t packetHeader[2]
+ {
+ packetHeaderWord0, // packet_family + packet_id + reserved + reserved
+ packetHeaderWord1 // data_length
+ };
+
+ // -----------
+ // Body header
+ // -----------
+
+ // Body header word 0:
+ // 16:31 [16] device_records_count: number of entries in the device_records_pointer_table
+ // 0:15 [16] reserved: all zeros
+ uint32_t bodyHeaderWord0 = static_cast<uint32_t>(deviceCount) << 16;
+
+ // Body header word 1:
+ // 0:31 [32] device_records_pointer_table_offset: offset to the device_records_pointer_table
+ uint32_t bodyHeaderWord1 = 0; // The offset is always zero here, as the device record pointer table field is always
+ // the first item in the pool
+
+ // Body header word 2:
+ // 16:31 [16] counter_set_count: number of entries in the counter_set_pointer_table
+ // 0:15 [16] reserved: all zeros
+ uint32_t bodyHeaderWord2 = static_cast<uint32_t>(counterSetCount) << 16;
+
+ // Body header word 3:
+ // 0:31 [32] counter_set_pointer_table_offset: offset to the counter_set_pointer_table
+ ARMNN_NO_CONVERSION_WARN_BEGIN
+ uint32_t bodyHeaderWord3 = deviceRecordOffsets.size() * uint32_t_size; // The size of the device records pointer
+ ARMNN_NO_CONVERSION_WARN_END // table
+
+ // Body header word 4:
+ // 16:31 [16] categories_count: number of entries in the categories_pointer_table
+ // 0:15 [16] reserved: all zeros
+ uint32_t bodyHeaderWord4 = static_cast<uint32_t>(categoryCount) << 16;
+
+ // Body header word 3:
+ // 0:31 [32] categories_pointer_table_offset: offset to the categories_pointer_table
+ ARMNN_NO_CONVERSION_WARN_BEGIN
+ uint32_t bodyHeaderWord5 = deviceRecordOffsets.size() * uint32_t_size + // The size of the device records pointer
+ counterSetRecordOffsets.size() * uint32_t_size; // table, plus the size of the counter
+ ARMNN_NO_CONVERSION_WARN_END // set pointer table
+
+ // Create the body header
+ uint32_t bodyHeader[6]
+ {
+ bodyHeaderWord0, // device_records_count + reserved
+ bodyHeaderWord1, // device_records_pointer_table_offset
+ bodyHeaderWord2, // counter_set_count + reserved
+ bodyHeaderWord3, // counter_set_pointer_table_offset
+ bodyHeaderWord4, // categories_count + reserved
+ bodyHeaderWord5 // categories_pointer_table_offset
+ };
+
+ // Create the counter directory packet
+ auto counterDirectoryPacketOffset = counterDirectoryPacket.begin();
+ // packet_header
+ std::copy(packetHeader, packetHeader + packetHeaderSize, counterDirectoryPacketOffset);
+ ARMNN_NO_CONVERSION_WARN_BEGIN
+ counterDirectoryPacketOffset += packetHeaderSize;
+ ARMNN_NO_CONVERSION_WARN_END
+ // body_header
+ std::copy(bodyHeader, bodyHeader + bodyHeaderSize, counterDirectoryPacketOffset);
+ ARMNN_NO_CONVERSION_WARN_BEGIN
+ counterDirectoryPacketOffset += bodyHeaderSize;
+ ARMNN_NO_CONVERSION_WARN_END
+ // device_records_pointer_table
+ std::copy(deviceRecordOffsets.begin(), deviceRecordOffsets.end(), counterDirectoryPacketOffset);
+ ARMNN_NO_CONVERSION_WARN_BEGIN
+ counterDirectoryPacketOffset += deviceRecordOffsets.size();
+ ARMNN_NO_CONVERSION_WARN_END
+ // counter_set_pointer_table
+ std::copy(counterSetRecordOffsets.begin(), counterSetRecordOffsets.end(), counterDirectoryPacketOffset);
+ ARMNN_NO_CONVERSION_WARN_BEGIN
+ counterDirectoryPacketOffset += counterSetRecordOffsets.size();
+ ARMNN_NO_CONVERSION_WARN_END
+ // category_pointer_table
+ std::copy(categoryRecordOffsets.begin(), categoryRecordOffsets.end(), counterDirectoryPacketOffset);
+ ARMNN_NO_CONVERSION_WARN_BEGIN
+ counterDirectoryPacketOffset += categoryRecordOffsets.size();
+ ARMNN_NO_CONVERSION_WARN_END
+ // device_records
+ for (const DeviceRecord& deviceRecord : deviceRecords)
+ {
+ std::copy(deviceRecord.begin(), deviceRecord.end(), counterDirectoryPacketOffset); // device_record
+ ARMNN_NO_CONVERSION_WARN_BEGIN
+ counterDirectoryPacketOffset += deviceRecord.size();
+ ARMNN_NO_CONVERSION_WARN_END
+ }
+ // counter_set_records
+ for (const CounterSetRecord& counterSetRecord : counterSetRecords)
+ {
+ std::copy(counterSetRecord.begin(), counterSetRecord.end(), counterDirectoryPacketOffset); // counter_set_record
+ ARMNN_NO_CONVERSION_WARN_BEGIN
+ counterDirectoryPacketOffset += counterSetRecord.size();
+ ARMNN_NO_CONVERSION_WARN_END
+ }
+ // category_records
+ for (const CategoryRecord& categoryRecord : categoryRecords)
+ {
+ std::copy(categoryRecord.begin(), categoryRecord.end(), counterDirectoryPacketOffset); // category_record
+ ARMNN_NO_CONVERSION_WARN_BEGIN
+ counterDirectoryPacketOffset += categoryRecord.size();
+ ARMNN_NO_CONVERSION_WARN_END
+ }
+
+ // Calculate the total size in bytes of the counter directory packet
+ ARMNN_NO_CONVERSION_WARN_BEGIN
+ uint32_t totalSize = counterDirectoryPacketSize * uint32_t_size;
+ ARMNN_NO_CONVERSION_WARN_END
+
+ // Reserve space in the buffer for the packet
+ uint32_t reserved = 0;
+ unsigned char* writeBuffer = m_Buffer.Reserve(totalSize, reserved);
+
+ // Check that the reserved buffer size is enough to hold the counter directory packet
+ if (reserved < totalSize)
+ {
+ CancelOperationAndThrow<BufferExhaustion>(
+ boost::str(boost::format("No space left in buffer. Unable to reserve (%1%) bytes")
+ % totalSize));
+ }
+
+ // Check the buffer handle is valid
+ if (writeBuffer == nullptr)
+ {
+ CancelOperationAndThrow<RuntimeException>("Error reserving buffer memory");
+ }
+
+ // Offset for writing to the buffer
+ uint32_t offset = 0;
+
+ // Write the counter directory packet to the buffer
+ for (uint32_t counterDirectoryPacketWord : counterDirectoryPacket)
+ {
+ WriteUint32(writeBuffer, offset, counterDirectoryPacketWord);
+ ARMNN_NO_CONVERSION_WARN_BEGIN
+ offset += uint32_t_size;
+ ARMNN_NO_CONVERSION_WARN_END
+ }
+
+ m_Buffer.Commit(totalSize);
}
void SendCounterPacket::SendPeriodicCounterCapturePacket(uint64_t timestamp, const IndexValuePairsVector& values)
diff --git a/src/profiling/SendCounterPacket.hpp b/src/profiling/SendCounterPacket.hpp
index 3238b351a3..2a2d5d4313 100644
--- a/src/profiling/SendCounterPacket.hpp
+++ b/src/profiling/SendCounterPacket.hpp
@@ -18,6 +18,11 @@ namespace profiling
class SendCounterPacket : public ISendCounterPacket
{
public:
+ using CategoryRecord = std::vector<uint32_t>;
+ using DeviceRecord = std::vector<uint32_t>;
+ using CounterSetRecord = std::vector<uint32_t>;
+ using EventRecord = std::vector<uint32_t>;
+
using IndexValuePairsVector = std::vector<std::pair<uint16_t, uint32_t>>;
SendCounterPacket(IBufferWrapper& buffer)
@@ -27,7 +32,7 @@ public:
void SendStreamMetaDataPacket() override;
- void SendCounterDirectoryPacket(const CounterDirectory& counterDirectory) override;
+ void SendCounterDirectoryPacket(const ICounterDirectory& counterDirectory) override;
void SendPeriodicCounterCapturePacket(uint64_t timestamp, const IndexValuePairsVector& values) override;
@@ -52,6 +57,22 @@ private:
IBufferWrapper& m_Buffer;
bool m_ReadyToRead;
+
+protected:
+ // Helper methods, protected for testing
+ bool CreateCategoryRecord(const CategoryPtr& category,
+ const Counters& counters,
+ CategoryRecord& categoryRecord,
+ std::string& errorMessage);
+ bool CreateDeviceRecord(const DevicePtr& device,
+ DeviceRecord& deviceRecord,
+ std::string& errorMessage);
+ bool CreateCounterSetRecord(const CounterSetPtr& counterSet,
+ CounterSetRecord& counterSetRecord,
+ std::string& errorMessage);
+ bool CreateEventRecord(const CounterPtr& counter,
+ EventRecord& eventRecord,
+ std::string& errorMessage);
};
} // namespace profiling
diff --git a/src/profiling/test/ProfilingTests.cpp b/src/profiling/test/ProfilingTests.cpp
index 51dbb07a58..e97068fbb4 100644
--- a/src/profiling/test/ProfilingTests.cpp
+++ b/src/profiling/test/ProfilingTests.cpp
@@ -1647,7 +1647,6 @@ BOOST_AUTO_TEST_CASE(CounterSelectionCommandHandlerParseData)
BOOST_TEST(((headerWord0 >> 16) & 0x3FF) == 4); // packet id
BOOST_TEST(headerWord1 == 4); // data lenght
BOOST_TEST(period == 11); // capture period
-
}
BOOST_AUTO_TEST_CASE(CheckSocketProfilingConnection)
@@ -1656,4 +1655,170 @@ BOOST_AUTO_TEST_CASE(CheckSocketProfilingConnection)
BOOST_CHECK_THROW(new SocketProfilingConnection(), armnn::Exception);
}
+BOOST_AUTO_TEST_CASE(SwTraceIsValidCharTest)
+{
+ // Only ASCII 7-bit encoding supported
+ for (unsigned char c = 0; c < 128; c++)
+ {
+ BOOST_CHECK(SwTraceCharPolicy::IsValidChar(c));
+ }
+
+ // Not ASCII
+ for (unsigned char c = 255; c >= 128; c++)
+ {
+ BOOST_CHECK(!SwTraceCharPolicy::IsValidChar(c));
+ }
+}
+
+BOOST_AUTO_TEST_CASE(SwTraceIsValidNameCharTest)
+{
+ // Only alpha-numeric and underscore ASCII 7-bit encoding supported
+ const unsigned char validChars[] = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789_";
+ for (unsigned char i = 0; i < sizeof(validChars) / sizeof(validChars[0]) - 1; i++)
+ {
+ BOOST_CHECK(SwTraceNameCharPolicy::IsValidChar(validChars[i]));
+ }
+
+ // Non alpha-numeric chars
+ for (unsigned char c = 0; c < 48; c++)
+ {
+ BOOST_CHECK(!SwTraceNameCharPolicy::IsValidChar(c));
+ }
+ for (unsigned char c = 58; c < 65; c++)
+ {
+ BOOST_CHECK(!SwTraceNameCharPolicy::IsValidChar(c));
+ }
+ for (unsigned char c = 91; c < 95; c++)
+ {
+ BOOST_CHECK(!SwTraceNameCharPolicy::IsValidChar(c));
+ }
+ for (unsigned char c = 96; c < 97; c++)
+ {
+ BOOST_CHECK(!SwTraceNameCharPolicy::IsValidChar(c));
+ }
+ for (unsigned char c = 123; c < 128; c++)
+ {
+ BOOST_CHECK(!SwTraceNameCharPolicy::IsValidChar(c));
+ }
+
+ // Not ASCII
+ for (unsigned char c = 255; c >= 128; c++)
+ {
+ BOOST_CHECK(!SwTraceNameCharPolicy::IsValidChar(c));
+ }
+}
+
+BOOST_AUTO_TEST_CASE(IsValidSwTraceStringTest)
+{
+ // Valid SWTrace strings
+ BOOST_CHECK(IsValidSwTraceString<SwTraceCharPolicy>(""));
+ BOOST_CHECK(IsValidSwTraceString<SwTraceCharPolicy>("_"));
+ BOOST_CHECK(IsValidSwTraceString<SwTraceCharPolicy>("0123"));
+ BOOST_CHECK(IsValidSwTraceString<SwTraceCharPolicy>("valid_string"));
+ BOOST_CHECK(IsValidSwTraceString<SwTraceCharPolicy>("VALID_string_456"));
+ BOOST_CHECK(IsValidSwTraceString<SwTraceCharPolicy>(" "));
+ BOOST_CHECK(IsValidSwTraceString<SwTraceCharPolicy>("valid string"));
+ BOOST_CHECK(IsValidSwTraceString<SwTraceCharPolicy>("!$%"));
+ BOOST_CHECK(IsValidSwTraceString<SwTraceCharPolicy>("valid|\\~string#123"));
+
+ // Invalid SWTrace strings
+ BOOST_CHECK(!IsValidSwTraceString<SwTraceCharPolicy>("€£"));
+ BOOST_CHECK(!IsValidSwTraceString<SwTraceCharPolicy>("invalid‡string"));
+ BOOST_CHECK(!IsValidSwTraceString<SwTraceCharPolicy>("12Ž34"));
+}
+
+BOOST_AUTO_TEST_CASE(IsValidSwTraceNameStringTest)
+{
+ // Valid SWTrace name strings
+ BOOST_CHECK(IsValidSwTraceString<SwTraceNameCharPolicy>(""));
+ BOOST_CHECK(IsValidSwTraceString<SwTraceNameCharPolicy>("_"));
+ BOOST_CHECK(IsValidSwTraceString<SwTraceNameCharPolicy>("0123"));
+ BOOST_CHECK(IsValidSwTraceString<SwTraceNameCharPolicy>("valid_string"));
+ BOOST_CHECK(IsValidSwTraceString<SwTraceNameCharPolicy>("VALID_string_456"));
+
+ // Invalid SWTrace name strings
+ BOOST_CHECK(!IsValidSwTraceString<SwTraceNameCharPolicy>(" "));
+ BOOST_CHECK(!IsValidSwTraceString<SwTraceNameCharPolicy>("invalid string"));
+ BOOST_CHECK(!IsValidSwTraceString<SwTraceNameCharPolicy>("!$%"));
+ BOOST_CHECK(!IsValidSwTraceString<SwTraceNameCharPolicy>("invalid|\\~string#123"));
+ BOOST_CHECK(!IsValidSwTraceString<SwTraceNameCharPolicy>("€£"));
+ BOOST_CHECK(!IsValidSwTraceString<SwTraceNameCharPolicy>("invalid‡string"));
+ BOOST_CHECK(!IsValidSwTraceString<SwTraceNameCharPolicy>("12Ž34"));
+}
+
+template <typename SwTracePolicy>
+void StringToSwTraceStringTestHelper(const std::string& testString, std::vector<uint32_t> buffer, size_t expectedSize)
+{
+ // Convert the test string to a SWTrace string
+ BOOST_CHECK(StringToSwTraceString<SwTracePolicy>(testString, buffer));
+
+ // The buffer must contain at least the length of the string
+ BOOST_CHECK(!buffer.empty());
+
+ // The buffer must be of the expected size (in words)
+ BOOST_CHECK(buffer.size() == expectedSize);
+
+ // The first word of the byte must be the length of the string including the null-terminator
+ BOOST_CHECK(buffer[0] == testString.size() + 1);
+
+ // The contents of the buffer must match the test string
+ BOOST_CHECK(std::memcmp(testString.data(), buffer.data() + 1, testString.size()) == 0);
+
+ // The buffer must include the null-terminator at the end of the string
+ size_t nullTerminatorIndex = sizeof(uint32_t) + testString.size();
+ BOOST_CHECK(reinterpret_cast<unsigned char*>(buffer.data())[nullTerminatorIndex] == '\0');
+}
+
+BOOST_AUTO_TEST_CASE(StringToSwTraceStringTest)
+{
+ std::vector<uint32_t> buffer;
+
+ // Valid SWTrace strings (expected size in words)
+ StringToSwTraceStringTestHelper<SwTraceCharPolicy>("", buffer, 2);
+ StringToSwTraceStringTestHelper<SwTraceCharPolicy>("_", buffer, 2);
+ StringToSwTraceStringTestHelper<SwTraceCharPolicy>("0123", buffer, 3);
+ StringToSwTraceStringTestHelper<SwTraceCharPolicy>("valid_string", buffer, 5);
+ StringToSwTraceStringTestHelper<SwTraceCharPolicy>("VALID_string_456", buffer, 6);
+ StringToSwTraceStringTestHelper<SwTraceCharPolicy>(" ", buffer, 2);
+ StringToSwTraceStringTestHelper<SwTraceCharPolicy>("valid string", buffer, 5);
+ StringToSwTraceStringTestHelper<SwTraceCharPolicy>("!$%", buffer, 2);
+ StringToSwTraceStringTestHelper<SwTraceCharPolicy>("valid|\\~string#123", buffer, 6);
+
+ // Invalid SWTrace strings
+ BOOST_CHECK(!StringToSwTraceString<SwTraceCharPolicy>("€£", buffer));
+ BOOST_CHECK(buffer.empty());
+ BOOST_CHECK(!StringToSwTraceString<SwTraceCharPolicy>("invalid‡string", buffer));
+ BOOST_CHECK(buffer.empty());
+ BOOST_CHECK(!StringToSwTraceString<SwTraceCharPolicy>("12Ž34", buffer));
+ BOOST_CHECK(buffer.empty());
+}
+
+BOOST_AUTO_TEST_CASE(StringToSwTraceNameStringTest)
+{
+ std::vector<uint32_t> buffer;
+
+ // Valid SWTrace namestrings (expected size in words)
+ StringToSwTraceStringTestHelper<SwTraceNameCharPolicy>("", buffer, 2);
+ StringToSwTraceStringTestHelper<SwTraceNameCharPolicy>("_", buffer, 2);
+ StringToSwTraceStringTestHelper<SwTraceNameCharPolicy>("0123", buffer, 3);
+ StringToSwTraceStringTestHelper<SwTraceNameCharPolicy>("valid_string", buffer, 5);
+ StringToSwTraceStringTestHelper<SwTraceNameCharPolicy>("VALID_string_456", buffer, 6);
+
+ // Invalid SWTrace namestrings
+ BOOST_CHECK(!StringToSwTraceString<SwTraceNameCharPolicy>(" ", buffer));
+ BOOST_CHECK(buffer.empty());
+ BOOST_CHECK(!StringToSwTraceString<SwTraceNameCharPolicy>("invalid string", buffer));
+ BOOST_CHECK(buffer.empty());
+ BOOST_CHECK(!StringToSwTraceString<SwTraceNameCharPolicy>("!$%", buffer));
+ BOOST_CHECK(buffer.empty());
+ BOOST_CHECK(!StringToSwTraceString<SwTraceNameCharPolicy>("invalid|\\~string#123", buffer));
+ BOOST_CHECK(buffer.empty());
+ BOOST_CHECK(!StringToSwTraceString<SwTraceNameCharPolicy>("€£", buffer));
+ BOOST_CHECK(buffer.empty());
+ BOOST_CHECK(!StringToSwTraceString<SwTraceNameCharPolicy>("invalid‡string", buffer));
+ BOOST_CHECK(buffer.empty());
+ BOOST_CHECK(!StringToSwTraceString<SwTraceNameCharPolicy>("12Ž34", buffer));
+ BOOST_CHECK(buffer.empty());
+}
+
BOOST_AUTO_TEST_SUITE_END()
diff --git a/src/profiling/test/SendCounterPacketTests.cpp b/src/profiling/test/SendCounterPacketTests.cpp
index c060f168cd..f32c3684c1 100644
--- a/src/profiling/test/SendCounterPacketTests.cpp
+++ b/src/profiling/test/SendCounterPacketTests.cpp
@@ -5,11 +5,12 @@
#include "SendCounterPacketTests.hpp"
-#include <ProfilingUtils.hpp>
#include <EncodeVersion.hpp>
+#include <ProfilingUtils.hpp>
#include <SendCounterPacket.hpp>
#include <armnn/Exceptions.hpp>
+#include <armnn/Conversion.hpp>
#include <boost/test/unit_test.hpp>
#include <boost/numeric/conversion/cast.hpp>
@@ -17,6 +18,8 @@
#include <chrono>
#include <iostream>
+using namespace armnn::profiling;
+
BOOST_AUTO_TEST_SUITE(SendCounterPacketTests)
BOOST_AUTO_TEST_CASE(MockSendCounterPacketTest)
@@ -48,7 +51,6 @@ BOOST_AUTO_TEST_CASE(MockSendCounterPacketTest)
sendCounterPacket.SendPeriodicCounterSelectionPacket(capturePeriod, selectedCounterIds);
BOOST_TEST(strcmp(buffer, "SendPeriodicCounterSelectionPacket") == 0);
-
}
BOOST_AUTO_TEST_CASE(SendPeriodicCounterSelectionPacketTest)
@@ -309,5 +311,1346 @@ BOOST_AUTO_TEST_CASE(SendStreamMetaDataPacketTest)
BOOST_TEST(offset == totalLength);
}
+BOOST_AUTO_TEST_CASE(CreateDeviceRecordTest)
+{
+ MockBuffer mockBuffer(0);
+ SendCounterPacketTest sendCounterPacketTest(mockBuffer);
+
+ // Create a device for testing
+ uint16_t deviceUid = 27;
+ const std::string deviceName = "some_device";
+ uint16_t deviceCores = 3;
+ const DevicePtr device = std::make_unique<Device>(deviceUid, deviceName, deviceCores);
+
+ // Create a device record
+ SendCounterPacket::DeviceRecord deviceRecord;
+ std::string errorMessage;
+ bool result = sendCounterPacketTest.CreateDeviceRecordTest(device, deviceRecord, errorMessage);
+
+ BOOST_CHECK(result);
+ BOOST_CHECK(errorMessage.empty());
+ BOOST_CHECK(deviceRecord.size() == 6); // Size in words: header [2] + device name [4]
+
+ uint16_t deviceRecordWord0[]
+ {
+ static_cast<uint16_t>(deviceRecord[0] >> 16),
+ static_cast<uint16_t>(deviceRecord[0])
+ };
+ BOOST_CHECK(deviceRecordWord0[0] == deviceUid); // uid
+ BOOST_CHECK(deviceRecordWord0[1] == deviceCores); // cores
+ BOOST_CHECK(deviceRecord[1] == 0); // name_offset
+ BOOST_CHECK(deviceRecord[2] == deviceName.size() + 1); // The length of the SWTrace string (name)
+ BOOST_CHECK(std::memcmp(deviceRecord.data() + 3, deviceName.data(), deviceName.size()) == 0); // name
+}
+
+BOOST_AUTO_TEST_CASE(CreateInvalidDeviceRecordTest)
+{
+ MockBuffer mockBuffer(0);
+ SendCounterPacketTest sendCounterPacketTest(mockBuffer);
+
+ // Create a device for testing
+ uint16_t deviceUid = 27;
+ const std::string deviceName = "some€£invalid‡device";
+ uint16_t deviceCores = 3;
+ const DevicePtr device = std::make_unique<Device>(deviceUid, deviceName, deviceCores);
+
+ // Create a device record
+ SendCounterPacket::DeviceRecord deviceRecord;
+ std::string errorMessage;
+ bool result = sendCounterPacketTest.CreateDeviceRecordTest(device, deviceRecord, errorMessage);
+
+ BOOST_CHECK(!result);
+ BOOST_CHECK(!errorMessage.empty());
+ BOOST_CHECK(deviceRecord.empty());
+}
+
+BOOST_AUTO_TEST_CASE(CreateCounterSetRecordTest)
+{
+ MockBuffer mockBuffer(0);
+ SendCounterPacketTest sendCounterPacketTest(mockBuffer);
+
+ // Create a counter set for testing
+ uint16_t counterSetUid = 27;
+ const std::string counterSetName = "some_counter_set";
+ uint16_t counterSetCount = 3421;
+ const CounterSetPtr counterSet = std::make_unique<CounterSet>(counterSetUid, counterSetName, counterSetCount);
+
+ // Create a counter set record
+ SendCounterPacket::CounterSetRecord counterSetRecord;
+ std::string errorMessage;
+ bool result = sendCounterPacketTest.CreateCounterSetRecordTest(counterSet, counterSetRecord, errorMessage);
+
+ BOOST_CHECK(result);
+ BOOST_CHECK(errorMessage.empty());
+ BOOST_CHECK(counterSetRecord.size() == 8); // Size in words: header [2] + counter set name [6]
+
+ uint16_t counterSetRecordWord0[]
+ {
+ static_cast<uint16_t>(counterSetRecord[0] >> 16),
+ static_cast<uint16_t>(counterSetRecord[0])
+ };
+ BOOST_CHECK(counterSetRecordWord0[0] == counterSetUid); // uid
+ BOOST_CHECK(counterSetRecordWord0[1] == counterSetCount); // cores
+ BOOST_CHECK(counterSetRecord[1] == 0); // name_offset
+ BOOST_CHECK(counterSetRecord[2] == counterSetName.size() + 1); // The length of the SWTrace string (name)
+ BOOST_CHECK(std::memcmp(counterSetRecord.data() + 3, counterSetName.data(), counterSetName.size()) == 0); // name
+}
+
+BOOST_AUTO_TEST_CASE(CreateInvalidCounterSetRecordTest)
+{
+ MockBuffer mockBuffer(0);
+ SendCounterPacketTest sendCounterPacketTest(mockBuffer);
+
+ // Create a counter set for testing
+ uint16_t counterSetUid = 27;
+ const std::string counterSetName = "some invalid_counter€£set";
+ uint16_t counterSetCount = 3421;
+ const CounterSetPtr counterSet = std::make_unique<CounterSet>(counterSetUid, counterSetName, counterSetCount);
+
+ // Create a counter set record
+ SendCounterPacket::CounterSetRecord counterSetRecord;
+ std::string errorMessage;
+ bool result = sendCounterPacketTest.CreateCounterSetRecordTest(counterSet, counterSetRecord, errorMessage);
+
+ BOOST_CHECK(!result);
+ BOOST_CHECK(!errorMessage.empty());
+ BOOST_CHECK(counterSetRecord.empty());
+}
+
+BOOST_AUTO_TEST_CASE(CreateEventRecordTest)
+{
+ MockBuffer mockBuffer(0);
+ SendCounterPacketTest sendCounterPacketTest(mockBuffer);
+
+ // Create a counter for testing
+ uint16_t counterUid = 7256;
+ uint16_t maxCounterUid = 132;
+ uint16_t deviceUid = 132;
+ uint16_t counterSetUid = 4497;
+ uint16_t counterClass = 1;
+ uint16_t counterInterpolation = 1;
+ double counterMultiplier = 1234.567f;
+ const std::string counterName = "some_valid_counter";
+ const std::string counterDescription = "a_counter_for_testing";
+ const std::string counterUnits = "Mrads2";
+ const CounterPtr counter = std::make_unique<Counter>(counterUid,
+ maxCounterUid,
+ counterClass,
+ counterInterpolation,
+ counterMultiplier,
+ counterName,
+ counterDescription,
+ counterUnits,
+ deviceUid,
+ counterSetUid);
+ BOOST_ASSERT(counter);
+
+ // Create an event record
+ SendCounterPacket::EventRecord eventRecord;
+ std::string errorMessage;
+ bool result = sendCounterPacketTest.CreateEventRecordTest(counter, eventRecord, errorMessage);
+
+ BOOST_CHECK(result);
+ BOOST_CHECK(errorMessage.empty());
+ BOOST_CHECK(eventRecord.size() == 24); // Size in words: header [8] + counter name [6] + description [7] + units [3]
+
+ uint16_t eventRecordWord0[]
+ {
+ static_cast<uint16_t>(eventRecord[0] >> 16),
+ static_cast<uint16_t>(eventRecord[0])
+ };
+ uint16_t eventRecordWord1[]
+ {
+ static_cast<uint16_t>(eventRecord[1] >> 16),
+ static_cast<uint16_t>(eventRecord[1])
+ };
+ uint16_t eventRecordWord2[]
+ {
+ static_cast<uint16_t>(eventRecord[2] >> 16),
+ static_cast<uint16_t>(eventRecord[2])
+ };
+ uint32_t eventRecordWord34[]
+ {
+ eventRecord[3],
+ eventRecord[4]
+ };
+ BOOST_CHECK(eventRecordWord0[0] == maxCounterUid); // max_counter_uid
+ BOOST_CHECK(eventRecordWord0[1] == counterUid); // counter_uid
+ BOOST_CHECK(eventRecordWord1[0] == deviceUid); // device
+ BOOST_CHECK(eventRecordWord1[1] == counterSetUid); // counter_set
+ BOOST_CHECK(eventRecordWord2[0] == counterClass); // class
+ BOOST_CHECK(eventRecordWord2[1] == counterInterpolation); // interpolation
+ BOOST_CHECK(std::memcmp(eventRecordWord34, &counterMultiplier, sizeof(counterMultiplier)) == 0); // multiplier
+
+ ARMNN_NO_CONVERSION_WARN_BEGIN
+ uint32_t counterNameOffset = 0; // The name is the first item in pool
+ uint32_t counterDescriptionOffset = counterNameOffset + // Counter name offset
+ 4u + // Counter name length (uint32_t)
+ counterName.size() + // 18u
+ 1u + // Null-terminator
+ 1u; // Rounding to the next word
+ size_t counterUnitsOffset = counterDescriptionOffset + // Counter description offset
+ 4u + // Counter description length (uint32_t)
+ counterDescription.size() + // 21u
+ 1u + // Null-terminator
+ 2u; // Rounding to the next word
+ ARMNN_NO_CONVERSION_WARN_END
+
+ BOOST_CHECK(eventRecord[5] == counterNameOffset); // name_offset
+ BOOST_CHECK(eventRecord[6] == counterDescriptionOffset); // description_offset
+ BOOST_CHECK(eventRecord[7] == counterUnitsOffset); // units_offset
+
+ auto eventRecordPool = reinterpret_cast<unsigned char*>(eventRecord.data() + 8u); // The start of the pool
+ size_t uint32_t_size = sizeof(uint32_t);
+
+ // The length of the SWTrace string (name)
+ BOOST_CHECK(eventRecordPool[counterNameOffset] == counterName.size() + 1);
+ // The counter name
+ BOOST_CHECK(std::memcmp(eventRecordPool +
+ counterNameOffset + // Offset
+ uint32_t_size /* The length of the name */,
+ counterName.data(),
+ counterName.size()) == 0); // name
+ // The null-terminator at the end of the name
+ BOOST_CHECK(eventRecordPool[counterNameOffset + uint32_t_size + counterName.size()] == '\0');
+
+ // The length of the SWTrace string (description)
+ BOOST_CHECK(eventRecordPool[counterDescriptionOffset] == counterDescription.size() + 1);
+ // The counter description
+ BOOST_CHECK(std::memcmp(eventRecordPool +
+ counterDescriptionOffset + // Offset
+ uint32_t_size /* The length of the description */,
+ counterDescription.data(),
+ counterDescription.size()) == 0); // description
+ // The null-terminator at the end of the description
+ BOOST_CHECK(eventRecordPool[counterDescriptionOffset + uint32_t_size + counterDescription.size()] == '\0');
+
+ // The length of the SWTrace namestring (units)
+ BOOST_CHECK(eventRecordPool[counterUnitsOffset] == counterUnits.size() + 1);
+ // The counter units
+ BOOST_CHECK(std::memcmp(eventRecordPool +
+ counterUnitsOffset + // Offset
+ uint32_t_size /* The length of the units */,
+ counterUnits.data(),
+ counterUnits.size()) == 0); // units
+ // The null-terminator at the end of the units
+ BOOST_CHECK(eventRecordPool[counterUnitsOffset + uint32_t_size + counterUnits.size()] == '\0');
+}
+
+BOOST_AUTO_TEST_CASE(CreateEventRecordNoUnitsTest)
+{
+ MockBuffer mockBuffer(0);
+ SendCounterPacketTest sendCounterPacketTest(mockBuffer);
+
+ // Create a counter for testing
+ uint16_t counterUid = 44312;
+ uint16_t maxCounterUid = 345;
+ uint16_t deviceUid = 101;
+ uint16_t counterSetUid = 34035;
+ uint16_t counterClass = 0;
+ uint16_t counterInterpolation = 1;
+ double counterMultiplier = 4435.0023f;
+ const std::string counterName = "some_valid_counter";
+ const std::string counterDescription = "a_counter_for_testing";
+ const CounterPtr counter = std::make_unique<Counter>(counterUid,
+ maxCounterUid,
+ counterClass,
+ counterInterpolation,
+ counterMultiplier,
+ counterName,
+ counterDescription,
+ "",
+ deviceUid,
+ counterSetUid);
+ BOOST_ASSERT(counter);
+
+ // Create an event record
+ SendCounterPacket::EventRecord eventRecord;
+ std::string errorMessage;
+ bool result = sendCounterPacketTest.CreateEventRecordTest(counter, eventRecord, errorMessage);
+
+ BOOST_CHECK(result);
+ BOOST_CHECK(errorMessage.empty());
+ BOOST_CHECK(eventRecord.size() == 21); // Size in words: header [8] + counter name [6] + description [7]
+
+ uint16_t eventRecordWord0[]
+ {
+ static_cast<uint16_t>(eventRecord[0] >> 16),
+ static_cast<uint16_t>(eventRecord[0])
+ };
+ uint16_t eventRecordWord1[]
+ {
+ static_cast<uint16_t>(eventRecord[1] >> 16),
+ static_cast<uint16_t>(eventRecord[1])
+ };
+ uint16_t eventRecordWord2[]
+ {
+ static_cast<uint16_t>(eventRecord[2] >> 16),
+ static_cast<uint16_t>(eventRecord[2])
+ };
+ uint32_t eventRecordWord34[]
+ {
+ eventRecord[3],
+ eventRecord[4]
+ };
+ BOOST_CHECK(eventRecordWord0[0] == maxCounterUid); // max_counter_uid
+ BOOST_CHECK(eventRecordWord0[1] == counterUid); // counter_uid
+ BOOST_CHECK(eventRecordWord1[0] == deviceUid); // device
+ BOOST_CHECK(eventRecordWord1[1] == counterSetUid); // counter_set
+ BOOST_CHECK(eventRecordWord2[0] == counterClass); // class
+ BOOST_CHECK(eventRecordWord2[1] == counterInterpolation); // interpolation
+ BOOST_CHECK(std::memcmp(eventRecordWord34, &counterMultiplier, sizeof(counterMultiplier)) == 0); // multiplier
+
+ ARMNN_NO_CONVERSION_WARN_BEGIN
+ uint32_t counterNameOffset = 0; // The name is the first item in pool
+ uint32_t counterDescriptionOffset = counterNameOffset + // Counter name offset
+ 4u + // Counter name length (uint32_t)
+ counterName.size() + // 18u
+ 1u + // Null-terminator
+ 1u; // Rounding to the next word
+ ARMNN_NO_CONVERSION_WARN_END
+
+ BOOST_CHECK(eventRecord[5] == counterNameOffset); // name_offset
+ BOOST_CHECK(eventRecord[6] == counterDescriptionOffset); // description_offset
+ BOOST_CHECK(eventRecord[7] == 0); // units_offset
+
+ auto eventRecordPool = reinterpret_cast<unsigned char*>(eventRecord.data() + 8u); // The start of the pool
+ size_t uint32_t_size = sizeof(uint32_t);
+
+ // The length of the SWTrace string (name)
+ BOOST_CHECK(eventRecordPool[counterNameOffset] == counterName.size() + 1);
+ // The counter name
+ BOOST_CHECK(std::memcmp(eventRecordPool +
+ counterNameOffset + // Offset
+ uint32_t_size, // The length of the name
+ counterName.data(),
+ counterName.size()) == 0); // name
+ // The null-terminator at the end of the name
+ BOOST_CHECK(eventRecordPool[counterNameOffset + uint32_t_size + counterName.size()] == '\0');
+
+ // The length of the SWTrace string (description)
+ BOOST_CHECK(eventRecordPool[counterDescriptionOffset] == counterDescription.size() + 1);
+ // The counter description
+ BOOST_CHECK(std::memcmp(eventRecordPool +
+ counterDescriptionOffset + // Offset
+ uint32_t_size, // The length of the description
+ counterDescription.data(),
+ counterDescription.size()) == 0); // description
+ // The null-terminator at the end of the description
+ BOOST_CHECK(eventRecordPool[counterDescriptionOffset + uint32_t_size + counterDescription.size()] == '\0');
+}
+
+BOOST_AUTO_TEST_CASE(CreateInvalidEventRecordTest1)
+{
+ MockBuffer mockBuffer(0);
+ SendCounterPacketTest sendCounterPacketTest(mockBuffer);
+
+ // Create a counter for testing
+ uint16_t counterUid = 7256;
+ uint16_t maxCounterUid = 132;
+ uint16_t deviceUid = 132;
+ uint16_t counterSetUid = 4497;
+ uint16_t counterClass = 1;
+ uint16_t counterInterpolation = 1;
+ double counterMultiplier = 1234.567f;
+ const std::string counterName = "some_invalid_counter £££"; // Invalid name
+ const std::string counterDescription = "a_counter_for_testing";
+ const std::string counterUnits = "Mrads2";
+ const CounterPtr counter = std::make_unique<Counter>(counterUid,
+ maxCounterUid,
+ counterClass,
+ counterInterpolation,
+ counterMultiplier,
+ counterName,
+ counterDescription,
+ counterUnits,
+ deviceUid,
+ counterSetUid);
+ BOOST_ASSERT(counter);
+
+ // Create an event record
+ SendCounterPacket::EventRecord eventRecord;
+ std::string errorMessage;
+ bool result = sendCounterPacketTest.CreateEventRecordTest(counter, eventRecord, errorMessage);
+
+ BOOST_CHECK(!result);
+ BOOST_CHECK(!errorMessage.empty());
+ BOOST_CHECK(eventRecord.empty());
+}
+
+BOOST_AUTO_TEST_CASE(CreateInvalidEventRecordTest2)
+{
+ MockBuffer mockBuffer(0);
+ SendCounterPacketTest sendCounterPacketTest(mockBuffer);
+
+ // Create a counter for testing
+ uint16_t counterUid = 7256;
+ uint16_t maxCounterUid = 132;
+ uint16_t deviceUid = 132;
+ uint16_t counterSetUid = 4497;
+ uint16_t counterClass = 1;
+ uint16_t counterInterpolation = 1;
+ double counterMultiplier = 1234.567f;
+ const std::string counterName = "some_invalid_counter";
+ const std::string counterDescription = "an invalid d€scription"; // Invalid description
+ const std::string counterUnits = "Mrads2";
+ const CounterPtr counter = std::make_unique<Counter>(counterUid,
+ maxCounterUid,
+ counterClass,
+ counterInterpolation,
+ counterMultiplier,
+ counterName,
+ counterDescription,
+ counterUnits,
+ deviceUid,
+ counterSetUid);
+ BOOST_ASSERT(counter);
+
+ // Create an event record
+ SendCounterPacket::EventRecord eventRecord;
+ std::string errorMessage;
+ bool result = sendCounterPacketTest.CreateEventRecordTest(counter, eventRecord, errorMessage);
+
+ BOOST_CHECK(!result);
+ BOOST_CHECK(!errorMessage.empty());
+ BOOST_CHECK(eventRecord.empty());
+}
+
+BOOST_AUTO_TEST_CASE(CreateInvalidEventRecordTest3)
+{
+ MockBuffer mockBuffer(0);
+ SendCounterPacketTest sendCounterPacketTest(mockBuffer);
+
+ // Create a counter for testing
+ uint16_t counterUid = 7256;
+ uint16_t maxCounterUid = 132;
+ uint16_t deviceUid = 132;
+ uint16_t counterSetUid = 4497;
+ uint16_t counterClass = 1;
+ uint16_t counterInterpolation = 1;
+ double counterMultiplier = 1234.567f;
+ const std::string counterName = "some_invalid_counter";
+ const std::string counterDescription = "a valid description";
+ const std::string counterUnits = "Mrad s2"; // Invalid units
+ const CounterPtr counter = std::make_unique<Counter>(counterUid,
+ maxCounterUid,
+ counterClass,
+ counterInterpolation,
+ counterMultiplier,
+ counterName,
+ counterDescription,
+ counterUnits,
+ deviceUid,
+ counterSetUid);
+ BOOST_ASSERT(counter);
+
+ // Create an event record
+ SendCounterPacket::EventRecord eventRecord;
+ std::string errorMessage;
+ bool result = sendCounterPacketTest.CreateEventRecordTest(counter, eventRecord, errorMessage);
+
+ BOOST_CHECK(!result);
+ BOOST_CHECK(!errorMessage.empty());
+ BOOST_CHECK(eventRecord.empty());
+}
+
+BOOST_AUTO_TEST_CASE(CreateCategoryRecordTest)
+{
+ MockBuffer mockBuffer(0);
+ SendCounterPacketTest sendCounterPacketTest(mockBuffer);
+
+ // Create a category for testing
+ const std::string categoryName = "some_category";
+ uint16_t deviceUid = 1302;
+ uint16_t counterSetUid = 20734;
+ const CategoryPtr category = std::make_unique<Category>(categoryName, deviceUid, counterSetUid);
+ BOOST_ASSERT(category);
+ category->m_Counters = { 11u, 23u, 5670u };
+
+ // Create a collection of counters
+ Counters counters;
+ counters.insert(std::make_pair<uint16_t, CounterPtr>(11,
+ CounterPtr(new Counter(11,
+ 1234,
+ 0,
+ 1,
+ 534.0003f,
+ "counter1",
+ "the first counter",
+ "millipi2",
+ 0,
+ 0))));
+ counters.insert(std::make_pair<uint16_t, CounterPtr>(23,
+ CounterPtr(new Counter(23,
+ 344,
+ 1,
+ 1,
+ 534.0003f,
+ "this is counter 2",
+ "the second counter",
+ "",
+ 0,
+ 0))));
+ counters.insert(std::make_pair<uint16_t, CounterPtr>(5670,
+ CounterPtr(new Counter(5670,
+ 31,
+ 0,
+ 0,
+ 534.0003f,
+ "and this is number 3",
+ "the third counter",
+ "blah_per_second",
+ 0,
+ 0))));
+ Counter* counter1 = counters.find(11)->second.get();
+ Counter* counter2 = counters.find(23)->second.get();
+ Counter* counter3 = counters.find(5670)->second.get();
+ BOOST_ASSERT(counter1);
+ BOOST_ASSERT(counter2);
+ BOOST_ASSERT(counter3);
+ uint16_t categoryEventCount = boost::numeric_cast<uint16_t>(counters.size());
+
+ // Create a category record
+ SendCounterPacket::CategoryRecord categoryRecord;
+ std::string errorMessage;
+ bool result = sendCounterPacketTest.CreateCategoryRecordTest(category, counters, categoryRecord, errorMessage);
+
+ BOOST_CHECK(result);
+ BOOST_CHECK(errorMessage.empty());
+ BOOST_CHECK(categoryRecord.size() == 80); // Size in words: header [4] + event pointer table [3] +
+ // category name [5] + event records [68 = 22 + 20 + 26]
+
+ uint16_t categoryRecordWord0[]
+ {
+ static_cast<uint16_t>(categoryRecord[0] >> 16),
+ static_cast<uint16_t>(categoryRecord[0])
+ };
+ uint16_t categoryRecordWord1[]
+ {
+ static_cast<uint16_t>(categoryRecord[1] >> 16),
+ static_cast<uint16_t>(categoryRecord[1])
+ };
+ BOOST_CHECK(categoryRecordWord0[0] == deviceUid); // device
+ BOOST_CHECK(categoryRecordWord0[1] == counterSetUid); // counter_set
+ BOOST_CHECK(categoryRecordWord1[0] == categoryEventCount); // event_count
+ BOOST_CHECK(categoryRecordWord1[1] == 0); // reserved
+
+ size_t uint32_t_size = sizeof(uint32_t);
+
+ ARMNN_NO_CONVERSION_WARN_BEGIN
+ uint32_t eventPointerTableOffset = 0; // The event pointer table is the first item in pool
+ uint32_t categoryNameOffset = eventPointerTableOffset + // Event pointer table offset
+ categoryEventCount * uint32_t_size; // The size of the event pointer table
+ ARMNN_NO_CONVERSION_WARN_END
+
+ BOOST_CHECK(categoryRecord[2] == eventPointerTableOffset); // event_pointer_table_offset
+ BOOST_CHECK(categoryRecord[3] == categoryNameOffset); // name_offset
+
+ auto categoryRecordPool = reinterpret_cast<unsigned char*>(categoryRecord.data() + 4u); // The start of the pool
+
+ // The event pointer table
+ uint32_t eventRecord0Offset = categoryRecordPool[eventPointerTableOffset + 0 * uint32_t_size];
+ uint32_t eventRecord1Offset = categoryRecordPool[eventPointerTableOffset + 1 * uint32_t_size];
+ uint32_t eventRecord2Offset = categoryRecordPool[eventPointerTableOffset + 2 * uint32_t_size];
+ BOOST_CHECK(eventRecord0Offset == 32);
+ BOOST_CHECK(eventRecord1Offset == 120);
+ BOOST_CHECK(eventRecord2Offset == 200);
+
+ // The length of the SWTrace namestring (name)
+ BOOST_CHECK(categoryRecordPool[categoryNameOffset] == categoryName.size() + 1);
+ // The category name
+ BOOST_CHECK(std::memcmp(categoryRecordPool +
+ categoryNameOffset + // Offset
+ uint32_t_size, // The length of the name
+ categoryName.data(),
+ categoryName.size()) == 0); // name
+ // The null-terminator at the end of the name
+ BOOST_CHECK(categoryRecordPool[categoryNameOffset + uint32_t_size + categoryName.size()] == '\0');
+
+ // For brevity, checking only the UIDs, max counter UIDs and names of the counters in the event records,
+ // as the event records already have a number of unit tests dedicated to them
+
+ // Counter1 UID and max counter UID
+ uint16_t eventRecord0Word0[2] = { 0u, 0u };
+ std::memcpy(eventRecord0Word0, categoryRecordPool + eventRecord0Offset, sizeof(eventRecord0Word0));
+ BOOST_CHECK(eventRecord0Word0[0] == counter1->m_Uid);
+ BOOST_CHECK(eventRecord0Word0[1] == counter1->m_MaxCounterUid);
+
+ // Counter1 name
+ uint32_t counter1NameOffset = 0;
+ std::memcpy(&counter1NameOffset, categoryRecordPool + eventRecord0Offset + 5u * uint32_t_size, uint32_t_size);
+ BOOST_CHECK(counter1NameOffset == 0);
+ // The length of the SWTrace string (name)
+ BOOST_CHECK(categoryRecordPool[eventRecord0Offset + // Offset to the event record
+ 8u * uint32_t_size + // Offset to the event record pool
+ counter1NameOffset // Offset to the name of the counter
+ ] == counter1->m_Name.size() + 1); // The length of the name including the
+ // null-terminator
+ // The counter1 name
+ BOOST_CHECK(std::memcmp(categoryRecordPool + // The beginning of the category pool
+ eventRecord0Offset + // Offset to the event record
+ 8u * uint32_t_size + // Offset to the event record pool
+ counter1NameOffset + // Offset to the name of the counter
+ uint32_t_size, // The length of the name
+ counter1->m_Name.data(),
+ counter1->m_Name.size()) == 0); // name
+ // The null-terminator at the end of the counter1 name
+ BOOST_CHECK(categoryRecordPool[eventRecord0Offset + // Offset to the event record
+ 8u * uint32_t_size + // Offset to the event record pool
+ counter1NameOffset + // Offset to the name of the counter
+ uint32_t_size + // The length of the name
+ counter1->m_Name.size() // The name of the counter
+ ] == '\0');
+
+ // Counter2 name
+ uint32_t counter2NameOffset = 0;
+ std::memcpy(&counter2NameOffset, categoryRecordPool + eventRecord1Offset + 5u * uint32_t_size, uint32_t_size);
+ BOOST_CHECK(counter2NameOffset == 0);
+ // The length of the SWTrace string (name)
+ BOOST_CHECK(categoryRecordPool[eventRecord1Offset + // Offset to the event record
+ 8u * uint32_t_size + // Offset to the event record pool
+ counter2NameOffset // Offset to the name of the counter
+ ] == counter2->m_Name.size() + 1); // The length of the name including the
+ // null-terminator
+ // The counter2 name
+ BOOST_CHECK(std::memcmp(categoryRecordPool + // The beginning of the category pool
+ eventRecord1Offset + // Offset to the event record
+ 8u * uint32_t_size + // Offset to the event record pool
+ counter2NameOffset + // Offset to the name of the counter
+ uint32_t_size, // The length of the name
+ counter2->m_Name.data(),
+ counter2->m_Name.size()) == 0); // name
+ // The null-terminator at the end of the counter2 name
+ BOOST_CHECK(categoryRecordPool[eventRecord1Offset + // Offset to the event record
+ 8u * uint32_t_size + // Offset to the event record pool
+ counter2NameOffset + // Offset to the name of the counter
+ uint32_t_size + // The length of the name
+ counter2->m_Name.size() // The name of the counter
+ ] == '\0');
+
+ // Counter3 name
+ uint32_t counter3NameOffset = 0;
+ std::memcpy(&counter3NameOffset, categoryRecordPool + eventRecord2Offset + 5u * uint32_t_size, uint32_t_size);
+ BOOST_CHECK(counter3NameOffset == 0);
+ // The length of the SWTrace string (name)
+ BOOST_CHECK(categoryRecordPool[eventRecord2Offset + // Offset to the event record
+ 8u * uint32_t_size + // Offset to the event record pool
+ counter3NameOffset // Offset to the name of the counter
+ ] == counter3->m_Name.size() + 1); // The length of the name including the
+ // null-terminator
+ // The counter3 name
+ BOOST_CHECK(std::memcmp(categoryRecordPool + // The beginning of the category pool
+ eventRecord2Offset + // Offset to the event record
+ 8u * uint32_t_size + // Offset to the event record pool
+ counter3NameOffset + // Offset to the name of the counter
+ uint32_t_size, // The length of the name
+ counter3->m_Name.data(),
+ counter3->m_Name.size()) == 0); // name
+ // The null-terminator at the end of the counter3 name
+ BOOST_CHECK(categoryRecordPool[eventRecord2Offset + // Offset to the event record
+ 8u * uint32_t_size + // Offset to the event record pool
+ counter3NameOffset + // Offset to the name of the counter
+ uint32_t_size + // The length of the name
+ counter3->m_Name.size() // The name of the counter
+ ] == '\0');
+}
+
+BOOST_AUTO_TEST_CASE(CreateInvalidCategoryRecordTest1)
+{
+ MockBuffer mockBuffer(0);
+ SendCounterPacketTest sendCounterPacketTest(mockBuffer);
+
+ // Create a category for testing
+ const std::string categoryName = "some invalid category";
+ uint16_t deviceUid = 1302;
+ uint16_t counterSetUid = 20734;
+ const CategoryPtr category = std::make_unique<Category>(categoryName, deviceUid, counterSetUid);
+ BOOST_CHECK(category);
+
+ // Create a category record
+ Counters counters;
+ SendCounterPacket::CategoryRecord categoryRecord;
+ std::string errorMessage;
+ bool result = sendCounterPacketTest.CreateCategoryRecordTest(category, counters, categoryRecord, errorMessage);
+
+ BOOST_CHECK(!result);
+ BOOST_CHECK(!errorMessage.empty());
+ BOOST_CHECK(categoryRecord.empty());
+}
+
+BOOST_AUTO_TEST_CASE(CreateInvalidCategoryRecordTest2)
+{
+ MockBuffer mockBuffer(0);
+ SendCounterPacketTest sendCounterPacketTest(mockBuffer);
+
+ // Create a category for testing
+ const std::string categoryName = "some_category";
+ uint16_t deviceUid = 1302;
+ uint16_t counterSetUid = 20734;
+ const CategoryPtr category = std::make_unique<Category>(categoryName, deviceUid, counterSetUid);
+ BOOST_CHECK(category);
+ category->m_Counters = { 11u, 23u, 5670u };
+
+ // Create a collection of counters
+ Counters counters;
+ counters.insert(std::make_pair<uint16_t, CounterPtr>(11,
+ CounterPtr(new Counter(11,
+ 1234,
+ 0,
+ 1,
+ 534.0003f,
+ "count€r1", // Invalid name
+ "the first counter",
+ "millipi2",
+ 0,
+ 0))));
+
+ Counter* counter1 = counters.find(11)->second.get();
+ BOOST_CHECK(counter1);
+
+ // Create a category record
+ SendCounterPacket::CategoryRecord categoryRecord;
+ std::string errorMessage;
+ bool result = sendCounterPacketTest.CreateCategoryRecordTest(category, counters, categoryRecord, errorMessage);
+
+ BOOST_CHECK(!result);
+ BOOST_CHECK(!errorMessage.empty());
+ BOOST_CHECK(categoryRecord.empty());
+}
+
+BOOST_AUTO_TEST_CASE(SendCounterDirectoryPacketTest1)
+{
+ // The counter directory used for testing
+ CounterDirectory counterDirectory;
+
+ // Register a device
+ const std::string device1Name = "device1";
+ const Device* device1 = nullptr;
+ BOOST_CHECK_NO_THROW(device1 = counterDirectory.RegisterDevice(device1Name, 3));
+ BOOST_CHECK(counterDirectory.GetDeviceCount() == 1);
+ BOOST_CHECK(device1);
+
+ // Register a device
+ const std::string device2Name = "device2";
+ const Device* device2 = nullptr;
+ BOOST_CHECK_NO_THROW(device2 = counterDirectory.RegisterDevice(device2Name));
+ BOOST_CHECK(counterDirectory.GetDeviceCount() == 2);
+ BOOST_CHECK(device2);
+
+ // Buffer with not enough space
+ MockBuffer mockBuffer(10);
+ SendCounterPacket sendCounterPacket(mockBuffer);
+ BOOST_CHECK_THROW(sendCounterPacket.SendCounterDirectoryPacket(counterDirectory),
+ armnn::profiling::BufferExhaustion);
+}
+
+BOOST_AUTO_TEST_CASE(SendCounterDirectoryPacketTest2)
+{
+ // The counter directory used for testing
+ CounterDirectory counterDirectory;
+
+ // Register a device
+ const std::string device1Name = "device1";
+ const Device* device1 = nullptr;
+ BOOST_CHECK_NO_THROW(device1 = counterDirectory.RegisterDevice(device1Name, 3));
+ BOOST_CHECK(counterDirectory.GetDeviceCount() == 1);
+ BOOST_CHECK(device1);
+
+ // Register a device
+ const std::string device2Name = "device2";
+ const Device* device2 = nullptr;
+ BOOST_CHECK_NO_THROW(device2 = counterDirectory.RegisterDevice(device2Name));
+ BOOST_CHECK(counterDirectory.GetDeviceCount() == 2);
+ BOOST_CHECK(device2);
+
+ // Register a counter set
+ const std::string counterSet1Name = "counterset1";
+ const CounterSet* counterSet1 = nullptr;
+ BOOST_CHECK_NO_THROW(counterSet1 = counterDirectory.RegisterCounterSet(counterSet1Name));
+ BOOST_CHECK(counterDirectory.GetCounterSetCount() == 1);
+ BOOST_CHECK(counterSet1);
+
+ // Register a category associated to "device1" and "counterset1"
+ const std::string category1Name = "category1";
+ const Category* category1 = nullptr;
+ BOOST_CHECK_NO_THROW(category1 = counterDirectory.RegisterCategory(category1Name,
+ device1->m_Uid,
+ counterSet1->m_Uid));
+ BOOST_CHECK(counterDirectory.GetCategoryCount() == 1);
+ BOOST_CHECK(category1);
+
+ // Register a category not associated to "device2" but no counter set
+ const std::string category2Name = "category2";
+ const Category* category2 = nullptr;
+ BOOST_CHECK_NO_THROW(category2 = counterDirectory.RegisterCategory(category2Name,
+ device2->m_Uid));
+ BOOST_CHECK(counterDirectory.GetCategoryCount() == 2);
+ BOOST_CHECK(category2);
+
+ // Register a counter associated to "category1"
+ const Counter* counter1 = nullptr;
+ BOOST_CHECK_NO_THROW(counter1 = counterDirectory.RegisterCounter(category1Name,
+ 0,
+ 1,
+ 123.45f,
+ "counter1",
+ "counter1description",
+ std::string("counter1units")));
+ BOOST_CHECK(counterDirectory.GetCounterCount() == 3);
+ BOOST_CHECK(counter1);
+
+ // Register a counter associated to "category1"
+ const Counter* counter2 = nullptr;
+ BOOST_CHECK_NO_THROW(counter2 = counterDirectory.RegisterCounter(category1Name,
+ 1,
+ 0,
+ 330.1245656765f,
+ "counter2",
+ "counter2description",
+ std::string("counter2units"),
+ armnn::EmptyOptional(),
+ device2->m_Uid,
+ 0));
+ BOOST_CHECK(counterDirectory.GetCounterCount() == 4);
+ BOOST_CHECK(counter2);
+
+ // Register a counter associated to "category2"
+ const Counter* counter3 = nullptr;
+ BOOST_CHECK_NO_THROW(counter3 = counterDirectory.RegisterCounter(category2Name,
+ 1,
+ 1,
+ 0.0000045399f,
+ "counter3",
+ "counter3description",
+ armnn::EmptyOptional(),
+ 5,
+ device2->m_Uid,
+ counterSet1->m_Uid));
+ BOOST_CHECK(counterDirectory.GetCounterCount() == 9);
+ BOOST_CHECK(counter3);
+
+ // Buffer with enough space
+ MockBuffer mockBuffer(1024);
+ SendCounterPacket sendCounterPacket(mockBuffer);
+ BOOST_CHECK_NO_THROW(sendCounterPacket.SendCounterDirectoryPacket(counterDirectory));
+
+ // Get the read buffer
+ unsigned int sizeRead = 0;
+ const unsigned char* readBuffer = mockBuffer.GetReadBuffer(sizeRead);
+
+ // Check the packet header
+ uint32_t packetHeaderWord0 = ReadUint32(readBuffer, 0);
+ uint32_t packetHeaderWord1 = ReadUint32(readBuffer, 4);
+ BOOST_TEST(((packetHeaderWord0 >> 26) & 0x3F) == 0); // packet_family
+ BOOST_TEST(((packetHeaderWord0 >> 16) & 0x3FF) == 2); // packet_id
+ BOOST_TEST(packetHeaderWord1 == 944); // data_length
+
+ // Check the body header
+ uint32_t bodyHeaderWord0 = ReadUint32(readBuffer, 8);
+ uint32_t bodyHeaderWord1 = ReadUint32(readBuffer, 12);
+ uint32_t bodyHeaderWord2 = ReadUint32(readBuffer, 16);
+ uint32_t bodyHeaderWord3 = ReadUint32(readBuffer, 20);
+ uint32_t bodyHeaderWord4 = ReadUint32(readBuffer, 24);
+ uint32_t bodyHeaderWord5 = ReadUint32(readBuffer, 28);
+ uint16_t deviceRecordCount = static_cast<uint16_t>(bodyHeaderWord0 >> 16);
+ uint16_t counterSetRecordCount = static_cast<uint16_t>(bodyHeaderWord2 >> 16);
+ uint16_t categoryRecordCount = static_cast<uint16_t>(bodyHeaderWord4 >> 16);
+ BOOST_TEST(deviceRecordCount == 2); // device_records_count
+ BOOST_TEST(bodyHeaderWord1 == 0); // device_records_pointer_table_offset
+ BOOST_TEST(counterSetRecordCount == 1); // counter_set_count
+ BOOST_TEST(bodyHeaderWord3 == 8); // counter_set_pointer_table_offset
+ BOOST_TEST(categoryRecordCount == 2); // categories_count
+ BOOST_TEST(bodyHeaderWord5 == 12); // categories_pointer_table_offset
+
+ // Check the device records pointer table
+ uint32_t deviceRecordOffset0 = ReadUint32(readBuffer, 32);
+ uint32_t deviceRecordOffset1 = ReadUint32(readBuffer, 36);
+ BOOST_TEST(deviceRecordOffset0 == 0); // Device record offset for "device1"
+ BOOST_TEST(deviceRecordOffset1 == 20); // Device record offset for "device2"
+
+ // Check the counter set pointer table
+ uint32_t counterSetRecordOffset0 = ReadUint32(readBuffer, 40);
+ BOOST_TEST(counterSetRecordOffset0 == 40); // Counter set record offset for "counterset1"
+
+ // Check the category pointer table
+ uint32_t categoryRecordOffset0 = ReadUint32(readBuffer, 44);
+ uint32_t categoryRecordOffset1 = ReadUint32(readBuffer, 48);
+ BOOST_TEST(categoryRecordOffset0 == 64); // Category record offset for "category1"
+ BOOST_TEST(categoryRecordOffset1 == 476); // Category record offset for "category2"
+
+ // Get the device record pool offset
+ uint32_t uint32_t_size = sizeof(uint32_t);
+ uint32_t packetBodyPoolOffset = 2u * uint32_t_size + // packet_header
+ 6u * uint32_t_size + // body_header
+ deviceRecordCount * uint32_t_size + // Size of device_records_pointer_table
+ counterSetRecordCount * uint32_t_size + // Size of counter_set_pointer_table
+ categoryRecordCount * uint32_t_size; // Size of categories_pointer_table
+
+ // Device record structure/collection used for testing
+ struct DeviceRecord
+ {
+ uint16_t uid;
+ uint16_t cores;
+ uint32_t name_offset;
+ uint32_t name_length;
+ std::string name;
+ };
+ std::vector<DeviceRecord> deviceRecords;
+ uint32_t deviceRecordsPointerTableOffset = 2u * uint32_t_size + // packet_header
+ 6u * uint32_t_size + // body_header
+ bodyHeaderWord1; // device_records_pointer_table_offset
+ for (uint32_t i = 0; i < deviceRecordCount; i++)
+ {
+ // Get the device record offset
+ uint32_t deviceRecordOffset = ReadUint32(readBuffer, deviceRecordsPointerTableOffset + i * uint32_t_size);
+
+ // Collect the data for the device record
+ uint32_t deviceRecordWord0 = ReadUint32(readBuffer,
+ packetBodyPoolOffset + deviceRecordOffset + 0 * uint32_t_size);
+ uint32_t deviceRecordWord1 = ReadUint32(readBuffer,
+ packetBodyPoolOffset + deviceRecordOffset + 1 * uint32_t_size);
+ DeviceRecord deviceRecord;
+ deviceRecord.uid = static_cast<uint16_t>(deviceRecordWord0 >> 16); // uid
+ deviceRecord.cores = static_cast<uint16_t>(deviceRecordWord0); // cores
+ deviceRecord.name_offset = deviceRecordWord1; // name_offset
+
+ uint32_t deviceRecordPoolOffset = packetBodyPoolOffset + // Packet body offset
+ deviceRecordOffset + // Device record offset
+ 2 * uint32_t_size + // Device record header
+ deviceRecord.name_offset; // Device name offset
+ uint32_t deviceRecordNameLength = ReadUint32(readBuffer, deviceRecordPoolOffset);
+ deviceRecord.name_length = deviceRecordNameLength; // name_length
+ unsigned char deviceRecordNameNullTerminator = // name null-terminator
+ ReadUint8(readBuffer, deviceRecordPoolOffset + uint32_t_size + deviceRecordNameLength - 1);
+ BOOST_CHECK(deviceRecordNameNullTerminator == '\0');
+ std::vector<unsigned char> deviceRecordNameBuffer(deviceRecord.name_length - 1);
+ std::memcpy(deviceRecordNameBuffer.data(),
+ readBuffer + deviceRecordPoolOffset + uint32_t_size, deviceRecordNameBuffer.size());
+ deviceRecord.name.assign(deviceRecordNameBuffer.begin(), deviceRecordNameBuffer.end()); // name
+
+ deviceRecords.push_back(deviceRecord);
+ }
+
+ // Check that the device records are correct
+ BOOST_CHECK(deviceRecords.size() == 2);
+ for (const DeviceRecord& deviceRecord : deviceRecords)
+ {
+ const Device* device = counterDirectory.GetDevice(deviceRecord.uid);
+ BOOST_CHECK(device);
+ BOOST_CHECK(device->m_Uid == deviceRecord.uid);
+ BOOST_CHECK(device->m_Cores == deviceRecord.cores);
+ BOOST_CHECK(device->m_Name == deviceRecord.name);
+ }
+
+ // Counter set record structure/collection used for testing
+ struct CounterSetRecord
+ {
+ uint16_t uid;
+ uint16_t count;
+ uint32_t name_offset;
+ uint32_t name_length;
+ std::string name;
+ };
+ std::vector<CounterSetRecord> counterSetRecords;
+ uint32_t counterSetRecordsPointerTableOffset = 2u * uint32_t_size + // packet_header
+ 6u * uint32_t_size + // body_header
+ bodyHeaderWord3; // counter_set_pointer_table_offset
+ for (uint32_t i = 0; i < counterSetRecordCount; i++)
+ {
+ // Get the counter set record offset
+ uint32_t counterSetRecordOffset = ReadUint32(readBuffer,
+ counterSetRecordsPointerTableOffset + i * uint32_t_size);
+
+ // Collect the data for the counter set record
+ uint32_t counterSetRecordWord0 = ReadUint32(readBuffer,
+ packetBodyPoolOffset + counterSetRecordOffset + 0 * uint32_t_size);
+ uint32_t counterSetRecordWord1 = ReadUint32(readBuffer,
+ packetBodyPoolOffset + counterSetRecordOffset + 1 * uint32_t_size);
+ CounterSetRecord counterSetRecord;
+ counterSetRecord.uid = static_cast<uint16_t>(counterSetRecordWord0 >> 16); // uid
+ counterSetRecord.count = static_cast<uint16_t>(counterSetRecordWord0); // count
+ counterSetRecord.name_offset = counterSetRecordWord1; // name_offset
+
+ uint32_t counterSetRecordPoolOffset = packetBodyPoolOffset + // Packet body offset
+ counterSetRecordOffset + // Counter set record offset
+ 2 * uint32_t_size + // Counter set record header
+ counterSetRecord.name_offset; // Counter set name offset
+ uint32_t counterSetRecordNameLength = ReadUint32(readBuffer, counterSetRecordPoolOffset);
+ counterSetRecord.name_length = counterSetRecordNameLength; // name_length
+ unsigned char counterSetRecordNameNullTerminator = // name null-terminator
+ ReadUint8(readBuffer, counterSetRecordPoolOffset + uint32_t_size + counterSetRecordNameLength - 1);
+ BOOST_CHECK(counterSetRecordNameNullTerminator == '\0');
+ std::vector<unsigned char> counterSetRecordNameBuffer(counterSetRecord.name_length - 1);
+ std::memcpy(counterSetRecordNameBuffer.data(),
+ readBuffer + counterSetRecordPoolOffset + uint32_t_size, counterSetRecordNameBuffer.size());
+ counterSetRecord.name.assign(counterSetRecordNameBuffer.begin(), counterSetRecordNameBuffer.end()); // name
+
+ counterSetRecords.push_back(counterSetRecord);
+ }
+
+ // Check that the counter set records are correct
+ BOOST_CHECK(counterSetRecords.size() == 1);
+ for (const CounterSetRecord& counterSetRecord : counterSetRecords)
+ {
+ const CounterSet* counterSet = counterDirectory.GetCounterSet(counterSetRecord.uid);
+ BOOST_CHECK(counterSet);
+ BOOST_CHECK(counterSet->m_Uid == counterSetRecord.uid);
+ BOOST_CHECK(counterSet->m_Count == counterSetRecord.count);
+ BOOST_CHECK(counterSet->m_Name == counterSetRecord.name);
+ }
+
+ // Event record structure/collection used for testing
+ struct EventRecord
+ {
+ uint16_t counter_uid;
+ uint16_t max_counter_uid;
+ uint16_t device;
+ uint16_t counter_set;
+ uint16_t counter_class;
+ uint16_t interpolation;
+ double multiplier;
+ uint32_t name_offset;
+ uint32_t name_length;
+ std::string name;
+ uint32_t description_offset;
+ uint32_t description_length;
+ std::string description;
+ uint32_t units_offset;
+ uint32_t units_length;
+ std::string units;
+ };
+ // Category record structure/collection used for testing
+ struct CategoryRecord
+ {
+ uint16_t device;
+ uint16_t counter_set;
+ uint16_t event_count;
+ uint32_t event_pointer_table_offset;
+ uint32_t name_offset;
+ uint32_t name_length;
+ std::string name;
+ std::vector<uint32_t> event_pointer_table;
+ std::vector<EventRecord> event_records;
+ };
+ std::vector<CategoryRecord> categoryRecords;
+ uint32_t categoryRecordsPointerTableOffset = 2u * uint32_t_size + // packet_header
+ 6u * uint32_t_size + // body_header
+ bodyHeaderWord5; // categories_pointer_table_offset
+ for (uint32_t i = 0; i < categoryRecordCount; i++)
+ {
+ // Get the category record offset
+ uint32_t categoryRecordOffset = ReadUint32(readBuffer, categoryRecordsPointerTableOffset + i * uint32_t_size);
+
+ // Collect the data for the category record
+ uint32_t categoryRecordWord0 = ReadUint32(readBuffer,
+ packetBodyPoolOffset + categoryRecordOffset + 0 * uint32_t_size);
+ uint32_t categoryRecordWord1 = ReadUint32(readBuffer,
+ packetBodyPoolOffset + categoryRecordOffset + 1 * uint32_t_size);
+ uint32_t categoryRecordWord2 = ReadUint32(readBuffer,
+ packetBodyPoolOffset + categoryRecordOffset + 2 * uint32_t_size);
+ uint32_t categoryRecordWord3 = ReadUint32(readBuffer,
+ packetBodyPoolOffset + categoryRecordOffset + 3 * uint32_t_size);
+ CategoryRecord categoryRecord;
+ categoryRecord.device = static_cast<uint16_t>(categoryRecordWord0 >> 16); // device
+ categoryRecord.counter_set = static_cast<uint16_t>(categoryRecordWord0); // counter_set
+ categoryRecord.event_count = static_cast<uint16_t>(categoryRecordWord1 >> 16); // event_count
+ categoryRecord.event_pointer_table_offset = categoryRecordWord2; // event_pointer_table_offset
+ categoryRecord.name_offset = categoryRecordWord3; // name_offset
+
+ uint32_t categoryRecordPoolOffset = packetBodyPoolOffset + // Packet body offset
+ categoryRecordOffset + // Category record offset
+ 4 * uint32_t_size; // Category record header
+
+ uint32_t categoryRecordNameLength = ReadUint32(readBuffer,
+ categoryRecordPoolOffset + categoryRecord.name_offset);
+ categoryRecord.name_length = categoryRecordNameLength; // name_length
+ unsigned char categoryRecordNameNullTerminator =
+ ReadUint8(readBuffer,
+ categoryRecordPoolOffset +
+ categoryRecord.name_offset +
+ uint32_t_size +
+ categoryRecordNameLength - 1); // name null-terminator
+ BOOST_CHECK(categoryRecordNameNullTerminator == '\0');
+ std::vector<unsigned char> categoryRecordNameBuffer(categoryRecord.name_length - 1);
+ std::memcpy(categoryRecordNameBuffer.data(),
+ readBuffer +
+ categoryRecordPoolOffset +
+ categoryRecord.name_offset +
+ uint32_t_size,
+ categoryRecordNameBuffer.size());
+ categoryRecord.name.assign(categoryRecordNameBuffer.begin(), categoryRecordNameBuffer.end()); // name
+
+ categoryRecord.event_pointer_table.resize(categoryRecord.event_count);
+ for (uint32_t eventIndex = 0; eventIndex < categoryRecord.event_count; eventIndex++)
+ {
+ uint32_t eventRecordOffset = ReadUint32(readBuffer,
+ categoryRecordPoolOffset +
+ categoryRecord.event_pointer_table_offset +
+ eventIndex * uint32_t_size);
+ categoryRecord.event_pointer_table[eventIndex] = eventRecordOffset;
+
+ // Collect the data for the event record
+ uint32_t eventRecordWord0 = ReadUint32(readBuffer,
+ categoryRecordPoolOffset + eventRecordOffset + 0 * uint32_t_size);
+ uint32_t eventRecordWord1 = ReadUint32(readBuffer,
+ categoryRecordPoolOffset + eventRecordOffset + 1 * uint32_t_size);
+ uint32_t eventRecordWord2 = ReadUint32(readBuffer,
+ categoryRecordPoolOffset + eventRecordOffset + 2 * uint32_t_size);
+ uint64_t eventRecordWord34 = ReadUint64(readBuffer,
+ categoryRecordPoolOffset + eventRecordOffset + 3 * uint32_t_size);
+ uint32_t eventRecordWord5 = ReadUint32(readBuffer,
+ categoryRecordPoolOffset + eventRecordOffset + 5 * uint32_t_size);
+ uint32_t eventRecordWord6 = ReadUint32(readBuffer,
+ categoryRecordPoolOffset + eventRecordOffset + 6 * uint32_t_size);
+ uint32_t eventRecordWord7 = ReadUint32(readBuffer,
+ categoryRecordPoolOffset + eventRecordOffset + 7 * uint32_t_size);
+ EventRecord eventRecord;
+ eventRecord.counter_uid = static_cast<uint16_t>(eventRecordWord0); // counter_uid
+ eventRecord.max_counter_uid = static_cast<uint16_t>(eventRecordWord0 >> 16); // max_counter_uid
+ eventRecord.device = static_cast<uint16_t>(eventRecordWord1 >> 16); // device
+ eventRecord.counter_set = static_cast<uint16_t>(eventRecordWord1); // counter_set
+ eventRecord.counter_class = static_cast<uint16_t>(eventRecordWord2 >> 16); // class
+ eventRecord.interpolation = static_cast<uint16_t>(eventRecordWord2); // interpolation
+ std::memcpy(&eventRecord.multiplier, &eventRecordWord34, sizeof(eventRecord.multiplier)); // multiplier
+ eventRecord.name_offset = static_cast<uint32_t>(eventRecordWord5); // name_offset
+ eventRecord.description_offset = static_cast<uint32_t>(eventRecordWord6); // description_offset
+ eventRecord.units_offset = static_cast<uint32_t>(eventRecordWord7); // units_offset
+
+ uint32_t eventRecordPoolOffset = categoryRecordPoolOffset + // Category record pool offset
+ eventRecordOffset + // Event record offset
+ 8 * uint32_t_size; // Event record header
+
+ uint32_t eventRecordNameLength = ReadUint32(readBuffer,
+ eventRecordPoolOffset + eventRecord.name_offset);
+ eventRecord.name_length = eventRecordNameLength; // name_length
+ unsigned char eventRecordNameNullTerminator =
+ ReadUint8(readBuffer,
+ eventRecordPoolOffset +
+ eventRecord.name_offset +
+ uint32_t_size +
+ eventRecordNameLength - 1); // name null-terminator
+ BOOST_CHECK(eventRecordNameNullTerminator == '\0');
+ std::vector<unsigned char> eventRecordNameBuffer(eventRecord.name_length - 1);
+ std::memcpy(eventRecordNameBuffer.data(),
+ readBuffer +
+ eventRecordPoolOffset +
+ eventRecord.name_offset +
+ uint32_t_size,
+ eventRecordNameBuffer.size());
+ eventRecord.name.assign(eventRecordNameBuffer.begin(), eventRecordNameBuffer.end()); // name
+
+ uint32_t eventRecordDescriptionLength = ReadUint32(readBuffer,
+ eventRecordPoolOffset + eventRecord.description_offset);
+ eventRecord.description_length = eventRecordDescriptionLength; // description_length
+ unsigned char eventRecordDescriptionNullTerminator =
+ ReadUint8(readBuffer,
+ eventRecordPoolOffset +
+ eventRecord.description_offset +
+ uint32_t_size +
+ eventRecordDescriptionLength - 1); // description null-terminator
+ BOOST_CHECK(eventRecordDescriptionNullTerminator == '\0');
+ std::vector<unsigned char> eventRecordDescriptionBuffer(eventRecord.description_length - 1);
+ std::memcpy(eventRecordDescriptionBuffer.data(),
+ readBuffer +
+ eventRecordPoolOffset +
+ eventRecord.description_offset +
+ uint32_t_size,
+ eventRecordDescriptionBuffer.size());
+ eventRecord.description.assign(eventRecordDescriptionBuffer.begin(),
+ eventRecordDescriptionBuffer.end()); // description
+
+ if (eventRecord.units_offset > 0)
+ {
+ uint32_t eventRecordUnitsLength = ReadUint32(readBuffer,
+ eventRecordPoolOffset + eventRecord.units_offset);
+ eventRecord.units_length = eventRecordUnitsLength; // units_length
+ unsigned char eventRecordUnitsNullTerminator =
+ ReadUint8(readBuffer,
+ eventRecordPoolOffset +
+ eventRecord.units_offset +
+ uint32_t_size +
+ eventRecordUnitsLength - 1); // units null-terminator
+ BOOST_CHECK(eventRecordUnitsNullTerminator == '\0');
+ std::vector<unsigned char> eventRecordUnitsBuffer(eventRecord.units_length - 1);
+ std::memcpy(eventRecordUnitsBuffer.data(),
+ readBuffer +
+ eventRecordPoolOffset +
+ eventRecord.units_offset +
+ uint32_t_size,
+ eventRecordUnitsBuffer.size());
+ eventRecord.units.assign(eventRecordUnitsBuffer.begin(), eventRecordUnitsBuffer.end()); // units
+ }
+
+ categoryRecord.event_records.push_back(eventRecord);
+ }
+
+ categoryRecords.push_back(categoryRecord);
+ }
+
+ // Check that the category records are correct
+ BOOST_CHECK(categoryRecords.size() == 2);
+ for (const CategoryRecord& categoryRecord : categoryRecords)
+ {
+ const Category* category = counterDirectory.GetCategory(categoryRecord.name);
+ BOOST_CHECK(category);
+ BOOST_CHECK(category->m_Name == categoryRecord.name);
+ BOOST_CHECK(category->m_DeviceUid == categoryRecord.device);
+ BOOST_CHECK(category->m_CounterSetUid == categoryRecord.counter_set);
+ BOOST_CHECK(category->m_Counters.size() == categoryRecord.event_count);
+
+ // Check that the event records are correct
+ for (const EventRecord& eventRecord : categoryRecord.event_records)
+ {
+ const Counter* counter = counterDirectory.GetCounter(eventRecord.counter_uid);
+ BOOST_CHECK(counter);
+ BOOST_CHECK(counter->m_MaxCounterUid == eventRecord.max_counter_uid);
+ BOOST_CHECK(counter->m_DeviceUid == eventRecord.device);
+ BOOST_CHECK(counter->m_CounterSetUid == eventRecord.counter_set);
+ BOOST_CHECK(counter->m_Class == eventRecord.counter_class);
+ BOOST_CHECK(counter->m_Interpolation == eventRecord.interpolation);
+ BOOST_CHECK(counter->m_Multiplier == eventRecord.multiplier);
+ BOOST_CHECK(counter->m_Name == eventRecord.name);
+ BOOST_CHECK(counter->m_Description == eventRecord.description);
+ BOOST_CHECK(counter->m_Units == eventRecord.units);
+ }
+ }
+}
+
+BOOST_AUTO_TEST_CASE(SendCounterDirectoryPacketTest3)
+{
+ // Using a mock counter directory that allows to register invalid objects
+ MockCounterDirectory counterDirectory;
+
+ // Register an invalid device
+ const std::string deviceName = "inv@lid dev!c€";
+ const Device* device = nullptr;
+ BOOST_CHECK_NO_THROW(device = counterDirectory.RegisterDevice(deviceName, 3));
+ BOOST_CHECK(counterDirectory.GetDeviceCount() == 1);
+ BOOST_CHECK(device);
+
+ // Buffer with enough space
+ MockBuffer mockBuffer(1024);
+ SendCounterPacket sendCounterPacket(mockBuffer);
+ BOOST_CHECK_THROW(sendCounterPacket.SendCounterDirectoryPacket(counterDirectory), armnn::RuntimeException);
+}
+
+BOOST_AUTO_TEST_CASE(SendCounterDirectoryPacketTest4)
+{
+ // Using a mock counter directory that allows to register invalid objects
+ MockCounterDirectory counterDirectory;
+
+ // Register an invalid counter set
+ const std::string counterSetName = "inv@lid count€rs€t";
+ const CounterSet* counterSet = nullptr;
+ BOOST_CHECK_NO_THROW(counterSet = counterDirectory.RegisterCounterSet(counterSetName));
+ BOOST_CHECK(counterDirectory.GetCounterSetCount() == 1);
+ BOOST_CHECK(counterSet);
+
+ // Buffer with enough space
+ MockBuffer mockBuffer(1024);
+ SendCounterPacket sendCounterPacket(mockBuffer);
+ BOOST_CHECK_THROW(sendCounterPacket.SendCounterDirectoryPacket(counterDirectory), armnn::RuntimeException);
+}
+
+BOOST_AUTO_TEST_CASE(SendCounterDirectoryPacketTest5)
+{
+ // Using a mock counter directory that allows to register invalid objects
+ MockCounterDirectory counterDirectory;
+
+ // Register an invalid category
+ const std::string categoryName = "c@t€gory";
+ const Category* category = nullptr;
+ BOOST_CHECK_NO_THROW(category = counterDirectory.RegisterCategory(categoryName));
+ BOOST_CHECK(counterDirectory.GetCategoryCount() == 1);
+ BOOST_CHECK(category);
+
+ // Buffer with enough space
+ MockBuffer mockBuffer(1024);
+ SendCounterPacket sendCounterPacket(mockBuffer);
+ BOOST_CHECK_THROW(sendCounterPacket.SendCounterDirectoryPacket(counterDirectory), armnn::RuntimeException);
+}
+
+BOOST_AUTO_TEST_CASE(SendCounterDirectoryPacketTest6)
+{
+ // Using a mock counter directory that allows to register invalid objects
+ MockCounterDirectory counterDirectory;
+
+ // Register an invalid device
+ const std::string deviceName = "inv@lid dev!c€";
+ const Device* device = nullptr;
+ BOOST_CHECK_NO_THROW(device = counterDirectory.RegisterDevice(deviceName, 3));
+ BOOST_CHECK(counterDirectory.GetDeviceCount() == 1);
+ BOOST_CHECK(device);
+
+ // Register an invalid counter set
+ const std::string counterSetName = "inv@lid count€rs€t";
+ const CounterSet* counterSet = nullptr;
+ BOOST_CHECK_NO_THROW(counterSet = counterDirectory.RegisterCounterSet(counterSetName));
+ BOOST_CHECK(counterDirectory.GetCounterSetCount() == 1);
+ BOOST_CHECK(counterSet);
+
+ // Register an invalid category associated to an invalid device and an invalid counter set
+ const std::string categoryName = "c@t€gory";
+ const Category* category = nullptr;
+ BOOST_CHECK_NO_THROW(category = counterDirectory.RegisterCategory(categoryName,
+ device->m_Uid,
+ counterSet->m_Uid));
+ BOOST_CHECK(counterDirectory.GetCategoryCount() == 1);
+ BOOST_CHECK(category);
+
+ // Buffer with enough space
+ MockBuffer mockBuffer(1024);
+ SendCounterPacket sendCounterPacket(mockBuffer);
+ BOOST_CHECK_THROW(sendCounterPacket.SendCounterDirectoryPacket(counterDirectory), armnn::RuntimeException);
+}
+
+BOOST_AUTO_TEST_CASE(SendCounterDirectoryPacketTest7)
+{
+ // Using a mock counter directory that allows to register invalid objects
+ MockCounterDirectory counterDirectory;
+
+ // Register an valid device
+ const std::string deviceName = "valid device";
+ const Device* device = nullptr;
+ BOOST_CHECK_NO_THROW(device = counterDirectory.RegisterDevice(deviceName, 3));
+ BOOST_CHECK(counterDirectory.GetDeviceCount() == 1);
+ BOOST_CHECK(device);
+
+ // Register an valid counter set
+ const std::string counterSetName = "valid counterset";
+ const CounterSet* counterSet = nullptr;
+ BOOST_CHECK_NO_THROW(counterSet = counterDirectory.RegisterCounterSet(counterSetName));
+ BOOST_CHECK(counterDirectory.GetCounterSetCount() == 1);
+ BOOST_CHECK(counterSet);
+
+ // Register an valid category associated to a valid device and a valid counter set
+ const std::string categoryName = "category";
+ const Category* category = nullptr;
+ BOOST_CHECK_NO_THROW(category = counterDirectory.RegisterCategory(categoryName,
+ device->m_Uid,
+ counterSet->m_Uid));
+ BOOST_CHECK(counterDirectory.GetCategoryCount() == 1);
+ BOOST_CHECK(category);
+
+ // Register an invalid counter associated to a valid category
+ const Counter* counter = nullptr;
+ BOOST_CHECK_NO_THROW(counter = counterDirectory.RegisterCounter(categoryName,
+ 0,
+ 1,
+ 123.45f,
+ "counter",
+ "counter description",
+ std::string("invalid counter units"),
+ 5,
+ device->m_Uid,
+ counterSet->m_Uid));
+ BOOST_CHECK(counterDirectory.GetCounterCount() == 5);
+ BOOST_CHECK(counter);
+
+ // Buffer with enough space
+ MockBuffer mockBuffer(1024);
+ SendCounterPacket sendCounterPacket(mockBuffer);
+ BOOST_CHECK_THROW(sendCounterPacket.SendCounterDirectoryPacket(counterDirectory), armnn::RuntimeException);
+}
BOOST_AUTO_TEST_SUITE_END()
diff --git a/src/profiling/test/SendCounterPacketTests.hpp b/src/profiling/test/SendCounterPacketTests.hpp
index a22d02bd63..6c7bb50362 100644
--- a/src/profiling/test/SendCounterPacketTests.hpp
+++ b/src/profiling/test/SendCounterPacketTests.hpp
@@ -66,7 +66,7 @@ public:
memcpy(buffer, message.c_str(), static_cast<unsigned int>(message.size()) + 1);
}
- void SendCounterDirectoryPacket(const CounterDirectory& counterDirectory) override
+ void SendCounterDirectoryPacket(const ICounterDirectory& counterDirectory) override
{
std::string message("SendCounterDirectoryPacket");
unsigned int reserved = 0;
@@ -99,3 +99,244 @@ public:
private:
IBufferWrapper& m_Buffer;
};
+
+class MockCounterDirectory : public ICounterDirectory
+{
+public:
+ MockCounterDirectory() = default;
+ ~MockCounterDirectory() = default;
+
+ // Register profiling objects
+ const Category* RegisterCategory(const std::string& categoryName,
+ const armnn::Optional<uint16_t>& deviceUid = armnn::EmptyOptional(),
+ const armnn::Optional<uint16_t>& counterSetUid = armnn::EmptyOptional())
+ {
+ // Get the device UID
+ uint16_t deviceUidValue = deviceUid.has_value() ? deviceUid.value() : 0;
+
+ // Get the counter set UID
+ uint16_t counterSetUidValue = counterSetUid.has_value() ? counterSetUid.value() : 0;
+
+ // Create the category
+ CategoryPtr category = std::make_unique<Category>(categoryName, deviceUidValue, counterSetUidValue);
+ BOOST_ASSERT(category);
+
+ // Get the raw category pointer
+ const Category* categoryPtr = category.get();
+ BOOST_ASSERT(categoryPtr);
+
+ // Register the category
+ m_Categories.insert(std::move(category));
+
+ return categoryPtr;
+ }
+
+ const Device* RegisterDevice(const std::string& deviceName,
+ uint16_t cores = 0,
+ const armnn::Optional<std::string>& parentCategoryName = armnn::EmptyOptional())
+ {
+ // Get the device UID
+ uint16_t deviceUid = GetNextUid();
+
+ // Create the device
+ DevicePtr device = std::make_unique<Device>(deviceUid, deviceName, cores);
+ BOOST_ASSERT(device);
+
+ // Get the raw device pointer
+ const Device* devicePtr = device.get();
+ BOOST_ASSERT(devicePtr);
+
+ // Register the device
+ m_Devices.insert(std::make_pair(deviceUid, std::move(device)));
+
+ // Connect the counter set to the parent category, if required
+ if (parentCategoryName.has_value())
+ {
+ // Set the counter set UID in the parent category
+ Category* parentCategory = const_cast<Category*>(GetCategory(parentCategoryName.value()));
+ BOOST_ASSERT(parentCategory);
+ parentCategory->m_DeviceUid = deviceUid;
+ }
+
+ return devicePtr;
+ }
+
+ const CounterSet* RegisterCounterSet(
+ const std::string& counterSetName,
+ uint16_t count = 0,
+ const armnn::Optional<std::string>& parentCategoryName = armnn::EmptyOptional())
+ {
+ // Get the counter set UID
+ uint16_t counterSetUid = GetNextUid();
+
+ // Create the counter set
+ CounterSetPtr counterSet = std::make_unique<CounterSet>(counterSetUid, counterSetName, count);
+ BOOST_ASSERT(counterSet);
+
+ // Get the raw counter set pointer
+ const CounterSet* counterSetPtr = counterSet.get();
+ BOOST_ASSERT(counterSetPtr);
+
+ // Register the counter set
+ m_CounterSets.insert(std::make_pair(counterSetUid, std::move(counterSet)));
+
+ // Connect the counter set to the parent category, if required
+ if (parentCategoryName.has_value())
+ {
+ // Set the counter set UID in the parent category
+ Category* parentCategory = const_cast<Category*>(GetCategory(parentCategoryName.value()));
+ BOOST_ASSERT(parentCategory);
+ parentCategory->m_CounterSetUid = counterSetUid;
+ }
+
+ return counterSetPtr;
+ }
+
+ const Counter* RegisterCounter(const std::string& parentCategoryName,
+ uint16_t counterClass,
+ uint16_t interpolation,
+ double multiplier,
+ const std::string& name,
+ const std::string& description,
+ const armnn::Optional<std::string>& units = armnn::EmptyOptional(),
+ const armnn::Optional<uint16_t>& numberOfCores = armnn::EmptyOptional(),
+ const armnn::Optional<uint16_t>& deviceUid = armnn::EmptyOptional(),
+ const armnn::Optional<uint16_t>& counterSetUid = armnn::EmptyOptional())
+ {
+ // Get the number of cores from the argument only
+ uint16_t deviceCores = numberOfCores.has_value() ? numberOfCores.value() : 0;
+
+ // Get the device UID
+ uint16_t deviceUidValue = deviceUid.has_value() ? deviceUid.value() : 0;
+
+ // Get the counter set UID
+ uint16_t counterSetUidValue = counterSetUid.has_value() ? counterSetUid.value() : 0;
+
+ // Get the counter UIDs and calculate the max counter UID
+ std::vector<uint16_t> counterUids = GetNextCounterUids(deviceCores);
+ BOOST_ASSERT(!counterUids.empty());
+ uint16_t maxCounterUid = deviceCores <= 1 ? counterUids.front() : counterUids.back();
+
+ // Get the counter units
+ const std::string unitsValue = units.has_value() ? units.value() : "";
+
+ // Create the counter
+ CounterPtr counter = std::make_shared<Counter>(counterUids.front(),
+ maxCounterUid,
+ counterClass,
+ interpolation,
+ multiplier,
+ name,
+ description,
+ unitsValue,
+ deviceUidValue,
+ counterSetUidValue);
+ BOOST_ASSERT(counter);
+
+ // Get the raw counter pointer
+ const Counter* counterPtr = counter.get();
+ BOOST_ASSERT(counterPtr);
+
+ // Process multiple counters if necessary
+ for (uint16_t counterUid : counterUids)
+ {
+ // Connect the counter to the parent category
+ Category* parentCategory = const_cast<Category*>(GetCategory(parentCategoryName));
+ BOOST_ASSERT(parentCategory);
+ parentCategory->m_Counters.push_back(counterUid);
+
+ // Register the counter
+ m_Counters.insert(std::make_pair(counterUid, counter));
+ }
+
+ return counterPtr;
+ }
+
+ // Getters for counts
+ uint16_t GetCategoryCount() const override { return boost::numeric_cast<uint16_t>(m_Categories.size()); }
+ uint16_t GetDeviceCount() const override { return boost::numeric_cast<uint16_t>(m_Devices.size()); }
+ uint16_t GetCounterSetCount() const override { return boost::numeric_cast<uint16_t>(m_CounterSets.size()); }
+ uint16_t GetCounterCount() const override { return boost::numeric_cast<uint16_t>(m_Counters.size()); }
+
+ // Getters for collections
+ const Categories& GetCategories() const override { return m_Categories; }
+ const Devices& GetDevices() const override { return m_Devices; }
+ const CounterSets& GetCounterSets() const override { return m_CounterSets; }
+ const Counters& GetCounters() const override { return m_Counters; }
+
+ // Getters for profiling objects
+ const Category* GetCategory(const std::string& name) const override
+ {
+ auto it = std::find_if(m_Categories.begin(), m_Categories.end(), [&name](const CategoryPtr& category)
+ {
+ BOOST_ASSERT(category);
+
+ return category->m_Name == name;
+ });
+
+ if (it == m_Categories.end())
+ {
+ return nullptr;
+ }
+
+ return it->get();
+ }
+
+ const Device* GetDevice(uint16_t uid) const override
+ {
+ return nullptr; // Not used by the unit tests
+ }
+
+ const CounterSet* GetCounterSet(uint16_t uid) const override
+ {
+ return nullptr; // Not used by the unit tests
+ }
+
+ const Counter* GetCounter(uint16_t uid) const override
+ {
+ return nullptr; // Not used by the unit tests
+ }
+
+private:
+ Categories m_Categories;
+ Devices m_Devices;
+ CounterSets m_CounterSets;
+ Counters m_Counters;
+};
+
+class SendCounterPacketTest : public SendCounterPacket
+{
+public:
+ SendCounterPacketTest(IBufferWrapper& buffer)
+ : SendCounterPacket(buffer)
+ {}
+
+ bool CreateDeviceRecordTest(const DevicePtr& device,
+ DeviceRecord& deviceRecord,
+ std::string& errorMessage)
+ {
+ return CreateDeviceRecord(device, deviceRecord, errorMessage);
+ }
+
+ bool CreateCounterSetRecordTest(const CounterSetPtr& counterSet,
+ CounterSetRecord& counterSetRecord,
+ std::string& errorMessage)
+ {
+ return CreateCounterSetRecord(counterSet, counterSetRecord, errorMessage);
+ }
+
+ bool CreateEventRecordTest(const CounterPtr& counter,
+ EventRecord& eventRecord,
+ std::string& errorMessage)
+ {
+ return CreateEventRecord(counter, eventRecord, errorMessage);
+ }
+
+ bool CreateCategoryRecordTest(const CategoryPtr& category,
+ const Counters& counters,
+ CategoryRecord& categoryRecord,
+ std::string& errorMessage)
+ {
+ return CreateCategoryRecord(category, counters, categoryRecord, errorMessage);
+ }
+};