webnet/module/wn_module_asp.c

/*
 * File      : wn_module_asp.c
 * This file is part of RT-Thread RTOS
 * COPYRIGHT (C) 2006 - 2018, RT-Thread Development Team
 *
 * This software is dual-licensed: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation. For the terms of this
 * license, see <http://www.gnu.org/licenses/>.
 *
 * You are free to use this software under the terms of the GNU General
 * Public License, but WITHOUT ANY WARRANTY; without even the implied
 * warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU General Public License for more details.
 *
 * Alternatively for commercial application, you can contact us
 * by email <business@rt-thread.com> for commercial license.
 *
 * Change Logs:
 * Date           Author       Notes
 * 2011-08-02     Bernard      the first version
 */

#include <string.h>
#include <rtthread.h>

#ifdef RT_USING_DFS
#if RT_VER_NUM >= 0x40100
#include <stdio.h> /* fix SEEK_END */
#include <fcntl.h> /* fix O_RDONLY */
#else
#include <dfs_posix.h>
#endif /*RT_VER_NUM >= 0x40100*/
#endif

#ifdef RT_USING_SAL
#include <sys/socket.h>
#else
#include <lwip/netif.h>
#endif /* RT_USING_SAL */

#include <webnet.h>
#include <wn_module.h>
#include <wn_utils.h>

#ifdef WEBNET_USING_ASP

struct webnet_asp_variable
{
    char* name;
    void (*handler)(struct webnet_session* session);
};
static struct webnet_asp_variable* _webnet_asp_vars = RT_NULL;
rt_uint32_t _webnet_asp_vars_count = 0;

void webnet_asp_add_var(const char* name, void (*handler)(struct webnet_session* session))
{
    if (_webnet_asp_vars == RT_NULL)
    {
        _webnet_asp_vars_count = 1;
        _webnet_asp_vars = (struct webnet_asp_variable*)wn_malloc (sizeof(struct webnet_asp_variable) *
                           _webnet_asp_vars_count);
    }
    else
    {
        _webnet_asp_vars_count += 1;
        _webnet_asp_vars = (struct webnet_asp_variable*) wn_realloc (_webnet_asp_vars, sizeof(struct webnet_asp_variable) *
                           _webnet_asp_vars_count);
    }

    RT_ASSERT(_webnet_asp_vars != RT_NULL);

    _webnet_asp_vars[_webnet_asp_vars_count - 1].name = wn_strdup(name);
    _webnet_asp_vars[_webnet_asp_vars_count - 1].handler = handler;
}
RTM_EXPORT(webnet_asp_add_var);

static void _default_asp_handler(struct webnet_session* session, const char* name)
{
    if (strncmp(name, "VERION", 6) == 0)
    {
        webnet_session_printf(session, "WebNet 1.0.0");
    }
    else if (strncmp(name, "REMOTE_ADDR", 11) == 0)
    {
        webnet_session_printf(session, "%s", inet_ntoa(session->cliaddr.sin_addr));
    }
    else if (strncmp(name, "REMOTE_PORT", 11) == 0)
    {
        webnet_session_printf(session, "%d", ntohs(session->cliaddr.sin_port));
    }
    else if (strncmp(name, "SERVER_ADDR", 11) == 0)
    {
#if defined(RT_USING_LWIP) && !defined(RT_USING_SAL)
        webnet_session_printf(session, "%s", inet_ntoa(*(struct in_addr*)&(netif_default->ip_addr)));
#endif
    }
    else if (strncmp(name, "SERVER_PORT", 11) == 0)
    {
        webnet_session_printf(session, "%d", WEBNET_PORT);
    }
    else if (strncmp(name, "DOCUMENT_ROOT", 13) == 0)
    {
        webnet_session_printf(session, "%s", webnet_get_root());
    }
    else if (strncmp(name, "SERVER", 6) == 0)
    {
        webnet_session_printf(session, "RT-Thread/WebNet");
    }
    else if (strncmp(name, "HOST", 4) == 0)
    {
        webnet_session_printf(session, "WebNet");
    }
    else if (strncmp(name, "DATE", 4) == 0)
    {
        webnet_session_printf(session, "2011/08/01");
    }
    else if (strncmp(name, "USER_AGENT", 10) == 0)
    {
        webnet_session_printf(session, "%s", session->request->user_agent);
    }
    else if (strncmp(name, "COOKIE", 6) == 0)
    {
        webnet_session_printf(session, "%s", session->request->cookie);
    }
    else if (strncmp(name, "MEMUSAGE", 8) == 0)
    {
        rt_uint32_t total, used, max_used;

        //rt_memory_info(&total, &used, &max_used);
        total = 1024*32;
        used = 1024*16;
        max_used = 1024*24;

        webnet_session_printf(session, "current %d/maximal used %d/total %d", used, max_used, total);
    }
    else if (strncmp(name, "TICK", 4) == 0)
    {
        rt_uint32_t tick;
        rt_uint32_t hour, min, second;

        tick = rt_tick_get()/RT_TICK_PER_SECOND;
        second = tick % 60;
        min    = (tick/60) % 60;
        hour   = tick / (60 * 60);

        webnet_session_printf(session, "%02d:%02d:%02d", hour, min, second);
    }
}

static void _webnet_asp_dofile(struct webnet_session* session, int fd)
{
    char *asp_begin, *asp_end;
    char *offset, *end;
    char *buffer;
    rt_uint32_t length, index;

    /* get file length */
    length = lseek(fd, 0, SEEK_END);
    lseek(fd, 0, SEEK_SET);

    /* allocate read buffer */
    buffer = (char*) wn_malloc (length);
    if (buffer == RT_NULL)
    {
        session->request->result_code = 500;
        return;
    }

    /* write page header */
    webnet_session_set_header(session, "text/html", 200, "OK", -1);
    /* read file to buffer */
    if (read(fd, buffer, (size_t)length) != length) /* read failed */
    {
        wn_free(buffer);
        session->request->result_code = 500;
        return;
    }

    offset = buffer;
    end = buffer + length;
    while (offset < end)
    {
        /* get beginning of asp variable */
        asp_begin = strstr(offset, "<%");
        if (asp_begin == RT_NULL)
        {
            /* write content directly */
            webnet_session_write(session, (const rt_uint8_t*)offset, end - offset);
            break;
        }

        /* get end of aps variable */
        asp_end = strstr(asp_begin, "%>");
        if (asp_end == RT_NULL)
        {
            /* write content directly */
            webnet_session_write(session, (const rt_uint8_t*)offset, end - offset);
            break;
        }
        else
        {
            /* write content */
            webnet_session_write(session, (const rt_uint8_t*)offset, asp_begin - offset);

            offset = asp_begin + 2;
            while ((*offset == ' ') || (*offset == '\t')) offset ++;

            /* extrace asp variable */
            for (index = 0; index < _webnet_asp_vars_count; index ++)
            {
                if (str_begin_with(offset, _webnet_asp_vars[index].name))
                {
                    /* found asp variable */
                    _webnet_asp_vars[index].handler(session);
                    break;
                }
            }
            if (index == _webnet_asp_vars_count)
            {
                /* use default handler */
                _default_asp_handler(session, offset);
            }

            /* move to the end of asp varaialbe */
            offset = asp_end + 2;
        }
    }

    /* release read buffer */
    wn_free(buffer);
}

int webnet_module_asp(struct webnet_session* session, int event)
{
    struct webnet_request* request;

    RT_ASSERT(session != RT_NULL);
    request = session->request;
    RT_ASSERT(request != RT_NULL);

    if (event == WEBNET_EVENT_URI_POST)
    {
        int fd;

        /* check whether a asp file */
        if ((strstr(request->path, ".asp") != RT_NULL) ||
                (strstr(request->path, ".ASP") != RT_NULL))
        {
            /* try to open this file */

            fd = open(request->path, O_RDONLY, 0);
            if ( fd >= 0)
            {
                _webnet_asp_dofile(session, fd);
                close(fd);
                return WEBNET_MODULE_FINISHED;
            }
            else
            {
                /* no this file */
                request->result_code = 404;
                return WEBNET_MODULE_FINISHED;
            }
        }
        else
        {
            /* try index.asp */
            char *asp_filename;

            asp_filename = (char*) wn_malloc (WEBNET_PATH_MAX);
            if (asp_filename != RT_NULL)
            {
                rt_snprintf(asp_filename, WEBNET_PATH_MAX, "%s/index.asp", request->path);
                fd = open(asp_filename, O_RDONLY, 0);

                if (fd >= 0)
                {
                    wn_free(asp_filename);
                    _webnet_asp_dofile(session, fd);
                    close(fd);
                    return WEBNET_MODULE_FINISHED;
                }
            }
            wn_free(asp_filename);
        }
    }

    return WEBNET_MODULE_CONTINUE;
}

#endif /* WEBNET_USING_ASP */