forked from openkylin/quarkai
209 lines
7.0 KiB
C++
209 lines
7.0 KiB
C++
/*
|
||
* Copyright (C) 2013 ~ 2018 National University of Defense Technology(NUDT) & Tianjin Kylin Ltd.
|
||
*
|
||
* Authors:
|
||
* Kobe Lee xiangli@ubuntukylin.com/kobe24_lixiang@126.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; version 3.
|
||
*
|
||
* 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/>.
|
||
*/
|
||
|
||
#include "filesystemworker.h"
|
||
|
||
#include <stddef.h>
|
||
#include <glibtop/mountlist.h>
|
||
#include <glibtop/fsusage.h>
|
||
/*For PRIu64*/
|
||
#include <inttypes.h>
|
||
|
||
//extern "C" {
|
||
//#include <gio/gio.h>
|
||
//}
|
||
|
||
typedef struct _DISK_INFO
|
||
{
|
||
char devname[256];
|
||
char mountdir[256];
|
||
char type[256];
|
||
gint percentage;
|
||
guint64 btotal;
|
||
guint64 bfree;
|
||
guint64 bavail;
|
||
guint64 bused;
|
||
gint valid;
|
||
} DISK_INFO;
|
||
|
||
static void fsusage_stats(const glibtop_fsusage *buf,
|
||
guint64 *bused, guint64 *bfree, guint64 *bavail, guint64 *btotal,
|
||
gint *percentage)
|
||
{
|
||
guint64 total = buf->blocks * buf->block_size;
|
||
|
||
if (!total) {
|
||
/* not a real device */
|
||
*btotal = *bfree = *bavail = *bused = 0ULL;
|
||
*percentage = 0;
|
||
} else {
|
||
int percent;
|
||
*btotal = total;
|
||
*bfree = buf->bfree * buf->block_size;
|
||
*bavail = buf->bavail * buf->block_size;
|
||
*bused = *btotal - *bfree;
|
||
/* percent = 100.0f * *bused / *btotal; */
|
||
percent = 100 * *bused / (*bused + *bavail);
|
||
*percentage = CLAMP(percent, 0, 100);
|
||
}
|
||
}
|
||
|
||
DISK_INFO add_disk(const glibtop_mountentry *entry, gboolean show_all_fs)
|
||
{
|
||
DISK_INFO disk;
|
||
memset(&disk, 0, sizeof(disk));
|
||
disk.valid = 0;
|
||
glibtop_fsusage usage;
|
||
guint64 bused, bfree, bavail, btotal;
|
||
gint percentage;
|
||
glibtop_get_fsusage(&usage, entry->mountdir);
|
||
if (usage.blocks == 0) {
|
||
return disk;
|
||
}
|
||
if(strcmp(entry->devname,"none")==0 || strcmp(entry->devname,"tmpfs")==0){
|
||
return disk;
|
||
}
|
||
if(strstr(entry->type, "tmpfs")) {
|
||
return disk;
|
||
}
|
||
fsusage_stats(&usage, &bused, &bfree, &bavail, &btotal, &percentage);
|
||
memcpy(disk.devname, entry->devname, strlen(entry->devname));
|
||
memcpy(disk.mountdir, entry->mountdir, strlen(entry->mountdir));
|
||
memcpy(disk.type, entry->type, strlen(entry->type));
|
||
disk.percentage = percentage;
|
||
disk.btotal = btotal;
|
||
disk.bfree = bfree;
|
||
disk.bavail = bavail;
|
||
disk.bused = bused;
|
||
disk.valid = 1;
|
||
// qDebug() << disk.devname;//设备
|
||
// qDebug() << disk.mountdir;//目录
|
||
// qDebug() << disk.type;//类型
|
||
// qDebug() << disk.percentage;
|
||
// qDebug() << g_format_size_full(disk.btotal, G_FORMAT_SIZE_DEFAULT);//总数
|
||
// qDebug() << g_format_size_full(disk.bavail, G_FORMAT_SIZE_DEFAULT);//可用
|
||
// qDebug() << g_format_size_full(disk.bused, G_FORMAT_SIZE_DEFAULT);//已用
|
||
return disk;
|
||
}
|
||
|
||
//void hello(gpointer data)
|
||
//{
|
||
// g_print ("Hello World\n");
|
||
//}
|
||
|
||
FileSystemWorker::FileSystemWorker(QObject *parent)
|
||
: QObject(parent)
|
||
{
|
||
// GVolumeMonitor * monitor;//GVolumeMonitor不是 thread-default-context aware，因此不能在除了主线程中的其他地方使用????
|
||
// monitor = g_volume_monitor_get();
|
||
// g_signal_connect(monitor, "mount-added", G_CALLBACK(hello), NULL);
|
||
|
||
|
||
// GVolumeMonitor* monitor = g_volume_monitor_get();
|
||
// GList* mountDeviceList = g_volume_monitor_get_mounts(monitor);
|
||
// GList* it = NULL;
|
||
// for(it = mountDeviceList; it; it = it->next) {
|
||
// qDebug() << "mount device list:" << it->data;
|
||
// }
|
||
// GList* mountVolumeList = g_volume_monitor_get_volumes(monitor);
|
||
// for(it = mountVolumeList; it; it = it->next) {
|
||
// qDebug() << "mount volume list:" << it->data;
|
||
// }
|
||
}
|
||
|
||
FileSystemWorker::~FileSystemWorker()
|
||
{
|
||
m_diskInfoList.clear();
|
||
}
|
||
|
||
void FileSystemWorker::onFileSystemListChanged()
|
||
{
|
||
QStringList newDiskList;
|
||
glibtop_mountentry *entries;
|
||
glibtop_mountlist mountlist;
|
||
guint i;
|
||
gboolean show_all_fs = TRUE;
|
||
entries = glibtop_get_mountlist(&mountlist, show_all_fs);
|
||
for (i = 0; i < mountlist.number; i++) {
|
||
DISK_INFO disk = add_disk(&entries[i], show_all_fs);
|
||
if (disk.valid == 1) {
|
||
QString dev_name = QString(QLatin1String(disk.devname));
|
||
newDiskList.append(dev_name);
|
||
|
||
if (!this->isDeviceContains(dev_name)) {
|
||
FileSystemData *info = new FileSystemData(this);
|
||
info->setDevName(dev_name);
|
||
info->updateDiskInfo(QString(QLatin1String(disk.mountdir)), QString(QLatin1String(disk.type)), QString(QLatin1String(g_format_size_full(disk.btotal, G_FORMAT_SIZE_DEFAULT))), QString(QLatin1String(g_format_size_full(disk.bfree, G_FORMAT_SIZE_DEFAULT))), QString(QLatin1String(g_format_size_full(disk.bavail, G_FORMAT_SIZE_DEFAULT))), QString(QLatin1String(g_format_size_full(disk.bused, G_FORMAT_SIZE_DEFAULT))), disk.percentage/*QString::number(disk.percentage).append("%")*/);
|
||
this->addDiskInfo(dev_name, info);
|
||
}
|
||
else {//update info which had exists
|
||
FileSystemData *info = this->getDiskInfo(dev_name);
|
||
if (info) {
|
||
info->updateDiskInfo(QString(QLatin1String(disk.mountdir)), QString(QLatin1String(disk.type)), QString(QLatin1String(g_format_size_full(disk.btotal, G_FORMAT_SIZE_DEFAULT))), QString(QLatin1String(g_format_size_full(disk.bfree, G_FORMAT_SIZE_DEFAULT))), QString(QLatin1String(g_format_size_full(disk.bavail, G_FORMAT_SIZE_DEFAULT))), QString(QLatin1String(g_format_size_full(disk.bused, G_FORMAT_SIZE_DEFAULT))), disk.percentage/*QString::number(disk.percentage).append("%")*/);
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
//remove the device whice not exists anymore
|
||
for (auto device : m_diskInfoList.keys()) {
|
||
bool foundDevice = false;
|
||
for (auto devName : newDiskList) {
|
||
if (devName == device) {
|
||
foundDevice = true;
|
||
break;
|
||
}
|
||
}
|
||
|
||
if (!foundDevice) {
|
||
m_diskInfoList.remove(device);//or erase???
|
||
}
|
||
}
|
||
|
||
g_free(entries);
|
||
}
|
||
|
||
FileSystemData *FileSystemWorker::getDiskInfo(const QString &devname)
|
||
{
|
||
return m_diskInfoList.value(devname, nullptr);
|
||
}
|
||
|
||
QList<FileSystemData *> FileSystemWorker::diskInfoList() const
|
||
{
|
||
return m_diskInfoList.values();
|
||
}
|
||
|
||
void FileSystemWorker::addDiskInfo(const QString &devname, FileSystemData *info)
|
||
{
|
||
if (!m_diskInfoList.contains(devname)) {
|
||
m_diskInfoList[devname] = info;
|
||
}
|
||
}
|
||
|
||
void FileSystemWorker::removeDiskItem(const QString &devname)
|
||
{
|
||
// FileSystemData *info = getDiskInfo(devname);
|
||
// m_diskInfoList.remove(devname);
|
||
}
|
||
|
||
bool FileSystemWorker::isDeviceContains(const QString &devname)
|
||
{
|
||
return m_diskInfoList.keys().contains(devname);
|
||
}
|