1362181Sdim/* ==================================================================== 2362181Sdim * Licensed to the Apache Software Foundation (ASF) under one 3362181Sdim * or more contributor license agreements. See the NOTICE file 4362181Sdim * distributed with this work for additional information 5362181Sdim * regarding copyright ownership. The ASF licenses this file 6362181Sdim * to you under the Apache License, Version 2.0 (the 7362181Sdim * "License"); you may not use this file except in compliance 8362181Sdim * with the License. You may obtain a copy of the License at 9251877Speter * 10362181Sdim * http://www.apache.org/licenses/LICENSE-2.0 11251877Speter * 12362181Sdim * Unless required by applicable law or agreed to in writing, 13362181Sdim * software distributed under the License is distributed on an 14362181Sdim * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY 15362181Sdim * KIND, either express or implied. See the License for the 16362181Sdim * specific language governing permissions and limitations 17362181Sdim * under the License. 18362181Sdim * ==================================================================== 19251877Speter */ 20251877Speter 21251877Speter#include <apr_pools.h> 22251877Speter 23251877Speter#include "serf.h" 24251877Speter#include "serf_bucket_util.h" 25251877Speter 26251877Speter 27251877Spetertypedef struct { 28251877Speter struct iovec *vecs; 29251877Speter 30251877Speter /* Total number of buffer stored in the vecs var. */ 31251877Speter int vecs_len; 32251877Speter /* Points to the first unread buffer. */ 33251877Speter int current_vec; 34251877Speter /* First buffer offset. */ 35251877Speter int offset; 36251877Speter} iovec_context_t; 37251877Speter 38251877Speterserf_bucket_t *serf_bucket_iovec_create( 39251877Speter struct iovec vecs[], 40251877Speter int len, 41251877Speter serf_bucket_alloc_t *allocator) 42251877Speter{ 43251877Speter iovec_context_t *ctx; 44251877Speter int i; 45251877Speter 46251877Speter ctx = serf_bucket_mem_alloc(allocator, sizeof(*ctx)); 47251877Speter ctx->vecs = serf_bucket_mem_alloc(allocator, len * sizeof(struct iovec)); 48251877Speter ctx->vecs_len = len; 49251877Speter ctx->current_vec = 0; 50251877Speter ctx->offset = 0; 51251877Speter 52251877Speter /* copy all buffers to our iovec. */ 53251877Speter for (i = 0; i < len; i++) { 54251877Speter ctx->vecs[i].iov_base = vecs[i].iov_base; 55251877Speter ctx->vecs[i].iov_len = vecs[i].iov_len; 56251877Speter } 57251877Speter 58251877Speter return serf_bucket_create(&serf_bucket_type_iovec, allocator, ctx); 59251877Speter} 60251877Speter 61251877Speterstatic apr_status_t serf_iovec_readline(serf_bucket_t *bucket, 62251877Speter int acceptable, int *found, 63251877Speter const char **data, apr_size_t *len) 64251877Speter{ 65251877Speter return APR_ENOTIMPL; 66251877Speter} 67251877Speter 68251877Speterstatic apr_status_t serf_iovec_read_iovec(serf_bucket_t *bucket, 69251877Speter apr_size_t requested, 70251877Speter int vecs_size, 71251877Speter struct iovec *vecs, 72251877Speter int *vecs_used) 73251877Speter{ 74251877Speter iovec_context_t *ctx = bucket->data; 75251877Speter 76251877Speter *vecs_used = 0; 77251877Speter 78251877Speter /* copy the requested amount of buffers to the provided iovec. */ 79251877Speter for (; ctx->current_vec < ctx->vecs_len; ctx->current_vec++) { 80251877Speter struct iovec vec = ctx->vecs[ctx->current_vec]; 81251877Speter apr_size_t remaining; 82251877Speter 83251877Speter if (requested != SERF_READ_ALL_AVAIL && requested <= 0) 84251877Speter break; 85251877Speter if (*vecs_used >= vecs_size) 86251877Speter break; 87251877Speter 88251877Speter vecs[*vecs_used].iov_base = (char*)vec.iov_base + ctx->offset; 89251877Speter remaining = vec.iov_len - ctx->offset; 90251877Speter 91251877Speter /* Less bytes requested than remaining in the current buffer. */ 92251877Speter if (requested != SERF_READ_ALL_AVAIL && requested < remaining) { 93251877Speter vecs[*vecs_used].iov_len = requested; 94251877Speter ctx->offset += requested; 95251877Speter requested = 0; 96251877Speter (*vecs_used)++; 97251877Speter break; 98251877Speter } else { 99251877Speter /* Copy the complete buffer. */ 100251877Speter vecs[*vecs_used].iov_len = remaining; 101251877Speter ctx->offset = 0; 102251877Speter if (requested != SERF_READ_ALL_AVAIL) 103251877Speter requested -= remaining; 104251877Speter (*vecs_used)++; 105251877Speter } 106251877Speter } 107251877Speter 108251877Speter if (ctx->current_vec == ctx->vecs_len && !ctx->offset) 109251877Speter return APR_EOF; 110251877Speter 111251877Speter return APR_SUCCESS; 112251877Speter} 113251877Speter 114251877Speterstatic apr_status_t serf_iovec_read(serf_bucket_t *bucket, 115251877Speter apr_size_t requested, 116251877Speter const char **data, apr_size_t *len) 117251877Speter{ 118251877Speter struct iovec vec[1]; 119251877Speter apr_status_t status; 120251877Speter int vecs_used; 121251877Speter 122251877Speter status = serf_iovec_read_iovec(bucket, requested, 1, vec, &vecs_used); 123251877Speter 124251877Speter if (vecs_used) { 125251877Speter *data = vec[0].iov_base; 126251877Speter *len = vec[0].iov_len; 127251877Speter } else { 128251877Speter *len = 0; 129251877Speter } 130251877Speter 131251877Speter return status; 132251877Speter} 133251877Speter 134251877Speterstatic apr_status_t serf_iovec_peek(serf_bucket_t *bucket, 135251877Speter const char **data, 136251877Speter apr_size_t *len) 137251877Speter{ 138251877Speter iovec_context_t *ctx = bucket->data; 139251877Speter 140251877Speter if (ctx->current_vec >= ctx->vecs_len) { 141251877Speter *len = 0; 142251877Speter return APR_EOF; 143251877Speter } 144251877Speter 145251877Speter /* Return the first unread buffer, don't bother combining all 146251877Speter remaining data. */ 147251877Speter *data = ctx->vecs[ctx->current_vec].iov_base; 148251877Speter *len = ctx->vecs[ctx->current_vec].iov_len; 149251877Speter 150251877Speter if (ctx->current_vec + 1 == ctx->vecs_len) 151251877Speter return APR_EOF; 152251877Speter 153251877Speter return APR_SUCCESS; 154251877Speter} 155251877Speter 156251877Speterstatic void serf_iovec_destroy(serf_bucket_t *bucket) 157251877Speter{ 158251877Speter iovec_context_t *ctx = bucket->data; 159251877Speter 160251877Speter serf_bucket_mem_free(bucket->allocator, ctx->vecs); 161251877Speter serf_default_destroy_and_data(bucket); 162251877Speter} 163251877Speter 164251877Speter 165251877Speterconst serf_bucket_type_t serf_bucket_type_iovec = { 166251877Speter "IOVEC", 167251877Speter serf_iovec_read, 168251877Speter serf_iovec_readline, 169251877Speter serf_iovec_read_iovec, 170251877Speter serf_default_read_for_sendfile, 171251877Speter serf_default_read_bucket, 172251877Speter serf_iovec_peek, 173251877Speter serf_iovec_destroy, 174251877Speter}; 175