mirror of
https://github.com/uroni/urbackup_backend.git
synced 2025-10-26 11:36:50 +00:00
87 lines
2.3 KiB
C++
87 lines
2.3 KiB
C++
/*************************************************************************
|
|
* UrBackup - Client/Server backup system
|
|
* Copyright (C) 2011-2016 Martin Raiber
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as published by
|
|
* the Free Software Foundation, either version 3 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 Affero General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
**************************************************************************/
|
|
|
|
#include "win_disk_mon.h"
|
|
#include <stdio.h>
|
|
#include <map>
|
|
#include "../stringtools.h"
|
|
#include "../urbackupcommon/os_functions.h"
|
|
|
|
namespace
|
|
{
|
|
std::vector<std::map<std::string, std::string> > read_kvs(const std::string& cmd)
|
|
{
|
|
std::string output;
|
|
int rc = os_popen(cmd.c_str(), output);
|
|
|
|
std::vector<std::map<std::string, std::string> > ret;
|
|
if (rc == 0)
|
|
{
|
|
std::map<std::string, std::string> curr;
|
|
int lc = linecount(output);
|
|
for (int i = 0; i < lc; ++i)
|
|
{
|
|
std::string line = trim(getline(i + 1, output));
|
|
|
|
if (line.empty())
|
|
{
|
|
if (!curr.empty())
|
|
{
|
|
ret.push_back(curr);
|
|
curr.clear();
|
|
}
|
|
}
|
|
else
|
|
{
|
|
std::string key = strlower(trim(getuntil("=", line)));
|
|
std::string value = trim(getafter("=", line));
|
|
|
|
curr[key] = value;
|
|
}
|
|
}
|
|
|
|
if (!curr.empty())
|
|
{
|
|
ret.push_back(curr);
|
|
}
|
|
}
|
|
return ret;
|
|
}
|
|
}
|
|
|
|
|
|
std::vector<SFailedDisk> get_failed_disks()
|
|
{
|
|
std::vector<std::map<std::string, std::string> > m = read_kvs("wmic diskdrive get caption,status,statusinfo /value");
|
|
|
|
std::vector<SFailedDisk> ret;
|
|
for (size_t i = 0; i < m.size(); ++i)
|
|
{
|
|
if (strlower(m[i]["status"]) != "ok")
|
|
{
|
|
SFailedDisk fd;
|
|
fd.name = m[i]["caption"];
|
|
fd.status = m[i]["status"];
|
|
fd.status_info = m[i]["statusinfo"];
|
|
ret.push_back(fd);
|
|
}
|
|
}
|
|
|
|
return ret;
|
|
}
|