This repository has been archived on 2025-09-14. You can view files and clone it, but cannot push or open issues or pull requests.
Files
tango-maat/src/maat_compile.c

1836 lines
61 KiB
C
Raw Normal View History

2023-01-30 21:59:35 +08:00
/*
**********************************************************************************************
* File: maat_hierarchy.cpp
* Description:
* Authors: Zheng Chao <zhengchao@geedgenetworks.com>
* Date: 2022-10-31
* Copyright: (c) 2018-2022 Geedge Networks, Inc. All rights reserved.
***********************************************************************************************
*/
#include <assert.h>
#include <pthread.h>
#include <linux/limits.h>
2023-01-30 21:59:35 +08:00
#include "maat_utils.h"
#include "log/log.h"
#include "uthash/utarray.h"
#include "uthash/uthash.h"
#include "bool_matcher.h"
#include "igraph/igraph.h"
#include "maat_compile.h"
#include "maat_garbage_collection.h"
#include "maat_group.h"
#include "maat_ex_data.h"
2023-01-30 21:59:35 +08:00
#include "rcu_hash.h"
2023-02-03 17:28:14 +08:00
#include "maat_table.h"
2023-01-30 21:59:35 +08:00
#define MODULE_COMPILE module_name_str("maat.compile")
#define MAX_TABLE_LINE_SIZE (1024 * 16)
struct compile_schema {
int compile_id_column;
2023-02-24 18:20:04 +08:00
int rule_tag_column;
int declared_clause_num_column;
2023-01-30 21:59:35 +08:00
int evaluation_order_column;
struct ex_data_schema *ex_schema;
2023-01-30 21:59:35 +08:00
int table_id; //ugly
2023-02-09 22:13:15 +08:00
struct table_manager *ref_tbl_mgr;
2023-02-24 18:20:04 +08:00
unsigned long long update_err_cnt;
unsigned long long unmatch_tag_cnt;
2023-01-30 21:59:35 +08:00
};
struct group2compile_schema {
int group_id_column;
int compile_id_column;
int not_flag_column;
2023-02-03 17:28:14 +08:00
int vtable_name_column;
2023-01-30 21:59:35 +08:00
int clause_index_column;
char associated_compile_table_id;
int table_id;//ugly
2023-02-09 22:13:15 +08:00
struct table_manager *ref_tbl_mgr;
2023-01-30 21:59:35 +08:00
};
struct compile_item {
2023-02-22 15:22:41 +08:00
long long compile_id;
int declared_clause_num;
2023-01-30 21:59:35 +08:00
int evaluation_order;
};
struct group2compile_item {
2023-02-22 15:22:41 +08:00
long long group_id;
long long compile_id;
2023-01-30 21:59:35 +08:00
int not_flag;
2023-02-03 17:28:14 +08:00
int vtable_id;
2023-01-30 21:59:35 +08:00
int clause_index;
int associated_compile_table_id;
};
/* compile_runtime and group2compile_runtime share compile_hash_map */
struct compile_runtime {
struct bool_matcher *bm;
struct maat_compile *compile_hash; // <compile_id, struct maat_compile>
unsigned long long clause_id_generator;
uint32_t rule_num;
uint32_t updating_rule_num;
2023-02-16 16:45:06 +08:00
pthread_rwlock_t rwlock; /* TODO: replaced with mutex? */
2023-02-03 17:28:14 +08:00
struct bool_expr_match *expr_match_buff;
2023-01-30 21:59:35 +08:00
struct maat_garbage_bin *ref_garbage_bin;
struct log_handle *logger;
};
struct group2compile_runtime {
long long not_flag_group;
struct compile_runtime *ref_compile_rt;
struct group2group_runtime *ref_g2g_rt;
};
struct maat_clause_state {
unsigned long long clause_id;
char not_flag;
char in_use;
UT_array *literal_ids;
};
struct maat_literal_id {
2023-02-22 15:22:41 +08:00
long long group_id;
2023-02-03 17:28:14 +08:00
int vtable_id;
2023-01-30 21:59:35 +08:00
};
struct maat_clause {
unsigned long long clause_id;
size_t n_literal_id;
struct maat_literal_id *literal_ids;
UT_hash_handle hh;
};
struct compile_sort_para {
double evaluation_order;
int declared_clause_num;
2023-02-22 15:22:41 +08:00
long long compile_id;
2023-01-30 21:59:35 +08:00
void *user;
};
#define MAAT_COMPILE_MAGIC 0x4a5b6c7d
struct maat_compile {
unsigned int magic;
2023-02-22 15:22:41 +08:00
long long compile_id;
2023-01-30 21:59:35 +08:00
int actual_clause_num;
int declared_clause_num;
int not_clause_cnt;
void *user_data;
void (*user_data_free)(void *);
UT_hash_handle hh;
struct maat_clause_state clause_states[MAX_ITEMS_PER_BOOL_EXPR];
};
struct maat_internal_hit_path {
int Nth_scan;
int Nth_hit_item;
2023-02-22 15:22:41 +08:00
long long item_id;
long long group_id;
2023-02-03 17:28:14 +08:00
int vtable_id;
2023-01-30 21:59:35 +08:00
};
struct maat_compile_state {
int thread_id;
int Nth_scan;
time_t hier_ver;
2023-02-03 17:28:14 +08:00
size_t this_scan_hit_item_cnt;
2023-01-30 21:59:35 +08:00
int not_clause_hitted_flag;
int is_no_count_scan;
size_t hit_path_cnt;
UT_array *internal_hit_paths;
2023-02-03 17:28:14 +08:00
UT_array *all_hit_clauses;
UT_array *this_scan_hit_clauses;
2023-01-30 21:59:35 +08:00
};
int compile_table_set_ex_data_schema(struct compile_schema *compile_schema,
2023-02-03 17:28:14 +08:00
int table_id,
maat_ex_new_func_t *new_func,
maat_ex_free_func_t *free_func,
maat_ex_dup_func_t *dup_func,
2023-01-30 21:59:35 +08:00
long argl, void *argp,
struct log_handle *logger)
{
if (compile_schema->ex_schema != NULL) {
2023-02-03 17:28:14 +08:00
log_error(logger, MODULE_COMPILE,
"compile ex schema has been set already, can't set anymore");
2023-01-30 21:59:35 +08:00
return -1;
}
compile_schema->ex_schema = ALLOC(struct ex_data_schema, 1);
compile_schema->ex_schema->argl = argl;
compile_schema->ex_schema->argp = argp;
compile_schema->ex_schema->new_func = new_func;
compile_schema->ex_schema->free_func = free_func;
compile_schema->ex_schema->dup_func = dup_func;
return 0;
2023-01-30 21:59:35 +08:00
}
2023-02-22 15:22:41 +08:00
void *compile_runtime_get_user_data(struct compile_runtime *compile_rt, long long compile_id, int is_dettach)
2023-02-09 22:13:15 +08:00
{
struct maat_compile *compile = NULL;
void *ret = NULL;
2023-02-16 16:45:06 +08:00
pthread_rwlock_rdlock(&compile_rt->rwlock);
2023-02-22 15:22:41 +08:00
HASH_FIND(hh, compile_rt->compile_hash, &compile_id, sizeof(long long), compile);
2023-02-09 22:13:15 +08:00
if (compile != NULL) {
ret = compile->user_data;
if (is_dettach) {
compile->user_data = NULL;
}
}
2023-02-16 16:45:06 +08:00
pthread_rwlock_unlock(&compile_rt->rwlock);
2023-02-09 22:13:15 +08:00
return ret;
}
void *rule_ex_data_new(int table_id, const char *table_line,
const struct ex_data_schema *ex_schema)
{
void *ex_data = NULL;
ex_schema->new_func(table_id, NULL, table_line, &ex_data, ex_schema->argl, ex_schema->argp);
return ex_data;
}
void rule_ex_data_free(int table_id, void **ex_data, const struct ex_data_schema *ex_schema)
{
ex_schema->free_func(table_id, ex_data, ex_schema->argl, ex_schema->argp);
}
void rule_ex_data_new_cb(void *user_data, void *param, int table_id)
{
struct ex_data_schema *ex_schema = (struct ex_data_schema *)param;
struct compile_rule *compile = (struct compile_rule *)user_data;
// if(compile->ref_table->table_id!=ex_desc->table_id)
// {
// return;
// }
void *ad = rule_ex_data_new(table_id, compile->table_line, ex_schema);
*compile->ex_data = ad;
}
void compile_runtime_user_data_iterate(struct compile_runtime *compile_rt,
void (*callback)(void *user_data, void *param, int table_id),
void *param, int table_id)
{
struct maat_compile *compile = NULL, *tmp_compile = NULL;
pthread_rwlock_rdlock(&compile_rt->rwlock);
HASH_ITER(hh, compile_rt->compile_hash, compile, tmp_compile) {
if (compile->user_data) {
callback(compile->user_data, param, table_id);
}
}
pthread_rwlock_unlock(&compile_rt->rwlock);
}
2023-01-30 21:59:35 +08:00
UT_icd ut_literal_id_icd = {sizeof(struct maat_literal_id), NULL, NULL, NULL};
2023-02-22 15:22:41 +08:00
UT_icd ut_clause_id_icd = {sizeof(long long), NULL, NULL, NULL};
2023-01-30 21:59:35 +08:00
UT_icd ut_hit_path_icd = {sizeof(struct maat_internal_hit_path), NULL, NULL, NULL};
2023-02-03 17:28:14 +08:00
void *compile_schema_new(cJSON *json, struct table_manager *tbl_mgr,
const char *table_name,
struct log_handle *logger)
2023-01-30 21:59:35 +08:00
{
int read_cnt = 0;
struct compile_schema *compile_schema = ALLOC(struct compile_schema, 1);
cJSON *custom_item = NULL;
cJSON *item = cJSON_GetObjectItem(json, "table_id");
2023-01-31 20:39:53 +08:00
if (item != NULL && item->type == cJSON_Number) {
compile_schema->table_id = item->valueint;
read_cnt++;
2023-01-30 21:59:35 +08:00
}
item = cJSON_GetObjectItem(json, "custom");
if (item == NULL || item->type != cJSON_Object) {
2023-02-03 17:28:14 +08:00
log_error(logger, MODULE_COMPILE,
"table %s has no custom column", table_name);
2023-01-30 21:59:35 +08:00
goto error;
}
custom_item = cJSON_GetObjectItem(item, "compile_id");
if (custom_item != NULL && custom_item->type == cJSON_Number) {
compile_schema->compile_id_column = custom_item->valueint;
read_cnt++;
}
custom_item = cJSON_GetObjectItem(item, "tags");
if (custom_item != NULL && custom_item->type == cJSON_Number) {
2023-02-24 18:20:04 +08:00
compile_schema->rule_tag_column = custom_item->valueint;
2023-01-30 21:59:35 +08:00
read_cnt++;
}
custom_item = cJSON_GetObjectItem(item, "clause_num");
if (custom_item != NULL && custom_item->type == cJSON_Number) {
compile_schema->declared_clause_num_column = custom_item->valueint;
2023-01-30 21:59:35 +08:00
read_cnt++;
}
custom_item = cJSON_GetObjectItem(item, "evaluation_order");
if (custom_item != NULL && custom_item->type == cJSON_Number) {
compile_schema->evaluation_order_column = custom_item->valueint;
read_cnt++;
}
2023-02-09 22:13:15 +08:00
compile_schema->ref_tbl_mgr = tbl_mgr;
2023-02-03 17:28:14 +08:00
2023-02-24 18:20:04 +08:00
if (read_cnt < 5) {
2023-01-30 21:59:35 +08:00
goto error;
}
return compile_schema;
error:
FREE(compile_schema);
return NULL;
}
void compile_schema_free(void *compile_schema)
{
FREE(compile_schema);
}
2023-02-03 17:28:14 +08:00
void *group2compile_schema_new(cJSON *json, struct table_manager *tbl_mgr,
const char *table_name, struct log_handle *logger)
2023-01-30 21:59:35 +08:00
{
int read_cnt = 0;
struct group2compile_schema *g2c_schema = ALLOC(struct group2compile_schema, 1);
cJSON *custom_item = NULL;
cJSON *item = cJSON_GetObjectItem(json, "table_id");
2023-01-31 20:39:53 +08:00
if (item != NULL && item->type == cJSON_Number) {
g2c_schema->table_id = item->valueint;
read_cnt++;
}
item = cJSON_GetObjectItem(json, "associated_compile_table_id");
if (item != NULL && item->type == cJSON_Number) {
g2c_schema->associated_compile_table_id = item->valueint;
read_cnt++;
2023-01-30 21:59:35 +08:00
}
item = cJSON_GetObjectItem(json, "custom");
if (item == NULL || item->type != cJSON_Object) {
log_error(logger, MODULE_COMPILE, "table %s has no custom column", table_name);
goto error;
}
custom_item = cJSON_GetObjectItem(item, "group_id");
if (custom_item != NULL && custom_item->type == cJSON_Number) {
g2c_schema->group_id_column = custom_item->valueint;
read_cnt++;
}
custom_item = cJSON_GetObjectItem(item, "compile_id");
if (custom_item != NULL && custom_item->type == cJSON_Number) {
g2c_schema->compile_id_column = custom_item->valueint;
read_cnt++;
}
custom_item = cJSON_GetObjectItem(item, "not_flag");
if (custom_item != NULL && custom_item->type == cJSON_Number) {
g2c_schema->not_flag_column = custom_item->valueint;
read_cnt++;
}
custom_item = cJSON_GetObjectItem(item, "virtual_table_name");
if (custom_item != NULL && custom_item->type == cJSON_Number) {
2023-02-03 17:28:14 +08:00
g2c_schema->vtable_name_column = custom_item->valueint;
2023-01-30 21:59:35 +08:00
read_cnt++;
}
custom_item = cJSON_GetObjectItem(item, "clause_index");
if (custom_item != NULL && custom_item->type == cJSON_Number) {
g2c_schema->clause_index_column = custom_item->valueint;
read_cnt++;
}
2023-02-09 22:13:15 +08:00
g2c_schema->ref_tbl_mgr = tbl_mgr;
2023-02-03 17:28:14 +08:00
2023-01-31 20:39:53 +08:00
if (read_cnt < 7) {
2023-01-30 21:59:35 +08:00
goto error;
}
return g2c_schema;
error:
FREE(g2c_schema);
return NULL;
}
void group2compile_schema_free(void *g2c_schema)
{
FREE(g2c_schema);
}
2023-01-31 20:39:53 +08:00
int group2compile_associated_compile_table_id(void *g2c_schema)
{
struct group2compile_schema *schema = (struct group2compile_schema *)g2c_schema;
return schema->associated_compile_table_id;
}
2023-02-24 18:20:04 +08:00
int compile_accept_tag_match(struct compile_schema *schema, const char *line,
struct log_handle *logger)
{
size_t column_offset = 0;
size_t column_len = 0;
size_t n_tag = table_manager_accept_tags_count(schema->ref_tbl_mgr);
if (schema->rule_tag_column > 0 && n_tag > 0) {
int ret = get_column_pos(line, schema->rule_tag_column,
&column_offset, &column_len);
if (ret < 0) {
log_error(logger, MODULE_COMPILE,
"compile table(table_id:%d) has no rule_tag, line:%s",
schema->table_id, line);
schema->update_err_cnt++;
return TAG_MATCH_ERR;
}
if (column_len > 2) {
char *tag_str = ALLOC(char, column_len + 1);
memcpy(tag_str, (line + column_offset), column_len);
ret = table_manager_accept_tags_match(schema->ref_tbl_mgr, tag_str);
FREE(tag_str);
if (TAG_MATCH_ERR == ret) {
log_error(logger, MODULE_COMPILE,
"compile table(table_id:%d) has invalid tag format, line:%s",
schema->table_id, line);
schema->update_err_cnt++;
return TAG_MATCH_ERR;
}
if (TAG_MATCH_UNMATCHED == ret) {
schema->unmatch_tag_cnt++;
return TAG_MATCH_UNMATCHED;
}
}
}
return TAG_MATCH_MATCHED;
}
2023-01-30 21:59:35 +08:00
struct compile_item *
2023-02-03 17:28:14 +08:00
compile_item_new(const char *line, struct compile_schema *compile_schema,
struct log_handle *logger)
2023-01-30 21:59:35 +08:00
{
2023-02-24 18:20:04 +08:00
int ret = compile_accept_tag_match(compile_schema, line, logger);
if (ret == TAG_MATCH_UNMATCHED) {
return NULL;
}
2023-01-30 21:59:35 +08:00
size_t column_offset = 0;
size_t column_len = 0;
struct compile_item *compile_item = ALLOC(struct compile_item, 1);
2023-02-24 18:20:04 +08:00
ret = get_column_pos(line, compile_schema->compile_id_column,
2023-02-03 17:28:14 +08:00
&column_offset, &column_len);
2023-01-30 21:59:35 +08:00
if (ret < 0) {
2023-02-03 17:28:14 +08:00
log_error(logger, MODULE_COMPILE,
"compile table(table_id:%d) line:%s has no compile_id",
2023-01-30 21:59:35 +08:00
compile_schema->table_id, line);
goto error;
}
compile_item->compile_id = atoll(line + column_offset);
2023-01-30 21:59:35 +08:00
ret = get_column_pos(line, compile_schema->declared_clause_num_column,
2023-02-03 17:28:14 +08:00
&column_offset, &column_len);
2023-01-30 21:59:35 +08:00
if (ret < 0) {
log_error(logger, MODULE_COMPILE,
"compile table(table_id:%d) line:%s has no clause_num",
compile_schema->table_id, line);
goto error;
}
compile_item->declared_clause_num = atoi(line + column_offset);
2023-01-30 21:59:35 +08:00
2023-02-03 17:28:14 +08:00
ret = get_column_pos(line, compile_schema->evaluation_order_column,
&column_offset, &column_len);
2023-01-30 21:59:35 +08:00
if (ret < 0) {
log_error(logger, MODULE_COMPILE,
"compile table(table_id:%d) line:%s has no evaluation_order",
compile_schema->table_id, line);
goto error;
}
compile_item->evaluation_order = atoi(line + column_offset);
return compile_item;
error:
FREE(compile_item);
return NULL;
}
void compile_item_free(struct compile_item *compile_item)
{
if (NULL == compile_item) {
return;
}
FREE(compile_item);
}
2023-02-03 17:28:14 +08:00
void *compile_runtime_new(void *compile_schema, int max_thread_num,
struct maat_garbage_bin *garbage_bin,
2023-01-30 21:59:35 +08:00
struct log_handle *logger)
{
if (NULL == compile_schema) {
return NULL;
}
struct compile_runtime *compile_rt = ALLOC(struct compile_runtime, 1);
2023-02-03 17:28:14 +08:00
compile_rt->expr_match_buff = ALLOC(struct bool_expr_match, max_thread_num * MAX_SCANNER_HIT_COMPILE_NUM);
2023-01-30 21:59:35 +08:00
compile_rt->logger = logger;
compile_rt->ref_garbage_bin = garbage_bin;
2023-02-16 16:45:06 +08:00
pthread_rwlock_init(&compile_rt->rwlock, NULL);
2023-01-30 21:59:35 +08:00
return compile_rt;
}
2023-01-31 20:39:53 +08:00
void maat_compile_free(struct maat_compile *compile)
{
struct maat_clause_state *clause_state = NULL;
if (compile->user_data && compile->user_data_free) {
compile->user_data_free(compile->user_data);
}
for (int i = 0; i < MAX_ITEMS_PER_BOOL_EXPR; i++) {
clause_state = compile->clause_states + i;
utarray_free(clause_state->literal_ids);
clause_state->literal_ids = NULL;
clause_state->in_use = 0;
}
compile->magic = 0;
free(compile);
}
void maat_compile_hash_free(struct maat_compile **compile_hash)
{
struct maat_compile *compile = NULL, *tmp_compile = NULL;
HASH_ITER(hh, *compile_hash, compile, tmp_compile) {
HASH_DEL(*compile_hash, compile);
maat_compile_free(compile);
}
assert(*compile_hash == NULL);
}
2023-01-30 21:59:35 +08:00
void compile_runtime_free(void *compile_runtime)
{
if (NULL == compile_runtime) {
return;
}
struct compile_runtime *compile_rt = (struct compile_runtime *)compile_runtime;
2023-02-16 16:45:06 +08:00
pthread_rwlock_wrlock(&compile_rt->rwlock);
2023-01-30 21:59:35 +08:00
if (compile_rt->bm != NULL) {
bool_matcher_free(compile_rt->bm);
}
if (compile_rt->compile_hash != NULL) {
maat_compile_hash_free(&(compile_rt->compile_hash));
}
2023-02-16 16:45:06 +08:00
pthread_rwlock_unlock(&compile_rt->rwlock);
2023-01-30 21:59:35 +08:00
2023-02-03 17:28:14 +08:00
if (compile_rt->expr_match_buff != NULL) {
FREE(compile_rt->expr_match_buff);
}
2023-01-30 21:59:35 +08:00
FREE(compile_rt);
}
2023-02-03 17:28:14 +08:00
void *group2compile_runtime_new(void *g2c_schema, int max_thread_num,
struct maat_garbage_bin *garbage_bin,
2023-01-30 21:59:35 +08:00
struct log_handle *logger)
{
struct group2compile_runtime *g2c_rt = ALLOC(struct group2compile_runtime, 1);
return g2c_rt;
}
2023-02-03 17:28:14 +08:00
void group2compile_runtime_init(void *g2c_runtime, void *compile_runtime,
void *g2g_runtime)
2023-01-31 20:39:53 +08:00
{
struct group2compile_runtime *g2c_rt = (struct group2compile_runtime *)g2c_runtime;
g2c_rt->ref_compile_rt = (struct compile_runtime *)compile_runtime;
g2c_rt->ref_g2g_rt = (struct group2group_runtime *)g2g_runtime;
}
2023-01-30 21:59:35 +08:00
void group2compile_runtime_free(void *g2c_runtime)
{
if (NULL == g2c_runtime) {
return;
}
FREE(g2c_runtime);
}
int is_valid_table_name(const char *str)
{
size_t integer_cnt=0;
for (size_t i = 0; i < strlen(str); i++) {
if (str[i] >= '0' && str[i] <= '9') {
integer_cnt++;
}
}
2023-02-03 17:28:14 +08:00
if (strlen(str) == 0 || integer_cnt == strlen(str) ||
0 == strcasecmp(str, "null")) {
2023-01-30 21:59:35 +08:00
return 0;
}
return 1;
}
struct group2compile_item *
group2compile_item_new(const char *line, struct group2compile_schema *g2c_schema,
struct log_handle *logger)
{
size_t column_offset = 0;
size_t column_len = 0;
2023-02-03 17:28:14 +08:00
char vtable_name[NAME_MAX] = {0};
2023-01-30 21:59:35 +08:00
struct group2compile_item *g2c_item = ALLOC(struct group2compile_item, 1);
int ret = get_column_pos(line, g2c_schema->group_id_column, &column_offset, &column_len);
if (ret < 0) {
log_error(logger, MODULE_COMPILE,
"group2compile table(table_id:%d) line:%s has no group_id",
g2c_schema->table_id, line);
goto error;
}
g2c_item->group_id = atoll(line + column_offset);
2023-01-30 21:59:35 +08:00
ret = get_column_pos(line, g2c_schema->compile_id_column, &column_offset, &column_len);
if (ret < 0) {
log_error(logger, MODULE_COMPILE,
"group2compile table(table_id:%d) line:%s has no compile_id",
g2c_schema->table_id, line);
goto error;
}
g2c_item->compile_id = atoll(line + column_offset);
2023-01-30 21:59:35 +08:00
ret = get_column_pos(line, g2c_schema->not_flag_column, &column_offset, &column_len);
if (ret < 0) {
log_error(logger, MODULE_COMPILE,
"group2compile table(table_id:%d) line:%s has no NOT_flag",
g2c_schema->table_id, line);
goto error;
}
g2c_item->not_flag = atoi(line + column_offset);
2023-02-03 17:28:14 +08:00
ret = get_column_pos(line, g2c_schema->vtable_name_column, &column_offset, &column_len);
2023-01-30 21:59:35 +08:00
if (ret < 0) {
log_error(logger, MODULE_COMPILE,
"group2compile table(table_id:%d) line:%s has no virtual_table_name",
g2c_schema->table_id, line);
goto error;
}
if (column_len > NAME_MAX) {
log_error(logger, MODULE_COMPILE,
"group2compile table(table_id:%d) line:%s virtual_table_name length too long",
g2c_schema->table_id, line);
goto error;
}
2023-01-31 20:39:53 +08:00
2023-02-03 17:28:14 +08:00
memcpy(vtable_name, (line + column_offset), column_len);
2023-01-30 21:59:35 +08:00
2023-02-03 17:28:14 +08:00
if (is_valid_table_name(vtable_name)) {
g2c_item->vtable_id = table_manager_get_table_id(g2c_schema->ref_tbl_mgr, vtable_name);
if (g2c_item->vtable_id < 0) {
2023-01-30 21:59:35 +08:00
log_error(logger, MODULE_COMPILE,
"group2compile table(table_id:%d) line:%s unknown virtual table:%s",
2023-02-03 17:28:14 +08:00
g2c_schema->table_id, line, vtable_name);
2023-01-30 21:59:35 +08:00
goto error;
}
}
ret = get_column_pos(line, g2c_schema->clause_index_column, &column_offset, &column_len);
if (ret < 0) {
log_error(logger, MODULE_COMPILE,
"group2compile table(table_id:%d) line:%s has no clause_index",
g2c_schema->table_id, line);
goto error;
}
g2c_item->clause_index = atoi(line + column_offset);
return g2c_item;
error:
FREE(g2c_item);
return NULL;
}
void group2compile_item_free(struct group2compile_item *g2c_item)
{
if (NULL == g2c_item) {
return;
}
FREE(g2c_item);
}
#define MAAT_HIER_COMPILE_MAGIC 0x4a5b6c7d
2023-02-22 15:22:41 +08:00
struct maat_compile *maat_compile_new(long long compile_id)
2023-01-30 21:59:35 +08:00
{
struct maat_compile *compile = ALLOC(struct maat_compile, 1);
compile->magic = MAAT_HIER_COMPILE_MAGIC;
compile->compile_id = compile_id;
for(int i = 0; i < MAX_ITEMS_PER_BOOL_EXPR; i++) {
utarray_new(compile->clause_states[i].literal_ids, &ut_literal_id_icd);
compile->clause_states[i].in_use=0;
}
return compile;
}
2023-02-03 17:28:14 +08:00
int maat_compile_set(struct maat_compile *compile, int declared_clause_num,
void *user_data, void (*user_data_free)(void *))
2023-01-30 21:59:35 +08:00
{
if (user_data != NULL && NULL == user_data_free) {
return -1;
}
compile->declared_clause_num = declared_clause_num;
compile->user_data = user_data;
compile->user_data_free = user_data_free;
return 0;
}
2023-02-22 15:22:41 +08:00
int maat_compile_hash_add(struct maat_compile **compile_hash, long long compile_id,
2023-02-03 17:28:14 +08:00
struct maat_compile *compile)
2023-01-30 21:59:35 +08:00
{
int ret = 0;
2023-02-16 16:45:06 +08:00
assert(compile->declared_clause_num >= 0);
2023-02-22 15:22:41 +08:00
HASH_ADD(hh, *compile_hash, compile_id, sizeof(long long), compile);
2023-01-30 21:59:35 +08:00
//TODO:mytest need to delete
#if 0
2023-03-02 10:18:20 +08:00
if (compile_id == 141)
{
size_t compile_cnt = HASH_COUNT(*compile_hash);
struct maat_compile *compile1 = NULL, *tmp_compile1 = NULL;
HASH_ITER(hh, *compile_hash, compile1, tmp_compile1)
{
printf("<maat_compile_hash_add> compile_id:%lld, compile_cnt:%zu\n",
compile1->compile_id, compile_cnt);
}
}
2023-01-30 21:59:35 +08:00
#endif
2023-03-02 10:18:20 +08:00
return ret;
2023-01-30 21:59:35 +08:00
}
2023-02-22 15:22:41 +08:00
void maat_compile_hash_set(struct maat_compile **compile_hash, long long compile_id,
2023-02-16 16:45:06 +08:00
struct maat_compile *compile)
2023-01-30 21:59:35 +08:00
{
2023-02-16 16:45:06 +08:00
struct maat_compile *tmp_compile = NULL;
2023-01-30 21:59:35 +08:00
2023-02-22 15:22:41 +08:00
HASH_FIND(hh, *compile_hash, &compile_id, sizeof(long long), tmp_compile);
2023-02-16 16:45:06 +08:00
assert(tmp_compile != NULL);
2023-01-30 21:59:35 +08:00
2023-02-16 16:45:06 +08:00
assert(tmp_compile->user_data == NULL);
maat_compile_set(tmp_compile, compile->declared_clause_num,
compile->user_data, compile->user_data_free);
}
int maat_compile_hash_remove(struct maat_compile **compile_hash, struct maat_compile *compile,
struct maat_garbage_bin *garbage_bin)
2023-02-16 16:45:06 +08:00
{
2023-02-23 14:50:07 +08:00
if (compile->user_data_free && compile->user_data) {
compile->user_data_free(compile->user_data);
compile->user_data = NULL;
}
2023-01-30 21:59:35 +08:00
if (0 == compile->actual_clause_num) {
HASH_DEL(*compile_hash, compile);
maat_garbage_bagging(garbage_bin, compile, (void (*)(void *))maat_compile_free);
}
//TODO:mytest need to delete
#if 0
size_t compile_cnt = HASH_COUNT(*compile_hash);
struct maat_compile *compile1 = NULL, *tmp_compile1 = NULL;
HASH_ITER (hh, *compile_hash, compile1, tmp_compile1) {
printf("<maat_compile_hash_remove> compile_id:%lu, compile_cnt:%zu\n",
compile1->compile_id, compile_cnt);
2023-01-30 21:59:35 +08:00
}
#endif
return 0;
}
2023-02-22 15:22:41 +08:00
struct maat_compile *maat_compile_hash_find(struct maat_compile **compile_hash, long long compile_id)
2023-01-30 21:59:35 +08:00
{
struct maat_compile *compile = NULL;
HASH_FIND(hh, *compile_hash, &compile_id, sizeof(compile_id), compile);
return compile;
2023-01-30 21:59:35 +08:00
}
2023-02-09 22:13:15 +08:00
size_t maat_compile_in_use_count(struct maat_compile *compile_hash)
2023-01-30 21:59:35 +08:00
{
2023-02-09 22:13:15 +08:00
struct maat_compile *compile = NULL, *tmp_compile = NULL;
size_t in_use_compile_cnt = 0;
struct maat_clause_state *clause_state = NULL;
HASH_ITER(hh, compile_hash, compile, tmp_compile) {
//find how much compile whose clause is in_use
for (int i = 0; i < MAX_ITEMS_PER_BOOL_EXPR; i++) {
clause_state = compile->clause_states + i;
if (clause_state->in_use) {
in_use_compile_cnt++;
break;
}
}
}
return in_use_compile_cnt;
2023-01-30 21:59:35 +08:00
}
int compare_literal_id(const void *pa, const void *pb)
{
struct maat_literal_id *la = (struct maat_literal_id *)pa;
struct maat_literal_id *lb = (struct maat_literal_id *)pb;
2023-02-03 17:28:14 +08:00
int ret = la->vtable_id - lb->vtable_id;
2023-01-30 21:59:35 +08:00
if (0 == ret) {
ret = la->group_id - lb->group_id;
}
return ret;
}
2023-01-31 20:39:53 +08:00
int maat_compile_clause_add_literal(struct maat_compile *compile,
struct maat_literal_id *literal_id,
2023-01-30 21:59:35 +08:00
int clause_index, int clause_not_flag)
{
struct maat_clause_state *clause_state = compile->clause_states + clause_index;
clause_state->not_flag = clause_not_flag;
if (!clause_state->in_use) {
clause_state->in_use = 1;
compile->actual_clause_num++;
}
2023-01-31 20:39:53 +08:00
struct maat_literal_id *tmp = NULL;
tmp = (struct maat_literal_id *)utarray_find(clause_state->literal_ids,
literal_id, compare_literal_id);
2023-01-30 21:59:35 +08:00
if (tmp) {
assert(tmp->group_id == literal_id->group_id);
assert(tmp->vtable_id == literal_id->vtable_id);
2023-01-30 21:59:35 +08:00
return -1;
} else {
utarray_push_back(clause_state->literal_ids, literal_id);
utarray_sort(clause_state->literal_ids, compare_literal_id);
}
return 0;
}
2023-01-31 20:39:53 +08:00
int maat_compile_clause_remove_literal(struct maat_compile *compile,
struct maat_literal_id *literal_id,
int clause_index)
2023-01-30 21:59:35 +08:00
{
struct maat_clause_state* clause_state = compile->clause_states + clause_index;
2023-01-31 20:39:53 +08:00
struct maat_literal_id *tmp = NULL;
tmp = (struct maat_literal_id *)utarray_find(clause_state->literal_ids,
literal_id, compare_literal_id);
2023-01-30 21:59:35 +08:00
if (tmp) {
assert(*(unsigned long long*)tmp == *(unsigned long long*)(literal_id));
} else {
return -1;
}
size_t remove_idx = utarray_eltidx(clause_state->literal_ids, tmp);
utarray_erase(clause_state->literal_ids, remove_idx, 1);
if (0 == utarray_len(clause_state->literal_ids)) {
clause_state->in_use=0;
compile->actual_clause_num--;
}
return 0;
}
static const struct maat_clause *
2023-01-31 20:39:53 +08:00
maat_clause_hash_fetch_clause(struct maat_clause **clause_hash,
unsigned long long *clause_id_generator,
2023-02-03 17:28:14 +08:00
struct maat_literal_id *literal_ids,
size_t n_literal_id)
2023-01-30 21:59:35 +08:00
{
struct maat_clause *clause = NULL;
HASH_FIND(hh, *clause_hash, literal_ids, n_literal_id * sizeof(struct maat_literal_id), clause);
if (!clause) {
clause = ALLOC(struct maat_clause, 1);
clause->clause_id = *clause_id_generator;
clause->n_literal_id = n_literal_id;
clause->literal_ids = ALLOC(struct maat_literal_id, n_literal_id);
memcpy(clause->literal_ids, literal_ids, n_literal_id * sizeof(struct maat_literal_id));
(*clause_id_generator)++;
2023-02-03 17:28:14 +08:00
HASH_ADD_KEYPTR(hh, *clause_hash, clause->literal_ids,
n_literal_id * sizeof(struct maat_literal_id), clause);
2023-01-30 21:59:35 +08:00
}
return clause;
}
static void maat_clause_hash_free(struct maat_clause *clause_hash)
{
struct maat_clause *clause = NULL, *tmp_clause = NULL;
HASH_ITER (hh, clause_hash, clause, tmp_clause) {
HASH_DELETE(hh, clause_hash, clause);
free(clause->literal_ids);
clause->n_literal_id = 0;
free(clause);
}
}
2023-01-31 20:39:53 +08:00
struct bool_matcher *
maat_compile_bool_matcher_new(struct maat_compile *compile_hash,
unsigned long long *clause_id_generator,
struct log_handle *logger)
2023-01-30 21:59:35 +08:00
{
if (NULL == compile_hash || NULL == logger) {
return NULL;
}
size_t i = 0, j = 0;
int has_clause_num = 0;
struct bool_matcher *bm = NULL;
struct maat_clause_state *clause_state = NULL;
const struct maat_clause *clause = NULL;
struct maat_clause *clause_hash = NULL; // <literal_id, struct maat_clause>
//STEP 1, update clause_id of each compile and literal
struct maat_compile *compile = NULL, *tmp_compile = NULL;
struct maat_literal_id *literal_ids = NULL;
size_t n_literal_id = 0;
HASH_ITER(hh, compile_hash, compile, tmp_compile) {
has_clause_num = 0;
for (i = 0; i < MAX_ITEMS_PER_BOOL_EXPR; i++) {
clause_state = compile->clause_states + i;
clause_state->clause_id = 0;
if (!clause_state->in_use) {
continue;
}
has_clause_num++;
literal_ids = (struct maat_literal_id *)utarray_eltptr(clause_state->literal_ids, 0);
n_literal_id = utarray_len(clause_state->literal_ids);
2023-02-03 17:28:14 +08:00
clause = maat_clause_hash_fetch_clause(&clause_hash, clause_id_generator,
literal_ids, n_literal_id);
2023-01-30 21:59:35 +08:00
clause_state->clause_id = clause->clause_id;
}
assert(has_clause_num == compile->actual_clause_num);
}
//STEP 2, serial compile clause states to a bool expression array
size_t expr_cnt = 0;
2023-02-09 22:13:15 +08:00
size_t compile_cnt = maat_compile_in_use_count(compile_hash);
2023-01-30 21:59:35 +08:00
struct bool_expr *bool_expr_array = ALLOC(struct bool_expr, compile_cnt);
HASH_ITER(hh, compile_hash, compile, tmp_compile) {
for (i = 0, j = 0; i < MAX_ITEMS_PER_BOOL_EXPR; i++) {
if (compile->clause_states[i].in_use) {
if (compile->clause_states[i].not_flag) {
compile->not_clause_cnt++;
}
//TODO:mytest need to delete
2023-02-09 22:13:15 +08:00
#if 0
2023-01-30 21:59:35 +08:00
struct maat_literal_id *p = NULL;
for(p = (struct maat_literal_id *)utarray_front(compile->clause_states[i].literal_ids); p!=NULL; p=(struct maat_literal_id *)utarray_next(compile->clause_states[i].literal_ids,p)) {
printf("<before bool_matcher_new> compile_id:%llu, clause_id:%llu, literal{%llu: %d}\n",
2023-02-03 17:28:14 +08:00
compile->compile_id, compile->clause_states[i].clause_id, p->group_id, p->vtable_id);
2023-01-30 21:59:35 +08:00
}
#endif
bool_expr_array[expr_cnt].items[j].item_id = compile->clause_states[i].clause_id;
bool_expr_array[expr_cnt].items[j].not_flag = compile->clause_states[i].not_flag;
j++;
}
}
2023-02-16 16:45:06 +08:00
// printf("bool_matcher_new compile_id:%lu j:%zu, compile->declared_clause_num:%d\n",
// compile->compile_id, j, compile->declared_clause_num);
2023-01-30 21:59:35 +08:00
//some compile may have zero groups, e.g. default policy.
if (j == (size_t)compile->declared_clause_num && j > 0) {
bool_expr_array[expr_cnt].expr_id = compile->compile_id;
bool_expr_array[expr_cnt].user_tag = compile;
bool_expr_array[expr_cnt].item_num = j;
expr_cnt++;
}
}
//size_t expr_index = 0, item_index = 0;
// STEP 3, build bool matcher
size_t mem_size = 0;
if (0 == expr_cnt) {
log_error(logger, MODULE_COMPILE, "No bool expression to build bool matcher.");
goto error;
}
//TODO:mytest need to delete
#if 0
2023-02-16 11:13:23 +08:00
printf("bool_matcher_new....................expr_cnt:%zu\n", expr_cnt);
2023-01-30 21:59:35 +08:00
for (expr_index = 0; expr_index < expr_cnt; expr_index++) {
printf("bool_expr_array[%zu].expr_id:%llu, item_num:%zu\n", expr_index, bool_expr_array[expr_index].expr_id,
bool_expr_array[expr_index].item_num);
for (item_index = 0; item_index < bool_expr_array[expr_index].item_num; item_index++) {
printf("bool_expr_array[%zu].items[%zu]:%llu, not_flag:%d\n", expr_index, item_index,
bool_expr_array[expr_index].items[item_index].item_id,
bool_expr_array[expr_index].items[item_index].not_flag);
}
printf("\n");
}
#endif
bm = bool_matcher_new(bool_expr_array, expr_cnt, &mem_size);
if (bm != NULL) {
log_info(logger, MODULE_COMPILE,
"Build bool matcher of %zu expressions with %zu bytes memory.", expr_cnt, mem_size);
} else {
log_error(logger, MODULE_COMPILE, "Build bool matcher failed!");
}
error:
maat_clause_hash_free(clause_hash);
FREE(bool_expr_array);
return bm;
}
void maat_compile_bool_matcher_free(struct bool_matcher *bm)
{
bool_matcher_free(bm);
}
static int maat_compile_has_clause(struct maat_compile *compile,
unsigned long long clause_id)
{
struct maat_clause_state *clause_state = NULL;
for (size_t i = 0; i < MAX_ITEMS_PER_BOOL_EXPR; i++) {
clause_state = compile->clause_states + i;
if (!clause_state->in_use) {
continue;
}
if (clause_state->clause_id == clause_id) {
return 1;
}
}
return 0;
}
static size_t compile_state_if_new_hit_compile(struct maat_compile_state *compile_state,
struct maat_compile *compile)
{
size_t r_in_c_cnt = 0;
int ret = 0;
unsigned long long new_hit_clause_id = 0;
for (size_t i = 0; i < utarray_len(compile_state->this_scan_hit_clauses); i++) {
new_hit_clause_id = *(unsigned long long*)utarray_eltptr(compile_state->this_scan_hit_clauses, i);
ret = maat_compile_has_clause(compile, new_hit_clause_id);
if (ret) {
r_in_c_cnt++;
}
}
return r_in_c_cnt;
}
2023-02-03 17:28:14 +08:00
size_t maat_compile_bool_matcher_match(struct bool_matcher *bm, int is_last_scan,
struct maat_compile_state *compile_state,
2023-01-31 20:39:53 +08:00
void **user_data_array, size_t ud_array_size)
2023-01-30 21:59:35 +08:00
{
struct maat_compile *compile = NULL;
struct bool_expr_match *expr_match = ALLOC(struct bool_expr_match, MAX_SCANNER_HIT_COMPILE_NUM);
size_t ud_result_cnt = 0;
//TODO:mytest need to delete
2023-02-09 22:13:15 +08:00
#if 0
2023-01-30 21:59:35 +08:00
unsigned long long *p;
printf("utarray_len:%u\n", utarray_len(compile_state->all_hit_clauses));
for (p = (unsigned long long *)utarray_front(compile_state->all_hit_clauses); p != NULL; p = (unsigned long long *)utarray_next(compile_state->all_hit_clauses, p))
2023-01-30 21:59:35 +08:00
{
printf("before bool_matcher_match compile_state clause_id:%llu\n", *p);
2023-01-30 21:59:35 +08:00
}
#endif
int bool_match_ret = bool_matcher_match(bm, (unsigned long long *)utarray_eltptr(compile_state->all_hit_clauses, 0),
2023-02-03 17:28:14 +08:00
utarray_len(compile_state->all_hit_clauses),
expr_match, MAX_SCANNER_HIT_COMPILE_NUM);
2023-01-30 21:59:35 +08:00
for (int i = 0; i < bool_match_ret && ud_result_cnt < ud_array_size; i++) {
compile = (struct maat_compile *)expr_match[i].user_tag;
assert(compile->magic == MAAT_COMPILE_MAGIC);
assert((unsigned long long)compile->compile_id == expr_match[i].expr_id);
if (0 == compile->actual_clause_num) {
continue;
}
size_t n_new_hit_compile = compile_state_if_new_hit_compile(compile_state, compile);
size_t n_this_scan_hit_item = compile_state->this_scan_hit_item_cnt;
2023-02-03 17:28:14 +08:00
if ((compile->not_clause_cnt > 0) && (LAST_SCAN_UNSET == is_last_scan)) {
compile_state->not_clause_hitted_flag = 1;
}
2023-01-30 21:59:35 +08:00
//TODO: not_clause
if (compile->user_data) {
if (n_new_hit_compile > 0 || 0 == n_this_scan_hit_item) {
/* compile hit because of new item or
hit a compile that refer a NOT-logic group in previous scan */
user_data_array[ud_result_cnt] = compile->user_data;
ud_result_cnt++;
}
2023-01-30 21:59:35 +08:00
}
}
2023-02-09 22:13:15 +08:00
compile_state->this_scan_hit_item_cnt = 0;
2023-01-30 21:59:35 +08:00
return ud_result_cnt;
}
int maat_add_group_to_compile(struct maat_compile **compile_hash, struct group2compile_item *g2c_item,
struct log_handle *logger)
2023-01-30 21:59:35 +08:00
{
int ret = -1;
2023-01-31 20:39:53 +08:00
struct maat_compile *compile = maat_compile_hash_find(compile_hash, g2c_item->compile_id);
2023-01-30 21:59:35 +08:00
if (!compile) {
2023-01-31 20:39:53 +08:00
compile = maat_compile_new(g2c_item->compile_id);
ret = maat_compile_hash_add(compile_hash, g2c_item->compile_id, compile);
2023-01-30 21:59:35 +08:00
if (ret < 0) {
return -1;
}
}
2023-02-03 17:28:14 +08:00
struct maat_literal_id literal_id = {g2c_item->group_id, g2c_item->vtable_id};
2023-02-09 22:13:15 +08:00
2023-02-03 17:28:14 +08:00
ret = maat_compile_clause_add_literal(compile, &literal_id, g2c_item->clause_index,
g2c_item->not_flag);
2023-01-30 21:59:35 +08:00
if (ret < 0) {
2023-02-03 17:28:14 +08:00
log_error(logger, MODULE_COMPILE,
2023-02-09 22:13:15 +08:00
"add literal_id{group_id:%d, vtable_id:%d} to clause_index: %d of compile %d failed",
2023-02-03 17:28:14 +08:00
g2c_item->group_id, g2c_item->vtable_id, g2c_item->clause_index,
g2c_item->compile_id);
2023-01-30 21:59:35 +08:00
ret = -1;
} else {
ret = 0;
}
// printf("group2compile update compile_id:%lu, compile->declared_clause_num:%d\n",
// compile->compile_id, compile->declared_clause_num);
2023-01-30 21:59:35 +08:00
return ret;
}
int maat_remove_group_from_compile(struct maat_compile **compile_hash,
2023-01-31 20:39:53 +08:00
struct group2compile_item *g2c_item,
struct maat_garbage_bin *garbage_bin,
2023-01-30 21:59:35 +08:00
struct log_handle *logger)
{
struct maat_compile *compile = NULL;
2023-01-31 20:39:53 +08:00
HASH_FIND(hh, *compile_hash, &(g2c_item->compile_id), sizeof(g2c_item->compile_id), compile);
2023-01-30 21:59:35 +08:00
if (!compile) {
2023-01-31 20:39:53 +08:00
log_error(logger, MODULE_COMPILE,
"Remove group %d from compile %d failed, compile is not exisited.",
g2c_item->group_id, g2c_item->compile_id);
2023-01-30 21:59:35 +08:00
return -1;
}
2023-02-03 17:28:14 +08:00
struct maat_literal_id literal_id = {g2c_item->group_id, g2c_item->vtable_id};
2023-01-31 20:39:53 +08:00
int ret = maat_compile_clause_remove_literal(compile, &literal_id, g2c_item->clause_index);
2023-01-30 21:59:35 +08:00
if (ret < 0) {
log_error(logger, MODULE_COMPILE,
2023-02-03 17:28:14 +08:00
"Remove group %d vtable_id %d from clause %d of compile %d failed, literal is not in compile.",
g2c_item->group_id, g2c_item->vtable_id, g2c_item->clause_index, g2c_item->compile_id);
2023-01-30 21:59:35 +08:00
return -1;
}
if (0 == compile->actual_clause_num && NULL == compile->user_data) {
HASH_DEL(*compile_hash, compile);
maat_garbage_bagging(garbage_bin, compile, (void (*)(void*))maat_compile_free);
}
return 0;
}
static inline int compare_clause_id(const void *a, const void *b)
{
long long ret = *(const unsigned long long *)a - *(const unsigned long long *)b;
if (0 == ret) {
return 0;
} else if(ret < 0) {
return -1;
} else {
return 1;
}
}
struct maat_compile_state *maat_compile_state_new(int thread_id)
{
struct maat_compile_state *compile_state = ALLOC(struct maat_compile_state, 1);
compile_state->thread_id = thread_id;
//compile_state->hier_ver = hier->version;
utarray_new(compile_state->internal_hit_paths, &ut_hit_path_icd);
utarray_new(compile_state->all_hit_clauses, &ut_clause_id_icd);
utarray_new(compile_state->this_scan_hit_clauses, &ut_clause_id_icd);
2023-01-30 21:59:35 +08:00
return compile_state;
}
void maat_compile_state_free(struct maat_compile_state *compile_state)
{
utarray_free(compile_state->internal_hit_paths);
utarray_free(compile_state->all_hit_clauses);
utarray_free(compile_state->this_scan_hit_clauses);
2023-01-30 21:59:35 +08:00
free(compile_state);
}
2023-02-22 15:22:41 +08:00
static int maat_compile_hit_path_add(UT_array *hit_paths, long long item_id, long long group_id,
2023-02-03 17:28:14 +08:00
int vtable_id, int Nth_scan, int Nth_item_result)
2023-01-30 21:59:35 +08:00
{
struct maat_internal_hit_path new_path;
new_path.item_id = item_id;
new_path.Nth_hit_item = Nth_item_result;
new_path.Nth_scan = Nth_scan;
new_path.group_id = group_id;
2023-02-03 17:28:14 +08:00
new_path.vtable_id = vtable_id;
2023-01-30 21:59:35 +08:00
utarray_push_back(hit_paths, &new_path);
2023-02-03 17:28:14 +08:00
2023-01-30 21:59:35 +08:00
return 1;
}
2023-02-03 17:28:14 +08:00
static int maat_compile_has_literal(struct maat_compile* compile,
struct maat_literal_id *literal_id)
2023-01-30 21:59:35 +08:00
{
int i = 0;
struct maat_literal_id *tmp = NULL;
struct maat_clause_state *clause_state = NULL;
for (i = 0; i < MAX_ITEMS_PER_BOOL_EXPR; i++) {
clause_state = compile->clause_states+i;
if(!clause_state->in_use) {
continue;
}
2023-02-03 17:28:14 +08:00
tmp = (struct maat_literal_id*)utarray_find(clause_state->literal_ids,
literal_id, compare_literal_id);
2023-01-30 21:59:35 +08:00
if (tmp) {
2023-02-03 17:28:14 +08:00
assert(tmp->group_id == literal_id->group_id &&
tmp->vtable_id == literal_id->vtable_id);
2023-01-30 21:59:35 +08:00
return 1;
}
}
return 0;
}
static int maat_compile_is_hit_path_existed(const struct maat_hit_path *hit_paths,
2023-02-03 17:28:14 +08:00
size_t n_path, const struct maat_hit_path *find)
2023-01-30 21:59:35 +08:00
{
for (size_t i = 0; i < n_path; i++) {
if (0 == memcmp(hit_paths + i, find, sizeof(*find))) {
return 1;
}
}
return 0;
}
2023-01-31 20:39:53 +08:00
size_t compile_runtime_get_hit_paths(struct compile_runtime *compile_rt,
struct group2group_runtime *g2g_rt,
2023-01-30 21:59:35 +08:00
struct maat_compile_state *compile_state,
2023-02-03 17:28:14 +08:00
struct maat_hit_path *hit_paths,
size_t hit_path_index, size_t hit_path_size)
2023-01-30 21:59:35 +08:00
{
size_t i = 0, j = 0;
2023-01-30 21:59:35 +08:00
struct maat_internal_hit_path *internal_path = NULL;
size_t hit_path_cnt = hit_path_index;
2023-02-03 17:28:14 +08:00
size_t new_hit_path_cnt = 0;
2023-01-30 21:59:35 +08:00
for (i = 0; i < utarray_len(compile_state->internal_hit_paths); i++) {
internal_path = (struct maat_internal_hit_path *)utarray_eltptr(compile_state->internal_hit_paths, i);
2023-02-03 17:28:14 +08:00
/*
NOTE: maybe one item has been deleted, but it's item_id still exist in internal_hit_paths
*/
2023-02-22 15:22:41 +08:00
long long top_group_ids[MAX_SCANNER_HIT_GROUP_NUM];
memset(top_group_ids, 0, sizeof(top_group_ids));
2023-02-03 17:28:14 +08:00
int top_group_cnt = group2group_runtime_get_top_groups(g2g_rt, &(internal_path->group_id),
1, top_group_ids);
if (top_group_cnt <= 0) {
/*
item->group_id has no top group, this group can only be referenced by compile
------------------------------------------------------------------------------
for example:
compile1 -> group1 -> group2 -> item1
group3 -> item2
group1 and group3 has no top group
group1 is referenced by compile1, group3 is not referenced by any compile
NOTE: Add the hit path as long as the item is hit
*/
top_group_cnt = 1; //add one hit path which top_group_ids[0] = -1
}
for (int j = 0; j < top_group_cnt && hit_path_cnt < hit_path_size; j++, hit_path_cnt++) {
hit_paths[hit_path_cnt].Nth_scan = internal_path->Nth_scan;
hit_paths[hit_path_cnt].item_id = internal_path->item_id;
hit_paths[hit_path_cnt].sub_group_id = internal_path->group_id;
hit_paths[hit_path_cnt].top_group_id = top_group_ids[j]; //top_group_id may be -1
hit_paths[hit_path_cnt].vtable_id = internal_path->vtable_id;
hit_paths[hit_path_cnt].compile_id = -1;
}
2023-01-30 21:59:35 +08:00
}
2023-02-03 17:28:14 +08:00
/* assign hit_paths[].compile_id */
2023-01-30 21:59:35 +08:00
struct maat_compile *compile = NULL;
struct maat_literal_id literal_id = {0, 0};
struct maat_hit_path tmp_path;
2023-02-03 17:28:14 +08:00
struct bool_expr_match *expr_match = compile_rt->expr_match_buff + compile_state->thread_id * MAX_SCANNER_HIT_COMPILE_NUM;
int bool_match_ret = bool_matcher_match(compile_rt->bm,
(unsigned long long *)utarray_eltptr(compile_state->all_hit_clauses, 0),
utarray_len(compile_state->all_hit_clauses), expr_match,
2023-02-03 17:28:14 +08:00
MAX_SCANNER_HIT_COMPILE_NUM);
2023-01-30 21:59:35 +08:00
for (int idx = 0; idx < bool_match_ret; idx++) {
compile = (struct maat_compile *)expr_match[idx].user_tag;
assert(compile->magic == MAAT_COMPILE_MAGIC);
assert((unsigned long long)compile->compile_id == expr_match[idx].expr_id);
if (0 == compile->actual_clause_num) {
continue;
}
2023-02-03 17:28:14 +08:00
for (j = 0; j < hit_path_cnt && (hit_path_cnt + new_hit_path_cnt) < hit_path_size; j++) {
if (hit_paths[j].top_group_id < 0) {
2023-01-30 21:59:35 +08:00
continue;
}
literal_id.group_id = hit_paths[j].top_group_id;
2023-02-03 17:28:14 +08:00
literal_id.vtable_id = hit_paths[j].vtable_id;
2023-01-30 21:59:35 +08:00
if (maat_compile_has_literal(compile, &literal_id)) {
if (hit_paths[j].compile_id < 0) {
hit_paths[j].compile_id = compile->compile_id;
} else {
2023-02-03 17:28:14 +08:00
// means same literal_id hit more than one compile_id
2023-01-30 21:59:35 +08:00
tmp_path = hit_paths[j];
tmp_path.compile_id = compile->compile_id;
2023-02-03 17:28:14 +08:00
if(maat_compile_is_hit_path_existed(hit_paths, hit_path_cnt + new_hit_path_cnt, &tmp_path)) {
hit_paths[hit_path_cnt + new_hit_path_cnt] = tmp_path;
new_hit_path_cnt++;
2023-01-30 21:59:35 +08:00
}
}
}
}
}
2023-02-03 17:28:14 +08:00
return hit_path_cnt + new_hit_path_cnt;
2023-01-30 21:59:35 +08:00
}
2023-02-03 17:28:14 +08:00
void maat_compile_state_update_hit_path(struct maat_compile_state *compile_state,
2023-02-22 15:22:41 +08:00
long long item_id, long long group_id, int vtable_id,
2023-02-03 17:28:14 +08:00
int Nth_scan, int Nth_item_result)
2023-01-30 21:59:35 +08:00
{
if (compile_state->Nth_scan != Nth_scan) {
2023-02-03 17:28:14 +08:00
assert(compile_state->this_scan_hit_item_cnt == 0);
2023-01-30 21:59:35 +08:00
compile_state->Nth_scan = Nth_scan;
2023-02-03 17:28:14 +08:00
utarray_clear(compile_state->this_scan_hit_clauses);
2023-01-30 21:59:35 +08:00
}
2023-02-03 17:28:14 +08:00
maat_compile_hit_path_add(compile_state->internal_hit_paths, item_id, group_id,
vtable_id, Nth_scan, Nth_item_result);
2023-01-30 21:59:35 +08:00
compile_state->hit_path_cnt++;
2023-02-03 17:28:14 +08:00
compile_state->this_scan_hit_item_cnt++;
2023-01-30 21:59:35 +08:00
}
2023-01-31 20:39:53 +08:00
void maat_compile_state_update_hit_clause(struct maat_compile_state *compile_state,
2023-02-22 15:22:41 +08:00
void *compile_runtime, long long group_id,
2023-02-03 17:28:14 +08:00
int vtable_id)
2023-01-30 21:59:35 +08:00
{
2023-01-31 20:39:53 +08:00
if (NULL == compile_state || NULL == compile_runtime) {
return;
}
2023-01-30 21:59:35 +08:00
struct maat_compile *compile = NULL, *tmp_compile = NULL;
struct maat_clause_state *clause_state = NULL;
2023-02-03 17:28:14 +08:00
struct maat_literal_id literal_id = {group_id, vtable_id};
2023-01-31 20:39:53 +08:00
struct maat_literal_id *tmp = NULL;
2023-02-03 17:28:14 +08:00
unsigned long long *clause_id = 0;
2023-02-16 16:45:06 +08:00
struct compile_runtime *compile_rt = (struct compile_runtime *)compile_runtime;
pthread_rwlock_rdlock(&compile_rt->rwlock);
2023-02-16 16:45:06 +08:00
assert(compile_rt->compile_hash != NULL);
2023-01-31 20:39:53 +08:00
HASH_ITER(hh, compile_rt->compile_hash, compile, tmp_compile) {
2023-01-30 21:59:35 +08:00
for (size_t i = 0; i < MAX_ITEMS_PER_BOOL_EXPR; i++) {
clause_state = compile->clause_states + i;
if (!clause_state->in_use) {
continue;
}
size_t new_clause_idx = utarray_len(compile_state->this_scan_hit_clauses);
2023-02-03 17:28:14 +08:00
tmp = (struct maat_literal_id *)utarray_find(clause_state->literal_ids,
&literal_id, compare_literal_id);
2023-01-30 21:59:35 +08:00
if (tmp) {
//Deduplication
2023-02-03 17:28:14 +08:00
if (utarray_find(compile_state->all_hit_clauses, &(clause_state->clause_id),
compare_clause_id)) {
2023-01-30 21:59:35 +08:00
continue;
}
2023-02-03 17:28:14 +08:00
utarray_push_back(compile_state->this_scan_hit_clauses, &(clause_state->clause_id));
2023-01-30 21:59:35 +08:00
}
2023-02-03 17:28:14 +08:00
//means this scan hit new clause
if ((utarray_len(compile_state->this_scan_hit_clauses) - new_clause_idx) > 0) {
utarray_reserve(compile_state->all_hit_clauses,
utarray_len(compile_state->this_scan_hit_clauses) - new_clause_idx);
for (i = new_clause_idx; i < utarray_len(compile_state->this_scan_hit_clauses); i++) {
clause_id = (unsigned long long *)utarray_eltptr(compile_state->this_scan_hit_clauses, i);
utarray_push_back(compile_state->all_hit_clauses, clause_id);
}
utarray_sort(compile_state->all_hit_clauses, compare_clause_id);
}
}
2023-01-30 21:59:35 +08:00
}
2023-02-16 16:45:06 +08:00
pthread_rwlock_unlock(&compile_rt->rwlock);
2023-01-30 21:59:35 +08:00
}
int maat_compile_state_has_NOT_clause(struct maat_compile_state *compile_state)
{
return compile_state->not_clause_hitted_flag;
}
2023-01-31 20:39:53 +08:00
void compile_item_to_compile_rule(struct compile_item *compile_item,
struct compile_schema *compile_schema,
struct compile_rule *compile_rule,
const char *table_line)
2023-01-31 20:39:53 +08:00
{
compile_rule->magic_num = COMPILE_RULE_MAGIC;
compile_rule->declared_clause_num = compile_item->declared_clause_num;
2023-01-31 20:39:53 +08:00
compile_rule->ref_table = compile_schema;
compile_rule->ex_data = ALLOC(void *, 1);
compile_rule->table_line_len = strlen(table_line) + 1;
compile_rule->table_line = ALLOC(char, compile_rule->table_line_len);
memcpy(compile_rule->table_line, table_line, compile_rule->table_line_len);
2023-01-31 20:39:53 +08:00
compile_rule->evaluation_order = compile_item->evaluation_order;
if (compile_schema->ex_schema != NULL) {
*(compile_rule->ex_data) = rule_ex_data_new(compile_schema->table_id, compile_rule->table_line,
compile_schema->ex_schema);
2023-01-31 20:39:53 +08:00
}
2023-01-31 20:39:53 +08:00
compile_rule->compile_id = compile_item->compile_id;
pthread_rwlock_init(&compile_rule->rwlock, NULL);
}
void destroy_compile_rule(struct compile_rule *compile_rule)
{
struct compile_schema *schema = compile_rule->ref_table;
assert(compile_rule->magic_num==COMPILE_RULE_MAGIC);
if (schema->ex_schema != NULL) {
rule_ex_data_free(schema->table_id, compile_rule->ex_data, schema->ex_schema);
*compile_rule->ex_data = NULL;
}
FREE(compile_rule->ex_data);
2023-01-31 20:39:53 +08:00
compile_rule->declared_clause_num = -1;
FREE(compile_rule->table_line);
FREE(compile_rule);
2023-01-31 20:39:53 +08:00
}
void compile_runtime_ex_data_iterate(struct compile_runtime *compile_rt,
struct compile_schema *compile_schema)
{
if (NULL == compile_rt || NULL == compile_schema ||
NULL == compile_schema->ex_schema) {
return;
}
compile_runtime_user_data_iterate(compile_rt, rule_ex_data_new_cb,
compile_schema->ex_schema,
compile_schema->table_id);
}
void *compile_runtime_get_ex_data(struct compile_runtime *compile_rt,
struct compile_schema *compile_schema,
long long compile_id)
{
if (NULL == compile_rt || NULL == compile_schema || compile_id < 0) {
return NULL;
}
struct compile_rule *compile_rule = NULL;
compile_rule = (struct compile_rule *)compile_runtime_get_user_data(compile_rt, compile_id, 0);
if (NULL == compile_rule) {
return NULL;
}
void *ex_data = NULL;
struct ex_data_schema *ex_schema = compile_schema->ex_schema;
ex_schema->dup_func(compile_schema->table_id, &ex_data, compile_rule->ex_data,
ex_schema->argl, ex_schema->argp);
return ex_data;
}
2023-02-03 17:28:14 +08:00
int compile_runtime_update(void *compile_runtime, void *compile_schema,
const char *line, int valid_column)
2023-01-30 21:59:35 +08:00
{
2023-02-03 17:28:14 +08:00
if (NULL == compile_runtime || NULL == compile_schema ||
NULL == line) {
2023-01-30 21:59:35 +08:00
return -1;
}
int ret = -1;
struct maat_compile *compile = NULL;
struct compile_item *compile_item = NULL;
struct compile_schema *schema = (struct compile_schema *)compile_schema;
struct compile_runtime *compile_rt = (struct compile_runtime *)compile_runtime;
int is_valid = get_column_value(line, valid_column);
if (is_valid < 0) {
return -1;
2023-02-16 16:45:06 +08:00
}
long long compile_id = get_column_value(line, schema->compile_id_column);
if (compile_id < 0) {
return -1;
}
2023-02-16 16:45:06 +08:00
if (0 == is_valid) {
2023-01-30 21:59:35 +08:00
//delete
2023-02-16 16:45:06 +08:00
pthread_rwlock_wrlock(&compile_rt->rwlock);
compile = maat_compile_hash_find(&(compile_rt->compile_hash), compile_id);
if (NULL == compile) {
pthread_rwlock_unlock(&compile_rt->rwlock);
return -1;
}
ret = maat_compile_hash_remove(&(compile_rt->compile_hash), compile,
compile_rt->ref_garbage_bin);
2023-02-16 16:45:06 +08:00
pthread_rwlock_unlock(&compile_rt->rwlock);
2023-01-30 21:59:35 +08:00
if (ret < 0) {
2023-01-31 20:39:53 +08:00
log_error(compile_rt->logger, MODULE_COMPILE,
"remove compile table(table_id:%d) compile(compile_id:%lld) from compile_hash failed",
2023-01-30 21:59:35 +08:00
schema->table_id, compile_id);
return -1;
}
} else {
//add
2023-02-16 16:45:06 +08:00
pthread_rwlock_wrlock(&compile_rt->rwlock);
2023-01-30 21:59:35 +08:00
compile_item = compile_item_new(line, schema, compile_rt->logger);
2023-01-31 20:39:53 +08:00
if (NULL == compile_item) {
2023-02-16 16:45:06 +08:00
pthread_rwlock_unlock(&compile_rt->rwlock);
2023-01-31 20:39:53 +08:00
return -1;
}
2023-01-30 21:59:35 +08:00
struct compile_rule *compile_rule = ALLOC(struct compile_rule, 1);
compile_item_to_compile_rule(compile_item, schema, compile_rule, line);
2023-01-30 21:59:35 +08:00
compile_item_free(compile_item);
compile_item = NULL;
compile = maat_compile_new(compile_rule->compile_id);
if (NULL == compile) {
destroy_compile_rule(compile_rule);
2023-02-16 16:45:06 +08:00
pthread_rwlock_unlock(&compile_rt->rwlock);
2023-01-31 20:39:53 +08:00
log_error(compile_rt->logger, MODULE_COMPILE,
2023-01-30 21:59:35 +08:00
"maat_compile_new failed, compile_table(table_id:%d) compile_id:%d",
schema->table_id, compile_item->compile_id);
return -1;
}
2023-02-03 17:28:14 +08:00
maat_compile_set(compile, compile_rule->declared_clause_num, compile_rule,
(void (*)(void *))destroy_compile_rule);
2023-02-16 16:45:06 +08:00
struct maat_compile *tmp_compile = maat_compile_hash_find(&(compile_rt->compile_hash), compile_id);
if (tmp_compile != NULL) {
maat_compile_hash_set(&(compile_rt->compile_hash), compile_id, compile);
} else {
maat_compile_hash_add(&(compile_rt->compile_hash), compile_id, compile);
2023-01-30 21:59:35 +08:00
}
// printf("compile_runtime_update compile_id:%lu, compile->declared_clause_num:%d\n",
// compile->compile_id, compile->declared_clause_num);
2023-02-16 16:45:06 +08:00
pthread_rwlock_unlock(&compile_rt->rwlock);
2023-01-30 21:59:35 +08:00
}
return 0;
}
2023-02-03 17:28:14 +08:00
int group2compile_runtime_update(void *g2c_runtime, void *g2c_schema,
const char *line, int valid_column)
2023-01-30 21:59:35 +08:00
{
if (NULL == g2c_runtime || NULL == g2c_schema || NULL == line) {
return -1;
}
struct group2compile_schema *schema = (struct group2compile_schema *)g2c_schema;
struct group2compile_runtime *g2c_rt = (struct group2compile_runtime *)g2c_runtime;
struct compile_runtime *compile_rt = g2c_rt->ref_compile_rt;
struct group2group_runtime *g2g_rt = g2c_rt->ref_g2g_rt;
int is_valid = get_column_value(line, valid_column);
if (is_valid < 0) {
return -1;
}
int ret = -1;
2023-01-31 20:39:53 +08:00
struct group2compile_item *g2c_item = group2compile_item_new(line, schema, compile_rt->logger);
if (NULL == g2c_item) {
return -1;
}
struct maat_group *group = NULL;
2023-01-30 21:59:35 +08:00
if (0 == is_valid) {
//delete
group = group2group_runtime_find_group(g2g_rt, g2c_item->group_id);
if (!group) {
log_error(compile_rt->logger, MODULE_COMPILE,
"Remove group %d from compile %d failed, group is not exisited.",
g2c_item->group_id, g2c_item->compile_id);
return -1;
}
pthread_rwlock_wrlock(&compile_rt->rwlock);
ret = maat_remove_group_from_compile(&(compile_rt->compile_hash), g2c_item,
2023-01-31 20:39:53 +08:00
compile_rt->ref_garbage_bin, compile_rt->logger);
pthread_rwlock_unlock(&compile_rt->rwlock);
2023-01-30 21:59:35 +08:00
if (0 == ret) {
if (g2c_item->not_flag) {
g2c_rt->not_flag_group--;
}
maat_group_ref_dec(group);
2023-01-30 21:59:35 +08:00
}
} else {
//add
group = group2group_runtime_find_group(g2g_rt, g2c_item->group_id);
if (!group) {
group = group2group_runtime_add_group(g2g_rt, g2c_item->group_id);
}
pthread_rwlock_wrlock(&compile_rt->rwlock);
ret = maat_add_group_to_compile(&(compile_rt->compile_hash), g2c_item, compile_rt->logger);
pthread_rwlock_unlock(&compile_rt->rwlock);
2023-01-30 21:59:35 +08:00
if (0 == ret) {
if (g2c_item->not_flag) {
g2c_rt->not_flag_group++;
}
maat_group_ref_inc(group);
2023-01-30 21:59:35 +08:00
}
}
2023-01-30 21:59:35 +08:00
group2compile_item_free(g2c_item);
return ret;
}
2023-02-09 22:13:15 +08:00
int compile_runtime_commit(void *compile_runtime, const char *table_name)
2023-01-30 21:59:35 +08:00
{
2023-01-31 20:39:53 +08:00
if (NULL == compile_runtime) {
return -1;
}
2023-01-30 21:59:35 +08:00
2023-01-31 20:39:53 +08:00
struct compile_runtime *compile_rt = (struct compile_runtime *)compile_runtime;
2023-02-16 16:45:06 +08:00
//TODO: no need add rdlock?
2023-02-09 22:13:15 +08:00
size_t compile_cnt = maat_compile_in_use_count(compile_rt->compile_hash);
2023-01-30 21:59:35 +08:00
if (0 == compile_cnt) {
return 0;
}
struct bool_matcher *old_bool_matcher = NULL;
struct bool_matcher *new_bool_matcher = NULL;
2023-01-31 20:39:53 +08:00
log_info(compile_rt->logger, MODULE_COMPILE,
2023-02-09 22:13:15 +08:00
"table[%s] committing %zu compile rules for rebuilding compile bool_matcher engine",
table_name, compile_cnt);
2023-01-31 20:39:53 +08:00
int ret = 0;
new_bool_matcher = maat_compile_bool_matcher_new(compile_rt->compile_hash,
&compile_rt->clause_id_generator,
compile_rt->logger);
2023-01-30 21:59:35 +08:00
if (NULL == new_bool_matcher) {
2023-01-31 20:39:53 +08:00
log_error(compile_rt->logger, MODULE_COMPILE,
2023-02-09 22:13:15 +08:00
"table[%s] rebuild compile bool_matcher engine failed when update %zu compile rules",
table_name, compile_cnt);
2023-01-30 21:59:35 +08:00
ret = -1;
}
2023-02-16 16:45:06 +08:00
pthread_rwlock_wrlock(&compile_rt->rwlock);
2023-01-30 21:59:35 +08:00
old_bool_matcher = compile_rt->bm;
compile_rt->bm = new_bool_matcher;
2023-02-16 16:45:06 +08:00
pthread_rwlock_unlock(&compile_rt->rwlock);
2023-02-03 17:28:14 +08:00
maat_garbage_bagging(compile_rt->ref_garbage_bin, old_bool_matcher,
(void (*)(void*))maat_compile_bool_matcher_free);
2023-01-30 21:59:35 +08:00
compile_rt->rule_num = compile_cnt;
return ret;
}
2023-02-03 17:28:14 +08:00
static int compile_sort_para_compare(const struct compile_sort_para *a,
const struct compile_sort_para *b)
2023-01-30 21:59:35 +08:00
{
//If both of compile rule's evaluation order are specified, compile rule with small evaluation order is priority.
if (a->evaluation_order != 0 && b->evaluation_order != 0) {
if (a->evaluation_order - b->evaluation_order < 0) {
return -1;
} else if(a->evaluation_order - b->evaluation_order > 0) {
return 1;
}
} else if(a->evaluation_order + b->evaluation_order!= 0) {
//If one of compile rule's evaluation order is zero, compile rule with big evaluation order is priority.
return (a->evaluation_order - b->evaluation_order > 0) ? -1 : 1;
}
//If compile rule's execute sequences are not specified or equal.
if (a->declared_clause_num != b->declared_clause_num) {
return (a->declared_clause_num - b->declared_clause_num);
} else {
return (b->compile_id - a->compile_id);
}
}
2023-02-03 17:28:14 +08:00
static void compile_sort_para_set(struct compile_sort_para *para,
const struct compile_rule *compile_relation,
void *user)
2023-01-30 21:59:35 +08:00
{
para->compile_id = compile_relation->compile_id;
para->evaluation_order = compile_relation->evaluation_order;
para->declared_clause_num = compile_relation->declared_clause_num;
para->user = user;
}
static int compare_compile_rule(const void *a, const void *b)
{
const struct compile_rule *ra = *(const struct compile_rule **)a;
const struct compile_rule *rb = *(const struct compile_rule **)b;
struct compile_sort_para sa, sb;
compile_sort_para_set(&sa, ra, NULL);
compile_sort_para_set(&sb, rb, NULL);
return compile_sort_para_compare(&sa, &sb);
}
2023-02-03 17:28:14 +08:00
int compile_runtime_match(struct compile_runtime *compile_rt,
2023-02-22 15:22:41 +08:00
long long *compile_ids, size_t compile_ids_size,
2023-02-03 17:28:14 +08:00
struct maat_state *state)
2023-01-30 21:59:35 +08:00
{
2023-02-03 17:28:14 +08:00
struct maat_compile_state *compile_state = state->compile_state;
int is_last_scan = state->is_last_scan;
struct compile_rule *compile_rules[compile_ids_size];
2023-01-30 21:59:35 +08:00
// all hit clause_id -> compile_id
2023-02-16 16:45:06 +08:00
pthread_rwlock_rdlock(&compile_rt->rwlock);
2023-02-03 17:28:14 +08:00
size_t bool_match_ret = maat_compile_bool_matcher_match(compile_rt->bm,
is_last_scan, compile_state,
(void **)compile_rules,
compile_ids_size);
2023-02-16 16:45:06 +08:00
pthread_rwlock_unlock(&compile_rt->rwlock);
2023-01-30 21:59:35 +08:00
if (bool_match_ret > 0) {
2023-02-03 17:28:14 +08:00
qsort(compile_rules, bool_match_ret, sizeof(struct compile_rule *),
compare_compile_rule);
2023-01-30 21:59:35 +08:00
}
for (size_t i = 0; i < bool_match_ret; i++) {
2023-02-03 17:28:14 +08:00
compile_ids[i] = compile_rules[i]->compile_id;
2023-01-30 21:59:35 +08:00
}
return MIN(bool_match_ret, compile_ids_size);
2023-02-03 17:28:14 +08:00
}
int maat_compile_state_update(struct maat_item *item_hash, int vtable_id,
2023-02-22 15:22:41 +08:00
long long *hit_item_ids, size_t hit_item_cnt,
2023-02-03 17:28:14 +08:00
size_t *n_hit_group_id, struct maat_state *state)
{
struct maat_item *item = NULL;
2023-02-22 15:22:41 +08:00
long long hit_group_ids[MAX_SCANNER_HIT_GROUP_NUM];
memset(hit_group_ids, 0, sizeof(hit_group_ids));
size_t hit_group_cnt = 0;
2023-02-03 17:28:14 +08:00
void *g2g_rt = table_manager_get_runtime(state->maat_instance->tbl_mgr,
state->maat_instance->g2g_table_id);
if (NULL == g2g_rt) {
return -1;
}
for (size_t i = 0; i < hit_item_cnt; i++) {
2023-02-22 15:22:41 +08:00
HASH_FIND(hh, item_hash, &(hit_item_ids[i]), sizeof(long long), item);
2023-02-16 11:13:23 +08:00
//assert(item != NULL);
2023-02-03 17:28:14 +08:00
if (!item) {
2023-02-16 11:13:23 +08:00
// item config has been deleted
2023-02-03 17:28:14 +08:00
continue;
}
if (hit_group_cnt >= MAX_SCANNER_HIT_GROUP_NUM) {
hit_group_cnt = MAX_SCANNER_HIT_GROUP_NUM;
2023-02-03 17:28:14 +08:00
//Prevent group_id_array out of bounds
} else {
hit_group_ids[hit_group_cnt++] = item->group_id;
2023-02-03 17:28:14 +08:00
}
// update hit path
maat_compile_state_update_hit_path(state->compile_state, hit_item_ids[i],
item->group_id, vtable_id, state->scan_cnt, i);
}
*n_hit_group_id = hit_group_cnt;
/* update hit clause */
int compile_table_ids[MAX_COMPILE_TABLE_NUM] = {0};
size_t compile_table_cnt = 0;
if (0 == state->n_compile_table) {
compile_table_ids[0] = state->maat_instance->default_compile_table_id;
compile_table_cnt = 1;
2023-02-03 17:28:14 +08:00
} else {
for (size_t i = 0; i < state->n_compile_table; i++) {
2023-03-01 17:44:07 +08:00
compile_table_ids[i] = maat_get_table_id(state->maat_instance, state->compile_tables[i]);
}
compile_table_cnt = state->n_compile_table;
}
for (size_t idx = 0; idx < compile_table_cnt; idx++) {
void *compile_rt = table_manager_get_runtime(state->maat_instance->tbl_mgr,
compile_table_ids[idx]);
for (size_t i = 0; i < hit_group_cnt; i++) {
2023-02-22 15:22:41 +08:00
long long top_group_ids[MAX_SCANNER_HIT_GROUP_NUM];
memset(top_group_ids, 0, sizeof(top_group_ids));
int top_group_cnt = group2group_runtime_get_top_groups(g2g_rt, &hit_group_ids[i],
1, top_group_ids);
for (int j = 0; j < top_group_cnt; j++) {
maat_compile_state_update_hit_clause(state->compile_state, compile_rt,
top_group_ids[j], vtable_id);
}
2023-02-03 17:28:14 +08:00
}
}
return 0;
2023-02-22 15:22:41 +08:00
}