mirror of https://github.com/OISF/suricata
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
101 lines
2.6 KiB
C
101 lines
2.6 KiB
C
11 years ago
|
/* Copyright (C) 2014 Open Information Security Foundation
|
||
|
|
*
|
||
|
|
* You can copy, redistribute or modify this Program under the terms of
|
||
|
|
* the GNU General Public License version 2 as published by the Free
|
||
|
|
* Software Foundation.
|
||
|
|
*
|
||
|
|
* This program is distributed in the hope that it will be useful,
|
||
|
|
* 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.
|
||
|
|
*
|
||
|
|
* You should have received a copy of the GNU General Public License
|
||
|
|
* version 2 along with this program; if not, write to the Free Software
|
||
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
|
||
|
|
* 02110-1301, USA.
|
||
|
|
*/
|
||
|
|
|
||
|
|
|
||
|
|
/**
|
||
|
|
* \file
|
||
|
|
*
|
||
|
|
* \author Eric Leblond <eric@regit.org>
|
||
|
|
*
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include "suricata-common.h"
|
||
|
|
#include "debug.h"
|
||
|
|
#include "detect.h"
|
||
|
|
#include "pkt-var.h"
|
||
|
|
#include "conf.h"
|
||
|
|
|
||
|
|
#include "threads.h"
|
||
|
|
#include "threadvars.h"
|
||
|
|
#include "tm-threads.h"
|
||
|
|
|
||
|
|
#include "util-print.h"
|
||
|
|
#include "util-unittest.h"
|
||
|
|
|
||
|
|
#include "util-debug.h"
|
||
|
|
|
||
|
|
#include "output.h"
|
||
|
|
#include "app-layer-dns-common.h"
|
||
|
|
#include "app-layer.h"
|
||
|
|
#include "app-layer-parser.h"
|
||
|
|
#include "util-privs.h"
|
||
|
|
#include "util-buffer.h"
|
||
|
|
#include "util-proto-name.h"
|
||
|
|
#include "util-logopenfile.h"
|
||
|
|
#include "util-time.h"
|
||
|
|
|
||
|
|
#ifdef HAVE_LUA
|
||
|
|
|
||
|
|
#include <lua.h>
|
||
|
|
#include <lualib.h>
|
||
|
|
#include <lauxlib.h>
|
||
|
|
|
||
|
|
#include "util-lua.h"
|
||
|
|
#include "util-lua-common.h"
|
||
|
|
|
||
|
|
static int DnsGetDnsRrname(lua_State *luastate)
|
||
|
|
{
|
||
|
|
if (!(LuaStateNeedProto(luastate, ALPROTO_DNS)))
|
||
|
|
return LuaCallbackError(luastate, "error: protocol not dns");
|
||
|
|
|
||
|
|
DNSTransaction *tx = LuaStateGetTX(luastate);
|
||
|
|
if (tx == NULL)
|
||
|
|
return LuaCallbackError(luastate, "internal error: no tx");
|
||
|
|
|
||
|
|
DNSQueryEntry *query = NULL;
|
||
|
|
TAILQ_FOREACH(query, &tx->query_list, next) {
|
||
|
|
char *c;
|
||
|
|
size_t input_len;
|
||
|
|
c = BytesToString((uint8_t *)((uint8_t *)query + sizeof(DNSQueryEntry)), query->len);
|
||
|
|
if (c != NULL) {
|
||
|
|
int ret;
|
||
|
|
input_len = strlen(c);
|
||
|
|
/* sanity check */
|
||
|
|
if (input_len > (size_t)(2 * query->len)) {
|
||
|
|
SCFree(c);
|
||
|
|
return LuaCallbackError(luastate, "invalid length");
|
||
|
|
}
|
||
|
|
ret = LuaPushStringBuffer(luastate, (uint8_t *)c, input_len);
|
||
|
|
SCFree(c);
|
||
|
|
return ret;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return LuaCallbackError(luastate, "no query");
|
||
|
|
}
|
||
|
|
|
||
|
|
/** \brief register http lua extensions in a luastate */
|
||
|
|
int LuaRegisterDnsFunctions(lua_State *luastate)
|
||
|
|
{
|
||
|
|
/* registration of the callbacks */
|
||
|
|
lua_pushcfunction(luastate, DnsGetDnsRrname);
|
||
|
|
lua_setglobal(luastate, "DnsGetDnsRrname");
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif /* HAVE_LUA */
|