/* ====================================================================

 *    Licensed to the Apache Software Foundation (ASF) under one

 *    or more contributor license agreements.  See the NOTICE file

 *    distributed with this work for additional information

 *    regarding copyright ownership.  The ASF licenses this file

 *    to you under the Apache License, Version 2.0 (the

 *    "License"); you may not use this file except in compliance

 *    with the License.  You may obtain a copy of the License at

 *

 *      http://www.apache.org/licenses/LICENSE-2.0

 *

 *    Unless required by applicable law or agreed to in writing,

 *    software distributed under the License is distributed on an

 *    "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY

 *    KIND, either express or implied.  See the License for the

 *    specific language governing permissions and limitations

 *    under the License.

 * ====================================================================

 */

#ifndef TEST_SERVER_H

#define TEST_SERVER_H

/* Test logging facilities, set flag to 1 to enable console logging for

   the test suite. */

#define TEST_VERBOSE 0

#define TEST_SERVER_DUMP 1

/* Default port for our test server. */

#define SERV_PORT 12345

#define SERV_PORT_STR "12345"

#define PROXY_PORT 23456

typedef struct serv_ctx_t serv_ctx_t;

typedef apr_status_t (*send_func_t)(serv_ctx_t *serv_ctx, const char *data,

                                    apr_size_t *len);

typedef apr_status_t (*receive_func_t)(serv_ctx_t *serv_ctx, char *data,

                                       apr_size_t *len);

typedef apr_status_t (*handshake_func_t)(serv_ctx_t *serv_ctx);

typedef apr_status_t (*reset_conn_func_t)(serv_ctx_t *serv_ctx);

typedef struct

{

    enum {

        SERVER_RECV,

        SERVER_SEND,

        SERVER_RESPOND,

        SERVER_IGNORE_AND_KILL_CONNECTION,

        SERVER_KILL_CONNECTION,

        PROXY_FORWARD,

    } kind;

    const char *text;

} test_server_action_t;

typedef struct

{

    const char *text;

} test_server_message_t;

struct serv_ctx_t {

    /* Pool for resource allocation. */

    apr_pool_t *pool;

    serf_bucket_alloc_t *allocator;

    apr_int32_t options;

    /* Array of actions which server will replay when client connected. */

    test_server_action_t *action_list;

    /* Size of action_list array. */

    apr_size_t action_count;

    /* Index of current action. */

    apr_size_t cur_action;

    /* Array of messages the server will receive from the client. */

    test_server_message_t *message_list;

    /* Size of message_list array. */

    apr_size_t message_count;

    /* Index of current message. */

    apr_size_t cur_message;

    /* Number of messages received that the server didn't respond to yet. */

    apr_size_t outstanding_responses;

    /* Position in message buffer (incoming messages being read). */

    apr_size_t message_buf_pos;

    /* Position in action buffer. (outgoing messages being sent). */

    apr_size_t action_buf_pos;

    /* Address for server binding. */

    apr_sockaddr_t *serv_addr;

    apr_socket_t *serv_sock;

    /* Accepted client socket. NULL if there is no client socket. */

    apr_socket_t *client_sock;

    /* Client socket to a server, in case this server acts as a proxy. */

    apr_socket_t *proxy_client_sock;

    serf_bucket_t *clientstream;

    serf_bucket_t *servstream;

    send_func_t send;

    receive_func_t read;

    /* SSL related variables */

    handshake_func_t handshake;

    reset_conn_func_t reset;

    void *ssl_ctx;

    const char *client_cn;

    apr_status_t bio_read_status;

};

void setup_test_server(serv_ctx_t **servctx_p,

                       apr_sockaddr_t *address,

                       test_server_message_t *message_list,

                       apr_size_t message_count,

                       test_server_action_t *action_list,

                       apr_size_t action_count,

                       apr_int32_t options,

                       apr_pool_t *pool);

void setup_https_test_server(serv_ctx_t **servctx_p,

                             apr_sockaddr_t *address,

                             test_server_message_t *message_list,

                             apr_size_t message_count,

                             test_server_action_t *action_list,

                             apr_size_t action_count,

                             apr_int32_t options,

                             const char *keyfile,

                             const char **certfiles,

                             const char *client_cn,

                             apr_pool_t *pool);

apr_status_t start_test_server(serv_ctx_t *serv_ctx);

apr_status_t run_test_server(serv_ctx_t *servctx,

                             apr_short_interval_time_t duration,

                             apr_pool_t *pool);

#ifndef APR_VERSION_AT_LEAST /* Introduced in APR 1.3.0 */

#define APR_VERSION_AT_LEAST(major,minor,patch)                  \

(((major) < APR_MAJOR_VERSION)                                       \

 || ((major) == APR_MAJOR_VERSION && (minor) < APR_MINOR_VERSION)    \

 || ((major) == APR_MAJOR_VERSION && (minor) == APR_MINOR_VERSION && \

     (patch) <= APR_PATCH_VERSION))

#endif /* APR_VERSION_AT_LEAST */

#endif /* TEST_SERVER_H */