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
g_winhttp_file_output_stream_finalize(GObject * object)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)->funcs->pWinHttpCloseHandle (winhttp_stream->connection);
68
69 G_OBJECT_CLASS (g_winhttp_file_output_stream_parent_class)->finalize (object);
70 }
71
72 static void
g_winhttp_file_output_stream_class_init(GWinHttpFileOutputStreamClass * klass)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
g_winhttp_file_output_stream_init(GWinHttpFileOutputStream * info)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 *
_g_winhttp_file_output_stream_new(GWinHttpFile * file,HINTERNET connection)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
g_winhttp_file_output_stream_write(GOutputStream * stream,const void * buffer,gsize count,GCancellable * cancellable,GError ** error)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)->funcs->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 _g_winhttp_set_error (error, GetLastError (), "PUT request");
136
137 return -1;
138 }
139
140 headers = g_strdup_printf ("Content-Range: bytes %" G_GINT64_FORMAT "-%" G_GINT64_FORMAT "/*\r\n",
141 winhttp_stream->offset, winhttp_stream->offset + count);
142 wheaders = g_utf8_to_utf16 (headers, -1, NULL, NULL, NULL);
143 g_free (headers);
144
145 if (!G_WINHTTP_VFS_GET_CLASS (winhttp_stream->file->vfs)->funcs->pWinHttpSendRequest
146 (request,
147 wheaders, -1,
148 NULL, 0,
149 count,
150 0))
151 {
152 _g_winhttp_set_error (error, GetLastError (), "PUT request");
153
154 G_WINHTTP_VFS_GET_CLASS (winhttp_stream->file->vfs)->funcs->pWinHttpCloseHandle (request);
155 g_free (wheaders);
156
157 return -1;
158 }
159
160 g_free (wheaders);
161
162 if (!G_WINHTTP_VFS_GET_CLASS (winhttp_stream->file->vfs)->funcs->pWinHttpWriteData
163 (request, buffer, count, &bytes_written))
164 {
165 _g_winhttp_set_error (error, GetLastError (), "PUT request");
166
167 G_WINHTTP_VFS_GET_CLASS (winhttp_stream->file->vfs)->funcs->pWinHttpCloseHandle (request);
168
169 return -1;
170 }
171
172 winhttp_stream->offset += bytes_written;
173
174 if (!_g_winhttp_response (winhttp_stream->file->vfs,
175 request,
176 error,
177 "PUT request"))
178 {
179 G_WINHTTP_VFS_GET_CLASS (winhttp_stream->file->vfs)->funcs->pWinHttpCloseHandle (request);
180
181 return -1;
182 }
183
184 G_WINHTTP_VFS_GET_CLASS (winhttp_stream->file->vfs)->funcs->pWinHttpCloseHandle (request);
185
186 return bytes_written;
187 }
188