feat(message): Implemented message class

Contains the full can message from one or more frames.

2 files changed, 64 insertions, 0 deletions

diff --git a/src/message.cpp b/src/message.cpp
new file mode 100644
index 0000000..dfeb582
--- /dev/null
+++ b/src/message.cpp
@@ -0,0 +1,33 @@
+#include <cstring>
+#include "message.h"
+
+Message::Message(unsigned int can_id, const Frame& first_frame) : can_id(can_id), received(0) {
+	this->size = first_frame.get_size();
+	this->msg = (unsigned char*)malloc(sizeof(unsigned char) * this->size);
+	this->receive_frame(first_frame);
+}
+
+Message::Message(const Message& rhs) : can_id(rhs.can_id), size(rhs.size), received(rhs.received) {
+	this->msg = (unsigned char*)malloc(sizeof(unsigned char) * this->size);
+	std::memcpy(this->msg, rhs.msg, this->size);
+}
+
+void Message::receive_frame(const Frame& f) {
+	size_t expects = std::min(this->size - this->received, f.get_data_size());
+	std::memcpy(this->msg + this->received, f.get_data(), expects);
+	this->received += expects;
+}
+
+bool Message::expects_more_frames() const {
+	return this->received < this->size;
+}
+
+std::ostream& operator<<(std::ostream& os, const Message& m) {
+	os << m.can_id << ":";
+	for (int i = 0; i < m.received; ++i) {
+		os << " " << (unsigned)m.msg[i];
+	}
+	if (m.expects_more_frames())
+		os << "...";
+	return os;
+}
diff --git a/src/message.h b/src/message.h
new file mode 100644
index 0000000..c988059
--- /dev/null
+++ b/src/message.h
@@ -0,0 +1,31 @@
+#ifndef ISO_TP_MESSAGE
+#define ISO_TP_MESSAGE
+
+#include <ostream>
+#include "frame.h"
+
+class Message {
+	unsigned int can_id;
+	unsigned int size;
+	unsigned int received;
+	unsigned char* msg;
+
+public:
+	Message() = default;
+	Message(unsigned int can_id, const Frame& first_frame);
+	Message(const Message& rhs);
+
+	~Message() {
+		delete[] this->msg;
+	}
+
+	void receive_frame(const Frame& f);
+
+	bool expects_more_frames() const;
+
+	friend std::ostream& operator<<(std::ostream& os, const Message& m);
+};
+
+std::ostream& operator<<(std::ostream& os, const Message& m);
+
+#endif /* ISO_TP_MESSAGE */