opt/pfpacket-pps.c

   1 /* Compile with -lrt */
   2
   3 #include <stdio.h>
   4 #include <stdint.h>
   5 #include <unistd.h>
   6 #include <signal.h>
   7 #include <stdlib.h>
   8 #include <string.h>
   9 #include <ctype.h>
  10 #include <assert.h>
  11 #include <sys/socket.h>
  12 #include <sys/time.h>
  13 #include <time.h>
  14 #include <sys/socket.h>
  15 #include <linux/if_packet.h>
  16 #include <linux/if_ether.h>
  17 #include <linux/if_arp.h>
  18 #include <arpa/inet.h>
  19
  20 static sig_atomic_t sigint = 0;
  21
  22 static inline double timespec_to_double(const struct timespec *time)
  23 {
  24         return time->tv_nsec * 1E-9 + (double) time->tv_sec;
  25 }
  26
  27 static void sig_handler(int sig)
  28 {
  29         if (sig == SIGINT)
  30                 sigint = 1;
  31 }
  32
  33 int main(void)
  34 {
  35         int sock, ret;
  36         char buff[1600];
  37         struct timespec before, after;
  38         unsigned long long pkts = 0, byte = 0;
  39         double x1, x2, elapsed;
  40
  41         if (geteuid() != 0) {
  42                 fprintf(stderr, "Not root?!\n");
  43                 exit(EXIT_FAILURE);
  44         }
  45
  46         signal(SIGINT, sig_handler);
  47
  48         sock = socket(AF_PACKET, SOCK_RAW, htons(ETH_P_ALL));
  49         if (sock < 0) {
  50                 perror("socket");
  51                 return 0;
  52         }
  53
  54         printf("Hit key to start!\n");
  55         getchar();
  56         printf("Abort with ^C\n");
  57
  58         memset(&before, 0, sizeof(before));
  59         memset(&after, 0, sizeof(after));
  60
  61         clock_gettime(CLOCK_REALTIME, &before);
  62         while (!sigint) {
  63                 ret = recv(sock, buff, sizeof(buff), 0);
  64                 if (ret < 0) {
  65                         perror("recvmsg");
  66                         continue;
  67                 } else {
  68                         pkts++;
  69                         byte += ret;
  70                 }
  71         }
  72         clock_gettime(CLOCK_REALTIME, &after);
  73
  74         x1 = timespec_to_double(&after);
  75         x2 = timespec_to_double(&before);
  76         elapsed = x1 - x2;
  77
  78         printf("\n\n");
  79         fflush(stdout);
  80         printf("time: %lf, pkts/s: %.2lf, bytes/s: %.2lf\n",
  81                elapsed, 1.0 * pkts / elapsed, 1.0 * byte / elapsed);
  82
  83         close(sock);
  84         return 0;
  85 }
  86