snoopctl.c

   1 #include <string.h>
   2 #include "l2tpns.h"
   3 #include "plugin.h"
   4 #include "control.h"
   5
   6 /* snoop control */
   7
   8 char const *cvs_id = "$Id: snoopctl.c,v 1.4 2004/12/16 08:49:53 bodea Exp $";
   9
  10 int plugin_api_version = PLUGIN_API_VERSION;
  11 static struct pluginfuncs *p = 0;
  12
  13 char *plugin_control_help[] = {
  14         "  snoop USER|SID IP PORT                      Intercept user traffic",
  15         "  unsnoop USER|SID                            Stop intercepting user",
  16         0
  17 };
  18
  19 int plugin_init(struct pluginfuncs *funcs)
  20 {
  21         if (!funcs)
  22                 return 0;
  23
  24         p = funcs;
  25         return 1;
  26 }
  27
  28 int plugin_control(struct param_control *data)
  29 {
  30         sessionidt session;
  31         sessiont *s = 0;
  32         int flag;
  33         char *end;
  34
  35         if (data->argc < 1)
  36                 return PLUGIN_RET_OK;
  37
  38         if (strcmp(data->argv[0], "snoop") && strcmp(data->argv[0], "unsnoop"))
  39                 return PLUGIN_RET_OK; // not for us
  40
  41         if (!data->iam_master)
  42                 return PLUGIN_RET_NOTMASTER;
  43
  44         flag = data->argv[0][0] != 'u';
  45
  46         if (flag)
  47         {
  48                 if (data->argc != 4)
  49                 {
  50                         data->response = NSCTL_RES_ERR;
  51                         data->additional = "requires username or session id and host, port";
  52                         return PLUGIN_RET_STOP;
  53                 }
  54         }
  55         else
  56         {
  57                 if (data->argc != 2)
  58                 {
  59                         data->response = NSCTL_RES_ERR;
  60                         data->additional = "requires username or session id";
  61                         return PLUGIN_RET_STOP;
  62                 }
  63         }
  64
  65         if (!(session = strtol(data->argv[1], &end, 10)) || *end)
  66                 session = p->get_session_by_username(data->argv[1]);
  67
  68         if (session)
  69                 s = p->get_session_by_id(session);
  70
  71         if (!s || !s->ip)
  72         {
  73                 data->response = NSCTL_RES_ERR;
  74                 data->additional = "session not found";
  75                 return PLUGIN_RET_STOP;
  76         }
  77
  78         if (flag)
  79         {
  80                 in_addr_t ip = inet_addr(data->argv[2]);
  81                 uint16_t port = atoi(data->argv[3]);
  82
  83                 if (!ip || ip == INADDR_NONE)
  84                 {
  85                         data->response = NSCTL_RES_ERR;
  86                         data->additional = "invalid ip address";
  87                         return PLUGIN_RET_STOP;
  88                 }
  89
  90                 if (!port)
  91                 {
  92                         data->response = NSCTL_RES_ERR;
  93                         data->additional = "invalid port";
  94                         return PLUGIN_RET_STOP;
  95                 }
  96
  97                 if (ip == s->snoop_ip && port == s->snoop_port)
  98                 {
  99                         data->response = NSCTL_RES_ERR;
 100                         data->additional = "already intercepted";
 101                         return PLUGIN_RET_STOP;
 102                 }
 103
 104                 s->snoop_ip = ip;
 105                 s->snoop_port = port;
 106         }
 107         else
 108         {
 109                 if (!s->snoop_ip)
 110                 {
 111                         data->response = NSCTL_RES_ERR;
 112                         data->additional = "not intercepted";
 113                         return PLUGIN_RET_STOP;
 114                 }
 115
 116                 s->snoop_ip = 0;
 117                 s->snoop_port = 0;
 118         }
 119
 120         p->session_changed(session);
 121
 122         data->response = NSCTL_RES_OK;
 123         data->additional = 0;
 124
 125         return PLUGIN_RET_STOP;
 126 }