-rw-r--r-- | noncore/net/wellenreiter/daemon/source/daemon.cc | 107 | ||||
-rw-r--r-- | noncore/net/wellenreiter/daemon/source/daemon.hh | 3 |
2 files changed, 68 insertions, 42 deletions
diff --git a/noncore/net/wellenreiter/daemon/source/daemon.cc b/noncore/net/wellenreiter/daemon/source/daemon.cc index b57e6a0..4407436 100644 --- a/noncore/net/wellenreiter/daemon/source/daemon.cc +++ b/noncore/net/wellenreiter/daemon/source/daemon.cc @@ -1,128 +1,123 @@ /* * Startup functions of wellenreiter * * $Id$ */ #include "config.hh" #include "daemon.hh" -// temporary solution, will be removed soon +/* should be parsed from cfg-file */ #define MAXCHANNEL 13 -int card_type; -char sniffer_device[6]; -int channel=0; -int timedout=1; - -void chanswitch(int blah) -{ - if(channel >= MAXCHANNEL) - { - channel=1; - } - else - { - channel++; - } - card_set_channel(sniffer_device, channel, card_type); - timedout=0; - alarm(1); -} +#define CHANINTERVAL 100000 /* Main function of wellenreiterd */ int main(int argc, char **argv) { int sock, maxfd, retval; char buffer[WL_SOCKBUF]; struct pcap_pkthdr header; struct sockaddr_in saddr; pcap_t *handletopcap; + wl_cardtype_t cardtype; + pthread_t sub; const unsigned char *packet; fd_set rset; fprintf(stderr, "wellenreiterd %s\n\n", VERSION); fprintf(stderr, "(c) 2002 by M-M-M\n\n"); if(argc < 3) usage(); - // removed soon, see above - signal(SIGALRM, chanswitch); - alarm(1); /* Set sniffer device */ - memset(sniffer_device, 0, sizeof(sniffer_device)); - strncpy(sniffer_device, (char *)argv[1], sizeof(sniffer_device) - 1); + memset(cardtype.iface, 0, sizeof(cardtype.iface)); + strncpy(cardtype.iface, (char *)argv[1], sizeof(cardtype.iface) - 1); /* Set card type */ - card_type = atoi(argv[2]); - if(card_type < 1 || card_type > 4) + cardtype.type = atoi(argv[2]); + if(cardtype.type < 1 || cardtype.type > 4) usage(); - if(!card_into_monitormode(&handletopcap, sniffer_device, card_type)) + if(!card_into_monitormode(&handletopcap, cardtype.iface, cardtype.type)) { wl_logerr("Cannot initialize the wireless-card, aborting"); - exit(-1); + exit(EXIT_FAILURE); } wl_loginfo("Set card into monitor mode"); + /////// following line will be moved to lib as soon as possible //////////// - if((handletopcap = pcap_open_live(sniffer_device, BUFSIZ, 1, 0, NULL)) == NULL) + if((handletopcap = pcap_open_live(cardtype.iface, BUFSIZ, 1, 0, NULL)) == NULL) { wl_logerr("pcap_open_live() failed: %s", strerror(errno)); - exit(-1); + exit(EXIT_FAILURE); } #ifdef HAVE_PCAP_NONBLOCK pcap_setnonblock(handletopcap, 1, NULL); #endif - //////////////////////////////////////// + /* Setup socket for incoming commands */ if((sock=wl_setupsock(DAEMONADDR, DAEMONPORT, saddr)) < 0) { wl_logerr("Cannot setup socket"); - exit(-1); + exit(EXIT_FAILURE); } wl_loginfo("Set up socket '%d' for GUI communication", sock); + /* Create channelswitching thread */ + if(pthread_create(&sub, NULL, channel_switcher, (void *)&cardtype) != 0) + { + wl_logerr("Cannot create thread: %s", strerror(errno)); + close(sock); + exit(EXIT_FAILURE); + } + if(pthread_detach(sub)) + { + wl_logerr("Error detaching thread"); + close(sock); + pthread_exit((pthread_t *)sub); + exit(EXIT_FAILURE); + } + wl_loginfo("Created and detached channel switching thread"); + FD_ZERO(&rset); /* Start main loop */ wl_loginfo("Starting main loop"); while(1) { FD_SET(sock, &rset); FD_SET(pcap_fileno(handletopcap), &rset); - // blah - timedout=1; - /* socket or pcap handle bigger? Will be cleaned up, have to check pcap */ - maxfd = (sock > pcap_fileno(handletopcap) ? sock : pcap_fileno(handletopcap)) + 1; + maxfd = (sock > pcap_fileno(handletopcap) ? sock + 1: pcap_fileno(handletopcap)) + 1; - if(select(maxfd, &rset, NULL, NULL, NULL) < 0 && timedout) + if(select(maxfd, &rset, NULL, NULL, NULL) < 0) { wl_logerr("Error calling select: %s", strerror(errno)); break; } /* Got data on local socket from GUI */ if(FD_ISSET(sock, &rset)) { /* Receive data from socket */ - if((retval=wl_recv(&sock, saddr, buffer, sizeof(buffer))) < 0 && timedout) + if((retval=wl_recv(&sock, saddr, buffer, sizeof(buffer))) < 0) { wl_logerr("Error trying to read: %s", strerror(errno)); break; } else { /* check type of packet and start function according to it */ switch(retval) { case 98: wl_loginfo("Received STARTSNIFF command"); break; @@ -141,25 +136,55 @@ int main(int argc, char **argv) { /* Grab one single packet */ packet = pcap_next(handletopcap, &header); /* process the packet */ process_packets(&header,*&packet, GUIADDR, GUIPORT); } } /* while(1) */ close(sock); - exit(0); + exit(EXIT_SUCCESS); } void usage(void) { fprintf(stderr, "Usage: wellenreiter <device> <cardtype>\n" \ "\t<device> = Wirelessdevice (e.g. wlan0)\n" \ "\t<cardtype> = Cardtype:\tCisco\t= 1\n" \ "\t\t\t\tNG\t= 2\n" \ "\t\t\t\tHOSTAP\t= 3\n" \ "\t\t\t\tLUCENT\t= 4\n"); - exit(-1); + exit(EXIT_FAILURE); +} + +void * +channel_switcher(void *cardtypeptr) +{ + wl_cardtype_t *cardtype; + int channel=1; + + /* Get card info struct */ + cardtype = (wl_cardtype_t *)cardtypeptr; + + while(1) + { + + /* If channel bigger than maxchannel, set to 1 */ + if(channel > MAXCHANNEL) + channel=1; + + /* Set channel */ + if(!card_set_channel(cardtype->iface, channel, cardtype->type)) + { + wl_logerr("Cannot set channel, thread exiting"); + pthread_exit(NULL); + } + + /* sleep */ + usleep(CHANINTERVAL); + + channel++; + } /* while */ } diff --git a/noncore/net/wellenreiter/daemon/source/daemon.hh b/noncore/net/wellenreiter/daemon/source/daemon.hh index d50487a..f9ac45e 100644 --- a/noncore/net/wellenreiter/daemon/source/daemon.hh +++ b/noncore/net/wellenreiter/daemon/source/daemon.hh @@ -1,26 +1,27 @@ /* $Id$ */ #ifndef DAEMON_HH #define DAEMON_HH #include <stdio.h> #include <string.h> #include <sys/types.h> #include <sys/time.h> #include <sys/socket.h> #include <netinet/in.h> #include <arpa/inet.h> +#include <pthread.h> #include <unistd.h> #include <errno.h> -#include <signal.h> #include <libwellenreiter/source/wl_sock.hh> #include <libwellenreiter/source/wl_log.hh> #include <libwellenreiter/source/wl_types.hh> #include <libwellenreiter/source/wl_proto.hh> #include <libwellenreiter/source/cardmode.hh> #include <libwellenreiter/source/sniff.hh> void usage(void); +void *channel_switcher(void *); #endif /* DAEMON_HH */ |