filter_options.cpp

#include "filter_options.h"
#include "options.h"


cOptionsFilter::cOptionsFilter(const char *filter) {
	setFilter(filter);
}

void cOptionsFilter::setFilter(const char *filter) {
	JsonItem jsonData;
	jsonData.parse(filter);
	map<string, string> filterData;
	for(unsigned int i = 0; i < jsonData.getLocalCount(); i++) {
		JsonItem *item = jsonData.getLocalItem(i);
		string filterTypeName = item->getLocalName();
		string filterValue = item->getLocalValue();
		if(filterValue.empty()) {
			continue;
		}
		filterData[filterTypeName] = filterValue;
	}
	if(!filterData["ip_src"].empty() &&
	   filterData["ip_src_dst_type"] == "0") {
		cRecordFilterItem_IP *filter1 =  new cRecordFilterItem_IP(this, of_ip_src);
		filter1->addWhite(filterData["ip_src"].c_str());
		cRecordFilterItem_IP *filter2 = new cRecordFilterItem_IP(this, of_ip_dst);
		filter2->addWhite(filterData["ip_src"].c_str());
		addFilter(filter1, filter2);
	} else {
		cRecordFilterItems gItems(cRecordFilterItems::_and);
		if(!filterData["ip_src"].empty()) {
			cRecordFilterItem_IP *filter = new cRecordFilterItem_IP(this, of_ip_src);
			filter->addWhite(filterData["ip_src"].c_str());
			gItems.addFilter(filter);
		}
		if(!filterData["ip_dst"].empty()) {
			cRecordFilterItem_IP *filter = new cRecordFilterItem_IP(this, of_ip_dst);
			filter->addWhite(filterData["ip_dst"].c_str());
			gItems.addFilter(filter);
		}
		if(gItems.isSet()) {
			addFilter(&gItems);
		}
	}
	if(!filterData["number_src"].empty() &&
	   filterData["number_src_dst_type"] == "0") {
		cRecordFilterItem_CheckString *filter1 = new cRecordFilterItem_CheckString(this, of_number_src);
		filter1->addWhite(filterData["number_src"].c_str());
		cRecordFilterItem_CheckString *filter2 = new cRecordFilterItem_CheckString(this, of_number_dst);
		filter2->addWhite(filterData["number_src"].c_str());
		addFilter(filter1, filter2);
	} else {
		cRecordFilterItems gItems(cRecordFilterItems::_and);
		if(!filterData["number_src"].empty()) {
			cRecordFilterItem_CheckString *filter = new cRecordFilterItem_CheckString(this, of_number_src);
			filter->addWhite(filterData["number_src"].c_str());
			gItems.addFilter(filter);
		}
		if(!filterData["number_dst"].empty()) {
			cRecordFilterItem_CheckString *filter = new cRecordFilterItem_CheckString(this, of_number_dst);
			filter->addWhite(filterData["number_dst"].c_str());
			gItems.addFilter(filter);
		}
		if(gItems.isSet()) {
			addFilter(&gItems);
		}
	}
	if(!filterData["domain_src"].empty() &&
	   filterData["domain_src_dst_type"] == "0") {
		cRecordFilterItem_CheckString *filter1 = new cRecordFilterItem_CheckString(this, of_domain_src);
		filter1->addWhite(filterData["domain_src"].c_str());
		cRecordFilterItem_CheckString *filter2 = new cRecordFilterItem_CheckString(this, of_domain_dst);
		filter2->addWhite(filterData["domain_src"].c_str());
		addFilter(filter1, filter2);
	} else {
		cRecordFilterItems gItems(cRecordFilterItems::_and);
		if(!filterData["domain_src"].empty()) {
			cRecordFilterItem_CheckString *filter = new cRecordFilterItem_CheckString(this, of_domain_src);
			filter->addWhite(filterData["domain_src"].c_str());
			gItems.addFilter(filter);
		}
		if(!filterData["domain_dst"].empty()) {
			cRecordFilterItem_CheckString *filter = new cRecordFilterItem_CheckString(this, of_domain_dst);
			filter->addWhite(filterData["domain_dst"].c_str());
			gItems.addFilter(filter);
		}
		if(gItems.isSet()) {
			addFilter(&gItems);
		}
	}
	if(atoi(filterData["qualify_state"].c_str()) == 1) {
		cRecordFilterItem_numList *filter = new cRecordFilterItem_numList(this, of_qualify_ok);
		filter->addNum(1);
		addFilter(filter);
	} else if(atoi(filterData["qualify_state"].c_str()) == 2) {
		cRecordFilterItem_numList *filter = new cRecordFilterItem_numList(this, of_qualify_ok);
		filter->addNum(0);
		filter->addNum(-1);
		addFilter(filter);
	}
	if(!filterData["response_time_ge"].empty()) {
		cRecordFilterItem_numInterval *filter = new cRecordFilterItem_numInterval(this, of_response_time_ms, atol(filterData["response_time_ge"].c_str()), cRecordFilterItem_base::_ge);
		addFilter(filter);
	}
	if(!filterData["response_time_lt"].empty()) {
		cRecordFilterItem_numInterval *filter1 = new cRecordFilterItem_numInterval(this, of_response_time_ms, atol(filterData["response_time_lt"].c_str()), cRecordFilterItem_base::_lt);
		addFilter(filter1);
		cRecordFilterItem_numInterval *filter2 = new cRecordFilterItem_numInterval(this, of_response_time_ms, 0, cRecordFilterItem_base::_ge);
		addFilter(filter2);
	}
	if(!filterData["response_number"].empty()) {
		vector<string> response_numbers_str = split(filterData["response_number"].c_str(), split(",|;| |", "|"), true);
		vector<int> response_numbers;
		for(unsigned i = 0; i < response_numbers_str.size(); i++) {
			int response_number = atoi(response_numbers_str[i].c_str());
			if(response_number > 0) {
				response_numbers.push_back(response_number);
			}
		}
		if(response_numbers.size()) {
			cRecordFilterItem_numList *filter = new cRecordFilterItem_numList(this, of_last_response_number);
			for(unsigned i = 0; i < response_numbers.size(); i++) {
				filter->addNum(response_numbers[i]);
			}
			addFilter(filter);
		}
	}
	if(!filterData["sensor_id"].empty()) {
		cRecordFilterItem_numList *filter = new cRecordFilterItem_numList(this, of_id_sensor);
		filter->addNum(atoi(filterData["sensor_id"].c_str()) >= 0 ? atoi(filterData["sensor_id"].c_str()) : -1);
		addFilter(filter);
	}
}