Mirror/youtubeUnblock
1
0
Fork 0
mirror of https://github.com/Waujito/youtubeUnblock.git synced 2026-01-01 22:28:49 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add more error handlers

Browse Source
This commit is contained in:
Vadim Vetrov
2024-07-23 22:24:43 +03:00
parent 8b4b14bddf
commit 692add84c5
1 changed files with 22 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
youtubeUnblock.c
View File

@@ -218,7 +218,14 @@ static int ipv4_frag(struct pkt_buff *pktb, size_t payload_offset,
nfq_ip_set_checksum(f2_hdr); nfq_ip_set_checksum(f2_hdr);
*frag1 = pktb_alloc(AF_INET, buff1, f1_dlen, 0); *frag1 = pktb_alloc(AF_INET, buff1, f1_dlen, 0);
if (*frag1 == NULL)
return -1;
*frag2 = pktb_alloc(AF_INET, buff2, f2_dlen, 0); *frag2 = pktb_alloc(AF_INET, buff2, f2_dlen, 0);
if (*frag2 == NULL) {
pktb_free(*frag1);
return -1;
}
return 0; return 0;
} }
@@ -291,7 +298,15 @@ static int tcp4_frag(struct pkt_buff *pktb, size_t payload_offset,
nfq_tcp_compute_checksum_ipv4(s2_tcph, s2_hdr); nfq_tcp_compute_checksum_ipv4(s2_tcph, s2_hdr);
*seg1 = pktb_alloc(AF_INET, buff1, s1_dlen, 0); *seg1 = pktb_alloc(AF_INET, buff1, s1_dlen, 0);
if (*seg1 == NULL)
return -1;
*seg2 = pktb_alloc(AF_INET, buff2, s2_dlen, 0); *seg2 = pktb_alloc(AF_INET, buff2, s2_dlen, 0);
if (*seg2 == NULL) {
pktb_free(*seg1);
return -1;
}
return 0; return 0;
} }
@@ -308,9 +323,12 @@ static int send_raw_socket(struct pkt_buff *pktb) {
struct pkt_buff *buff2; struct pkt_buff *buff2;
#ifdef USE_TCP_SEGMENTATION #ifdef USE_TCP_SEGMENTATION
tcp4_frag(pktb, AVAILABLE_MTU-128, &buff1, &buff2); if (tcp4_frag(pktb, AVAILABLE_MTU-128, &buff1, &buff2) < 0)
return -1;
#else #else
ipv4_frag(pktb, AVAILABLE_MTU-128, &buff1, &buff2); if (ipv4_frag(pktb, AVAILABLE_MTU-128, &buff1, &buff2) < 0)
return -1;
#endif #endif
int sent = 0; int sent = 0;
@@ -335,7 +353,6 @@ static int send_raw_socket(struct pkt_buff *pktb) {
return sent; return sent;
} }
struct iphdr *iph = nfq_ip_get_hdr(pktb); struct iphdr *iph = nfq_ip_get_hdr(pktb);
if (iph == NULL) if (iph == NULL)
return -1; return -1;
@@ -350,6 +367,7 @@ static int send_raw_socket(struct pkt_buff *pktb) {
if (tcph != NULL) { if (tcph != NULL) {
sin_port = tcph->dest; sin_port = tcph->dest;
errno = 0;
} else if (udph != NULL) { } else if (udph != NULL) {
sin_port = udph->dest; sin_port = udph->dest;
} else { } else {
@@ -645,7 +663,6 @@ static int process_packet(const struct packet_data packet) {
mid_offset += 8 - mid_offset % 8; mid_offset += 8 - mid_offset % 8;
if (ipv4_frag(pktb, mid_offset, &frag1, &frag2) < 0) { if (ipv4_frag(pktb, mid_offset, &frag1, &frag2) < 0) {
perror("ipv4_frag"); perror("ipv4_frag");
goto fallback; goto fallback;