Add option to specify 2nd packet delay

This commit is contained in:
Vadim Vetrov 2024-08-01 12:35:03 +03:00
parent 13e78bde6d
commit 4f78c490f4
No known key found for this signature in database
GPG Key ID: E8A308689D7A73A5

View File

@ -19,6 +19,7 @@
#include <linux/netfilter.h> #include <linux/netfilter.h>
#include <linux/if_ether.h> #include <linux/if_ether.h>
#include <sys/socket.h> #include <sys/socket.h>
#include <pthread.h>
#ifndef NOUSE_GSO #ifndef NOUSE_GSO
#define USE_GSO #define USE_GSO
@ -29,6 +30,7 @@
#endif #endif
#define RAWSOCKET_MARK 0xfc70 #define RAWSOCKET_MARK 0xfc70
#define SEG2_DELAY 100
static struct { static struct {
uint32_t queue_num; uint32_t queue_num;
@ -564,7 +566,23 @@ nextMessage:
return vrd; return vrd;
} }
struct dps_t {
struct pkt_buff *pkt;
// Time for the packet in milliseconds
uint32_t timer;
};
// Note that the thread will automatically release dps_t and pkt_buff
void *delay_packet_send(void *data) {
struct dps_t *dpdt = data;
struct pkt_buff *pkt = dpdt->pkt;
usleep(dpdt->timer * 1000);
send_raw_socket(pkt);
pktb_free(pkt);
free(dpdt);
return NULL;
}
static int process_packet(const struct packet_data packet) { static int process_packet(const struct packet_data packet) {
char buf[MNL_SOCKET_BUFFER_SIZE]; char buf[MNL_SOCKET_BUFFER_SIZE];
@ -644,11 +662,31 @@ static int process_packet(const struct packet_data packet) {
goto fallback; goto fallback;
} }
if ((send_raw_socket(frag2) == -1) || (send_raw_socket(frag1) == -1)) { int ret = send_raw_socket(frag2);
if (ret < 0) {
errno = ret;
perror("raw frags send"); perror("raw frags send");
pktb_free(frag1);
goto err;
} }
pktb_free(frag1);
#ifdef SEG2_DELAY
struct dps_t *dpdt = malloc(sizeof(struct dps_t));
dpdt->pkt = frag1;
dpdt->timer = SEG2_DELAY;
pthread_t thr;
pthread_create(&thr, NULL, delay_packet_send, dpdt);
pthread_detach(thr);
#else
ret = send_raw_socket(frag1);
if (ret < 0) {
free(frag1);
goto err;
}
free(frag1);
#endif
err:
pktb_free(frag2); pktb_free(frag2);
pktb_free(pktb); pktb_free(pktb);