src/lxc/tools/lxc_wait.c

   1 /*
   2  * lxc: linux Container library
   3  *
   4  * (C) Copyright IBM Corp. 2007, 2008
   5  *
   6  * Authors:
   7  * Daniel Lezcano <daniel.lezcano at free.fr>
   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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  22  */
  23 #include <stdio.h>
  24 #include <string.h>
  25 #include <libgen.h>
  26 #include <unistd.h>
  27 #include <stdlib.h>
  28 #include <signal.h>
  29 #include <sys/types.h>
  30
  31 #include <lxc/lxccontainer.h>
  32
  33 #include "lxc.h"
  34 #include "log.h"
  35 #include "arguments.h"
  36
  37 lxc_log_define(lxc_wait_ui, lxc);
  38
  39 static int my_checker(const struct lxc_arguments* args)
  40 {
  41         if (!args->states) {
  42                 lxc_error(args, "missing state option to wait for.");
  43                 return -1;
  44         }
  45         return 0;
  46 }
  47
  48 static int my_parser(struct lxc_arguments* args, int c, char* arg)
  49 {
  50         switch (c) {
  51         case 's': args->states = optarg; break;
  52         case 't': args->timeout = atol(optarg); break;
  53         }
  54         return 0;
  55 }
  56
  57 static const struct option my_longopts[] = {
  58         {"state", required_argument, 0, 's'},
  59         {"timeout", required_argument, 0, 't'},
  60         LXC_COMMON_OPTIONS
  61 };
  62
  63 static struct lxc_arguments my_args = {
  64         .progname = "lxc-wait",
  65         .help     = "\
  66 --name=NAME --state=STATE\n\
  67 \n\
  68 lxc-wait waits for NAME container state to reach STATE\n\
  69 \n\
  70 Options :\n\
  71   -n, --name=NAME   NAME of the container\n\
  72   -s, --state=STATE ORed states to wait for\n\
  73                     STOPPED, STARTING, RUNNING, STOPPING,\n\
  74                     ABORTING, FREEZING, FROZEN, THAWED\n\
  75   -t, --timeout=TMO Seconds to wait for state changes\n\
  76   --rcfile=FILE     Load configuration file FILE\n",
  77         .options  = my_longopts,
  78         .parser   = my_parser,
  79         .checker  = my_checker,
  80         .timeout = -1,
  81 };
  82
  83 int main(int argc, char *argv[])
  84 {
  85         struct lxc_container *c;
  86
  87         if (lxc_arguments_parse(&my_args, argc, argv))
  88                 return 1;
  89
  90         if (!my_args.log_file)
  91                 my_args.log_file = "none";
  92
  93         if (lxc_log_init(my_args.name, my_args.log_file, my_args.log_priority,
  94                          my_args.progname, my_args.quiet, my_args.lxcpath[0]))
  95                 return 1;
  96         lxc_log_options_no_override();
  97
  98         c = lxc_container_new(my_args.name, my_args.lxcpath[0]);
  99         if (!c)
 100                 return 1;
 101
 102         if (!c->may_control(c)) {
 103                 fprintf(stderr, "Insufficent privileges to control %s\n", c->name);
 104                 lxc_container_put(c);
 105                 return 1;
 106         }
 107
 108         if (my_args.rcfile) {
 109                 c->clear_config(c);
 110                 if (!c->load_config(c, my_args.rcfile)) {
 111                         fprintf(stderr, "Failed to load rcfile\n");
 112                         lxc_container_put(c);
 113                         return 1;
 114                 }
 115         }
 116
 117         if (!c->wait(c, my_args.states, my_args.timeout)) {
 118                 lxc_container_put(c);
 119                 return 1;
 120         }
 121         return 0;
 122 }