Rename some things, stub of async version, download rapidjson in cmake file
This commit is contained in:
133
src/connection_win.cpp
Normal file
133
src/connection_win.cpp
Normal file
@ -0,0 +1,133 @@
|
||||
#include "connection.h"
|
||||
|
||||
#include <stdio.h>
|
||||
|
||||
#define WIN32_LEAN_AND_MEAN
|
||||
#define NOMCX
|
||||
#define NOSERVICE
|
||||
#define NOIME
|
||||
#include <windows.h>
|
||||
|
||||
#include "yolojson.h"
|
||||
|
||||
const int RpcVersion = 1;
|
||||
const int NumFrames = 3;
|
||||
static int LastErrorCode = 0;
|
||||
static const char* LastErrorMessage = "";
|
||||
|
||||
struct WinRpcConnection : public RpcConnection {
|
||||
HANDLE pipe{INVALID_HANDLE_VALUE};
|
||||
RpcMessageFrame frames[NumFrames];
|
||||
int nextFrame{0};
|
||||
int lastErrorCode{0};
|
||||
char lastErrorMessage[1024];
|
||||
|
||||
void HandleError(RpcMessageFrame* frame) {
|
||||
if (frame->opcode == OPCODE::CLOSE) {
|
||||
lastErrorCode = 1; // todo
|
||||
StringCopy(lastErrorMessage, frame->message, sizeof(lastErrorMessage));
|
||||
printf("got a close message: %d: %s\n", lastErrorCode, lastErrorMessage);
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
static const wchar_t* PipeName = L"\\\\?\\pipe\\discord-ipc";
|
||||
|
||||
/*static*/ RpcConnection* RpcConnection::Create(const char* applicationId)
|
||||
{
|
||||
auto connection = new WinRpcConnection;
|
||||
StringCopy(connection->appId, applicationId, sizeof(connection->appId));
|
||||
return connection;
|
||||
}
|
||||
|
||||
/*static*/ void RpcConnection::Destroy(RpcConnection*& c)
|
||||
{
|
||||
auto self = reinterpret_cast<WinRpcConnection*>(c);
|
||||
delete self;
|
||||
c = nullptr;
|
||||
}
|
||||
|
||||
void RpcConnection::Open()
|
||||
{
|
||||
auto self = reinterpret_cast<WinRpcConnection*>(this);
|
||||
for (;;) {
|
||||
self->pipe = ::CreateFileW(PipeName, GENERIC_READ | GENERIC_WRITE, 0, nullptr, OPEN_EXISTING, 0, nullptr);
|
||||
if (self->pipe != INVALID_HANDLE_VALUE) {
|
||||
break;
|
||||
}
|
||||
|
||||
if (GetLastError() != ERROR_PIPE_BUSY) {
|
||||
printf("Could not open pipe. Error: %d\n", GetLastError());
|
||||
return;
|
||||
}
|
||||
|
||||
if (!WaitNamedPipeW(PipeName, 10000)) {
|
||||
printf("Could not open pipe: 10 second wait timed out.\n");
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
RpcMessageFrame* frame = GetNextFrame();
|
||||
frame->opcode = OPCODE::HANDSHAKE;
|
||||
char* msg = frame->message;
|
||||
JsonWriteHandshakeObj(msg, RpcVersion, appId);
|
||||
frame->length = msg - frame->message;
|
||||
WriteFrame(frame);
|
||||
|
||||
if (self->onConnect) {
|
||||
self->onConnect();
|
||||
}
|
||||
}
|
||||
|
||||
void RpcConnection::Close()
|
||||
{
|
||||
auto self = reinterpret_cast<WinRpcConnection*>(this);
|
||||
::CloseHandle(self->pipe);
|
||||
self->pipe = INVALID_HANDLE_VALUE;
|
||||
if (self->onDisconnect) {
|
||||
self->onDisconnect(LastErrorCode, LastErrorMessage);
|
||||
LastErrorCode = 0;
|
||||
LastErrorMessage = "";
|
||||
}
|
||||
}
|
||||
|
||||
void RpcConnection::Write(const void* data, size_t length)
|
||||
{
|
||||
auto self = reinterpret_cast<WinRpcConnection*>(this);
|
||||
const int retries = 3;
|
||||
for (int i = 0; i < retries; ++i) {
|
||||
if (self->pipe == INVALID_HANDLE_VALUE) {
|
||||
self->Open();
|
||||
if (self->pipe == INVALID_HANDLE_VALUE) {
|
||||
break;
|
||||
}
|
||||
}
|
||||
BOOL success = ::WriteFile(self->pipe, data, length, nullptr, nullptr);
|
||||
if (success) {
|
||||
break;
|
||||
}
|
||||
LastErrorCode = -1;
|
||||
LastErrorMessage = "Pipe closed";
|
||||
self->Close();
|
||||
}
|
||||
}
|
||||
|
||||
RpcMessageFrame* RpcConnection::Read()
|
||||
{
|
||||
// todo
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
RpcMessageFrame* RpcConnection::GetNextFrame()
|
||||
{
|
||||
auto self = reinterpret_cast<WinRpcConnection*>(this);
|
||||
auto result = &(self->frames[self->nextFrame]);
|
||||
self->nextFrame = (self->nextFrame + 1) % NumFrames;
|
||||
return result;
|
||||
}
|
||||
|
||||
void RpcConnection::WriteFrame(RpcMessageFrame* frame)
|
||||
{
|
||||
auto self = reinterpret_cast<WinRpcConnection*>(this);
|
||||
self->Write(frame, 8 + frame->length);
|
||||
}
|
Reference in New Issue
Block a user