summaryrefslogtreecommitdiffstats
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/MAX_PROC_NR2
-rw-r--r--src/guestfs-actions.c92
-rw-r--r--src/guestfs-actions.h1
-rw-r--r--src/guestfs_protocol.c20
-rw-r--r--src/guestfs_protocol.h17
-rw-r--r--src/guestfs_protocol.x9
6 files changed, 139 insertions, 2 deletions
diff --git a/src/MAX_PROC_NR b/src/MAX_PROC_NR
index 0a3e7b04..c75acbe2 100644
--- a/src/MAX_PROC_NR
+++ b/src/MAX_PROC_NR
@@ -1 +1 @@
-126
+127
diff --git a/src/guestfs-actions.c b/src/guestfs-actions.c
index 745a52bc..b34c8017 100644
--- a/src/guestfs-actions.c
+++ b/src/guestfs-actions.c
@@ -11557,3 +11557,95 @@ char *guestfs_df_h (guestfs_h *g)
return ctx.ret.output; /* caller will free */
}
+struct du_ctx {
+ /* This flag is set by the callbacks, so we know we've done
+ * the callbacks as expected, and in the right sequence.
+ * 0 = not called, 1 = reply_cb called.
+ */
+ int cb_sequence;
+ struct guestfs_message_header hdr;
+ struct guestfs_message_error err;
+ struct guestfs_du_ret ret;
+};
+
+static void du_reply_cb (guestfs_h *g, void *data, XDR *xdr)
+{
+ guestfs_main_loop *ml = guestfs_get_main_loop (g);
+ struct du_ctx *ctx = (struct du_ctx *) data;
+
+ /* This should definitely not happen. */
+ if (ctx->cb_sequence != 0) {
+ ctx->cb_sequence = 9999;
+ error (g, "%s: internal error: reply callback called twice", "guestfs_du");
+ return;
+ }
+
+ ml->main_loop_quit (ml, g);
+
+ if (!xdr_guestfs_message_header (xdr, &ctx->hdr)) {
+ error (g, "%s: failed to parse reply header", "guestfs_du");
+ return;
+ }
+ if (ctx->hdr.status == GUESTFS_STATUS_ERROR) {
+ if (!xdr_guestfs_message_error (xdr, &ctx->err)) {
+ error (g, "%s: failed to parse reply error", "guestfs_du");
+ return;
+ }
+ goto done;
+ }
+ if (!xdr_guestfs_du_ret (xdr, &ctx->ret)) {
+ error (g, "%s: failed to parse reply", "guestfs_du");
+ return;
+ }
+ done:
+ ctx->cb_sequence = 1;
+}
+
+int64_t guestfs_du (guestfs_h *g,
+ const char *path)
+{
+ struct guestfs_du_args args;
+ struct du_ctx ctx;
+ guestfs_main_loop *ml = guestfs_get_main_loop (g);
+ int serial;
+
+ if (check_state (g, "guestfs_du") == -1) return -1;
+ guestfs_set_busy (g);
+
+ memset (&ctx, 0, sizeof ctx);
+
+ args.path = (char *) path;
+ serial = guestfs__send_sync (g, GUESTFS_PROC_DU,
+ (xdrproc_t) xdr_guestfs_du_args, (char *) &args);
+ if (serial == -1) {
+ guestfs_end_busy (g);
+ return -1;
+ }
+
+ guestfs__switch_to_receiving (g);
+ ctx.cb_sequence = 0;
+ guestfs_set_reply_callback (g, du_reply_cb, &ctx);
+ (void) ml->main_loop_run (ml, g);
+ guestfs_set_reply_callback (g, NULL, NULL);
+ if (ctx.cb_sequence != 1) {
+ error (g, "%s reply failed, see earlier error messages", "guestfs_du");
+ guestfs_end_busy (g);
+ return -1;
+ }
+
+ if (check_reply_header (g, &ctx.hdr, GUESTFS_PROC_DU, serial) == -1) {
+ guestfs_end_busy (g);
+ return -1;
+ }
+
+ if (ctx.hdr.status == GUESTFS_STATUS_ERROR) {
+ error (g, "%s", ctx.err.error_message);
+ free (ctx.err.error_message);
+ guestfs_end_busy (g);
+ return -1;
+ }
+
+ guestfs_end_busy (g);
+ return ctx.ret.sizekb;
+}
+
diff --git a/src/guestfs-actions.h b/src/guestfs-actions.h
index cbf5ff76..3c983325 100644
--- a/src/guestfs-actions.h
+++ b/src/guestfs-actions.h
@@ -197,3 +197,4 @@ extern char **guestfs_tail (guestfs_h *handle, const char *path);
extern char **guestfs_tail_n (guestfs_h *handle, int nrlines, const char *path);
extern char *guestfs_df (guestfs_h *handle);
extern char *guestfs_df_h (guestfs_h *handle);
+extern int64_t guestfs_du (guestfs_h *handle, const char *path);
diff --git a/src/guestfs_protocol.c b/src/guestfs_protocol.c
index 2830bf2a..050d9282 100644
--- a/src/guestfs_protocol.c
+++ b/src/guestfs_protocol.c
@@ -2151,6 +2151,26 @@ xdr_guestfs_df_h_ret (XDR *xdrs, guestfs_df_h_ret *objp)
}
bool_t
+xdr_guestfs_du_args (XDR *xdrs, guestfs_du_args *objp)
+{
+ register int32_t *buf;
+
+ if (!xdr_string (xdrs, &objp->path, ~0))
+ return FALSE;
+ return TRUE;
+}
+
+bool_t
+xdr_guestfs_du_ret (XDR *xdrs, guestfs_du_ret *objp)
+{
+ register int32_t *buf;
+
+ if (!xdr_quad_t (xdrs, &objp->sizekb))
+ return FALSE;
+ return TRUE;
+}
+
+bool_t
xdr_guestfs_procedure (XDR *xdrs, guestfs_procedure *objp)
{
register int32_t *buf;
diff --git a/src/guestfs_protocol.h b/src/guestfs_protocol.h
index 16db25af..c8231a0d 100644
--- a/src/guestfs_protocol.h
+++ b/src/guestfs_protocol.h
@@ -1101,6 +1101,16 @@ struct guestfs_df_h_ret {
};
typedef struct guestfs_df_h_ret guestfs_df_h_ret;
+struct guestfs_du_args {
+ char *path;
+};
+typedef struct guestfs_du_args guestfs_du_args;
+
+struct guestfs_du_ret {
+ quad_t sizekb;
+};
+typedef struct guestfs_du_ret guestfs_du_ret;
+
enum guestfs_procedure {
GUESTFS_PROC_MOUNT = 1,
GUESTFS_PROC_SYNC = 2,
@@ -1228,7 +1238,8 @@ enum guestfs_procedure {
GUESTFS_PROC_TAIL_N = 124,
GUESTFS_PROC_DF = 125,
GUESTFS_PROC_DF_H = 126,
- GUESTFS_PROC_NR_PROCS = 126 + 1,
+ GUESTFS_PROC_DU = 127,
+ GUESTFS_PROC_NR_PROCS = 127 + 1,
};
typedef enum guestfs_procedure guestfs_procedure;
#define GUESTFS_MESSAGE_MAX 4194304
@@ -1456,6 +1467,8 @@ extern bool_t xdr_guestfs_tail_n_args (XDR *, guestfs_tail_n_args*);
extern bool_t xdr_guestfs_tail_n_ret (XDR *, guestfs_tail_n_ret*);
extern bool_t xdr_guestfs_df_ret (XDR *, guestfs_df_ret*);
extern bool_t xdr_guestfs_df_h_ret (XDR *, guestfs_df_h_ret*);
+extern bool_t xdr_guestfs_du_args (XDR *, guestfs_du_args*);
+extern bool_t xdr_guestfs_du_ret (XDR *, guestfs_du_ret*);
extern bool_t xdr_guestfs_procedure (XDR *, guestfs_procedure*);
extern bool_t xdr_guestfs_message_direction (XDR *, guestfs_message_direction*);
extern bool_t xdr_guestfs_message_status (XDR *, guestfs_message_status*);
@@ -1642,6 +1655,8 @@ extern bool_t xdr_guestfs_tail_n_args ();
extern bool_t xdr_guestfs_tail_n_ret ();
extern bool_t xdr_guestfs_df_ret ();
extern bool_t xdr_guestfs_df_h_ret ();
+extern bool_t xdr_guestfs_du_args ();
+extern bool_t xdr_guestfs_du_ret ();
extern bool_t xdr_guestfs_procedure ();
extern bool_t xdr_guestfs_message_direction ();
extern bool_t xdr_guestfs_message_status ();
diff --git a/src/guestfs_protocol.x b/src/guestfs_protocol.x
index 702df5bf..a895b6d5 100644
--- a/src/guestfs_protocol.x
+++ b/src/guestfs_protocol.x
@@ -842,6 +842,14 @@ struct guestfs_df_h_ret {
string output<>;
};
+struct guestfs_du_args {
+ string path<>;
+};
+
+struct guestfs_du_ret {
+ hyper sizekb;
+};
+
enum guestfs_procedure {
GUESTFS_PROC_MOUNT = 1,
GUESTFS_PROC_SYNC = 2,
@@ -969,6 +977,7 @@ enum guestfs_procedure {
GUESTFS_PROC_TAIL_N = 124,
GUESTFS_PROC_DF = 125,
GUESTFS_PROC_DF_H = 126,
+ GUESTFS_PROC_DU = 127,
GUESTFS_PROC_NR_PROCS
};
generated by cgit (git 2.34.1) at 2025-09-24 23:02:02 +0000