streamcpy.c
2.64 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
/* GNU Mailutils -- a suite of utilities for electronic mail
Copyright (C) 2009, 2010 Free Software Foundation, Inc.
This library is free software; you can redistribute it and/or modify
it under the terms of the GNU Lesser General Public License as published by
the Free Software Foundation; either version 3, or (at your option)
any later version.
This library is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Lesser General Public License for more details.
You should have received a copy of the GNU Lesser General Public License
along with GNU Mailutils. If not, see <http://www.gnu.org/licenses/>. */
#ifdef HAVE_CONFIG_H
# include <config.h>
#endif
#include <stdlib.h>
#include <mailutils/types.h>
#include <mailutils/alloc.h>
#include <mailutils/error.h>
#include <mailutils/errno.h>
#include <mailutils/stream.h>
#include <mailutils/sys/stream.h>
#define STREAMCPY_MIN_BUF_SIZE 2
#define STREAMCPY_MAX_BUF_SIZE 16384
/* Copy SIZE bytes from SRC to DST. If SIZE is 0, copy everything up to
EOF. */
int
mu_stream_copy (mu_stream_t dst, mu_stream_t src, mu_off_t size,
mu_off_t *pcsz)
{
int status;
size_t bufsize, n;
char *buf;
mu_off_t total = 0;
if (pcsz)
*pcsz = 0;
if (size == 0)
{
status = mu_stream_size (src, &size);
switch (status)
{
case 0:
break;
case ENOSYS:
size = 0;
break;
default:
return status;
}
if (size)
{
mu_off_t pos;
status = mu_stream_seek (src, 0, MU_SEEK_CUR, &pos);
switch (status)
{
case 0:
if (pos > size)
return ESPIPE;
size -= pos;
break;
case EACCES:
mu_stream_clearerr (src);
case ENOSYS:
size = 0;
break;
default:
return status;
}
}
}
bufsize = size;
if (!bufsize)
bufsize = STREAMCPY_MAX_BUF_SIZE;
for (; (buf = malloc (bufsize)) == NULL; bufsize >>= 1)
if (bufsize < STREAMCPY_MIN_BUF_SIZE)
return ENOMEM;
if (size)
while (size)
{
size_t rdsize = bufsize < size ? bufsize : size;
status = mu_stream_read (src, buf, rdsize, &n);
if (status)
break;
if (n == 0)
{
status = EIO;
break;
}
status = mu_stream_write (dst, buf, n, NULL);
if (status)
break;
size -= n;
total += n;
}
else
while ((status = mu_stream_read (src, buf, bufsize, &n)) == 0
&& n > 0)
{
status = mu_stream_write (dst, buf, n, NULL);
if (status)
break;
total += n;
}
if (pcsz)
*pcsz = total;
free (buf);
return status;
}