diff --git a/source/controler.cc b/source/controler.cc index 822fb5d..38f889a 100644 --- a/source/controler.cc +++ b/source/controler.cc @@ -21,7 +21,7 @@ bool Back_Back::wrapper_off_quotas(string quotas_type) { flag_status = obj_quotas_sys.off_quota(partition, cmd_hw, cmd_sw); } else if (flag_mode == "ubconfig") { - flag_status = obj_quotas_ubconfig.off_quota(partition, quotas_type); + flag_status = obj_quotas_ubconfig.off_quota(partition, quotas_type, "disable"); } return flag_status; } @@ -35,6 +35,7 @@ bool Back_Back::wrapper_on_quotas(string quotas_type) { flag_status = obj_quotas_sys.wrapper_on_quotas(partition, quotas_type); } else if (flag_mode == "ubconfig") { + flag_status = obj_quotas_sys.wrapper_on_quotas(partition, quotas_type, "enable"); flag_status = obj_quotas_ubconfig.format_cmd_quotas(); } return flag_status; diff --git a/source/ubconfig_quotas.cc b/source/ubconfig_quotas.cc index e420db6..88bcb30 100644 --- a/source/ubconfig_quotas.cc +++ b/source/ubconfig_quotas.cc @@ -9,10 +9,10 @@ void Quotas_ubconfig::format_cmd_quotas(struct struct_quotas& _struct_quota) { (*map_gui_cfg)[key] = value; } -bool Quotas_ubconfig::off_quota(string partition, string quota_type) { +bool Quotas_ubconfig::off_quota(string partition, string quota_type, string state) { string key = "DISK_QUOTA[" + quota_type + ":" + partition + "]"; string value = "disable"; - (*map_gui_cfg)[key] = value; + (*map_gui_cfg)[key] = state; return true; } void Quotas_ubconfig::set_map_gui(map_str_str &map_gui) { diff --git a/source/ubconfig_quotas.h b/source/ubconfig_quotas.h index 3df6291..f9a0d4d 100644 --- a/source/ubconfig_quotas.h +++ b/source/ubconfig_quotas.h @@ -17,7 +17,7 @@ class Quotas_ubconfig{ My_Process::My_Process_system obj_process_system = My_Process::My_Process_system(); My_Process::My_Process_call obj_process_call = My_Process::My_Process_call(); Quotas_ubconfig(); - bool off_quota(string partition, string quota_type); + bool off_quota(string partition, string quota_type, string state); void format_cmd_quotas(struct struct_quotas& _struct_quota); void set_map_gui(map_str_str &map_gui); };