blob: b7cbe562aecbde3165d2f189e72968c61f566017 [file] [log] [blame]
/* ====================================================================
* The Apache Software License, Version 1.1
*
* Copyright (c) 2000-2001 The Apache Software Foundation. All rights
* reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
*
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
*
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in
* the documentation and/or other materials provided with the
* distribution.
*
* 3. The end-user documentation included with the redistribution,
* if any, must include the following acknowledgment:
* "This product includes software developed by the
* Apache Software Foundation (http://www.apache.org/)."
* Alternately, this acknowledgment may appear in the software itself,
* if and wherever such third-party acknowledgments normally appear.
*
* 4. The names "Apache" and "Apache Software Foundation" must
* not be used to endorse or promote products derived from this
* software without prior written permission. For written
* permission, please contact apache@apache.org.
*
* 5. Products derived from this software may not be called "Apache",
* nor may "Apache" appear in their name, without prior written
* permission of the Apache Software Foundation.
*
* THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
* WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL THE APACHE SOFTWARE FOUNDATION OR
* ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
* USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
* OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* <http://www.apache.org/>.
*
*/
#include <stdlib.h> /* For atexit */
#include <apr.h>
#include <apr_general.h> /* For apr_initialize */
#include <apr_strings.h>
#include <apr_file_io.h>
#include <flood_profile.h>
/* FIXME: Base this on RCS ID? */
#define FLOOD_VERSION ".001"
#define CRLF "\r\n"
/* FIXME: replace the apr_recv logic with something sane. */
#define MAX_DOC_LENGTH 8192
#define LOCAL_SOCKET_TIMEOUT 30 * APR_USEC_PER_SEC
/* Win32 doesn't have stdout or stderr. */
apr_file_t *local_stdout, *local_stderr;
/* Should be the new apr_sms_t struct? Not ready yet. */
apr_pool_t *local_pool;
/* Construct a request */
static void construct_request(request_t *r)
{
/* FIXME: Handle keepalives. */
switch (r->method)
{
case GET:
r->rbuf = apr_psprintf(local_pool,
"GET %s HTTP/1.1" CRLF
"User-Agent: Flood/" FLOOD_VERSION CRLF
"Host: %s" CRLF CRLF,
r->parsed_uri->path,
r->parsed_uri->hostinfo);
r->rbuftype = POOL;
r->rbufsize = strlen(r->rbuf);
break;
case HEAD:
r->rbuf = apr_psprintf(local_pool,
"HEAD %s HTTP/1.1" CRLF
"User-Agent: Flood/" FLOOD_VERSION CRLF
"Host: %s" CRLF CRLF,
r->parsed_uri->path,
r->parsed_uri->hostinfo);
r->rbuftype = POOL;
r->rbufsize = strlen(r->rbuf);
break;
case POST:
/* FIXME */
r->rbuf = apr_psprintf(local_pool,
"POST %s HTTP/1.1" CRLF
"User-Agent: Flood/" FLOOD_VERSION CRLF
"Host: %s" CRLF CRLF,
r->parsed_uri->path,
r->parsed_uri->hostinfo);
r->rbuftype = POOL;
r->rbufsize = strlen(r->rbuf);
break;
}
}
/* Open the TCP connection to the server */
static apr_socket_t* open_socket(request_t *r)
{
apr_status_t rv = 0;
apr_sockaddr_t *destsa;
/* FIXME: apr_socket_t == socket_t - fix profile.h */
apr_socket_t *socket;
if ((rv = apr_sockaddr_info_get(&destsa, r->parsed_uri->hostname, APR_INET,
r->parsed_uri->port, 0, local_pool))
!= APR_SUCCESS) {
return NULL;
}
if ((rv = apr_socket_create(&socket, APR_INET, SOCK_STREAM,
local_pool)) != APR_SUCCESS) {
return NULL;
}
if ((rv = apr_connect(socket, destsa)) != APR_SUCCESS) {
if (APR_STATUS_IS_EINPROGRESS(rv)) {
/* FIXME: Handle better */
apr_socket_close(socket);
return NULL;
}
else {
/* FIXME: Handle */
apr_socket_close(socket);
return NULL;
}
}
apr_setsocketopt(socket, APR_SO_TIMEOUT, LOCAL_SOCKET_TIMEOUT);
return socket;
}
/* close down TCP socket */
static void close_socket(apr_socket_t *s)
{
/* FIXME: recording and other stuff here? */
apr_socket_close(s);
}
static apr_status_t write_socket(request_t *r, apr_socket_t *s)
{
apr_size_t l;
apr_status_t e;
l = r->rbufsize;
e = apr_send(s, r->rbuf, &l);
/* FIXME: Better error and allow restarts? */
if (l != r->rbufsize)
return APR_EGENERAL;
return APR_SUCCESS;
}
/* FIXME: poll implementation? */
static response_t *read_socket(apr_socket_t *s)
{
apr_status_t status;
response_t *resp;
resp = apr_pcalloc(local_pool, sizeof(response_t));
resp->rbuftype = POOL;
resp->rbufsize = MAX_DOC_LENGTH;
resp->rbuf = apr_pcalloc(local_pool, resp->rbufsize);
status = apr_recv(s, resp->rbuf, &resp->rbufsize);
if (status != APR_SUCCESS && status != APR_EOF)
return NULL;
return resp;
}
int main(int argc, char** argv)
{
request_t r;
apr_socket_t *s;
/* FIXME: Where is Roy's change to return the global pool... */
apr_initialize();
atexit(apr_terminate);
apr_pool_create(&local_pool, NULL);
/* The pool should close these file descriptors */
apr_file_open_stdout(&local_stdout, local_pool);
apr_file_open_stderr(&local_stderr, local_pool);
r.uri = "http://www.apachelabs.org/";
r.method = GET;
r.parsed_uri = apr_pcalloc(local_pool, sizeof(uri_components));
/* FIXME: This is the private copy. */
ap_parse_uri_components(local_pool, r.uri, r.parsed_uri);
if (!r.parsed_uri->port)
r.parsed_uri->port = DEFAULT_HTTP_PORT;
apr_file_printf(local_stdout, "%s\n", r.parsed_uri->hostname);
construct_request(&r);
s = open_socket(&r);
if (s)
{
response_t *resp;
if (write_socket(&r, s) == APR_SUCCESS)
{
/* Wait .25 seconds... */
/* FIXME: replace with poll. */
apr_sleep(0.25 * APR_USEC_PER_SEC);
resp = read_socket(s);
if (resp)
apr_file_printf(local_stdout, "%s\n", (char*)resp->rbuf);
}
close_socket(s);
}
return EXIT_SUCCESS;
}