mirror of
https://github.com/badaix/snapcast.git
synced 2025-08-03 08:39:49 +02:00
move message dir into common
This commit is contained in:
badaix
parent
3f16a6ce88
commit
54a8ca824f
23 changed files with 14 additions and 12 deletions
314
common/message/message.h
Normal file
314
common/message/message.h
Normal file
|
|
@ -0,0 +1,314 @@
|
|||
/***
|
||||
This file is part of snapcast
|
||||
Copyright (C) 2014-2018 Johannes Pohl
|
||||
|
||||
This program is free software: you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
the Free Software Foundation, either version 3 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
***/
|
||||
|
||||
#ifndef MESSAGE_H
|
||||
#define MESSAGE_H
|
||||
|
||||
#include <cstdlib>
|
||||
#include <cstring>
|
||||
#include <iostream>
|
||||
#include <streambuf>
|
||||
#include <vector>
|
||||
#include <sys/time.h>
|
||||
#include "common/endian.hpp"
|
||||
#include "common/timeDefs.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
|
||||
{
|
||||
kBase = 0,
|
||||
kCodecHeader = 1,
|
||||
kWireChunk = 2,
|
||||
kServerSettings = 3,
|
||||
kTime = 4,
|
||||
kHello = 5,
|
||||
kStreamTags = 6,
|
||||
|
||||
kFirst = kBase,
|
||||
kLast = kStreamTags
|
||||
};
|
||||
|
||||
|
||||
|
||||
struct tv
|
||||
{
|
||||
tv()
|
||||
{
|
||||
timeval t;
|
||||
chronos::systemtimeofday(&t);
|
||||
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;
|
||||
|
||||
tv operator+(const tv& other) const
|
||||
{
|
||||
tv result(*this);
|
||||
result.sec += other.sec;
|
||||
result.usec += other.usec;
|
||||
if (result.usec > 1000000)
|
||||
{
|
||||
result.sec += result.usec / 1000000;
|
||||
result.usec %= 1000000;
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
tv operator-(const tv& other) const
|
||||
{
|
||||
tv result(*this);
|
||||
result.sec -= other.sec;
|
||||
result.usec -= other.usec;
|
||||
while (result.usec < 0)
|
||||
{
|
||||
result.sec -= 1;
|
||||
result.usec += 1000000;
|
||||
}
|
||||
return result;
|
||||
}
|
||||
};
|
||||
|
||||
namespace msg
|
||||
{
|
||||
|
||||
const size_t max_size = 1000000;
|
||||
|
||||
struct BaseMessage;
|
||||
|
||||
using message_ptr = std::shared_ptr<msg::BaseMessage>;
|
||||
|
||||
struct BaseMessage
|
||||
{
|
||||
BaseMessage() : type(kBase), id(0), refersTo(0)
|
||||
{
|
||||
}
|
||||
|
||||
BaseMessage(message_type type_) : type(type_), id(0), refersTo(0)
|
||||
{
|
||||
}
|
||||
|
||||
virtual ~BaseMessage()
|
||||
{
|
||||
}
|
||||
|
||||
virtual void read(std::istream& stream)
|
||||
{
|
||||
readVal(stream, type);
|
||||
readVal(stream, id);
|
||||
readVal(stream, refersTo);
|
||||
readVal(stream, sent.sec);
|
||||
readVal(stream, sent.usec);
|
||||
readVal(stream, received.sec);
|
||||
readVal(stream, received.usec);
|
||||
readVal(stream, size);
|
||||
}
|
||||
|
||||
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) const
|
||||
{
|
||||
writeVal(stream, type);
|
||||
writeVal(stream, id);
|
||||
writeVal(stream, refersTo);
|
||||
writeVal(stream, sent.sec);
|
||||
writeVal(stream, sent.usec);
|
||||
writeVal(stream, received.sec);
|
||||
writeVal(stream, received.usec);
|
||||
size = getSize();
|
||||
writeVal(stream, size);
|
||||
doserialize(stream);
|
||||
}
|
||||
|
||||
virtual uint32_t getSize() const
|
||||
{
|
||||
return 3*sizeof(uint16_t) + 2*sizeof(tv) + sizeof(uint32_t);
|
||||
};
|
||||
|
||||
uint16_t type;
|
||||
mutable uint16_t id;
|
||||
uint16_t refersTo;
|
||||
tv received;
|
||||
mutable tv sent;
|
||||
mutable uint32_t size;
|
||||
|
||||
protected:
|
||||
void writeVal(std::ostream& stream, const bool& val) const
|
||||
{
|
||||
char c = val?1:0;
|
||||
writeVal(stream, c);
|
||||
}
|
||||
|
||||
void writeVal(std::ostream& stream, const char& val) const
|
||||
{
|
||||
stream.write(reinterpret_cast<const char*>(&val), sizeof(char));
|
||||
}
|
||||
|
||||
void writeVal(std::ostream& stream, const uint16_t& val) const
|
||||
{
|
||||
uint16_t v = SWAP_16(val);
|
||||
stream.write(reinterpret_cast<const char*>(&v), sizeof(uint16_t));
|
||||
}
|
||||
|
||||
void writeVal(std::ostream& stream, const int16_t& val) const
|
||||
{
|
||||
uint16_t v = SWAP_16(val);
|
||||
stream.write(reinterpret_cast<const char*>(&v), sizeof(int16_t));
|
||||
}
|
||||
|
||||
void writeVal(std::ostream& stream, const uint32_t& val) const
|
||||
{
|
||||
uint32_t v = SWAP_32(val);
|
||||
stream.write(reinterpret_cast<const char*>(&v), sizeof(uint32_t));
|
||||
}
|
||||
|
||||
void writeVal(std::ostream& stream, const int32_t& val) const
|
||||
{
|
||||
uint32_t v = SWAP_32(val);
|
||||
stream.write(reinterpret_cast<const char*>(&v), sizeof(int32_t));
|
||||
}
|
||||
|
||||
void writeVal(std::ostream& stream, const char* payload, const uint32_t& size) const
|
||||
{
|
||||
writeVal(stream, size);
|
||||
stream.write(payload, size);
|
||||
}
|
||||
|
||||
void writeVal(std::ostream& stream, const std::string& val) const
|
||||
{
|
||||
uint32_t size = val.size();
|
||||
writeVal(stream, val.c_str(), size);
|
||||
}
|
||||
|
||||
|
||||
|
||||
void readVal(std::istream& stream, bool& val) const
|
||||
{
|
||||
char c;
|
||||
readVal(stream, c);
|
||||
val = (c != 0);
|
||||
}
|
||||
|
||||
void readVal(std::istream& stream, char& val) const
|
||||
{
|
||||
stream.read(reinterpret_cast<char*>(&val), sizeof(char));
|
||||
}
|
||||
|
||||
void readVal(std::istream& stream, uint16_t& val) const
|
||||
{
|
||||
stream.read(reinterpret_cast<char*>(&val), sizeof(uint16_t));
|
||||
val = SWAP_16(val);
|
||||
}
|
||||
|
||||
void readVal(std::istream& stream, int16_t& val) const
|
||||
{
|
||||
stream.read(reinterpret_cast<char*>(&val), sizeof(int16_t));
|
||||
val = SWAP_16(val);
|
||||
}
|
||||
|
||||
void readVal(std::istream& stream, uint32_t& val) const
|
||||
{
|
||||
stream.read(reinterpret_cast<char*>(&val), sizeof(uint32_t));
|
||||
val = SWAP_32(val);
|
||||
}
|
||||
|
||||
void readVal(std::istream& stream, int32_t& val) const
|
||||
{
|
||||
stream.read(reinterpret_cast<char*>(&val), sizeof(int32_t));
|
||||
val = SWAP_32(val);
|
||||
}
|
||||
|
||||
void readVal(std::istream& stream, char** payload, uint32_t& size) const
|
||||
{
|
||||
readVal(stream, size);
|
||||
*payload = (char*)realloc(*payload, size);
|
||||
stream.read(*payload, size);
|
||||
}
|
||||
|
||||
void readVal(std::istream& stream, std::string& val) const
|
||||
{
|
||||
uint32_t size;
|
||||
readVal(stream, size);
|
||||
val.resize(size);
|
||||
stream.read(&val[0], size);
|
||||
}
|
||||
|
||||
|
||||
virtual void doserialize(std::ostream& stream) const
|
||||
{
|
||||
};
|
||||
};
|
||||
|
||||
|
||||
struct SerializedMessage
|
||||
{
|
||||
~SerializedMessage()
|
||||
{
|
||||
free(buffer);
|
||||
}
|
||||
|
||||
BaseMessage message;
|
||||
char* buffer;
|
||||
};
|
||||
|
||||
}
|
||||
|
||||
#endif
|
||||
|
||||
|
||||
Loading…
Add table
Add a link
Reference in a new issue