mirror of
https://github.com/Chatterino/chatterino2.git
synced 2024-11-13 19:49:51 +01:00
105 lines
2.6 KiB
C++
105 lines
2.6 KiB
C++
#include "BrowserExtension.hpp"
|
|
|
|
#include "singletons/NativeMessaging.hpp"
|
|
|
|
#include <QStringList>
|
|
#include <QTimer>
|
|
#include <fstream>
|
|
#include <iostream>
|
|
#include <memory>
|
|
|
|
#ifdef Q_OS_WIN
|
|
# include <fcntl.h>
|
|
# include <io.h>
|
|
# include <stdio.h>
|
|
#endif
|
|
|
|
namespace chatterino {
|
|
|
|
namespace {
|
|
void initFileMode()
|
|
{
|
|
#ifdef Q_OS_WIN
|
|
_setmode(_fileno(stdin), _O_BINARY);
|
|
_setmode(_fileno(stdout), _O_BINARY);
|
|
#endif
|
|
}
|
|
|
|
void runLoop(NativeMessagingClient &client)
|
|
{
|
|
auto received_message = std::make_shared<std::atomic_bool>(true);
|
|
|
|
auto thread = std::thread([=]() {
|
|
while (true)
|
|
{
|
|
using namespace std::chrono_literals;
|
|
if (!received_message->exchange(false))
|
|
{
|
|
_Exit(1);
|
|
}
|
|
std::this_thread::sleep_for(5s);
|
|
}
|
|
});
|
|
|
|
while (true)
|
|
{
|
|
char size_c[4];
|
|
std::cin.read(size_c, 4);
|
|
|
|
if (std::cin.eof())
|
|
{
|
|
break;
|
|
}
|
|
|
|
auto size = *reinterpret_cast<uint32_t *>(size_c);
|
|
|
|
#if 0
|
|
bool bigEndian = isBigEndian();
|
|
// To avoid breaking strict-aliasing rules and potentially inducing undefined behaviour, the following code can be run instead
|
|
uint32_t size = 0;
|
|
if (bigEndian) {
|
|
size = size_c[3] | static_cast<uint32_t>(size_c[2]) << 8 |
|
|
static_cast<uint32_t>(size_c[1]) << 16 | static_cast<uint32_t>(size_c[0]) << 24;
|
|
} else {
|
|
size = size_c[0] | static_cast<uint32_t>(size_c[1]) << 8 |
|
|
static_cast<uint32_t>(size_c[2]) << 16 | static_cast<uint32_t>(size_c[3]) << 24;
|
|
}
|
|
#endif
|
|
|
|
std::unique_ptr<char[]> buffer(new char[size + 1]);
|
|
std::cin.read(buffer.get(), size);
|
|
*(buffer.get() + size) = '\0';
|
|
|
|
auto data = QByteArray::fromRawData(buffer.get(),
|
|
static_cast<int32_t>(size));
|
|
auto doc = QJsonDocument();
|
|
|
|
if (doc.object().value("type") == "nm_pong")
|
|
{
|
|
received_message->store(true);
|
|
}
|
|
|
|
received_message->store(true);
|
|
|
|
client.sendMessage(data);
|
|
}
|
|
}
|
|
} // namespace
|
|
|
|
bool shouldRunBrowserExtensionHost(const QStringList &args)
|
|
{
|
|
return args.size() > 0 && (args[0].startsWith("chrome-extension://") ||
|
|
args[0].endsWith(".json"));
|
|
}
|
|
|
|
void runBrowserExtensionHost()
|
|
{
|
|
initFileMode();
|
|
|
|
NativeMessagingClient client;
|
|
|
|
runLoop(client);
|
|
}
|
|
|
|
} // namespace chatterino
|