| -rw-r--r-- | noncore/net/wellenreiter/daemon/source/daemon.cc | 18 |
1 files changed, 11 insertions, 7 deletions
diff --git a/noncore/net/wellenreiter/daemon/source/daemon.cc b/noncore/net/wellenreiter/daemon/source/daemon.cc index d77d987..3a28217 100644 --- a/noncore/net/wellenreiter/daemon/source/daemon.cc +++ b/noncore/net/wellenreiter/daemon/source/daemon.cc @@ -99,55 +99,59 @@ int main(int argc, char **argv) /* maxfd = biggest filefd */ maxfd = (sock > pcap_fileno(handletopcap) ? sock + 1 : pcap_fileno(handletopcap)) + 1; 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) { 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; - case 99: - wl_loginfo("Received STOPSNIFF command"); - break; - default: + case STARTSNIFF: + wl_loginfo("Received STARTSNIFF command"); + if(!send_ok(GUIADDR, GUIPORT, STARTSNIFF)) + wl_logerr("Cannot set OK_CMD to GUI"); + break; + case STOPSNIFF: + wl_loginfo("Received STOPSNIFF command"); + if(!send_ok(GUIADDR, GUIPORT, STOPSNIFF)) + wl_logerr("Cannot set FAIL_CMD to GUI"); + break; + default: wl_logerr("Received unknown command: %d", retval); break; } } } /* FD_ISSET */ /* Check pcap lib for packets */ if(FD_ISSET(pcap_fileno(handletopcap), &rset)) { /* Grab one single packet */ packet = pcap_next(handletopcap, &header); /* process the packet */ process_packets(&header,*&packet, GUIADDR, GUIPORT); } } /* while(1) */ close(sock); exit(EXIT_SUCCESS); } void |