debugfs/quota.c

   1 /*
   2  * quota.c --- debugfs quota commands
   3  *
   4  * Copyright (C) 2014 Theodore Ts'o.  This file may be redistributed
   5  * under the terms of the GNU Public License.
   6  */
   7
   8 #include "config.h"
   9 #include <stdio.h>
  10 #include <unistd.h>
  11 #include <stdlib.h>
  12 #include <ctype.h>
  13 #include <string.h>
  14 #include <time.h>
  15 #ifdef HAVE_ERRNO_H
  16 #include <errno.h>
  17 #endif
  18 #include <sys/types.h>
  19 #ifdef HAVE_GETOPT_H
  20 #include <getopt.h>
  21 #else
  22 extern int optind;
  23 extern char *optarg;
  24 #endif
  25
  26 #include "debugfs.h"
  27
  28 const char *quota_type[] = { "user", "group", "project", NULL };
  29
  30 static int load_quota_ctx(char *progname)
  31 {
  32         errcode_t       retval;
  33
  34         if (check_fs_open(progname))
  35                 return 1;
  36
  37         if (!ext2fs_has_feature_quota(current_fs->super)) {
  38                 com_err(progname, 0, "quota feature not enabled");
  39                 return 1;
  40         }
  41
  42         if (current_qctx)
  43                 return 0;
  44
  45         retval = quota_init_context(&current_qctx, current_fs, 0);
  46         if (retval) {
  47                 com_err(current_fs->device_name, retval,
  48                         "while trying to load quota information");
  49                 return 1;
  50         }
  51         return 0;
  52 }
  53
  54 static int parse_quota_type(const char *cmdname, const char *str)
  55 {
  56         errcode_t       retval;
  57         char            *t;
  58         int             flags = 0;
  59         int             i;
  60
  61         for (i = 0; i < MAXQUOTAS; i++) {
  62                 if (strcasecmp(str, quota_type[i]) == 0)
  63                         break;
  64         }
  65         if (i >= MAXQUOTAS) {
  66                 i = strtol(str, &t, 0);
  67                 if (*t)
  68                         i = -1;
  69         }
  70         if (i < 0 || i >= MAXQUOTAS) {
  71                 com_err(0, 0, "Invalid quota type: %s", str);
  72                 printf("Valid quota types are: ");
  73                 for (i = 0; i < MAXQUOTAS; i++)
  74                         printf("%s ", quota_type[i]);
  75                 printf("\n");
  76                 return -1;
  77         }
  78
  79         if (current_fs->flags & EXT2_FLAG_RW)
  80                 flags |= EXT2_FILE_WRITE;
  81
  82         retval = quota_file_open(current_qctx, NULL, 0, i, -1, flags);
  83         if (retval) {
  84                 com_err(cmdname, retval,
  85                         "while opening quota inode (type %d)", i);
  86                 return -1;
  87         }
  88         return i;
  89 }
  90
  91
  92 static int list_quota_callback(struct dquot *dq,
  93                                void *cb_data EXT2FS_ATTR((unused)))
  94 {
  95         printf("%10u   %8lld %8lld %8lld    %8lld %8lld %8lld\n",
  96                dq->dq_id, (long long)dq->dq_dqb.dqb_curspace,
  97                (long long)dq->dq_dqb.dqb_bsoftlimit,
  98                (long long)dq->dq_dqb.dqb_bhardlimit,
  99                (long long)dq->dq_dqb.dqb_curinodes,
 100                (long long)dq->dq_dqb.dqb_isoftlimit,
 101                (long long)dq->dq_dqb.dqb_ihardlimit);
 102         return 0;
 103 }
 104
 105 void do_list_quota(int argc, char *argv[], int sci_idx EXT2FS_ATTR((unused)),
 106                    void *infop EXT2FS_ATTR((unused)))
 107 {
 108         errcode_t       retval;
 109         int             type;
 110         struct quota_handle *qh;
 111
 112         if (load_quota_ctx(argv[0]))
 113                 return;
 114
 115         if (argc != 2) {
 116                 com_err(0, 0, "Usage: list_quota <quota_type>\n");
 117                 return;
 118         }
 119
 120         type = parse_quota_type(argv[0], argv[1]);
 121         if (type < 0)
 122                 return;
 123
 124         printf("%7s %2s   %8s %8s %8s    %8s %8s %8s\n",
 125                quota_type[type], "id",
 126                "blocks", "quota", "limit", "inodes", "quota", "limit");
 127         qh = current_qctx->quota_file[type];
 128         retval = qh->qh_ops->scan_dquots(qh, list_quota_callback, NULL);
 129         if (retval) {
 130                 com_err(argv[0], retval, "while scanning dquots");
 131                 return;
 132         }
 133 }
 134
 135 void do_get_quota(int argc, char *argv[], int sci_idx EXT2FS_ATTR((unused)),
 136                   void *infop EXT2FS_ATTR((unused)))
 137 {
 138         int             err, type;
 139         struct quota_handle *qh;
 140         struct dquot    *dq;
 141         qid_t           id;
 142
 143         if (load_quota_ctx(argv[0]))
 144                 return;
 145
 146         if (argc != 3) {
 147                 com_err(0, 0, "Usage: get_quota <quota_type> <id>\n");
 148                 return;
 149         }
 150
 151         type = parse_quota_type(argv[0], argv[1]);
 152         if (type < 0)
 153                 return;
 154
 155         id = parse_ulong(argv[2], argv[0], "id", &err);
 156         if (err)
 157                 return;
 158
 159         printf("%7s %2s   %8s %8s %8s    %8s %8s %8s\n",
 160                quota_type[type], "id",
 161                "blocks", "quota", "limit", "inodes", "quota", "limit");
 162
 163         qh = current_qctx->quota_file[type];
 164
 165         dq = qh->qh_ops->read_dquot(qh, id);
 166         if (dq) {
 167                 list_quota_callback(dq, NULL);
 168                 ext2fs_free_mem(&dq);
 169         } else {
 170                 com_err(argv[0], 0, "couldn't read quota record");
 171         }
 172 }