/src/httpd/srclib/apr/file_io/unix/fullrw.c
Line | Count | Source (jump to first uncovered line) |
1 | | /* Licensed to the Apache Software Foundation (ASF) under one or more |
2 | | * contributor license agreements. See the NOTICE file distributed with |
3 | | * this work for additional information regarding copyright ownership. |
4 | | * The ASF licenses this file to You under the Apache License, Version 2.0 |
5 | | * (the "License"); you may not use this file except in compliance with |
6 | | * the License. You may obtain a copy of the License at |
7 | | * |
8 | | * http://www.apache.org/licenses/LICENSE-2.0 |
9 | | * |
10 | | * Unless required by applicable law or agreed to in writing, software |
11 | | * distributed under the License is distributed on an "AS IS" BASIS, |
12 | | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
13 | | * See the License for the specific language governing permissions and |
14 | | * limitations under the License. |
15 | | */ |
16 | | |
17 | | #include "apr_file_io.h" |
18 | | |
19 | | |
20 | | APR_DECLARE(apr_status_t) apr_file_read_full(apr_file_t *thefile, void *buf, |
21 | | apr_size_t nbytes, |
22 | | apr_size_t *bytes_read) |
23 | 0 | { |
24 | 0 | apr_status_t status; |
25 | 0 | apr_size_t total_read = 0; |
26 | |
|
27 | 0 | do { |
28 | 0 | apr_size_t amt = nbytes; |
29 | |
|
30 | 0 | status = apr_file_read(thefile, buf, &amt); |
31 | 0 | buf = (char *)buf + amt; |
32 | 0 | nbytes -= amt; |
33 | 0 | total_read += amt; |
34 | 0 | } while (status == APR_SUCCESS && nbytes > 0); |
35 | |
|
36 | 0 | if (bytes_read != NULL) |
37 | 0 | *bytes_read = total_read; |
38 | |
|
39 | 0 | return status; |
40 | 0 | } |
41 | | |
42 | | APR_DECLARE(apr_status_t) apr_file_write_full(apr_file_t *thefile, |
43 | | const void *buf, |
44 | | apr_size_t nbytes, |
45 | | apr_size_t *bytes_written) |
46 | 0 | { |
47 | 0 | apr_status_t status; |
48 | 0 | apr_size_t total_written = 0; |
49 | |
|
50 | 0 | do { |
51 | 0 | apr_size_t amt = nbytes; |
52 | |
|
53 | 0 | status = apr_file_write(thefile, buf, &amt); |
54 | 0 | buf = (char *)buf + amt; |
55 | 0 | nbytes -= amt; |
56 | 0 | total_written += amt; |
57 | 0 | } while (status == APR_SUCCESS && nbytes > 0); |
58 | |
|
59 | 0 | if (bytes_written != NULL) |
60 | 0 | *bytes_written = total_written; |
61 | |
|
62 | 0 | return status; |
63 | 0 | } |
64 | | |
65 | | APR_DECLARE(apr_status_t) apr_file_writev_full(apr_file_t *thefile, |
66 | | const struct iovec *vec, |
67 | | apr_size_t nvec, |
68 | | apr_size_t *bytes_written) |
69 | 0 | { |
70 | 0 | apr_status_t rv = APR_SUCCESS; |
71 | 0 | apr_size_t i; |
72 | 0 | apr_size_t amt = 0; |
73 | 0 | apr_size_t total = 0; |
74 | |
|
75 | 0 | for (i = 0; i < nvec; i++) { |
76 | 0 | total += vec[i].iov_len; |
77 | 0 | } |
78 | |
|
79 | 0 | rv = apr_file_writev(thefile, vec, nvec, &amt); |
80 | |
|
81 | 0 | if (bytes_written != NULL) |
82 | 0 | *bytes_written = amt; |
83 | |
|
84 | 0 | if (rv != APR_SUCCESS || (amt == total)) { |
85 | 0 | return rv; |
86 | 0 | } |
87 | | |
88 | 0 | for (i = 0; i < nvec && amt; i++) { |
89 | 0 | if (amt >= vec[i].iov_len) { |
90 | 0 | amt -= vec[i].iov_len; |
91 | 0 | } |
92 | 0 | else { |
93 | 0 | break; |
94 | 0 | } |
95 | 0 | } |
96 | |
|
97 | 0 | if (amt) { |
98 | 0 | rv = apr_file_write_full(thefile, (const char *)vec[i].iov_base + amt, |
99 | 0 | vec[i].iov_len - amt, NULL); |
100 | 0 | } |
101 | |
|
102 | 0 | for (; i < nvec && rv == APR_SUCCESS; i++) { |
103 | 0 | rv = apr_file_write_full(thefile, vec[i].iov_base, |
104 | 0 | vec[i].iov_len, &amt); |
105 | 0 | } |
106 | |
|
107 | 0 | if (bytes_written != NULL) |
108 | 0 | *bytes_written = total; |
109 | |
|
110 | 0 | return rv; |
111 | 0 | } |