src/vnet/udp/udp_api.c

   1 /*
   2  * Copyright (c) 2018-2019 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/vnet.h>
  17 #include <vlibmemory/api.h>
  18
  19 #include <vnet/udp/udp_local.h>
  20 #include <vnet/udp/udp_encap.h>
  21 #include <vnet/fib/fib_table.h>
  22 #include <vnet/ip/ip_types_api.h>
  23 #include <vnet/udp/udp.h>
  24
  25 #include <vnet/format_fns.h>
  26 #include <vnet/udp/udp.api_enum.h>
  27 #include <vnet/udp/udp.api_types.h>
  28
  29 #define REPLY_MSG_ID_BASE udp_main.msg_id_base
  30 #include <vlibapi/api_helper_macros.h>
  31
  32 static void
  33 send_udp_encap_details (const udp_encap_t * ue, vl_api_registration_t * reg,
  34                         u32 context)
  35 {
  36   vl_api_udp_encap_details_t *mp;
  37
  38   mp = vl_msg_api_alloc (sizeof (*mp));
  39   clib_memset (mp, 0, sizeof (*mp));
  40   mp->_vl_msg_id = ntohs (REPLY_MSG_ID_BASE + VL_API_UDP_ENCAP_DETAILS);
  41   mp->context = context;
  42
  43   if (FIB_PROTOCOL_IP4 == ue->ue_ip_proto)
  44     {
  45       clib_memcpy (&mp->udp_encap.src_ip.un.ip4,
  46                    &ue->ue_hdrs.ip4.ue_ip4.src_address, 4);
  47       clib_memcpy (&mp->udp_encap.dst_ip.un.ip4,
  48                    &ue->ue_hdrs.ip4.ue_ip4.dst_address, 4);
  49       mp->udp_encap.dst_ip.af = ip_address_family_encode (AF_IP4);
  50       mp->udp_encap.src_ip.af = ip_address_family_encode (AF_IP4);
  51
  52       /* ports aren't byte swapped because they are stored in network
  53        * byte order */
  54       mp->udp_encap.src_port = ue->ue_hdrs.ip4.ue_udp.src_port;
  55       mp->udp_encap.dst_port = ue->ue_hdrs.ip4.ue_udp.dst_port;
  56     }
  57   else
  58     {
  59       clib_memcpy (&mp->udp_encap.src_ip.un.ip6,
  60                    &ue->ue_hdrs.ip6.ue_ip6.src_address, 16);
  61       clib_memcpy (&mp->udp_encap.dst_ip.un.ip6,
  62                    &ue->ue_hdrs.ip6.ue_ip6.dst_address, 16);
  63       mp->udp_encap.dst_ip.af = ip_address_family_encode (AF_IP6);
  64       mp->udp_encap.src_ip.af = ip_address_family_encode (AF_IP6);
  65
  66       /* ports aren't byte swapped because they are stored in network
  67        * byte order */
  68       mp->udp_encap.src_port = ue->ue_hdrs.ip6.ue_udp.src_port;
  69       mp->udp_encap.dst_port = ue->ue_hdrs.ip6.ue_udp.dst_port;
  70     }
  71
  72   mp->udp_encap.table_id =
  73     htonl (fib_table_get_table_id (ue->ue_fib_index, ue->ue_ip_proto));
  74   mp->udp_encap.id = htonl (ue - udp_encap_pool);
  75
  76   vl_api_send_msg (reg, (u8 *) mp);
  77 }
  78
  79 static void
  80 vl_api_udp_encap_dump_t_handler (vl_api_udp_encap_dump_t *mp)
  81 {
  82   vl_api_registration_t *reg;
  83   udp_encap_t *ue;
  84
  85   reg = vl_api_client_index_to_registration (mp->client_index);
  86   if (!reg)
  87     return;
  88
  89   pool_foreach (ue, udp_encap_pool)
  90    {
  91     send_udp_encap_details(ue, reg, mp->context);
  92   }
  93 }
  94
  95 static void
  96 vl_api_udp_encap_add_t_handler (vl_api_udp_encap_add_t *mp)
  97 {
  98   vl_api_udp_encap_add_reply_t *rmp;
  99   ip46_address_t src_ip, dst_ip;
 100   udp_encap_fixup_flags_t flags;
 101   u32 fib_index, table_id;
 102   fib_protocol_t fproto;
 103   ip46_type_t itype;
 104   index_t uei;
 105   int rv = 0;
 106
 107   uei = INDEX_INVALID;
 108   table_id = ntohl (mp->udp_encap.table_id);
 109
 110   itype = ip_address_decode (&mp->udp_encap.src_ip, &src_ip);
 111   itype = ip_address_decode (&mp->udp_encap.dst_ip, &dst_ip);
 112   fproto = fib_proto_from_ip46 (itype);
 113   fib_index = fib_table_find (fproto, table_id);
 114
 115   if (~0 == fib_index)
 116     {
 117       rv = VNET_API_ERROR_NO_SUCH_TABLE;
 118       goto done;
 119     }
 120
 121   flags = UDP_ENCAP_FIXUP_NONE;
 122   if (mp->udp_encap.src_port == 0)
 123     flags |= UDP_ENCAP_FIXUP_UDP_SRC_PORT_ENTROPY;
 124
 125   uei = udp_encap_add_and_lock (fproto, fib_index, &src_ip, &dst_ip,
 126                                 ntohs (mp->udp_encap.src_port),
 127                                 ntohs (mp->udp_encap.dst_port), flags);
 128
 129 done:
 130   REPLY_MACRO2 (VL_API_UDP_ENCAP_ADD_REPLY,
 131   ({
 132     rmp->id = ntohl (uei);
 133   }));
 134
 135 }
 136
 137 static void
 138 vl_api_udp_encap_del_t_handler (vl_api_udp_encap_del_t *mp)
 139 {
 140   vl_api_udp_encap_del_reply_t *rmp;
 141   int rv = 0;
 142
 143   udp_encap_unlock (ntohl (mp->id));
 144
 145   REPLY_MACRO (VL_API_UDP_ENCAP_DEL_REPLY);
 146 }
 147
 148 u32
 149 udp_api_decap_proto_to_index (vlib_main_t *vm,
 150                               vl_api_udp_decap_next_proto_t iproto)
 151 {
 152   switch (iproto)
 153     {
 154     case UDP_API_DECAP_PROTO_IP4:
 155       return vlib_get_node_by_name (vm, (u8 *) "ip4-input")->index;
 156     case UDP_API_DECAP_PROTO_IP6:
 157       return vlib_get_node_by_name (vm, (u8 *) "ip6-input")->index;
 158     case UDP_API_DECAP_PROTO_MPLS:
 159       return vlib_get_node_by_name (vm, (u8 *) "mpls-input")->index;
 160     }
 161   return ~0;
 162 }
 163
 164 static void
 165 vl_api_udp_decap_add_del_t_handler (vl_api_udp_decap_add_del_t *mp)
 166 {
 167   vl_api_udp_decap_add_del_reply_t *rmp;
 168   vlib_main_t *vm = vlib_get_main ();
 169   int rv = 0;
 170
 171   if (mp->is_add)
 172     {
 173       u32 node_index =
 174         udp_api_decap_proto_to_index (vm, ntohl (mp->udp_decap.next_proto));
 175       if (node_index == ~0)
 176         rv = VNET_API_ERROR_INVALID_PROTOCOL;
 177       else
 178         udp_register_dst_port (vm, ntohs (mp->udp_decap.port), node_index,
 179                                mp->udp_decap.is_ip4);
 180     }
 181   else
 182     udp_unregister_dst_port (vm, ntohs (mp->udp_decap.port),
 183                              mp->udp_decap.is_ip4);
 184   REPLY_MACRO (VL_API_UDP_DECAP_ADD_DEL_REPLY);
 185 }
 186
 187 #include <vnet/udp/udp.api.c>
 188 static clib_error_t *
 189 udp_api_hookup (vlib_main_t * vm)
 190 {
 191   api_main_t *am = vlibapi_get_main ();
 192
 193   /*
 194    * Set up the (msg_name, crc, message-id) table
 195    */
 196   REPLY_MSG_ID_BASE = setup_message_id_table ();
 197
 198   /* Mark these APIs as mp safe */
 199   vl_api_set_msg_thread_safe (am, REPLY_MSG_ID_BASE + VL_API_UDP_ENCAP_ADD, 1);
 200   vl_api_set_msg_thread_safe (am, REPLY_MSG_ID_BASE + VL_API_UDP_ENCAP_DEL, 1);
 201   vl_api_set_msg_thread_safe (am, REPLY_MSG_ID_BASE + VL_API_UDP_ENCAP_DUMP,
 202                               1);
 203
 204   return 0;
 205 }
 206
 207 VLIB_API_INIT_FUNCTION (udp_api_hookup);
 208
 209 /*
 210  * fd.io coding-style-patch-verification: ON
 211  *
 212  * Local Variables:
 213  * eval: (c-set-style "gnu")
 214  * End:
 215  */