blob: b681317cbd70be7f46cb510d6ee8d188d5e2ee9e (
plain)
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
|
/*
Copyright (C) 2012 Sebastian Herbord. All rights reserved.
This file is part of Mod Organizer.
Mod Organizer is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
Mod Organizer is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with Mod Organizer. If not, see <http://www.gnu.org/licenses/>.
*/
#include "singleinstance.h"
#include "utility.h"
#include <QLocalSocket>
#include <report.h>
static const char s_Key[] = "mo-43d1a3ad-eeb0-4818-97c9-eda5216c29b5";
static const int s_Timeout = 5000;
using MOBase::reportError;
SingleInstance::SingleInstance(bool forcePrimary, QObject* parent) : QObject(parent), m_PrimaryInstance(false) {
m_SharedMem.setKey(s_Key);
if (!m_SharedMem.create(1)) {
if (forcePrimary) {
while (m_SharedMem.error() == QSharedMemory::AlreadyExists) {
Sleep(500);
if (m_SharedMem.create(1)) {
m_PrimaryInstance = true;
break;
}
}
}
if (m_SharedMem.error() == QSharedMemory::AlreadyExists) {
m_SharedMem.attach();
m_PrimaryInstance = false;
}
if ((m_SharedMem.error() != QSharedMemory::NoError) && (m_SharedMem.error() != QSharedMemory::AlreadyExists)) {
throw MOBase::MyException(tr("SHM error: %1").arg(m_SharedMem.errorString()));
}
} else {
m_PrimaryInstance = true;
}
if (m_PrimaryInstance) {
connect(&m_Server, SIGNAL(newConnection()), this, SLOT(receiveMessage()));
// has to be called before listen
m_Server.setSocketOptions(QLocalServer::WorldAccessOption);
m_Server.listen(s_Key);
}
}
void SingleInstance::sendMessage(const QString& message) {
if (m_PrimaryInstance) {
// nobody there to receive the message
return;
}
QLocalSocket socket(this);
bool connected = false;
for (int i = 0; i < 2 && !connected; ++i) {
if (i > 0) {
Sleep(250);
}
// other instance may be just starting up
socket.connectToServer(s_Key, QIODevice::WriteOnly);
connected = socket.waitForConnected(s_Timeout);
}
if (!connected) {
reportError(tr("failed to connect to running instance: %1").arg(socket.errorString()));
return;
}
socket.write(message.toUtf8());
if (!socket.waitForBytesWritten(s_Timeout)) {
reportError(tr("failed to communicate with running instance: %1").arg(socket.errorString()));
return;
}
socket.disconnectFromServer();
}
void SingleInstance::receiveMessage() {
QLocalSocket* socket = m_Server.nextPendingConnection();
if (!socket->waitForReadyRead(s_Timeout)) {
reportError(tr("failed to receive data from secondary instance: %1").arg(socket->errorString()));
return;
}
QString message = QString::fromUtf8(socket->readAll().constData());
emit messageSent(message);
socket->disconnectFromServer();
}
|
