gio/gmemoryinputstream.c

   1 /* GIO - GLib Input, Output and Streaming Library
   2  *
   3  * Copyright (C) 2006-2007 Red Hat, Inc.
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General
  16  * Public License along with this library; if not, write to the
  17  * Free Software Foundation, Inc., 59 Temple Place, Suite 330,
  18  * Boston, MA 02111-1307, USA.
  19  *
  20  * Author: Christian Kellner <gicmo@gnome.org>
  21  */
  22
  23 #include "config.h"
  24 #include "gmemoryinputstream.h"
  25 #include "gpollableinputstream.h"
  26 #include "ginputstream.h"
  27 #include "gseekable.h"
  28 #include "string.h"
  29 #include "gtask.h"
  30 #include "gioerror.h"
  31 #include "glibintl.h"
  32
  33
  34 /**
  35  * SECTION:gmemoryinputstream
  36  * @short_description: Streaming input operations on memory chunks
  37  * @include: gio/gio.h
  38  * @see_also: #GMemoryOutputStream
  39  *
  40  * #GMemoryInputStream is a class for using arbitrary
  41  * memory chunks as input for GIO streaming input operations.
  42  *
  43  * As of GLib 2.34, #GMemoryInputStream implements
  44  * #GPollableInputStream.
  45  */
  46
  47 struct _GMemoryInputStreamPrivate {
  48   GSList *chunks;
  49   gsize   len;
  50   gsize   pos;
  51 };
  52
  53 static gssize   g_memory_input_stream_read         (GInputStream         *stream,
  54                                                     void                 *buffer,
  55                                                     gsize                 count,
  56                                                     GCancellable         *cancellable,
  57                                                     GError              **error);
  58 static gssize   g_memory_input_stream_skip         (GInputStream         *stream,
  59                                                     gsize                 count,
  60                                                     GCancellable         *cancellable,
  61                                                     GError              **error);
  62 static gboolean g_memory_input_stream_close        (GInputStream         *stream,
  63                                                     GCancellable         *cancellable,
  64                                                     GError              **error);
  65 static void     g_memory_input_stream_skip_async   (GInputStream         *stream,
  66                                                     gsize                 count,
  67                                                     int                   io_priority,
  68                                                     GCancellable         *cancellabl,
  69                                                     GAsyncReadyCallback   callback,
  70                                                     gpointer              datae);
  71 static gssize   g_memory_input_stream_skip_finish  (GInputStream         *stream,
  72                                                     GAsyncResult         *result,
  73                                                     GError              **error);
  74 static void     g_memory_input_stream_close_async  (GInputStream         *stream,
  75                                                     int                   io_priority,
  76                                                     GCancellable         *cancellabl,
  77                                                     GAsyncReadyCallback   callback,
  78                                                     gpointer              data);
  79 static gboolean g_memory_input_stream_close_finish (GInputStream         *stream,
  80                                                     GAsyncResult         *result,
  81                                                     GError              **error);
  82
  83 static void     g_memory_input_stream_seekable_iface_init (GSeekableIface  *iface);
  84 static goffset  g_memory_input_stream_tell                (GSeekable       *seekable);
  85 static gboolean g_memory_input_stream_can_seek            (GSeekable       *seekable);
  86 static gboolean g_memory_input_stream_seek                (GSeekable       *seekable,
  87                                                            goffset          offset,
  88                                                            GSeekType        type,
  89                                                            GCancellable    *cancellable,
  90                                                            GError         **error);
  91 static gboolean g_memory_input_stream_can_truncate        (GSeekable       *seekable);
  92 static gboolean g_memory_input_stream_truncate            (GSeekable       *seekable,
  93                                                            goffset          offset,
  94                                                            GCancellable    *cancellable,
  95                                                            GError         **error);
  96
  97 static void     g_memory_input_stream_pollable_iface_init (GPollableInputStreamInterface *iface);
  98 static gboolean g_memory_input_stream_is_readable         (GPollableInputStream *stream);
  99 static GSource *g_memory_input_stream_create_source       (GPollableInputStream *stream,
 100                                                            GCancellable          *cancellable);
 101
 102 static void     g_memory_input_stream_finalize            (GObject         *object);
 103
 104 G_DEFINE_TYPE_WITH_CODE (GMemoryInputStream, g_memory_input_stream, G_TYPE_INPUT_STREAM,
 105                          G_ADD_PRIVATE (GMemoryInputStream)
 106                          G_IMPLEMENT_INTERFACE (G_TYPE_SEEKABLE,
 107                                                 g_memory_input_stream_seekable_iface_init);
 108                          G_IMPLEMENT_INTERFACE (G_TYPE_POLLABLE_INPUT_STREAM,
 109                                                 g_memory_input_stream_pollable_iface_init);
 110                          )
 111
 112
 113 static void
 114 g_memory_input_stream_class_init (GMemoryInputStreamClass *klass)
 115 {
 116   GObjectClass *object_class;
 117   GInputStreamClass *istream_class;
 118
 119   object_class = G_OBJECT_CLASS (klass);
 120   object_class->finalize     = g_memory_input_stream_finalize;
 121
 122   istream_class = G_INPUT_STREAM_CLASS (klass);
 123   istream_class->read_fn  = g_memory_input_stream_read;
 124   istream_class->skip  = g_memory_input_stream_skip;
 125   istream_class->close_fn = g_memory_input_stream_close;
 126
 127   istream_class->skip_async  = g_memory_input_stream_skip_async;
 128   istream_class->skip_finish  = g_memory_input_stream_skip_finish;
 129   istream_class->close_async = g_memory_input_stream_close_async;
 130   istream_class->close_finish = g_memory_input_stream_close_finish;
 131 }
 132
 133 static void
 134 g_memory_input_stream_finalize (GObject *object)
 135 {
 136   GMemoryInputStream        *stream;
 137   GMemoryInputStreamPrivate *priv;
 138
 139   stream = G_MEMORY_INPUT_STREAM (object);
 140   priv = stream->priv;
 141
 142   g_slist_free_full (priv->chunks, (GDestroyNotify)g_bytes_unref);
 143
 144   G_OBJECT_CLASS (g_memory_input_stream_parent_class)->finalize (object);
 145 }
 146
 147 static void
 148 g_memory_input_stream_seekable_iface_init (GSeekableIface *iface)
 149 {
 150   iface->tell         = g_memory_input_stream_tell;
 151   iface->can_seek     = g_memory_input_stream_can_seek;
 152   iface->seek         = g_memory_input_stream_seek;
 153   iface->can_truncate = g_memory_input_stream_can_truncate;
 154   iface->truncate_fn  = g_memory_input_stream_truncate;
 155 }
 156
 157 static void
 158 g_memory_input_stream_pollable_iface_init (GPollableInputStreamInterface *iface)
 159 {
 160   iface->is_readable   = g_memory_input_stream_is_readable;
 161   iface->create_source = g_memory_input_stream_create_source;
 162 }
 163
 164 static void
 165 g_memory_input_stream_init (GMemoryInputStream *stream)
 166 {
 167   stream->priv = g_memory_input_stream_get_private (stream);
 168 }
 169
 170 /**
 171  * g_memory_input_stream_new:
 172  *
 173  * Creates a new empty #GMemoryInputStream.
 174  *
 175  * Returns: a new #GInputStream
 176  */
 177 GInputStream *
 178 g_memory_input_stream_new (void)
 179 {
 180   GInputStream *stream;
 181
 182   stream = g_object_new (G_TYPE_MEMORY_INPUT_STREAM, NULL);
 183
 184   return stream;
 185 }
 186
 187 /**
 188  * g_memory_input_stream_new_from_data:
 189  * @data: (array length=len) (element-type guint8) (transfer full): input data
 190  * @len: length of the data, may be -1 if @data is a nul-terminated string
 191  * @destroy: (allow-none): function that is called to free @data, or %NULL
 192  *
 193  * Creates a new #GMemoryInputStream with data in memory of a given size.
 194  *
 195  * Returns: new #GInputStream read from @data of @len bytes.
 196  **/
 197 GInputStream *
 198 g_memory_input_stream_new_from_data (const void     *data,
 199                                      gssize          len,
 200                                      GDestroyNotify  destroy)
 201 {
 202   GInputStream *stream;
 203
 204   stream = g_memory_input_stream_new ();
 205
 206   g_memory_input_stream_add_data (G_MEMORY_INPUT_STREAM (stream),
 207                                   data, len, destroy);
 208
 209   return stream;
 210 }
 211
 212 /**
 213  * g_memory_input_stream_new_from_bytes:
 214  * @bytes: a #GBytes
 215  *
 216  * Creates a new #GMemoryInputStream with data from the given @bytes.
 217  *
 218  * Returns: new #GInputStream read from @bytes
 219  *
 220  * Since: 2.34
 221  **/
 222 GInputStream *
 223 g_memory_input_stream_new_from_bytes (GBytes  *bytes)
 224 {
 225
 226   GInputStream *stream;
 227
 228   stream = g_memory_input_stream_new ();
 229
 230   g_memory_input_stream_add_bytes (G_MEMORY_INPUT_STREAM (stream),
 231                                    bytes);
 232
 233   return stream;
 234 }
 235
 236 /**
 237  * g_memory_input_stream_add_data:
 238  * @stream: a #GMemoryInputStream
 239  * @data: (array length=len) (element-type guint8) (transfer full): input data
 240  * @len: length of the data, may be -1 if @data is a nul-terminated string
 241  * @destroy: (allow-none): function that is called to free @data, or %NULL
 242  *
 243  * Appends @data to data that can be read from the input stream
 244  */
 245 void
 246 g_memory_input_stream_add_data (GMemoryInputStream *stream,
 247                                 const void         *data,
 248                                 gssize              len,
 249                                 GDestroyNotify      destroy)
 250 {
 251   GBytes *bytes;
 252
 253   if (len == -1)
 254     len = strlen (data);
 255
 256   /* It's safe to discard the const here because we're chaining the
 257    * destroy callback.
 258    */
 259   bytes = g_bytes_new_with_free_func (data, len, destroy, (void*)data);
 260
 261   g_memory_input_stream_add_bytes (stream, bytes);
 262
 263   g_bytes_unref (bytes);
 264 }
 265
 266 /**
 267  * g_memory_input_stream_add_bytes:
 268  * @stream: a #GMemoryInputStream
 269  * @bytes: input data
 270  *
 271  * Appends @bytes to data that can be read from the input stream.
 272  *
 273  * Since: 2.34
 274  */
 275 void
 276 g_memory_input_stream_add_bytes (GMemoryInputStream *stream,
 277                                  GBytes             *bytes)
 278 {
 279   GMemoryInputStreamPrivate *priv;
 280
 281   g_return_if_fail (G_IS_MEMORY_INPUT_STREAM (stream));
 282   g_return_if_fail (bytes != NULL);
 283
 284   priv = stream->priv;
 285
 286   priv->chunks = g_slist_append (priv->chunks, g_bytes_ref (bytes));
 287   priv->len += g_bytes_get_size (bytes);
 288 }
 289
 290 static gssize
 291 g_memory_input_stream_read (GInputStream  *stream,
 292                             void          *buffer,
 293                             gsize          count,
 294                             GCancellable  *cancellable,
 295                             GError       **error)
 296 {
 297   GMemoryInputStream *memory_stream;
 298   GMemoryInputStreamPrivate *priv;
 299   GSList *l;
 300   GBytes *chunk;
 301   gsize len;
 302   gsize offset, start, rest, size;
 303
 304   memory_stream = G_MEMORY_INPUT_STREAM (stream);
 305   priv = memory_stream->priv;
 306
 307   count = MIN (count, priv->len - priv->pos);
 308
 309   offset = 0;
 310   for (l = priv->chunks; l; l = l->next)
 311     {
 312       chunk = (GBytes *)l->data;
 313       len = g_bytes_get_size (chunk);
 314
 315       if (offset + len > priv->pos)
 316         break;
 317
 318       offset += len;
 319     }
 320
 321   start = priv->pos - offset;
 322   rest = count;
 323
 324   for (; l && rest > 0; l = l->next)
 325     {
 326       const guint8* chunk_data;
 327       chunk = (GBytes *)l->data;
 328
 329       chunk_data = g_bytes_get_data (chunk, &len);
 330
 331       size = MIN (rest, len - start);
 332
 333       memcpy ((guint8 *)buffer + (count - rest), chunk_data + start, size);
 334       rest -= size;
 335
 336       start = 0;
 337     }
 338
 339   priv->pos += count;
 340
 341   return count;
 342 }
 343
 344 static gssize
 345 g_memory_input_stream_skip (GInputStream  *stream,
 346                             gsize          count,
 347                             GCancellable  *cancellable,
 348                             GError       **error)
 349 {
 350   GMemoryInputStream *memory_stream;
 351   GMemoryInputStreamPrivate *priv;
 352
 353   memory_stream = G_MEMORY_INPUT_STREAM (stream);
 354   priv = memory_stream->priv;
 355
 356   count = MIN (count, priv->len - priv->pos);
 357   priv->pos += count;
 358
 359   return count;
 360 }
 361
 362 static gboolean
 363 g_memory_input_stream_close (GInputStream  *stream,
 364                              GCancellable  *cancellable,
 365                              GError       **error)
 366 {
 367   return TRUE;
 368 }
 369
 370 static void
 371 g_memory_input_stream_skip_async (GInputStream        *stream,
 372                                   gsize                count,
 373                                   int                  io_priority,
 374                                   GCancellable        *cancellable,
 375                                   GAsyncReadyCallback  callback,
 376                                   gpointer             user_data)
 377 {
 378   GTask *task;
 379   gssize nskipped;
 380   GError *error = NULL;
 381
 382   nskipped = G_INPUT_STREAM_GET_CLASS (stream)->skip (stream, count, cancellable, &error);
 383   task = g_task_new (stream, cancellable, callback, user_data);
 384   if (error)
 385     g_task_return_error (task, error);
 386   else
 387     g_task_return_int (task, nskipped);
 388   g_object_unref (task);
 389 }
 390
 391 static gssize
 392 g_memory_input_stream_skip_finish (GInputStream  *stream,
 393                                    GAsyncResult  *result,
 394                                    GError       **error)
 395 {
 396   g_return_val_if_fail (g_task_is_valid (result, stream), -1);
 397
 398   return g_task_propagate_int (G_TASK (result), error);
 399 }
 400
 401 static void
 402 g_memory_input_stream_close_async (GInputStream        *stream,
 403                                    int                  io_priority,
 404                                    GCancellable        *cancellable,
 405                                    GAsyncReadyCallback  callback,
 406                                    gpointer             user_data)
 407 {
 408   GTask *task;
 409
 410   task = g_task_new (stream, cancellable, callback, user_data);
 411   g_task_return_boolean (task, TRUE);
 412   g_object_unref (task);
 413 }
 414
 415 static gboolean
 416 g_memory_input_stream_close_finish (GInputStream  *stream,
 417                                     GAsyncResult  *result,
 418                                     GError       **error)
 419 {
 420   return TRUE;
 421 }
 422
 423 static goffset
 424 g_memory_input_stream_tell (GSeekable *seekable)
 425 {
 426   GMemoryInputStream *memory_stream;
 427   GMemoryInputStreamPrivate *priv;
 428
 429   memory_stream = G_MEMORY_INPUT_STREAM (seekable);
 430   priv = memory_stream->priv;
 431
 432   return priv->pos;
 433 }
 434
 435 static
 436 gboolean g_memory_input_stream_can_seek (GSeekable *seekable)
 437 {
 438   return TRUE;
 439 }
 440
 441 static gboolean
 442 g_memory_input_stream_seek (GSeekable     *seekable,
 443                             goffset        offset,
 444                             GSeekType      type,
 445                             GCancellable  *cancellable,
 446                             GError       **error)
 447 {
 448   GMemoryInputStream *memory_stream;
 449   GMemoryInputStreamPrivate *priv;
 450   goffset absolute;
 451
 452   memory_stream = G_MEMORY_INPUT_STREAM (seekable);
 453   priv = memory_stream->priv;
 454
 455   switch (type)
 456     {
 457     case G_SEEK_CUR:
 458       absolute = priv->pos + offset;
 459       break;
 460
 461     case G_SEEK_SET:
 462       absolute = offset;
 463       break;
 464
 465     case G_SEEK_END:
 466       absolute = priv->len + offset;
 467       break;
 468
 469     default:
 470       g_set_error_literal (error,
 471                            G_IO_ERROR,
 472                            G_IO_ERROR_INVALID_ARGUMENT,
 473                            _("Invalid GSeekType supplied"));
 474
 475       return FALSE;
 476     }
 477
 478   if (absolute < 0 || absolute > priv->len)
 479     {
 480       g_set_error_literal (error,
 481                            G_IO_ERROR,
 482                            G_IO_ERROR_INVALID_ARGUMENT,
 483                            _("Invalid seek request"));
 484       return FALSE;
 485     }
 486
 487   priv->pos = absolute;
 488
 489   return TRUE;
 490 }
 491
 492 static gboolean
 493 g_memory_input_stream_can_truncate (GSeekable *seekable)
 494 {
 495   return FALSE;
 496 }
 497
 498 static gboolean
 499 g_memory_input_stream_truncate (GSeekable     *seekable,
 500                                 goffset        offset,
 501                                 GCancellable  *cancellable,
 502                                 GError       **error)
 503 {
 504   g_set_error_literal (error,
 505                        G_IO_ERROR,
 506                        G_IO_ERROR_NOT_SUPPORTED,
 507                        _("Cannot truncate GMemoryInputStream"));
 508   return FALSE;
 509 }
 510
 511 static gboolean
 512 g_memory_input_stream_is_readable (GPollableInputStream *stream)
 513 {
 514   return TRUE;
 515 }
 516
 517 static GSource *
 518 g_memory_input_stream_create_source (GPollableInputStream *stream,
 519                                      GCancellable         *cancellable)
 520 {
 521   GSource *base_source, *pollable_source;
 522
 523   base_source = g_timeout_source_new (0);
 524   pollable_source = g_pollable_source_new_full (stream, base_source,
 525                                                 cancellable);
 526   g_source_unref (base_source);
 527
 528   return pollable_source;
 529 }