145 lines
2.9 KiB
C++
145 lines
2.9 KiB
C++
#pragma once
|
|
#include <stdexcept>
|
|
#include <string>
|
|
#include <vector>
|
|
#include <queue>
|
|
#include <memory>
|
|
|
|
#include "communicate/commu_options.h"
|
|
#include "utils/sha256_generator.h"
|
|
|
|
using std::string;
|
|
using std::vector;
|
|
using std::shared_ptr;
|
|
using std::queue;
|
|
|
|
namespace Net {
|
|
|
|
class Option {
|
|
public:
|
|
Option(const string &key, const string &value);
|
|
|
|
Option(const Option &o);
|
|
|
|
Option(Option &&o) noexcept;
|
|
|
|
Option& operator=(const Option &o);
|
|
|
|
Option& operator=(Option &&o) noexcept;
|
|
|
|
const string &getKey() const;
|
|
|
|
const string &getValue() const;
|
|
|
|
void updateValue(const string &value);
|
|
|
|
private:
|
|
shared_ptr<string> key;
|
|
shared_ptr<string> value;
|
|
};
|
|
|
|
class MessageFactory;
|
|
|
|
class Message {
|
|
public:
|
|
|
|
Message() noexcept;
|
|
|
|
void setHead(int32_t tid, int16_t type);
|
|
|
|
void addOption(const string &key, const string &value);
|
|
|
|
const string &getValue(const string &key) const;
|
|
|
|
void pushData(void *const buf, size_t size);
|
|
|
|
int32_t getTID() const;
|
|
|
|
const vector<char> &getData() const;
|
|
|
|
void clear();
|
|
|
|
private:
|
|
friend MessageFactory;
|
|
|
|
int32_t tid = -1;
|
|
int16_t version = MSG_VERSION;
|
|
int16_t type = 0x0;
|
|
vector<Option> options = vector<Option>();
|
|
vector<char> data = vector<char>();
|
|
static const string msg_head;
|
|
static const string msg_tail;
|
|
|
|
};
|
|
|
|
class MessageFactory {
|
|
public:
|
|
bool encodeMessage(Message &msg, vector<char> &raw_data);
|
|
|
|
bool decodeMessage(const vector<char> &raw_data, Message &msg);
|
|
|
|
bool decodeMessageHead(const vector<char> &raw_data, Message &msg);
|
|
|
|
bool decodeMessageOption(const vector<char> &raw_data, Message &msg);
|
|
|
|
bool decodeMessageBody(const vector<char> &raw_data, Message &msg);
|
|
|
|
bool decodeMessageTail(const vector<char> &raw_data, Message &msg);
|
|
|
|
bool checkMessageSign(const Message &msg);
|
|
|
|
private:
|
|
void encode_head(const Message &msg, vector<char> &raw_data);
|
|
|
|
void encode_options(const Message &msg, vector<char> &raw_data);
|
|
|
|
void encode_body(const Message &msg, vector<char> &raw_data);
|
|
|
|
void encode_tail(const Message &msg, vector<char> &raw_data);
|
|
|
|
ssize_t decode_head(const vector<char> &raw_data, Message &msg);
|
|
|
|
ssize_t decode_options(const vector<char> &raw_data, size_t offset, Message &msg);
|
|
|
|
ssize_t decode_body(const vector<char> &raw_data, size_t offset, Message &msg);
|
|
|
|
ssize_t decode_tail(const vector<char> &raw_Data, size_t offset, Message &msg);
|
|
|
|
SHA256Generator sign_gen;
|
|
};
|
|
|
|
class MessageParser {
|
|
public:
|
|
|
|
void parse(const void *buf, size_t size);
|
|
|
|
shared_ptr<Message> getMessage();
|
|
|
|
size_t getMessageCount();
|
|
|
|
private:
|
|
int head_state = -5;
|
|
int option_state = -4;
|
|
ssize_t body_state = -1;
|
|
int tail_state = -4;
|
|
MessageFactory factory;
|
|
queue<shared_ptr<Message>> msgs;
|
|
queue<char> buffer;
|
|
vector<char> temp_buffer;
|
|
|
|
shared_ptr<Message> temp_msg;
|
|
|
|
void reset_state();
|
|
|
|
void locate_head();
|
|
|
|
void locate_option();
|
|
|
|
void locate_body();
|
|
|
|
void locate_tail();
|
|
|
|
};
|
|
|
|
} // namespace Net
|