Line data    Source code

       1             : /*
       2             :  * Copyright (c) 2016 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 <urpf/urpf.h>
      17             : #include <vnet/plugin/plugin.h>
      18             : #include <vnet/ip/ip_types_api.h>
      19             : 
      20             : #include <vpp/app/version.h>
      21             : 
      22             : #include <vlibapi/api.h>
      23             : #include <vlibmemory/api.h>
      24             : 
      25             : /* define message IDs */
      26             : #include <vnet/format_fns.h>
      27             : #include <urpf/urpf.api_enum.h>
      28             : #include <urpf/urpf.api_types.h>
      29             : #include <vnet/fib/fib_table.h>
      30             : #include <vnet/ip/ip_types.h>
      31             : 
      32             : /**
      33             :  * Base message ID fot the plugin
      34             :  */
      35             : static u32 urpf_base_msg_id;
      36             : #define REPLY_MSG_ID_BASE urpf_base_msg_id
      37             : 
      38             : #include <vlibapi/api_helper_macros.h>
      39             : 
      40             : static int
      41          12 : urpf_mode_decode (vl_api_urpf_mode_t in, urpf_mode_t * out)
      42             : {
      43             :   if (0)
      44             :     ;
      45             : #define _(a,b)                                  \
      46             :   else if (URPF_API_MODE_##a == in)             \
      47             :     {                                           \
      48             :       *out = URPF_MODE_##a;                     \
      49             :       return (0);                               \
      50             :     }
      51          12 :   foreach_urpf_mode
      52             : #undef _
      53           0 :     return (VNET_API_ERROR_INVALID_VALUE);
      54             : }
      55             : 
      56             : static void
      57          12 : vl_api_urpf_update_t_handler (vl_api_urpf_update_t * mp)
      58             : {
      59             :   vl_api_urpf_update_reply_t *rmp;
      60             :   ip_address_family_t af;
      61             :   urpf_mode_t mode;
      62          12 :   int rv = 0;
      63             : 
      64          12 :   VALIDATE_SW_IF_INDEX (mp);
      65             : 
      66          12 :   rv = urpf_mode_decode (mp->mode, &mode);
      67          12 :   if (rv)
      68           0 :     goto done;
      69             : 
      70          12 :   rv = ip_address_family_decode (mp->af, &af);
      71          12 :   if (rv)
      72           0 :     goto done;
      73             : 
      74          12 :   rv = urpf_update (mode, htonl (mp->sw_if_index), af,
      75          12 :                     (mp->is_input ? VLIB_RX : VLIB_TX), 0);
      76          12 :   if (rv)
      77           0 :     goto done;
      78             : 
      79          12 :   BAD_SW_IF_INDEX_LABEL;
      80          12 : done:
      81          12 :   REPLY_MACRO (VL_API_URPF_UPDATE_REPLY);
      82             : }
      83             : 
      84             : static void
      85           0 : vl_api_urpf_update_v2_t_handler (vl_api_urpf_update_v2_t *mp)
      86             : {
      87             :   vl_api_urpf_update_reply_t *rmp;
      88             :   ip_address_family_t af;
      89             :   urpf_mode_t mode;
      90           0 :   int rv = 0;
      91             : 
      92           0 :   VALIDATE_SW_IF_INDEX (mp);
      93             : 
      94           0 :   rv = urpf_mode_decode (mp->mode, &mode);
      95           0 :   if (rv)
      96           0 :     goto done;
      97             : 
      98           0 :   rv = ip_address_family_decode (mp->af, &af);
      99             : 
     100           0 :   if (rv)
     101           0 :     goto done;
     102             : 
     103           0 :   rv = urpf_update (mode, htonl (mp->sw_if_index), af,
     104           0 :                     (mp->is_input ? VLIB_RX : VLIB_TX), ntohl (mp->table_id));
     105             : 
     106           0 :   if (rv)
     107           0 :     goto done;
     108             : 
     109           0 :   BAD_SW_IF_INDEX_LABEL;
     110           0 : done:
     111           0 :   REPLY_MACRO (VL_API_URPF_UPDATE_V2_REPLY);
     112             : }
     113             : 
     114             : #include <urpf/urpf.api.c>
     115             : 
     116             : static clib_error_t *
     117         559 : urpf_api_init (vlib_main_t * vm)
     118             : {
     119             :   /* Ask for a correctly-sized block of API message decode slots */
     120         559 :   urpf_base_msg_id = setup_message_id_table ();
     121             : 
     122         559 :   return 0;
     123             : }
     124             : 
     125        1119 : VLIB_INIT_FUNCTION (urpf_api_init);
     126             : 
     127             : /* *INDENT-OFF* */
     128             : VLIB_PLUGIN_REGISTER () = {
     129             :     .version = VPP_BUILD_VER,
     130             :     .description = "Unicast Reverse Path Forwarding (uRPF)",
     131             : };
     132             : /* *INDENT-ON* */
     133             : 
     134             : /*
     135             :  * fd.io coding-style-patch-verification: ON
     136             :  *
     137             :  * Local Variables:
     138             :  * eval: (c-set-style "gnu")
     139             :  * End:
     140             :  */