X-Git-Url: http://developer.intra2net.com/git/?p=libipt_ACCOUNT;a=blobdiff_plain;f=iptaccount%2Fiptaccount.c;h=4854efda036b89b4a253a4d840fb6b663182e41c;hp=9de3a8cfbb56da589e9288aeabad53e96a67fd30;hb=e3ccc02f6d76e1c1495370812fd790fd84c7cab1;hpb=3a07e3fb3b03f93c64ca679c2d97dbd948f960a9

diff --git a/iptaccount/iptaccount.c b/iptaccount/iptaccount.c
index 9de3a8c..4854efd 100644
--- a/iptaccount/iptaccount.c
+++ b/iptaccount/iptaccount.c
@@ -14,59 +14,190 @@
 
 #include <stdio.h>
 #include <stdlib.h>
+#include <getopt.h>
+#include <signal.h>
 
 #include <ipt_ACCOUNT_cl.h>
 
+char exit_now = 0;
+static void sig_term(int signr)
+{
+    signal(SIGINT, SIG_IGN);
+    signal(SIGQUIT, SIG_IGN);
+    signal(SIGTERM, SIG_IGN);
+    
+    exit_now=1;
+}
+
 char *addr_to_dotted(unsigned int addr)
 {
-    static char buf[20];
+    static char buf[17];
     const unsigned char *bytep;
 
     bytep = (const unsigned char *) &addr;
-    sprintf(buf, "%u.%u.%u.%u", bytep[0], bytep[1], bytep[2], bytep[3]);
+    snprintf(buf, 16, "%u.%u.%u.%u", bytep[0], bytep[1], bytep[2], bytep[3]);
+    buf[16] = 0;
     return buf;
 }
 
+void show_usage(void)
+{
+    printf ("Unknown command line option. Try: [-u] [-h] [-a] [-f] [-l name]\n");
+    printf("[-u] show kernel handle usage\n");
+    printf("[-h] free all kernel handles (experts only!)\n");
+    printf("[-a] list all table names\n\n");
+    printf("[-l name] show table data\n");
+    printf("[-f] flush data after show\n");
+    printf("[-c] loop every second (abort with CTRL+C)\n");
+    printf("\n");
+}
+
 int main(int argc, char *argv[])
 {
     struct ipt_ACCOUNT_context ctx;
-    struct ipt_account_handle_ip *entry;
-    int i = 0;
+    struct ipt_acc_handle_ip *entry;
+    int i;
+    char optchar, doHandleUsage=0, doHandleFree=0, doTableNames=0,
+                  doFlush=0, doContinue=0;
+                  
+    char *table_name = NULL;
+    
+    printf("\nipt_ACCOUNT userspace accounting tool v%s\n\n", VERSION);
+
+    if (argc == 1)
+    {
+        show_usage();
+        exit(0);
+    }
+        
+    while ((optchar = getopt (argc, argv, "uhacfl:")) != -1)
+    {
+        switch (optchar)
+        {
+            case 'u':
+                doHandleUsage=1;
+                break;
+            case 'h':
+                doHandleFree=1;
+                break;
+            case 'a':
+                doTableNames=1;
+                break;
+            case 'f':
+                doFlush=1;
+                break;
+            case 'c':
+                doContinue=1;
+                break;
+            case 'l':
+                table_name = (char *)strdup(optarg);
+                break;
+            case '?':
+            default:
+                show_usage();
+                exit (0);
+                break;
+        }
+    }
     
-    if (argc != 2)
+    // install exit handler
+    if (signal(SIGTERM, sig_term) == SIG_ERR)
+    {
+        printf("can't install signal handler for SIGTERM\n");
+        exit (-1);
+    }
+    if (signal(SIGINT, sig_term) == SIG_ERR)
+    {
+        printf("can't install signal handler for SIGINT\n");
+        exit (-1);
+    }
+    if (signal(SIGQUIT, sig_term) == SIG_ERR)
     {
-        printf("Syntax: %s TABLE-NAME\n", argv[0]);
+        printf("can't install signal handler for SIGQUIT\n");
         exit (-1);
     }
     
+    
     if(ipt_ACCOUNT_init(&ctx))
     {
         printf("Init failed: %s\n", ctx.error_str);
         exit (-1);
     }
 
-    while (1)
+    // Get handle usage?
+    if (doHandleUsage)
     {
-        printf("Run #%d\n", i);
+        int rtn = ipt_ACCOUNT_get_handle_usage(&ctx);
+        if (rtn < 0)
+        {
+            printf("get_handle_usage failed: %s\n", ctx.error_str);
+            exit (-1);
+        }
         
-        // Get entries from table test
-        if (ipt_ACCOUNT_read_entries(&ctx, argv[1], 0))
+        printf("Current kernel handle usage: %d\n", ctx.handle.itemcount);
+    }
+    
+    if (doHandleFree)
+    {
+        int rtn = ipt_ACCOUNT_free_all_handles(&ctx);
+        if (rtn < 0)
         {
-            printf("Read failed: %s\n", ctx.error_str);
-            ipt_ACCOUNT_deinit(&ctx);
+            printf("handle_free_all failed: %s\n", ctx.error_str);
             exit (-1);
         }
         
-        // Output and free entries
-        while ((entry = ipt_ACCOUNT_get_next_entry(&ctx)) != NULL)
+        printf("Freed all handles in kernel space\n");
+    }
+
+    if (doTableNames)
+    {
+        int rtn = ipt_ACCOUNT_get_table_names(&ctx);
+        if (rtn < 0)
         {
-            printf("IP: %s SRC packets: %u bytes: %u DST packets: %u bytes: %u\n",
-                   addr_to_dotted(entry->ip), entry->src_packets, entry->src_bytes, entry->dst_packets, entry->dst_bytes);
+            printf("get_table_names failed: %s\n", ctx.error_str);
+            exit (-1);
         }
-        sleep(1);
-	i++;
+        const char *name;
+        while ((name = ipt_ACCOUNT_get_next_name(&ctx)) != 0)
+            printf("Found table: %s\n", name);
     }
+        
+    if (table_name)
+    {
+        // Read out data
+        printf("Showing table: %s\n", table_name);
+        i = 0;
+        while (!exit_now)
+        {
+            // Get entries from table test
+            if (ipt_ACCOUNT_read_entries(&ctx, table_name, !doFlush))
+            {
+                printf("Read failed: %s\n", ctx.error_str);
+                ipt_ACCOUNT_deinit(&ctx);
+                exit (-1);
+            }
+            
+            printf("Run #%d - %u %s found\n", i, ctx.handle.itemcount,
+                   ctx.handle.itemcount == 1 ? "item" : "items");
+            
+            // Output and free entries
+            while ((entry = ipt_ACCOUNT_get_next_entry(&ctx)) != NULL)
+            {
+                printf("IP: %s SRC packets: %u bytes: %u DST packets: %u bytes: %u\n",
+                    addr_to_dotted(entry->ip), entry->src_packets, entry->src_bytes,
+                                   entry->dst_packets, entry->dst_bytes);
+            }
             
+            if (doContinue)
+            {
+                sleep(1);
+                i++;
+            } else
+                exit_now = 1;
+        }
+    }            
+        
+    printf("Finished.\n");
     ipt_ACCOUNT_deinit(&ctx);
     exit (0);
 }