Line data    Source code

       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         447 : igmp_src_free (igmp_src_t * src)
      24             : {
      25         447 :   igmp_main_t *im = &igmp_main;
      26             : 
      27         447 :   IGMP_DBG ("free-src: (%U)", format_igmp_key, src->key);
      28             : 
      29         447 :   igmp_timer_retire (&src->timers[IGMP_SRC_TIMER_EXP]);
      30             : 
      31         447 :   clib_mem_free (src->key);
      32         447 :   pool_put (im->srcs, src);
      33         447 : }
      34             : 
      35             : static void
      36           9 : igmp_src_exp (u32 obj, void *dat)
      37             : {
      38             :   igmp_group_t *group;
      39             :   igmp_src_t *src;
      40             : 
      41           9 :   src = pool_elt_at_index (igmp_main.srcs, obj);
      42           9 :   group = igmp_group_get (src->group);
      43             : 
      44           9 :   IGMP_DBG ("src-exp: %U", format_igmp_key, src->key);
      45             : 
      46           9 :   igmp_timer_retire (&src->timers[IGMP_SRC_TIMER_EXP]);
      47             : 
      48           9 :   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           9 :       group = igmp_group_get (src->group);
      57           9 :       config = igmp_config_get (group->config);
      58             : 
      59           9 :       igmp_event (IGMP_FILTER_MODE_EXCLUDE,
      60           9 :                   config->sw_if_index, src->key, group->key);
      61             : 
      62           9 :       igmp_proxy_device_block_src (config, group, src);
      63             :     }
      64             : 
      65           9 :   igmp_group_src_remove (group, src);
      66           9 :   igmp_src_free (src);
      67             : 
      68           9 :   if (0 == igmp_group_n_srcs (group, IGMP_FILTER_MODE_INCLUDE))
      69           5 :     igmp_group_clear (&group);
      70           9 : }
      71             : 
      72             : igmp_src_t *
      73         447 : igmp_src_alloc (u32 group_index, const igmp_key_t * skey, igmp_mode_t mode)
      74             : {
      75         447 :   igmp_main_t *im = &igmp_main;
      76             :   igmp_src_t *src;
      77             : 
      78         447 :   IGMP_DBG ("new-src: (%U)", format_igmp_key, skey);
      79             : 
      80         447 :   pool_get (im->srcs, src);
      81         447 :   clib_memset (src, 0, sizeof (igmp_src_t));
      82         447 :   src->mode = mode;
      83         447 :   src->key = clib_mem_alloc (sizeof (*skey));
      84         447 :   src->group = group_index;
      85         447 :   clib_memcpy (src->key, skey, sizeof (*skey));
      86             : 
      87         447 :   if (IGMP_MODE_ROUTER == mode)
      88             :     {
      89             :       igmp_config_t *config;
      90             :       igmp_group_t *group;
      91             :       /*
      92             :        * start a timer that determines whether the source is still
      93             :        * active on the link
      94             :        */
      95          24 :       src->timers[IGMP_SRC_TIMER_EXP] =
      96          12 :         igmp_timer_schedule (igmp_timer_type_get (IGMP_TIMER_SRC),
      97          12 :                              src - im->srcs, igmp_src_exp, NULL);
      98             : 
      99             :       /*
     100             :        * inform interest parties
     101             :        */
     102          12 :       group = igmp_group_get (src->group);
     103          12 :       config = igmp_config_get (group->config);
     104             : 
     105          12 :       igmp_event (IGMP_FILTER_MODE_INCLUDE,
     106          12 :                   config->sw_if_index, src->key, group->key);
     107             :     }
     108             :   else
     109             :     {
     110         435 :       src->timers[IGMP_SRC_TIMER_EXP] = IGMP_TIMER_ID_INVALID;
     111             :     }
     112             : 
     113         447 :   return (src);
     114             : }
     115             : 
     116             : void
     117           2 : igmp_src_refresh (igmp_src_t * src)
     118             : {
     119           2 :   IGMP_DBG ("refresh-src: (%U)", format_igmp_key, src->key);
     120             : 
     121           2 :   igmp_timer_retire (&src->timers[IGMP_SRC_TIMER_EXP]);
     122             : 
     123           2 :   src->timers[IGMP_SRC_TIMER_EXP] =
     124           2 :     igmp_timer_schedule (igmp_timer_type_get (IGMP_TIMER_SRC),
     125             :                          igmp_src_index (src), igmp_src_exp, NULL);
     126           2 : }
     127             : 
     128             : void
     129           5 : igmp_src_blocked (igmp_src_t * src)
     130             : {
     131           5 :   IGMP_DBG ("block-src: (%U)", format_igmp_key, src->key);
     132             : 
     133           5 :   igmp_timer_retire (&src->timers[IGMP_SRC_TIMER_EXP]);
     134             : 
     135           5 :   src->timers[IGMP_SRC_TIMER_EXP] =
     136           5 :     igmp_timer_schedule (igmp_timer_type_get (IGMP_TIMER_LEAVE),
     137             :                          igmp_src_index (src), igmp_src_exp, NULL);
     138           5 : }
     139             : 
     140             : u32
     141         454 : igmp_src_index (igmp_src_t * src)
     142             : {
     143         454 :   return (src - igmp_main.srcs);
     144             : }
     145             : 
     146             : u8 *
     147           0 : format_igmp_src (u8 * s, va_list * args)
     148             : {
     149           0 :   igmp_src_t *src = va_arg (*args, igmp_src_t *);
     150           0 :   u32 indent = va_arg (*args, u32);
     151             : 
     152           0 :   s = format (s, "%U%U %U",
     153             :               format_white_space, indent,
     154             :               format_igmp_key, src->key,
     155             :               format_igmp_timer_id, src->timers[IGMP_SRC_TIMER_EXP]);
     156             : 
     157           0 :   return (s);
     158             : }
     159             : 
     160             : /*
     161             :  * fd.io coding-style-patch-verification: ON
     162             :  *
     163             :  * Local Variables:
     164             :  * eval: (c-set-style "gnu")
     165             :  * End:
     166             :  */