daxi20/smartdns
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

crash bug fix

Browse Source
This commit is contained in:
Nick Peng
2018-12-26 20:40:05 +08:00
parent 9398573e6f
commit 48112eb784
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/dns_server.c
View File

@@ -276,6 +276,10 @@ static int _dns_add_rrs(struct dns_packet *packet, struct dns_request *request)
static void _dns_server_client_release(struct dns_server_conn *client) static void _dns_server_client_release(struct dns_server_conn *client)
{ {
if (client == NULL) {
return;
}
int refcnt = atomic_dec_return(&client->refcnt); int refcnt = atomic_dec_return(&client->refcnt);
if (refcnt) { if (refcnt) {
@@ -297,6 +301,9 @@ static void _dns_server_client_release(struct dns_server_conn *client)
static void _dns_server_client_get(struct dns_server_conn *client) static void _dns_server_client_get(struct dns_server_conn *client)
{ {
if (client == NULL) {
return;
}
atomic_inc(&client->refcnt); atomic_inc(&client->refcnt);
} }
@@ -1263,7 +1270,7 @@ static int _dns_server_recv(struct dns_server_conn *client, unsigned char *inpac
int rr_count = 0; int rr_count = 0;
int i = 0; int i = 0;
int qclass; int qclass;
int qtype; int qtype = DNS_T_ALL;
_dns_server_client_get(client); _dns_server_client_get(client);