146 lines
3.6 KiB
C++
146 lines
3.6 KiB
C++
#include "discord_rpc_connection.h"
|
|
#include "discord_serialization.h"
|
|
|
|
namespace discord_rpc {
|
|
|
|
static const int RpcVersion = 1;
|
|
static RpcConnection Instance;
|
|
|
|
RpcConnection *RpcConnection::Create(const char *applicationId) {
|
|
|
|
Instance.connection = BaseConnection::Create();
|
|
StringCopy(Instance.appId, applicationId);
|
|
return &Instance;
|
|
|
|
}
|
|
|
|
void RpcConnection::Destroy(RpcConnection *&c) {
|
|
|
|
c->Close();
|
|
BaseConnection::Destroy(c->connection);
|
|
c = nullptr;
|
|
|
|
}
|
|
|
|
void RpcConnection::Open() {
|
|
|
|
if (state == State::Connected) {
|
|
return;
|
|
}
|
|
|
|
if (state == State::Disconnected && !connection->Open()) {
|
|
return;
|
|
}
|
|
|
|
if (state == State::SentHandshake) {
|
|
JsonDocument message;
|
|
if (Read(message)) {
|
|
auto cmd = GetStrMember(&message, "cmd");
|
|
auto evt = GetStrMember(&message, "evt");
|
|
if (cmd && evt && !strcmp(cmd, "DISPATCH") && !strcmp(evt, "READY")) {
|
|
state = State::Connected;
|
|
if (onConnect) {
|
|
onConnect(message);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
sendFrame.opcode = Opcode::Handshake;
|
|
sendFrame.length = static_cast<uint32_t>(JsonWriteHandshakeObj(sendFrame.message, sizeof(sendFrame.message), RpcVersion, appId));
|
|
|
|
if (connection->Write(&sendFrame, sizeof(MessageFrameHeader) + sendFrame.length)) {
|
|
state = State::SentHandshake;
|
|
}
|
|
else {
|
|
Close();
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
void RpcConnection::Close() {
|
|
|
|
if (onDisconnect && (state == State::Connected || state == State::SentHandshake)) {
|
|
onDisconnect(lastErrorCode, lastErrorMessage);
|
|
}
|
|
connection->Close();
|
|
state = State::Disconnected;
|
|
|
|
}
|
|
|
|
bool RpcConnection::Write(const void *data, size_t length) {
|
|
|
|
sendFrame.opcode = Opcode::Frame;
|
|
memcpy(sendFrame.message, data, length);
|
|
sendFrame.length = static_cast<uint32_t>(length);
|
|
if (!connection->Write(&sendFrame, sizeof(MessageFrameHeader) + length)) {
|
|
Close();
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
bool RpcConnection::Read(JsonDocument &message) {
|
|
|
|
if (state != State::Connected && state != State::SentHandshake) {
|
|
return false;
|
|
}
|
|
MessageFrame readFrame{};
|
|
for (;;) {
|
|
bool didRead = connection->Read(&readFrame, sizeof(MessageFrameHeader));
|
|
if (!didRead) {
|
|
if (!connection->isOpen) {
|
|
lastErrorCode = static_cast<int>(ErrorCode::PipeClosed);
|
|
StringCopy(lastErrorMessage, "Pipe closed");
|
|
Close();
|
|
}
|
|
return false;
|
|
}
|
|
|
|
if (readFrame.length > 0) {
|
|
didRead = connection->Read(readFrame.message, readFrame.length);
|
|
if (!didRead) {
|
|
lastErrorCode = static_cast<int>(ErrorCode::ReadCorrupt);
|
|
StringCopy(lastErrorMessage, "Partial data in frame");
|
|
Close();
|
|
return false;
|
|
}
|
|
readFrame.message[readFrame.length] = 0;
|
|
}
|
|
|
|
switch (readFrame.opcode) {
|
|
case Opcode::Close: {
|
|
message.ParseInsitu(readFrame.message);
|
|
lastErrorCode = GetIntMember(&message, "code");
|
|
StringCopy(lastErrorMessage, GetStrMember(&message, "message", ""));
|
|
Close();
|
|
return false;
|
|
}
|
|
case Opcode::Frame:
|
|
message.ParseInsitu(readFrame.message);
|
|
return true;
|
|
case Opcode::Ping:
|
|
readFrame.opcode = Opcode::Pong;
|
|
if (!connection->Write(&readFrame, sizeof(MessageFrameHeader) + readFrame.length)) {
|
|
Close();
|
|
}
|
|
break;
|
|
case Opcode::Pong:
|
|
break;
|
|
case Opcode::Handshake:
|
|
default:
|
|
// something bad happened
|
|
lastErrorCode = static_cast<int>(ErrorCode::ReadCorrupt);
|
|
StringCopy(lastErrorMessage, "Bad ipc frame");
|
|
Close();
|
|
return false;
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
} // namespace discord_rpc
|