From: Richard W.M. Jones <rjones@redhat.com>
Date: Fri, 1 Apr 2011 15:03:14 +0000 (+0100)
Subject: du: Add pulse mode progress messages.
X-Git-Tag: 1.9.16~1
X-Git-Url: http://git.annexia.org/?a=commitdiff_plain;h=9b60f3a9f3b38c085e9265e56ddb839dabe6e416;p=libguestfs.git

du: Add pulse mode progress messages.
---

diff --git a/daemon/du.c b/daemon/du.c
index 7cf2f4a..1c44b3b 100644
--- a/daemon/du.c
+++ b/daemon/du.c
@@ -1,5 +1,5 @@
 /* libguestfs - the guestfsd daemon
- * Copyright (C) 2009 Red Hat Inc.
+ * Copyright (C) 2009-2011 Red Hat Inc.
  *
  * This program is free software; you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
@@ -43,9 +43,12 @@ do_du (const char *path)
     return -1;
   }
 
+  pulse_mode_start ();
+
   r = command (&out, &err, "du", "-s", buf, NULL);
   free (buf);
   if (r == -1) {
+    pulse_mode_cancel ();
     reply_with_error ("%s: %s", path, err);
     free (out);
     free (err);
@@ -55,6 +58,7 @@ do_du (const char *path)
   free (err);
 
   if (sscanf (out, "%"SCNi64, &rv) != 1) {
+    pulse_mode_cancel ();
     reply_with_error ("%s: could not read output: %s", path, out);
     free (out);
     return -1;
@@ -62,5 +66,7 @@ do_du (const char *path)
 
   free (out);
 
+  pulse_mode_end ();
+
   return rv;
 }
diff --git a/generator/generator_actions.ml b/generator/generator_actions.ml
index a324d99..a10b7f7 100644
--- a/generator/generator_actions.ml
+++ b/generator/generator_actions.ml
@@ -3577,7 +3577,7 @@ This command is mostly useful for interactive sessions.  It
 is I<not> intended that you try to parse the output string.
 Use C<guestfs_statvfs> from programs.");
 
-  ("du", (RInt64 "sizekb", [Pathname "path"], []), 127, [],
+  ("du", (RInt64 "sizekb", [Pathname "path"], []), 127, [Progress],
    [InitISOFS, Always, TestOutputInt (
       [["du"; "/directory"]], 2 (* ISO fs blocksize is 2K *))],
    "estimate file space usage",