mirror of
https://github.com/Chatterino/chatterino2.git
synced 2024-11-13 19:49:51 +01:00
91 lines
2.2 KiB
C++
91 lines
2.2 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)
|
|
{
|
|
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';
|
|
|
|
client.sendMessage(QByteArray::fromRawData(
|
|
buffer.get(), static_cast<int32_t>(size)));
|
|
}
|
|
}
|
|
} // namespace
|
|
|
|
bool shouldRunBrowserExtensionHost(const QStringList &args)
|
|
{
|
|
return args.size() > 0 && (args[0].startsWith("chrome-extension://") ||
|
|
args[0].endsWith(".json"));
|
|
}
|
|
|
|
void runBrowserExtensionHost()
|
|
{
|
|
initFileMode();
|
|
|
|
std::atomic<bool> ping(false);
|
|
|
|
QTimer timer;
|
|
QObject::connect(&timer, &QTimer::timeout, [&ping] {
|
|
if (!ping.exchange(false))
|
|
{
|
|
_Exit(0);
|
|
}
|
|
});
|
|
timer.setInterval(11000);
|
|
timer.start();
|
|
|
|
NativeMessagingClient client;
|
|
|
|
runLoop(client);
|
|
}
|
|
|
|
} // namespace chatterino
|