git.gag.com Git - debian/amanda/blob - xfer-src/dest-buffer.c

   1 /*
   2  * Amanda, The Advanced Maryland Automatic Network Disk Archiver
   3  * Copyright (c) 2008,2009 Zmanda, Inc.  All Rights Reserved.
   4  *
   5  * This program is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 as published
   7  * by the Free Software Foundation.
   8  *
   9  * This program is distributed in the hope that it will be useful, but
  10  * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  11  * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12  * for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License along
  15  * with this program; if not, write to the Free Software Foundation, Inc.,
  16  * 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  17  *
  18  * Contact information: Zmanda Inc., 465 N Mathlida Ave, Suite 300
  19  * Sunnyvale, CA 94085, USA, or: http://www.zmanda.com
  20  */
  21
  22 #include "amanda.h"
  23 #include "amxfer.h"
  24
  25 /*
  26  * Class declaration
  27  *
  28  * This declaration is entirely private; nothing but xfer_dest_buffer() references
  29  * it directly.
  30  */
  31
  32 GType xfer_dest_buffer_get_type(void);
  33 #define XFER_DEST_BUFFER_TYPE (xfer_dest_buffer_get_type())
  34 #define XFER_DEST_BUFFER(obj) G_TYPE_CHECK_INSTANCE_CAST((obj), xfer_dest_buffer_get_type(), XferDestBuffer)
  35 #define XFER_DEST_BUFFER_CONST(obj) G_TYPE_CHECK_INSTANCE_CAST((obj), xfer_dest_buffer_get_type(), XferDestBuffer const)
  36 #define XFER_DEST_BUFFER_CLASS(klass) G_TYPE_CHECK_CLASS_CAST((klass), xfer_dest_buffer_get_type(), XferDestBufferClass)
  37 #define IS_XFER_DEST_BUFFER(obj) G_TYPE_CHECK_INSTANCE_TYPE((obj), xfer_dest_buffer_get_type ())
  38 #define XFER_DEST_BUFFER_GET_CLASS(obj) G_TYPE_INSTANCE_GET_CLASS((obj), xfer_dest_buffer_get_type(), XferDestBufferClass)
  39
  40 static GObjectClass *parent_class = NULL;
  41
  42 /*
  43  * Main object structure
  44  */
  45
  46 typedef struct XferDestBuffer {
  47     XferElement __parent__;
  48
  49     gsize max_size;
  50
  51     gpointer buf;
  52     gsize len;
  53     gsize allocated;
  54 } XferDestBuffer;
  55
  56 /*
  57  * Class definition
  58  */
  59
  60 typedef struct {
  61     XferElementClass __parent__;
  62
  63     void (*get)(XferDestBuffer *self, gpointer *buf, gsize *size);
  64 } XferDestBufferClass;
  65
  66 /*
  67  * Implementation
  68  */
  69
  70 static void
  71 get_impl(
  72     XferDestBuffer *self,
  73     gpointer *buf,
  74     gsize *size)
  75 {
  76     if (size)
  77         *size = self->len;
  78
  79     if (buf)
  80         *buf = self->buf;
  81 }
  82
  83 static void
  84 push_buffer_impl(
  85     XferElement *elt,
  86     gpointer buf,
  87     size_t len)
  88 {
  89     XferDestBuffer *self = (XferDestBuffer *)elt;
  90
  91     if (!buf)
  92         return;
  93
  94     /* make sure this isn't too much data */
  95     if (self->max_size && self->len + len > self->max_size) {
  96         xfer_cancel_with_error(elt,
  97             _("illegal attempt to transfer more than %zd bytes"), self->max_size);
  98         wait_until_xfer_cancelled(elt->xfer);
  99         amfree(buf);
 100         return;
 101     }
 102
 103     /* expand the buffer if necessary */
 104     if (self->len + len > self->allocated) {
 105         gsize new_size = self->allocated * 2;
 106         if (new_size < self->len+len)
 107             new_size = self->len+len;
 108         if (self->max_size && new_size > self->max_size)
 109             new_size = self->max_size;
 110
 111         self->buf = g_realloc(self->buf, new_size);
 112         self->allocated = new_size;
 113     }
 114
 115     g_memmove(((guint8 *)self->buf)+self->len, buf, len);
 116     self->len += len;
 117
 118     amfree(buf);
 119 }
 120
 121 static void
 122 finalize_impl(
 123     GObject * obj_self)
 124 {
 125     XferDestBuffer *self = XFER_DEST_BUFFER(obj_self);
 126
 127     if (self->buf)
 128         g_free(self->buf);
 129     self->buf = NULL;
 130
 131     /* chain up */
 132     G_OBJECT_CLASS(parent_class)->finalize(obj_self);
 133 }
 134
 135 static void
 136 class_init(
 137     XferDestBufferClass * selfc)
 138 {
 139     XferElementClass *klass = XFER_ELEMENT_CLASS(selfc);
 140     GObjectClass *goc = G_OBJECT_CLASS(selfc);
 141     static xfer_element_mech_pair_t mech_pairs[] = {
 142         { XFER_MECH_PUSH_BUFFER, XFER_MECH_NONE, 0, 0},
 143         { XFER_MECH_NONE, XFER_MECH_NONE, 0, 0},
 144     };
 145
 146     selfc->get = get_impl;
 147     klass->push_buffer = push_buffer_impl;
 148     goc->finalize = finalize_impl;
 149
 150     klass->perl_class = "Amanda::Xfer::Dest::Buffer";
 151     klass->mech_pairs = mech_pairs;
 152
 153     parent_class = g_type_class_peek_parent(selfc);
 154 }
 155
 156 GType
 157 xfer_dest_buffer_get_type (void)
 158 {
 159     static GType type = 0;
 160
 161     if G_UNLIKELY(type == 0) {
 162         static const GTypeInfo info = {
 163             sizeof (XferDestBufferClass),
 164             (GBaseInitFunc) NULL,
 165             (GBaseFinalizeFunc) NULL,
 166             (GClassInitFunc) class_init,
 167             (GClassFinalizeFunc) NULL,
 168             NULL /* class_data */,
 169             sizeof (XferDestBuffer),
 170             0 /* n_preallocs */,
 171             (GInstanceInitFunc) NULL,
 172             NULL
 173         };
 174
 175         type = g_type_register_static (XFER_ELEMENT_TYPE, "XferDestBuffer", &info, 0);
 176     }
 177
 178     return type;
 179 }
 180
 181 void
 182 xfer_dest_buffer_get(
 183     XferElement *elt,
 184     gpointer *buf,
 185     gsize *size)
 186 {
 187     XferDestBufferClass *klass;
 188     g_assert(IS_XFER_DEST_BUFFER(elt));
 189
 190     klass = XFER_DEST_BUFFER_GET_CLASS(elt);
 191     klass->get(XFER_DEST_BUFFER(elt), buf, size);
 192 }
 193
 194 /* create an element of this class; prototype is in xfer-element.h */
 195 XferElement *
 196 xfer_dest_buffer(
 197     gsize max_size)
 198 {
 199     XferDestBuffer *self = (XferDestBuffer *)g_object_new(XFER_DEST_BUFFER_TYPE, NULL);
 200     XferElement *elt = XFER_ELEMENT(self);
 201
 202     self->max_size = max_size;
 203
 204     return elt;
 205 }