|
|
#include "system_quotas.h"
|
|
|
#include "my_type.h"
|
|
|
|
|
|
namespace Quotas_sys {
|
|
|
|
|
|
Quotas_sys::Quotas_sys() {
|
|
|
map_hw_or_sw["ext2"] = "hw";
|
|
|
map_hw_or_sw["ext3"] = "hw";
|
|
|
map_hw_or_sw["nfs"] = "sw";
|
|
|
map_hw_or_sw["NFS"] = "sw";
|
|
|
}
|
|
|
|
|
|
void Quotas_sys::set_map_device(map_str_str &map_device) {
|
|
|
this->map_device = &map_device;
|
|
|
this->check_partition_mount();
|
|
|
}
|
|
|
|
|
|
void Quotas_sys::set_map_gui(map_str_str &map_gui) {
|
|
|
map_gui_cfg = &map_gui;
|
|
|
}
|
|
|
|
|
|
void Quotas_sys::check_partition_mount() {
|
|
|
string key = "";
|
|
|
string response = Utils::file_read("/proc/mounts");
|
|
|
for (const auto& iter: (*map_device)) {
|
|
|
key = iter.first;
|
|
|
if (response.find(key) != string::npos) {
|
|
|
disk_mount[key] = true;
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
|
|
|
string Quotas_sys::wrapper_hw_sw(string partition) {
|
|
|
|
|
|
if ((*map_device).find(partition) != (*map_device).end()) {
|
|
|
string fsystem = (*map_device)[partition];
|
|
|
if ((*map_device).find(fsystem) != (*map_device).end()) {
|
|
|
return map_hw_or_sw[fsystem];
|
|
|
}
|
|
|
}
|
|
|
return "";
|
|
|
|
|
|
}
|
|
|
|
|
|
bool Quotas_sys::check_on_quotas_system(string cmd) {
|
|
|
string response = obj_process_call.call_all_sections(cmd);
|
|
|
if (response.length() > 0) {
|
|
|
return true;
|
|
|
}
|
|
|
else {
|
|
|
return false;
|
|
|
}
|
|
|
}
|
|
|
|
|
|
struct status_quotas Quotas_sys::check_on_quotas_system_2_hw(string params) {
|
|
|
string cmd = "quotaon -ugPvp " + params;
|
|
|
string response = obj_process_call.call_all_sections(cmd);
|
|
|
struct status_quotas status;
|
|
|
if (response.length() > 0 && response.find("Cannot find mountpoint for device ") == string::npos) {
|
|
|
status.status = true;
|
|
|
vector<string> vec_str = Utils::split(response, '\n');
|
|
|
string array_str[3] = {"group", "user", "project"};
|
|
|
for (int index = 0; index < 3; index++) {
|
|
|
string u_g_p = array_str[index];
|
|
|
for (const string& vec_u_g_p : vec_str) {
|
|
|
if (vec_u_g_p.find(u_g_p) != string::npos && vec_u_g_p.find("accounting") != string::npos ) {
|
|
|
if (index == 0) {
|
|
|
status.group = true;
|
|
|
}
|
|
|
else if (index == 1) {
|
|
|
status.user = true;
|
|
|
}
|
|
|
else if (index == 2) {
|
|
|
status.project = true;
|
|
|
}
|
|
|
}
|
|
|
else if (vec_u_g_p.find(u_g_p) != string::npos && vec_u_g_p.find("enforced") != string::npos){
|
|
|
if (index == 0) {
|
|
|
status.group = false;
|
|
|
}
|
|
|
else if (index == 1) {
|
|
|
status.user = false;
|
|
|
}
|
|
|
else if (index == 2) {
|
|
|
status.project = false;
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
else {
|
|
|
status.group = false;
|
|
|
status.user = false;
|
|
|
status.project = false;
|
|
|
status.status = false;
|
|
|
}
|
|
|
return status;
|
|
|
|
|
|
}
|
|
|
|
|
|
bool Quotas_sys::wrapper_on_quotas(string partition, string flag_mode) {
|
|
|
// Todo
|
|
|
// Добавить выключение квот в зависимости от диска
|
|
|
cout << partition << "" << flag_mode << endl;
|
|
|
if (disk_mount.find(partition) == disk_mount.end()) {
|
|
|
|
|
|
}
|
|
|
else {
|
|
|
|
|
|
}
|
|
|
return true;
|
|
|
}
|
|
|
|
|
|
bool Quotas_sys::on_quota_hw(string partition, string cmd_hw, string cmd_sw) {
|
|
|
/*
|
|
|
Включить квоты, только для HW ext2,3,4:
|
|
|
Если ФС, уже существует и без HW атрибут квот ext2,3,4
|
|
|
Раздел должен быть отмонтирован
|
|
|
*/
|
|
|
if (this->wrapper_hw_sw(partition) == "hw") {
|
|
|
obj_process_system.call(cmd_hw, "");
|
|
|
obj_process_system.call("ubconfig set config SERVICESSTART+=,quotaon.service", "");
|
|
|
return true;
|
|
|
}
|
|
|
else if (this->wrapper_hw_sw(partition) == "sw") {
|
|
|
obj_process_system.call(cmd_sw, "");
|
|
|
obj_process_system.call("ubconfig set config SERVICESSTART+=,quotaon.service", "");
|
|
|
return true;
|
|
|
}
|
|
|
else{
|
|
|
return false;
|
|
|
}
|
|
|
return false;
|
|
|
}
|
|
|
|
|
|
bool Quotas_sys::on_quota_sw(string partition, int mode) {
|
|
|
/*
|
|
|
Включить квоты, только для SW, другие ФС и сетевые NFS:
|
|
|
У этого исполнения квот отсутствуют проектные квоты prjquota
|
|
|
Раздел должен быть примонтирован
|
|
|
*/
|
|
|
if (map_hw_or_sw.find(partition) != map_hw_or_sw.end()) {
|
|
|
string cmd = "";
|
|
|
if (mode == 0) {
|
|
|
cmd += "mount -vo remount,usrquota " + partition;
|
|
|
cmd += "; quotacheck -ucm " + partition;
|
|
|
cmd += "; quotaon -uv " + partition;
|
|
|
}
|
|
|
else if (mode == 1) {
|
|
|
cmd += "; mount -vo remount,grpquota " + partition;
|
|
|
cmd += "; quotacheck -gcm " + partition;
|
|
|
cmd += "; quotaon -gv " + partition;
|
|
|
}
|
|
|
else if (mode == 2) {
|
|
|
cmd += "; quotaon -ugPv " + partition;
|
|
|
}
|
|
|
else if (mode == 3) {
|
|
|
cmd += "; quotaon -augPv";
|
|
|
}
|
|
|
else {
|
|
|
return false;
|
|
|
}
|
|
|
cmd += "; ubconfig set config SERVICESSTART+=,quotaon.service";
|
|
|
obj_process_system.call(cmd, "");
|
|
|
return true;
|
|
|
}
|
|
|
return false;
|
|
|
}
|
|
|
|
|
|
bool Quotas_sys::off_quota(string partition, string quota_hw, string quota_sw) {
|
|
|
string cmd = "";
|
|
|
if (map_hw_or_sw.find(partition) != map_hw_or_sw.end()) {
|
|
|
if (this->wrapper_hw_sw(partition) == "hw") {
|
|
|
cmd = quota_hw + " " + partition;
|
|
|
obj_process_system.call(cmd, "");
|
|
|
return true;
|
|
|
}
|
|
|
else if (this->wrapper_hw_sw(partition) == "sw") {
|
|
|
cmd = quota_sw + " " + partition;
|
|
|
obj_process_system.call(cmd, "");
|
|
|
return true;
|
|
|
}
|
|
|
}
|
|
|
return false;
|
|
|
|
|
|
}
|
|
|
|
|
|
vector<data_all_quotas> Quotas_sys::pars_data(string cmd) {
|
|
|
string start_pars = "------\n";
|
|
|
string response = obj_process_call.call_all_sections(cmd);
|
|
|
size_t index_start = response.find(start_pars);
|
|
|
string line_pars = "";
|
|
|
struct data_all_quotas struct_data;
|
|
|
vector<data_all_quotas> vec_struct_data;
|
|
|
int index = 0;
|
|
|
int old_index = index;
|
|
|
if (index_start != string::npos) {
|
|
|
response = response.substr(index_start+start_pars.length(), response.length());
|
|
|
vector<int> vec_index_sep = Utils::find_all(response, "\n");
|
|
|
for (size_t i = 0; i < vec_index_sep.size(); i++) {
|
|
|
old_index = index;
|
|
|
index = vec_index_sep[i];
|
|
|
if (index-1 == old_index) {
|
|
|
break;
|
|
|
}
|
|
|
line_pars = response.substr(old_index, index);
|
|
|
struct_data = part_quotas_line(line_pars);
|
|
|
vec_struct_data.push_back(struct_data);
|
|
|
}
|
|
|
}
|
|
|
return vec_struct_data;
|
|
|
}
|
|
|
|
|
|
struct data_all_quotas Quotas_sys::part_quotas_line(string line) {
|
|
|
struct data_all_quotas struct_data;
|
|
|
vector<int> vec_index_sep = Utils::find_all(line, " ");
|
|
|
int old_index = 0;
|
|
|
string buff = "";
|
|
|
int index_data = 0;
|
|
|
for (int& index: vec_index_sep) {
|
|
|
if ((index - old_index) >= 2) {
|
|
|
buff = line.substr(old_index, index - old_index);
|
|
|
cout << buff << endl;
|
|
|
if (index_data == 0) {
|
|
|
struct_data.username = buff;
|
|
|
}
|
|
|
else if (index_data == 1) {
|
|
|
struct_data.status = buff;
|
|
|
}
|
|
|
else if (index_data == 2) {
|
|
|
struct_data.block_used = buff;
|
|
|
}
|
|
|
else if (index_data == 3) {
|
|
|
struct_data.block_soft = buff;
|
|
|
}
|
|
|
else if (index_data == 4) {
|
|
|
struct_data.block_hard = buff;
|
|
|
}
|
|
|
else if (index_data == 5) {
|
|
|
struct_data.block_grace = buff;
|
|
|
}
|
|
|
else if (index_data == 6) {
|
|
|
struct_data.file_used = buff;
|
|
|
}
|
|
|
else if (index_data == 7) {
|
|
|
struct_data.file_soft = buff;
|
|
|
}
|
|
|
else if (index_data == 8) {
|
|
|
struct_data.file_hard = buff;
|
|
|
}
|
|
|
else if (index == 9) {
|
|
|
struct_data.file_grace = buff;
|
|
|
}
|
|
|
index_data += 1;
|
|
|
}
|
|
|
old_index = index;
|
|
|
}
|
|
|
if (index_data != 0) {struct_data.is_validate = true; }
|
|
|
else {struct_data.is_validate = false; }
|
|
|
return struct_data;
|
|
|
}
|
|
|
|
|
|
} |
