gio/win32/gwinhttpfileinputstream.c

   1 /* GIO - GLib Input, Output and Streaming Library
   2  *
   3  * Copyright (C) 2006-2007 Red Hat, Inc.
   4  * Copyright (C) 2008 Novell, Inc.
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General
  17  * Public License along with this library; if not, write to the
  18  * Free Software Foundation, Inc., 59 Temple Place, Suite 330,
  19  * Boston, MA 02111-1307, USA.
  20  *
  21  * Author: Alexander Larsson <alexl@redhat.com>
  22  * Author: Tor Lillqvist <tml@novell.com>
  23  */
  24
  25 #include "config.h"
  26
  27 #include <glib.h>
  28
  29 #include "gcancellable.h"
  30 #include "gioerror.h"
  31 #include "gwinhttpfileinputstream.h"
  32 #include "glibintl.h"
  33
  34 #include "gioalias.h"
  35
  36 struct _GWinHttpFileInputStream
  37 {
  38   GFileInputStream parent_instance;
  39
  40   GWinHttpFile *file;
  41   gboolean request_sent;
  42   HINTERNET connection;
  43   HINTERNET request;
  44 };
  45
  46 struct _GWinHttpFileInputStreamClass
  47 {
  48   GFileInputStreamClass parent_class;
  49 };
  50
  51 #define g_winhttp_file_input_stream_get_type _g_winhttp_file_input_stream_get_type
  52 G_DEFINE_TYPE (GWinHttpFileInputStream, g_winhttp_file_input_stream, G_TYPE_FILE_INPUT_STREAM);
  53
  54 static gssize g_winhttp_file_input_stream_read (GInputStream    *stream,
  55                                                 void            *buffer,
  56                                                 gsize            count,
  57                                                 GCancellable    *cancellable,
  58                                                 GError         **error);
  59
  60 static void
  61 g_winhttp_file_input_stream_finalize (GObject *object)
  62 {
  63   GWinHttpFileInputStream *winhttp_stream;
  64
  65   winhttp_stream = G_WINHTTP_FILE_INPUT_STREAM (object);
  66
  67   if (winhttp_stream->request != NULL)
  68     G_WINHTTP_VFS_GET_CLASS (winhttp_stream->file->vfs)->pWinHttpCloseHandle (winhttp_stream->request);
  69   if (winhttp_stream->connection != NULL)
  70     G_WINHTTP_VFS_GET_CLASS (winhttp_stream->file->vfs)->pWinHttpCloseHandle (winhttp_stream->connection);
  71
  72   G_OBJECT_CLASS (g_winhttp_file_input_stream_parent_class)->finalize (object);
  73 }
  74
  75 static void
  76 g_winhttp_file_input_stream_class_init (GWinHttpFileInputStreamClass *klass)
  77 {
  78   GObjectClass *gobject_class = G_OBJECT_CLASS (klass);
  79   GInputStreamClass *stream_class = G_INPUT_STREAM_CLASS (klass);
  80
  81   gobject_class->finalize = g_winhttp_file_input_stream_finalize;
  82
  83   stream_class->read_fn = g_winhttp_file_input_stream_read;
  84 }
  85
  86 static void
  87 g_winhttp_file_input_stream_init (GWinHttpFileInputStream *info)
  88 {
  89 }
  90
  91 /**
  92  * g_winhttp_file_input_stream_new:
  93  * @file: the GWinHttpFile being read
  94  * @connection: handle to the HTTP connection, as from WinHttpConnect()
  95  * @request: handle to the HTTP request, as from WinHttpOpenRequest
  96  *
  97  * Returns: #GFileInputStream for the given request
  98  **/
  99 GFileInputStream *
 100 _g_winhttp_file_input_stream_new (GWinHttpFile *file,
 101                                   HINTERNET     connection,
 102                                   HINTERNET     request)
 103 {
 104   GWinHttpFileInputStream *stream;
 105
 106   stream = g_object_new (G_TYPE_WINHTTP_FILE_INPUT_STREAM, NULL);
 107
 108   stream->file = file;
 109   stream->request_sent = FALSE;
 110   stream->connection = connection;
 111   stream->request = request;
 112
 113   return G_FILE_INPUT_STREAM (stream);
 114 }
 115
 116 static gssize
 117 g_winhttp_file_input_stream_read (GInputStream  *stream,
 118                                   void          *buffer,
 119                                   gsize          count,
 120                                   GCancellable  *cancellable,
 121                                   GError       **error)
 122 {
 123   GWinHttpFileInputStream *winhttp_stream = G_WINHTTP_FILE_INPUT_STREAM (stream);
 124   DWORD bytes_read;
 125
 126   if (!winhttp_stream->request_sent)
 127     {
 128       if (!G_WINHTTP_VFS_GET_CLASS (winhttp_stream->file->vfs)->pWinHttpSendRequest
 129           (winhttp_stream->request,
 130            NULL, 0,
 131            NULL, 0,
 132            0,
 133            0))
 134         {
 135           char *emsg = _g_winhttp_error_message (GetLastError ());
 136
 137           g_set_error (error, G_IO_ERROR,
 138                        G_IO_ERROR_FAILED,
 139                        "%s", emsg);
 140           g_free (emsg);
 141
 142           return -1;
 143         }
 144
 145       if (!G_WINHTTP_VFS_GET_CLASS (winhttp_stream->file->vfs)->pWinHttpReceiveResponse
 146           (winhttp_stream->request, NULL))
 147         {
 148           char *emsg = _g_winhttp_error_message (GetLastError ());
 149           g_set_error (error, G_IO_ERROR, G_IO_ERROR_FAILED,
 150                        "%s", emsg);
 151           g_free (emsg);
 152
 153           return -1;
 154         }
 155       winhttp_stream->request_sent = TRUE;
 156     }
 157
 158   if (!G_WINHTTP_VFS_GET_CLASS (winhttp_stream->file->vfs)->pWinHttpReadData
 159       (winhttp_stream->request, buffer, count, &bytes_read))
 160     {
 161       char *emsg = _g_winhttp_error_message (GetLastError ());
 162       g_set_error (error, G_IO_ERROR, G_IO_ERROR_FAILED,
 163                    "%s", emsg);
 164       g_free (emsg);
 165
 166       return -1;
 167     }
 168
 169   return bytes_read;
 170 }