Fix this up a little so we retry a little right after failure.

This commit is contained in:
Chris Marsh 2017-07-10 16:00:05 -07:00
parent f41137e5cf
commit e3d663bc95
3 changed files with 22 additions and 12 deletions

View File

@ -11,10 +11,12 @@
#include "yolojson.h"
const int RpcVersion = 1;
const int NumFrames = 3;
struct WinRpcConnection : public RpcConnection {
HANDLE pipe{INVALID_HANDLE_VALUE};
RpcMessageFrame frame;
RpcMessageFrame frames[NumFrames];
int nextFrame{0};
};
static const wchar_t* PipeName = L"\\\\?\\pipe\\discord-ipc";
@ -39,9 +41,6 @@ void RpcConnection::Open()
for (;;) {
self->pipe = ::CreateFileW(PipeName, GENERIC_READ | GENERIC_WRITE, 0, nullptr, OPEN_EXISTING, 0, nullptr);
if (self->pipe != INVALID_HANDLE_VALUE) {
if (self->onConnect) {
self->onConnect();
}
break;
}
@ -62,6 +61,10 @@ void RpcConnection::Open()
JsonWriteHandshakeObj(msg, RpcVersion, appId);
frame->length = msg - frame->message;
WriteFrame(frame);
if (self->onConnect) {
self->onConnect();
}
}
void RpcConnection::Close()
@ -77,15 +80,18 @@ void RpcConnection::Close()
void RpcConnection::Write(const void* data, size_t length)
{
auto self = reinterpret_cast<WinRpcConnection*>(this);
if (self->pipe == INVALID_HANDLE_VALUE) {
self->Open();
const int retries = 3;
for (int i = 0; i < retries; ++i) {
if (self->pipe == INVALID_HANDLE_VALUE) {
return;
self->Open();
if (self->pipe == INVALID_HANDLE_VALUE) {
break;
}
}
BOOL success = ::WriteFile(self->pipe, data, length, nullptr, nullptr);
if (success) {
break;
}
}
BOOL success = ::WriteFile(self->pipe, data, length, nullptr, nullptr);
if (!success) {
self->Close();
}
}
@ -93,7 +99,9 @@ void RpcConnection::Write(const void* data, size_t length)
RpcMessageFrame* RpcConnection::GetNextFrame()
{
auto self = reinterpret_cast<WinRpcConnection*>(this);
return &(self->frame);
auto result = &(self->frames[self->nextFrame]);
self->nextFrame = (self->nextFrame + 1) % NumFrames;
return result;
}
void RpcConnection::WriteFrame(RpcMessageFrame* frame)

View File

@ -34,6 +34,7 @@ extern "C" void Discord_Shutdown()
extern "C" void Discord_UpdatePresence(const DiscordRichPresence* presence)
{
auto frame = MyConnection->GetNextFrame();
frame->opcode = OPCODE::FRAME;
char* jsonWrite = frame->message;
JsonWriteRichPresenceObj(jsonWrite, presence);
frame->length = jsonWrite - frame->message;

View File

@ -190,6 +190,7 @@ inline void JsonWriteRichPresenceObj(char*& dest, const DiscordRichPresence* pre
dest -= 1;
*(dest - 1) = '}';
*dest = 0;
}
inline void JsonWriteHandshakeObj(char*& dest, int version, const char* applicationId)