gio/win32/gwinhttpfileoutputstream.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 "gwinhttpfileoutputstream.h"
  32 #include "glibintl.h"
  33
  34 #include "gioalias.h"
  35
  36 struct _GWinHttpFileOutputStream
  37 {
  38   GFileOutputStream parent_instance;
  39
  40   GWinHttpFile *file;
  41   HINTERNET connection;
  42   goffset offset;
  43 };
  44
  45 struct _GWinHttpFileOutputStreamClass
  46 {
  47   GFileOutputStreamClass parent_class;
  48 };
  49
  50 #define g_winhttp_file_output_stream_get_type _g_winhttp_file_output_stream_get_type
  51 G_DEFINE_TYPE (GWinHttpFileOutputStream, g_winhttp_file_output_stream, G_TYPE_FILE_OUTPUT_STREAM);
  52
  53 static gssize     g_winhttp_file_output_stream_write      (GOutputStream     *stream,
  54                                                            const void        *buffer,
  55                                                            gsize              count,
  56                                                            GCancellable      *cancellable,
  57                                                            GError           **error);
  58
  59 static void
  60 g_winhttp_file_output_stream_finalize (GObject *object)
  61 {
  62   GWinHttpFileOutputStream *winhttp_stream;
  63
  64   winhttp_stream = G_WINHTTP_FILE_OUTPUT_STREAM (object);
  65
  66   if (winhttp_stream->connection != NULL)
  67     G_WINHTTP_VFS_GET_CLASS (winhttp_stream->file->vfs)->pWinHttpCloseHandle (winhttp_stream->connection);
  68
  69   G_OBJECT_CLASS (g_winhttp_file_output_stream_parent_class)->finalize (object);
  70 }
  71
  72 static void
  73 g_winhttp_file_output_stream_class_init (GWinHttpFileOutputStreamClass *klass)
  74 {
  75   GObjectClass *gobject_class = G_OBJECT_CLASS (klass);
  76   GOutputStreamClass *stream_class = G_OUTPUT_STREAM_CLASS (klass);
  77
  78   gobject_class->finalize = g_winhttp_file_output_stream_finalize;
  79
  80   stream_class->write_fn = g_winhttp_file_output_stream_write;
  81 }
  82
  83 static void
  84 g_winhttp_file_output_stream_init (GWinHttpFileOutputStream *info)
  85 {
  86 }
  87
  88 /**
  89  * g_winhttp_file_output_stream_new:
  90  * @file: the GWinHttpFile being read
  91  * @connection: handle to the HTTP connection, as from WinHttpConnect()
  92  * @request: handle to the HTTP request, as from WinHttpOpenRequest
  93  *
  94  * Returns: #GFileOutputStream for the given request
  95  **/
  96 GFileOutputStream *
  97 _g_winhttp_file_output_stream_new (GWinHttpFile *file,
  98                                    HINTERNET     connection)
  99 {
 100   GWinHttpFileOutputStream *stream;
 101
 102   stream = g_object_new (G_TYPE_WINHTTP_FILE_OUTPUT_STREAM, NULL);
 103
 104   stream->file = file;
 105   stream->connection = connection;
 106   stream->offset = 0;
 107
 108   return G_FILE_OUTPUT_STREAM (stream);
 109 }
 110
 111 static gssize
 112 g_winhttp_file_output_stream_write (GOutputStream  *stream,
 113                                     const void     *buffer,
 114                                     gsize           count,
 115                                     GCancellable   *cancellable,
 116                                     GError        **error)
 117 {
 118   GWinHttpFileOutputStream *winhttp_stream = G_WINHTTP_FILE_OUTPUT_STREAM (stream);
 119   HINTERNET request;
 120   char *headers;
 121   wchar_t *wheaders;
 122   DWORD bytes_written;
 123
 124   request = G_WINHTTP_VFS_GET_CLASS (winhttp_stream->file->vfs)->pWinHttpOpenRequest
 125     (winhttp_stream->connection,
 126      L"PUT",
 127      winhttp_stream->file->url.lpszUrlPath,
 128      NULL,
 129      WINHTTP_NO_REFERER,
 130      NULL,
 131      winhttp_stream->file->url.nScheme == INTERNET_SCHEME_HTTPS ? WINHTTP_FLAG_SECURE : 0);
 132
 133   if (request == NULL)
 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   headers = g_strdup_printf ("Content-Range: bytes %" G_GINT64_FORMAT "-%" G_GINT64_FORMAT "/*\r\n",
 146                              winhttp_stream->offset, winhttp_stream->offset + count);
 147   wheaders = g_utf8_to_utf16 (headers, -1, NULL, NULL, NULL);
 148   g_free (headers);
 149
 150   if (!G_WINHTTP_VFS_GET_CLASS (winhttp_stream->file->vfs)->pWinHttpSendRequest
 151       (request,
 152        wheaders, -1,
 153        NULL, 0,
 154        count,
 155        0))
 156     {
 157       char *emsg = _g_winhttp_error_message (GetLastError ());
 158
 159       g_set_error (error, G_IO_ERROR,
 160                    G_IO_ERROR_FAILED,
 161                    "%s", emsg);
 162       g_free (emsg);
 163
 164       G_WINHTTP_VFS_GET_CLASS (winhttp_stream->file->vfs)->pWinHttpCloseHandle (request);
 165       g_free (wheaders);
 166
 167       return -1;
 168     }
 169
 170   g_free (wheaders);
 171
 172   if (!G_WINHTTP_VFS_GET_CLASS (winhttp_stream->file->vfs)->pWinHttpWriteData
 173       (request, buffer, count, &bytes_written))
 174     {
 175       char *emsg = _g_winhttp_error_message (GetLastError ());
 176       g_set_error (error, G_IO_ERROR, G_IO_ERROR_FAILED,
 177                    "%s",
 178                    emsg);
 179       g_free (emsg);
 180
 181       G_WINHTTP_VFS_GET_CLASS (winhttp_stream->file->vfs)->pWinHttpCloseHandle (request);
 182
 183       return -1;
 184     }
 185
 186   winhttp_stream->offset += bytes_written;
 187
 188   if (!G_WINHTTP_VFS_GET_CLASS (winhttp_stream->file->vfs)->pWinHttpReceiveResponse
 189       (request, NULL))
 190     {
 191       char *emsg = _g_winhttp_error_message (GetLastError ());
 192       g_set_error (error, G_IO_ERROR, G_IO_ERROR_FAILED,
 193                    "%s",
 194                    emsg);
 195       g_free (emsg);
 196
 197       G_WINHTTP_VFS_GET_CLASS (winhttp_stream->file->vfs)->pWinHttpCloseHandle (request);
 198       return -1;
 199     }
 200
 201   G_WINHTTP_VFS_GET_CLASS (winhttp_stream->file->vfs)->pWinHttpCloseHandle (request);
 202
 203   return bytes_written;
 204 }