tango-maat/tools/maat_debug_tool.cpp

#include "Maat_rule.h"
#include "Maat_command.h"
#include <MESA/MESA_handle_logger.h>

#include <MESA/stream.h>
#include <stdio.h>     /* for printf */
#include <stdlib.h>    /* for exit */
#include <getopt.h>
#include <sys/stat.h>
#include <arpa/inet.h>
#include <unistd.h>		/* for sleep*/

void debug_maat_result_print(const char* table_name, int scan_ret, struct Maat_rule_t* result)
{
	printf("Scan table %s ", table_name);
	if(scan_ret==-1)
	{
		printf("error.\n");
	}
	else if(scan_ret==-2)
	{
		printf("hits group, but not compile.\n");
	}
	else if(scan_ret==0)
	{
		printf("not hit.");
	}
	else
	{
		printf("hits ");
		for(int i=0; i<scan_ret; i++)
		{
			printf("%d, ", result[i].config_id);
		}
	}
	printf("\n");

}
int debug_maat_str_scan(Maat_feather_t feather, const char* table_name, const char* district, char* data, size_t sz)
{
	int table_id=0,ret=0;
	int scan_result_max=64;
	struct Maat_rule_t result[scan_result_max];
	scan_status_t mid=NULL;


	table_id=Maat_table_register(feather, table_name);
	if(table_id<0)
	{
		printf("Unkown table %s\n", table_name);
	}
	
	struct Maat_hit_detail_t *hit_detail=(struct Maat_hit_detail_t *)malloc(sizeof(struct Maat_hit_detail_t)*10);

	enum MAAT_CHARSET maat_charset=CHARSET_UTF8;

	if(strlen(district)>0)
	{	
		ret=Maat_set_scan_status(feather, &mid, MAAT_SET_SCAN_DISTRICT, district, strlen(district));
	}
	
	ret=Maat_full_scan_string(feather, table_id, maat_charset, data, sz, 
						result, NULL, scan_result_max,
						&mid, 0);
	debug_maat_result_print(table_name, ret, result);

	free(hit_detail);
	Maat_clean_status(&mid);
	return 0;
}
int debug_maat_ip_scan(Maat_feather_t feather, const char* table_name, const char* sip)
{
	int table_id=0,ret=0;
	int scan_result_max=64;

	struct Maat_rule_t result[scan_result_max];
	scan_status_t mid=NULL;
	struct ipaddr ipv4_addr;
	struct stream_tuple4_v4 v4_addr;
	ipv4_addr.addrtype=ADDR_TYPE_IPV4;
	inet_pton(AF_INET, sip, &(v4_addr.saddr));
	v4_addr.source=htons(5210);
	inet_pton(AF_INET, "127,0.0.1", &(v4_addr.daddr));
	v4_addr.dest=htons(7400);
	ipv4_addr.v4=&v4_addr;


	table_id=Maat_table_register(feather, table_name);
	ret=Maat_scan_proto_addr(feather, table_id, &ipv4_addr, 6, result, scan_result_max, &mid, 0);
	
	debug_maat_result_print(table_name, ret, result);

	Maat_clean_status(&mid);
	return 0;
}

enum tool_arg_type{
	ARG_TABLE_INFO=0,
	ARG_TABLE_NAME,
	ARG_SCAN_FILE,
	ARG_SCAN_STRING,
	ARG_SCAN_DISTRICT,
	ARG_SCAN_CHARSET,
	ARG_INPUT_JSON,
	ARG_INPUT_FULL_INDEX,
	ARG_REDIS_IP,
	ARG_REDIS_PORT,
	ARG_DECRYPT_KEY,
	ARG_ACCEPT_TAGS,
	ARG_SCAN_IPv4,
	ARG_SLEEP_SECONDS,
	__ARG_MAX
};

int main(int argc, char ** argv)
{
	FILE* fp=NULL;
	char* file_buff=NULL;
	size_t file_size=0;
	struct stat file_info;	
	char arg_value[__ARG_MAX][1024];
	memset(arg_value, 0, sizeof(arg_value));
	if(argc<2)
	{
		return 0;
	}
	static struct option long_options[]={
		{"table-info",		required_argument, 0, 0},
		{"table-name",		required_argument, 0, 0},
		{"scan-file",		optional_argument, 0, 0},
		{"scan-string", optional_argument, 0, 0},
		{"scan-district", optional_argument, 0, 0},
		{"scan-charset",	optional_argument, 0, 0},
		{"maat-json",		required_argument, 0, 0},
		{"full-index",	required_argument, 0, 0},
		{"redis-ip",	required_argument, 0, 0},
		{"redis-port",	required_argument, 0, 0},
		{"decrypt-key", optional_argument, 0, 0},
		{"accept-tags", optional_argument, 0, 0},
		{"scan-ipv4", required_argument, 0, 0},	
		{"sleep-seconds", required_argument, 0, 0},	
		{0, 0, 0, 0}
	};

	int c=0;
	while (1) {
		int option_index = 0;


		c = getopt_long(argc, argv, "ab", long_options, &option_index);
		if (c == -1)
			break;
		if(c!=0)
		{
			printf("Invalid parameter, long options only.\n");
		}
		else
		{
			strcpy(arg_value[option_index], optarg);
		}
	}

	const char* log_file="./test.log";
	int scan_detail=0, ret=0;
	Maat_feather_t feather=NULL;
	void *g_logger=NULL;
	int redis_port=6379, sleep_seconds=0;

	g_logger=MESA_create_runtime_log_handle(log_file, 0);

	feather=Maat_feather(4, arg_value[ARG_TABLE_INFO], g_logger);
	Maat_set_feather_opt(feather, MAAT_OPT_INSTANCE_NAME, "debugtool", strlen("debugtool")+1);
	if(strlen(arg_value[ARG_DECRYPT_KEY])>0)
	{
		Maat_set_feather_opt(feather, MAAT_OPT_DECRYPT_KEY, arg_value[ARG_DECRYPT_KEY], strlen(arg_value[ARG_DECRYPT_KEY])+1);
	}
	if(strlen(arg_value[ARG_ACCEPT_TAGS])>0)
	{
		Maat_set_feather_opt(feather, MAAT_OPT_ACCEPT_TAGS, arg_value[ARG_ACCEPT_TAGS], strlen(arg_value[ARG_ACCEPT_TAGS])+1);
	}
	if(strlen(arg_value[ARG_INPUT_JSON])>0)
	{
		ret=Maat_set_feather_opt(feather, MAAT_OPT_JSON_FILE_PATH, arg_value[ARG_INPUT_JSON], strlen(arg_value[ARG_INPUT_JSON])+1);
		if(ret!=0)
		{
			printf("Read %s failed, invalid maat json.\n", arg_value[ARG_INPUT_JSON]);
			ret=-1;
			goto clean_up;
		}
	}
	else if(strlen(arg_value[ARG_INPUT_FULL_INDEX]))
	{
		ret=Maat_set_feather_opt(feather, MAAT_OPT_FULL_CFG_DIR, arg_value[ARG_INPUT_FULL_INDEX], strlen(arg_value[ARG_INPUT_FULL_INDEX])+1);
		if(ret!=0)
		{
			printf("Set %s failed, invalid full cfg directory.\n", arg_value[ARG_INPUT_FULL_INDEX]);
			ret=-1;
			goto clean_up;
		}
	}
	else if(strlen(arg_value[ARG_REDIS_IP]))
	{
		ret=Maat_set_feather_opt(feather, MAAT_OPT_REDIS_IP, arg_value[ARG_REDIS_IP], strlen(arg_value[ARG_REDIS_IP])+1);
		if(ret!=0)
		{
			printf("Set %s failed, redis IP.\n", arg_value[ARG_REDIS_IP]);
			ret=-1;
			goto clean_up;
		}
		if(strlen(arg_value[ARG_REDIS_PORT]))
		{
			redis_port=atoi(arg_value[ARG_REDIS_PORT]);
		}
		else
		{
			redis_port=6379;
		}
		ret=Maat_set_feather_opt(feather, MAAT_OPT_REDIS_PORT, &redis_port, sizeof(redis_port));
	}
	else
	{
		printf("Error: One of --%s and --%s should be specified.\n", long_options[ARG_INPUT_JSON].name, long_options[ARG_INPUT_FULL_INDEX].name);
		ret=-1;
		goto clean_up;
	}
	Maat_set_feather_opt(feather, MAAT_OPT_SCAN_DETAIL, &scan_detail, sizeof(scan_detail));
	Maat_initiate_feather(feather);

	if(strlen(arg_value[ARG_SCAN_FILE])>0)
	{
	
		if(stat(arg_value[ARG_SCAN_FILE], &file_info)<0)
		{		
			printf("Stat %s failed.\n", arg_value[ARG_SCAN_FILE]);
			goto clean_up;
		}
		file_size=file_info.st_size;
		fp=fopen(arg_value[ARG_SCAN_FILE], "r");
		if(fp==NULL)
		{
			printf("Open %s failed.\n", arg_value[ARG_SCAN_FILE]);
			goto clean_up;
		}
		file_buff=(char*)malloc(file_size);
		fread(file_buff,1,file_size,fp);
		fclose(fp);
		
		debug_maat_str_scan(feather, arg_value[ARG_TABLE_NAME], arg_value[ARG_SCAN_DISTRICT], file_buff, file_size);
		free(file_buff);
	}
	if(strlen(arg_value[ARG_SCAN_IPv4])>0)
	{
		debug_maat_ip_scan(feather, arg_value[ARG_TABLE_NAME], arg_value[ARG_SCAN_IPv4]);
	}
	if(strlen(arg_value[ARG_SLEEP_SECONDS])>0)
	{
		sleep_seconds=atoi(arg_value[ARG_SLEEP_SECONDS]);
		sleep(sleep_seconds);
	}
clean_up:
	Maat_burn_feather(feather);
	MESA_destroy_runtime_log_handle(g_logger);
	return ret;
}