src/plugins/igmp/igmp_src.c

   1 /*
   2  *------------------------------------------------------------------
   3  * Copyright (c) 2018 Cisco and/or its affiliates.
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at:
   7  *
   8  *     http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  *------------------------------------------------------------------
  16  */
  17
  18 #include <igmp/igmp_src.h>
  19 #include <igmp/igmp_group.h>
  20 #include <igmp/igmp.h>
  21
  22 void
  23 igmp_src_free (igmp_src_t * src)
  24 {
  25   igmp_main_t *im = &igmp_main;
  26
  27   IGMP_DBG ("free-src: (%U)", format_igmp_key, src->key);
  28
  29   igmp_timer_retire (&src->timers[IGMP_SRC_TIMER_EXP]);
  30
  31   clib_mem_free (src->key);
  32   pool_put (im->srcs, src);
  33 }
  34
  35 static void
  36 igmp_src_exp (u32 obj, void *dat)
  37 {
  38   igmp_group_t *group;
  39   igmp_src_t *src;
  40
  41   src = pool_elt_at_index (igmp_main.srcs, obj);
  42   group = igmp_group_get (src->group);
  43
  44   IGMP_DBG ("src-exp: %U", format_igmp_key, src->key);
  45
  46   igmp_timer_retire (&src->timers[IGMP_SRC_TIMER_EXP]);
  47
  48   if (IGMP_MODE_ROUTER == src->mode)
  49     {
  50       igmp_config_t *config;
  51       igmp_group_t *group;
  52
  53       /*
  54        * inform interest parties
  55        */
  56       group = igmp_group_get (src->group);
  57       config = igmp_config_get (group->config);
  58
  59       igmp_event (IGMP_FILTER_MODE_EXCLUDE,
  60                   config->sw_if_index, src->key, group->key);
  61     }
  62
  63   igmp_group_src_remove (group, src);
  64   igmp_src_free (src);
  65
  66   if (0 == igmp_group_n_srcs (group, IGMP_FILTER_MODE_INCLUDE))
  67     igmp_group_clear (group);
  68 }
  69
  70 igmp_src_t *
  71 igmp_src_alloc (u32 group_index, const igmp_key_t * skey, igmp_mode_t mode)
  72 {
  73   igmp_main_t *im = &igmp_main;
  74   igmp_src_t *src;
  75
  76   IGMP_DBG ("new-src: (%U)", format_igmp_key, skey);
  77
  78   pool_get (im->srcs, src);
  79   memset (src, 0, sizeof (igmp_src_t));
  80   src->mode = mode;
  81   src->key = clib_mem_alloc (sizeof (*skey));
  82   src->group = group_index;
  83   clib_memcpy (src->key, skey, sizeof (*skey));
  84
  85   if (IGMP_MODE_ROUTER == mode)
  86     {
  87       igmp_config_t *config;
  88       igmp_group_t *group;
  89       /*
  90        * start a timer that determines whether the source is still
  91        * active on the link
  92        */
  93       src->timers[IGMP_SRC_TIMER_EXP] =
  94         igmp_timer_schedule (igmp_timer_type_get (IGMP_TIMER_SRC),
  95                              src - im->srcs, igmp_src_exp, NULL);
  96
  97       /*
  98        * inform interest parties
  99        */
 100       group = igmp_group_get (src->group);
 101       config = igmp_config_get (group->config);
 102
 103       igmp_event (IGMP_FILTER_MODE_INCLUDE,
 104                   config->sw_if_index, src->key, group->key);
 105     }
 106   else
 107     {
 108       src->timers[IGMP_SRC_TIMER_EXP] = IGMP_TIMER_ID_INVALID;
 109     }
 110
 111   return (src);
 112 }
 113
 114 void
 115 igmp_src_refresh (igmp_src_t * src)
 116 {
 117   IGMP_DBG ("refresh-src: (%U)", format_igmp_key, src->key);
 118
 119   igmp_timer_retire (&src->timers[IGMP_SRC_TIMER_EXP]);
 120
 121   src->timers[IGMP_SRC_TIMER_EXP] =
 122     igmp_timer_schedule (igmp_timer_type_get (IGMP_TIMER_SRC),
 123                          igmp_src_index (src), igmp_src_exp, NULL);
 124 }
 125
 126 void
 127 igmp_src_blocked (igmp_src_t * src)
 128 {
 129   IGMP_DBG ("block-src: (%U)", format_igmp_key, src->key);
 130
 131   igmp_timer_retire (&src->timers[IGMP_SRC_TIMER_EXP]);
 132
 133   src->timers[IGMP_SRC_TIMER_EXP] =
 134     igmp_timer_schedule (igmp_timer_type_get (IGMP_TIMER_LEAVE),
 135                          igmp_src_index (src), igmp_src_exp, NULL);
 136 }
 137
 138 u32
 139 igmp_src_index (igmp_src_t * src)
 140 {
 141   return (src - igmp_main.srcs);
 142 }
 143
 144
 145 /*
 146  * fd.io coding-style-patch-verification: ON
 147  *
 148  * Local Variables:
 149  * eval: (c-set-style "gnu")
 150  * End:
 151  */