X-Git-Url: http://git.annexia.org/?p=libguestfs.git;a=blobdiff_plain;f=daemon%2Fdebug.c;h=ce760e6f3e373c23672ba64d81d3158026edd41a;hp=22baa9bff8a5037181c7cb39f3bcf8771e3f8cb2;hb=57d2dfab18ad3d987d9273bb7c1f42e73e0bbcb2;hpb=713283f4a67b3a1960ca96ce6608d046fc202d2d

diff --git a/daemon/debug.c b/daemon/debug.c
index 22baa9b..ce760e6 100644
--- a/daemon/debug.c
+++ b/daemon/debug.c
@@ -21,7 +21,10 @@
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
+#include <sys/types.h>
+#include <sys/stat.h>
 #include <unistd.h>
+#include <dirent.h>
 
 #include "../src/guestfs_protocol.h"
 #include "daemon.h"
@@ -42,27 +45,23 @@ struct cmd {
 };
 
 static char *debug_help (const char *subcmd, int argc, char *const *const argv);
-#if 0
+static char *debug_env (const char *subcmd, int argc, char *const *const argv);
 static char *debug_fds (const char *subcmd, int argc, char *const *const argv);
-static char *debug_free (const char *subcmd, int argc, char *const *const argv);
-static char *debug_mem (const char *subcmd, int argc, char *const *const argv);
-static char *debug_ps (const char *subcmd, int argc, char *const *const argv);
-#endif
+static char *debug_segv (const char *subcmd, int argc, char *const *const argv);
+static char *debug_sh (const char *subcmd, int argc, char *const *const argv);
 
 static struct cmd cmds[] = {
   { "help", debug_help },
-#if 0
+  { "env", debug_env },
   { "fds", debug_fds },
-  { "free", debug_free },
-  { "mem", debug_mem },
-  { "ps", debug_free },
-#endif
+  { "segv", debug_segv },
+  { "sh", debug_sh },
   { NULL, NULL }
 };
 #endif
 
 char *
-do_debug (const char *subcmd, char *const *const argv)
+do_debug (char *subcmd, char **argv)
 {
 #if ENABLE_DEBUG_COMMAND
   int argc, i;
@@ -114,25 +113,120 @@ debug_help (const char *subcmd, int argc, char *const *const argv)
   return r;
 }
 
-#if 0
+/* Show open FDs. */
 static char *
 debug_fds (const char *subcmd, int argc, char *const *const argv)
 {
+  int r;
+  char *out;
+  size_t size;
+  FILE *fp;
+  DIR *dir;
+  struct dirent *d;
+  char fname[256], link[256];
+  struct stat statbuf;
+
+  fp = open_memstream (&out, &size);
+  if (!fp) {
+    reply_with_perror ("open_memstream");
+    return NULL;
+  }
+
+  dir = opendir ("/proc/self/fd");
+  if (!dir) {
+    reply_with_perror ("opendir: /proc/self/fd");
+    fclose (fp);
+    return NULL;
+  }
+
+  while ((d = readdir (dir)) != NULL) {
+    if (strcmp (d->d_name, ".") == 0 || strcmp (d->d_name, "..") == 0)
+      continue;
+
+    snprintf (fname, sizeof fname, "/proc/self/fd/%s", d->d_name);
+
+    r = lstat (fname, &statbuf);
+    if (r == -1) {
+      reply_with_perror ("stat: %s", fname);
+      fclose (fp);
+      free (out);
+      closedir (dir);
+      return NULL;
+    }
+
+    if (S_ISLNK (statbuf.st_mode)) {
+      r = readlink (fname, link, sizeof link - 1);
+      if (r == -1) {
+	reply_with_perror ("readline: %s", fname);
+	fclose (fp);
+	free (out);
+	closedir (dir);
+	return NULL;
+      }
+      link[r] = '\0';
+
+      fprintf (fp, "%2s %s\n", d->d_name, link);
+    } else
+      fprintf (fp, "%2s 0%o\n", d->d_name, statbuf.st_mode);
+  }
+
+  fclose (fp);
+
+  if (closedir (dir) == -1) {
+    reply_with_perror ("closedir");
+    free (out);
+    return NULL;
+  }
+
+  return out;
 }
 
+/* Force a segfault in the daemon. */
 static char *
-debug_free (const char *subcmd, int argc, char *const *const argv)
+debug_segv (const char *subcmd, int argc, char *const *const argv)
 {
+  *(int*)0 = 0;
+  return NULL;
 }
 
+/* Run an arbitrary shell command. */
 static char *
-debug_mem (const char *subcmd, int argc, char *const *const argv)
+debug_sh (const char *subcmd, int argc, char *const *const argv)
 {
+  int r;
+  char *out, *err;
+
+  r = commandv (&out, &err, argv);
+  if (r == -1) {
+    reply_with_error ("sh: %s", err);
+    free (out);
+    free (err);
+    return NULL;
+  }
+
+  free (err);
+
+  return out;
 }
 
+/* Print the environment that commands get (by running external printenv). */
 static char *
-debug_ps (const char *subcmd, int argc, char *const *const argv)
+debug_env (const char *subcmd, int argc, char *const *const argv)
 {
+  int r;
+  char *out, *err;
+
+  r = command (&out, &err, "printenv", NULL);
+  if (r == -1) {
+    reply_with_error ("printenv: %s", err);
+    free (out);
+    free (err);
+    return NULL;
+  }
+
+  free (err);
+
+  return out;
 }
-#endif
+
 #endif /* ENABLE_DEBUG_COMMAND */