git.gag.com Git - debian/amanda/blob - xfer-src/xfer.h

   1 /*
   2  * Copyright (c) 2008-2012 Zmanda, Inc.  All Rights Reserved.
   3  *
   4  * This program is free software; you can redistribute it and/or modify it
   5  * under the terms of the GNU General Public License version 2 as published
   6  * by the Free Software Foundation.
   7  *
   8  * This program is distributed in the hope that it will be useful, but
   9  * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  10  * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  11  * for more details.
  12  *
  13  * You should have received a copy of the GNU General Public License along
  14  * with this program; if not, write to the Free Software Foundation, Inc.,
  15  * 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16  *
  17  * Contact information: Zmanda Inc., 465 S. Mathilda Ave., Suite 300
  18  * Sunnyvale, CA 94085, USA, or: http://www.zmanda.com
  19  */
  20
  21 /* An Xfer abstracts an active data transfer through the Amanda core.
  22  */
  23
  24 #ifndef XFER_H
  25 #define XFER_H
  26
  27 #include <glib.h>
  28
  29 /* An Xfer represents a flow of data from a source, via zero or more filters,
  30  * to a destination.  Sources, filters, and destinations are "transfer elements".
  31  * The job of the Xfer is glue together a sequence of elements, and provide a
  32  * dispatch point for messages from those elements to the caller.
  33  *
  34  * Xfers are not implemented as GObjects because there is no reason to subclass an
  35  * Xfer or apply any of the other features that come with GObject.
  36  */
  37
  38 /* The moment-to-moment state of a transfer */
  39 typedef enum {
  40     XFER_INIT = 1,      /* initializing */
  41     XFER_START = 2,     /* starting */
  42     XFER_RUNNING = 3,   /* data flowing */
  43     XFER_CANCELLING = 4,/* cancellation begun */
  44     XFER_CANCELLED = 5, /* all elements cancelled; draining data */
  45     XFER_DONE = 6,      /* data no longer flowing */
  46 } xfer_status;
  47
  48 /* forward declarations */
  49 struct XferElement;
  50 struct XMsgSource;
  51 struct XMsg;
  52
  53 /*
  54  * "Class" declaration
  55  */
  56
  57 struct Xfer {
  58     /* The current status of this transfer.  This is read-only, and
  59      * must only be accessed from the main thread or with status_mutex
  60      * held. */
  61     xfer_status status;
  62
  63     /* lock this while checking status in a thread
  64      * other than the main thread */
  65     GMutex *status_mutex;
  66
  67     /* and wait on this for status changes */
  68     GCond *status_cond;
  69
  70     /* -- remaining fields are private -- */
  71
  72     gint refcount;
  73
  74     /* All transfer elements for this transfer, in order from
  75      * source to destination.  This is initialized when the Xfer is
  76      * created. */
  77     GPtrArray *elements;
  78
  79     /* temporary string for a representation of this transfer */
  80     char *repr;
  81
  82     /* GSource and queue for incoming messages */
  83     struct XMsgSource *msg_source;
  84     GAsyncQueue *queue;
  85
  86     /* Number of active elements remaining (a.k.a. the number of
  87      * XMSG_DONE messages to expect) */
  88     gint num_active_elements;
  89
  90     /* Used to coordinate handing off file descriptors among elements of this
  91      * xfer */
  92     GMutex *fd_mutex;
  93
  94     int cancelled;
  95 };
  96
  97 typedef struct Xfer Xfer;
  98
  99 /* Note that all functions must be called from the main thread unless
 100  * otherwise noted */
 101
 102 /* Create a new Xfer object, which should later be freed with xfref_free.
 103  *
 104  * This function adds a reference to each element.  The caller should
 105  * unreference the elements if it does not intend to use them directly.
 106  * The Xfer returned has a refcount of one.
 107  *
 108  * @param elements: array of pointers to transfer elements, in order from source
 109  *     to destination
 110  * @param nelements: length of 'elements'
 111  * @returns: new Xfer object
 112  */
 113 Xfer *xfer_new(struct XferElement **elements, unsigned int nelements);
 114
 115 /* Increase the reference count of a transfer.
 116  *
 117  * @param xfer: the transfer
 118  */
 119 void xfer_ref(Xfer *xfer);
 120
 121 /* Decrease the reference count of a transfer, possibly freeing it.  A running
 122  * transfer (state neither XFER_INIT nor XFER_DONE) will not be freed.
 123  *
 124  * @param xfer: the transfer
 125  */
 126 void xfer_unref(Xfer *xfer);
 127
 128 /* Get a GSource which will produce events corresponding to messages from
 129  * this transfer.  This is a "peek" operation, so the reference count for the
 130  * GSource is not affected.  Note that the same GSource is returned on every
 131  * call for a particular transfer.
 132  *
 133  * @returns: GSource object
 134  */
 135 GSource *xfer_get_source(Xfer *xfer);
 136
 137 /* Typedef for the callback to be set on the GSource returned from
 138  * xfer_get_source.
 139  */
 140 typedef void (*XMsgCallback)(gpointer data, struct XMsg *msg, Xfer *xfer);
 141
 142 /* Queue a message for delivery via this transfer's GSource.  This can
 143  * be called in any thread.
 144  *
 145  * @param xfer: the transfer
 146  * @param msg: the message to queue
 147  */
 148 void xfer_queue_message(Xfer *xfer, struct XMsg *msg);
 149
 150 /* Get a representation of this transfer.  The string belongs to the transfer, and
 151  * will be freed when the transfer is freed.
 152  *
 153  * @param xfer: the Xfer object
 154  * @returns: statically allocated string
 155  */
 156 char *xfer_repr(Xfer *xfer);
 157
 158 /* Start a transfer.  This function will fail with an error message if it is
 159  * unable to set up the transfer (e.g., if the elements cannot be connected
 160  * correctly).
 161  *
 162  * @param xfer: the Xfer object
 163  * @param offset: the offset to start the transfer from (must be 0)
 164  * @param size: the Xfer object: the number of bytes to transfer.
 165  */
 166 void xfer_start(Xfer *xfer, gint64 offset, gint64 size);
 167
 168 /* Abort a running transfer.  This essentially tells the source to stop
 169  * producing data and allows the remainder of the transfer to "drain".  Thus
 170  * the transfer will signal its completion "normally" some time after
 171  * xfer_cancel returns.  In particular, the state transitions will occur
 172  * as follows:
 173  *
 174  * - XFER_RUNNING
 175  * - xfer_cancel()  (note state may still be XFER_RUNNING on return)
 176  * - XFER_CANCELLING
 177  * - (individual elements' cancel() methods are invoked)
 178  * - XFER_CANCELLED
 179  * - (data drains from the transfer)
 180  * - XFER_DONE
 181  *
 182  * This function can be called from any thread at any time.  It will return
 183  * without blocking.
 184  *
 185  * @param xfer: the Xfer object
 186  */
 187 void xfer_cancel(Xfer *xfer);
 188
 189 /*
 190  * Utilities
 191  */
 192
 193 /* Wait for the xfer's state to become CANCELLED or DONE; this is useful to
 194  * wait until a cancelletion is in progress before returning an EOF or
 195  * otherwise handling a failure.  If you call this in the main thread, you'll
 196  * be waiting for a while.
 197  *
 198  * @param xfer: the transfer object
 199  * @returns: the new status (XFER_CANCELLED or XFER_DONE)
 200  */
 201 xfer_status wait_until_xfer_cancelled(Xfer *xfer);
 202
 203 /* Wait for the xfer's state to become anything but START; this is
 204  * called *automatically* for every xfer_element_pull_buffer call, as the
 205  * upstream element may not be running and ready for a pull just yet.  But
 206  * the function may be useful in other places, too.
 207  *
 208  * @param xfer: the transfer object
 209  * @returns: the new status (XFER_CANCELLED or XFER_DONE)
 210  */
 211 xfer_status wait_until_xfer_running(Xfer *xfer);
 212
 213 /* Send an XMSG_ERROR constructed with the given format and arguments, then
 214  * cancel the transfer, then wait until the transfer is completely cancelled.
 215  * This is the most common error-handling process for transfer elements.  All
 216  * that remains to be done on return is to branch to the appropriate point in
 217  * the cancellation-handling portion of the transfer.
 218  *
 219  * @param elt: the transfer element producing the error
 220  * @param fmt: the format for the error message
 221  * @param ...: arguments corresponding to the format
 222  */
 223 void xfer_cancel_with_error(struct XferElement *elt, const char *fmt, ...)
 224         G_GNUC_PRINTF(2,3);
 225
 226 /* Return the fd in *FDP and set *FDP to NEWFD, all in one step.  The operation
 227  * is atomic with respect to all other such operations in this transfer, making
 228  * this a good way to "move" a file descriptor from one element to another.  If
 229  * xfer is NULL, the operation proceeds with no locking.
 230  *
 231  * @param xfer: the xfer within which this fd is used
 232  * @param fdp: pointer to the file descriptor to swap
 233  * @param newfd: the new value for *FDP
 234  * @returns: the previous contents of *fdp (may be -1)
 235  */
 236 gint xfer_atomic_swap_fd(Xfer *xfer, gint *fdp, gint newfd);
 237
 238 #endif /* XFER_H */