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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
|
/* -*- Mode: C++; indent-tabs-mode: nil; tab-width: 4 -*-
* -*- coding: utf-8 -*-
*
* Copyright (C) 2023 KylinSoft Co., Ltd.
*
* 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 3 of the License, or
* 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, see <http://www.gnu.org/licenses/>.
*/
#ifndef PROCADDSERVER_H
#define PROCADDSERVER_H
#include <QObject>
#include <QThread>
#include <QQueue>
#include <QDir>
#include <QMap>
#include <QDBusConnection>
#include <QDBusInterface>
#include <QDBusReply>
#include <QtDBus/QDBusMetaType>
#include <QThreadPool>
#include <QRunnable>
#include <QMetaObject>
#include <QTime>
extern "C"
{
#include <sys/socket.h>
#include <linux/netlink.h>
#include <linux/connector.h>
#include <linux/cn_proc.h>
#include <signal.h>
#include <errno.h>
#include <stdbool.h>
#include <unistd.h>
}
#define THREAD_MAXNUM 5
#define READLINE_MAXSIZE 256
#define PROCINFOKEY_TYPE "type"
#define PROCINFOKEY_PID "pid"
#define PROCINFOKEY_CMDLINE "cmdline"
#define PROCINFOKEY_UID "uid"
#define PROCINFOKEY_NAME "name"
#define PROCINFOKEY_STATE "state"
#define PROCINFOKEY_TGID "tgid"
#define PROCINFOKEY_DESKTOP "desktop"
#define FILEKEY_DESKTOP "GIO_LAUNCHED_DESKTOP_FILE="
#define FILEKEY_NAME "Name:"
#define FILEKEY_STATE "State:"
#define FILEKEY_TGID "Tgid:"
#define FILEKEY_UID "Uid:"
typedef QMap<QString, QString> MyMap;
struct ky_cn_msg
{
struct cb_id id;
__u32 seq;
__u32 ack;
__u32 len;
__u8 data[0];
};
class ListenThObject : public QObject
{
Q_OBJECT
public:
explicit ListenThObject(QObject *parent = nullptr);
inline void setExitStat(bool stat)
{
m_exitStat = stat;
}
inline void setSockFd(int sockFd)
{
m_sockFd = sockFd;
}
private:
bool m_exitStat = false;
int m_sockFd;
public Q_SLOTS:
int handleProcEv();
Q_SIGNALS:
void procAdd(int pid);
};
class DealData : public QRunnable
{
public:
explicit DealData(int pid, QObject *obj);
void run() override;
private:
QString getFileMsg(QString filePath);
bool isSystemCmd(QString cmdline);
QString getEnvironMsg(QString environ);
void getStatusMsg(QString fileMsg, QMap<QString, QString> *infoMap);
private:
int m_pid;
QObject *m_obj;
};
class ProcAddServer : public QObject
{
Q_OBJECT
Q_CLASSINFO("D-Bus Interface","com.settings.daemon.interface")
public:
explicit ProcAddServer(QObject *parent = nullptr);
private:
bool netlinkConnect();
bool setProcEvListen(int nl_sock, bool enable);
private Q_SLOTS:
void emitSignal(MyMap procmap);
public Q_SLOTS:
void startListen(int pid);
void stopListen(int pid);
private:
int m_sockFd = -1;
QThread *m_listenTh = nullptr;
ListenThObject *m_listenThObj = nullptr;
QThreadPool *m_queDutyPool = nullptr;
QList<int> m_usedModelList;
Q_SIGNALS:
void procAdd(QMap<QString, QString> procinfo);
};
#endif // PROCADDSERVER_H
|
