message sub directory

git-svn-id: svn://elaine/murooma/trunk@287 d8a302eb-03bc-478d-80e4-98257eca68ef

message/message.h

@@ -0,0 +1,186 @@
+#ifndef MESSAGE_H
+#define MESSAGE_H
+
+#include <cstring>
+#include <iostream>
+#include <streambuf>
+#include <vector>
+#include <sys/time.h>
+
+
+template<typename CharT, typename TraitsT = std::char_traits<CharT> >
+class vectorwrapbuf : public std::basic_streambuf<CharT, TraitsT>
+{
+public:
+	vectorwrapbuf(std::vector<CharT> &vec)
+	{
+		this->setg(vec.data(), vec.data(), vec.data() + vec.size());
+	}
+};
+
+
+struct membuf : public std::basic_streambuf<char>
+{
+	membuf(char* begin, char* end)
+	{
+		this->setg(begin, begin, end);
+	}
+};
+
+
+enum message_type
+{
+	base = 0,
+	header = 1,
+	payload = 2,
+	sampleformat = 3,
+	serversettings = 4,
+	timemsg = 5,
+	requestmsg = 6,
+	ackMsg = 7
+};
+
+
+
+struct tv
+{
+	tv()
+	{
+		timeval t;
+		gettimeofday(&t, NULL);
+		sec = t.tv_sec;
+		usec = t.tv_usec;
+	}
+	tv(timeval tv) : sec(tv.tv_sec), usec(tv.tv_usec) {};
+	tv(int32_t _sec, int32_t _usec) : sec(_sec), usec(_usec) {};
+
+	int32_t sec;
+	int32_t usec;
+	/*
+	5.3 - 6.2 = -0.9
+	-1
+	0.1
+
+	5.3 - 6.4 = -1.1
+	-1
+	-0.1
+	*/
+//(timeMsg.received.sec - timeMsg.sent.sec) * 1000000 + (timeMsg.received.usec - timeMsg.sent.usec)
+	tv operator-(const tv& other)
+	{
+		tv result(*this);
+		result.sec -= other.sec;
+		result.usec -= other.usec;
+		if (result.usec > 0)
+		{
+			result.sec += 1;
+			result.usec = 1000000 - result.usec;
+		}
+		else if (result.usec < 0)
+		{
+			result.usec *= -1;
+		}
+		/*		else if (result.usec >= 1000000)
+				{
+					result.usec -= 1000000;
+					result.sec += 1;
+				}
+		*/		return result;
+	}
+};
+
+
+struct BaseMessage
+{
+	BaseMessage() : type(base), id(0), refersTo(0)
+	{
+	}
+
+	BaseMessage(message_type type_) : type(type_), id(0), refersTo(0)
+	{
+	}
+
+	virtual ~BaseMessage()
+	{
+	}
+
+	virtual void read(std::istream& stream)
+	{
+		stream.read(reinterpret_cast<char*>(&type), sizeof(uint16_t));
+		stream.read(reinterpret_cast<char*>(&id), sizeof(uint16_t));
+		stream.read(reinterpret_cast<char*>(&refersTo), sizeof(uint16_t));
+		stream.read(reinterpret_cast<char *>(&sent.sec), sizeof(int32_t));
+		stream.read(reinterpret_cast<char *>(&sent.usec), sizeof(int32_t));
+		stream.read(reinterpret_cast<char *>(&received.sec), sizeof(int32_t));
+		stream.read(reinterpret_cast<char *>(&received.usec), sizeof(int32_t));
+		stream.read(reinterpret_cast<char*>(&size), sizeof(uint32_t));
+	}
+
+	void deserialize(char* payload)
+	{
+		membuf databuf(payload, payload + BaseMessage::getSize());
+		std::istream is(&databuf);
+		read(is);
+	}
+
+	void deserialize(const BaseMessage& baseMessage, char* payload)
+	{
+		type = baseMessage.type;
+		id = baseMessage.id;
+		refersTo = baseMessage.refersTo;
+		sent = baseMessage.sent;
+		received = baseMessage.received;
+		size = baseMessage.size;
+		membuf databuf(payload, payload + size);
+		std::istream is(&databuf);
+		read(is);
+	}
+
+	virtual void serialize(std::ostream& stream)
+	{
+		stream.write(reinterpret_cast<char*>(&type), sizeof(uint16_t));
+		stream.write(reinterpret_cast<char*>(&id), sizeof(uint16_t));
+		stream.write(reinterpret_cast<char*>(&refersTo), sizeof(uint16_t));
+		stream.write(reinterpret_cast<char *>(&sent.sec), sizeof(int32_t));
+		stream.write(reinterpret_cast<char *>(&sent.usec), sizeof(int32_t));
+		stream.write(reinterpret_cast<char *>(&received.sec), sizeof(int32_t));
+		stream.write(reinterpret_cast<char *>(&received.usec), sizeof(int32_t));
+		size = getSize();
+		stream.write(reinterpret_cast<char*>(&size), sizeof(uint32_t));
+		doserialize(stream);
+	}
+
+	virtual uint32_t getSize()
+	{
+		return 3*sizeof(uint16_t) + 2*sizeof(tv) + sizeof(uint32_t);
+	};
+
+	uint16_t type;
+	uint16_t id;
+	uint16_t refersTo;
+	tv sent;
+	tv received;
+	uint32_t size;
+
+protected:
+	virtual void doserialize(std::ostream& stream)
+	{
+	};
+};
+
+
+struct SerializedMessage
+{
+	~SerializedMessage()
+	{
+		free(buffer);
+	}
+
+	BaseMessage message;
+	char* buffer;
+};
+
+
+#endif
+
+