git.sur5r.net Git - minitube/blob - src/qtsingleapplication/qtlocalpeer.cpp

   1 /****************************************************************************
   2 **
   3 ** This file is part of a Qt Solutions component.
   4 **
   5 ** Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).
   6 **
   7 ** Contact:  Qt Software Information (qt-info@nokia.com)
   8 **
   9 ** Commercial Usage
  10 ** Licensees holding valid Qt Commercial licenses may use this file in
  11 ** accordance with the Qt Solutions Commercial License Agreement provided
  12 ** with the Software or, alternatively, in accordance with the terms
  13 ** contained in a written agreement between you and Nokia.
  14 **
  15 ** GNU Lesser General Public License Usage
  16 ** Alternatively, this file may be used under the terms of the GNU Lesser
  17 ** General Public License version 2.1 as published by the Free Software
  18 ** Foundation and appearing in the file LICENSE.LGPL included in the
  19 ** packaging of this file.  Please review the following information to
  20 ** ensure the GNU Lesser General Public License version 2.1 requirements
  21 ** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
  22 **
  23 ** In addition, as a special exception, Nokia gives you certain
  24 ** additional rights. These rights are described in the Nokia Qt LGPL
  25 ** Exception version 1.0, included in the file LGPL_EXCEPTION.txt in this
  26 ** package.
  27 **
  28 ** GNU General Public License Usage
  29 ** Alternatively, this file may be used under the terms of the GNU
  30 ** General Public License version 3.0 as published by the Free Software
  31 ** Foundation and appearing in the file LICENSE.GPL included in the
  32 ** packaging of this file.  Please review the following information to
  33 ** ensure the GNU General Public License version 3.0 requirements will be
  34 ** met: http://www.gnu.org/copyleft/gpl.html.
  35 **
  36 ** Please note Third Party Software included with Qt Solutions may impose
  37 ** additional restrictions and it is the user's responsibility to ensure
  38 ** that they have met the licensing requirements of the GPL, LGPL, or Qt
  39 ** Solutions Commercial license and the relevant license of the Third
  40 ** Party Software they are using.
  41 **
  42 ** If you are unsure which license is appropriate for your use, please
  43 ** contact the sales department at qt-sales@nokia.com.
  44 **
  45 ****************************************************************************/
  46
  47
  48 #include "qtlocalpeer.h"
  49 #include <QtCore/QCoreApplication>
  50 #include <QtCore/QTime>
  51 #include <QDataStream>
  52 #include <QDebug>
  53
  54 #if defined(Q_OS_WIN)
  55 #include <QtCore/QLibrary>
  56 #include <QtCore/qt_windows.h>
  57 typedef BOOL(WINAPI*PProcessIdToSessionId)(DWORD,DWORD*);
  58 static PProcessIdToSessionId pProcessIdToSessionId = 0;
  59 #endif
  60 #if defined(Q_OS_UNIX)
  61 #include <unistd.h>
  62 #include <sys/types.h>
  63 #include <time.h>
  64 #endif
  65
  66 namespace QtLP_Private {
  67 #include "qtlockedfile.cpp"
  68 #if defined(Q_OS_WIN)
  69 #include "qtlockedfile_win.cpp"
  70 #else
  71 #include "qtlockedfile_unix.cpp"
  72 #endif
  73 }
  74
  75 const char* QtLocalPeer::ack = "ack";
  76
  77 QtLocalPeer::QtLocalPeer(QObject* parent, const QString &appId)
  78     : QObject(parent), id(appId)
  79 {
  80     QString prefix = id;
  81     if (id.isEmpty()) {
  82         id = QCoreApplication::applicationFilePath();
  83 #if defined(Q_OS_WIN)
  84         id = id.toLower();
  85 #endif
  86         prefix = id.section(QLatin1Char('/'), -1);
  87     }
  88     prefix.remove(QRegExp(QLatin1String("[^a-zA-Z]")));
  89     prefix.truncate(6);
  90
  91     QByteArray idc = id.toUtf8();
  92     quint16 idNum = qChecksum(idc.constData(), idc.size());
  93     socketName = QLatin1String("qtsingleapp-") + prefix
  94                  + QLatin1Char('-') + QString::number(idNum, 16);
  95
  96 #if defined(Q_OS_WIN)
  97     if (!pProcessIdToSessionId) {
  98         QLibrary lib("kernel32");
  99         pProcessIdToSessionId = (PProcessIdToSessionId)lib.resolve("ProcessIdToSessionId");
 100     }
 101     if (pProcessIdToSessionId) {
 102         DWORD sessionId = 0;
 103         pProcessIdToSessionId(GetCurrentProcessId(), &sessionId);
 104         socketName += QLatin1Char('-') + QString::number(sessionId, 16);
 105     }
 106 #else
 107     socketName += QLatin1Char('-') + QString::number(::getuid(), 16);
 108 #endif
 109
 110     server = new QLocalServer(this);
 111     QString lockName = QDir(QDir::tempPath()).absolutePath()
 112                        + QLatin1Char('/') + socketName
 113                        + QLatin1String("-lockfile");
 114     lockFile.setFileName(lockName);
 115     lockFile.open(QIODevice::ReadWrite);
 116 }
 117
 118
 119
 120 bool QtLocalPeer::isClient()
 121 {
 122     if (lockFile.isLocked())
 123         return false;
 124
 125     if (!lockFile.lock(QtLP_Private::QtLockedFile::WriteLock, false))
 126         return true;
 127
 128     bool res = server->listen(socketName);
 129 #if defined(Q_OS_UNIX) && (QT_VERSION >= QT_VERSION_CHECK(4,5,0))
 130     // ### Workaround
 131     if (!res && server->serverError() == QAbstractSocket::AddressInUseError) {
 132         QFile::remove(QDir::cleanPath(QDir::tempPath())+QLatin1Char('/')+socketName);
 133         res = server->listen(socketName);
 134     }
 135 #endif
 136     if (!res)
 137         qWarning("QtSingleCoreApplication: listen on local socket failed, %s", qPrintable(server->errorString()));
 138     QObject::connect(server, SIGNAL(newConnection()), SLOT(receiveConnection()));
 139     return false;
 140 }
 141
 142
 143 bool QtLocalPeer::sendMessage(const QString &message, int timeout)
 144 {
 145     if (!isClient())
 146         return false;
 147
 148     QLocalSocket socket;
 149     bool connOk = false;
 150     for(int i = 0; i < 2; i++) {
 151         // Try twice, in case the other instance is just starting up
 152         socket.connectToServer(socketName);
 153         connOk = socket.waitForConnected(timeout/2);
 154         if (connOk || i)
 155             break;
 156         int ms = 250;
 157 #if defined(Q_OS_WIN)
 158         Sleep(DWORD(ms));
 159 #else
 160         struct timespec ts = { ms / 1000, (ms % 1000) * 1000 * 1000 };
 161         nanosleep(&ts, NULL);
 162 #endif
 163     }
 164     if (!connOk)
 165         return false;
 166
 167     QByteArray uMsg(message.toUtf8());
 168     QDataStream ds(&socket);
 169     ds.writeBytes(uMsg.constData(), uMsg.size());
 170     bool res = socket.waitForBytesWritten(timeout);
 171     res &= socket.waitForReadyRead(timeout);   // wait for ack
 172     res &= (socket.read(qstrlen(ack)) == ack);
 173     return res;
 174 }
 175
 176
 177 void QtLocalPeer::receiveConnection()
 178 {
 179     QLocalSocket* socket = server->nextPendingConnection();
 180     if (!socket)
 181         return;
 182
 183     while (socket->bytesAvailable() < (int)sizeof(quint32))
 184         socket->waitForReadyRead();
 185     QDataStream ds(socket);
 186     QByteArray uMsg;
 187     quint32 remaining;
 188     ds >> remaining;
 189     uMsg.resize(remaining);
 190     int got = 0;
 191     char* uMsgBuf = uMsg.data();
 192     do {
 193         got = ds.readRawData(uMsgBuf, remaining);
 194         remaining -= got;
 195         uMsgBuf += got;
 196     } while (remaining && got >= 0 && socket->waitForReadyRead(2000));
 197     if (got < 0) {
 198         qWarning() << "QtLocalPeer: Message reception failed" << socket->errorString();
 199         delete socket;
 200         return;
 201     }
 202     QString message(QString::fromUtf8(uMsg));
 203     socket->write(ack, qstrlen(ack));
 204     socket->waitForBytesWritten(1000);
 205     delete socket;
 206     emit messageReceived(message); //### (might take a long time to return)
 207 }