httpd.c
上传用户:kjfoods
上传日期:2020-07-06
资源大小:29949k
文件大小:13k
源码类别:

midi

开发平台:

Unix_Linux

  1. /*****************************************************************************
  2.  * httpd.c: HTTPd wrapper
  3.  *****************************************************************************
  4.  * Copyright (C) 2007-2008 the VideoLAN team
  5.  * $Id: 09ca8a4ec47e8cacb5eea92c09d842814091b789 $
  6.  *
  7.  * Authors: Antoine Cellerier <dionoea at videolan tod org>
  8.  *
  9.  * This program is free software; you can redistribute it and/or modify
  10.  * it under the terms of the GNU General Public License as published by
  11.  * the Free Software Foundation; either version 2 of the License, or
  12.  * (at your option) any later version.
  13.  *
  14.  * This program is distributed in the hope that it will be useful,
  15.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17.  * GNU General Public License for more details.
  18.  *
  19.  * You should have received a copy of the GNU General Public License
  20.  * along with this program; if not, write to the Free Software
  21.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  22.  *****************************************************************************/
  23. /*****************************************************************************
  24.  * Preamble
  25.  *****************************************************************************/
  26. #ifndef  _GNU_SOURCE
  27. #   define  _GNU_SOURCE
  28. #endif
  29. #ifdef HAVE_CONFIG_H
  30. # include "config.h"
  31. #endif
  32. #include <vlc_common.h>
  33. #include <vlc_httpd.h>
  34. #include <lua.h>        /* Low level lua C API */
  35. #include <lauxlib.h>    /* Higher level C API */
  36. #include <lualib.h>     /* Lua libs */
  37. #include "../vlc.h"
  38. #include "../libs.h"
  39. /*****************************************************************************
  40.  * Local prototypes
  41.  *****************************************************************************/
  42. static uint8_t *vlclua_todata( lua_State *L, int narg, int *i_data );
  43. static int vlclua_httpd_host_delete( lua_State * );
  44. static int vlclua_httpd_handler_new( lua_State * );
  45. static int vlclua_httpd_handler_delete( lua_State * );
  46. static int vlclua_httpd_file_new( lua_State * );
  47. static int vlclua_httpd_file_delete( lua_State * );
  48. static int vlclua_httpd_redirect_new( lua_State * );
  49. static int vlclua_httpd_redirect_delete( lua_State * );
  50. /*****************************************************************************
  51.  * HTTPD Host
  52.  *****************************************************************************/
  53. static const luaL_Reg vlclua_httpd_reg[] = {
  54.     { "handler", vlclua_httpd_handler_new },
  55.     { "file", vlclua_httpd_file_new },
  56.     { "redirect", vlclua_httpd_redirect_new },
  57.     { NULL, NULL }
  58. };
  59. static int vlclua_httpd_tls_host_new( lua_State *L )
  60. {
  61.     vlc_object_t *p_this = vlclua_get_this( L );
  62.     const char *psz_host = luaL_checkstring( L, 1 );
  63.     int i_port = luaL_checkint( L, 2 );
  64.     const char *psz_cert = luaL_optstring( L, 3, NULL );
  65.     const char *psz_key = luaL_optstring( L, 4, NULL );
  66.     const char *psz_ca = luaL_optstring( L, 5, NULL );
  67.     const char *psz_crl = luaL_optstring( L, 6, NULL );
  68.     httpd_host_t *p_host = httpd_TLSHostNew( p_this, psz_host, i_port,
  69.                                                    psz_cert, psz_key,
  70.                                                    psz_ca, psz_crl );
  71.     if( !p_host )
  72.         return luaL_error( L, "Failed to create HTTP TLS host "%s:%d" "
  73.                            "(cert: "%s", key: "%s", ca: "%s", "
  74.                            "crl: "%s").", psz_host, i_port,
  75.                            psz_cert, psz_key, psz_ca, psz_crl );
  76.     httpd_host_t **pp_host = lua_newuserdata( L, sizeof( httpd_host_t * ) );
  77.     *pp_host = p_host;
  78.     if( luaL_newmetatable( L, "httpd_host" ) )
  79.     {
  80.         lua_newtable( L );
  81.         luaL_register( L, NULL, vlclua_httpd_reg );
  82.         lua_setfield( L, -2, "__index" );
  83.         lua_pushcfunction( L, vlclua_httpd_host_delete );
  84.         lua_setfield( L, -2, "__gc" );
  85.     }
  86.     lua_setmetatable( L, -2 );
  87.     return 1;
  88. }
  89. static int vlclua_httpd_host_delete( lua_State *L )
  90. {
  91.     httpd_host_t **pp_host = (httpd_host_t **)luaL_checkudata( L, 1, "httpd_host" );
  92.     httpd_HostDelete( *pp_host );
  93.     return 0;
  94. }
  95. /*****************************************************************************
  96.  * HTTPd Handler
  97.  *****************************************************************************/
  98. struct httpd_handler_sys_t
  99. {
  100.     lua_State *L;
  101.     int ref;
  102. };
  103. static int vlclua_httpd_handler_callback(
  104.      httpd_handler_sys_t *p_sys, httpd_handler_t *p_handler, char *psz_url,
  105.      uint8_t *psz_request, int i_type, uint8_t *p_in, int i_in,
  106.      char *psz_remote_addr, char *psz_remote_host,
  107.      uint8_t **pp_data, int *pi_data )
  108. {
  109.     VLC_UNUSED(p_handler);
  110.     lua_State *L = p_sys->L;
  111.     /* function data */
  112.     lua_pushvalue( L, 1 );
  113.     lua_pushvalue( L, 2 );
  114.     /* function data function data */
  115.     lua_pushstring( L, psz_url );
  116.     /* function data function data url */
  117.     lua_pushstring( L, (const char *)psz_request );
  118.     /* function data function data url request */
  119.     lua_pushinteger( L, i_type ); /* Q: what does i_type stand for? */
  120.     /* function data function data url request type */
  121.     lua_pushlstring( L, (const char *)p_in, i_in ); /* Q: what do p_in contain? */
  122.     /* function data function data url request type in */
  123.     lua_pushstring( L, psz_remote_addr );
  124.     /* function data function data url request type in addr */
  125.     lua_pushstring( L, psz_remote_host );
  126.     /* function data function data url request type in addr host */
  127.     if( lua_pcall( L, 7, 1, 0 ) )
  128.     {
  129.         /* function data err */
  130.         vlc_object_t *p_this = vlclua_get_this( L );
  131.         const char *psz_err = lua_tostring( L, -1 );
  132.         msg_Err( p_this, "Error while runing the lua HTTPd handler "
  133.                  "callback: %s", psz_err );
  134.         lua_settop( L, 2 );
  135.         /* function data */
  136.         return VLC_EGENERIC;
  137.     }
  138.     /* function data outdata */
  139.     *pp_data = vlclua_todata( L, -1, pi_data );
  140.     lua_pop( L, 1 );
  141.     /* function data */
  142.     return VLC_SUCCESS;
  143. }
  144. static int vlclua_httpd_handler_new( lua_State * L )
  145. {
  146.     httpd_host_t **pp_host = (httpd_host_t **)luaL_checkudata( L, 1, "httpd_host" );
  147.     const char *psz_url = luaL_checkstring( L, 2 );
  148.     const char *psz_user = luaL_nilorcheckstring( L, 3 );
  149.     const char *psz_password = luaL_nilorcheckstring( L, 4 );
  150.     const vlc_acl_t *p_acl = lua_isnil( L, 5 ) ? NULL : luaL_checkudata( L, 5, "acl" );
  151.     /* Stack item 6 is the callback function */
  152.     luaL_argcheck( L, lua_isfunction( L, 6 ), 6, "Should be a function" );
  153.     /* Stack item 7 is the callback data */
  154.     lua_settop( L, 7 );
  155.     httpd_handler_sys_t *p_sys = (httpd_handler_sys_t*)
  156.                                  malloc( sizeof( httpd_handler_sys_t ) );
  157.     if( !p_sys )
  158.         return luaL_error( L, "Failed to allocate private buffer." );
  159.     p_sys->L = lua_newthread( L );
  160.     p_sys->ref = luaL_ref( L, LUA_REGISTRYINDEX ); /* pops the object too */
  161.     /* use lua_xmove to move the lua callback function and data to
  162.      * the callback's stack. */
  163.     lua_xmove( L, p_sys->L, 2 );
  164.     httpd_handler_t *p_handler = httpd_HandlerNew(
  165.                             *pp_host, psz_url, psz_user, psz_password,
  166.                             p_acl, vlclua_httpd_handler_callback, p_sys );
  167.     if( !p_handler )
  168.     {
  169.         free( p_sys );
  170.         return luaL_error( L, "Failed to create HTTPd handler." );
  171.     }
  172.     httpd_handler_t **pp_handler = lua_newuserdata( L, sizeof( httpd_handler_t * ) );
  173.     *pp_handler = p_handler;
  174.     if( luaL_newmetatable( L, "httpd_handler" ) )
  175.     {
  176.         lua_pushcfunction( L, vlclua_httpd_handler_delete );
  177.         lua_setfield( L, -2, "__gc" );
  178.     }
  179.     lua_setmetatable( L, -2 );
  180.     return 1;
  181. }
  182. static int vlclua_httpd_handler_delete( lua_State *L )
  183. {
  184.     httpd_handler_t **pp_handler = (httpd_handler_t**)luaL_checkudata( L, 1, "httpd_handler" );
  185.     httpd_handler_sys_t *p_sys = httpd_HandlerDelete( *pp_handler );
  186.     luaL_unref( p_sys->L, LUA_REGISTRYINDEX, p_sys->ref );
  187.     free( p_sys );
  188.     return 0;
  189. }
  190. /*****************************************************************************
  191.  * HTTPd File
  192.  *****************************************************************************/
  193. struct httpd_file_sys_t
  194. {
  195.     lua_State *L;
  196.     int ref;
  197. };
  198. static int vlclua_httpd_file_callback(
  199.     httpd_file_sys_t *p_sys, httpd_file_t *p_file, uint8_t *psz_request,
  200.     uint8_t **pp_data, int *pi_data )
  201. {
  202.     VLC_UNUSED(p_file);
  203.     lua_State *L = p_sys->L;
  204.     /* function data */
  205.     lua_pushvalue( L, 1 );
  206.     lua_pushvalue( L, 2 );
  207.     /* function data function data */
  208.     lua_pushstring( L, (const char *)psz_request );
  209.     /* function data function data request */
  210.     if( lua_pcall( L, 2, 1, 0 ) )
  211.     {
  212.         /* function data err */
  213.         vlc_object_t *p_this = vlclua_get_this( L );
  214.         const char *psz_err = lua_tostring( L, -1 );
  215.         msg_Err( p_this, "Error while runing the lua HTTPd file callback: %s",
  216.                  psz_err );
  217.         lua_settop( L, 2 );
  218.         /* function data */
  219.         return VLC_EGENERIC;
  220.     }
  221.     /* function data outdata */
  222.     *pp_data = vlclua_todata( L, -1, pi_data );
  223.     lua_pop( L, 1 );
  224.     /* function data */
  225.     return VLC_SUCCESS;
  226. }
  227. static int vlclua_httpd_file_new( lua_State *L )
  228. {
  229.     httpd_host_t **pp_host = (httpd_host_t **)luaL_checkudata( L, 1, "httpd_host" );
  230.     const char *psz_url = luaL_checkstring( L, 2 );
  231.     const char *psz_mime = luaL_nilorcheckstring( L, 3 );
  232.     const char *psz_user = luaL_nilorcheckstring( L, 4 );
  233.     const char *psz_password = luaL_nilorcheckstring( L, 5 );
  234.     const vlc_acl_t *p_acl = lua_isnil( L, 6 ) ? NULL : luaL_checkudata( L, 6, "acl" );
  235.     /* Stack item 7 is the callback function */
  236.     luaL_argcheck( L, lua_isfunction( L, 7 ), 7, "Should be a function" );
  237.     /* Stack item 8 is the callback data */
  238.     httpd_file_sys_t *p_sys = (httpd_file_sys_t *)
  239.                               malloc( sizeof( httpd_file_sys_t ) );
  240.     if( !p_sys )
  241.         return luaL_error( L, "Failed to allocate private buffer." );
  242.     p_sys->L = lua_newthread( L );
  243.     p_sys->ref = luaL_ref( L, LUA_REGISTRYINDEX ); /* pops the object too */
  244.     lua_xmove( L, p_sys->L, 2 );
  245.     httpd_file_t *p_file = httpd_FileNew( *pp_host, psz_url, psz_mime,
  246.                                           psz_user, psz_password, p_acl,
  247.                                           vlclua_httpd_file_callback, p_sys );
  248.     if( !p_file )
  249.     {
  250.         free( p_sys );
  251.         return luaL_error( L, "Failed to create HTTPd file." );
  252.     }
  253.     httpd_file_t **pp_file = lua_newuserdata( L, sizeof( httpd_file_t * ) );
  254.     *pp_file = p_file;
  255.     if( luaL_newmetatable( L, "httpd_file" ) )
  256.     {
  257.         lua_pushcfunction( L, vlclua_httpd_file_delete );
  258.         lua_setfield( L, -2, "__gc" );
  259.     }
  260.     lua_setmetatable( L, -2 );
  261.     return 1;
  262. }
  263. static int vlclua_httpd_file_delete( lua_State *L )
  264. {
  265.     httpd_file_t **pp_file = (httpd_file_t**)luaL_checkudata( L, 1, "httpd_file" );
  266.     httpd_file_sys_t *p_sys = httpd_FileDelete( *pp_file );
  267.     luaL_unref( p_sys->L, LUA_REGISTRYINDEX, p_sys->ref );
  268.     free( p_sys );
  269.     return 0;
  270. }
  271. /*****************************************************************************
  272.  * HTTPd Redirect
  273.  *****************************************************************************/
  274. static int vlclua_httpd_redirect_new( lua_State *L )
  275. {
  276.     httpd_host_t **pp_host = (httpd_host_t **)luaL_checkudata( L, 1, "httpd_host" );
  277.     const char *psz_url_dst = luaL_checkstring( L, 2 );
  278.     const char *psz_url_src = luaL_checkstring( L, 3 );
  279.     httpd_redirect_t *p_redirect = httpd_RedirectNew( *pp_host,
  280.                                                       psz_url_dst,
  281.                                                       psz_url_src );
  282.     if( !p_redirect )
  283.         return luaL_error( L, "Failed to create HTTPd redirect." );
  284.     httpd_redirect_t **pp_redirect = lua_newuserdata( L, sizeof( httpd_redirect_t * ) );
  285.     *pp_redirect = p_redirect;
  286.     if( luaL_newmetatable( L, "httpd_redirect" ) )
  287.     {
  288.         lua_pushcfunction( L, vlclua_httpd_redirect_delete );
  289.         lua_setfield( L, -2, "__gc" );
  290.     }
  291.     lua_setmetatable( L, -2 );
  292.     return 1;
  293. }
  294. static int vlclua_httpd_redirect_delete( lua_State *L )
  295. {
  296.     httpd_redirect_t **pp_redirect = (httpd_redirect_t**)luaL_checkudata( L, 1, "httpd_redirect" );
  297.     httpd_RedirectDelete( *pp_redirect );
  298.     return 0;
  299. }
  300. /*****************************************************************************
  301.  * Utils
  302.  *****************************************************************************/
  303. static uint8_t *vlclua_todata( lua_State *L, int narg, int *pi_data )
  304. {
  305.     size_t i_data;
  306.     const char *psz_data = lua_tolstring( L, narg, &i_data );
  307.     uint8_t *p_data = (uint8_t*)malloc( i_data * sizeof(uint8_t) );
  308.     *pi_data = (int)i_data;
  309.     if( !p_data )
  310.     {
  311.         luaL_error( L, "Error while allocating buffer." );
  312.         return NULL; /* To please gcc even though luaL_error longjmp-ed out of here */
  313.     }
  314.     memcpy( p_data, psz_data, i_data );
  315.     return p_data;
  316. }
  317. /*****************************************************************************
  318.  *
  319.  *****************************************************************************/
  320. void luaopen_httpd( lua_State *L )
  321. {
  322.     lua_pushcfunction( L, vlclua_httpd_tls_host_new );
  323.     lua_setfield( L, -2, "httpd" );
  324. }