2023-01-30 21:59:35 +08:00
|
|
|
/*
|
|
|
|
|
**********************************************************************************************
|
2023-05-04 17:10:19 +08:00
|
|
|
* File: maat_compile.c
|
2023-01-30 21:59:35 +08:00
|
|
|
* Description:
|
2023-05-04 17:10:19 +08:00
|
|
|
* Authors: Liu wentan <liuwentan@geedgenetworks.com>
|
2023-01-30 21:59:35 +08:00
|
|
|
* Date: 2022-10-31
|
2023-05-04 17:10:19 +08:00
|
|
|
* Copyright: (c) Since 2022 Geedge Networks, Ltd. All rights reserved.
|
2023-01-30 21:59:35 +08:00
|
|
|
***********************************************************************************************
|
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
#include <assert.h>
|
|
|
|
|
#include <pthread.h>
|
2023-02-21 11:27:18 +08:00
|
|
|
#include <linux/limits.h>
|
2023-01-30 21:59:35 +08:00
|
|
|
|
|
|
|
|
#include "maat_utils.h"
|
|
|
|
|
#include "log/log.h"
|
2023-06-05 15:14:45 +08:00
|
|
|
#include "maat.h"
|
2023-01-30 21:59:35 +08:00
|
|
|
#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"
|
2023-02-23 11:37:02 +08:00
|
|
|
#include "maat_ex_data.h"
|
2023-02-03 17:28:14 +08:00
|
|
|
#include "maat_table.h"
|
2023-06-08 17:43:05 +08:00
|
|
|
#include "alignment.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;
|
2023-02-21 11:27:18 +08:00
|
|
|
int declared_clause_num_column;
|
2023-04-13 18:53:36 +08:00
|
|
|
int set_flag;
|
|
|
|
|
struct ex_data_schema ex_schema;
|
2023-02-09 22:13:15 +08:00
|
|
|
struct table_manager *ref_tbl_mgr;
|
2023-04-05 21:09:19 +08:00
|
|
|
struct log_handle *logger;
|
2023-04-21 11:18:30 +08:00
|
|
|
int table_id; //ugly
|
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;
|
2023-02-21 11:27:18 +08:00
|
|
|
int declared_clause_num;
|
2023-01-30 21:59:35 +08:00
|
|
|
};
|
|
|
|
|
|
|
|
|
|
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;
|
|
|
|
|
};
|
|
|
|
|
|
2023-04-21 17:19:43 +08:00
|
|
|
struct maat_literal_id {
|
|
|
|
|
long long group_id;
|
|
|
|
|
long long vtable_id;
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
struct maat_clause {
|
|
|
|
|
long long clause_id;
|
|
|
|
|
size_t n_literal_id;
|
|
|
|
|
struct maat_literal_id *literal_ids;
|
|
|
|
|
UT_hash_handle hh;
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
struct literal_clause {
|
|
|
|
|
struct maat_literal_id key;
|
|
|
|
|
UT_array *clause_ids;
|
|
|
|
|
UT_hash_handle hh;
|
|
|
|
|
};
|
|
|
|
|
|
2023-05-11 11:21:46 +08:00
|
|
|
struct compile_rule {
|
2023-06-09 16:44:47 +08:00
|
|
|
uint32_t magic_num;
|
|
|
|
|
int declared_clause_num;
|
2023-05-11 11:21:46 +08:00
|
|
|
long long compile_id;
|
|
|
|
|
char *table_line;
|
|
|
|
|
size_t table_line_len;
|
|
|
|
|
struct compile_schema *ref_schema;
|
|
|
|
|
void **ex_data;
|
2023-06-09 16:44:47 +08:00
|
|
|
char table_name[MAX_NAME_STR_LEN];
|
2023-05-11 11:21:46 +08:00
|
|
|
};
|
|
|
|
|
|
2023-05-17 14:34:34 +08:00
|
|
|
struct group_reference {
|
|
|
|
|
long long group_id;
|
|
|
|
|
size_t ref_by_compile_cnt;
|
|
|
|
|
UT_hash_handle hh;
|
|
|
|
|
};
|
|
|
|
|
|
2023-01-30 21:59:35 +08:00
|
|
|
/* compile_runtime and group2compile_runtime share compile_hash_map */
|
|
|
|
|
struct compile_runtime {
|
|
|
|
|
struct bool_matcher *bm;
|
2023-04-14 11:32:59 +08:00
|
|
|
struct rcu_hash_table *cfg_hash_tbl; // <compile_id, struct maat_compile>
|
2023-03-15 11:36:54 +08:00
|
|
|
struct maat_runtime *ref_maat_rt;
|
2023-03-29 22:25:14 +08:00
|
|
|
time_t version;
|
2023-03-29 14:29:34 +08:00
|
|
|
struct maat_clause *clause_by_literals_hash;
|
2023-04-21 17:19:43 +08:00
|
|
|
struct literal_clause *literal2clause_hash;
|
2023-05-17 14:34:34 +08:00
|
|
|
struct group_reference *group_ref_hash;
|
|
|
|
|
pthread_mutex_t mutex;
|
2023-04-12 19:20:05 +08:00
|
|
|
long long rule_num;
|
2023-04-20 15:34:56 +08:00
|
|
|
long long update_err_cnt;
|
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;
|
2023-04-12 19:20:05 +08:00
|
|
|
long long rule_num;
|
2023-04-20 15:34:56 +08:00
|
|
|
long long update_err_cnt;
|
2023-01-30 21:59:35 +08:00
|
|
|
struct compile_runtime *ref_compile_rt;
|
|
|
|
|
};
|
2023-04-10 14:13:41 +08:00
|
|
|
|
2023-01-30 21:59:35 +08:00
|
|
|
struct maat_clause_state {
|
2023-03-15 11:36:54 +08:00
|
|
|
long long clause_id;
|
|
|
|
|
UT_array *ut_literal_ids;
|
2023-05-04 17:10:19 +08:00
|
|
|
char not_flag; // 1 byte
|
2023-04-10 14:13:41 +08:00
|
|
|
char in_use; // 1 byte
|
|
|
|
|
char pad[6]; // for 8 bytes alignment
|
2023-01-30 21:59:35 +08:00
|
|
|
};
|
|
|
|
|
|
|
|
|
|
struct compile_sort_para {
|
|
|
|
|
int declared_clause_num;
|
2023-02-22 15:22:41 +08:00
|
|
|
long long compile_id;
|
2023-01-30 21:59:35 +08:00
|
|
|
};
|
|
|
|
|
|
|
|
|
|
#define MAAT_COMPILE_MAGIC 0x4a5b6c7d
|
|
|
|
|
struct maat_compile {
|
2023-06-09 16:44:47 +08:00
|
|
|
uint32_t magic;
|
|
|
|
|
int actual_clause_num;
|
2023-01-30 21:59:35 +08:00
|
|
|
int declared_clause_num;
|
2023-05-04 17:10:19 +08:00
|
|
|
int not_clause_cnt;
|
2023-06-09 16:44:47 +08:00
|
|
|
long long compile_id;
|
|
|
|
|
char table_name[MAX_NAME_STR_LEN];
|
2023-01-30 21:59:35 +08:00
|
|
|
void *user_data;
|
|
|
|
|
void (*user_data_free)(void *);
|
|
|
|
|
struct maat_clause_state clause_states[MAX_ITEMS_PER_BOOL_EXPR];
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
struct maat_internal_hit_path {
|
2023-06-09 16:44:47 +08:00
|
|
|
long long item_id;
|
|
|
|
|
long long group_id;
|
2023-01-30 21:59:35 +08:00
|
|
|
int Nth_scan;
|
|
|
|
|
int Nth_hit_item;
|
2023-02-03 17:28:14 +08:00
|
|
|
int vtable_id;
|
2023-01-30 21:59:35 +08:00
|
|
|
};
|
|
|
|
|
|
|
|
|
|
struct maat_compile_state {
|
2023-06-09 16:44:47 +08:00
|
|
|
uint8_t this_scan_hit_item_flag;
|
|
|
|
|
uint8_t not_clause_hit_flag;
|
2023-01-30 21:59:35 +08:00
|
|
|
int Nth_scan;
|
2023-03-29 22:25:14 +08:00
|
|
|
time_t compile_rt_version;
|
2023-01-30 21:59:35 +08:00
|
|
|
|
|
|
|
|
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
|
|
|
};
|
|
|
|
|
|
2023-04-14 11:32:59 +08:00
|
|
|
UT_icd ut_literal_id_icd = {sizeof(struct maat_literal_id), NULL, NULL, NULL};
|
|
|
|
|
UT_icd ut_clause_id_icd = {sizeof(long long), NULL, NULL, NULL};
|
2023-06-05 15:14:45 +08:00
|
|
|
UT_icd ut_hit_group_icd = {sizeof(struct maat_hit_group), NULL, NULL, NULL};
|
2023-04-14 11:32:59 +08:00
|
|
|
UT_icd ut_hit_path_icd = {sizeof(struct maat_internal_hit_path), NULL, NULL, NULL};
|
|
|
|
|
|
|
|
|
|
struct maat_compile *maat_compile_new(long long compile_id)
|
|
|
|
|
{
|
|
|
|
|
struct maat_compile *compile = ALLOC(struct maat_compile, 1);
|
|
|
|
|
|
2023-05-11 11:21:46 +08:00
|
|
|
compile->magic = MAAT_COMPILE_MAGIC;
|
2023-04-14 11:32:59 +08:00
|
|
|
compile->compile_id = compile_id;
|
|
|
|
|
|
|
|
|
|
for(int i = 0; i < MAX_ITEMS_PER_BOOL_EXPR; i++) {
|
|
|
|
|
utarray_new(compile->clause_states[i].ut_literal_ids, &ut_literal_id_icd);
|
|
|
|
|
compile->clause_states[i].in_use=0;
|
|
|
|
|
compile->clause_states[i].clause_id = -1;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return compile;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
int maat_compile_set(struct maat_compile *compile, const char *table_name,
|
|
|
|
|
int declared_clause_num, void *user_data,
|
|
|
|
|
void (*user_data_free)(void *))
|
|
|
|
|
{
|
|
|
|
|
if (user_data != NULL && NULL == user_data_free) {
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
memset(compile->table_name, 0, sizeof(compile->table_name));
|
|
|
|
|
memcpy(compile->table_name, table_name, sizeof(compile->table_name));
|
|
|
|
|
compile->declared_clause_num = declared_clause_num;
|
|
|
|
|
compile->user_data = user_data;
|
|
|
|
|
compile->user_data_free = user_data_free;
|
|
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
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);
|
2023-04-22 10:46:21 +08:00
|
|
|
compile->user_data = NULL;
|
2023-04-14 11:32:59 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
for (int i = 0; i < MAX_ITEMS_PER_BOOL_EXPR; i++) {
|
|
|
|
|
clause_state = compile->clause_states + i;
|
|
|
|
|
|
|
|
|
|
if (clause_state->ut_literal_ids != NULL) {
|
|
|
|
|
utarray_free(clause_state->ut_literal_ids);
|
|
|
|
|
clause_state->ut_literal_ids = NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
clause_state->in_use = 0;
|
|
|
|
|
}
|
|
|
|
|
compile->magic = 0;
|
2023-05-23 03:23:39 +00:00
|
|
|
FREE(compile);
|
2023-04-14 11:32:59 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void rcu_maat_compile_free(void *user_ctx, void *data)
|
|
|
|
|
{
|
|
|
|
|
struct maat_compile *compile = (struct maat_compile *)data;
|
|
|
|
|
maat_compile_free(compile);
|
|
|
|
|
}
|
|
|
|
|
|
2023-04-05 21:09:19 +08:00
|
|
|
int compile_table_set_ex_data_schema(struct compile_schema *compile_schema, int table_id,
|
2023-03-02 14:52:31 +08:00
|
|
|
maat_ex_new_func_t *new_func,
|
|
|
|
|
maat_ex_free_func_t *free_func,
|
|
|
|
|
maat_ex_dup_func_t *dup_func,
|
2023-04-05 21:09:19 +08:00
|
|
|
long argl, void *argp)
|
2023-01-30 21:59:35 +08:00
|
|
|
{
|
2023-04-05 21:09:19 +08:00
|
|
|
if (1 == compile_schema->set_flag) {
|
|
|
|
|
log_error(compile_schema->logger, MODULE_COMPILE,
|
|
|
|
|
"[%s:%d] compile table(table_id:%d)ex schema has been set already, can't set again",
|
2023-03-02 14:52:31 +08:00
|
|
|
__FUNCTION__, __LINE__, table_id);
|
2023-01-30 21:59:35 +08:00
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
|
2023-04-05 21:09:19 +08:00
|
|
|
compile_schema->ex_schema.new_func = new_func;
|
|
|
|
|
compile_schema->ex_schema.free_func = free_func;
|
|
|
|
|
compile_schema->ex_schema.dup_func = dup_func;
|
|
|
|
|
compile_schema->ex_schema.argl = argl;
|
|
|
|
|
compile_schema->ex_schema.argp = argp;
|
|
|
|
|
compile_schema->set_flag = 1;
|
|
|
|
|
|
2023-02-23 11:37:02 +08:00
|
|
|
return 0;
|
2023-01-30 21:59:35 +08:00
|
|
|
}
|
|
|
|
|
|
2023-04-14 11:32:59 +08:00
|
|
|
void *compile_runtime_get_user_data(struct compile_runtime *compile_rt, long long compile_id)
|
2023-02-09 22:13:15 +08:00
|
|
|
{
|
2023-04-14 11:32:59 +08:00
|
|
|
struct maat_compile *compile = rcu_hash_find(compile_rt->cfg_hash_tbl,
|
|
|
|
|
(char *)&compile_id, sizeof(long long));
|
2023-02-09 22:13:15 +08:00
|
|
|
void *ret = NULL;
|
|
|
|
|
if (compile != NULL) {
|
|
|
|
|
ret = compile->user_data;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
|
}
|
|
|
|
|
|
2023-03-29 22:25:14 +08:00
|
|
|
void *rule_ex_data_new(const char *table_name, int table_id, const char *table_line,
|
2023-04-05 21:09:19 +08:00
|
|
|
struct ex_data_schema *ex_schema)
|
2023-02-21 11:27:18 +08:00
|
|
|
{
|
|
|
|
|
void *ex_data = NULL;
|
|
|
|
|
|
2023-03-29 22:25:14 +08:00
|
|
|
ex_schema->new_func(table_name, table_id, NULL, table_line, &ex_data,
|
|
|
|
|
ex_schema->argl, ex_schema->argp);
|
2023-02-21 11:27:18 +08:00
|
|
|
return ex_data;
|
|
|
|
|
}
|
|
|
|
|
|
2023-02-23 11:37:02 +08:00
|
|
|
void rule_ex_data_free(int table_id, void **ex_data, const struct ex_data_schema *ex_schema)
|
2023-02-21 11:27:18 +08:00
|
|
|
{
|
2023-02-23 11:37:02 +08:00
|
|
|
ex_schema->free_func(table_id, ex_data, ex_schema->argl, ex_schema->argp);
|
2023-02-21 11:27:18 +08:00
|
|
|
}
|
|
|
|
|
|
2023-03-29 22:25:14 +08:00
|
|
|
void rule_ex_data_new_cb(void *user_data, void *param, const char *table_name, int table_id)
|
2023-02-21 11:27:18 +08:00
|
|
|
{
|
2023-02-23 11:37:02 +08:00
|
|
|
struct ex_data_schema *ex_schema = (struct ex_data_schema *)param;
|
2023-02-21 11:27:18 +08:00
|
|
|
struct compile_rule *compile = (struct compile_rule *)user_data;
|
2023-02-23 11:37:02 +08:00
|
|
|
|
2023-03-29 22:25:14 +08:00
|
|
|
void *ad = rule_ex_data_new(table_name, table_id, compile->table_line, ex_schema);
|
2023-02-23 11:37:02 +08:00
|
|
|
*compile->ex_data = ad;
|
2023-02-21 11:27:18 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void compile_runtime_user_data_iterate(struct compile_runtime *compile_rt,
|
2023-03-29 22:25:14 +08:00
|
|
|
void (*callback)(void *user_data, void *param, const char *table_name, int table_id),
|
2023-02-23 11:37:02 +08:00
|
|
|
void *param, int table_id)
|
2023-02-21 11:27:18 +08:00
|
|
|
{
|
2023-04-14 11:32:59 +08:00
|
|
|
/* I'm in background_update_mutex, config update can't happen, so no need to lock cfg_hash_tbl */
|
|
|
|
|
void **data_array = NULL;
|
|
|
|
|
size_t data_cnt = rcu_hash_list(compile_rt->cfg_hash_tbl, &data_array);
|
2023-02-21 11:27:18 +08:00
|
|
|
|
2023-04-14 11:32:59 +08:00
|
|
|
for (size_t i = 0; i < data_cnt; i++) {
|
|
|
|
|
struct maat_compile *compile = (struct maat_compile *)data_array[i];
|
2023-02-21 11:27:18 +08:00
|
|
|
if (compile->user_data) {
|
2023-03-29 22:25:14 +08:00
|
|
|
callback(compile->user_data, param, compile->table_name, table_id);
|
2023-02-21 11:27:18 +08:00
|
|
|
}
|
|
|
|
|
}
|
2023-05-11 11:21:46 +08:00
|
|
|
|
2023-04-14 11:32:59 +08:00
|
|
|
FREE(data_array);
|
2023-02-21 11:27:18 +08:00
|
|
|
}
|
|
|
|
|
|
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
|
|
|
{
|
|
|
|
|
struct compile_schema *compile_schema = ALLOC(struct compile_schema, 1);
|
2023-04-05 21:09:19 +08:00
|
|
|
compile_schema->logger = logger;
|
2023-01-30 21:59:35 +08:00
|
|
|
|
|
|
|
|
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;
|
2023-03-27 15:52:47 +08:00
|
|
|
} else {
|
|
|
|
|
log_error(logger, MODULE_COMPILE,
|
2023-05-23 03:23:39 +00:00
|
|
|
"[%s:%d] table: <%s> schema has no table_id column",
|
|
|
|
|
__FUNCTION__, __LINE__, table_name);
|
2023-03-27 15:52:47 +08:00
|
|
|
goto error;
|
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,
|
2023-05-23 03:23:39 +00:00
|
|
|
"[%s:%d] table: <%s> schema has no custom column",
|
2023-03-02 14:52:31 +08:00
|
|
|
__FUNCTION__, __LINE__, 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;
|
2023-03-27 15:52:47 +08:00
|
|
|
} else {
|
|
|
|
|
log_error(logger, MODULE_COMPILE,
|
2023-05-23 03:23:39 +00:00
|
|
|
"[%s:%d] table: <%s> schema has no compile_id column",
|
|
|
|
|
__FUNCTION__, __LINE__, table_name);
|
2023-03-27 15:52:47 +08:00
|
|
|
goto error;
|
2023-01-30 21:59:35 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
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
|
|
|
}
|
|
|
|
|
|
|
|
|
|
custom_item = cJSON_GetObjectItem(item, "clause_num");
|
|
|
|
|
if (custom_item != NULL && custom_item->type == cJSON_Number) {
|
2023-02-21 11:27:18 +08:00
|
|
|
compile_schema->declared_clause_num_column = custom_item->valueint;
|
2023-03-27 15:52:47 +08:00
|
|
|
} else {
|
|
|
|
|
log_error(logger, MODULE_COMPILE,
|
2023-05-23 03:23:39 +00:00
|
|
|
"[%s:%d] table: <%s> schema has no clause_num column",
|
|
|
|
|
__FUNCTION__, __LINE__, table_name);
|
2023-03-27 15:52:47 +08:00
|
|
|
goto error;
|
2023-01-30 21:59:35 +08:00
|
|
|
}
|
|
|
|
|
|
2023-02-09 22:13:15 +08:00
|
|
|
compile_schema->ref_tbl_mgr = tbl_mgr;
|
2023-01-30 21:59:35 +08:00
|
|
|
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
|
|
|
{
|
|
|
|
|
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;
|
2023-03-27 15:52:47 +08:00
|
|
|
} else {
|
|
|
|
|
log_error(logger, MODULE_COMPILE,
|
2023-05-23 03:23:39 +00:00
|
|
|
"[%s:%d] table: <%s> schema has no table_id column",
|
2023-03-27 15:52:47 +08:00
|
|
|
__FUNCTION__, __LINE__, table_name);
|
|
|
|
|
goto error;
|
2023-01-31 20:39:53 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
item = cJSON_GetObjectItem(json, "associated_compile_table_id");
|
|
|
|
|
if (item != NULL && item->type == cJSON_Number) {
|
|
|
|
|
g2c_schema->associated_compile_table_id = item->valueint;
|
2023-03-27 15:52:47 +08:00
|
|
|
} else {
|
|
|
|
|
log_error(logger, MODULE_COMPILE,
|
2023-05-23 03:23:39 +00:00
|
|
|
"[%s:%d] table: <%s> schema has no associated_compile_table_id column",
|
2023-03-27 15:52:47 +08:00
|
|
|
__FUNCTION__, __LINE__, table_name);
|
|
|
|
|
goto error;
|
2023-01-30 21:59:35 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
item = cJSON_GetObjectItem(json, "custom");
|
|
|
|
|
if (item == NULL || item->type != cJSON_Object) {
|
2023-03-02 14:52:31 +08:00
|
|
|
log_error(logger, MODULE_COMPILE,
|
2023-05-23 03:23:39 +00:00
|
|
|
"[%s:%d] table: <%s> schema has no custom column",
|
2023-03-02 14:52:31 +08:00
|
|
|
__FUNCTION__, __LINE__, table_name);
|
2023-01-30 21:59:35 +08:00
|
|
|
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;
|
2023-03-27 15:52:47 +08:00
|
|
|
} else {
|
|
|
|
|
log_error(logger, MODULE_COMPILE,
|
2023-05-23 03:23:39 +00:00
|
|
|
"[%s:%d] table: <%s> schema has no group_id column",
|
2023-03-27 15:52:47 +08:00
|
|
|
__FUNCTION__, __LINE__, table_name);
|
|
|
|
|
goto error;
|
2023-01-30 21:59:35 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
custom_item = cJSON_GetObjectItem(item, "compile_id");
|
|
|
|
|
if (custom_item != NULL && custom_item->type == cJSON_Number) {
|
|
|
|
|
g2c_schema->compile_id_column = custom_item->valueint;
|
2023-03-27 15:52:47 +08:00
|
|
|
} else {
|
|
|
|
|
log_error(logger, MODULE_COMPILE,
|
2023-05-23 03:23:39 +00:00
|
|
|
"[%s:%d] table: <%s> schema has no compile_id column",
|
2023-03-27 15:52:47 +08:00
|
|
|
__FUNCTION__, __LINE__, table_name);
|
|
|
|
|
goto error;
|
2023-01-30 21:59:35 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
custom_item = cJSON_GetObjectItem(item, "not_flag");
|
|
|
|
|
if (custom_item != NULL && custom_item->type == cJSON_Number) {
|
|
|
|
|
g2c_schema->not_flag_column = custom_item->valueint;
|
2023-03-27 15:52:47 +08:00
|
|
|
} else {
|
|
|
|
|
log_error(logger, MODULE_COMPILE,
|
2023-05-23 03:23:39 +00:00
|
|
|
"[%s:%d] table: <%s> schema has no not_flag column",
|
2023-03-27 15:52:47 +08:00
|
|
|
__FUNCTION__, __LINE__, table_name);
|
|
|
|
|
goto error;
|
2023-01-30 21:59:35 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
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-03-27 15:52:47 +08:00
|
|
|
} else {
|
|
|
|
|
log_error(logger, MODULE_COMPILE,
|
2023-05-23 03:23:39 +00:00
|
|
|
"[%s:%d] table: <%s> schema has no virtual_table_name column",
|
2023-03-27 15:52:47 +08:00
|
|
|
__FUNCTION__, __LINE__, table_name);
|
|
|
|
|
goto error;
|
2023-01-30 21:59:35 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
custom_item = cJSON_GetObjectItem(item, "clause_index");
|
|
|
|
|
if (custom_item != NULL && custom_item->type == cJSON_Number) {
|
|
|
|
|
g2c_schema->clause_index_column = custom_item->valueint;
|
2023-03-27 15:52:47 +08:00
|
|
|
} else {
|
|
|
|
|
log_error(logger, MODULE_COMPILE,
|
2023-05-23 03:23:39 +00:00
|
|
|
"[%s:%d] table: <%s> schema has no clause_index column",
|
2023-03-27 15:52:47 +08:00
|
|
|
__FUNCTION__, __LINE__, table_name);
|
|
|
|
|
goto error;
|
2023-01-30 21:59:35 +08:00
|
|
|
}
|
|
|
|
|
|
2023-02-09 22:13:15 +08:00
|
|
|
g2c_schema->ref_tbl_mgr = tbl_mgr;
|
2023-01-30 21:59:35 +08:00
|
|
|
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,
|
2023-04-14 11:32:59 +08:00
|
|
|
const char *table_name, struct log_handle *logger)
|
2023-02-24 18:20:04 +08:00
|
|
|
{
|
|
|
|
|
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,
|
2023-05-23 03:23:39 +00:00
|
|
|
"[%s:%d] table: <%s> has no rule_tag in line:%s",
|
2023-04-14 11:32:59 +08:00
|
|
|
__FUNCTION__, __LINE__, table_name, line);
|
2023-02-24 18:20:04 +08:00
|
|
|
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,
|
2023-05-23 03:23:39 +00:00
|
|
|
"[%s:%d] table: <%s> has invalid tag format in line:%s",
|
2023-04-14 11:32:59 +08:00
|
|
|
__FUNCTION__, __LINE__, table_name, line);
|
2023-02-24 18:20:04 +08:00
|
|
|
return TAG_MATCH_ERR;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (TAG_MATCH_UNMATCHED == ret) {
|
2023-05-23 03:23:39 +00:00
|
|
|
log_error(logger, MODULE_COMPILE,
|
|
|
|
|
"[%s:%d] table: <%s> has unmatched tag in line:%s",
|
|
|
|
|
__FUNCTION__, __LINE__, table_name, line);
|
2023-02-24 18:20:04 +08:00
|
|
|
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,
|
2023-04-14 11:32:59 +08:00
|
|
|
const char *table_name, struct log_handle *logger)
|
2023-01-30 21:59:35 +08:00
|
|
|
{
|
2023-04-14 11:32:59 +08:00
|
|
|
int ret = compile_accept_tag_match(compile_schema, line, table_name, logger);
|
2023-02-24 18:20:04 +08:00
|
|
|
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-04-14 11:32:59 +08:00
|
|
|
log_error(logger, MODULE_COMPILE,
|
2023-05-23 03:23:39 +00:00
|
|
|
"[%s:%d] table: <%s> has no compile_id in line:%s",
|
2023-04-14 11:32:59 +08:00
|
|
|
__FUNCTION__, __LINE__, table_name, line);
|
2023-01-30 21:59:35 +08:00
|
|
|
goto error;
|
|
|
|
|
}
|
2023-02-22 15:08:52 +08:00
|
|
|
compile_item->compile_id = atoll(line + column_offset);
|
2023-01-30 21:59:35 +08:00
|
|
|
|
2023-02-21 11:27:18 +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,
|
2023-05-23 03:23:39 +00:00
|
|
|
"[%s:%d] table: <%s> has no clause_num in line:%s",
|
2023-04-14 11:32:59 +08:00
|
|
|
__FUNCTION__, __LINE__, table_name, line);
|
2023-01-30 21:59:35 +08:00
|
|
|
goto error;
|
|
|
|
|
}
|
2023-02-21 11:27:18 +08:00
|
|
|
compile_item->declared_clause_num = atoi(line + column_offset);
|
2023-01-30 21:59:35 +08:00
|
|
|
|
|
|
|
|
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-04-20 15:34:56 +08:00
|
|
|
void *compile_runtime_new(void *compile_schema, size_t max_thread_num,
|
2023-04-13 14:56:35 +08:00
|
|
|
struct maat_garbage_bin *garbage_bin,
|
|
|
|
|
struct log_handle *logger)
|
2023-01-30 21:59:35 +08:00
|
|
|
{
|
|
|
|
|
if (NULL == compile_schema) {
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
struct compile_runtime *compile_rt = ALLOC(struct compile_runtime, 1);
|
|
|
|
|
|
2023-05-23 03:23:39 +00:00
|
|
|
compile_rt->expr_match_buff = ALLOC(struct bool_expr_match,
|
|
|
|
|
max_thread_num * MAX_SCANNER_HIT_COMPILE_NUM);
|
2023-03-29 22:25:14 +08:00
|
|
|
compile_rt->version = time(NULL);
|
2023-04-14 11:32:59 +08:00
|
|
|
compile_rt->cfg_hash_tbl = rcu_hash_new(rcu_maat_compile_free, NULL);
|
2023-03-29 14:29:34 +08:00
|
|
|
compile_rt->clause_by_literals_hash = NULL;
|
2023-05-17 14:34:34 +08:00
|
|
|
compile_rt->literal2clause_hash = NULL;
|
|
|
|
|
compile_rt->group_ref_hash = NULL;
|
2023-01-30 21:59:35 +08:00
|
|
|
compile_rt->logger = logger;
|
|
|
|
|
compile_rt->ref_garbage_bin = garbage_bin;
|
2023-05-17 14:34:34 +08:00
|
|
|
pthread_mutex_init(&(compile_rt->mutex), NULL);
|
2023-01-30 21:59:35 +08:00
|
|
|
|
|
|
|
|
return compile_rt;
|
|
|
|
|
}
|
|
|
|
|
|
2023-04-21 17:19:43 +08:00
|
|
|
static void maat_clause_hash_free(struct maat_clause *clause_hash)
|
2023-03-29 14:29:34 +08:00
|
|
|
{
|
|
|
|
|
struct maat_clause *clause = NULL, *tmp_clause = NULL;
|
|
|
|
|
|
2023-04-21 17:19:43 +08:00
|
|
|
HASH_ITER (hh, clause_hash, clause, tmp_clause) {
|
|
|
|
|
HASH_DEL(clause_hash, clause);
|
2023-03-29 14:29:34 +08:00
|
|
|
FREE(clause->literal_ids);
|
|
|
|
|
clause->n_literal_id = 0;
|
|
|
|
|
FREE(clause);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2023-05-17 14:34:34 +08:00
|
|
|
static void literal2clause_hash_free(struct literal_clause *hash)
|
2023-04-21 17:19:43 +08:00
|
|
|
{
|
|
|
|
|
struct literal_clause *l2c_val = NULL, *tmp_l2c_val = NULL;
|
|
|
|
|
|
|
|
|
|
HASH_ITER(hh, hash, l2c_val, tmp_l2c_val) {
|
|
|
|
|
HASH_DEL(hash, l2c_val);
|
2023-05-08 18:51:53 +08:00
|
|
|
if (l2c_val->clause_ids != NULL) {
|
|
|
|
|
utarray_free(l2c_val->clause_ids);
|
|
|
|
|
l2c_val->clause_ids = NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
FREE(l2c_val);
|
2023-04-21 17:19:43 +08:00
|
|
|
}
|
|
|
|
|
assert(hash == NULL);
|
|
|
|
|
}
|
|
|
|
|
|
2023-05-17 14:34:34 +08:00
|
|
|
static void group_reference_hash_free(struct group_reference *group_ref_hash)
|
|
|
|
|
{
|
|
|
|
|
struct group_reference *group_ref = NULL, *tmp_group_ref = NULL;
|
|
|
|
|
|
|
|
|
|
HASH_ITER(hh, group_ref_hash, group_ref, tmp_group_ref) {
|
|
|
|
|
HASH_DEL(group_ref_hash, group_ref);
|
|
|
|
|
FREE(group_ref);
|
|
|
|
|
}
|
|
|
|
|
assert(group_ref_hash == NULL);
|
|
|
|
|
}
|
|
|
|
|
|
2023-04-21 17:19:43 +08:00
|
|
|
void garbage_literal2clause_hash_free(void *l2c_hash, void *arg)
|
|
|
|
|
{
|
|
|
|
|
literal2clause_hash_free((struct literal_clause *)l2c_hash);
|
|
|
|
|
}
|
|
|
|
|
|
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
|
|
|
|
2023-01-30 21:59:35 +08:00
|
|
|
if (compile_rt->bm != NULL) {
|
|
|
|
|
bool_matcher_free(compile_rt->bm);
|
2023-03-29 14:29:34 +08:00
|
|
|
compile_rt->bm = NULL;
|
2023-01-30 21:59:35 +08:00
|
|
|
}
|
|
|
|
|
|
2023-04-14 11:32:59 +08:00
|
|
|
if (compile_rt->cfg_hash_tbl != NULL) {
|
|
|
|
|
rcu_hash_free(compile_rt->cfg_hash_tbl);
|
|
|
|
|
compile_rt->cfg_hash_tbl = NULL;
|
2023-01-30 21:59:35 +08:00
|
|
|
}
|
2023-03-15 11:36:54 +08:00
|
|
|
|
2023-04-21 17:19:43 +08:00
|
|
|
if (compile_rt->literal2clause_hash != NULL) {
|
|
|
|
|
literal2clause_hash_free(compile_rt->literal2clause_hash);
|
|
|
|
|
compile_rt->literal2clause_hash = NULL;
|
|
|
|
|
}
|
|
|
|
|
|
2023-03-29 14:29:34 +08:00
|
|
|
if (compile_rt->clause_by_literals_hash != NULL) {
|
2023-04-21 17:19:43 +08:00
|
|
|
maat_clause_hash_free(compile_rt->clause_by_literals_hash);
|
2023-03-29 14:29:34 +08:00
|
|
|
compile_rt->clause_by_literals_hash = NULL;
|
|
|
|
|
}
|
2023-03-30 21:03:59 +08:00
|
|
|
|
2023-05-17 14:34:34 +08:00
|
|
|
pthread_mutex_lock(&(compile_rt->mutex));
|
|
|
|
|
if (compile_rt->group_ref_hash != NULL) {
|
|
|
|
|
group_reference_hash_free(compile_rt->group_ref_hash);
|
|
|
|
|
compile_rt->group_ref_hash = NULL;
|
|
|
|
|
}
|
|
|
|
|
pthread_mutex_unlock(&(compile_rt->mutex));
|
|
|
|
|
pthread_mutex_destroy(&(compile_rt->mutex));
|
|
|
|
|
|
2023-02-03 17:28:14 +08:00
|
|
|
if (compile_rt->expr_match_buff != NULL) {
|
|
|
|
|
FREE(compile_rt->expr_match_buff);
|
|
|
|
|
}
|
2023-03-30 21:03:59 +08:00
|
|
|
|
2023-01-30 21:59:35 +08:00
|
|
|
FREE(compile_rt);
|
|
|
|
|
}
|
|
|
|
|
|
2023-03-15 11:36:54 +08:00
|
|
|
void compile_runtime_init(void *compile_runtime, struct maat_runtime *maat_rt)
|
|
|
|
|
{
|
|
|
|
|
if (NULL == compile_runtime) {
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
struct compile_runtime *compile_rt = (struct compile_runtime *)compile_runtime;
|
|
|
|
|
compile_rt->ref_maat_rt = maat_rt;
|
|
|
|
|
}
|
|
|
|
|
|
2023-04-20 15:34:56 +08:00
|
|
|
void *group2compile_runtime_new(void *g2c_schema, size_t max_thread_num,
|
2023-04-13 14:56:35 +08:00
|
|
|
struct maat_garbage_bin *garbage_bin,
|
2023-01-30 21:59:35 +08:00
|
|
|
struct log_handle *logger)
|
|
|
|
|
{
|
2023-04-13 14:56:35 +08:00
|
|
|
if (NULL == g2c_schema) {
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
2023-01-30 21:59:35 +08:00
|
|
|
struct group2compile_runtime *g2c_rt = ALLOC(struct group2compile_runtime, 1);
|
2023-04-13 14:56:35 +08:00
|
|
|
|
2023-01-30 21:59:35 +08:00
|
|
|
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;
|
|
|
|
|
}
|
|
|
|
|
|
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,
|
2023-03-29 22:25:14 +08:00
|
|
|
const char *table_name, struct log_handle *logger)
|
2023-01-30 21:59:35 +08:00
|
|
|
{
|
|
|
|
|
size_t column_offset = 0;
|
|
|
|
|
size_t column_len = 0;
|
2023-06-09 16:44:47 +08:00
|
|
|
char vtable_name[MAX_NAME_STR_LEN] = {0};
|
2023-01-30 21:59:35 +08:00
|
|
|
struct group2compile_item *g2c_item = ALLOC(struct group2compile_item, 1);
|
|
|
|
|
|
2023-05-30 16:16:18 +08:00
|
|
|
int ret = get_column_pos(line, g2c_schema->group_id_column, &column_offset,
|
|
|
|
|
&column_len);
|
2023-01-30 21:59:35 +08:00
|
|
|
if (ret < 0) {
|
|
|
|
|
log_error(logger, MODULE_COMPILE,
|
2023-05-30 16:16:18 +08:00
|
|
|
"[%s:%d] g2c table:<%s> has no group_id in line:%s",
|
2023-03-29 22:25:14 +08:00
|
|
|
__FUNCTION__, __LINE__, table_name, line);
|
2023-01-30 21:59:35 +08:00
|
|
|
goto error;
|
|
|
|
|
}
|
2023-02-22 15:08:52 +08:00
|
|
|
g2c_item->group_id = atoll(line + column_offset);
|
2023-01-30 21:59:35 +08:00
|
|
|
|
2023-05-30 16:16:18 +08:00
|
|
|
ret = get_column_pos(line, g2c_schema->compile_id_column, &column_offset,
|
|
|
|
|
&column_len);
|
2023-01-30 21:59:35 +08:00
|
|
|
if (ret < 0) {
|
|
|
|
|
log_error(logger, MODULE_COMPILE,
|
2023-05-30 16:16:18 +08:00
|
|
|
"[%s:%d] g2c table:<%s> has no compile_id in line:%s",
|
2023-03-29 22:25:14 +08:00
|
|
|
__FUNCTION__, __LINE__, table_name, line);
|
2023-01-30 21:59:35 +08:00
|
|
|
goto error;
|
|
|
|
|
}
|
2023-02-22 15:08:52 +08:00
|
|
|
g2c_item->compile_id = atoll(line + column_offset);
|
2023-01-30 21:59:35 +08:00
|
|
|
|
2023-05-30 16:16:18 +08:00
|
|
|
ret = get_column_pos(line, g2c_schema->not_flag_column, &column_offset,
|
|
|
|
|
&column_len);
|
2023-01-30 21:59:35 +08:00
|
|
|
if (ret < 0) {
|
|
|
|
|
log_error(logger, MODULE_COMPILE,
|
2023-05-30 16:16:18 +08:00
|
|
|
"[%s:%d] g2c table:<%s> has no NOT_flag in line:%s ",
|
2023-03-29 22:25:14 +08:00
|
|
|
__FUNCTION__, __LINE__, table_name, line);
|
2023-01-30 21:59:35 +08:00
|
|
|
goto error;
|
|
|
|
|
}
|
|
|
|
|
g2c_item->not_flag = atoi(line + column_offset);
|
|
|
|
|
|
2023-05-30 16:16:18 +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,
|
2023-05-30 16:16:18 +08:00
|
|
|
"[%s:%d] g2c table:<%s>has no virtual_table_name in line:%s",
|
2023-03-29 22:25:14 +08:00
|
|
|
__FUNCTION__, __LINE__, table_name, line);
|
2023-01-30 21:59:35 +08:00
|
|
|
goto error;
|
|
|
|
|
}
|
|
|
|
|
|
2023-06-09 16:44:47 +08:00
|
|
|
if (column_len > MAX_NAME_STR_LEN) {
|
2023-01-30 21:59:35 +08:00
|
|
|
log_error(logger, MODULE_COMPILE,
|
2023-05-30 16:16:18 +08:00
|
|
|
"[%s:%d] g2c table:<%s> virtual_table_name length exceed "
|
|
|
|
|
"maxium:%d in line:%s", __FUNCTION__, __LINE__, table_name,
|
2023-06-09 16:44:47 +08:00
|
|
|
MAX_NAME_STR_LEN, line);
|
2023-01-30 21:59:35 +08:00
|
|
|
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)) {
|
2023-05-30 16:16:18 +08:00
|
|
|
g2c_item->vtable_id = table_manager_get_table_id(g2c_schema->ref_tbl_mgr,
|
|
|
|
|
vtable_name);
|
2023-02-03 17:28:14 +08:00
|
|
|
if (g2c_item->vtable_id < 0) {
|
2023-01-30 21:59:35 +08:00
|
|
|
log_error(logger, MODULE_COMPILE,
|
2023-05-30 16:16:18 +08:00
|
|
|
"[%s:%d] g2c table:<%s> has unknown virtual table:%s in line:%s",
|
2023-05-23 03:23:39 +00:00
|
|
|
__FUNCTION__, __LINE__, table_name, vtable_name, line);
|
2023-01-30 21:59:35 +08:00
|
|
|
goto error;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2023-05-30 16:16:18 +08:00
|
|
|
ret = get_column_pos(line, g2c_schema->clause_index_column, &column_offset,
|
|
|
|
|
&column_len);
|
2023-01-30 21:59:35 +08:00
|
|
|
if (ret < 0) {
|
|
|
|
|
log_error(logger, MODULE_COMPILE,
|
2023-05-30 16:16:18 +08:00
|
|
|
"[%s:%d] g2c table:<%s> has no clause_index in line:%s",
|
2023-03-29 22:25:14 +08:00
|
|
|
__FUNCTION__, __LINE__, table_name, line);
|
2023-01-30 21:59:35 +08:00
|
|
|
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);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
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-04-04 09:31:20 +08:00
|
|
|
long long 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;
|
|
|
|
|
|
2023-05-04 17:10:19 +08:00
|
|
|
clause_state->not_flag = clause_not_flag;
|
2023-01-30 21:59:35 +08:00
|
|
|
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;
|
2023-03-15 11:36:54 +08:00
|
|
|
tmp = (struct maat_literal_id *)utarray_find(clause_state->ut_literal_ids,
|
2023-01-31 20:39:53 +08:00
|
|
|
literal_id, compare_literal_id);
|
2023-01-30 21:59:35 +08:00
|
|
|
if (tmp) {
|
2023-02-22 15:08:52 +08:00
|
|
|
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 {
|
2023-03-15 11:36:54 +08:00
|
|
|
utarray_push_back(clause_state->ut_literal_ids, literal_id);
|
|
|
|
|
utarray_sort(clause_state->ut_literal_ids, compare_literal_id);
|
2023-01-30 21:59:35 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
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;
|
2023-03-15 11:36:54 +08:00
|
|
|
tmp = (struct maat_literal_id *)utarray_find(clause_state->ut_literal_ids,
|
2023-01-31 20:39:53 +08:00
|
|
|
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;
|
|
|
|
|
}
|
|
|
|
|
|
2023-03-15 11:36:54 +08:00
|
|
|
size_t remove_idx = utarray_eltidx(clause_state->ut_literal_ids, tmp);
|
|
|
|
|
utarray_erase(clause_state->ut_literal_ids, remove_idx, 1);
|
2023-01-30 21:59:35 +08:00
|
|
|
|
2023-03-15 11:36:54 +08:00
|
|
|
if (0 == utarray_len(clause_state->ut_literal_ids)) {
|
|
|
|
|
clause_state->in_use = 0;
|
2023-01-30 21:59:35 +08:00
|
|
|
compile->actual_clause_num--;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static const struct maat_clause *
|
2023-03-29 14:29:34 +08:00
|
|
|
maat_clause_hash_fetch_clause(struct compile_runtime *compile_rt,
|
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;
|
2023-04-05 21:09:19 +08:00
|
|
|
|
2023-03-29 14:29:34 +08:00
|
|
|
HASH_FIND(hh, compile_rt->clause_by_literals_hash, literal_ids,
|
|
|
|
|
n_literal_id * sizeof(struct maat_literal_id), clause);
|
2023-04-04 09:31:20 +08:00
|
|
|
if (NULL == clause) {
|
2023-01-30 21:59:35 +08:00
|
|
|
clause = ALLOC(struct maat_clause, 1);
|
2023-03-29 14:29:34 +08:00
|
|
|
clause->clause_id = maat_runtime_get_sequence(compile_rt->ref_maat_rt, "clause_id");
|
2023-01-30 21:59:35 +08:00
|
|
|
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));
|
|
|
|
|
|
2023-03-29 14:29:34 +08:00
|
|
|
HASH_ADD_KEYPTR(hh, compile_rt->clause_by_literals_hash, clause->literal_ids,
|
2023-02-03 17:28:14 +08:00
|
|
|
n_literal_id * sizeof(struct maat_literal_id), clause);
|
2023-01-30 21:59:35 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return clause;
|
|
|
|
|
}
|
|
|
|
|
|
2023-05-30 16:16:18 +08:00
|
|
|
struct bool_matcher *maat_compile_bool_matcher_new(struct compile_runtime *compile_rt,
|
|
|
|
|
size_t *compile_cnt)
|
2023-01-30 21:59:35 +08:00
|
|
|
{
|
2023-03-15 11:36:54 +08:00
|
|
|
if (NULL == compile_rt) {
|
2023-04-14 11:32:59 +08:00
|
|
|
return NULL;
|
|
|
|
|
}
|
2023-01-30 21:59:35 +08:00
|
|
|
|
|
|
|
|
size_t i = 0, j = 0;
|
|
|
|
|
int has_clause_num = 0;
|
2023-04-14 11:32:59 +08:00
|
|
|
const struct maat_clause *clause = NULL;
|
|
|
|
|
|
|
|
|
|
// STEP 1, update clause_id of each compile and literal
|
|
|
|
|
void **data_array = NULL;
|
|
|
|
|
size_t idx = 0;
|
|
|
|
|
struct maat_compile *iter_compile = NULL;
|
|
|
|
|
size_t rule_cnt = rcu_updating_hash_list(compile_rt->cfg_hash_tbl, &data_array);
|
|
|
|
|
*compile_cnt = rule_cnt;
|
|
|
|
|
|
|
|
|
|
for (idx = 0; idx < rule_cnt; idx++) {
|
|
|
|
|
iter_compile = (struct maat_compile *)data_array[idx];
|
|
|
|
|
has_clause_num = 0;
|
|
|
|
|
for (i = 0; i < MAX_ITEMS_PER_BOOL_EXPR; i++) {
|
|
|
|
|
struct maat_clause_state *clause_state = iter_compile->clause_states + i;
|
|
|
|
|
if (!clause_state->in_use) {
|
|
|
|
|
continue;
|
|
|
|
|
}
|
2023-01-30 21:59:35 +08:00
|
|
|
|
2023-04-14 11:32:59 +08:00
|
|
|
has_clause_num++;
|
|
|
|
|
struct maat_literal_id *literal_ids = (struct maat_literal_id *)utarray_eltptr(clause_state->ut_literal_ids, 0);
|
|
|
|
|
size_t n_literal_id = utarray_len(clause_state->ut_literal_ids);
|
|
|
|
|
clause = maat_clause_hash_fetch_clause(compile_rt, literal_ids, n_literal_id);
|
|
|
|
|
clause_state->clause_id = clause->clause_id;
|
|
|
|
|
}
|
|
|
|
|
assert(has_clause_num == iter_compile->actual_clause_num);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// STEP 2, serial compile clause states to a bool expression array
|
2023-01-30 21:59:35 +08:00
|
|
|
size_t expr_cnt = 0;
|
2023-04-14 11:32:59 +08:00
|
|
|
struct bool_expr *bool_expr_array = ALLOC(struct bool_expr, rule_cnt);
|
|
|
|
|
|
|
|
|
|
for (idx = 0; idx < rule_cnt; idx++) {
|
|
|
|
|
iter_compile = (struct maat_compile *)data_array[idx];
|
|
|
|
|
for (i = 0, j = 0; i < MAX_ITEMS_PER_BOOL_EXPR; i++) {
|
|
|
|
|
if (iter_compile->clause_states[i].in_use) {
|
|
|
|
|
if (iter_compile->clause_states[i].not_flag) {
|
|
|
|
|
iter_compile->not_clause_cnt++;
|
|
|
|
|
}
|
|
|
|
|
// TODO:mytest need to delete
|
|
|
|
|
#if 0
|
2023-01-30 21:59:35 +08:00
|
|
|
struct maat_literal_id *p = NULL;
|
2023-04-14 11:32:59 +08:00
|
|
|
for(p = (struct maat_literal_id *)utarray_front(iter_compile->clause_states[i].ut_literal_ids); p!=NULL;
|
2023-05-23 03:23:39 +00:00
|
|
|
p = (struct maat_literal_id *)utarray_next(iter_compile->clause_states[i].ut_literal_ids, p)) {
|
2023-04-14 11:32:59 +08:00
|
|
|
printf("<before bool_matcher_new> compile_rt:%p compile_id:%lld, clause_id:%llu, literal{%lld: %lld}\n",
|
|
|
|
|
compile_rt, iter_compile->compile_id, iter_compile->clause_states[i].clause_id, p->group_id, p->vtable_id);
|
2023-01-30 21:59:35 +08:00
|
|
|
}
|
2023-04-14 11:32:59 +08:00
|
|
|
#endif
|
|
|
|
|
bool_expr_array[expr_cnt].items[j].item_id = iter_compile->clause_states[i].clause_id;
|
|
|
|
|
bool_expr_array[expr_cnt].items[j].not_flag = iter_compile->clause_states[i].not_flag;
|
2023-01-30 21:59:35 +08:00
|
|
|
|
2023-04-14 11:32:59 +08:00
|
|
|
j++;
|
|
|
|
|
}
|
|
|
|
|
}
|
2023-02-16 16:45:06 +08:00
|
|
|
|
2023-04-14 11:32:59 +08:00
|
|
|
// some compile may have zero groups, e.g. default policy.
|
|
|
|
|
if (j == (size_t)iter_compile->declared_clause_num && j > 0) {
|
|
|
|
|
bool_expr_array[expr_cnt].expr_id = iter_compile->compile_id;
|
|
|
|
|
bool_expr_array[expr_cnt].user_tag = iter_compile;
|
|
|
|
|
bool_expr_array[expr_cnt].item_num = j;
|
|
|
|
|
expr_cnt++;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
FREE(data_array);
|
2023-01-30 21:59:35 +08:00
|
|
|
|
|
|
|
|
// STEP 3, build bool matcher
|
|
|
|
|
size_t mem_size = 0;
|
2023-04-14 11:32:59 +08:00
|
|
|
if (0 == expr_cnt) {
|
2023-05-23 03:23:39 +00:00
|
|
|
log_error(compile_rt->logger, MODULE_COMPILE,
|
|
|
|
|
"[%s:%d] No bool expression to build bool matcher.", __FUNCTION__, __LINE__);
|
2023-04-14 11:32:59 +08:00
|
|
|
FREE(bool_expr_array);
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
2023-01-30 21:59:35 +08:00
|
|
|
|
2023-04-14 11:32:59 +08:00
|
|
|
struct bool_matcher *bm = bool_matcher_new(bool_expr_array, expr_cnt, &mem_size);
|
|
|
|
|
if (bm != NULL) {
|
|
|
|
|
log_info(compile_rt->logger, MODULE_COMPILE,
|
2023-05-23 03:23:39 +00:00
|
|
|
"Build bool matcher of %zu expressions with %zu bytes memory.",
|
|
|
|
|
expr_cnt, mem_size);
|
2023-04-14 11:32:59 +08:00
|
|
|
} else {
|
|
|
|
|
log_error(compile_rt->logger, MODULE_COMPILE, "[%s:%d] Build bool matcher failed!",
|
2023-03-02 14:52:31 +08:00
|
|
|
__FUNCTION__, __LINE__);
|
2023-04-14 11:32:59 +08:00
|
|
|
}
|
2023-01-30 21:59:35 +08:00
|
|
|
|
|
|
|
|
FREE(bool_expr_array);
|
|
|
|
|
return bm;
|
|
|
|
|
}
|
|
|
|
|
|
2023-04-21 17:19:43 +08:00
|
|
|
static inline int compare_clause_id(const void *a, const void *b)
|
|
|
|
|
{
|
|
|
|
|
long long ret = *(const long long *)a - *(const long long *)b;
|
|
|
|
|
|
|
|
|
|
if (0 == ret) {
|
|
|
|
|
return 0;
|
|
|
|
|
} else if(ret < 0) {
|
|
|
|
|
return -1;
|
|
|
|
|
} else {
|
|
|
|
|
return 1;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2023-06-05 15:14:45 +08:00
|
|
|
static inline int compare_hit_group(const void *pa, const void *pb)
|
|
|
|
|
{
|
|
|
|
|
struct maat_hit_group *la=(struct maat_hit_group *)pa;
|
|
|
|
|
struct maat_hit_group *lb=(struct maat_hit_group *)pb;
|
|
|
|
|
|
|
|
|
|
long long ret = la->group_id - lb->group_id;
|
|
|
|
|
if (ret == 0) {
|
|
|
|
|
ret = la->vtable_id - lb->vtable_id;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
|
}
|
|
|
|
|
|
2023-04-21 17:19:43 +08:00
|
|
|
struct literal_clause *maat_compile_build_literal2clause_hash(struct compile_runtime *compile_rt)
|
|
|
|
|
{
|
|
|
|
|
if (NULL == compile_rt) {
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void **data_array = NULL;
|
|
|
|
|
struct maat_clause_state *clause_state = NULL;
|
|
|
|
|
struct maat_literal_id *tmp_literal_id = NULL;
|
|
|
|
|
struct literal_clause *l2c_value = NULL;
|
|
|
|
|
struct literal_clause *literal2clause_hash = NULL;
|
|
|
|
|
size_t compile_cnt = rcu_updating_hash_list(compile_rt->cfg_hash_tbl, &data_array);
|
|
|
|
|
|
|
|
|
|
for (size_t idx = 0; idx < compile_cnt; idx++) {
|
|
|
|
|
struct maat_compile *compile = (struct maat_compile *)data_array[idx];
|
|
|
|
|
for (size_t i = 0; i < MAX_ITEMS_PER_BOOL_EXPR; i++) {
|
|
|
|
|
clause_state = compile->clause_states + i;
|
|
|
|
|
if (!clause_state->in_use) {
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
for (size_t j = 0; j < utarray_len(clause_state->ut_literal_ids); j++) {
|
|
|
|
|
tmp_literal_id = (struct maat_literal_id *)utarray_eltptr(clause_state->ut_literal_ids, j);
|
|
|
|
|
HASH_FIND(hh, literal2clause_hash, tmp_literal_id, sizeof(struct maat_literal_id), l2c_value);
|
|
|
|
|
if (NULL == l2c_value) {
|
|
|
|
|
l2c_value = ALLOC(struct literal_clause, 1);
|
|
|
|
|
l2c_value->key = *tmp_literal_id;
|
|
|
|
|
utarray_new(l2c_value->clause_ids, &ut_clause_id_icd);
|
|
|
|
|
HASH_ADD(hh, literal2clause_hash, key, sizeof(l2c_value->key), l2c_value);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (utarray_find(l2c_value->clause_ids, &(clause_state->clause_id), compare_clause_id)) {
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
utarray_push_back(l2c_value->clause_ids, &(clause_state->clause_id));
|
|
|
|
|
utarray_sort(l2c_value->clause_ids, compare_clause_id);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
FREE(data_array);
|
|
|
|
|
return literal2clause_hash;
|
|
|
|
|
}
|
|
|
|
|
|
2023-03-15 11:36:54 +08:00
|
|
|
static int maat_compile_has_clause(struct maat_compile *compile, long long clause_id)
|
2023-03-01 13:12:22 +08:00
|
|
|
{
|
|
|
|
|
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;
|
|
|
|
|
|
2023-03-15 11:36:54 +08:00
|
|
|
long long new_hit_clause_id = 0;
|
2023-03-01 13:12:22 +08:00
|
|
|
for (size_t i = 0; i < utarray_len(compile_state->this_scan_hit_clauses); i++) {
|
2023-03-15 11:36:54 +08:00
|
|
|
new_hit_clause_id = *(long long*)utarray_eltptr(compile_state->this_scan_hit_clauses, i);
|
2023-03-01 13:12:22 +08:00
|
|
|
ret = maat_compile_has_clause(compile, new_hit_clause_id);
|
|
|
|
|
if (ret) {
|
|
|
|
|
r_in_c_cnt++;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return r_in_c_cnt;
|
|
|
|
|
}
|
|
|
|
|
|
2023-03-15 11:36:54 +08:00
|
|
|
size_t maat_compile_bool_matcher_match(struct compile_runtime *compile_rt, int is_last_scan,
|
2023-06-09 16:44:47 +08:00
|
|
|
struct maat_compile_state *compile_state, int thread_id,
|
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
|
|
|
{
|
2023-03-15 11:36:54 +08:00
|
|
|
size_t ud_result_cnt = 0;
|
2023-01-30 21:59:35 +08:00
|
|
|
struct maat_compile *compile = NULL;
|
2023-06-09 16:44:47 +08:00
|
|
|
struct bool_expr_match *expr_match = compile_rt->expr_match_buff +
|
|
|
|
|
(thread_id * MAX_SCANNER_HIT_COMPILE_NUM);
|
|
|
|
|
assert(thread_id >= 0);
|
2023-01-30 21:59:35 +08:00
|
|
|
|
2023-03-29 22:25:14 +08:00
|
|
|
if (0 == compile_state->compile_rt_version) {
|
|
|
|
|
compile_state->compile_rt_version = compile_rt->version;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (NULL == compile_rt->bm || 0 == utarray_len(compile_state->all_hit_clauses)
|
|
|
|
|
|| compile_state->compile_rt_version != compile_rt->version) {
|
2023-06-09 16:44:47 +08:00
|
|
|
compile_state->this_scan_hit_item_flag = 0;
|
2023-03-29 22:25:14 +08:00
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
|
2023-01-30 21:59:35 +08:00
|
|
|
//TODO:mytest need to delete
|
2023-02-09 22:13:15 +08:00
|
|
|
#if 0
|
2023-03-15 11:36:54 +08:00
|
|
|
unsigned long long *p = utarray_eltptr(compile_state->all_hit_clauses, 0);
|
2023-04-21 11:18:30 +08:00
|
|
|
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-03-15 11:36:54 +08:00
|
|
|
printf("before bool_matcher_match compile_rt:%p compile_state clause_id:%llu\n", compile_rt, *p);
|
2023-01-30 21:59:35 +08:00
|
|
|
}
|
|
|
|
|
#endif
|
|
|
|
|
|
2023-04-21 11:18:30 +08:00
|
|
|
int bool_match_ret = bool_matcher_match(compile_rt->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;
|
|
|
|
|
}
|
|
|
|
|
|
2023-03-01 13:12:22 +08:00
|
|
|
size_t n_new_hit_compile = compile_state_if_new_hit_compile(compile_state, compile);
|
2023-06-09 16:44:47 +08:00
|
|
|
int this_scan_hit_item_flag = compile_state->this_scan_hit_item_flag;
|
2023-02-03 17:28:14 +08:00
|
|
|
if ((compile->not_clause_cnt > 0) && (LAST_SCAN_UNSET == is_last_scan)) {
|
2023-06-09 16:44:47 +08:00
|
|
|
compile_state->not_clause_hit_flag = 1;
|
2023-03-06 10:45:36 +08:00
|
|
|
} else if (compile->user_data) {
|
2023-06-09 16:44:47 +08:00
|
|
|
if (n_new_hit_compile > 0 || 0 == this_scan_hit_item_flag) {
|
2023-03-01 13:12:22 +08:00
|
|
|
/* compile hit because of new item or
|
2023-03-06 10:45:36 +08:00
|
|
|
hit a compile that refer a NOT-logic group in previous scan */
|
2023-03-01 13:12:22 +08:00
|
|
|
user_data_array[ud_result_cnt] = compile->user_data;
|
|
|
|
|
ud_result_cnt++;
|
|
|
|
|
}
|
2023-03-06 10:45:36 +08:00
|
|
|
}
|
2023-01-30 21:59:35 +08:00
|
|
|
}
|
|
|
|
|
|
2023-06-09 16:44:47 +08:00
|
|
|
compile_state->this_scan_hit_item_flag = 0;
|
2023-03-15 11:36:54 +08:00
|
|
|
|
2023-01-30 21:59:35 +08:00
|
|
|
return ud_result_cnt;
|
|
|
|
|
}
|
|
|
|
|
|
2023-05-11 11:21:46 +08:00
|
|
|
#define COMPILE_RULE_MAGIC 0x1a2b3c4d
|
2023-04-14 11:32:59 +08:00
|
|
|
struct compile_rule *compile_rule_new(struct compile_item *compile_item,
|
|
|
|
|
struct compile_schema *schema,
|
|
|
|
|
const char *table_name,
|
|
|
|
|
const char *table_line)
|
|
|
|
|
{
|
|
|
|
|
struct compile_rule *compile_rule = ALLOC(struct compile_rule, 1);
|
|
|
|
|
|
|
|
|
|
compile_rule->magic_num = COMPILE_RULE_MAGIC;
|
|
|
|
|
compile_rule->declared_clause_num = compile_item->declared_clause_num;
|
|
|
|
|
compile_rule->ref_schema = schema;
|
|
|
|
|
compile_rule->ex_data = ALLOC(void *, 1);
|
|
|
|
|
memcpy(compile_rule->table_name, table_name, sizeof(compile_rule->table_name));
|
|
|
|
|
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);
|
|
|
|
|
|
|
|
|
|
if (1 == schema->set_flag)
|
|
|
|
|
{
|
|
|
|
|
*(compile_rule->ex_data) = rule_ex_data_new(table_name, schema->table_id,
|
|
|
|
|
compile_rule->table_line,
|
|
|
|
|
&(schema->ex_schema));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
compile_rule->compile_id = compile_item->compile_id;
|
|
|
|
|
|
|
|
|
|
return compile_rule;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
struct compile_rule *compile_rule_clone(struct compile_rule *rule)
|
|
|
|
|
{
|
|
|
|
|
struct compile_rule *new_rule = ALLOC(struct compile_rule, 1);
|
|
|
|
|
|
|
|
|
|
new_rule->magic_num = rule->magic_num;
|
|
|
|
|
new_rule->declared_clause_num = rule->declared_clause_num;
|
|
|
|
|
new_rule->ref_schema = rule->ref_schema;
|
|
|
|
|
new_rule->ex_data = ALLOC(void *, 1);
|
|
|
|
|
memcpy(new_rule->table_name, rule->table_name, sizeof(new_rule->table_name));
|
|
|
|
|
new_rule->table_line_len = rule->table_line_len;
|
|
|
|
|
new_rule->table_line = ALLOC(char, new_rule->table_line_len);
|
|
|
|
|
memcpy(new_rule->table_line, rule->table_line, new_rule->table_line_len);
|
|
|
|
|
|
2023-06-08 07:32:41 +00:00
|
|
|
if (1 == rule->ref_schema->set_flag) {
|
2023-04-14 11:32:59 +08:00
|
|
|
*(new_rule->ex_data) = rule_ex_data_new(rule->table_name, rule->ref_schema->table_id,
|
|
|
|
|
rule->table_line, &(rule->ref_schema->ex_schema));
|
|
|
|
|
}
|
|
|
|
|
new_rule->compile_id = rule->compile_id;
|
|
|
|
|
|
|
|
|
|
return new_rule;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void compile_rule_free(struct compile_rule *compile_rule)
|
|
|
|
|
{
|
|
|
|
|
struct compile_schema *schema = compile_rule->ref_schema;
|
|
|
|
|
assert(compile_rule->magic_num == COMPILE_RULE_MAGIC);
|
|
|
|
|
|
2023-05-23 03:23:39 +00:00
|
|
|
if (1 == schema->set_flag) {
|
2023-04-14 11:32:59 +08:00
|
|
|
rule_ex_data_free(schema->table_id, compile_rule->ex_data, &(schema->ex_schema));
|
|
|
|
|
*compile_rule->ex_data = NULL;
|
|
|
|
|
}
|
2023-05-23 03:23:39 +00:00
|
|
|
|
|
|
|
|
if (compile_rule->ex_data != NULL) {
|
|
|
|
|
FREE(compile_rule->ex_data);
|
|
|
|
|
}
|
|
|
|
|
|
2023-04-14 11:32:59 +08:00
|
|
|
compile_rule->declared_clause_num = -1;
|
2023-05-23 03:23:39 +00:00
|
|
|
|
|
|
|
|
if (compile_rule->table_line != NULL) {
|
|
|
|
|
FREE(compile_rule->table_line);
|
|
|
|
|
}
|
|
|
|
|
|
2023-04-14 11:32:59 +08:00
|
|
|
FREE(compile_rule);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
struct maat_compile *maat_compile_clone(struct maat_compile *compile, int deep_copy)
|
|
|
|
|
{
|
|
|
|
|
struct maat_compile *new_compile = ALLOC(struct maat_compile, 1);
|
|
|
|
|
|
|
|
|
|
new_compile->magic = compile->magic;
|
|
|
|
|
new_compile->compile_id = compile->compile_id;
|
|
|
|
|
new_compile->actual_clause_num = compile->actual_clause_num;
|
|
|
|
|
new_compile->declared_clause_num = compile->declared_clause_num;
|
|
|
|
|
memcpy(new_compile->table_name, compile->table_name, sizeof(new_compile->table_name));
|
|
|
|
|
new_compile->not_clause_cnt = compile->not_clause_cnt;
|
|
|
|
|
new_compile->user_data_free = compile->user_data_free;
|
2023-05-30 16:16:18 +08:00
|
|
|
if (1 == deep_copy && compile->user_data != NULL) {
|
2023-04-14 11:32:59 +08:00
|
|
|
new_compile->user_data = compile_rule_clone((struct compile_rule *)compile->user_data);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
struct maat_literal_id *literal_id = NULL;
|
2023-05-30 16:16:18 +08:00
|
|
|
for (int i = 0; i < MAX_ITEMS_PER_BOOL_EXPR; i++) {
|
2023-04-14 11:32:59 +08:00
|
|
|
new_compile->clause_states[i].clause_id = compile->clause_states[i].clause_id;
|
|
|
|
|
new_compile->clause_states[i].in_use = compile->clause_states[i].in_use;
|
|
|
|
|
new_compile->clause_states[i].not_flag = compile->clause_states[i].not_flag;
|
|
|
|
|
utarray_new(new_compile->clause_states[i].ut_literal_ids, &ut_literal_id_icd);
|
2023-05-30 16:16:18 +08:00
|
|
|
for (int j = 0; j < utarray_len(compile->clause_states[i].ut_literal_ids); j++) {
|
2023-04-14 11:32:59 +08:00
|
|
|
literal_id = (struct maat_literal_id *)utarray_eltptr(compile->clause_states[i].ut_literal_ids, j);
|
|
|
|
|
utarray_push_back(new_compile->clause_states[i].ut_literal_ids, literal_id);
|
|
|
|
|
}
|
|
|
|
|
|
2023-05-30 16:16:18 +08:00
|
|
|
for (int k = 0; k < utarray_len(new_compile->clause_states[i].ut_literal_ids); k++) {
|
2023-04-14 11:32:59 +08:00
|
|
|
literal_id = (struct maat_literal_id *)utarray_eltptr(new_compile->clause_states[i].ut_literal_ids, k);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return new_compile;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
int maat_add_group_to_compile(struct rcu_hash_table *hash_tbl, struct group2compile_item *g2c_item,
|
2023-02-22 15:08:52 +08:00
|
|
|
struct log_handle *logger)
|
2023-01-30 21:59:35 +08:00
|
|
|
{
|
|
|
|
|
int ret = -1;
|
2023-04-14 11:32:59 +08:00
|
|
|
long long compile_id = g2c_item->compile_id;
|
|
|
|
|
struct maat_compile *compile = NULL;
|
|
|
|
|
struct maat_literal_id literal_id = {g2c_item->group_id, g2c_item->vtable_id};
|
|
|
|
|
|
|
|
|
|
int updating_flag = rcu_hash_is_updating(hash_tbl);
|
|
|
|
|
if (1 == updating_flag) {
|
|
|
|
|
compile = rcu_updating_hash_find(hash_tbl, (char *)&compile_id, sizeof(long long));
|
|
|
|
|
if (compile != NULL) {
|
|
|
|
|
/* compile found in updating hash(added by compile runtime), it can be modified directly */
|
2023-05-30 16:16:18 +08:00
|
|
|
ret = maat_compile_clause_add_literal(compile, &literal_id, g2c_item->clause_index,
|
|
|
|
|
g2c_item->not_flag);
|
2023-04-14 11:32:59 +08:00
|
|
|
if (ret < 0) {
|
|
|
|
|
log_error(logger, MODULE_COMPILE,
|
2023-06-07 11:53:08 +08:00
|
|
|
"[%s:%d] add literal_id{group_id:%lld, vtable_id:%d} to clause_index: %d"
|
2023-05-30 16:16:18 +08:00
|
|
|
" of compile %d failed", __FUNCTION__, __LINE__, g2c_item->group_id,
|
|
|
|
|
g2c_item->vtable_id, g2c_item->clause_index, compile_id);
|
2023-04-14 11:32:59 +08:00
|
|
|
}
|
|
|
|
|
} else {
|
|
|
|
|
/* compile neither in effective hash nor in updating hash, so new one */
|
|
|
|
|
compile = maat_compile_new(compile_id);
|
|
|
|
|
assert(compile != NULL);
|
2023-05-30 16:16:18 +08:00
|
|
|
ret = maat_compile_clause_add_literal(compile, &literal_id, g2c_item->clause_index,
|
|
|
|
|
g2c_item->not_flag);
|
2023-04-14 11:32:59 +08:00
|
|
|
if (ret < 0) {
|
|
|
|
|
log_error(logger, MODULE_COMPILE,
|
2023-06-07 11:53:08 +08:00
|
|
|
"[%s:%d] add literal_id{group_id:%lld, vtable_id:%d} to clause_index: %d"
|
2023-05-30 16:16:18 +08:00
|
|
|
" of compile %d failed", __FUNCTION__, __LINE__, g2c_item->group_id,
|
|
|
|
|
g2c_item->vtable_id, g2c_item->clause_index, compile_id);
|
2023-04-14 11:32:59 +08:00
|
|
|
}
|
|
|
|
|
rcu_hash_add(hash_tbl, (char *)&compile_id, sizeof(long long), compile);
|
|
|
|
|
}
|
|
|
|
|
} else {
|
|
|
|
|
compile = rcu_hash_find(hash_tbl, (char *)&compile_id, sizeof(long long));
|
|
|
|
|
if (compile != NULL) {
|
|
|
|
|
/*******************************************************************
|
|
|
|
|
compile found in effective hash(added by compile runtime), which means
|
|
|
|
|
|
|
|
|
|
1. rcu_hash_add(hash_tbl, compile) ==> finished
|
|
|
|
|
2. rcu_hash_commit(hash_tbl) ==> finished
|
|
|
|
|
|
|
|
|
|
can only be deleted but not modified
|
|
|
|
|
before delete it, we need to make a copy for further use
|
|
|
|
|
*********************************************************************/
|
|
|
|
|
struct maat_compile *copy_compile = maat_compile_clone(compile, 1);
|
|
|
|
|
assert(copy_compile != NULL);
|
|
|
|
|
|
|
|
|
|
/* delete compile from rcu hash */
|
|
|
|
|
rcu_hash_del(hash_tbl, (char *)&compile_id, sizeof(long long));
|
|
|
|
|
|
|
|
|
|
ret = maat_compile_clause_add_literal(copy_compile, &literal_id, g2c_item->clause_index,
|
|
|
|
|
g2c_item->not_flag);
|
|
|
|
|
if (ret < 0) {
|
|
|
|
|
log_error(logger, MODULE_COMPILE,
|
2023-06-07 11:53:08 +08:00
|
|
|
"[%s:%d] add literal_id{group_id:%lld, vtable_id:%d} to clause_index: %d"
|
2023-05-30 16:16:18 +08:00
|
|
|
" of compile %d failed", __FUNCTION__, __LINE__, g2c_item->group_id,
|
|
|
|
|
g2c_item->vtable_id, g2c_item->clause_index, compile_id);
|
2023-04-14 11:32:59 +08:00
|
|
|
}
|
2023-03-06 10:45:36 +08:00
|
|
|
|
2023-04-14 11:32:59 +08:00
|
|
|
rcu_hash_add(hash_tbl, (char *)&compile_id, sizeof(long long), copy_compile);
|
|
|
|
|
} else {
|
|
|
|
|
compile = maat_compile_new(compile_id);
|
|
|
|
|
assert(compile != NULL);
|
|
|
|
|
ret = maat_compile_clause_add_literal(compile, &literal_id, g2c_item->clause_index,
|
|
|
|
|
g2c_item->not_flag);
|
|
|
|
|
if (ret < 0) {
|
|
|
|
|
log_error(logger, MODULE_COMPILE,
|
2023-06-07 11:53:08 +08:00
|
|
|
"[%s:%d] add literal_id{group_id:%lld, vtable_id:%d} to clause_index: %d"
|
2023-05-30 16:16:18 +08:00
|
|
|
" of compile %d failed", __FUNCTION__, __LINE__, g2c_item->group_id,
|
|
|
|
|
g2c_item->vtable_id, g2c_item->clause_index, compile_id);
|
2023-04-14 11:32:59 +08:00
|
|
|
}
|
|
|
|
|
rcu_hash_add(hash_tbl, (char *)&compile_id, sizeof(long long), compile);
|
|
|
|
|
}
|
|
|
|
|
}
|
2023-01-30 21:59:35 +08:00
|
|
|
|
2023-04-14 11:32:59 +08:00
|
|
|
return ret;
|
2023-01-30 21:59:35 +08:00
|
|
|
}
|
|
|
|
|
|
2023-04-14 11:32:59 +08:00
|
|
|
int maat_remove_group_from_compile(struct rcu_hash_table *hash_tbl,
|
2023-01-31 20:39:53 +08:00
|
|
|
struct group2compile_item *g2c_item,
|
2023-04-14 11:32:59 +08:00
|
|
|
struct log_handle *logger)
|
2023-01-30 21:59:35 +08:00
|
|
|
{
|
2023-04-14 11:32:59 +08:00
|
|
|
int ret = -1;
|
|
|
|
|
long long compile_id = g2c_item->compile_id;
|
|
|
|
|
struct maat_compile *compile = NULL;
|
|
|
|
|
struct maat_literal_id literal_id = {g2c_item->group_id, g2c_item->vtable_id};
|
2023-01-30 21:59:35 +08:00
|
|
|
|
2023-04-14 11:32:59 +08:00
|
|
|
int updating_flag = rcu_hash_is_updating(hash_tbl);
|
|
|
|
|
if (1 == updating_flag) {
|
2023-05-30 16:16:18 +08:00
|
|
|
compile = rcu_updating_hash_find(hash_tbl, (char *)&compile_id,
|
|
|
|
|
sizeof(long long));
|
2023-04-14 11:32:59 +08:00
|
|
|
if (NULL == compile) {
|
|
|
|
|
log_error(logger, MODULE_COMPILE,
|
2023-06-07 11:53:08 +08:00
|
|
|
"[%s:%d] Remove group_id:%lld from compile_id:%lld failed, compile"
|
2023-05-30 16:16:18 +08:00
|
|
|
" is not exisited.", __FUNCTION__, __LINE__, g2c_item->group_id,
|
|
|
|
|
compile_id);
|
2023-04-14 11:32:59 +08:00
|
|
|
return -1;
|
|
|
|
|
} else {
|
|
|
|
|
/* compile found in updating hash, it can be modified directly */
|
2023-05-30 16:16:18 +08:00
|
|
|
ret = maat_compile_clause_remove_literal(compile, &literal_id,
|
|
|
|
|
g2c_item->clause_index);
|
2023-04-14 11:32:59 +08:00
|
|
|
if (ret < 0) {
|
|
|
|
|
log_error(logger, MODULE_COMPILE,
|
2023-06-07 11:53:08 +08:00
|
|
|
"[%s:%d] Remove group_id:%lld vtable_id %d from clause %d of "
|
|
|
|
|
"compile_id:%lld failed, literal is not in compile.", __FUNCTION__,
|
2023-05-30 16:16:18 +08:00
|
|
|
__LINE__, g2c_item->group_id, g2c_item->vtable_id,
|
|
|
|
|
g2c_item->clause_index, compile_id);
|
2023-04-14 11:32:59 +08:00
|
|
|
}
|
2023-01-30 21:59:35 +08:00
|
|
|
|
2023-04-14 11:32:59 +08:00
|
|
|
if (0 == compile->actual_clause_num && NULL == compile->user_data) {
|
|
|
|
|
rcu_hash_del(hash_tbl, (char *)&compile_id, sizeof(long long));
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
} else {
|
|
|
|
|
//find in effetive hash
|
|
|
|
|
compile = rcu_hash_find(hash_tbl, (char *)&compile_id, sizeof(long long));
|
|
|
|
|
if (compile != NULL) {
|
|
|
|
|
/*******************************************************************
|
|
|
|
|
compile found in effective hash, which means
|
|
|
|
|
|
|
|
|
|
1. rcu_hash_add(hash_tbl, compile) ==> finished
|
|
|
|
|
2. rcu_hash_commit(hash_tbl) ==> finished
|
|
|
|
|
|
|
|
|
|
can only be deleted but not modified
|
|
|
|
|
before delete it, we need to make a copy for further use
|
|
|
|
|
*********************************************************************/
|
|
|
|
|
struct maat_compile *copy_compile = maat_compile_clone(compile, 1);
|
|
|
|
|
assert(copy_compile != NULL);
|
|
|
|
|
|
|
|
|
|
/* delete compile from rcu hash */
|
|
|
|
|
rcu_hash_del(hash_tbl, (char *)&compile_id, sizeof(long long));
|
|
|
|
|
|
2023-05-30 16:16:18 +08:00
|
|
|
ret = maat_compile_clause_remove_literal(copy_compile, &literal_id,
|
|
|
|
|
g2c_item->clause_index);
|
2023-04-14 11:32:59 +08:00
|
|
|
if (ret < 0) {
|
|
|
|
|
log_error(logger, MODULE_COMPILE,
|
2023-06-07 11:53:08 +08:00
|
|
|
"[%s:%d] Remove group_id:%lld vtable_id %d from clause %d of compile_id:"
|
|
|
|
|
"%lld failed, literal is not in compile.", __FUNCTION__, __LINE__,
|
2023-05-30 16:16:18 +08:00
|
|
|
g2c_item->group_id, g2c_item->vtable_id, g2c_item->clause_index,
|
2023-04-14 11:32:59 +08:00
|
|
|
compile_id);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (0 == copy_compile->actual_clause_num && NULL == copy_compile->user_data) {
|
|
|
|
|
maat_compile_free(copy_compile);
|
|
|
|
|
} else {
|
|
|
|
|
rcu_hash_add(hash_tbl, (char *)&compile_id, sizeof(long long), copy_compile);
|
|
|
|
|
}
|
|
|
|
|
} else {
|
|
|
|
|
log_error(logger, MODULE_COMPILE,
|
2023-06-07 11:53:08 +08:00
|
|
|
"[%s:%d] Remove group_id:%lld from compile_id:%lld failed, compile is not exisited.",
|
2023-04-14 11:32:59 +08:00
|
|
|
__FUNCTION__, __LINE__, g2c_item->group_id, compile_id);
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return ret;
|
2023-01-30 21:59:35 +08:00
|
|
|
}
|
|
|
|
|
|
2023-06-09 16:44:47 +08:00
|
|
|
struct maat_compile_state *maat_compile_state_new(void)
|
2023-01-30 21:59:35 +08:00
|
|
|
{
|
|
|
|
|
struct maat_compile_state *compile_state = ALLOC(struct maat_compile_state, 1);
|
|
|
|
|
|
|
|
|
|
utarray_new(compile_state->internal_hit_paths, &ut_hit_path_icd);
|
2023-02-07 11:25:31 +08:00
|
|
|
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;
|
|
|
|
|
}
|
|
|
|
|
|
2023-03-23 11:57:17 +08:00
|
|
|
void maat_compile_state_reset(struct maat_compile_state *compile_state)
|
|
|
|
|
{
|
|
|
|
|
if (NULL == compile_state) {
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
compile_state->Nth_scan = 0;
|
2023-03-29 22:25:14 +08:00
|
|
|
compile_state->compile_rt_version = 0;
|
2023-06-09 16:44:47 +08:00
|
|
|
compile_state->this_scan_hit_item_flag = 0;
|
|
|
|
|
compile_state->not_clause_hit_flag = 0;
|
2023-03-23 11:57:17 +08:00
|
|
|
|
|
|
|
|
utarray_clear(compile_state->internal_hit_paths);
|
|
|
|
|
utarray_clear(compile_state->all_hit_clauses);
|
|
|
|
|
utarray_clear(compile_state->this_scan_hit_clauses);
|
|
|
|
|
}
|
|
|
|
|
|
2023-06-09 17:30:20 +08:00
|
|
|
void maat_compile_state_free(struct maat_compile_state *compile_state,
|
|
|
|
|
struct maat *maat_instance, int thread_id)
|
2023-01-30 21:59:35 +08:00
|
|
|
{
|
2023-03-23 11:57:17 +08:00
|
|
|
if (NULL == compile_state) {
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
2023-06-09 17:30:20 +08:00
|
|
|
long long free_bytes = 0;
|
2023-05-08 18:51:53 +08:00
|
|
|
if (compile_state->internal_hit_paths != NULL) {
|
2023-06-09 17:30:20 +08:00
|
|
|
free_bytes += utarray_len(compile_state->internal_hit_paths) * sizeof(struct maat_internal_hit_path);
|
2023-05-08 18:51:53 +08:00
|
|
|
utarray_free(compile_state->internal_hit_paths);
|
|
|
|
|
compile_state->internal_hit_paths = NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (compile_state->all_hit_clauses != NULL) {
|
2023-06-09 17:30:20 +08:00
|
|
|
free_bytes += utarray_len(compile_state->all_hit_clauses) * sizeof(long long);
|
2023-05-08 18:51:53 +08:00
|
|
|
utarray_free(compile_state->all_hit_clauses);
|
|
|
|
|
compile_state->all_hit_clauses = NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (compile_state->this_scan_hit_clauses != NULL) {
|
2023-06-09 17:30:20 +08:00
|
|
|
free_bytes += utarray_len(compile_state->this_scan_hit_clauses) * sizeof(long long);
|
2023-05-08 18:51:53 +08:00
|
|
|
utarray_free(compile_state->this_scan_hit_clauses);
|
|
|
|
|
compile_state->this_scan_hit_clauses = NULL;
|
|
|
|
|
}
|
2023-06-09 17:30:20 +08:00
|
|
|
|
2023-05-23 03:23:39 +00:00
|
|
|
FREE(compile_state);
|
2023-06-09 17:30:20 +08:00
|
|
|
|
|
|
|
|
free_bytes += sizeof(struct maat_compile_state);
|
|
|
|
|
alignment_int64_array_add(maat_instance->stat->maat_state_free_bytes, thread_id, free_bytes);
|
2023-01-30 21:59:35 +08:00
|
|
|
}
|
|
|
|
|
|
2023-05-30 16:16:18 +08:00
|
|
|
static int maat_compile_hit_path_add(UT_array *hit_paths, long long item_id,
|
|
|
|
|
long long group_id, 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-05-17 14:34:34 +08:00
|
|
|
static int maat_compile_has_literal(struct maat_compile *compile,
|
2023-02-03 17:28:14 +08:00
|
|
|
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-03-15 11:36:54 +08:00
|
|
|
tmp = (struct maat_literal_id*)utarray_find(clause_state->ut_literal_ids,
|
2023-02-03 17:28:14 +08:00
|
|
|
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;
|
|
|
|
|
}
|
|
|
|
|
|
2023-02-20 10:57:40 +08:00
|
|
|
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-06-09 16:44:47 +08:00
|
|
|
size_t compile_runtime_get_hit_paths(struct compile_runtime *compile_rt, int thread_id,
|
2023-01-30 21:59:35 +08:00
|
|
|
struct maat_compile_state *compile_state,
|
2023-03-06 16:45:34 +08:00
|
|
|
struct maat_hit_path *hit_path_array,
|
2023-03-27 15:52:47 +08:00
|
|
|
size_t array_size, size_t n_internal_hit_path)
|
2023-01-30 21:59:35 +08:00
|
|
|
{
|
2023-03-06 16:45:34 +08:00
|
|
|
/* assign hit_path_array[].compile_id */
|
2023-02-03 17:28:14 +08:00
|
|
|
size_t new_hit_path_cnt = 0;
|
2023-01-30 21:59:35 +08:00
|
|
|
struct maat_compile *compile = NULL;
|
|
|
|
|
struct maat_literal_id literal_id = {0, 0};
|
2023-05-23 03:23:39 +00:00
|
|
|
struct bool_expr_match *expr_match = compile_rt->expr_match_buff +
|
2023-06-09 16:44:47 +08:00
|
|
|
(thread_id * MAX_SCANNER_HIT_COMPILE_NUM);
|
|
|
|
|
assert(thread_id >= 0);
|
2023-03-06 16:45:34 +08:00
|
|
|
|
2023-03-29 22:25:14 +08:00
|
|
|
if (compile_state->compile_rt_version != compile_rt->version) {
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
|
2023-02-03 17:28:14 +08:00
|
|
|
int bool_match_ret = bool_matcher_match(compile_rt->bm,
|
2023-02-07 11:25:31 +08:00
|
|
|
(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);
|
2023-03-27 15:52:47 +08:00
|
|
|
if (0 == compile->actual_clause_num || NULL == compile->user_data) {
|
2023-01-30 21:59:35 +08:00
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
|
2023-03-27 15:52:47 +08:00
|
|
|
for (size_t j = 0; j < n_internal_hit_path && (n_internal_hit_path + new_hit_path_cnt) < array_size; j++) {
|
2023-03-06 16:45:34 +08:00
|
|
|
if (hit_path_array[j].top_group_id < 0) {
|
2023-04-24 02:14:26 +00:00
|
|
|
literal_id.group_id = hit_path_array[j].sub_group_id;
|
|
|
|
|
} else {
|
|
|
|
|
literal_id.group_id = hit_path_array[j].top_group_id;
|
|
|
|
|
}
|
2023-01-30 21:59:35 +08:00
|
|
|
|
2023-03-06 16:45:34 +08:00
|
|
|
literal_id.vtable_id = hit_path_array[j].vtable_id;
|
2023-01-30 21:59:35 +08:00
|
|
|
if (maat_compile_has_literal(compile, &literal_id)) {
|
2023-04-24 02:14:26 +00:00
|
|
|
if (hit_path_array[j].top_group_id < 0) {
|
|
|
|
|
hit_path_array[j].top_group_id = hit_path_array[j].sub_group_id;
|
|
|
|
|
}
|
|
|
|
|
|
2023-03-06 16:45:34 +08:00
|
|
|
if (hit_path_array[j].compile_id < 0) {
|
|
|
|
|
hit_path_array[j].compile_id = compile->compile_id;
|
2023-01-30 21:59:35 +08:00
|
|
|
} else {
|
2023-02-03 17:28:14 +08:00
|
|
|
// means same literal_id hit more than one compile_id
|
2023-05-17 14:34:34 +08:00
|
|
|
struct maat_hit_path tmp_path = hit_path_array[j];
|
2023-01-30 21:59:35 +08:00
|
|
|
tmp_path.compile_id = compile->compile_id;
|
2023-03-27 15:52:47 +08:00
|
|
|
if(maat_compile_is_hit_path_existed(hit_path_array, n_internal_hit_path + new_hit_path_cnt, &tmp_path)) {
|
|
|
|
|
hit_path_array[n_internal_hit_path + new_hit_path_cnt] = tmp_path;
|
2023-02-03 17:28:14 +08:00
|
|
|
new_hit_path_cnt++;
|
2023-01-30 21:59:35 +08:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
2023-03-15 11:36:54 +08:00
|
|
|
|
2023-03-27 15:52:47 +08:00
|
|
|
return (n_internal_hit_path + 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-06-09 16:44:47 +08:00
|
|
|
assert(compile_state->this_scan_hit_item_flag == 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
|
|
|
|
2023-06-09 16:44:47 +08:00
|
|
|
compile_state->this_scan_hit_item_flag = 1;
|
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-02-03 17:28:14 +08:00
|
|
|
struct maat_literal_id literal_id = {group_id, vtable_id};
|
2023-04-21 17:19:43 +08:00
|
|
|
struct literal_clause *l2c_val = NULL;
|
2023-03-15 11:36:54 +08:00
|
|
|
long long *clause_id = 0;
|
2023-02-16 16:45:06 +08:00
|
|
|
struct compile_runtime *compile_rt = (struct compile_runtime *)compile_runtime;
|
|
|
|
|
|
2023-04-21 17:19:43 +08:00
|
|
|
HASH_FIND(hh, compile_rt->literal2clause_hash, &literal_id, sizeof(literal_id), l2c_val);
|
|
|
|
|
if (!l2c_val) {
|
|
|
|
|
return;
|
|
|
|
|
}
|
2023-02-03 17:28:14 +08:00
|
|
|
|
2023-04-21 17:19:43 +08:00
|
|
|
size_t i = 0;
|
|
|
|
|
size_t new_clause_idx = utarray_len(compile_state->this_scan_hit_clauses);
|
|
|
|
|
for (i = 0; i < utarray_len(l2c_val->clause_ids); i++) {
|
|
|
|
|
clause_id = (long long *)utarray_eltptr(l2c_val->clause_ids, i);
|
|
|
|
|
if (utarray_find(compile_state->all_hit_clauses, clause_id, compare_clause_id)) {
|
|
|
|
|
continue;
|
2023-02-03 17:28:14 +08:00
|
|
|
}
|
2023-04-21 17:19:43 +08:00
|
|
|
utarray_push_back(compile_state->this_scan_hit_clauses, clause_id);
|
|
|
|
|
}
|
2023-04-14 11:32:59 +08:00
|
|
|
|
2023-04-21 17:19:43 +08:00
|
|
|
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 = (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
|
|
|
}
|
|
|
|
|
|
|
|
|
|
int maat_compile_state_has_NOT_clause(struct maat_compile_state *compile_state)
|
|
|
|
|
{
|
2023-06-09 16:44:47 +08:00
|
|
|
return compile_state->not_clause_hit_flag;
|
2023-01-30 21:59:35 +08:00
|
|
|
}
|
|
|
|
|
|
2023-03-01 13:12:22 +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 ||
|
2023-04-05 21:09:19 +08:00
|
|
|
(0 == compile_schema->set_flag)) {
|
2023-03-01 13:12:22 +08:00
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
compile_runtime_user_data_iterate(compile_rt, rule_ex_data_new_cb,
|
2023-04-05 21:09:19 +08:00
|
|
|
&(compile_schema->ex_schema),
|
2023-03-01 13:12:22 +08:00
|
|
|
compile_schema->table_id);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void *compile_runtime_get_ex_data(struct compile_runtime *compile_rt,
|
|
|
|
|
struct compile_schema *compile_schema,
|
|
|
|
|
long long compile_id)
|
|
|
|
|
{
|
2023-04-05 21:09:19 +08:00
|
|
|
if (NULL == compile_rt || NULL == compile_schema || compile_id < 0 ||
|
|
|
|
|
(0 == compile_schema->set_flag)) {
|
2023-03-01 13:12:22 +08:00
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
struct compile_rule *compile_rule = NULL;
|
2023-04-14 11:32:59 +08:00
|
|
|
compile_rule = (struct compile_rule *)compile_runtime_get_user_data(compile_rt, compile_id);
|
2023-03-01 13:12:22 +08:00
|
|
|
if (NULL == compile_rule) {
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void *ex_data = NULL;
|
2023-04-05 21:09:19 +08:00
|
|
|
compile_schema->ex_schema.dup_func(compile_schema->table_id, &ex_data,
|
|
|
|
|
compile_rule->ex_data,
|
|
|
|
|
compile_schema->ex_schema.argl,
|
|
|
|
|
compile_schema->ex_schema.argp);
|
2023-03-01 13:12:22 +08:00
|
|
|
return ex_data;
|
|
|
|
|
}
|
|
|
|
|
|
2023-04-20 15:34:56 +08:00
|
|
|
int compile_runtime_add_compile(struct compile_runtime *compile_rt, struct compile_schema *schema,
|
2023-04-14 11:32:59 +08:00
|
|
|
long long compile_id, const char *table_name, const char *line)
|
|
|
|
|
{
|
|
|
|
|
struct compile_item *compile_item = NULL;
|
|
|
|
|
struct maat_compile *compile = NULL;
|
|
|
|
|
|
|
|
|
|
compile_item = compile_item_new(line, schema, table_name, compile_rt->logger);
|
|
|
|
|
if (NULL == compile_item) {
|
2023-04-20 15:34:56 +08:00
|
|
|
return -1;
|
2023-04-14 11:32:59 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
struct compile_rule *compile_rule = compile_rule_new(compile_item, schema, table_name, line);
|
|
|
|
|
compile_item_free(compile_item);
|
|
|
|
|
|
|
|
|
|
int updating_flag = rcu_hash_is_updating(compile_rt->cfg_hash_tbl);
|
|
|
|
|
if (1 == updating_flag) {
|
|
|
|
|
compile = rcu_updating_hash_find(compile_rt->cfg_hash_tbl, (char *)&compile_id,
|
|
|
|
|
sizeof(long long));
|
|
|
|
|
if (compile != NULL) {
|
|
|
|
|
/****************************************************************
|
|
|
|
|
compile found in updating hash(added by group2compile runtime), which means
|
|
|
|
|
|
|
|
|
|
1. rcu_hash_add(htable, compile) ==> finished
|
|
|
|
|
2. rcu_hash_commit(htable) ==> undo
|
|
|
|
|
because it's in updating hash, we can modify it directly
|
|
|
|
|
******************************************************************/
|
|
|
|
|
|
|
|
|
|
/* compile has group2compile_table info, so set compile_table info */
|
|
|
|
|
maat_compile_set(compile, table_name, compile_rule->declared_clause_num,
|
|
|
|
|
compile_rule, (void (*)(void *))compile_rule_free);
|
|
|
|
|
} else {
|
|
|
|
|
// compile neither in effective hash nor in updating hash
|
|
|
|
|
compile = maat_compile_new(compile_rule->compile_id);
|
|
|
|
|
assert(compile != NULL);
|
|
|
|
|
maat_compile_set(compile, table_name, compile_rule->declared_clause_num,
|
|
|
|
|
compile_rule, (void (*)(void *))compile_rule_free);
|
|
|
|
|
rcu_hash_add(compile_rt->cfg_hash_tbl, (char *)&compile_id, sizeof(long long), compile);
|
|
|
|
|
}
|
|
|
|
|
} else {
|
|
|
|
|
compile = rcu_hash_find(compile_rt->cfg_hash_tbl, (char *)&compile_id, sizeof(long long));
|
|
|
|
|
if (compile != NULL) {
|
|
|
|
|
/********************************************************************************
|
|
|
|
|
compile found in effective hash(added by group2compile runtime), which means
|
|
|
|
|
|
|
|
|
|
1. rcu_hash_add(htable, compile) ==> finished
|
|
|
|
|
2. rcu_hash_commit(htable) ==> finished
|
|
|
|
|
|
|
|
|
|
can only be deleted but not modified
|
|
|
|
|
before delete it, we need to make a copy for further use
|
|
|
|
|
***********************************************************************************/
|
|
|
|
|
struct maat_compile *copy_compile = maat_compile_clone(compile, 0);
|
|
|
|
|
assert(copy_compile != NULL);
|
|
|
|
|
|
|
|
|
|
/* delete compile from rcu hash */
|
|
|
|
|
rcu_hash_del(compile_rt->cfg_hash_tbl, (char *)&compile_id, sizeof(long long));
|
|
|
|
|
|
|
|
|
|
/* copy_compile has group2compile_table info, so set compile_table info */
|
|
|
|
|
maat_compile_set(copy_compile, table_name, compile_rule->declared_clause_num,
|
|
|
|
|
compile_rule, (void (*)(void *))compile_rule_free);
|
|
|
|
|
/* add copy_compile to rcu hash */
|
|
|
|
|
rcu_hash_add(compile_rt->cfg_hash_tbl, (char *)&compile_id, sizeof(long long), copy_compile);
|
|
|
|
|
} else {
|
2023-05-23 17:50:53 +08:00
|
|
|
compile = maat_compile_new(compile_rule->compile_id);
|
2023-04-14 11:32:59 +08:00
|
|
|
assert(compile != NULL);
|
|
|
|
|
maat_compile_set(compile, table_name, compile_rule->declared_clause_num,
|
|
|
|
|
compile_rule, (void (*)(void *))compile_rule_free);
|
|
|
|
|
rcu_hash_add(compile_rt->cfg_hash_tbl, (char *)&compile_id, sizeof(long long), compile);
|
|
|
|
|
}
|
|
|
|
|
}
|
2023-04-20 15:34:56 +08:00
|
|
|
|
|
|
|
|
return 0;
|
2023-04-14 11:32:59 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void compile_runtime_del_compile(struct compile_runtime *compile_rt, long long compile_id)
|
|
|
|
|
{
|
|
|
|
|
struct maat_compile *compile = NULL;
|
|
|
|
|
|
|
|
|
|
int updating_flag = rcu_hash_is_updating(compile_rt->cfg_hash_tbl);
|
|
|
|
|
|
|
|
|
|
if (1 == updating_flag) {
|
|
|
|
|
// find in updating hash
|
|
|
|
|
compile = rcu_updating_hash_find(compile_rt->cfg_hash_tbl, (char *)&compile_id,
|
|
|
|
|
sizeof(long long));
|
|
|
|
|
if (compile != NULL) {
|
|
|
|
|
/****************************************************************
|
|
|
|
|
compile found in updating hash, which means
|
|
|
|
|
|
|
|
|
|
1. rcu_hash_del(htable, compile) ==> finished
|
|
|
|
|
2. rcu_hash_commit(htable) ==> undo
|
|
|
|
|
because it's in updating hash, we can modify it directly
|
|
|
|
|
******************************************************************/
|
|
|
|
|
if (compile->user_data_free && compile->user_data) {
|
|
|
|
|
compile->user_data_free(compile->user_data);
|
|
|
|
|
compile->user_data = NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (0 == compile->actual_clause_num) {
|
|
|
|
|
rcu_hash_del(compile_rt->cfg_hash_tbl, (char *)&compile_id, sizeof(long long));
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
} else {
|
|
|
|
|
// find in effective hash
|
|
|
|
|
compile = rcu_hash_find(compile_rt->cfg_hash_tbl, (char *)&compile_id, sizeof(long long));
|
|
|
|
|
if (compile != NULL) {
|
|
|
|
|
/*******************************************************************
|
|
|
|
|
compile found in effective hash, which means
|
|
|
|
|
|
|
|
|
|
1. rcu_hash_add(htable, compile) ==> finished
|
|
|
|
|
2. rcu_hash_commit(htable) ==> finished
|
|
|
|
|
|
|
|
|
|
can only be deleted but not modified
|
|
|
|
|
before delete it, we need to make a copy for further use
|
|
|
|
|
*********************************************************************/
|
|
|
|
|
struct maat_compile *copy_compile = maat_compile_clone(compile, 0);
|
|
|
|
|
assert(copy_compile != NULL);
|
|
|
|
|
|
|
|
|
|
/* delete compile from rcu hash */
|
|
|
|
|
rcu_hash_del(compile_rt->cfg_hash_tbl, (char *)&compile_id, sizeof(long long));
|
|
|
|
|
|
|
|
|
|
if (0 == copy_compile->actual_clause_num) {
|
|
|
|
|
maat_compile_free(copy_compile);
|
|
|
|
|
} else {
|
|
|
|
|
rcu_hash_add(compile_rt->cfg_hash_tbl, (char *)&compile_id,
|
|
|
|
|
sizeof(long long), copy_compile);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2023-02-03 17:28:14 +08:00
|
|
|
int compile_runtime_update(void *compile_runtime, void *compile_schema,
|
2023-03-29 22:25:14 +08:00
|
|
|
const char *table_name, const char *line,
|
|
|
|
|
int valid_column)
|
2023-01-30 21:59:35 +08:00
|
|
|
{
|
2023-04-14 11:32:59 +08:00
|
|
|
if (NULL == compile_runtime || NULL == compile_schema || NULL == line) {
|
2023-01-30 21:59:35 +08:00
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
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) {
|
2023-05-30 16:16:18 +08:00
|
|
|
log_error(compile_rt->logger, MODULE_COMPILE,
|
|
|
|
|
"[%s:%d] compile table:<%s> has no is_valid(column seq:%d)"
|
|
|
|
|
" in table_line:%s", __FUNCTION__, __LINE__, table_name,
|
|
|
|
|
valid_column, line);
|
2023-04-20 15:34:56 +08:00
|
|
|
compile_rt->update_err_cnt++;
|
2023-01-30 21:59:35 +08:00
|
|
|
return -1;
|
2023-02-16 16:45:06 +08:00
|
|
|
}
|
|
|
|
|
|
2023-02-22 15:08:52 +08:00
|
|
|
long long compile_id = get_column_value(line, schema->compile_id_column);
|
|
|
|
|
if (compile_id < 0) {
|
2023-05-30 16:16:18 +08:00
|
|
|
log_error(compile_rt->logger, MODULE_COMPILE,
|
|
|
|
|
"[%s:%d] compile table:<%s> has no compile_id(column seq:%d)"
|
|
|
|
|
" in table_line:%s", __FUNCTION__, __LINE__, table_name,
|
|
|
|
|
schema->compile_id_column, line);
|
2023-04-20 15:34:56 +08:00
|
|
|
compile_rt->update_err_cnt++;
|
2023-02-22 15:08:52 +08:00
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
|
2023-02-16 16:45:06 +08:00
|
|
|
if (0 == is_valid) {
|
2023-04-14 11:32:59 +08:00
|
|
|
// delete
|
|
|
|
|
compile_runtime_del_compile(compile_rt, compile_id);
|
2023-01-30 21:59:35 +08:00
|
|
|
} else {
|
2023-04-14 11:32:59 +08:00
|
|
|
// add
|
2023-04-20 15:34:56 +08:00
|
|
|
int ret = compile_runtime_add_compile(compile_rt, schema, compile_id,
|
|
|
|
|
table_name, line);
|
|
|
|
|
if (ret < 0) {
|
|
|
|
|
compile_rt->update_err_cnt++;
|
|
|
|
|
}
|
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,
|
2023-03-29 22:25:14 +08:00
|
|
|
const char *table_name, 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;
|
2023-04-24 02:14:26 +00:00
|
|
|
|
2023-01-30 21:59:35 +08:00
|
|
|
int is_valid = get_column_value(line, valid_column);
|
|
|
|
|
if (is_valid < 0) {
|
2023-05-30 16:16:18 +08:00
|
|
|
log_error(compile_rt->logger, MODULE_COMPILE,
|
|
|
|
|
"[%s:%d] g2c table:<%s> has no is_valid(column seq:%d)"
|
|
|
|
|
" in table_line:%s", __FUNCTION__, __LINE__, table_name,
|
|
|
|
|
valid_column, line);
|
2023-04-20 15:34:56 +08:00
|
|
|
g2c_rt->update_err_cnt++;
|
2023-01-30 21:59:35 +08:00
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
int ret = -1;
|
2023-03-29 22:25:14 +08:00
|
|
|
struct group2compile_item *g2c_item = group2compile_item_new(line, schema, table_name,
|
|
|
|
|
compile_rt->logger);
|
2023-01-31 20:39:53 +08:00
|
|
|
if (NULL == g2c_item) {
|
2023-04-20 15:34:56 +08:00
|
|
|
g2c_rt->update_err_cnt++;
|
2023-01-31 20:39:53 +08:00
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
|
2023-05-17 14:34:34 +08:00
|
|
|
struct group_reference *group_ref = NULL;
|
2023-01-30 21:59:35 +08:00
|
|
|
if (0 == is_valid) {
|
|
|
|
|
//delete
|
2023-05-17 14:34:34 +08:00
|
|
|
pthread_mutex_lock(&(compile_rt->mutex));
|
|
|
|
|
HASH_FIND(hh, compile_rt->group_ref_hash, (char *)&(g2c_item->group_id),
|
|
|
|
|
sizeof(long long), group_ref);
|
|
|
|
|
if (group_ref != NULL) {
|
|
|
|
|
if (0 == group_ref->ref_by_compile_cnt) {
|
|
|
|
|
HASH_DEL(compile_rt->group_ref_hash, group_ref);
|
|
|
|
|
FREE(group_ref);
|
|
|
|
|
} else {
|
|
|
|
|
group_ref->ref_by_compile_cnt--;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
pthread_mutex_unlock(&(compile_rt->mutex));
|
|
|
|
|
|
|
|
|
|
ret = maat_remove_group_from_compile(compile_rt->cfg_hash_tbl, g2c_item,
|
|
|
|
|
compile_rt->logger);
|
2023-01-30 21:59:35 +08:00
|
|
|
if (0 == ret) {
|
|
|
|
|
if (g2c_item->not_flag) {
|
|
|
|
|
g2c_rt->not_flag_group--;
|
|
|
|
|
}
|
2023-04-12 19:20:05 +08:00
|
|
|
g2c_rt->rule_num--;
|
2023-04-20 15:34:56 +08:00
|
|
|
} else {
|
|
|
|
|
g2c_rt->update_err_cnt++;
|
2023-01-30 21:59:35 +08:00
|
|
|
}
|
|
|
|
|
} else {
|
|
|
|
|
//add
|
2023-05-17 14:34:34 +08:00
|
|
|
pthread_mutex_lock(&(compile_rt->mutex));
|
|
|
|
|
HASH_FIND(hh, compile_rt->group_ref_hash, (char *)&(g2c_item->group_id),
|
|
|
|
|
sizeof(long long), group_ref);
|
|
|
|
|
if (NULL == group_ref) {
|
|
|
|
|
group_ref = ALLOC(struct group_reference, 1);
|
|
|
|
|
group_ref->group_id = g2c_item->group_id;
|
|
|
|
|
HASH_ADD_KEYPTR(hh, compile_rt->group_ref_hash, (char *)&(group_ref->group_id),
|
2023-05-23 03:23:39 +00:00
|
|
|
sizeof(long long), group_ref);
|
2023-05-17 14:34:34 +08:00
|
|
|
}
|
|
|
|
|
group_ref->ref_by_compile_cnt++;
|
|
|
|
|
pthread_mutex_unlock(&(compile_rt->mutex));
|
|
|
|
|
|
2023-04-14 11:32:59 +08:00
|
|
|
ret = maat_add_group_to_compile(compile_rt->cfg_hash_tbl, g2c_item, compile_rt->logger);
|
2023-01-30 21:59:35 +08:00
|
|
|
if (0 == ret) {
|
|
|
|
|
if (g2c_item->not_flag) {
|
|
|
|
|
g2c_rt->not_flag_group++;
|
|
|
|
|
}
|
2023-04-12 19:20:05 +08:00
|
|
|
g2c_rt->rule_num++;
|
2023-04-20 15:34:56 +08:00
|
|
|
} else {
|
|
|
|
|
g2c_rt->update_err_cnt++;
|
2023-01-30 21:59:35 +08:00
|
|
|
}
|
|
|
|
|
}
|
2023-02-22 15:08:52 +08:00
|
|
|
|
2023-01-30 21:59:35 +08:00
|
|
|
group2compile_item_free(g2c_item);
|
|
|
|
|
return ret;
|
|
|
|
|
}
|
|
|
|
|
|
2023-05-17 14:34:34 +08:00
|
|
|
int group_referenced_by_compile(struct compile_runtime *compile_rt, long long group_id)
|
|
|
|
|
{
|
|
|
|
|
struct group_reference *group_ref = NULL;
|
2023-05-23 03:23:39 +00:00
|
|
|
|
2023-05-17 14:34:34 +08:00
|
|
|
pthread_mutex_lock(&(compile_rt->mutex));
|
|
|
|
|
HASH_FIND(hh, compile_rt->group_ref_hash, (char *)&group_id, sizeof(long long), group_ref);
|
|
|
|
|
if (group_ref != NULL) {
|
|
|
|
|
pthread_mutex_unlock(&(compile_rt->mutex));
|
|
|
|
|
return 1;
|
|
|
|
|
}
|
|
|
|
|
pthread_mutex_unlock(&(compile_rt->mutex));
|
|
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
|
2023-04-20 15:34:56 +08:00
|
|
|
long long group2compile_runtime_not_group_count(void *g2c_runtime)
|
|
|
|
|
{
|
|
|
|
|
if (NULL == g2c_runtime) {
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
struct group2compile_runtime *g2c_rt = (struct group2compile_runtime *)g2c_runtime;
|
|
|
|
|
return g2c_rt->not_flag_group;
|
|
|
|
|
}
|
|
|
|
|
|
2023-04-12 19:20:05 +08:00
|
|
|
long long group2compile_runtime_rule_count(void *g2c_runtime)
|
|
|
|
|
{
|
|
|
|
|
if (NULL == g2c_runtime) {
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
struct group2compile_runtime *g2c_rt = (struct group2compile_runtime *)g2c_runtime;
|
|
|
|
|
return g2c_rt->rule_num;
|
|
|
|
|
}
|
|
|
|
|
|
2023-04-20 15:34:56 +08:00
|
|
|
long long group2compile_runtime_update_err_count(void *g2c_runtime)
|
|
|
|
|
{
|
|
|
|
|
if (NULL == g2c_runtime) {
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
struct group2compile_runtime *g2c_rt = (struct group2compile_runtime *)g2c_runtime;
|
|
|
|
|
return g2c_rt->update_err_cnt;
|
|
|
|
|
}
|
|
|
|
|
|
2023-05-30 16:16:18 +08:00
|
|
|
int compile_runtime_commit(void *compile_runtime, const char *table_name,
|
|
|
|
|
long long maat_rt_version)
|
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-03-15 11:36:54 +08:00
|
|
|
|
2023-04-14 11:32:59 +08:00
|
|
|
int updating_flag = rcu_hash_is_updating(compile_rt->cfg_hash_tbl);
|
2023-03-30 21:03:59 +08:00
|
|
|
if (0 == updating_flag) {
|
2023-03-15 15:11:07 +08:00
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
|
2023-01-31 20:39:53 +08:00
|
|
|
int ret = 0;
|
2023-04-14 11:32:59 +08:00
|
|
|
size_t compile_cnt = 0;
|
|
|
|
|
struct bool_matcher *old_bool_matcher = NULL;
|
|
|
|
|
struct bool_matcher *new_bool_matcher = NULL;
|
|
|
|
|
|
|
|
|
|
new_bool_matcher = maat_compile_bool_matcher_new(compile_rt, &compile_cnt);
|
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-05-30 16:16:18 +08:00
|
|
|
"[%s:%d] table[%s] rebuild compile bool_matcher failed, compile"
|
|
|
|
|
" rules count:%zu", __FUNCTION__, __LINE__, table_name, compile_cnt);
|
2023-01-30 21:59:35 +08:00
|
|
|
ret = -1;
|
2023-04-14 11:32:59 +08:00
|
|
|
} else {
|
|
|
|
|
log_info(compile_rt->logger, MODULE_COMPILE,
|
2023-05-30 16:16:18 +08:00
|
|
|
"table[%s] commit %zu compile rules and rebuild compile bool_matcher"
|
|
|
|
|
" completed, version:%lld", table_name, compile_cnt, maat_rt_version);
|
2023-01-30 21:59:35 +08:00
|
|
|
}
|
|
|
|
|
|
2023-04-21 17:19:43 +08:00
|
|
|
struct literal_clause *old_literal2clause = NULL;
|
|
|
|
|
struct literal_clause *new_literal2clause = NULL;
|
|
|
|
|
|
|
|
|
|
new_literal2clause = maat_compile_build_literal2clause_hash(compile_rt);
|
|
|
|
|
|
|
|
|
|
old_literal2clause = compile_rt->literal2clause_hash;
|
2023-01-30 21:59:35 +08:00
|
|
|
old_bool_matcher = compile_rt->bm;
|
2023-04-21 17:19:43 +08:00
|
|
|
|
2023-01-30 21:59:35 +08:00
|
|
|
compile_rt->bm = new_bool_matcher;
|
2023-04-21 17:19:43 +08:00
|
|
|
compile_rt->literal2clause_hash = new_literal2clause;
|
|
|
|
|
|
2023-04-14 11:32:59 +08:00
|
|
|
rcu_hash_commit(compile_rt->cfg_hash_tbl);
|
2023-04-13 14:56:35 +08:00
|
|
|
|
2023-03-29 22:25:14 +08:00
|
|
|
maat_garbage_bagging(compile_rt->ref_garbage_bin, old_bool_matcher, NULL,
|
|
|
|
|
garbage_bool_matcher_free);
|
2023-04-21 17:19:43 +08:00
|
|
|
maat_garbage_bagging(compile_rt->ref_garbage_bin, old_literal2clause, NULL,
|
|
|
|
|
garbage_literal2clause_hash_free);
|
|
|
|
|
|
2023-04-14 11:32:59 +08:00
|
|
|
compile_rt->rule_num = rcu_hash_count(compile_rt->cfg_hash_tbl);
|
2023-01-30 21:59:35 +08:00
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
|
}
|
|
|
|
|
|
2023-04-12 19:20:05 +08:00
|
|
|
long long compile_runtime_rule_count(void *compile_runtime)
|
|
|
|
|
{
|
|
|
|
|
if (NULL == compile_runtime) {
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
struct compile_runtime *compile_rt = (struct compile_runtime *)compile_runtime;
|
|
|
|
|
return compile_rt->rule_num;
|
|
|
|
|
}
|
|
|
|
|
|
2023-04-20 15:34:56 +08:00
|
|
|
long long compile_runtime_update_err_count(void *compile_runtime)
|
|
|
|
|
{
|
|
|
|
|
if (NULL == compile_runtime) {
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
struct compile_runtime *compile_rt = (struct compile_runtime *)compile_runtime;
|
|
|
|
|
return compile_rt->update_err_cnt;
|
|
|
|
|
}
|
|
|
|
|
|
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 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,
|
2023-05-31 14:04:35 +08:00
|
|
|
const struct compile_rule *rule)
|
2023-01-30 21:59:35 +08:00
|
|
|
{
|
2023-05-31 14:04:35 +08:00
|
|
|
para->compile_id = rule->compile_id;
|
|
|
|
|
para->declared_clause_num = rule->declared_clause_num;
|
2023-01-30 21:59:35 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
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;
|
2023-05-31 14:04:35 +08:00
|
|
|
compile_sort_para_set(&sa, ra);
|
|
|
|
|
compile_sort_para_set(&sb, rb);
|
2023-01-30 21:59:35 +08:00
|
|
|
|
|
|
|
|
return compile_sort_para_compare(&sa, &sb);
|
|
|
|
|
}
|
|
|
|
|
|
2023-03-06 10:45:36 +08:00
|
|
|
int compile_runtime_match(struct compile_runtime *compile_rt, long long *compile_ids,
|
2023-03-22 20:40:36 +08:00
|
|
|
size_t compile_ids_size, 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-05-04 17:10:19 +08:00
|
|
|
size_t bool_match_ret = maat_compile_bool_matcher_match(compile_rt, is_last_scan,
|
2023-06-09 16:44:47 +08:00
|
|
|
compile_state, state->thread_id,
|
2023-05-04 17:10:19 +08:00
|
|
|
(void **)compile_rules,
|
|
|
|
|
compile_ids_size);
|
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-03-22 20:40:36 +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
|
|
|
}
|
|
|
|
|
|
2023-05-07 23:09:33 +08:00
|
|
|
void maat_compile_state_update(int vtable_id, struct maat_item *hit_items,
|
|
|
|
|
size_t n_hit_item, struct maat_state *state)
|
2023-02-03 17:28:14 +08:00
|
|
|
{
|
2023-05-07 23:09:33 +08:00
|
|
|
size_t i = 0;
|
|
|
|
|
size_t hit_cnt = n_hit_item;
|
2023-02-22 15:22:41 +08:00
|
|
|
long long hit_group_ids[MAX_SCANNER_HIT_GROUP_NUM];
|
2023-02-03 17:28:14 +08:00
|
|
|
|
2023-05-07 23:09:33 +08:00
|
|
|
if (hit_cnt >= MAX_SCANNER_HIT_GROUP_NUM) {
|
|
|
|
|
hit_cnt = MAX_SCANNER_HIT_GROUP_NUM;
|
2023-02-03 17:28:14 +08:00
|
|
|
}
|
|
|
|
|
|
2023-06-08 17:43:05 +08:00
|
|
|
struct maat *maat_instance = state->maat_instance;
|
2023-06-01 14:47:20 +08:00
|
|
|
if (NULL == state->compile_state) {
|
2023-06-09 16:44:47 +08:00
|
|
|
state->compile_state = maat_compile_state_new();
|
2023-06-08 17:43:05 +08:00
|
|
|
alignment_int64_array_add(maat_instance->stat->maat_compile_state_cnt,
|
|
|
|
|
state->thread_id, 1);
|
2023-06-01 14:47:20 +08:00
|
|
|
}
|
|
|
|
|
|
2023-05-07 23:09:33 +08:00
|
|
|
for (i = 0; i < hit_cnt; i++) {
|
|
|
|
|
maat_compile_state_update_hit_path(state->compile_state, hit_items[i].item_id,
|
|
|
|
|
hit_items[i].group_id, vtable_id, state->scan_cnt, i);
|
|
|
|
|
hit_group_ids[i] = hit_items[i].group_id;
|
|
|
|
|
}
|
2023-02-03 17:28:14 +08:00
|
|
|
|
|
|
|
|
/* update hit clause */
|
2023-03-22 20:40:36 +08:00
|
|
|
int compile_table_id = -1;
|
2023-03-30 20:49:59 +08:00
|
|
|
if (state->compile_table_id > 0) {
|
2023-03-22 20:40:36 +08:00
|
|
|
compile_table_id = state->compile_table_id;
|
2023-02-03 17:28:14 +08:00
|
|
|
} else {
|
2023-06-08 17:43:05 +08:00
|
|
|
compile_table_id = maat_instance->default_compile_table_id;
|
2023-02-20 10:57:40 +08:00
|
|
|
}
|
|
|
|
|
|
2023-06-08 17:43:05 +08:00
|
|
|
void *compile_rt = table_manager_get_runtime(maat_instance->tbl_mgr,
|
2023-03-22 20:40:36 +08:00
|
|
|
compile_table_id);
|
2023-03-30 20:49:59 +08:00
|
|
|
if (NULL == compile_rt) {
|
2023-05-07 23:09:33 +08:00
|
|
|
return;
|
2023-03-30 20:49:59 +08:00
|
|
|
}
|
2023-03-22 20:40:36 +08:00
|
|
|
|
2023-06-08 17:43:05 +08:00
|
|
|
void *g2g_rt = table_manager_get_runtime(maat_instance->tbl_mgr,
|
|
|
|
|
maat_instance->g2g_table_id);
|
2023-03-30 22:53:56 +08:00
|
|
|
if (NULL == g2g_rt) {
|
2023-05-07 23:09:33 +08:00
|
|
|
return;
|
2023-03-30 22:53:56 +08:00
|
|
|
}
|
|
|
|
|
|
2023-05-07 23:09:33 +08:00
|
|
|
|
2023-05-04 17:10:19 +08:00
|
|
|
long long super_group_ids[MAX_SCANNER_HIT_GROUP_NUM];
|
|
|
|
|
size_t super_group_cnt = group2group_runtime_get_super_groups(g2g_rt, hit_group_ids,
|
2023-05-07 23:09:33 +08:00
|
|
|
hit_cnt, super_group_ids,
|
2023-05-04 17:10:19 +08:00
|
|
|
MAX_SCANNER_HIT_GROUP_NUM);
|
|
|
|
|
if (super_group_cnt >= MAX_SCANNER_HIT_GROUP_NUM) {
|
|
|
|
|
super_group_cnt = MAX_SCANNER_HIT_GROUP_NUM;
|
2023-03-30 16:50:05 +08:00
|
|
|
}
|
|
|
|
|
|
2023-05-04 17:10:19 +08:00
|
|
|
for (int j = 0; j < super_group_cnt; j++) {
|
2023-03-30 16:50:05 +08:00
|
|
|
maat_compile_state_update_hit_clause(state->compile_state, compile_rt,
|
2023-05-04 17:10:19 +08:00
|
|
|
super_group_ids[j], vtable_id);
|
2023-02-03 17:28:14 +08:00
|
|
|
}
|
|
|
|
|
|
2023-05-07 23:09:33 +08:00
|
|
|
for (int j = 0; j < hit_cnt; j++) {
|
2023-04-24 02:14:26 +00:00
|
|
|
maat_compile_state_update_hit_clause(state->compile_state, compile_rt,
|
|
|
|
|
hit_group_ids[j], vtable_id);
|
|
|
|
|
}
|
2023-02-22 15:22:41 +08:00
|
|
|
}
|
2023-03-06 16:45:34 +08:00
|
|
|
|
2023-06-05 15:14:45 +08:00
|
|
|
size_t maat_compile_state_get_hit_groups(struct maat_compile_state *compile_state,
|
2023-06-09 16:44:47 +08:00
|
|
|
struct group2group_runtime *g2g_rt,
|
2023-06-05 15:14:45 +08:00
|
|
|
struct maat_hit_group *hit_group_array,
|
|
|
|
|
size_t array_size)
|
|
|
|
|
{
|
|
|
|
|
if (NULL == compile_state) {
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
size_t i = 0;
|
2023-06-09 16:44:47 +08:00
|
|
|
UT_array *all_hit_groups;
|
|
|
|
|
utarray_new(all_hit_groups, &ut_hit_group_icd);
|
|
|
|
|
struct maat_internal_hit_path *internal_path = NULL;
|
|
|
|
|
|
|
|
|
|
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);
|
|
|
|
|
|
|
|
|
|
long long super_group_ids[MAX_SCANNER_HIT_GROUP_NUM];
|
|
|
|
|
size_t super_group_cnt = group2group_runtime_get_super_groups(g2g_rt, &(internal_path->group_id), 1,
|
|
|
|
|
super_group_ids, MAX_SCANNER_HIT_GROUP_NUM);
|
|
|
|
|
if (super_group_cnt + 1 <= MAX_SCANNER_HIT_GROUP_NUM) {
|
|
|
|
|
super_group_ids[super_group_cnt++] = internal_path->group_id;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
for (size_t idx = 0; idx < super_group_cnt; idx++) {
|
|
|
|
|
struct maat_hit_group hit_group;
|
|
|
|
|
hit_group.group_id = super_group_ids[idx];
|
|
|
|
|
hit_group.vtable_id = internal_path->vtable_id;
|
|
|
|
|
if (utarray_find(all_hit_groups, &hit_group, compare_hit_group)) {
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
utarray_push_back(all_hit_groups, &hit_group);
|
|
|
|
|
utarray_sort(all_hit_groups, compare_hit_group);
|
|
|
|
|
}
|
|
|
|
|
}
|
2023-06-05 15:14:45 +08:00
|
|
|
|
2023-06-09 16:44:47 +08:00
|
|
|
struct maat_hit_group *tmp = NULL;
|
|
|
|
|
for (i = 0; i < utarray_len(all_hit_groups) && i < array_size; i++) {
|
|
|
|
|
tmp = (struct maat_hit_group *)utarray_eltptr(all_hit_groups, i);
|
|
|
|
|
hit_group_array[i] = *tmp;
|
2023-06-05 15:14:45 +08:00
|
|
|
}
|
2023-06-09 16:44:47 +08:00
|
|
|
utarray_free(all_hit_groups);
|
2023-06-05 15:14:45 +08:00
|
|
|
|
|
|
|
|
return i;
|
|
|
|
|
}
|
|
|
|
|
|
2023-05-17 14:34:34 +08:00
|
|
|
UT_icd ut_compile_group_id_icd = {sizeof(long long), NULL, NULL, NULL};
|
2023-03-27 15:52:47 +08:00
|
|
|
size_t maat_compile_state_get_internal_hit_paths(struct maat_compile_state *compile_state,
|
2023-05-17 14:34:34 +08:00
|
|
|
struct compile_runtime *compile_rt,
|
2023-05-04 17:10:19 +08:00
|
|
|
struct group2group_runtime *g2g_rt,
|
|
|
|
|
struct maat_hit_path *hit_path_array,
|
|
|
|
|
size_t array_size)
|
2023-03-06 16:45:34 +08:00
|
|
|
{
|
|
|
|
|
size_t hit_path_cnt = 0;
|
|
|
|
|
struct maat_internal_hit_path *internal_path = NULL;
|
|
|
|
|
|
|
|
|
|
for (int 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);
|
|
|
|
|
/*
|
|
|
|
|
NOTE: maybe one item has been deleted, but it's item_id still exist in internal_hit_paths
|
|
|
|
|
*/
|
2023-05-04 17:10:19 +08:00
|
|
|
long long super_group_ids[MAX_SCANNER_HIT_GROUP_NUM];
|
2023-05-17 14:34:34 +08:00
|
|
|
UT_array *valid_super_group_ids;
|
|
|
|
|
utarray_new(valid_super_group_ids, &ut_compile_group_id_icd);
|
|
|
|
|
|
2023-05-08 16:39:58 +08:00
|
|
|
size_t super_group_cnt = group2group_runtime_get_super_groups(g2g_rt, &(internal_path->group_id), 1,
|
|
|
|
|
super_group_ids, MAX_SCANNER_HIT_GROUP_NUM);
|
2023-05-17 14:34:34 +08:00
|
|
|
/* if super group is not referenced by compile, drop it */
|
|
|
|
|
for (size_t idx = 0; idx < super_group_cnt; idx++) {
|
|
|
|
|
if (0 == group_referenced_by_compile(compile_rt, super_group_ids[idx])) {
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
utarray_push_back(valid_super_group_ids, &super_group_ids[idx]);
|
2023-03-06 16:45:34 +08:00
|
|
|
}
|
|
|
|
|
|
2023-05-17 14:34:34 +08:00
|
|
|
/*
|
|
|
|
|
if internal_path->group_id can be referenced directly by compile,
|
|
|
|
|
so add it to hit_path which super_group_ids is -1
|
|
|
|
|
------------------------------------------------------------------------------
|
|
|
|
|
NOTE: Add the hit path as long as the item is hit
|
|
|
|
|
*/
|
|
|
|
|
if (0 == utarray_len(valid_super_group_ids)) {
|
|
|
|
|
long long super_group_id = -1;
|
|
|
|
|
utarray_push_back(valid_super_group_ids, &super_group_id);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
long long *p = NULL;
|
2023-03-06 16:45:34 +08:00
|
|
|
struct maat_hit_path tmp_path;
|
2023-05-17 14:34:34 +08:00
|
|
|
|
|
|
|
|
for (p = utarray_front(valid_super_group_ids); p != NULL && hit_path_cnt < array_size;
|
|
|
|
|
p = utarray_next(valid_super_group_ids, p)) {
|
2023-03-06 16:45:34 +08:00
|
|
|
memset(&tmp_path, 0, sizeof(tmp_path));
|
|
|
|
|
tmp_path.Nth_scan = internal_path->Nth_scan;
|
|
|
|
|
tmp_path.item_id = internal_path->item_id;
|
|
|
|
|
tmp_path.sub_group_id = internal_path->group_id;
|
2023-05-17 14:34:34 +08:00
|
|
|
tmp_path.top_group_id = *p;
|
2023-03-06 16:45:34 +08:00
|
|
|
tmp_path.vtable_id = internal_path->vtable_id;
|
|
|
|
|
tmp_path.compile_id = -1;
|
|
|
|
|
|
2023-05-17 14:34:34 +08:00
|
|
|
/* check if internal_path is duplicated from hit_path_array[] element */
|
2023-03-06 16:45:34 +08:00
|
|
|
if (hit_path_cnt > 0) {
|
|
|
|
|
if (maat_compile_is_hit_path_existed(hit_path_array, hit_path_cnt, &tmp_path)) {
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
}
|
2023-05-17 14:34:34 +08:00
|
|
|
|
2023-03-06 16:45:34 +08:00
|
|
|
hit_path_array[hit_path_cnt] = tmp_path;
|
2023-04-24 02:14:26 +00:00
|
|
|
hit_path_cnt++;
|
2023-03-06 16:45:34 +08:00
|
|
|
}
|
2023-05-17 14:34:34 +08:00
|
|
|
utarray_free(valid_super_group_ids);
|
2023-03-06 16:45:34 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return hit_path_cnt;
|
|
|
|
|
}
|