1 /* ====================================================================
2 * Licensed to the Apache Software Foundation (ASF) under one
3 * or more contributor license agreements. See the NOTICE file
4 * distributed with this work for additional information
5 * regarding copyright ownership. The ASF licenses this file
6 * to you under the Apache License, Version 2.0 (the
7 * "License"); you may not use this file except in compliance
8 * with the License. You may obtain a copy of the License at
9 *
10 * http://www.apache.org/licenses/LICENSE-2.0
11 *
12 * Unless required by applicable law or agreed to in writing,
13 * software distributed under the License is distributed on an
14 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15 * KIND, either express or implied. See the License for the
16 * specific language governing permissions and limitations
17 * under the License.
18 * ====================================================================
19 */
20
21 #include <apr_pools.h>
22
23 #include "serf.h"
24 #include "serf_bucket_util.h"
25
26
27 typedef struct {
28 serf_bucket_t *stream;
29 } barrier_context_t;
30
31
serf_bucket_barrier_create(serf_bucket_t * stream,serf_bucket_alloc_t * allocator)32 serf_bucket_t *serf_bucket_barrier_create(
33 serf_bucket_t *stream,
34 serf_bucket_alloc_t *allocator)
35 {
36 barrier_context_t *ctx;
37
38 ctx = serf_bucket_mem_alloc(allocator, sizeof(*ctx));
39 ctx->stream = stream;
40
41 return serf_bucket_create(&serf_bucket_type_barrier, allocator, ctx);
42 }
43
serf_barrier_read(serf_bucket_t * bucket,apr_size_t requested,const char ** data,apr_size_t * len)44 static apr_status_t serf_barrier_read(serf_bucket_t *bucket,
45 apr_size_t requested,
46 const char **data, apr_size_t *len)
47 {
48 barrier_context_t *ctx = bucket->data;
49
50 return serf_bucket_read(ctx->stream, requested, data, len);
51 }
52
serf_barrier_read_iovec(serf_bucket_t * bucket,apr_size_t requested,int vecs_size,struct iovec * vecs,int * vecs_used)53 static apr_status_t serf_barrier_read_iovec(serf_bucket_t *bucket,
54 apr_size_t requested,
55 int vecs_size, struct iovec *vecs,
56 int *vecs_used)
57 {
58 barrier_context_t *ctx = bucket->data;
59
60 return serf_bucket_read_iovec(ctx->stream, requested, vecs_size, vecs,
61 vecs_used);
62 }
63
serf_barrier_readline(serf_bucket_t * bucket,int acceptable,int * found,const char ** data,apr_size_t * len)64 static apr_status_t serf_barrier_readline(serf_bucket_t *bucket,
65 int acceptable, int *found,
66 const char **data, apr_size_t *len)
67 {
68 barrier_context_t *ctx = bucket->data;
69
70 return serf_bucket_readline(ctx->stream, acceptable, found, data, len);
71 }
72
serf_barrier_peek(serf_bucket_t * bucket,const char ** data,apr_size_t * len)73 static apr_status_t serf_barrier_peek(serf_bucket_t *bucket,
74 const char **data,
75 apr_size_t *len)
76 {
77 barrier_context_t *ctx = bucket->data;
78
79 return serf_bucket_peek(ctx->stream, data, len);
80 }
81
serf_barrier_destroy(serf_bucket_t * bucket)82 static void serf_barrier_destroy(serf_bucket_t *bucket)
83 {
84 /* The intent of this bucket is not to let our wrapped buckets be
85 * destroyed. */
86
87 /* The option is for us to go ahead and 'eat' this bucket now,
88 * or just ignore the deletion entirely.
89 */
90 serf_default_destroy_and_data(bucket);
91 }
92
93 const serf_bucket_type_t serf_bucket_type_barrier = {
94 "BARRIER",
95 serf_barrier_read,
96 serf_barrier_readline,
97 serf_barrier_read_iovec,
98 serf_default_read_for_sendfile,
99 serf_default_read_bucket,
100 serf_barrier_peek,
101 serf_barrier_destroy,
102 };
103