Differential D21970 Diff 60295 src/util/externalcommand_polkitbackend.cpp

Changeset View

Standalone View

src/util/externalcommand_polkitbackend.cpp

This file was added.

1		/*************************************************************************
2		* Copyright (C) 2019 by Shubham <shubham.jangra@kdemail.net> *
3		* *
4		* This program is free software; you can redistribute it and/or *
5		* modify it under the terms of the GNU General Public License as *
6		* published by the Free Software Foundation; either version 3 of *
7		* the License, or (at your option) any later version. *
8		* *
9		* This program is distributed in the hope that it will be useful, *
10		* but WITHOUT ANY WARRANTY; without even the implied warranty of *
11		* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
12		* GNU General Public License for more details. *
13		* *
14		* You should have received a copy of the GNU General Public License *
15		* along with this program. If not, see <http://www.gnu.org/licenses/>.*
16		*************************************************************************/
17
18		#include "util/externalcommand_polkitbackend.h"
19
20		#include <QApplication>
21		#include <QByteArray>
22		#include <QCoreApplication>
23		#include <QDebug>
24		#include <QDBusConnection>
25		#include <QDBusConnectionInterface>
26		#include <QEventLoop>
27		#include <QString>
28		#include <QTimer>
29		#include <QWidget>
30
31		#include <PolkitQt1/Authority>
32		#include <PolkitQt1/ActionDescription>
33		#include <PolkitQt1/Subject>
34
35		using namespace PolkitQt1;
36
37		namespace Auth
38		{
39
40		PolkitEventLoop::PolkitEventLoop(QObject *parent)
41		: QEventLoop(parent),
42		m_result(result())
43		{
44
45		}
46
47		PolkitEventLoop::~PolkitEventLoop()
48		{
49
50		}
51
52		void PolkitEventLoop::requestQuit(const Authority::Result &result)
53		{
54		m_result = result;
55		quit();
56		}
57
58		Authority::Result PolkitEventLoop::result() const
59		{
60		return m_result;
61		}
62
63		PolkitQt1Backend::PolkitQt1Backend()
64		: m_flyingActions(false)
65		{
66		// Connect various useful Polkit signals
67		connect(Authority::instance(), &Authority::configChanged, this, &Auth::PolkitQt1Backend::checkForResultChanged);
68		connect(Authority::instance(), &Authority::consoleKitDBChanged, this, &Auth::PolkitQt1Backend::checkForResultChanged);
69
70		//connect(Authority::instance(), /SIGNAL(consoleKitDBChanged())/&Gui::dataChanged, this, /SLOT(checkForResultChanged())/ &Auth::PolkitQt1Backend::checkForResultChanged);
71		//connect(Authority::instance(), /SIGNAL(enumerateActionsFinished(PolkitQt1::ActionDescription::List))/,this, /SLOT(updateCachedActions(PolkitQt1::ActionDescription::List))/);
72
73		// Take the current action into account.
74		m_flyingActions = true;
75
76
77		}
78
79		PolkitQt1Backend::~PolkitQt1Backend()
80		{
81
82		}
83
84		void PolkitQt1Backend::initPolkitAgent(const QString &action, QWidget parent /= nullptr*/)
85		{
86		if (!parent) {
87		qWarning() << "Parent widget does not exists, can not proceed further";
88		return;
89		}
90
91		// Check if KDE authentication agent exists
92		if (QDBusConnection::sessionBus().interface()->isServiceRegistered(QLatin1String("org.kde.polkit-kde-authentication-agent-1"))) {
	stikonasUnsubmitted Done I'll have to read a bit more about polkit, but this requirement sounds strange. It shouldn't be necessary to have kde-authentication-agent-1. Some users might be on other desktop environments and use their own polkit authentication agent, e.g. `polkit-gnome-authentication-agent-1`. stikonas: I'll have to read a bit more about polkit, but this requirement sounds strange. It shouldn't…
93		// Check if we are running terminal session or GUI session
94		if (!qApp \|\| !qobject_cast<QApplication *>(qApp)) {
95		qWarning() << "We are running a TTY session";
96		}
97
98		// Retrieve the dialog root window Id
99		qulonglong wId = parent->effectiveWinId();
100
101		// Send it over the bus to our agent
102		QDBusMessage methodCall =
103		QDBusMessage::createMethodCall(QLatin1String("org.kde.polkit-kde-authentication-agent-1"), QLatin1String("/org/kde/Polkit1AuthAgent"), QLatin1String("org.kde.Polkit1AuthAgent"),
104		QLatin1String("setWIdForAction"));
105
106		methodCall << action;
107		methodCall << wId;
108
109		QDBusPendingCall call = QDBusConnection::sessionBus().asyncCall(methodCall);
110		call.waitForFinished();
111
112		if (call.isError()) {
113		qWarning() << "ERROR while streaming the parent!!" << call.error();
114		}
115		} else {
116		qInfo() << "KDE polkit agent appears too old or not registered on the bus";
117		}
118		}
119
120		Authority::Result PolkitQt1Backend::actionStatus(const QString &action, QByteArray &callerID)
121		{
122		// Get the process ID of the caller
123		SystemBusNameSubject subject(QString::fromUtf8(callerID));
124
125		auto authority = PolkitQt1::Authority::instance();
126
127		auto result = authority->checkAuthorizationSync(action, subject, Authority::None);
128
129		if (authority->hasError()) {
130		qDebug() << "Encountered error while checking action status, error code:" << authority->lastError() << "\nError Details:" << authority->errorDetails();
131
132		// Clear all the errors from the buffer so that hasError() does not give previous error as a result when called later
133		authority->clearError();
134		}
135
136		return result;
137		}
138
139		QByteArray PolkitQt1Backend::callerID() const
140		{
141		// Get the current Application process ID
142		return QDBusConnection::systemBus().baseService().toUtf8();
143		}
144
145		bool PolkitQt1Backend::authorizeAction(const QString &action, QByteArray &callerID)
146		{
147		SystemBusNameSubject subject(QString::fromUtf8(callerID));
148		Authority *authority = Authority::instance();
149
150		PolkitEventLoop event;
151
152		connect(authority, &Authority::checkAuthorizationFinished, &event, &PolkitEventLoop::requestQuit);
153
154		authority->checkAuthorizationSync(action, subject, Authority::None);
155
156		event.exec();
157
158		if (authority->hasError()) {
159		qWarning() << "Encountered error while checking authorization, Error code:" << authority->lastError() << "\n" <<"Error details:" << authority->errorDetails();
160
161		// Clear all the errors from the buffer so that hasError() does not give previous error as a result when called later
162		authority->clearError();
163		}
164
165		if (event.result() == Authority::Yes) {
166		// Emit signal signalling that current action is authorized by the authority instance
167		emit actionAuthorized(true);
168		return true;
169		} else {
170		emit actionAuthorized(false);
171		return false;
172		}
173		}
174
175		// Dummy function for calling QTimer
176		void PolkitQt1Backend::quit() const
177		{
178
179		}
180
181		void PolkitQt1Backend::sendProgressStep(const int &percent)
182		{
183		Q_UNUSED(percent)
184		}
185
186		void PolkitQt1Backend::sendProgressStep(const QVariantMap &step)
187		{
188		Q_UNUSED(step)
189		}
190
191		void PolkitQt1Backend::checkForResultChanged()
192		{
193		for (auto it = m_cachedResults.begin(); it != m_cachedResults.end(); ++it) {
194		const QString action = it.key();
195		QByteArray pid = QDBusConnection::systemBus().baseService().toUtf8();
196		if (it.value() != actionStatus(action, pid)) {
197		*it = actionStatus(action, pid);
198		}
199		}
200
201		// Force updating known actions
202		Authority::instance()->enumerateActions();
203		m_flyingActions = true;
204		}
205
206		void PolkitQt1Backend::updateCachedActions(const ActionDescription::List &actions)
207		{
208		Q_UNUSED(actions)
209		}
210
211		} // namespace Auth
212
213
214
215