Mirror/zeek
1
0
Fork 0
mirror of https://github.com/zeek/zeek.git synced 2025-10-16 21:48:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Initial import of svn+ssh:://svn.icir.org/bro/trunk/bro as of r7088

Browse source
This commit is contained in:
Robin Sommer 2010-09-27 20:42:30 -07:00
commit 61757ac78b
1383 changed files with 380824 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

95
src/bif_arg.cc Normal file
View file

@ -0,0 +1,95 @@
// $Id: bif_arg.cc 3234 2006-06-08 02:38:11Z vern $
#include "config.h"
#include <set>
#include <string>
using namespace std;
#include <string.h>
#include "bif_arg.h"
static struct {
const char* bif_type;
const char* bro_type;
const char* c_type;
const char* accessor;
const char* constructor;
} builtin_func_arg_type[] = {
#define DEFINE_BIF_TYPE(id, bif_type, bro_type, c_type, accessor, constructor) \
{bif_type, bro_type, c_type, accessor, constructor},
#include "bif_type.def"
#undef DEFINE_BIF_TYPE
};
extern const char* arg_list_name;
extern set<string> enum_types;
BuiltinFuncArg::BuiltinFuncArg(const char* arg_name, int arg_type)
{
name = arg_name;
type = arg_type;
type_str = "";
}
BuiltinFuncArg::BuiltinFuncArg(const char* arg_name, const char* arg_type_str)
{
name = arg_name;
type = TYPE_OTHER;
type_str = arg_type_str;
for ( int i = 0; builtin_func_arg_type[i].bif_type[0] != '\0'; ++i )
if ( ! strcmp(builtin_func_arg_type[i].bif_type, arg_type_str) )
{
type = i;
type_str = "";
}
if ( enum_types.find(type_str) != enum_types.end() )
type = TYPE_ENUM;
}
void BuiltinFuncArg::PrintBro(FILE* fp)
{
fprintf(fp, "%s: %s%s", name, builtin_func_arg_type[type].bro_type, type_str);
}
void BuiltinFuncArg::PrintCDef(FILE* fp, int n)
{
fprintf(fp,
"\t%s %s = (%s) (",
builtin_func_arg_type[type].c_type,
name,
builtin_func_arg_type[type].c_type);
char buf[1024];
snprintf(buf, sizeof(buf), "(*%s)[%d]", arg_list_name, n);
// Print the accessor expression.
fprintf(fp, builtin_func_arg_type[type].accessor, buf);
fprintf(fp, ");\n");
}
void BuiltinFuncArg::PrintCArg(FILE* fp, int n)
{
const char* ctype = builtin_func_arg_type[type].c_type;
char buf[1024];
if ( type == TYPE_ENUM )
{
snprintf(buf, sizeof(buf),
builtin_func_arg_type[type].c_type, type_str);
ctype = buf;
}
fprintf(fp, "%s %s", ctype, name);
}
void BuiltinFuncArg::PrintBroValConstructor(FILE* fp)
{
if ( type == TYPE_ENUM )
fprintf(fp, builtin_func_arg_type[type].constructor,
name, type_str);
else
fprintf(fp, builtin_func_arg_type[type].constructor, name);
}