src/vnet/lawful-intercept/lawful_intercept.c

   1 /*
   2  * Copyright (c) 2015 Cisco and/or its affiliates.
   3  * Licensed under the Apache License, Version 2.0 (the "License");
   4  * you may not use this file except in compliance with the License.
   5  * You may obtain a copy of the License at:
   6  *
   7  *     http://www.apache.org/licenses/LICENSE-2.0
   8  *
   9  * Unless required by applicable law or agreed to in writing, software
  10  * distributed under the License is distributed on an "AS IS" BASIS,
  11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12  * See the License for the specific language governing permissions and
  13  * limitations under the License.
  14  */
  15
  16 #include <vnet/lawful-intercept/lawful_intercept.h>
  17
  18 li_main_t li_main;
  19
  20 static clib_error_t *
  21 set_li_command_fn (vlib_main_t * vm,
  22                  unformat_input_t * input,
  23                  vlib_cli_command_t * cmd)
  24 {
  25   li_main_t * lm = &li_main;
  26   ip4_address_t collector;
  27   u8 collector_set = 0;
  28   ip4_address_t src;
  29   u8 src_set = 0;
  30   u32 tmp;
  31   u16 udp_port = 0;
  32   u8 is_add = 1;
  33   int i;
  34
  35   while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT) {
  36     if (unformat (input, "collector %U", unformat_ip4_address, &collector))
  37       collector_set = 1;
  38     if (unformat (input, "src %U", unformat_ip4_address, &src))
  39       src_set = 1;
  40     else if (unformat (input, "udp-port %d", &tmp))
  41       udp_port = tmp;
  42     else if (unformat (input, "del"))
  43       is_add = 0;
  44     else
  45       break;
  46   }
  47
  48   if (collector_set == 0)
  49     return clib_error_return (0, "collector must be set...");
  50   if (src_set == 0)
  51     return clib_error_return (0, "src must be set...");
  52   if (udp_port == 0)
  53     return clib_error_return (0, "udp-port must be set...");
  54
  55   if (is_add == 1)
  56     {
  57       for (i = 0; i < vec_len (lm->collectors); i++)
  58         {
  59           if (lm->collectors[i].as_u32 == collector.as_u32)
  60             {
  61               if (lm->ports[i] == udp_port)
  62                 return clib_error_return
  63                   (0, "collector %U:%d already configured",
  64                    &collector, udp_port);
  65               else
  66                 return clib_error_return
  67                   (0, "collector %U already configured with port %d",
  68                    &collector, (int)(lm->ports[i]));
  69             }
  70         }
  71       vec_add1 (lm->collectors, collector);
  72       vec_add1 (lm->ports, udp_port);
  73       vec_add1 (lm->src_addrs, src);
  74       return 0;
  75     }
  76   else
  77     {
  78       for (i = 0; i < vec_len (lm->collectors); i++)
  79         {
  80           if ((lm->collectors[i].as_u32 == collector.as_u32)
  81               && lm->ports[i] == udp_port)
  82             {
  83               vec_delete (lm->collectors, 1, i);
  84               vec_delete (lm->ports, 1, i);
  85               vec_delete (lm->src_addrs, 1, i);
  86               return 0;
  87             }
  88         }
  89       return clib_error_return (0, "collector %U:%d not configured",
  90                                 &collector, udp_port);
  91     }
  92   return 0;
  93 }
  94
  95 VLIB_CLI_COMMAND (set_li_command, static) = {
  96     .path = "set li",
  97     .short_help =
  98     "set li src <ip4-address> collector <ip4-address> udp-port <nnnn>",
  99     .function = set_li_command_fn,
 100 };
 101
 102 static clib_error_t *
 103 li_init (vlib_main_t * vm)
 104 {
 105   li_main_t * lm = &li_main;
 106
 107   lm->vlib_main = vm;
 108   lm->vnet_main = vnet_get_main();
 109   lm->hit_node_index = li_hit_node.index;
 110   return 0;
 111 }
 112
 113 VLIB_INIT_FUNCTION(li_init);
 114