/* 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.
 */
/*
 * testssl: Simple APR SSL sockets test.
 */

#include "apr.h"
#include "apr_general.h"
#include "apr_pools.h"
#include "apr_errno.h"
#include "apr_getopt.h"
#include "apr_time.h"
#define APR_WANT_STRFUNC
#include "apr_want.h"

#include "apr_ssl.h"
#include "apr_network_io.h"

#if APR_HAVE_STDIO_H
#include <stdio.h>
#endif
#if APR_HAVE_UNISTD_H
#include <unistd.h>
#endif
#include <stdlib.h>     /* for atexit(), malloc() */
#include <string.h>

struct sslTestCase {
    char *host;
    int port;
    const char *request;
    int result;
} tests[] = {
    { "svn.apache.org", 443, "GET / HTTP/1.0\n\n", 1 },
    { NULL }
};

static apr_ssl_socket_t *createSocket(apr_ssl_factory_t *asf, 
                                      apr_pollset_t *pollset,
                                      apr_pool_t *pool, int blocking)
{
    apr_ssl_socket_t *sock;
    apr_status_t rv;
    printf("::Creating SSL socket\n");
    rv = apr_ssl_socket_create(&sock, AF_INET, SOCK_STREAM, 0, asf, NULL);
    if (rv != APR_SUCCESS) {
        printf("\tFailed to create socket\n");
        return NULL;
    }
    rv = apr_pollset_add_ssl_socket(pollset, sock);
    if (rv != APR_SUCCESS) {
        printf("\tFailed to add to pollset\n");
        return NULL;
    }
    printf("\tOK\n");
    return sock;
}

static apr_status_t connectSocket(apr_ssl_socket_t *sock,
                                  const char *host, int port,
                                  apr_pool_t *pool)
{
    apr_status_t rv;
    apr_sockaddr_t *remoteSA;

    printf("::Connecting socket\n");
    rv = apr_sockaddr_info_get(&remoteSA, host, APR_UNSPEC, port, 0, pool);
    if (rv != APR_SUCCESS) {
        printf("\tFailed to get address for '%s', port %d\n", host, port);
        return rv;
    }
    rv = apr_ssl_socket_connect(sock, remoteSA);
    if (rv != APR_SUCCESS) {
        printf("\tFailed to connect to '%s' port %d\n", host, port);
        return rv;
    }
    printf("\tOK\n");
    return rv;
}

static apr_status_t socketRead(apr_ssl_socket_t *sock,
                               apr_pollset_t *pollset,
                               char *buf, apr_size_t *len)
{
    int lrv;
    const apr_pollfd_t *descs = NULL;
    apr_status_t rv;

    printf("::Reading from socket\n");
    rv = apr_ssl_socket_set_poll_events(sock, APR_POLLIN);
    if (rv != APR_SUCCESS) {
        printf("\tUnable to change socket poll events!\n");
        return rv;
    }

    rv = apr_pollset_poll(pollset, 30 * APR_USEC_PER_SEC, &lrv, &descs);
    if (APR_STATUS_IS_TIMEUP(rv)) {
        printf("\tTime up!\n");
        return rv;
    }

    if (lrv != 1) {
        printf("\tIncorrect return count, %d\n", lrv);
        return rv;
    }
    if (descs[0].client_data != sock) {
        printf("\tWrong socket returned?!\n");
        return rv;
    }
    if ((descs[0].rtnevents & APR_POLLIN) == 0) {
        printf("\tSocket wasn't ready? huh? req [%08x] vs rtn [%08x]\n",
               descs[0].reqevents, descs[0].rtnevents);
        return rv;
    }
    rv = apr_ssl_socket_recv(sock, buf, len);
    if (rv == APR_SUCCESS)
        printf("\tOK, read %d bytes\n", *len);
    else
        printf("\tFailed\n");
    return rv;
}

static apr_status_t socketWrite(apr_ssl_socket_t *sock,
                                apr_pollset_t *pollset,
                                const char *buf, apr_size_t *len)
{
    int lrv;
    const apr_pollfd_t *descs = NULL;
    apr_status_t rv;

    printf("::Writing to socket\n");
    rv = apr_ssl_socket_set_poll_events(sock, APR_POLLOUT);
    if (rv != APR_SUCCESS) {
        printf("\tUnable to change socket poll events!\n");
        return rv;
    }

    rv = apr_pollset_poll(pollset, 30 * APR_USEC_PER_SEC, &lrv, &descs);
    if (APR_STATUS_IS_TIMEUP(rv)) {
        printf("\tTime up!\n");
        return rv;
    }
    if (lrv != 1) {
        printf("\tIncorrect return count, %d\n", lrv);
        return rv;
    }
    if (descs[0].client_data != sock) {
        printf("\tWrong socket returned?!\n");
        return rv;
    }
    if ((descs[0].rtnevents & APR_POLLOUT) == 0) {
        printf("\tSocket wasn't ready? huh?\n");
        return rv;
    }
    rv = apr_ssl_socket_send(sock, buf, len);
    if (rv == APR_SUCCESS)
        printf("\tOK, wrote %d bytes\n", *len);
    else
        printf("\tFailed\n");
    return rv;
}

apr_status_t socketClose(apr_ssl_socket_t *sock, apr_pollset_t *pollset)
{
    apr_status_t rv;
    printf("::Closing socket\n");
    rv = apr_pollset_remove_ssl_socket(sock);
    if (rv != APR_SUCCESS)
        printf("\tUnable to remove socket from pollset?\n");
    rv = apr_ssl_socket_close(sock);
    if (rv != APR_SUCCESS)
        printf("\tFailed to close SSL socket\n");
    else
        printf("\tOK\n");
    return rv;
}


int main(int argc, const char * const * argv)
{
    apr_pool_t *pool;
    apr_ssl_factory_t *asf = NULL;
    apr_status_t rv;
    apr_pollset_t *pollset;

    (void) apr_initialize();
    apr_pool_create(&pool, NULL);
    atexit(apr_terminate);

    printf("SSL Library: %s\n", apr_ssl_library_name());

    if (apr_pollset_create(&pollset, 1, pool, 0) != APR_SUCCESS) {
        printf("Failed to create pollset!\n");
        exit(1);
    }

    if (apr_ssl_factory_create(&asf, NULL, NULL, NULL, 
                               APR_SSL_FACTORY_CLIENT, pool) != APR_SUCCESS) {
        fprintf(stderr, "Unable to create client factory\n");
    } else {
        int i;
        for(i = 0; tests[i].host; i++) {
            apr_ssl_socket_t *sslSock = createSocket(asf, pollset, pool, 0);
            if (!sslSock)
                continue;

            rv = connectSocket(sslSock, tests[i].host, tests[i].port, pool);
            if (rv == APR_SUCCESS) {
                apr_size_t len = strlen(tests[i].request);
                rv = socketWrite(sslSock, pollset, tests[i].request, &len);
                if (rv == APR_SUCCESS) {
                    char buffer[4096];
                    len = 4096;
                    rv = socketRead(sslSock, pollset, buffer, &len);
                }
            }
            socketClose(sslSock, pollset);
        }
    }

    apr_pollset_destroy(pollset);
    apr_pool_destroy(pool);

    return 0;
}