468 lines
		
	
	
		
			12 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			468 lines
		
	
	
		
			12 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
/*
 | 
						|
 * Copyright (c) 2016 MariaDB Corporation Ab
 | 
						|
 *
 | 
						|
 * Use of this software is governed by the Business Source License included
 | 
						|
 * in the LICENSE.TXT file and at www.mariadb.com/bsl11.
 | 
						|
 *
 | 
						|
 * Change Date: 2023-01-01
 | 
						|
 *
 | 
						|
 * On the date above, in accordance with the Business Source License, use
 | 
						|
 * of this software will be governed by version 2 or later of the General
 | 
						|
 * Public License.
 | 
						|
 */
 | 
						|
 | 
						|
/**
 | 
						|
 * @file httpd.c - HTTP daemon protocol module
 | 
						|
 *
 | 
						|
 * The httpd protocol module is intended as a mechanism to allow connections
 | 
						|
 * into the gateway for the purpose of accessing information within
 | 
						|
 * the gateway with a REST interface
 | 
						|
 * databases.
 | 
						|
 *
 | 
						|
 * In the first instance it is intended to allow a debug connection to access
 | 
						|
 * internal data structures, however it may also be used to manage the
 | 
						|
 * configuration of the gateway via REST interface.
 | 
						|
 *
 | 
						|
 * @verbatim
 | 
						|
 * Revision History
 | 
						|
 * Date         Who                     Description
 | 
						|
 * 08/07/2013   Massimiliano Pinto      Initial version
 | 
						|
 * 09/07/2013   Massimiliano Pinto      Added /show?dcb|session for all dcbs|sessions
 | 
						|
 *
 | 
						|
 * @endverbatim
 | 
						|
 */
 | 
						|
 | 
						|
#define MXS_MODULE_NAME "HTTPD"
 | 
						|
 | 
						|
#include "httpd.hh"
 | 
						|
#include <ctype.h>
 | 
						|
#include <maxbase/alloc.h>
 | 
						|
#include <maxscale/protocol.hh>
 | 
						|
#include <maxscale/modinfo.h>
 | 
						|
#include <maxscale/resultset.hh>
 | 
						|
 | 
						|
#define ISspace(x) isspace((int)(x))
 | 
						|
#define HTTP_SERVER_STRING "MaxScale(c) v.1.0.0"
 | 
						|
 | 
						|
static int   httpd_read_event(DCB* dcb);
 | 
						|
static int   httpd_write_event(DCB* dcb);
 | 
						|
static int   httpd_write(DCB* dcb, GWBUF* queue);
 | 
						|
static int   httpd_error(DCB* dcb);
 | 
						|
static int   httpd_hangup(DCB* dcb);
 | 
						|
static int   httpd_accept(DCB*);
 | 
						|
static int   httpd_close(DCB* dcb);
 | 
						|
static int   httpd_get_line(int sock, char* buf, int size);
 | 
						|
static void  httpd_send_headers(DCB* dcb, int final, bool auth_ok);
 | 
						|
static char* httpd_default_auth();
 | 
						|
 | 
						|
extern "C"
 | 
						|
{
 | 
						|
/**
 | 
						|
 * The module entry point routine. It is this routine that
 | 
						|
 * must populate the structure that is referred to as the
 | 
						|
 * "module object", this is a structure with the set of
 | 
						|
 * external entry points for this module.
 | 
						|
 *
 | 
						|
 * @return The module object
 | 
						|
 */
 | 
						|
MXS_MODULE* MXS_CREATE_MODULE()
 | 
						|
{
 | 
						|
    static MXS_PROTOCOL MyObject =
 | 
						|
    {
 | 
						|
        httpd_read_event,       /**< Read - EPOLLIN handler        */
 | 
						|
        httpd_write,            /**< Write - data from gateway     */
 | 
						|
        httpd_write_event,      /**< WriteReady - EPOLLOUT handler */
 | 
						|
        httpd_error,            /**< Error - EPOLLERR handler      */
 | 
						|
        httpd_hangup,           /**< HangUp - EPOLLHUP handler     */
 | 
						|
        httpd_accept,           /**< Accept                        */
 | 
						|
        NULL,                   /**< Connect                       */
 | 
						|
        httpd_close,            /**< Close                         */
 | 
						|
        NULL,                   /**< Authentication                */
 | 
						|
        httpd_default_auth,     /**< Default authenticator         */
 | 
						|
        NULL,                   /**< Connection limit reached      */
 | 
						|
        NULL,
 | 
						|
        NULL,
 | 
						|
    };
 | 
						|
 | 
						|
    static MXS_MODULE info =
 | 
						|
    {
 | 
						|
        MXS_MODULE_API_PROTOCOL,
 | 
						|
        MXS_MODULE_IN_DEVELOPMENT,
 | 
						|
        MXS_PROTOCOL_VERSION,
 | 
						|
        "An experimental HTTPD implementation for use in administration",
 | 
						|
        "V1.2.0",
 | 
						|
        MXS_NO_MODULE_CAPABILITIES,
 | 
						|
        &MyObject,
 | 
						|
        NULL,       /* Process init. */
 | 
						|
        NULL,       /* Process finish. */
 | 
						|
        NULL,       /* Thread init. */
 | 
						|
        NULL,       /* Thread finish. */
 | 
						|
        {
 | 
						|
            {MXS_END_MODULE_PARAMS}
 | 
						|
        }
 | 
						|
    };
 | 
						|
 | 
						|
    return &info;
 | 
						|
}
 | 
						|
}
 | 
						|
/*lint +e14 */
 | 
						|
 | 
						|
static const char* default_auth = "NullAuthAllow";
 | 
						|
 | 
						|
/**
 | 
						|
 * The default authenticator name for this protocol
 | 
						|
 *
 | 
						|
 * @return name of authenticator
 | 
						|
 */
 | 
						|
static char* httpd_default_auth()
 | 
						|
{
 | 
						|
    return (char*)default_auth;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Read event for EPOLLIN on the httpd protocol module.
 | 
						|
 *
 | 
						|
 * @param dcb   The descriptor control block
 | 
						|
 * @return
 | 
						|
 */
 | 
						|
static int httpd_read_event(DCB* dcb)
 | 
						|
{
 | 
						|
    MXS_SESSION* session = dcb->session;
 | 
						|
 | 
						|
    int numchars = 1;
 | 
						|
    char buf[HTTPD_REQUESTLINE_MAXLEN - 1] = "";
 | 
						|
    char* query_string = NULL;
 | 
						|
    char method[HTTPD_METHOD_MAXLEN - 1] = "";
 | 
						|
    char url[HTTPD_SMALL_BUFFER] = "";
 | 
						|
    size_t i, j;
 | 
						|
    int headers_read = 0;
 | 
						|
    HTTPD_session* client_data = NULL;
 | 
						|
    GWBUF* uri;
 | 
						|
 | 
						|
    client_data = static_cast<HTTPD_session*>(dcb->data);
 | 
						|
 | 
						|
    /**
 | 
						|
     * get the request line
 | 
						|
     * METHOD URL HTTP_VER\r\n
 | 
						|
     */
 | 
						|
 | 
						|
    numchars = httpd_get_line(dcb->fd, buf, sizeof(buf));
 | 
						|
 | 
						|
    i = 0;
 | 
						|
    j = 0;
 | 
						|
    while (!ISspace(buf[j]) && (i < sizeof(method) - 1))
 | 
						|
    {
 | 
						|
        method[i] = buf[j];
 | 
						|
        i++;
 | 
						|
        j++;
 | 
						|
    }
 | 
						|
    method[i] = '\0';
 | 
						|
 | 
						|
    strcpy(client_data->method, method);
 | 
						|
 | 
						|
    /* check allowed http methods */
 | 
						|
    if (strcasecmp(method, "GET") && strcasecmp(method, "POST"))
 | 
						|
    {
 | 
						|
        // httpd_unimplemented(dcb->fd);
 | 
						|
        return 0;
 | 
						|
    }
 | 
						|
 | 
						|
    i = 0;
 | 
						|
 | 
						|
    while ((j < sizeof(buf)) && ISspace(buf[j]))
 | 
						|
    {
 | 
						|
        j++;
 | 
						|
    }
 | 
						|
 | 
						|
    while ((j < sizeof(buf) - 1) && !ISspace(buf[j]) && (i < sizeof(url) - 1))
 | 
						|
    {
 | 
						|
        url[i] = buf[j];
 | 
						|
        i++;
 | 
						|
        j++;
 | 
						|
    }
 | 
						|
 | 
						|
    url[i] = '\0';
 | 
						|
 | 
						|
    /**
 | 
						|
     * Get the query string if availble
 | 
						|
     */
 | 
						|
 | 
						|
    if (strcasecmp(method, "GET") == 0)
 | 
						|
    {
 | 
						|
        query_string = url;
 | 
						|
        while ((*query_string != '?') && (*query_string != '\0'))
 | 
						|
        {
 | 
						|
            query_string++;
 | 
						|
        }
 | 
						|
        if (*query_string == '?')
 | 
						|
        {
 | 
						|
            *query_string = '\0';
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    /** If listener->authenticator is the default authenticator, it means that
 | 
						|
     * we don't need to check the user credentials. All other authenticators
 | 
						|
     * cause a 401 Unauthorized to be returned on the first try. */
 | 
						|
    bool auth_ok = httpd_default_auth() == std::string(dcb->session->listener->authenticator());
 | 
						|
 | 
						|
    /**
 | 
						|
     * Get the request headers
 | 
						|
     */
 | 
						|
 | 
						|
    while ((numchars > 0) && strcmp("\n", buf))
 | 
						|
    {
 | 
						|
        char* value = NULL;
 | 
						|
        char* end = NULL;
 | 
						|
        numchars = httpd_get_line(dcb->fd, buf, sizeof(buf));
 | 
						|
        if ((value = strchr(buf, ':')))
 | 
						|
        {
 | 
						|
            *value = '\0';
 | 
						|
            value++;
 | 
						|
            end = &value[strlen(value) - 1];
 | 
						|
            *end = '\0';
 | 
						|
 | 
						|
            if (strncasecmp(buf, "Hostname", 6) == 0)
 | 
						|
            {
 | 
						|
                strcpy(client_data->hostname, value);
 | 
						|
            }
 | 
						|
            if (strncasecmp(buf, "useragent", 9) == 0)
 | 
						|
            {
 | 
						|
                strcpy(client_data->useragent, value);
 | 
						|
            }
 | 
						|
 | 
						|
            if (strcmp(buf, "Authorization") == 0)
 | 
						|
            {
 | 
						|
                GWBUF* auth_data = gwbuf_alloc_and_load(strlen(value), value);
 | 
						|
                MXS_OOM_IFNULL(auth_data);
 | 
						|
 | 
						|
                if (auth_data)
 | 
						|
                {
 | 
						|
                    /** The freeing entry point is called automatically when
 | 
						|
                     * the client DCB is closed */
 | 
						|
                    dcb->authfunc.extract(dcb, auth_data);
 | 
						|
                    auth_ok = dcb->authfunc.authenticate(dcb) == MXS_AUTH_SUCCEEDED;
 | 
						|
                    gwbuf_free(auth_data);
 | 
						|
                }
 | 
						|
            }
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    if (numchars)
 | 
						|
    {
 | 
						|
        headers_read = 1;
 | 
						|
        memcpy(&client_data->headers_received, &headers_read, sizeof(int));
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Now begins the server reply
 | 
						|
     */
 | 
						|
 | 
						|
    /* send all the basic headers and close with \r\n */
 | 
						|
    httpd_send_headers(dcb, 1, auth_ok);
 | 
						|
 | 
						|
#if 0
 | 
						|
    /**
 | 
						|
     * ToDO: launch proper content handling based on the requested URI, later REST interface
 | 
						|
     *
 | 
						|
     */
 | 
						|
    if (strcmp(url, "/show") == 0)
 | 
						|
    {
 | 
						|
        if (query_string && strlen(query_string))
 | 
						|
        {
 | 
						|
            if (strcmp(query_string, "dcb") == 0)
 | 
						|
            {
 | 
						|
                dprintAllDCBs(dcb);
 | 
						|
            }
 | 
						|
            if (strcmp(query_string, "session") == 0)
 | 
						|
            {
 | 
						|
                dprintAllSessions(dcb);
 | 
						|
            }
 | 
						|
        }
 | 
						|
    }
 | 
						|
    if (strcmp(url, "/services") == 0)
 | 
						|
    {
 | 
						|
        ResultSet* set, * seviceGetList();
 | 
						|
        if ((set = serviceGetList()) != NULL)
 | 
						|
        {
 | 
						|
            resultset_stream_json(set, dcb);
 | 
						|
            resultset_free(set);
 | 
						|
        }
 | 
						|
    }
 | 
						|
#endif
 | 
						|
    if (auth_ok && (uri = gwbuf_alloc(strlen(url) + 1)) != NULL)
 | 
						|
    {
 | 
						|
        strcpy((char*)GWBUF_DATA(uri), url);
 | 
						|
        gwbuf_set_type(uri, GWBUF_TYPE_HTTP);
 | 
						|
        MXS_SESSION_ROUTE_QUERY(session, uri);
 | 
						|
    }
 | 
						|
 | 
						|
    /* force the client connecton close */
 | 
						|
    dcb_close(dcb);
 | 
						|
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * EPOLLOUT handler for the HTTPD protocol module.
 | 
						|
 *
 | 
						|
 * @param dcb   The descriptor control block
 | 
						|
 * @return
 | 
						|
 */
 | 
						|
static int httpd_write_event(DCB* dcb)
 | 
						|
{
 | 
						|
    return dcb_drain_writeq(dcb);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Write routine for the HTTPD protocol module.
 | 
						|
 *
 | 
						|
 * Writes the content of the buffer queue to the socket
 | 
						|
 * observing the non-blocking principles of the gateway.
 | 
						|
 *
 | 
						|
 * @param dcb   Descriptor Control Block for the socket
 | 
						|
 * @param queue Linked list of buffes to write
 | 
						|
 */
 | 
						|
static int httpd_write(DCB* dcb, GWBUF* queue)
 | 
						|
{
 | 
						|
    int rc;
 | 
						|
    rc = dcb_write(dcb, queue);
 | 
						|
    return rc;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Handler for the EPOLLERR event.
 | 
						|
 *
 | 
						|
 * @param dcb   The descriptor control block
 | 
						|
 */
 | 
						|
static int httpd_error(DCB* dcb)
 | 
						|
{
 | 
						|
    dcb_close(dcb);
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Handler for the EPOLLHUP event.
 | 
						|
 *
 | 
						|
 * @param dcb   The descriptor control block
 | 
						|
 */
 | 
						|
static int httpd_hangup(DCB* dcb)
 | 
						|
{
 | 
						|
    dcb_close(dcb);
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Handler for the EPOLLIN event when the DCB refers to the listening
 | 
						|
 * socket for the protocol.
 | 
						|
 *
 | 
						|
 * @param listener   The descriptor control block
 | 
						|
 */
 | 
						|
static int httpd_accept(DCB* client_dcb)
 | 
						|
{
 | 
						|
    HTTPD_session* client_data = NULL;
 | 
						|
 | 
						|
    /* create the session data for HTTPD */
 | 
						|
    if ((client_data = (HTTPD_session*)MXS_CALLOC(1, sizeof(HTTPD_session))) == NULL)
 | 
						|
    {
 | 
						|
        dcb_close(client_dcb);
 | 
						|
        return 0;
 | 
						|
    }
 | 
						|
    client_dcb->data = client_data;
 | 
						|
 | 
						|
    if (!session_start(client_dcb->session) || poll_add_dcb(client_dcb) == -1)
 | 
						|
    {
 | 
						|
        dcb_close(client_dcb);
 | 
						|
        return 0;
 | 
						|
    }
 | 
						|
 | 
						|
    return 1;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * The close handler for the descriptor. Called by the gateway to
 | 
						|
 * explicitly close a connection.
 | 
						|
 *
 | 
						|
 * @param dcb   The descriptor control block
 | 
						|
 */
 | 
						|
 | 
						|
static int httpd_close(DCB* dcb)
 | 
						|
{
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * HTTPD get line from client
 | 
						|
 */
 | 
						|
static int httpd_get_line(int sock, char* buf, int size)
 | 
						|
{
 | 
						|
    int i = 0;
 | 
						|
    char c = '\0';
 | 
						|
    int n;
 | 
						|
 | 
						|
    while ((i < size - 1) && (c != '\n'))
 | 
						|
    {
 | 
						|
        n = recv(sock, &c, 1, 0);
 | 
						|
        /* DEBUG printf("%02X\n", c); */
 | 
						|
        if (n > 0)
 | 
						|
        {
 | 
						|
            if (c == '\r')
 | 
						|
            {
 | 
						|
                n = recv(sock, &c, 1, MSG_PEEK);
 | 
						|
                /* DEBUG printf("%02X\n", c); */
 | 
						|
                if ((n > 0) && (c == '\n'))
 | 
						|
                {
 | 
						|
                    if (recv(sock, &c, 1, 0) < 0)
 | 
						|
                    {
 | 
						|
                        c = '\n';
 | 
						|
                    }
 | 
						|
                }
 | 
						|
                else
 | 
						|
                {
 | 
						|
                    c = '\n';
 | 
						|
                }
 | 
						|
            }
 | 
						|
            buf[i] = c;
 | 
						|
            i++;
 | 
						|
        }
 | 
						|
        else
 | 
						|
        {
 | 
						|
            c = '\n';
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    buf[i] = '\0';
 | 
						|
 | 
						|
    return i;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * HTTPD send basic headers with 200 OK
 | 
						|
 */
 | 
						|
static void httpd_send_headers(DCB* dcb, int final, bool auth_ok)
 | 
						|
{
 | 
						|
    char date[64] = "";
 | 
						|
    const char* fmt = "%a, %d %b %Y %H:%M:%S GMT";
 | 
						|
    time_t httpd_current_time = time(NULL);
 | 
						|
 | 
						|
    struct tm tm;
 | 
						|
    localtime_r(&httpd_current_time, &tm);
 | 
						|
    strftime(date, sizeof(date), fmt, &tm);
 | 
						|
    const char* response = auth_ok ? "200 OK" : "401 Unauthorized";
 | 
						|
    dcb_printf(dcb,
 | 
						|
               "HTTP/1.1 %s\r\n"
 | 
						|
               "Date: %s\r\n"
 | 
						|
               "Server: %s\r\n"
 | 
						|
               "Connection: close\r\n"
 | 
						|
               "WWW-Authenticate: Basic realm=\"MaxInfo\"\r\n"
 | 
						|
               "Content-Type: application/json\r\n",
 | 
						|
               response,
 | 
						|
               date,
 | 
						|
               HTTP_SERVER_STRING);
 | 
						|
 | 
						|
    /* close the headers */
 | 
						|
    if (final)
 | 
						|
    {
 | 
						|
        dcb_printf(dcb, "\r\n");
 | 
						|
    }
 | 
						|
}
 |