Differential D23288 Diff 64157 processplugins/xres/xres.cpp

Changeset View

Standalone View

processplugins/xres/xres.cpp

This file was added.

1		/* This file is part of the KDE project
2
	apolUnsubmitted Done What's the copyright? apol: What's the copyright?
3		Copyright (C) 2019 David Edmundson <davidedmundson@kde.org>
4
5		This library is free software; you can redistribute it and/or
6		modify it under the terms of the GNU Library General Public
7		License as published by the Free Software Foundation; either
	apolUnsubmitted Done Scope shouldn't be necessary apol: Scope shouldn't be necessary
8		version 2 of the License, or (at your option) any later version.
9
10		This library is distributed in the hope that it will be useful,
11		but WITHOUT ANY WARRANTY; without even the implied warranty of
12		MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13		Library General Public License for more details.
14
15		You should have received a copy of the GNU Library General Public License
16		along with this library; see the file COPYING.LIB. If not, write to
17		the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
18		Boston, MA 02110-1301, USA.
19		*/
20
21		#include "xres.h"
22
23		#include <QDebug>
24		#include <QHash>
25		#include <QGuiApplication>
26		#include <QX11Info>
27
28		#include <KLocalizedString>
29		#include <KPluginFactory>
30
31		#include <processcore/process.h>
32
33		#include <X11/extensions/XRes.h>
34
35		using namespace KSysGuard;
36
37		XResPlugin::XResPlugin(QObject *parent, const QVariantList &args)
38		: ProcessDataProvider(parent, args)
39		{
40		if (!QGuiApplication::instance()) {
41		return;
42		}
43		if (!QX11Info::QX11Info::isPlatformX11()) {
44		return;
45		}
46		mIsX11 = true;
47		m_xresAttribute = new ProcessAttribute(QStringLiteral("xres"), i18n("X11 Memory"), this);
48		m_xresAttribute->setUnit(KSysGuard::UnitByte);
49		addProcessAttribute(m_xresAttribute);
50		}
51
52		bool XResPlugin::updateXResClientData() {
53		if (!mIsX11) {
54		return false;
55		}
56		XResClient *clients;
57		int count;
58
59		XResQueryClients(QX11Info::display(), &count, &clients);
60
61		mXResClientResources.clear();
62		for (int i=0; i < count; i++)
63		mXResClientResources.insert(-(qlonglong)(clients[i].resource_base), clients[i].resource_mask);
64
65		if(clients)
66		XFree(clients);
67		return true;
68		}
69
70		void XResPlugin::update() {
71		if (!mIsX11) {
72		return;
73		}
74		updateXResClientData();
75		Window *children, dummy;
76		unsigned int count;
77		Status result = XQueryTree(QX11Info::display(), QX11Info::appRootWindow(), &dummy, &dummy, &children, &count);
78		if(!result)
79		return;
80		if(!updateXResClientData())
81		return;
82		for (uint i=0; i < count; ++i) {
83		WId wid = children[i];
84		QMap<qlonglong, XID>::iterator iter = mXResClientResources.lowerBound(-(qlonglong)(wid));
85		if(iter == mXResClientResources.end())
86		continue; //We couldn't find it this time :-/
87
88		if(-iter.key() != (qlonglong)(wid & ~iter.value()))
89		continue; //Already added this window
90
91		//Get the PID for this window if we do not know it
92		NETWinInfo info(QX11Info::connection(), wid, QX11Info::appRootWindow(), NET::WMPid, NET::Properties2());
93
94		qlonglong pid = info.pid();
95		if(!pid)
96		continue;
97		//We found a window with this client
98		mXResClientResources.erase(iter);
99
100		KSysGuard::Process *process = getProcess(pid);
101		if(!process) {
102		continue; //can in race condition etc
103		}
104
105		//Now update the pixmap bytes for this window
106		unsigned long memUsage;
107		bool success = XResQueryClientPixmapBytes(QX11Info::display(), wid, &memUsage);
108
109		if(!success) {
110		m_xresAttribute->clearData(process);
111		} else {
112		qulonglong memUsageCast = memUsage;
113		m_xresAttribute->setData(process, memUsageCast);
114		}
115
116		}
117		if(children)
118		XFree((char*)children);
119		}
120
121		K_PLUGIN_FACTORY_WITH_JSON(PluginFactory, "xres.json", registerPlugin<XResPlugin>();)
122
123		#include "xres.moc"