Command to wait a specific state for the container

diff --git a/src/lxc/lxc_wait.c b/src/lxc/lxc_wait.c
new file mode 100644 (file)
index 0000000..f7d49b2
--- /dev/null
+++ b/src/lxc/lxc_wait.c
@@ -0,0 +1,116 @@
+/*
+ * lxc: linux Container library
+ *
+ * (C) Copyright IBM Corp. 2007, 2008
+ *
+ * Authors:
+ * Daniel Lezcano <dlezcano at fr.ibm.com>
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+ */
+#include <stdio.h>
+#include <string.h>
+#include <libgen.h>
+#include <unistd.h>
+#include <sys/types.h>
+
+#include <lxc/lxc.h>
+
+void usage(char *cmd)
+{
+       fprintf(stderr, "%s <command>\n", basename(cmd));
+       fprintf(stderr, "\t -n <name>   : name of the container\n");
+       fprintf(stderr, "\t -s <states> : ORed states to wait for STOPPED, " \
+               "STARTING, RUNNING, STOPPING, ABORTING, FREEZING, FROZEN\n");
+       _exit(1);
+}
+
+static int fillwaitedstates(char *strstates, int *states)
+{
+       char *token, *saveptr = NULL;
+       int state;
+
+       token = strtok_r(strstates, "|", &saveptr);
+       while (token) {
+
+               state = lxc_str2state(token);
+               if (state < 0)
+                       return -1;
+               states[state] = 1;
+
+               token = strtok_r(NULL, "|", &saveptr);
+               
+       }
+       return 0;
+}
+
+int main(int argc, char *argv[])
+{
+       char opt, *name = NULL, *states = NULL;
+       struct lxc_msg msg;
+       int s[MAX_STATE] = { }, fd;
+
+       while ((opt = getopt(argc, argv, "s:n:")) != -1) {
+               switch (opt) {
+               case 'n':
+                       name = optarg;
+                       break;
+               case 's':
+                       states = optarg;
+                       break;
+               }
+       }
+
+       if (!name || !states)
+               usage(argv[0]);
+
+       if (fillwaitedstates(states, s)) {
+               fprintf(stderr, "invalid states specified\n");
+               usage(argv[0]);
+       }
+
+       
+       fd = lxc_monitor_open(name);
+       if (fd < 0) {
+               fprintf(stderr, "failed to open monitor for '%s'\n", name);
+               return -1;
+       }
+
+       for (;;) {
+               if (lxc_monitor_read(fd, &msg) < 0) {
+                       fprintf(stderr, 
+                               "failed to read monitor's message for '%s'\n", 
+                               name);
+                       return -1;
+               }
+
+               switch (msg.type) {
+               case lxc_msg_state:
+                       if (msg.value < 0 || msg.value >= MAX_STATE) {
+                               fprintf(stderr, "Receive an invalid state number '%d'\n", msg.value);
+                               return -1;
+                       }
+
+                       if (s[msg.value])
+                               return 0;
+                       break;
+               default:
+                       printf("invalid msg format\n");
+                       break;
+               }
+       }
+
+       return 0;
+}