workspace/kcontrol/kfontinst/kcmfontinst/PolicyKitAuthenticator.cpp

   1 /*
   2  * KFontInst - KDE Font Installer
   3  *
   4  * Copyright 2003-2008 Craig Drummond <craig@kde.org>
   5  *
   6  * ----
   7  *
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License as published by
  10  * the Free Software Foundation; either version 2 of the License, or
  11  * (at your option) any later version.
  12  *
  13  * This program is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  16  * General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * along with this program; see the file COPYING.  If not, write to
  20  * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
  21  * Boston, MA 02110-1301, USA.
  22  */
  23
  24 #include "PolicyKitAuthenticator.h"
  25 #include <QtGui/QWidget>
  26 #include <QtGui/QApplication>
  27 #include <QtCore/QEventLoop>
  28 #include <QtDBus/QDBusMessage>
  29 #include <QtDBus/QDBusInterface>
  30 #include <KDE/KGlobal>
  31 #include <sys/types.h>
  32 #include <unistd.h>
  33 #include <set>
  34
  35 K_GLOBAL_STATIC(PolicyKitAuthenticator, theInstance)
  36
  37 struct PolicyKitAuthenticatorKey
  38 {
  39     PolicyKitAuthenticatorKey(const QString &a=QString(), uint p=0)
  40         : action(a), pid(p) { }
  41
  42     bool operator<(const PolicyKitAuthenticatorKey &o) const
  43     {
  44         return pid<o.pid || (pid==o.pid && action<o.action);
  45     }
  46
  47     QString action;
  48     uint    pid;
  49 };
  50
  51 class PolicyKitAuthenticatorPrivate
  52 {
  53     public:
  54
  55     PolicyKitAuthenticatorPrivate()
  56         : dbus("org.freedesktop.PolicyKit.AuthenticationAgent",
  57                "/",
  58                "org.freedesktop.PolicyKit.AuthenticationAgent",
  59                QDBusConnection::sessionBus())
  60     {
  61     }
  62
  63     virtual ~PolicyKitAuthenticatorPrivate() { }
  64
  65     std::set<PolicyKitAuthenticatorKey> sucesses;
  66     QDBusInterface                      dbus;
  67     bool                                replyStatus;
  68 };
  69
  70 PolicyKitAuthenticator::PolicyKitAuthenticator()
  71                       : d_ptr(new PolicyKitAuthenticatorPrivate)
  72 {
  73 }
  74
  75 PolicyKitAuthenticator::~PolicyKitAuthenticator()
  76 {
  77     delete d_ptr;
  78 }
  79
  80 PolicyKitAuthenticator * PolicyKitAuthenticator::instance()
  81 {
  82     return theInstance;
  83 }
  84
  85 bool PolicyKitAuthenticator::authenticate(const QString &action, QWidget *widet, bool gui)
  86 {
  87     return authenticate(action, widet ? widet->window()->winId() : 0, (uint)getpid(), gui);
  88 }
  89
  90 bool PolicyKitAuthenticator::authenticate(const QString &action, uint winId, uint pid, bool gui)
  91 {
  92     static const int constDBusTimeout=20*60*1000; // 20mins
  93
  94     Q_D(PolicyKitAuthenticator);
  95     PolicyKitAuthenticatorKey key(action, pid);
  96
  97     if(d->sucesses.end()!=d->sucesses.find(key))
  98         return true;
  99
 100     QList<QVariant> args;
 101
 102     args << action << winId << pid;
 103
 104     QDBusMessage message=QDBusMessage::createMethodCall(d->dbus.service(), d->dbus.path(), d->dbus.interface(), "ObtainAuthorization");
 105     message.setArguments(args);
 106
 107     if(gui)
 108     {
 109         QEventLoop eventLoop;
 110
 111         d->dbus.connection().callWithCallback(message, this, SLOT(reply(QDBusMessage)), SLOT(error(QDBusError, QDBusMessage)), constDBusTimeout);
 112         connect(this, SIGNAL(quitLoop()), &eventLoop, SLOT(quit()));
 113         d->replyStatus=false;
 114         eventLoop.exec(QEventLoop::ExcludeUserInputEvents);
 115
 116         if(d->replyStatus)
 117         {
 118             d->sucesses.insert(key);
 119             return true;
 120         }
 121     }
 122     else if(isAuthenticated(d->dbus.connection().call(message, QDBus::Block, constDBusTimeout)))
 123     {
 124         d->sucesses.insert(key);
 125         return true;
 126     }
 127
 128     return false;
 129 }
 130
 131 void PolicyKitAuthenticator::reply(const QDBusMessage &msg)
 132 {
 133     if(isAuthenticated(msg))
 134     {
 135         Q_D(PolicyKitAuthenticator);
 136         d->replyStatus=true;
 137     }
 138     emit quitLoop();
 139 }
 140
 141 void PolicyKitAuthenticator::error(const QDBusError &, const QDBusMessage &)
 142 {
 143     emit quitLoop();
 144 }
 145
 146 bool PolicyKitAuthenticator::isAuthenticated(const QDBusMessage &msg)
 147 {
 148     QList<QVariant> args=msg.arguments();
 149
 150     return QDBusMessage::ReplyMessage==msg.type() && 1==args.count() &&
 151            QVariant::Bool==args[0].type() && args[0].toBool();
 152 }
 153
 154 #include "PolicyKitAuthenticator.moc"