Coverage Report

Created: 2025-06-13 06:55

/src/glib/gio/gpollableoutputstream.c
Line
Count
Source (jump to first uncovered line)
1
/* GIO - GLib Input, Output and Streaming Library
2
 *
3
 * Copyright (C) 2010 Red Hat, Inc.
4
 *
5
 * SPDX-License-Identifier: LGPL-2.1-or-later
6
 *
7
 * This library is free software; you can redistribute it and/or
8
 * modify it under the terms of the GNU Lesser General Public
9
 * License as published by the Free Software Foundation; either
10
 * version 2.1 of the License, or (at your option) any later version.
11
 *
12
 * This library is distributed in the hope that it will be useful,
13
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15
 * Lesser General Public License for more details.
16
 *
17
 * You should have received a copy of the GNU Lesser General
18
 * Public License along with this library; if not, see <http://www.gnu.org/licenses/>.
19
 */
20
21
#include "config.h"
22
23
#include <errno.h>
24
25
#include "gpollableoutputstream.h"
26
#include "gasynchelper.h"
27
#include "gfiledescriptorbased.h"
28
#include "glibintl.h"
29
30
/**
31
 * SECTION:gpollableoutputstream
32
 * @short_description: Interface for pollable output streams
33
 * @include: gio/gio.h
34
 * @see_also: #GOutputStream, #GFileDescriptorBased, #GPollableInputStream
35
 *
36
 * #GPollableOutputStream is implemented by #GOutputStreams that
37
 * can be polled for readiness to write. This can be used when
38
 * interfacing with a non-GIO API that expects
39
 * UNIX-file-descriptor-style asynchronous I/O rather than GIO-style.
40
 *
41
 * Some classes may implement #GPollableOutputStream but have only certain
42
 * instances of that class be pollable. If g_pollable_output_stream_can_poll()
43
 * returns %FALSE, then the behavior of other #GPollableOutputStream methods is
44
 * undefined.
45
 *
46
 * Since: 2.28
47
 */
48
49
G_DEFINE_INTERFACE (GPollableOutputStream, g_pollable_output_stream, G_TYPE_OUTPUT_STREAM)
50
51
static gboolean g_pollable_output_stream_default_can_poll           (GPollableOutputStream *stream);
52
static gssize   g_pollable_output_stream_default_write_nonblocking  (GPollableOutputStream  *stream,
53
                     const void             *buffer,
54
                     gsize                   count,
55
                     GError                **error);
56
static GPollableReturn g_pollable_output_stream_default_writev_nonblocking (GPollableOutputStream  *stream,
57
                      const GOutputVector    *vectors,
58
                      gsize                   n_vectors,
59
                      gsize                  *bytes_written,
60
                      GError                **error);
61
62
static void
63
g_pollable_output_stream_default_init (GPollableOutputStreamInterface *iface)
64
1
{
65
1
  iface->can_poll           = g_pollable_output_stream_default_can_poll;
66
1
  iface->write_nonblocking  = g_pollable_output_stream_default_write_nonblocking;
67
1
  iface->writev_nonblocking = g_pollable_output_stream_default_writev_nonblocking;
68
1
}
69
70
static gboolean
71
g_pollable_output_stream_default_can_poll (GPollableOutputStream *stream)
72
0
{
73
0
  return TRUE;
74
0
}
75
76
/**
77
 * g_pollable_output_stream_can_poll:
78
 * @stream: a #GPollableOutputStream.
79
 *
80
 * Checks if @stream is actually pollable. Some classes may implement
81
 * #GPollableOutputStream but have only certain instances of that
82
 * class be pollable. If this method returns %FALSE, then the behavior
83
 * of other #GPollableOutputStream methods is undefined.
84
 *
85
 * For any given stream, the value returned by this method is constant;
86
 * a stream cannot switch from pollable to non-pollable or vice versa.
87
 *
88
 * Returns: %TRUE if @stream is pollable, %FALSE if not.
89
 *
90
 * Since: 2.28
91
 */
92
gboolean
93
g_pollable_output_stream_can_poll (GPollableOutputStream *stream)
94
0
{
95
0
  g_return_val_if_fail (G_IS_POLLABLE_OUTPUT_STREAM (stream), FALSE);
96
97
0
  return G_POLLABLE_OUTPUT_STREAM_GET_INTERFACE (stream)->can_poll (stream);
98
0
}
99
100
/**
101
 * g_pollable_output_stream_is_writable:
102
 * @stream: a #GPollableOutputStream.
103
 *
104
 * Checks if @stream can be written.
105
 *
106
 * Note that some stream types may not be able to implement this 100%
107
 * reliably, and it is possible that a call to g_output_stream_write()
108
 * after this returns %TRUE would still block. To guarantee
109
 * non-blocking behavior, you should always use
110
 * g_pollable_output_stream_write_nonblocking(), which will return a
111
 * %G_IO_ERROR_WOULD_BLOCK error rather than blocking.
112
 *
113
 * The behaviour of this method is undefined if
114
 * g_pollable_output_stream_can_poll() returns %FALSE for @stream.
115
 *
116
 * Returns: %TRUE if @stream is writable, %FALSE if not. If an error
117
 *   has occurred on @stream, this will result in
118
 *   g_pollable_output_stream_is_writable() returning %TRUE, and the
119
 *   next attempt to write will return the error.
120
 *
121
 * Since: 2.28
122
 */
123
gboolean
124
g_pollable_output_stream_is_writable (GPollableOutputStream *stream)
125
0
{
126
0
  g_return_val_if_fail (G_IS_POLLABLE_OUTPUT_STREAM (stream), FALSE);
127
128
0
  return G_POLLABLE_OUTPUT_STREAM_GET_INTERFACE (stream)->is_writable (stream);
129
0
}
130
131
/**
132
 * g_pollable_output_stream_create_source:
133
 * @stream: a #GPollableOutputStream.
134
 * @cancellable: (nullable): a #GCancellable, or %NULL
135
 *
136
 * Creates a #GSource that triggers when @stream can be written, or
137
 * @cancellable is triggered or an error occurs. The callback on the
138
 * source is of the #GPollableSourceFunc type.
139
 *
140
 * As with g_pollable_output_stream_is_writable(), it is possible that
141
 * the stream may not actually be writable even after the source
142
 * triggers, so you should use g_pollable_output_stream_write_nonblocking()
143
 * rather than g_output_stream_write() from the callback.
144
 *
145
 * The behaviour of this method is undefined if
146
 * g_pollable_output_stream_can_poll() returns %FALSE for @stream.
147
 *
148
 * Returns: (transfer full): a new #GSource
149
 *
150
 * Since: 2.28
151
 */
152
GSource *
153
g_pollable_output_stream_create_source (GPollableOutputStream *stream,
154
          GCancellable          *cancellable)
155
0
{
156
0
  g_return_val_if_fail (G_IS_POLLABLE_OUTPUT_STREAM (stream), NULL);
157
158
0
  return G_POLLABLE_OUTPUT_STREAM_GET_INTERFACE (stream)->
159
0
    create_source (stream, cancellable);
160
0
}
161
162
static gssize
163
g_pollable_output_stream_default_write_nonblocking (GPollableOutputStream  *stream,
164
                const void             *buffer,
165
                gsize                   count,
166
                GError                **error)
167
0
{
168
0
  if (!g_pollable_output_stream_is_writable (stream))
169
0
    {
170
0
      g_set_error_literal (error, G_IO_ERROR, G_IO_ERROR_WOULD_BLOCK,
171
0
                           g_strerror (EAGAIN));
172
0
      return -1;
173
0
    }
174
175
0
  return G_OUTPUT_STREAM_GET_CLASS (stream)->
176
0
    write_fn (G_OUTPUT_STREAM (stream), buffer, count, NULL, error);
177
0
}
178
179
static GPollableReturn
180
g_pollable_output_stream_default_writev_nonblocking (GPollableOutputStream  *stream,
181
                 const GOutputVector    *vectors,
182
                 gsize                   n_vectors,
183
                 gsize                  *bytes_written,
184
                 GError                **error)
185
0
{
186
0
  gsize _bytes_written = 0;
187
0
  GPollableOutputStreamInterface *iface = G_POLLABLE_OUTPUT_STREAM_GET_INTERFACE (stream);
188
0
  gsize i;
189
0
  GError *err = NULL;
190
191
0
  for (i = 0; i < n_vectors; i++)
192
0
    {
193
0
      gssize res;
194
195
      /* Would we overflow here? In that case simply return and let the caller
196
       * handle this like a short write */
197
0
      if (_bytes_written > G_MAXSIZE - vectors[i].size)
198
0
        break;
199
200
0
      res = iface->write_nonblocking (stream, vectors[i].buffer, vectors[i].size, &err);
201
0
      if (res == -1)
202
0
        {
203
0
          if (bytes_written)
204
0
            *bytes_written = _bytes_written;
205
206
          /* If something was written already we handle this like a short
207
           * write and assume that the next call would either give the same
208
           * error again or successfully finish writing without errors or data
209
           * loss
210
           */
211
0
          if (_bytes_written > 0)
212
0
            {
213
0
              g_clear_error (&err);
214
0
              return G_POLLABLE_RETURN_OK;
215
0
            }
216
0
          else if (g_error_matches (err, G_IO_ERROR, G_IO_ERROR_WOULD_BLOCK))
217
0
            {
218
0
              g_clear_error (&err);
219
0
              return G_POLLABLE_RETURN_WOULD_BLOCK;
220
0
            }
221
0
          else
222
0
            {
223
0
              g_propagate_error (error, err);
224
0
              return G_POLLABLE_RETURN_FAILED;
225
0
            }
226
0
        }
227
228
0
      _bytes_written += res;
229
      /* if we had a short write break the loop here */
230
0
      if ((gsize) res < vectors[i].size)
231
0
        break;
232
0
    }
233
234
0
  if (bytes_written)
235
0
    *bytes_written = _bytes_written;
236
237
0
  return G_POLLABLE_RETURN_OK;
238
0
}
239
240
/**
241
 * g_pollable_output_stream_write_nonblocking:
242
 * @stream: a #GPollableOutputStream
243
 * @buffer: (array length=count) (element-type guint8): a buffer to write
244
 *     data from
245
 * @count: the number of bytes you want to write
246
 * @cancellable: (nullable): a #GCancellable, or %NULL
247
 * @error: #GError for error reporting, or %NULL to ignore.
248
 *
249
 * Attempts to write up to @count bytes from @buffer to @stream, as
250
 * with g_output_stream_write(). If @stream is not currently writable,
251
 * this will immediately return %G_IO_ERROR_WOULD_BLOCK, and you can
252
 * use g_pollable_output_stream_create_source() to create a #GSource
253
 * that will be triggered when @stream is writable.
254
 *
255
 * Note that since this method never blocks, you cannot actually
256
 * use @cancellable to cancel it. However, it will return an error
257
 * if @cancellable has already been cancelled when you call, which
258
 * may happen if you call this method after a source triggers due
259
 * to having been cancelled.
260
 *
261
 * Also note that if %G_IO_ERROR_WOULD_BLOCK is returned some underlying
262
 * transports like D/TLS require that you re-send the same @buffer and
263
 * @count in the next write call.
264
 *
265
 * The behaviour of this method is undefined if
266
 * g_pollable_output_stream_can_poll() returns %FALSE for @stream.
267
 *
268
 * Virtual: write_nonblocking
269
 * Returns: the number of bytes written, or -1 on error (including
270
 *   %G_IO_ERROR_WOULD_BLOCK).
271
 */
272
gssize
273
g_pollable_output_stream_write_nonblocking (GPollableOutputStream  *stream,
274
              const void             *buffer,
275
              gsize                   count,
276
              GCancellable           *cancellable,
277
              GError                **error)
278
0
{
279
0
  gssize res;
280
281
0
  g_return_val_if_fail (G_IS_POLLABLE_OUTPUT_STREAM (stream), -1);
282
0
  g_return_val_if_fail (buffer != NULL, 0);
283
284
0
  if (g_cancellable_set_error_if_cancelled (cancellable, error))
285
0
    return -1;
286
287
0
  if (count == 0)
288
0
    return 0;
289
290
0
  if (((gssize) count) < 0)
291
0
    {
292
0
      g_set_error (error, G_IO_ERROR, G_IO_ERROR_INVALID_ARGUMENT,
293
0
       _("Too large count value passed to %s"), G_STRFUNC);
294
0
      return -1;
295
0
    }
296
297
0
  if (cancellable)
298
0
    g_cancellable_push_current (cancellable);
299
300
0
  res = G_POLLABLE_OUTPUT_STREAM_GET_INTERFACE (stream)->
301
0
    write_nonblocking (stream, buffer, count, error);
302
303
0
  if (cancellable)
304
0
    g_cancellable_pop_current (cancellable);
305
306
0
  return res;
307
0
}
308
309
/**
310
 * g_pollable_output_stream_writev_nonblocking:
311
 * @stream: a #GPollableOutputStream
312
 * @vectors: (array length=n_vectors): the buffer containing the #GOutputVectors to write.
313
 * @n_vectors: the number of vectors to write
314
 * @bytes_written: (out) (optional): location to store the number of bytes that were
315
 *     written to the stream
316
 * @cancellable: (nullable): a #GCancellable, or %NULL
317
 * @error: #GError for error reporting, or %NULL to ignore.
318
 *
319
 * Attempts to write the bytes contained in the @n_vectors @vectors to @stream,
320
 * as with g_output_stream_writev(). If @stream is not currently writable,
321
 * this will immediately return %@G_POLLABLE_RETURN_WOULD_BLOCK, and you can
322
 * use g_pollable_output_stream_create_source() to create a #GSource
323
 * that will be triggered when @stream is writable. @error will *not* be
324
 * set in that case.
325
 *
326
 * Note that since this method never blocks, you cannot actually
327
 * use @cancellable to cancel it. However, it will return an error
328
 * if @cancellable has already been cancelled when you call, which
329
 * may happen if you call this method after a source triggers due
330
 * to having been cancelled.
331
 *
332
 * Also note that if %G_POLLABLE_RETURN_WOULD_BLOCK is returned some underlying
333
 * transports like D/TLS require that you re-send the same @vectors and
334
 * @n_vectors in the next write call.
335
 *
336
 * The behaviour of this method is undefined if
337
 * g_pollable_output_stream_can_poll() returns %FALSE for @stream.
338
 *
339
 * Virtual: writev_nonblocking
340
 *
341
 * Returns: %@G_POLLABLE_RETURN_OK on success, %G_POLLABLE_RETURN_WOULD_BLOCK
342
 * if the stream is not currently writable (and @error is *not* set), or
343
 * %G_POLLABLE_RETURN_FAILED if there was an error in which case @error will
344
 * be set.
345
 *
346
 * Since: 2.60
347
 */
348
GPollableReturn
349
g_pollable_output_stream_writev_nonblocking (GPollableOutputStream  *stream,
350
               const GOutputVector    *vectors,
351
               gsize                   n_vectors,
352
               gsize                  *bytes_written,
353
               GCancellable           *cancellable,
354
               GError                **error)
355
0
{
356
0
  GPollableOutputStreamInterface *iface;
357
0
  GPollableReturn res;
358
0
  gsize _bytes_written = 0;
359
360
0
  if (bytes_written)
361
0
    *bytes_written = 0;
362
363
0
  g_return_val_if_fail (G_IS_POLLABLE_OUTPUT_STREAM (stream), G_POLLABLE_RETURN_FAILED);
364
0
  g_return_val_if_fail (vectors != NULL || n_vectors == 0, G_POLLABLE_RETURN_FAILED);
365
0
  g_return_val_if_fail (cancellable == NULL || G_IS_CANCELLABLE (cancellable), G_POLLABLE_RETURN_FAILED);
366
0
  g_return_val_if_fail (error == NULL || *error == NULL, G_POLLABLE_RETURN_FAILED);
367
368
0
  if (g_cancellable_set_error_if_cancelled (cancellable, error))
369
0
    return G_POLLABLE_RETURN_FAILED;
370
371
0
  if (n_vectors == 0)
372
0
    return G_POLLABLE_RETURN_OK;
373
374
0
  iface = G_POLLABLE_OUTPUT_STREAM_GET_INTERFACE (stream);
375
0
  g_return_val_if_fail (iface->writev_nonblocking != NULL, G_POLLABLE_RETURN_FAILED);
376
377
0
  if (cancellable)
378
0
    g_cancellable_push_current (cancellable);
379
380
0
  res = iface->
381
0
    writev_nonblocking (stream, vectors, n_vectors, &_bytes_written, error);
382
383
0
  if (cancellable)
384
0
    g_cancellable_pop_current (cancellable);
385
386
0
  if (res == G_POLLABLE_RETURN_FAILED)
387
0
    g_warn_if_fail (error == NULL || (*error != NULL && !g_error_matches (*error, G_IO_ERROR, G_IO_ERROR_WOULD_BLOCK)));
388
0
  else if (res == G_POLLABLE_RETURN_WOULD_BLOCK)
389
0
    g_warn_if_fail (error == NULL || *error == NULL);
390
391
  /* in case of not-OK nothing must've been written */
392
0
  g_warn_if_fail (res == G_POLLABLE_RETURN_OK || _bytes_written == 0);
393
394
0
  if (bytes_written)
395
0
    *bytes_written = _bytes_written;
396
397
0
  return res;
398
0
}