src/lxc/lxc_cgroup.c

   1 /*
   2  * lxc: linux Container library
   3  *
   4  * (C) Copyright IBM Corp. 2007, 2008
   5  *
   6  * Authors:
   7  * Daniel Lezcano <dlezcano at fr.ibm.com>
   8  *
   9  * This library is free software; you can redistribute it and/or
  10  * modify it under the terms of the GNU Lesser General Public
  11  * License as published by the Free Software Foundation; either
  12  * version 2.1 of the License, or (at your option) any later version.
  13  *
  14  * This library is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  17  * Lesser General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU Lesser General Public
  20  * License along with this library; if not, write to the Free Software
  21  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  22  */
  23
  24 #include <stdio.h>
  25 #include <unistd.h>
  26 #include <libgen.h>
  27 #include <sys/types.h>
  28
  29 #include <lxc/lxc.h>
  30 #include <lxc/log.h>
  31
  32 #include "arguments.h"
  33
  34 lxc_log_define(lxc_cgroup_ui, lxc_cgroup);
  35
  36 static int my_checker(const struct lxc_arguments* args)
  37 {
  38         if (!args->argc) {
  39                 lxc_error(args, "missing cgroup subsystem");
  40                 return -1;
  41         }
  42         return 0;
  43 }
  44
  45 static const struct option my_longopts[] = {
  46         LXC_COMMON_OPTIONS
  47 };
  48
  49 static struct lxc_arguments my_args = {
  50         .progname = "lxc-cgroup",
  51         .help     = "\
  52 --name=NAME subsystem [value]\n\
  53 \n\
  54 lxc-cgroup get or set subsystem value of cgroup\n\
  55 associated with the NAME container\n\
  56 \n\
  57 Options :\n\
  58   -n, --name=NAME      NAME for name of the container",
  59         .options  = my_longopts,
  60         .parser   = NULL,
  61         .checker  = my_checker,
  62 };
  63
  64 int main(int argc, char *argv[])
  65 {
  66         char *subsystem = NULL, *value = NULL;
  67
  68         if (lxc_arguments_parse(&my_args, argc, argv))
  69                 return -1;
  70
  71         if (lxc_log_init(my_args.log_file, my_args.log_priority,
  72                          my_args.progname, my_args.quiet))
  73                 return -1;
  74
  75         subsystem = my_args.argv[0];
  76
  77         if ((argc) > 1)
  78                 value = my_args.argv[1];
  79
  80         if (value) {
  81                 if (lxc_cgroup_set(my_args.name, subsystem, value)) {
  82                         ERROR("failed to assign '%s' value to '%s' for '%s'",
  83                                 value, subsystem, my_args.name);
  84                         return -1;
  85                 }
  86         } else {
  87                 const unsigned long len = 4096;
  88                 int ret;
  89                 char buffer[len];
  90
  91                 ret = lxc_cgroup_get(my_args.name, subsystem, buffer, len);
  92                 if (ret < 0) {
  93                         ERROR("failed to retrieve value of '%s' for '%s'",
  94                               subsystem, my_args.name);
  95                         return -1;
  96                 }
  97
  98                 printf("%*s", ret, buffer);
  99         }
 100
 101         return 0;
 102 }