Skip to content

feat: Add silent mode #24

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
MikeWang000000 merged 1 commit into from
Jun 5, 2025
Merged

feat: Add silent mode #24

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@ Options:
-m <mark> fwmark for bypassing the queue
-n <number> netfilter queue number
-r <repeat> duplicate generated packets for <repeat> times
-s enable silent mode
-t <ttl> TTL for generated packets
-w <file> write log to <file> instead of stderr
-x <mask> set the mask for fwmark
Expand Down
58 changes: 34 additions & 24 deletions src/fakehttp.c
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -44,10 +44,15 @@

#define E(...) logger(__func__, __FILE__, __LINE__, __VA_ARGS__)
#define E_RAW(...) logger_raw(__VA_ARGS__)
#define E_INFO(...) \
if (!g_silent) { \
E(__VA_ARGS__); \
}

static FILE *g_logfp = NULL;
static int g_sockfd = 0;
static int g_exit = 0;
static int g_silent = 0;
static int g_repeat = 3;
static uint32_t g_fwmark = 0x8000;
static uint32_t g_fwmask = 0;
Expand All @@ -68,6 +73,7 @@ static void print_usage(const char *name)
" -n <number> netfilter queue number\n"
" -r <repeat> duplicate generated packets for <repeat> "
"times\n"
" -s enable silent mode\n"
" -t <ttl> TTL for generated packets\n"
" -w <file> write log to <file> instead of stderr\n"
" -x <mask> set the mask for fwmark\n"
Expand Down Expand Up @@ -610,25 +616,26 @@ static int callback(struct nfq_q_handle *qh, struct nfgenmsg *nfmsg,

tcph = (struct tcphdr *) (pkt_data + iph_len);
tcph_len = tcph->doff * 4;
tcp_payload_len = pkt_len - iph_len - tcph_len;

if (!inet_ntop(AF_INET, &iph->saddr, src_ip, sizeof(src_ip))) {
strncpy(src_ip, "INVALID", sizeof(src_ip) - 1);
src_ip[sizeof(src_ip) - 1] = '\0';
}
if (!inet_ntop(AF_INET, &iph->daddr, dst_ip, sizeof(dst_ip))) {
strncpy(dst_ip, "INVALID", sizeof(dst_ip) - 1);
src_ip[sizeof(src_ip) - 1] = '\0';
if (!g_silent) {
if (!inet_ntop(AF_INET, &iph->saddr, src_ip, sizeof(src_ip))) {
strncpy(src_ip, "INVALID", sizeof(src_ip) - 1);
src_ip[sizeof(src_ip) - 1] = '\0';
}
if (!inet_ntop(AF_INET, &iph->daddr, dst_ip, sizeof(dst_ip))) {
strncpy(dst_ip, "INVALID", sizeof(dst_ip) - 1);
src_ip[sizeof(src_ip) - 1] = '\0';
}
}

tcp_payload_len = pkt_len - iph_len - tcph_len;

if (tcp_payload_len > 0) {
E("%s:%u ===PAYLOAD(?)===> %s:%u", src_ip, ntohs(tcph->source), dst_ip,
ntohs(tcph->dest));
E_INFO("%s:%u ===PAYLOAD(?)===> %s:%u", src_ip, ntohs(tcph->source),
dst_ip, ntohs(tcph->dest));
goto ret_mark_repeat;
} else if (tcph->syn && tcph->ack) {
E("%s:%u ===SYN-ACK===> %s:%u", src_ip, ntohs(tcph->source), dst_ip,
ntohs(tcph->dest));
E_INFO("%s:%u ===SYN-ACK===> %s:%u", src_ip, ntohs(tcph->source),
dst_ip, ntohs(tcph->dest));

ack_new = ntohl(tcph->seq);
ack_new++;
Expand All @@ -642,8 +649,8 @@ static int callback(struct nfq_q_handle *qh, struct nfgenmsg *nfmsg,
goto ret_accept;
}
}
E("%s:%u <===ACK(*)=== %s:%u", src_ip, ntohs(tcph->source), dst_ip,
ntohs(tcph->dest));
E_INFO("%s:%u <===ACK(*)=== %s:%u", src_ip, ntohs(tcph->source),
dst_ip, ntohs(tcph->dest));

for (i = 0; i < g_repeat; i++) {
res = send_http(iph->daddr, iph->saddr, tcph->dest, tcph->source,
Expand All @@ -653,13 +660,13 @@ static int callback(struct nfq_q_handle *qh, struct nfgenmsg *nfmsg,
goto ret_accept;
}
}
E("%s:%u <===HTTP(*)=== %s:%u", src_ip, ntohs(tcph->source), dst_ip,
ntohs(tcph->dest));
E_INFO("%s:%u <===HTTP(*)=== %s:%u", src_ip, ntohs(tcph->source),
dst_ip, ntohs(tcph->dest));

goto ret_mark_repeat;
} else if (tcph->ack) {
E("%s:%u ===ACK===> %s:%u", src_ip, ntohs(tcph->source), dst_ip,
ntohs(tcph->dest));
E_INFO("%s:%u ===ACK===> %s:%u", src_ip, ntohs(tcph->source), dst_ip,
ntohs(tcph->dest));

for (i = 0; i < g_repeat; i++) {
res = send_http(iph->daddr, iph->saddr, tcph->dest, tcph->source,
Expand All @@ -669,13 +676,13 @@ static int callback(struct nfq_q_handle *qh, struct nfgenmsg *nfmsg,
goto ret_accept;
}
}
E("%s:%u <===HTTP(*)=== %s:%u", src_ip, ntohs(tcph->source), dst_ip,
ntohs(tcph->dest));
E_INFO("%s:%u <===HTTP(*)=== %s:%u", src_ip, ntohs(tcph->source),
dst_ip, ntohs(tcph->dest));

goto ret_mark_repeat;
} else {
E("%s:%u ===(?)===> %s:%u", src_ip, ntohs(tcph->source), dst_ip,
ntohs(tcph->dest));
E_INFO("%s:%u ===(?)===> %s:%u", src_ip, ntohs(tcph->source), dst_ip,
ntohs(tcph->dest));
goto ret_accept;
}

Expand Down Expand Up @@ -705,7 +712,7 @@ int main(int argc, char *argv[])
return EXIT_FAILURE;
}

while ((opt = getopt(argc, argv, "h:i:m:n:r:t:w:x:")) != -1) {
while ((opt = getopt(argc, argv, "h:i:m:n:r:st:w:x:")) != -1) {
switch (opt) {
case 'h':
if (strlen(optarg) > _POSIX_HOST_NAME_MAX) {
Expand Down Expand Up @@ -751,6 +758,9 @@ int main(int argc, char *argv[])
}
g_repeat = tmp;
break;
case 's':
g_silent = 1;
break;
case 't':
if (sscanf(optarg, "%llu", &tmp) != 1 || !tmp ||
tmp > UINT8_MAX) {
Expand Down