blob: d983a9f0ceec512ecfb3a9ccfdc4fb1bf78a1df6 [file] [log] [blame] [raw]
/*
* Copyright (C) 2000-2005 SWsoft. All rights reserved.
*
* This file may be distributed under the terms of the Q Public License
* as defined by Trolltech AS of Norway and appearing in the file
* LICENSE.QPL included in the packaging of this file.
*
* This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
* WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
*/
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include "types.h"
#include "quota.h"
#include "vzerror.h"
#include "util.h"
#include "script.h"
#include "logger.h"
#define VZQUOTA "/usr/sbin/vzquota"
void quouta_inc(dq_param *param, int delta)
{
if (param->enable != YES)
return;
if (param->diskspace != NULL) {
param->diskspace[0] += delta;
param->diskspace[1] += delta;
}
if (param->diskinodes != NULL) {
param->diskinodes[0] += delta;
param->diskinodes[1] += delta;
}
}
int quota_set(envid_t veid, char *private, dq_param *param)
{
int i, ret;
char buf[64];
char *arg[24];
if (param == NULL)
return -1;
if (param->diskspace == NULL &&
param->diskinodes == NULL &&
param->exptime == NULL &&
param->ugidlimit == NULL &&
private == NULL)
{
return 0;
}
i = 0;
arg[i++] = strdup("vzquota");
arg[i++] = strdup("setlimit");
snprintf(buf, sizeof(buf), "%d", veid);
arg[i++] = strdup(buf);
if (private != NULL) {
arg[i++] = strdup("-p");
arg[i++] = strdup(private);
}
if (param->diskspace != NULL) {
arg[i++] = strdup("-b");
snprintf(buf, sizeof(buf), "%lu", param->diskspace[0]);
arg[i++] = strdup(buf);
arg[i++] = strdup("-B");
snprintf(buf, sizeof(buf), "%lu", param->diskspace[1]);
arg[i++] = strdup(buf);
}
if (param->diskinodes != NULL) {
arg[i++] = strdup("-i");
snprintf(buf, sizeof(buf), "%lu", param->diskinodes[0]);
arg[i++] = strdup(buf);
arg[i++] = strdup("-I");
snprintf(buf, sizeof(buf), "%lu", param->diskinodes[1]);
arg[i++] = strdup(buf);
}
if (param->exptime != NULL) {
arg[i++] = strdup("-e");
snprintf(buf, sizeof(buf), "%lu", param->exptime[0]);
arg[i++] = strdup(buf);
arg[i++] = strdup("-n");
snprintf(buf, sizeof(buf), "%lu", param->exptime[0]);
arg[i++] = strdup(buf);
}
/* Set ugid limit */
if (param->ugidlimit != NULL && *param->ugidlimit) {
arg[i++] = strdup("-u");
snprintf(buf, sizeof(buf), "%lu", *param->ugidlimit);
arg[i++] = strdup(buf);
}
arg[i] = NULL;
if ((ret = run_script(VZQUOTA, arg, NULL, 0))) {
logger(0, 0, "vzquota setlimit failed [%d]", ret);
ret = VZ_DQ_SET;
}
free_arg(arg);
return ret;
}
int quota_init(envid_t veid, char *private, dq_param *param)
{
int i, ret;
char buf[64];
char *arg[24];
if (param == NULL)
return -1;
if (check_var(private,
"Error: Not enough parameters, private not set"))
{
return VZ_FS_NOPRVT;
}
if (check_var(param->diskspace,
"Error: Not enough parameters, diskspace quota not set"))
{
return VZ_DISKSPACE_NOT_SET;
}
if (check_var(param->diskinodes,
"Error: Not enough parameters, diskinodes quota not set"))
{
return VZ_DISKINODES_NOT_SET;
}
i = 0;
arg[i++] = strdup(VZQUOTA);
arg[i++] = strdup("init");
snprintf(buf, sizeof(buf), "%d", veid);
arg[i++] = strdup(buf);
/* Disk space */
arg[i++] = strdup("-b");
snprintf(buf, sizeof(buf), "%lu", param->diskspace[0]);
arg[i++] = strdup(buf);
arg[i++] = strdup("-B");
snprintf(buf, sizeof(buf), "%lu", param->diskspace[1]);
arg[i++] = strdup(buf);
/*Disk inodes */
arg[i++] = strdup("-i");
snprintf(buf, sizeof(buf), "%lu", param->diskinodes[0]);
arg[i++] = strdup(buf);
arg[i++] = strdup("-I");
snprintf(buf, sizeof(buf), "%lu", param->diskinodes[1]);
arg[i++] = strdup(buf);
/* VPS private */
arg[i++] = strdup("-p");
arg[i++] = strdup(private);
/* Expiration time */
arg[i++] = strdup("-e");
snprintf(buf, sizeof(buf), "%lu",
param->exptime == NULL ? QUOTA_EXPTIME : param->exptime[0]);
arg[i++] = strdup(buf);
arg[i++] = strdup("-n");
arg[i++] = strdup(buf);
/* ugid limit */
arg[i++] = strdup("-s");
if (param->ugidlimit != NULL && *param->ugidlimit) {
arg[i++] = strdup("1");
arg[i++] = strdup("-u");
snprintf(buf, sizeof(buf), "%lu", *param->ugidlimit);
arg[i++] = strdup(buf);
} else
arg[i++] = strdup("0");
arg[i] = NULL;
if ((ret = run_script(VZQUOTA, arg, NULL, 0))) {
logger(0, 0, "vzquota init failed [%d]", ret);
ret = VZ_DQ_INIT;
}
free_arg(arg);
return ret;
}
/** Turn disk quota on.
*
* @param veid VPS id.
* @param private VPS private area path.
* @param dq disk quota parameters.
* @return 0 on success.
*/
int quota_on(envid_t veid, char *private, dq_param *param)
{
int i, ret, retry = 0;
char buf[64];
char *arg[24];
if (param == NULL)
return -1;
if (check_var(param->diskspace,
"Error: Not enough parameters, diskspace quota not set"))
{
return VZ_DISKSPACE_NOT_SET;
}
if (check_var(param->diskinodes,
"Error: Not enough parameters, diskinodes quota not set"))
{
return VZ_DISKINODES_NOT_SET;
}
i = 0;
arg[i++] = strdup(VZQUOTA);
arg[i++] = strdup("on");
snprintf(buf, sizeof(buf), "%d", veid);
arg[i++] = strdup(buf);
arg[i++] = strdup("-r");
arg[i++] = strdup("0");
/* Disk space */
arg[i++] = strdup("-b");
snprintf(buf, sizeof(buf), "%lu", param->diskspace[0]);
arg[i++] = strdup(buf);
arg[i++] = strdup("-B");
snprintf(buf, sizeof(buf), "%lu", param->diskspace[1]);
arg[i++] = strdup(buf);
/*Disk inodes */
arg[i++] = strdup("-i");
snprintf(buf, sizeof(buf), "%lu", param->diskinodes[0]);
arg[i++] = strdup(buf);
arg[i++] = strdup("-I");
snprintf(buf, sizeof(buf), "%lu", param->diskinodes[1]);
arg[i++] = strdup(buf);
/* Expiration time */
arg[i++] = strdup("-e");
snprintf(buf, sizeof(buf), "%lu",
param->exptime == NULL ? QUOTA_EXPTIME : param->exptime[0]);
arg[i++] = strdup(buf);
arg[i++] = strdup("-n");
arg[i++] = strdup(buf);
/* ugid limit */
arg[i++] = strdup("-s");
if (param->ugidlimit != NULL && *param->ugidlimit) {
arg[i++] = strdup("1");
arg[i++] = strdup("-u");
snprintf(buf, sizeof(buf), "%lu", *param->ugidlimit);
arg[i++] = strdup(buf);
} else
arg[i++] = strdup("0");
arg[i] = 0;
/* if (gparam->skipquotacheck == YES)
arg[i++] = strdup("--nocheck");
*/
retry:
if ((ret = run_script(VZQUOTA, arg, NULL, 0))) {
switch (ret) {
case EXITCODE_QUOTNOTEXIST:
ret = quota_init(veid, private, param);
if (!ret && !retry) {
retry++;
goto retry;
}
break;
case EXITCODE_QUOTARUN:
ret = quota_set(veid, private, param);
break;
}
if (ret) {
logger(0, 0, "vzquota on failed [%d]", ret);
ret = VZ_DQ_SET;
}
}
free_arg(arg);
return ret;
}
/** Turn disk quota off.
*
* @param veid VPS id.
* @param dq disk quota parameters.
* @return 0 on success.
*/
int quota_off(envid_t veid, int force)
{
int i, ret;
char buf[64];
char *arg[5];
i = 0;
arg[i++] = strdup(VZQUOTA);
arg[i++] = strdup("off");
snprintf(buf, sizeof(buf), "%d", veid);
arg[i++] = strdup(buf);
if (force)
arg[i++] = strdup("-f");
arg[i] = NULL;
if ((ret = run_script(VZQUOTA, arg, NULL, 0))) {
if (ret != EXITCODE_QUOTANOTRUN) {
logger(0, 0, "vzquota setlimit -f failed [%d]", ret);
ret = VZ_DQ_OFF;
}
}
free_arg(arg);
return ret;
}
/** Disk quota managment wraper.
*
* @param veid VPS id.
* @param cmd quota commands (QUOTA_MARKDURTY QUOTA_DROP QUOTA_STAT)
* @return 0 on success.
*/
int quota_ctl(envid_t veid, int cmd)
{
int i, ret;
char buf[64];
char *arg[5];
int quiet = 0;
int errcode = 0;
i = 0;
arg[i++] = strdup(VZQUOTA);
switch (cmd) {
case QUOTA_MARKDURTY:
arg[i++] =strdup( "setlimit");
snprintf(buf, sizeof(buf), "%d", veid);
arg[i++] = strdup(buf);
arg[i++] = strdup("-f");
quiet = 1;
break;
case QUOTA_DROP :
arg[i++] = strdup("drop");
snprintf(buf, sizeof(buf), "%d", veid);
arg[i++] = strdup(buf);
quiet = 1;
break;
case QUOTA_STAT :
arg[i++] = strdup("stat");
snprintf(buf, sizeof(buf), "%d", veid);
arg[i++] = strdup(buf);
arg[i++] = strdup("-f");
quiet = 1;
break;
case QUOTA_STAT2:
arg[i++] = strdup("stat");
snprintf(buf, sizeof(buf), "%d", veid);
arg[i++] = strdup(buf);
arg[i++] = strdup("-f");
arg[i++] = strdup("-t");
errcode = VZ_DQ_UGID_NOTINITIALIZED;
quiet = 1;
break;
case QUOTA_SHOW:
arg[i++] = strdup("show");
snprintf(buf, sizeof(buf), "%d", veid);
arg[i++] = strdup(buf);
quiet = 1;
break;
default :
logger(0, 0, "quota_ctl: Unknown action %d", cmd);
return VZ_SYSTEM_ERROR;
}
arg[i] = NULL;
if ((ret = run_script(VZQUOTA, arg, NULL, quiet))) {
if (errcode)
ret = errcode;
}
free_arg(arg);
return ret;
}
/** Setup disk quota limits.
*
* @param veid VPS id.
* @param dq disk quota parameters.
* @return 0 on success.
*/
int vps_set_quota(envid_t veid, dq_param *dq)
{
int ret;
unsigned long *tmp_ugidlimit = NULL;
if (dq->enable == NO)
return 0;
if (dq->diskspace == NULL &&
dq->diskinodes == NULL &&
dq->exptime == NULL &&
dq->ugidlimit == NULL)
{
return 0;
}
if (quota_ctl(veid, QUOTA_STAT)) {
logger(0, 0, "Error: Unable to apply new quota values"
" quota not running");
return -1;
}
if (dq->ugidlimit != NULL) {
ret = quota_ctl(veid, QUOTA_STAT2);
if (ret && *dq->ugidlimit) {
logger(0, 0, "Unable to apply new quota values:"
" ugid quota not initialized");
return VZ_DQ_UGID_NOTINITIALIZED;
// tmp_ugidlimit = dq->ugidlimit;
// dq->ugidlimit = NULL;
} else if (!ret && !*dq->ugidlimit) {
logger(0, 0, "WARNING: Unable to turn ugid quota"
" off. new parameters will be applyed"
" on next start");
tmp_ugidlimit = dq->ugidlimit;
dq->ugidlimit = NULL;
}
}
ret = quota_set(veid, NULL, dq);
if (tmp_ugidlimit != NULL)
dq->ugidlimit = tmp_ugidlimit;
return ret;
}
int vps_quotaon(envid_t veid, char *private, dq_param *dq)
{
int ret;
if (dq == NULL)
return 0;
if (dq->enable == NO) {
// quota_ctl(veid, QUOTA_MARKDURTY);
return 0;
} else {
if (quota_ctl(veid, QUOTA_SHOW) == EXITCODE_QUOTNOTEXIST) {
logger(0, 0, "Initializing quota ...");
if ((ret = quota_init(veid, private, dq)))
return ret;
}
}
if ((ret = quota_on(veid, private, dq)))
return ret;
return 0;
}
int vps_quotaoff(envid_t veid, dq_param *dq)
{
if (dq == NULL)
return 0;
if (dq->enable == NO)
return 0;
return quota_off(veid, 0);
}