|
Packit |
3adb1e |
/* ====================================================================
|
|
Packit |
3adb1e |
* Licensed to the Apache Software Foundation (ASF) under one
|
|
Packit |
3adb1e |
* or more contributor license agreements. See the NOTICE file
|
|
Packit |
3adb1e |
* distributed with this work for additional information
|
|
Packit |
3adb1e |
* regarding copyright ownership. The ASF licenses this file
|
|
Packit |
3adb1e |
* to you under the Apache License, Version 2.0 (the
|
|
Packit |
3adb1e |
* "License"); you may not use this file except in compliance
|
|
Packit |
3adb1e |
* with the License. You may obtain a copy of the License at
|
|
Packit |
3adb1e |
*
|
|
Packit |
3adb1e |
* http://www.apache.org/licenses/LICENSE-2.0
|
|
Packit |
3adb1e |
*
|
|
Packit |
3adb1e |
* Unless required by applicable law or agreed to in writing,
|
|
Packit |
3adb1e |
* software distributed under the License is distributed on an
|
|
Packit |
3adb1e |
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
Packit |
3adb1e |
* KIND, either express or implied. See the License for the
|
|
Packit |
3adb1e |
* specific language governing permissions and limitations
|
|
Packit |
3adb1e |
* under the License.
|
|
Packit |
3adb1e |
* ====================================================================
|
|
Packit |
3adb1e |
*/
|
|
Packit |
3adb1e |
|
|
Packit |
3adb1e |
#include <stdlib.h>
|
|
Packit |
3adb1e |
|
|
Packit |
3adb1e |
#include <apr.h>
|
|
Packit |
3adb1e |
#include <apr_uri.h>
|
|
Packit |
3adb1e |
#include <apr_strings.h>
|
|
Packit |
3adb1e |
#include <apr_atomic.h>
|
|
Packit |
3adb1e |
#include <apr_version.h>
|
|
Packit |
3adb1e |
|
|
Packit |
3adb1e |
#include "serf.h"
|
|
Packit |
3adb1e |
|
|
Packit |
3adb1e |
typedef struct {
|
|
Packit |
3adb1e |
const char *resp_file;
|
|
Packit |
3adb1e |
serf_bucket_t *bkt;
|
|
Packit |
3adb1e |
} accept_baton_t;
|
|
Packit |
3adb1e |
|
|
Packit |
3adb1e |
static serf_bucket_t* accept_response(void *acceptor_baton,
|
|
Packit |
3adb1e |
serf_bucket_alloc_t *bkt_alloc,
|
|
Packit |
3adb1e |
apr_pool_t *pool)
|
|
Packit |
3adb1e |
{
|
|
Packit |
3adb1e |
accept_baton_t *ctx = acceptor_baton;
|
|
Packit |
3adb1e |
serf_bucket_t *c;
|
|
Packit |
3adb1e |
apr_file_t *file;
|
|
Packit |
3adb1e |
apr_status_t status;
|
|
Packit |
3adb1e |
|
|
Packit |
3adb1e |
status = apr_file_open(&file, ctx->resp_file,
|
|
Packit |
3adb1e |
APR_READ, APR_OS_DEFAULT, pool);
|
|
Packit |
3adb1e |
if (status) {
|
|
Packit |
3adb1e |
return NULL;
|
|
Packit |
3adb1e |
}
|
|
Packit |
3adb1e |
|
|
Packit |
3adb1e |
c = ctx->bkt = serf_bucket_file_create(file, bkt_alloc);
|
|
Packit |
3adb1e |
|
|
Packit |
3adb1e |
c = serf_bucket_barrier_create(c, bkt_alloc);
|
|
Packit |
3adb1e |
|
|
Packit |
3adb1e |
return serf_bucket_response_create(c, bkt_alloc);
|
|
Packit |
3adb1e |
}
|
|
Packit |
3adb1e |
|
|
Packit |
3adb1e |
typedef struct {
|
|
Packit |
3adb1e |
#if APR_MAJOR_VERSION > 0
|
|
Packit |
3adb1e |
apr_uint32_t requests_outstanding;
|
|
Packit |
3adb1e |
#else
|
|
Packit |
3adb1e |
apr_atomic_t requests_outstanding;
|
|
Packit |
3adb1e |
#endif
|
|
Packit |
3adb1e |
} handler_baton_t;
|
|
Packit |
3adb1e |
|
|
Packit |
3adb1e |
/* Kludges for APR 0.9 support. */
|
|
Packit |
3adb1e |
#if APR_MAJOR_VERSION == 0
|
|
Packit |
3adb1e |
#define apr_atomic_inc32 apr_atomic_inc
|
|
Packit |
3adb1e |
#define apr_atomic_dec32 apr_atomic_dec
|
|
Packit |
3adb1e |
#define apr_atomic_read32 apr_atomic_read
|
|
Packit |
3adb1e |
#endif
|
|
Packit |
3adb1e |
|
|
Packit |
3adb1e |
static apr_status_t handle_response(serf_request_t *request,
|
|
Packit |
3adb1e |
serf_bucket_t *response,
|
|
Packit |
3adb1e |
void *handler_baton,
|
|
Packit |
3adb1e |
apr_pool_t *pool)
|
|
Packit |
3adb1e |
{
|
|
Packit |
3adb1e |
const char *data, *s;
|
|
Packit |
3adb1e |
apr_size_t len;
|
|
Packit |
3adb1e |
serf_status_line sl;
|
|
Packit |
3adb1e |
apr_status_t status;
|
|
Packit |
3adb1e |
handler_baton_t *ctx = handler_baton;
|
|
Packit |
3adb1e |
|
|
Packit |
3adb1e |
status = serf_bucket_response_status(response, &sl);
|
|
Packit |
3adb1e |
if (status) {
|
|
Packit |
3adb1e |
if (APR_STATUS_IS_EAGAIN(status)) {
|
|
Packit |
3adb1e |
return APR_SUCCESS;
|
|
Packit |
3adb1e |
}
|
|
Packit |
3adb1e |
abort();
|
|
Packit |
3adb1e |
}
|
|
Packit |
3adb1e |
|
|
Packit |
3adb1e |
status = serf_bucket_read(response, 2048, &data, &len;;
|
|
Packit |
3adb1e |
|
|
Packit |
3adb1e |
if (!status || APR_STATUS_IS_EOF(status)) {
|
|
Packit |
3adb1e |
if (len) {
|
|
Packit |
3adb1e |
s = apr_pstrmemdup(pool, data, len);
|
|
Packit |
3adb1e |
printf("%s", s);
|
|
Packit |
3adb1e |
}
|
|
Packit |
3adb1e |
}
|
|
Packit |
3adb1e |
else if (APR_STATUS_IS_EAGAIN(status)) {
|
|
Packit |
3adb1e |
status = APR_SUCCESS;
|
|
Packit |
3adb1e |
}
|
|
Packit |
3adb1e |
if (APR_STATUS_IS_EOF(status)) {
|
|
Packit |
3adb1e |
serf_bucket_t *hdrs;
|
|
Packit |
3adb1e |
const char *v;
|
|
Packit |
3adb1e |
|
|
Packit |
3adb1e |
hdrs = serf_bucket_response_get_headers(response);
|
|
Packit |
3adb1e |
v = serf_bucket_headers_get(hdrs, "Trailer-Test");
|
|
Packit |
3adb1e |
if (v) {
|
|
Packit |
3adb1e |
printf("Trailer-Test: %s\n", v);
|
|
Packit |
3adb1e |
}
|
|
Packit |
3adb1e |
|
|
Packit |
3adb1e |
apr_atomic_dec32(&ctx->requests_outstanding);
|
|
Packit |
3adb1e |
}
|
|
Packit |
3adb1e |
|
|
Packit |
3adb1e |
return status;
|
|
Packit |
3adb1e |
}
|
|
Packit |
3adb1e |
|
|
Packit |
3adb1e |
int main(int argc, const char **argv)
|
|
Packit |
3adb1e |
{
|
|
Packit |
3adb1e |
apr_status_t status;
|
|
Packit |
3adb1e |
apr_pool_t *pool;
|
|
Packit |
3adb1e |
serf_bucket_t *resp_bkt;
|
|
Packit |
3adb1e |
accept_baton_t accept_ctx;
|
|
Packit |
3adb1e |
handler_baton_t handler_ctx;
|
|
Packit |
3adb1e |
serf_bucket_alloc_t *allocator;
|
|
Packit |
3adb1e |
|
|
Packit |
3adb1e |
if (argc != 2) {
|
|
Packit |
3adb1e |
printf("%s: [Resp. File]\n", argv[0]);
|
|
Packit |
3adb1e |
exit(-1);
|
|
Packit |
3adb1e |
}
|
|
Packit |
3adb1e |
accept_ctx.resp_file = argv[1];
|
|
Packit |
3adb1e |
accept_ctx.bkt = NULL;
|
|
Packit |
3adb1e |
|
|
Packit |
3adb1e |
apr_initialize();
|
|
Packit |
3adb1e |
atexit(apr_terminate);
|
|
Packit |
3adb1e |
|
|
Packit |
3adb1e |
apr_pool_create(&pool, NULL);
|
|
Packit |
3adb1e |
apr_atomic_init(pool);
|
|
Packit |
3adb1e |
/* serf_initialize(); */
|
|
Packit |
3adb1e |
|
|
Packit |
3adb1e |
allocator = serf_bucket_allocator_create(pool, NULL, NULL);
|
|
Packit |
3adb1e |
|
|
Packit |
3adb1e |
handler_ctx.requests_outstanding = 0;
|
|
Packit |
3adb1e |
apr_atomic_inc32(&handler_ctx.requests_outstanding);
|
|
Packit |
3adb1e |
|
|
Packit |
3adb1e |
resp_bkt = accept_response(&accept_ctx, allocator, pool);
|
|
Packit |
3adb1e |
while (1) {
|
|
Packit |
3adb1e |
status = handle_response(NULL, resp_bkt, &handler_ctx, pool);
|
|
Packit |
3adb1e |
if (APR_STATUS_IS_TIMEUP(status))
|
|
Packit |
3adb1e |
continue;
|
|
Packit |
3adb1e |
if (SERF_BUCKET_READ_ERROR(status)) {
|
|
Packit |
3adb1e |
printf("Error running context: %d\n", status);
|
|
Packit |
3adb1e |
exit(1);
|
|
Packit |
3adb1e |
}
|
|
Packit |
3adb1e |
if (!apr_atomic_read32(&handler_ctx.requests_outstanding)) {
|
|
Packit |
3adb1e |
break;
|
|
Packit |
3adb1e |
}
|
|
Packit |
3adb1e |
}
|
|
Packit |
3adb1e |
serf_bucket_destroy(resp_bkt);
|
|
Packit |
3adb1e |
serf_bucket_destroy(accept_ctx.bkt);
|
|
Packit |
3adb1e |
|
|
Packit |
3adb1e |
apr_pool_destroy(pool);
|
|
Packit |
3adb1e |
|
|
Packit |
3adb1e |
return 0;
|
|
Packit |
3adb1e |
}
|