161 lines
3.8 KiB
C++
161 lines
3.8 KiB
C++
/*
|
|
* Copyright 2017 Discord, Inc.
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy of
|
|
* this software and associated documentation files (the "Software"), to deal in
|
|
* the Software without restriction, including without limitation the rights to
|
|
* use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
|
|
* of the Software, and to permit persons to whom the Software is furnished to do
|
|
* so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included in all
|
|
* copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
* SOFTWARE.
|
|
*
|
|
*/
|
|
|
|
#include "discord_connection.h"
|
|
|
|
#include <cerrno>
|
|
#include <fcntl.h>
|
|
#include <cstdio>
|
|
#include <cstring>
|
|
#include <sys/socket.h>
|
|
#include <sys/types.h>
|
|
#include <sys/un.h>
|
|
#include <unistd.h>
|
|
|
|
namespace discord_rpc {
|
|
|
|
int GetProcessId() {
|
|
return ::getpid();
|
|
}
|
|
|
|
struct BaseConnectionUnix : public BaseConnection {
|
|
int sock { -1 };
|
|
};
|
|
|
|
static BaseConnectionUnix Connection;
|
|
static sockaddr_un PipeAddr {};
|
|
#ifdef MSG_NOSIGNAL
|
|
static int MsgFlags = MSG_NOSIGNAL;
|
|
#else
|
|
static int MsgFlags = 0;
|
|
#endif
|
|
|
|
static const char *GetTempPath() {
|
|
|
|
const char *temp = getenv("XDG_RUNTIME_DIR");
|
|
temp = temp ? temp : getenv("TMPDIR");
|
|
temp = temp ? temp : getenv("TMP");
|
|
temp = temp ? temp : getenv("TEMP");
|
|
temp = temp ? temp : "/tmp";
|
|
|
|
return temp;
|
|
|
|
}
|
|
|
|
BaseConnection *BaseConnection::Create() {
|
|
PipeAddr.sun_family = AF_UNIX;
|
|
return &Connection;
|
|
}
|
|
|
|
void BaseConnection::Destroy(BaseConnection *&c) {
|
|
|
|
auto self = reinterpret_cast<BaseConnectionUnix*>(c);
|
|
self->Close();
|
|
c = nullptr;
|
|
|
|
}
|
|
|
|
bool BaseConnection::Open() {
|
|
|
|
const char *tempPath = GetTempPath();
|
|
auto self = reinterpret_cast<BaseConnectionUnix*>(this);
|
|
self->sock = socket(AF_UNIX, SOCK_STREAM, 0);
|
|
if (self->sock == -1) {
|
|
return false;
|
|
}
|
|
fcntl(self->sock, F_SETFL, O_NONBLOCK);
|
|
#ifdef SO_NOSIGPIPE
|
|
int optval = 1;
|
|
setsockopt(self->sock, SOL_SOCKET, SO_NOSIGPIPE, &optval, sizeof(optval));
|
|
#endif
|
|
|
|
for (int pipeNum = 0; pipeNum < 10; ++pipeNum) {
|
|
snprintf(PipeAddr.sun_path, sizeof(PipeAddr.sun_path), "%s/discord-ipc-%d", tempPath, pipeNum);
|
|
int err = connect(self->sock, reinterpret_cast<const sockaddr*>(&PipeAddr), sizeof(PipeAddr));
|
|
if (err == 0) {
|
|
self->isOpen = true;
|
|
return true;
|
|
}
|
|
}
|
|
self->Close();
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
bool BaseConnection::Close() {
|
|
|
|
auto self = reinterpret_cast<BaseConnectionUnix *>(this);
|
|
if (self->sock == -1) {
|
|
return false;
|
|
}
|
|
close(self->sock);
|
|
self->sock = -1;
|
|
self->isOpen = false;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
bool BaseConnection::Write(const void *data, size_t length) {
|
|
|
|
auto self = reinterpret_cast<BaseConnectionUnix*>(this);
|
|
|
|
if (self->sock == -1) {
|
|
return false;
|
|
}
|
|
|
|
ssize_t sentBytes = send(self->sock, data, length, MsgFlags);
|
|
if (sentBytes < 0) {
|
|
Close();
|
|
}
|
|
|
|
return sentBytes == static_cast<ssize_t>(length);
|
|
|
|
}
|
|
|
|
bool BaseConnection::Read(void *data, size_t length) {
|
|
|
|
auto self = reinterpret_cast<BaseConnectionUnix*>(this);
|
|
|
|
if (self->sock == -1) {
|
|
return false;
|
|
}
|
|
|
|
long res = recv(self->sock, data, length, MsgFlags);
|
|
if (res < 0) {
|
|
if (errno == EAGAIN) {
|
|
return false;
|
|
}
|
|
Close();
|
|
}
|
|
else if (res == 0) {
|
|
Close();
|
|
}
|
|
|
|
return static_cast<size_t>(res) == length;
|
|
|
|
}
|
|
|
|
} // namespace discord_rpc
|