him411/UA2F
1
0
Fork 0
mirror of https://github.com/Zxilly/UA2F.git synced 2025-12-25 04:28:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: add more guard

Browse Source
This commit is contained in:
Zxilly 2025-03-06 16:44:08 +08:00
parent 8da0500a11
commit 22d3f33687
1 changed files with 15 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/handler.c
View File

@ -289,10 +289,22 @@ void handle_packet(const struct nf_queue *queue, const struct nf_packet *pkt) {
char msg[300];
if (type == IPV4) {
syslog(LOG_WARNING, "Failed to set ipv4 transport header.");
nfq_ip_snprintf(msg, sizeof(msg), nfq_ip_get_hdr(pkt_buff));
const __auto_type ip_hdr = nfq_ip_get_hdr(pkt_buff);
if (ip_hdr != NULL) {
nfq_ip_snprintf(msg, sizeof(msg), ip_hdr);
} else {
syslog(LOG_WARNING, "Failed to get ipv4 ip header");
goto end;
}
} else {
syslog(LOG_WARNING, "Failed to set ipv6 transport header.");
nfq_ip6_snprintf(msg, sizeof(msg), nfq_ip6_get_hdr(pkt_buff));
const __auto_type ip_hdr = nfq_ip6_get_hdr(pkt_buff);
if (ip_hdr != NULL) {
nfq_ip6_snprintf(msg, sizeof(msg), ip_hdr);
} else {
syslog(LOG_WARNING, "Failed to get ipv6 ip header");
goto end;
}
}
syslog(LOG_WARNING, "Header: %s", msg);
goto end;
@ -392,9 +404,7 @@ void handle_packet(const struct nf_queue *queue, const struct nf_packet *pkt) {
send_verdict(queue, pkt, get_next_mark(pkt, has_ua), pkt_buff);
end:
if (pkt->payload != NULL) {
free(pkt->payload);
}
free(pkt->payload);
if (pkt_buff != NULL) {
pktb_free(pkt_buff);
}