|
1 /* |
|
2 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER. |
|
3 * |
|
4 * Copyright 2021 Mike Becker, Olaf Wintermann All rights reserved. |
|
5 * |
|
6 * Redistribution and use in source and binary forms, with or without |
|
7 * modification, are permitted provided that the following conditions are met: |
|
8 * |
|
9 * 1. Redistributions of source code must retain the above copyright |
|
10 * notice, this list of conditions and the following disclaimer. |
|
11 * |
|
12 * 2. Redistributions in binary form must reproduce the above copyright |
|
13 * notice, this list of conditions and the following disclaimer in the |
|
14 * documentation and/or other materials provided with the distribution. |
|
15 * |
|
16 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" |
|
17 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
|
18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
|
19 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE |
|
20 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR |
|
21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF |
|
22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS |
|
23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN |
|
24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) |
|
25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE |
|
26 * POSSIBILITY OF SUCH DAMAGE. |
|
27 */ |
|
28 |
|
29 #include "cx/streams.h" |
|
30 |
|
31 #ifndef CX_STREAM_BCOPY_BUF_SIZE |
|
32 #define CX_STREAM_BCOPY_BUF_SIZE 8192 |
|
33 #endif |
|
34 |
|
35 #ifndef CX_STREAM_COPY_BUF_SIZE |
|
36 #define CX_STREAM_COPY_BUF_SIZE 1024 |
|
37 #endif |
|
38 |
|
39 size_t cx_stream_bncopy( |
|
40 void *src, |
|
41 void *dest, |
|
42 cx_read_func rfnc, |
|
43 cx_write_func wfnc, |
|
44 char *buf, |
|
45 size_t bufsize, |
|
46 size_t n |
|
47 ) { |
|
48 if (n == 0) { |
|
49 return 0; |
|
50 } |
|
51 |
|
52 char *lbuf; |
|
53 size_t ncp = 0; |
|
54 |
|
55 if (buf) { |
|
56 if (bufsize == 0) return 0; |
|
57 lbuf = buf; |
|
58 } else { |
|
59 if (bufsize == 0) bufsize = CX_STREAM_BCOPY_BUF_SIZE; |
|
60 lbuf = malloc(bufsize); |
|
61 if (lbuf == NULL) { |
|
62 return 0; |
|
63 } |
|
64 } |
|
65 |
|
66 size_t r; |
|
67 size_t rn = bufsize > n ? n : bufsize; |
|
68 while ((r = rfnc(lbuf, 1, rn, src)) != 0) { |
|
69 r = wfnc(lbuf, 1, r, dest); |
|
70 ncp += r; |
|
71 n -= r; |
|
72 rn = bufsize > n ? n : bufsize; |
|
73 if (r == 0 || n == 0) { |
|
74 break; |
|
75 } |
|
76 } |
|
77 |
|
78 if (lbuf != buf) { |
|
79 free(lbuf); |
|
80 } |
|
81 |
|
82 return ncp; |
|
83 } |
|
84 |
|
85 size_t cx_stream_ncopy( |
|
86 void *src, |
|
87 void *dest, |
|
88 cx_read_func rfnc, |
|
89 cx_write_func wfnc, |
|
90 size_t n |
|
91 ) { |
|
92 char buf[CX_STREAM_COPY_BUF_SIZE]; |
|
93 return cx_stream_bncopy(src, dest, rfnc, wfnc, |
|
94 buf, CX_STREAM_COPY_BUF_SIZE, n); |
|
95 } |