mirror of
https://github.com/wpilibsuite/allwpilib
synced 2026-06-21 01:01:43 +00:00
114 lines
4.2 KiB
C++
114 lines
4.2 KiB
C++
/*----------------------------------------------------------------------------*/
|
|
/* Copyright (c) FIRST 2015. All Rights Reserved. */
|
|
/* Open Source Software - may be modified and shared by FRC teams. The code */
|
|
/* must be accompanied by the FIRST BSD license file in the root directory of */
|
|
/* the project. */
|
|
/*----------------------------------------------------------------------------*/
|
|
|
|
#ifndef NT_MESSAGE_H_
|
|
#define NT_MESSAGE_H_
|
|
|
|
#include <memory>
|
|
|
|
#include "Value.h"
|
|
|
|
namespace ntimpl {
|
|
|
|
class WireDecoder;
|
|
class WireEncoder;
|
|
|
|
class Message {
|
|
struct private_init {};
|
|
|
|
public:
|
|
enum MsgType {
|
|
kUnknown = -1,
|
|
kKeepAlive = 0x00,
|
|
kClientHello = 0x01,
|
|
kProtoUnsup = 0x02,
|
|
kServerHelloDone = 0x03,
|
|
kServerHello = 0x04,
|
|
kClientHelloDone = 0x05,
|
|
kEntryAssign = 0x10,
|
|
kEntryUpdate = 0x11,
|
|
kFlagsUpdate = 0x12,
|
|
kEntryDelete = 0x13,
|
|
kClearEntries = 0x14,
|
|
kExecuteRpc = 0x20,
|
|
kRpcResponse = 0x21
|
|
};
|
|
typedef NT_Type (*GetEntryTypeFunc)(unsigned int id);
|
|
|
|
Message() : m_type(kUnknown), m_id(0), m_flags(0), m_seq_num_uid(0) {}
|
|
Message(MsgType type, const private_init&)
|
|
: m_type(type), m_id(0), m_flags(0), m_seq_num_uid(0) {}
|
|
|
|
MsgType type() const { return m_type; }
|
|
bool Is(MsgType type) const { return type == m_type; }
|
|
|
|
// Read and write from wire representation
|
|
void Write(WireEncoder& encoder) const;
|
|
static std::shared_ptr<Message> Read(WireDecoder& decoder,
|
|
GetEntryTypeFunc get_entry_type);
|
|
|
|
// Create messages without data
|
|
static std::shared_ptr<Message> KeepAlive() {
|
|
return std::make_shared<Message>(kKeepAlive, private_init());
|
|
}
|
|
static std::shared_ptr<Message> ProtoUnsup() {
|
|
return std::make_shared<Message>(kProtoUnsup, private_init());
|
|
}
|
|
static std::shared_ptr<Message> ServerHelloDone() {
|
|
return std::make_shared<Message>(kServerHelloDone, private_init());
|
|
}
|
|
static std::shared_ptr<Message> ClientHelloDone() {
|
|
return std::make_shared<Message>(kClientHelloDone, private_init());
|
|
}
|
|
static std::shared_ptr<Message> ClearEntries() {
|
|
return std::make_shared<Message>(kClearEntries, private_init());
|
|
}
|
|
|
|
// Create messages with data
|
|
static std::shared_ptr<Message> ClientHello(llvm::StringRef self_id);
|
|
static std::shared_ptr<Message> ServerHello(unsigned int flags,
|
|
llvm::StringRef self_id);
|
|
static std::shared_ptr<Message> EntryAssign(llvm::StringRef name,
|
|
unsigned int id,
|
|
unsigned int seq_num,
|
|
std::shared_ptr<Value> value,
|
|
unsigned int flags);
|
|
static std::shared_ptr<Message> EntryUpdate(unsigned int id,
|
|
unsigned int seq_num,
|
|
std::shared_ptr<Value> value);
|
|
static std::shared_ptr<Message> FlagsUpdate(unsigned int id,
|
|
unsigned int flags);
|
|
static std::shared_ptr<Message> EntryDelete(unsigned int id);
|
|
static std::shared_ptr<Message> ExecuteRpc(unsigned int id, unsigned int uid,
|
|
llvm::ArrayRef<NT_Value> params);
|
|
static std::shared_ptr<Message> ExecuteRpc(unsigned int id, unsigned int uid,
|
|
llvm::StringRef params);
|
|
static std::shared_ptr<Message> RpcResponse(unsigned int id, unsigned int uid,
|
|
llvm::ArrayRef<NT_Value> results);
|
|
static std::shared_ptr<Message> RpcResponse(unsigned int id, unsigned int uid,
|
|
llvm::StringRef results);
|
|
|
|
Message(const Message&) = delete;
|
|
Message& operator=(const Message&) = delete;
|
|
Message(Message&&) = default;
|
|
Message& operator=(Message&&) = default;
|
|
|
|
private:
|
|
MsgType m_type;
|
|
|
|
// Message data. Use varies by message type.
|
|
StringValue m_str;
|
|
std::shared_ptr<Value> m_value;
|
|
unsigned int m_id; // also used for proto_rev
|
|
unsigned int m_flags;
|
|
unsigned int m_seq_num_uid;
|
|
};
|
|
|
|
} // namespace ntimpl
|
|
|
|
#endif // NT_MESSAGE_H_
|