gfwleak/tango-fw-dns-plug
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

支持从profile中选择应答记录内容，TSG-6653

Browse Source
This commit is contained in:
liuxueli
2021-06-11 09:39:29 +08:00
parent 2a25c20cea
commit 57c2feb69d
4 changed files with 314 additions and 241 deletions
Download Patch File Download Diff File Expand all files Collapse all files

209
src/fw_dns_rule.cpp
View File

@@ -1,4 +1,5 @@
#include <stdio.h>
#include <time.h>
#include <arpa/inet.h>
#include <MESA/stream.h>
@@ -12,10 +13,10 @@
#include "tsg_rule.h"
#include "fw_dns_rule.h"
fw_dns_rule_t g_fw_dns_rule_info;
struct fw_dns_rule g_fw_dns_rule_info;
char FW_DNS_RULE_VERSION_20191201=0;
struct _dns_str2idx str2index[]={{DNS_TYPE_CNAME, 5, (char *)"CNAME"},
struct dns_str2idx str2index[]={{DNS_TYPE_CNAME, 5, (char *)"CNAME"},
{DNS_TYPE_MX, 2, (char *)"MX"},
{DNS_TYPE_A, 1, (char *)"A"},
{DNS_TYPE_NS, 2, (char *)"NS"},
@@ -28,7 +29,7 @@ int fw_dns_type2index(char *type)
{
int i=0;
for(i=0; i<(int)(sizeof(str2index)/sizeof(struct _dns_str2idx)); i++)
for(i=0; i<(int)(sizeof(str2index)/sizeof(struct dns_str2idx)); i++)
{
if(str2index[i].len==(int)strlen(type) && (strncasecmp(str2index[i].type, type, str2index[i].len))==0)
{
@@ -39,65 +40,31 @@ int fw_dns_type2index(char *type)
return -1;
}
int get_cheat_opt(int record_id, int select_num, int ttl, int atype, cheat_pkt_opt_t* cheat_opt, int cheat_opt_num)
{
int i=0,idx=0;
int used_num=0,real_num=0;
cb_rule_t *cb_rule=NULL;
real_num=(select_num>cheat_opt_num) ? cheat_opt_num : select_num;
cb_rule=(cb_rule_t *)Maat_plugin_get_EX_data(g_tsg_maat_feather, g_fw_dns_rule_info.table_records_id, (char *)&record_id);
if(cb_rule!=NULL && real_num>0)
{
idx = (cb_rule->record_num>=real_num) ? (random()%(cb_rule->record_num-real_num+1)) : 0;
for(i=0; i<real_num && i+idx<cb_rule->record_num; i++)
{
cheat_opt[used_num].ttl=ttl;
cheat_opt[used_num].res_type=atype;
cheat_opt[used_num].cfg_type=atype;
switch(atype)
{
case DNS_TYPE_A:
cheat_opt[used_num].res_len=sizeof(unsigned int);
break;
case DNS_TYPE_AAAA:
cheat_opt[used_num].res_len=IPV6_ADDR_LEN;
break;
default:
cheat_opt[used_num].res_len=MIN(strlen((char *)(cb_rule->record_values[i+idx])), sizeof(cheat_opt[used_num].res_info)-1);
break;
}
memcpy(cheat_opt[used_num].res_info, (char *)(cb_rule->record_values[i+idx]), cheat_opt[used_num].res_len);
used_num++;
}
}
return used_num;
}
void fw_dns_EX_data_create(int table_id, const char* key, const char* table_line, MAAT_PLUGIN_EX_DATA* ad, long argl, void *argp)
{
int i=0;
cJSON *pSub=NULL;
cb_rule_t *cb_rule=(cb_rule_t *)calloc(1, sizeof(cb_rule_t));
char *tmp_records=(char *)calloc(1, strlen(table_line)+1);
sscanf(table_line, "%d %s %s %s %d", &cb_rule->record_id, cb_rule->record_name, cb_rule->record_type, tmp_records, &cb_rule->is_valid);
cJSON *one_record=NULL,*pSub=NULL;
int index=fw_dns_type2index(cb_rule->record_type);
struct dns_records *records=(struct dns_records *)calloc(1, sizeof(struct dns_records));
char *tmp_records=(char *)calloc(1, strlen(table_line)+1);
sscanf(table_line, "%d %s %s %s %d", &records->record_id, records->record_name, records->record_type, tmp_records, &records->is_valid);
int index=fw_dns_type2index(records->record_type);
cJSON *tmp_array=cJSON_Parse(tmp_records);
if(tmp_array != NULL)
if(tmp_array!=NULL)
{
cb_rule->record_num=cJSON_GetArraySize(tmp_array);
*cb_rule->record_values=(void *)malloc(cb_rule->record_num*sizeof(void *));
records->record_num=cJSON_GetArraySize(tmp_array);
records->record_values=(void **)malloc(records->record_num*sizeof(void *));
for(i=0; i<cb_rule->record_num; i++)
for(i=0; i<records->record_num; i++)
{
pSub = cJSON_GetArrayItem(tmp_array, i);
one_record=cJSON_GetArrayItem(tmp_array, i);
if(one_record==NULL)
{
continue;
}
pSub=cJSON_GetObjectItem(one_record, "value");
if(NULL==pSub )
{
continue;
@@ -106,12 +73,12 @@ void fw_dns_EX_data_create(int table_id, const char* key, const char* table_line
switch(index)
{
case DNS_TYPE_A:
cb_rule->record_values[i]=calloc(1, sizeof(unsigned int));
inet_pton(AF_INET, pSub->valuestring, cb_rule->record_values[i]);
records->record_values[i]=calloc(1, sizeof(unsigned int));
inet_pton(AF_INET, pSub->valuestring, records->record_values[i]);
break;
case DNS_TYPE_AAAA:
cb_rule->record_values[i]=calloc(1, IPV6_ADDR_LEN);
inet_pton(AF_INET6, pSub->valuestring, cb_rule->record_values[i]);
records->record_values[i]=calloc(1, IPV6_ADDR_LEN);
inet_pton(AF_INET6, pSub->valuestring, records->record_values[i]);
break;
case DNS_TYPE_MX:
break;
@@ -119,45 +86,141 @@ void fw_dns_EX_data_create(int table_id, const char* key, const char* table_line
case DNS_TYPE_TXT:
case DNS_TYPE_PTR:
case DNS_TYPE_CNAME:
cb_rule->record_values[i]=calloc(1, strlen(pSub->valuestring)+1);
memcpy(cb_rule->record_values[i], pSub->valuestring, strlen(pSub->valuestring));
records->record_values[i]=calloc(1, strlen(pSub->valuestring)+1);
memcpy(records->record_values[i], pSub->valuestring, strlen(pSub->valuestring));
break;
default:
continue;
}
}
records->table_id=table_id;
atomic_inc(&records->ref_cnt);
(*ad)=(MAAT_PLUGIN_EX_DATA)(records);
cJSON_Delete(tmp_array);
tmp_array=NULL;
}
else
{
free(records);
records=NULL;
}
free(tmp_records);
tmp_records=NULL;
return ;
}
void fw_dns_EX_data_free(int table_id, MAAT_PLUGIN_EX_DATA* ad, long argl, void *argp)
{
int i=0;
cb_rule_t *cb_rule=(cb_rule_t *)*ad;
struct dns_records *records=(struct dns_records *)*ad;
if(cb_rule!=NULL)
if(records!=NULL)
{
for(i=0; i<cb_rule->record_num; i++)
atomic_dec(&records->ref_cnt);
if(records->ref_cnt<=0)
{
free(cb_rule->record_values[i]);
cb_rule->record_values[i]=NULL;
}
free(*cb_rule->record_values);
*cb_rule->record_values=NULL;
for(i=0; i<records->record_num; i++)
{
free(records->record_values[i]);
records->record_values[i]=NULL;
}
free(*records->record_values);
*records->record_values=NULL;
free(cb_rule);
cb_rule=NULL;
free(records);
records=NULL;
}
}
return ;
}
void fw_dns_EX_data_dup(int table_id, MAAT_PLUGIN_EX_DATA *to, MAAT_PLUGIN_EX_DATA *from, long argl, void *argp)
{
struct dns_records *records=(struct dns_records *)(*from);
if(records!=NULL)
{
atomic_inc(&records->ref_cnt);
(*to)=(*from);
}
return ;
}
int fw_dns_EX_data_key2index(const char* key)
static int set_one_cheat_opt(cheat_pkt_opt_t *cheat_opt, int ttl, int atype, char *record_values)
{
return 0;
cheat_opt->ttl=ttl;
cheat_opt->res_type=atype;
cheat_opt->cfg_type=atype;
switch(atype)
{
case DNS_TYPE_A:
cheat_opt->res_len=sizeof(unsigned int);
break;
case DNS_TYPE_AAAA:
cheat_opt->res_len=IPV6_ADDR_LEN;
break;
case DNS_TYPE_NS:
case DNS_TYPE_TXT:
case DNS_TYPE_PTR:
case DNS_TYPE_CNAME:
cheat_opt->res_len=MIN(strlen(record_values), sizeof(cheat_opt->res_info)-1);
break;
default:
return 0;
}
memcpy(cheat_opt->res_info, record_values, cheat_opt->res_len);
return 1;
}
int dns_get_cheat_opt(int record_id, int select_num, int ttl, int atype, cheat_pkt_opt_t* cheat_opt, int cheat_opt_num)
{
int i=0,idx=0;
char record_id_str[32]={0};
int used_num=0,real_num=0;
struct dns_records *records=NULL;
real_num=MIN(select_num, cheat_opt_num);
if(real_num<=0)
{
return 0;
}
snprintf(record_id_str, sizeof(record_id_str), "%d", record_id);
records=(struct dns_records *)Maat_plugin_get_EX_data(g_tsg_maat_feather, g_fw_dns_rule_info.table_records_id, record_id_str);
if(records!=NULL)
{
if(records->record_num<=real_num)
{
for(i=0; i<records->record_num; i++)
{
used_num+=set_one_cheat_opt(&(cheat_opt[used_num]), ttl, atype, (char *)(records->record_values[i]));
}
}
else
{
srand(time(0));
idx=rand()%(records->record_num-real_num+1);
for(i=0; i<real_num; i++)
{
used_num+=set_one_cheat_opt(&(cheat_opt[used_num]), ttl, atype, (char *)(records->record_values[idx+i]));
}
}
fw_dns_EX_data_free(records->table_id, (MAAT_PLUGIN_EX_DATA *)&records, 0, NULL);
}
return used_num;
}
@@ -181,7 +244,7 @@ int fw_dns_rule_init(const char *conffile, void *logger)
fw_dns_EX_data_create,
fw_dns_EX_data_free,
fw_dns_EX_data_dup,
fw_dns_EX_data_key2index,
NULL,
arg,
NULL);