diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/guestfs-actions.c | 88 | ||||
-rw-r--r-- | src/guestfs-actions.h | 1 | ||||
-rw-r--r-- | src/guestfs_protocol.c | 22 | ||||
-rw-r--r-- | src/guestfs_protocol.h | 18 | ||||
-rw-r--r-- | src/guestfs_protocol.x | 10 |
5 files changed, 138 insertions, 1 deletions
diff --git a/src/guestfs-actions.c b/src/guestfs-actions.c index 1a632360..70d5ccec 100644 --- a/src/guestfs-actions.c +++ b/src/guestfs-actions.c @@ -5756,3 +5756,91 @@ int guestfs_download (guestfs_h *g, return 0; } +struct checksum_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 = send called, + * 1001 = reply called. + */ + int cb_sequence; + struct guestfs_message_header hdr; + struct guestfs_message_error err; + struct guestfs_checksum_ret ret; +}; + +static void checksum_reply_cb (guestfs_h *g, void *data, XDR *xdr) +{ + guestfs_main_loop *ml = guestfs_get_main_loop (g); + struct checksum_ctx *ctx = (struct checksum_ctx *) data; + + ml->main_loop_quit (ml, g); + + if (!xdr_guestfs_message_header (xdr, &ctx->hdr)) { + error (g, "%s: failed to parse reply header", "guestfs_checksum"); + 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_checksum"); + return; + } + goto done; + } + if (!xdr_guestfs_checksum_ret (xdr, &ctx->ret)) { + error (g, "%s: failed to parse reply", "guestfs_checksum"); + return; + } + done: + ctx->cb_sequence = 1001; +} + +char *guestfs_checksum (guestfs_h *g, + const char *csumtype, + const char *path) +{ + struct guestfs_checksum_args args; + struct checksum_ctx ctx; + guestfs_main_loop *ml = guestfs_get_main_loop (g); + int serial; + + if (check_state (g, "guestfs_checksum") == -1) return NULL; + guestfs_set_busy (g); + + memset (&ctx, 0, sizeof ctx); + + args.csumtype = (char *) csumtype; + args.path = (char *) path; + serial = guestfs__send_sync (g, GUESTFS_PROC_CHECKSUM, + (xdrproc_t) xdr_guestfs_checksum_args, (char *) &args); + if (serial == -1) { + guestfs_set_ready (g); + return NULL; + } + + read_reply: + guestfs__switch_to_receiving (g); + ctx.cb_sequence = 0; + guestfs_set_reply_callback (g, checksum_reply_cb, &ctx); + (void) ml->main_loop_run (ml, g); + guestfs_set_reply_callback (g, NULL, NULL); + if (ctx.cb_sequence != 1001) { + error (g, "%s reply failed, see earlier error messages", "guestfs_checksum"); + guestfs_set_ready (g); + return NULL; + } + + if (check_reply_header (g, &ctx.hdr, GUESTFS_PROC_CHECKSUM, serial) == -1) { + guestfs_set_ready (g); + return NULL; + } + + if (ctx.hdr.status == GUESTFS_STATUS_ERROR) { + error (g, "%s", ctx.err.error_message); + guestfs_set_ready (g); + return NULL; + } + + guestfs_set_ready (g); + return ctx.ret.checksum; /* caller will free */ +} + diff --git a/src/guestfs-actions.h b/src/guestfs-actions.h index e3c3cf89..c694ddda 100644 --- a/src/guestfs-actions.h +++ b/src/guestfs-actions.h @@ -105,3 +105,4 @@ extern int guestfs_blockdev_flushbufs (guestfs_h *handle, const char *device); extern int guestfs_blockdev_rereadpt (guestfs_h *handle, const char *device); extern int guestfs_upload (guestfs_h *handle, const char *filename, const char *remotefilename); extern int guestfs_download (guestfs_h *handle, const char *remotefilename, const char *filename); +extern char *guestfs_checksum (guestfs_h *handle, const char *csumtype, const char *path); diff --git a/src/guestfs_protocol.c b/src/guestfs_protocol.c index 24b14a07..fc2eae97 100644 --- a/src/guestfs_protocol.c +++ b/src/guestfs_protocol.c @@ -1207,6 +1207,28 @@ xdr_guestfs_download_args (XDR *xdrs, guestfs_download_args *objp) } bool_t +xdr_guestfs_checksum_args (XDR *xdrs, guestfs_checksum_args *objp) +{ + register int32_t *buf; + + if (!xdr_string (xdrs, &objp->csumtype, ~0)) + return FALSE; + if (!xdr_string (xdrs, &objp->path, ~0)) + return FALSE; + return TRUE; +} + +bool_t +xdr_guestfs_checksum_ret (XDR *xdrs, guestfs_checksum_ret *objp) +{ + register int32_t *buf; + + if (!xdr_string (xdrs, &objp->checksum, ~0)) + 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 4c3f1cd3..05bd40bb 100644 --- a/src/guestfs_protocol.h +++ b/src/guestfs_protocol.h @@ -626,6 +626,17 @@ struct guestfs_download_args { }; typedef struct guestfs_download_args guestfs_download_args; +struct guestfs_checksum_args { + char *csumtype; + char *path; +}; +typedef struct guestfs_checksum_args guestfs_checksum_args; + +struct guestfs_checksum_ret { + char *checksum; +}; +typedef struct guestfs_checksum_ret guestfs_checksum_ret; + enum guestfs_procedure { GUESTFS_PROC_MOUNT = 1, GUESTFS_PROC_SYNC = 2, @@ -694,7 +705,8 @@ enum guestfs_procedure { GUESTFS_PROC_BLOCKDEV_REREADPT = 65, GUESTFS_PROC_UPLOAD = 66, GUESTFS_PROC_DOWNLOAD = 67, - GUESTFS_PROC_NR_PROCS = 67 + 1, + GUESTFS_PROC_CHECKSUM = 68, + GUESTFS_PROC_NR_PROCS = 68 + 1, }; typedef enum guestfs_procedure guestfs_procedure; #define GUESTFS_MESSAGE_MAX 4194304 @@ -839,6 +851,8 @@ extern bool_t xdr_guestfs_blockdev_flushbufs_args (XDR *, guestfs_blockdev_flus extern bool_t xdr_guestfs_blockdev_rereadpt_args (XDR *, guestfs_blockdev_rereadpt_args*); extern bool_t xdr_guestfs_upload_args (XDR *, guestfs_upload_args*); extern bool_t xdr_guestfs_download_args (XDR *, guestfs_download_args*); +extern bool_t xdr_guestfs_checksum_args (XDR *, guestfs_checksum_args*); +extern bool_t xdr_guestfs_checksum_ret (XDR *, guestfs_checksum_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*); @@ -942,6 +956,8 @@ extern bool_t xdr_guestfs_blockdev_flushbufs_args (); extern bool_t xdr_guestfs_blockdev_rereadpt_args (); extern bool_t xdr_guestfs_upload_args (); extern bool_t xdr_guestfs_download_args (); +extern bool_t xdr_guestfs_checksum_args (); +extern bool_t xdr_guestfs_checksum_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 ff4713fd..ca2d0ced 100644 --- a/src/guestfs_protocol.x +++ b/src/guestfs_protocol.x @@ -483,6 +483,15 @@ struct guestfs_download_args { string remotefilename<>; }; +struct guestfs_checksum_args { + string csumtype<>; + string path<>; +}; + +struct guestfs_checksum_ret { + string checksum<>; +}; + enum guestfs_procedure { GUESTFS_PROC_MOUNT = 1, GUESTFS_PROC_SYNC = 2, @@ -551,6 +560,7 @@ enum guestfs_procedure { GUESTFS_PROC_BLOCKDEV_REREADPT = 65, GUESTFS_PROC_UPLOAD = 66, GUESTFS_PROC_DOWNLOAD = 67, + GUESTFS_PROC_CHECKSUM = 68, GUESTFS_PROC_NR_PROCS }; |