1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
|
#include "./singleinstance.h"
#include "resources/config.h"
#include <c++utilities/conversion/binaryconversion.h>
#include <c++utilities/io/ansiescapecodes.h>
#include <QCoreApplication>
#include <QFile>
#include <QLocalServer>
#include <QLocalSocket>
#include <QStringBuilder>
#include <QThread>
#include <iostream>
#include <memory>
#ifdef Q_OS_WINDOWS
#include <windows.h>
// needs to be included after windows.h
#include <sddl.h>
#else
#include <unistd.h>
#endif
using namespace CppUtilities;
using namespace CppUtilities::EscapeCodes;
namespace QtGui {
#ifdef Q_OS_WINDOWS
static QString getCurrentProcessSIDAsString()
{
auto res = QString();
auto processToken = HANDLE();
if (!OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &processToken)) {
std::cerr << Phrases::Error << "Unable to determine current user: OpenProcessToken failed with " << GetLastError() << Phrases::EndFlush;
return res;
}
auto bufferSize = DWORD();
if (!GetTokenInformation(processToken, TokenUser, nullptr, 0, &bufferSize) && (GetLastError() != ERROR_INSUFFICIENT_BUFFER)) {
std::cerr << Phrases::Error << "Unable to determine current user: GetTokenInformation failed with " << GetLastError() << Phrases::EndFlush;
CloseHandle(processToken);
return res;
}
auto buffer = std::vector<BYTE>();
buffer.resize(bufferSize);
auto userToken = reinterpret_cast<PTOKEN_USER>(buffer.data());
if (!GetTokenInformation(processToken, TokenUser, userToken, bufferSize, &bufferSize)) {
std::cerr << Phrases::Error << "Unable to determine current user: GetTokenInformation failed with " << GetLastError() << Phrases::EndFlush;
CloseHandle(processToken);
return res;
}
auto stringSid = LPWSTR();
if (!ConvertSidToStringSidW(userToken->User.Sid, &stringSid)) {
std::cerr << Phrases::Error << "Unable to determine current user: ConvertSidToStringSid failed with " << GetLastError() << Phrases::EndFlush;
CloseHandle(processToken);
return res;
}
res = QString::fromWCharArray(stringSid);
LocalFree(stringSid);
CloseHandle(processToken);
return res;
}
#endif
SingleInstance::SingleInstance(int argc, const char *const *argv, bool skipSingleInstanceBehavior, bool skipPassing, QObject *parent)
: QObject(parent)
, m_server(nullptr)
{
// just do nothing if supposed to skip single instance behavior
if (skipSingleInstanceBehavior) {
return;
}
// check for running instance; if there is one pass parameters and exit
static const auto appId = applicationId();
if (!skipPassing && passArgsToRunningInstance(argc, argv, appId)) {
std::exit(EXIT_SUCCESS);
}
// create local server; at this point no previous instance is running anymore
// -> cleanup possible leftover (previous instance might have crashed)
QLocalServer::removeServer(appId);
// -> setup server
m_server = new QLocalServer(this);
connect(m_server, &QLocalServer::newConnection, this, &SingleInstance::handleNewConnection);
if (!m_server->listen(appId)) {
std::cerr << Phrases::Error << "Unable to launch as single instance application as " << appId.toStdString() << Phrases::EndFlush;
} else {
std::cerr << Phrases::Info << "Single instance application ID: " << appId.toStdString() << Phrases::EndFlush;
}
}
const QString &SingleInstance::applicationId()
{
static const auto envOverride = qEnvironmentVariable(PROJECT_VARNAME_UPPER "_SINGLE_INSTANCE_ID");
if (!envOverride.isEmpty()) {
return envOverride;
}
static const auto id = QString(QCoreApplication::applicationName() % QChar('-') % QCoreApplication::organizationName() % QChar('-') %
#ifdef Q_OS_WINDOWS
getCurrentProcessSIDAsString()
#else
QString::number(getuid())
#endif
);
return id;
}
bool SingleInstance::passArgsToRunningInstance(int argc, const char *const *argv, const QString &appId, bool waitUntilGone)
{
if (argc < 0 || argc > 0xFFFF) {
std::cerr << Phrases::Error << "Unable to pass the specified number of arguments" << Phrases::EndFlush;
return false;
}
auto socket = QLocalSocket();
socket.connectToServer(appId, QLocalSocket::ReadWrite);
const auto fullServerName = socket.fullServerName();
if (!socket.waitForConnected(1000)) {
return false;
}
std::cerr << Phrases::Info << "Application already running, sending args to previous instance" << Phrases::EndFlush;
char buffer[2];
BE::getBytes(static_cast<std::uint16_t>(argc), buffer);
auto error = socket.write(buffer, 2) < 0;
*buffer = '\0';
for (const char *const *end = argv + argc; argv != end && !error; ++argv) {
error = socket.write(*argv) < 0 || socket.write(buffer, 1) < 0;
}
error = error || !socket.waitForBytesWritten(1000);
socket.disconnectFromServer();
if (socket.state() != QLocalSocket::UnconnectedState) {
error = !socket.waitForDisconnected(1000) || error;
}
if (error) {
std::cerr << Phrases::Error << "Unable to pass args to previous instance: " << socket.errorString().toStdString() << Phrases::EndFlush;
}
if (waitUntilGone && QFile::exists(fullServerName)) {
const auto fullServerNameStd = fullServerName.toStdString();
std::cerr << Phrases::Info << "Waiting for previous instance to shutdown (" << fullServerNameStd << " still exists)" << Phrases::EndFlush;
do {
QThread::msleep(500);
} while (QFile::exists(fullServerName));
}
return !error;
}
void SingleInstance::handleNewConnection()
{
const QLocalSocket *const socket = m_server->nextPendingConnection();
connect(socket, &QLocalSocket::readChannelFinished, this, &SingleInstance::readArgs);
}
void SingleInstance::readArgs()
{
auto *const socket = static_cast<QLocalSocket *>(sender());
const auto argData = socket->readAll();
if (argData.size() < 2) {
std::cerr << Phrases::Error << "Another application instance sent invalid argument data (payload only " << argData.size() << " bytes)."
<< Phrases::EndFlush;
return;
}
socket->close();
socket->deleteLater();
// reconstruct argc and argv array
const auto argc = BE::toInt<std::uint16_t>(argData.data());
auto args = std::vector<const char *>();
args.reserve(argc + 1);
std::cerr << Phrases::Info << "Evaluating " << argc << " arguments from another instance: " << Phrases::End;
for (const char *argv = argData.data() + 2, *end = argData.data() + argData.size(), *i = argv; i != end && *argv;) {
if (!*i) {
args.push_back(argv);
std::cerr << ' ' << argv;
argv = ++i;
} else {
++i;
}
}
args.push_back(nullptr);
std::cerr << '\n';
emit newInstance(static_cast<int>(args.size() - 1), args.data());
}
} // namespace QtGui
|
