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
|
// This is an example not a library
/***************************************************************************
* Copyright (C) 2008 Daniel Nicoletti <dantti85-pk@yahoo.com.br> *
* Copyright (C) 2009 Radek Novacek <rnovacek@redhat.com> *
* *
* This program 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 2 of the License, or *
* (at your option) any later version. *
* *
* This program 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 this program; if not, write to the *
* Free Software Foundation, Inc., *
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA . *
***************************************************************************/
#include "PkExampleHelper.h"
#include "examplesadaptor.h"
#include "polkitqt1-authority.h"
#include <QtDBus/QDBusConnection>
#include <QtCore/QTimer>
#include <QtCore/QDebug>
#include <QtXml/QDomDocument>
#define MINUTE 60000
using namespace PolkitQt1;
PkExampleHelper::PkExampleHelper(int &argc, char **argv)
: QCoreApplication(argc, argv)
{
qDebug() << "Creating Helper";
(void) new ExamplesAdaptor(this);
// Register the DBus service
if (!QDBusConnection::systemBus().registerService("org.qt.policykit.examples")) {
qDebug() << QDBusConnection::systemBus().lastError().message();;
QTimer::singleShot(0, this, SLOT(quit()));
return;
}
if (!QDBusConnection::systemBus().registerObject("/", this)) {
qDebug() << "unable to register service interface to dbus";
QTimer::singleShot(0, this, SLOT(quit()));
return;
}
// Normally you will set a timeout so your application can
// free some resources of the poor client machine ;)
QTimer::singleShot(MINUTE, this, SLOT(quit()));
}
PkExampleHelper::~PkExampleHelper()
{
qDebug() << "Destroying Helper";
}
bool PkExampleHelper::set(const QString &action)
{
qDebug() << "PkExampleHelper::set" << action;
// message().service() is the service name of the caller
// We can check if the caller is authorized to the following action
Authority::Result result;
SystemBusNameSubject subject(message().service());
result = Authority::instance()->checkAuthorizationSync("org.qt.policykit.examples.set",
subject , Authority::AllowUserInteraction);
if (result == Authority::Yes) {
qDebug() << message().service() << QString("Implicit authorization set to") << action;
// Caller is authorized so we can perform the action
return setValue(action);
} else {
qDebug() << message().service() << QString("Can't set the implicit authorization");
// Caller is not authorized so the action can't be performed
return false;
}
}
bool PkExampleHelper::setValue(const QString &action)
{
// This action must be authorized first. It will set the implicit
// authorization for the Shout action by editing the .policy file
QDomDocument doc = QDomDocument("policy");
QFile file("/usr/share/polkit-1/actions/org.qt.policykit.examples.policy");
if (!file.open(QIODevice::ReadOnly))
return false;
doc.setContent(&file);
file.close();
QDomElement el = doc.firstChildElement("policyconfig").
firstChildElement("action");
while (!el.isNull() && el.attribute("id", QString()) != "org.qt.policykit.examples.shout") {
el = el.nextSiblingElement("action");
}
el = el.firstChildElement("defaults");
el = el.firstChildElement("allow_active");
if (el.isNull())
return false;
el.firstChild().toText().setData(action);
if (!file.open(QIODevice::WriteOnly))
return false;
QTextStream stream(&file);
doc.save(stream, 2);
file.close();
return true;
}
|
