src/plugins/nat/nat44-ei/nat44_ei_api.c

   1 /*
   2  * Copyright (c) 2020 Cisco and/or its affiliates.
   3  *
   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 #include <vnet/ip/ip_types_api.h>
  18 #include <vlibmemory/api.h>
  19
  20 #include <vnet/fib/fib_table.h>
  21
  22 #include <nat/lib/nat_inlines.h>
  23 #include <nat/lib/ipfix_logging.h>
  24
  25 #include <nat/nat44-ei/nat44_ei.api_enum.h>
  26 #include <nat/nat44-ei/nat44_ei.api_types.h>
  27
  28 #include <nat/nat44-ei/nat44_ei_ha.h>
  29 #include <nat/nat44-ei/nat44_ei.h>
  30
  31 #define REPLY_MSG_ID_BASE nm->msg_id_base
  32 #include <vlibapi/api_helper_macros.h>
  33
  34 static void
  35 vl_api_nat44_ei_show_running_config_t_handler (
  36   vl_api_nat44_ei_show_running_config_t *mp)
  37 {
  38   vl_api_nat44_ei_show_running_config_reply_t *rmp;
  39   nat44_ei_main_t *nm = &nat44_ei_main;
  40   nat44_ei_config_t *rc = &nm->rconfig;
  41   int rv = 0;
  42
  43   REPLY_MACRO2_ZERO (
  44     VL_API_NAT44_EI_SHOW_RUNNING_CONFIG_REPLY, ({
  45       rmp->inside_vrf = htonl (rc->inside_vrf);
  46       rmp->outside_vrf = htonl (rc->outside_vrf);
  47       rmp->users = htonl (rc->users);
  48       rmp->sessions = htonl (rc->sessions);
  49       rmp->user_sessions = htonl (rc->user_sessions);
  50
  51       rmp->user_buckets = htonl (nm->user_buckets);
  52       rmp->translation_buckets = htonl (nm->translation_buckets);
  53
  54       rmp->timeouts.udp = htonl (nm->timeouts.udp);
  55       rmp->timeouts.tcp_established = htonl (nm->timeouts.tcp.established);
  56       rmp->timeouts.tcp_transitory = htonl (nm->timeouts.tcp.transitory);
  57       rmp->timeouts.icmp = htonl (nm->timeouts.icmp);
  58
  59       rmp->forwarding_enabled = nm->forwarding_enabled == 1;
  60       // consider how to split functionality between subplugins
  61       rmp->ipfix_logging_enabled = nat_ipfix_logging_enabled ();
  62
  63       if (rc->static_mapping_only)
  64         rmp->flags |= NAT44_EI_STATIC_MAPPING_ONLY;
  65       if (rc->connection_tracking)
  66         rmp->flags |= NAT44_EI_CONNECTION_TRACKING;
  67       if (rc->out2in_dpo)
  68         rmp->flags |= NAT44_EI_OUT2IN_DPO;
  69     }));
  70 }
  71
  72 static void
  73 vl_api_nat44_ei_set_workers_t_handler (vl_api_nat44_ei_set_workers_t *mp)
  74 {
  75   nat44_ei_main_t *nm = &nat44_ei_main;
  76   vl_api_nat44_ei_set_workers_reply_t *rmp;
  77   int rv = 0;
  78   uword *bitmap = 0;
  79   u64 mask;
  80
  81   mask = clib_net_to_host_u64 (mp->worker_mask);
  82
  83   if (nm->num_workers < 2)
  84     {
  85       rv = VNET_API_ERROR_FEATURE_DISABLED;
  86       goto send_reply;
  87     }
  88
  89   bitmap = clib_bitmap_set_multiple (bitmap, 0, mask, BITS (mask));
  90   rv = nat44_ei_set_workers (bitmap);
  91   clib_bitmap_free (bitmap);
  92
  93 send_reply:
  94   REPLY_MACRO (VL_API_NAT44_EI_SET_WORKERS_REPLY);
  95 }
  96
  97 static void
  98 send_nat_worker_details (u32 worker_index, vl_api_registration_t *reg,
  99                          u32 context)
 100 {
 101   vl_api_nat44_ei_worker_details_t *rmp;
 102   nat44_ei_main_t *nm = &nat44_ei_main;
 103   vlib_worker_thread_t *w =
 104     vlib_worker_threads + worker_index + nm->first_worker_index;
 105
 106   rmp = vl_msg_api_alloc (sizeof (*rmp));
 107   clib_memset (rmp, 0, sizeof (*rmp));
 108   rmp->_vl_msg_id = ntohs (VL_API_NAT44_EI_WORKER_DETAILS + nm->msg_id_base);
 109   rmp->context = context;
 110   rmp->worker_index = htonl (worker_index);
 111   rmp->lcore_id = htonl (w->cpu_id);
 112   strncpy ((char *) rmp->name, (char *) w->name, ARRAY_LEN (rmp->name) - 1);
 113
 114   vl_api_send_msg (reg, (u8 *) rmp);
 115 }
 116
 117 static void
 118 vl_api_nat44_ei_worker_dump_t_handler (vl_api_nat44_ei_worker_dump_t *mp)
 119 {
 120   vl_api_registration_t *reg;
 121   nat44_ei_main_t *nm = &nat44_ei_main;
 122   u32 *worker_index;
 123
 124   reg = vl_api_client_index_to_registration (mp->client_index);
 125   if (!reg)
 126     return;
 127
 128   vec_foreach (worker_index, nm->workers)
 129     {
 130       send_nat_worker_details (*worker_index, reg, mp->context);
 131     }
 132 }
 133
 134 static void
 135 vl_api_nat44_ei_set_log_level_t_handler (vl_api_nat44_ei_set_log_level_t *mp)
 136 {
 137   nat44_ei_main_t *nm = &nat44_ei_main;
 138   vl_api_nat44_ei_set_log_level_reply_t *rmp;
 139   int rv = 0;
 140
 141   if (nm->log_level > NAT_LOG_DEBUG)
 142     rv = VNET_API_ERROR_UNSUPPORTED;
 143   else
 144     nm->log_level = mp->log_level;
 145
 146   REPLY_MACRO (VL_API_NAT44_EI_SET_LOG_LEVEL_REPLY);
 147 }
 148
 149 static void
 150 vl_api_nat44_ei_plugin_enable_disable_t_handler (
 151   vl_api_nat44_ei_plugin_enable_disable_t *mp)
 152 {
 153   nat44_ei_main_t *nm = &nat44_ei_main;
 154   nat44_ei_config_t c = { 0 };
 155   vl_api_nat44_ei_plugin_enable_disable_reply_t *rmp;
 156   int rv = 0;
 157
 158   if (mp->enable)
 159     {
 160       c.static_mapping_only = mp->flags & NAT44_EI_STATIC_MAPPING_ONLY;
 161       c.connection_tracking = mp->flags & NAT44_EI_CONNECTION_TRACKING;
 162       c.out2in_dpo = mp->flags & NAT44_EI_OUT2IN_DPO;
 163
 164       c.inside_vrf = ntohl (mp->inside_vrf);
 165       c.outside_vrf = ntohl (mp->outside_vrf);
 166
 167       c.users = ntohl (mp->users);
 168
 169       c.sessions = ntohl (mp->sessions);
 170
 171       c.user_sessions = ntohl (mp->user_sessions);
 172
 173       rv = nat44_ei_plugin_enable (c);
 174     }
 175   else
 176     {
 177       rv = nat44_ei_plugin_disable ();
 178     }
 179
 180   REPLY_MACRO (VL_API_NAT44_EI_PLUGIN_ENABLE_DISABLE_REPLY);
 181 }
 182
 183 static void
 184 vl_api_nat44_ei_ipfix_enable_disable_t_handler (
 185   vl_api_nat44_ei_ipfix_enable_disable_t *mp)
 186 {
 187   nat44_ei_main_t *nm = &nat44_ei_main;
 188   vl_api_nat44_ei_ipfix_enable_disable_reply_t *rmp;
 189   int rv = 0;
 190
 191   rv = nat_ipfix_logging_enable_disable (mp->enable,
 192                                          clib_host_to_net_u32 (mp->domain_id),
 193                                          clib_host_to_net_u16 (mp->src_port));
 194
 195   REPLY_MACRO (VL_API_NAT44_EI_IPFIX_ENABLE_DISABLE_REPLY);
 196 }
 197
 198 static void
 199 vl_api_nat44_ei_set_timeouts_t_handler (vl_api_nat44_ei_set_timeouts_t *mp)
 200 {
 201   nat44_ei_main_t *nm = &nat44_ei_main;
 202   vl_api_nat44_ei_set_timeouts_reply_t *rmp;
 203   int rv = 0;
 204
 205   nm->timeouts.udp = ntohl (mp->udp);
 206   nm->timeouts.tcp.established = ntohl (mp->tcp_established);
 207   nm->timeouts.tcp.transitory = ntohl (mp->tcp_transitory);
 208   nm->timeouts.icmp = ntohl (mp->icmp);
 209
 210   REPLY_MACRO (VL_API_NAT44_EI_SET_TIMEOUTS_REPLY);
 211 }
 212
 213 static void
 214 vl_api_nat44_ei_set_addr_and_port_alloc_alg_t_handler (
 215   vl_api_nat44_ei_set_addr_and_port_alloc_alg_t *mp)
 216 {
 217   nat44_ei_main_t *nm = &nat44_ei_main;
 218   vl_api_nat44_ei_set_addr_and_port_alloc_alg_reply_t *rmp;
 219   int rv = 0;
 220   u16 port_start, port_end;
 221
 222   switch (mp->alg)
 223     {
 224     case NAT44_EI_ADDR_AND_PORT_ALLOC_ALG_DEFAULT:
 225       nat44_ei_set_alloc_default ();
 226       break;
 227     case NAT44_EI_ADDR_AND_PORT_ALLOC_ALG_MAPE:
 228       nat44_ei_set_alloc_mape (ntohs (mp->psid), mp->psid_offset,
 229                                mp->psid_length);
 230       break;
 231     case NAT44_EI_ADDR_AND_PORT_ALLOC_ALG_RANGE:
 232       port_start = ntohs (mp->start_port);
 233       port_end = ntohs (mp->end_port);
 234       if (port_end <= port_start)
 235         {
 236           rv = VNET_API_ERROR_INVALID_VALUE;
 237           goto send_reply;
 238         }
 239       nat44_ei_set_alloc_range (port_start, port_end);
 240       break;
 241     default:
 242       rv = VNET_API_ERROR_INVALID_VALUE;
 243       break;
 244     }
 245
 246 send_reply:
 247   REPLY_MACRO (VL_API_NAT44_EI_SET_ADDR_AND_PORT_ALLOC_ALG_REPLY);
 248 }
 249
 250 static void
 251 vl_api_nat44_ei_get_addr_and_port_alloc_alg_t_handler (
 252   vl_api_nat44_ei_get_addr_and_port_alloc_alg_t *mp)
 253 {
 254   nat44_ei_main_t *nm = &nat44_ei_main;
 255   vl_api_nat44_ei_get_addr_and_port_alloc_alg_reply_t *rmp;
 256   int rv = 0;
 257
 258   REPLY_MACRO2 (VL_API_NAT44_EI_GET_ADDR_AND_PORT_ALLOC_ALG_REPLY, ({
 259                   rmp->alg = nm->addr_and_port_alloc_alg;
 260                   rmp->psid_offset = nm->psid_offset;
 261                   rmp->psid_length = nm->psid_length;
 262                   rmp->psid = htons (nm->psid);
 263                   rmp->start_port = htons (nm->start_port);
 264                   rmp->end_port = htons (nm->end_port);
 265                 }))
 266 }
 267
 268 static void
 269 vl_api_nat44_ei_set_mss_clamping_t_handler (
 270   vl_api_nat44_ei_set_mss_clamping_t *mp)
 271 {
 272   nat44_ei_main_t *nm = &nat44_ei_main;
 273   vl_api_nat44_ei_set_mss_clamping_reply_t *rmp;
 274   int rv = 0;
 275
 276   if (mp->enable)
 277     nm->mss_clamping = ntohs (mp->mss_value);
 278   else
 279     nm->mss_clamping = 0;
 280
 281   REPLY_MACRO (VL_API_NAT44_EI_SET_MSS_CLAMPING_REPLY);
 282 }
 283
 284 static void
 285 vl_api_nat44_ei_get_mss_clamping_t_handler (
 286   vl_api_nat44_ei_get_mss_clamping_t *mp)
 287 {
 288   nat44_ei_main_t *nm = &nat44_ei_main;
 289   vl_api_nat44_ei_get_mss_clamping_reply_t *rmp;
 290   int rv = 0;
 291
 292   REPLY_MACRO2 (VL_API_NAT44_EI_GET_MSS_CLAMPING_REPLY, ({
 293                   rmp->enable = nm->mss_clamping ? 1 : 0;
 294                   rmp->mss_value = htons (nm->mss_clamping);
 295                 }))
 296 }
 297
 298 static void
 299 vl_api_nat44_ei_ha_set_listener_t_handler (
 300   vl_api_nat44_ei_ha_set_listener_t *mp)
 301 {
 302   nat44_ei_main_t *nm = &nat44_ei_main;
 303   vl_api_nat44_ei_ha_set_listener_reply_t *rmp;
 304   ip4_address_t addr;
 305   int rv;
 306
 307   memcpy (&addr, &mp->ip_address, sizeof (addr));
 308   rv = nat_ha_set_listener (vlib_get_main (), &addr,
 309                             clib_net_to_host_u16 (mp->port),
 310                             clib_net_to_host_u32 (mp->path_mtu));
 311
 312   REPLY_MACRO (VL_API_NAT44_EI_HA_SET_LISTENER_REPLY);
 313 }
 314
 315 static void
 316 vl_api_nat44_ei_ha_get_listener_t_handler (
 317   vl_api_nat44_ei_ha_get_listener_t *mp)
 318 {
 319   nat44_ei_main_t *nm = &nat44_ei_main;
 320   vl_api_nat44_ei_ha_get_listener_reply_t *rmp;
 321   int rv = 0;
 322   ip4_address_t addr;
 323   u16 port;
 324   u32 path_mtu;
 325
 326   nat_ha_get_listener (&addr, &port, &path_mtu);
 327
 328   REPLY_MACRO2 (VL_API_NAT44_EI_HA_GET_LISTENER_REPLY, ({
 329                   clib_memcpy (rmp->ip_address, &addr, sizeof (ip4_address_t));
 330                   rmp->port = clib_host_to_net_u16 (port);
 331                   rmp->path_mtu = clib_host_to_net_u32 (path_mtu);
 332                 }))
 333 }
 334
 335 static void
 336 vl_api_nat44_ei_ha_set_failover_t_handler (
 337   vl_api_nat44_ei_ha_set_failover_t *mp)
 338 {
 339   nat44_ei_main_t *nm = &nat44_ei_main;
 340   vl_api_nat44_ei_ha_set_failover_reply_t *rmp;
 341   ip4_address_t addr;
 342   int rv;
 343
 344   memcpy (&addr, &mp->ip_address, sizeof (addr));
 345   rv = nat_ha_set_failover (
 346     vlib_get_main (), &addr, clib_net_to_host_u16 (mp->port),
 347     clib_net_to_host_u32 (mp->session_refresh_interval));
 348
 349   REPLY_MACRO (VL_API_NAT44_EI_HA_SET_FAILOVER_REPLY);
 350 }
 351
 352 static void
 353 vl_api_nat44_ei_ha_get_failover_t_handler (
 354   vl_api_nat44_ei_ha_get_failover_t *mp)
 355 {
 356   nat44_ei_main_t *nm = &nat44_ei_main;
 357   vl_api_nat44_ei_ha_get_failover_reply_t *rmp;
 358   int rv = 0;
 359   ip4_address_t addr;
 360   u16 port;
 361   u32 session_refresh_interval;
 362
 363   nat_ha_get_failover (&addr, &port, &session_refresh_interval);
 364
 365   REPLY_MACRO2 (VL_API_NAT44_EI_HA_GET_FAILOVER_REPLY, ({
 366                   clib_memcpy (rmp->ip_address, &addr, sizeof (ip4_address_t));
 367                   rmp->port = clib_host_to_net_u16 (port);
 368                   rmp->session_refresh_interval =
 369                     clib_host_to_net_u32 (session_refresh_interval);
 370                 }))
 371 }
 372
 373 static void
 374 vl_api_nat44_ei_ha_flush_t_handler (vl_api_nat44_ei_ha_flush_t *mp)
 375 {
 376   nat44_ei_main_t *nm = &nat44_ei_main;
 377   vl_api_nat44_ei_ha_flush_reply_t *rmp;
 378   int rv = 0;
 379
 380   nat_ha_flush (0);
 381
 382   REPLY_MACRO (VL_API_NAT44_EI_HA_FLUSH_REPLY);
 383 }
 384
 385 static void
 386 nat_ha_resync_completed_event_cb (u32 client_index, u32 pid, u32 missed_count)
 387 {
 388   nat44_ei_main_t *nm = &nat44_ei_main;
 389   vl_api_registration_t *reg;
 390   vl_api_nat44_ei_ha_resync_completed_event_t *mp;
 391
 392   reg = vl_api_client_index_to_registration (client_index);
 393   if (!reg)
 394     return;
 395
 396   mp = vl_msg_api_alloc (sizeof (*mp));
 397   clib_memset (mp, 0, sizeof (*mp));
 398   mp->client_index = client_index;
 399   mp->pid = pid;
 400   mp->missed_count = clib_host_to_net_u32 (missed_count);
 401   mp->_vl_msg_id =
 402     ntohs (VL_API_NAT44_EI_HA_RESYNC_COMPLETED_EVENT + nm->msg_id_base);
 403
 404   vl_api_send_msg (reg, (u8 *) mp);
 405 }
 406
 407 static void
 408 vl_api_nat44_ei_ha_resync_t_handler (vl_api_nat44_ei_ha_resync_t *mp)
 409 {
 410   nat44_ei_main_t *nm = &nat44_ei_main;
 411   vl_api_nat44_ei_ha_resync_reply_t *rmp;
 412   int rv;
 413
 414   rv = nat_ha_resync (
 415     mp->client_index, mp->pid,
 416     mp->want_resync_event ? nat_ha_resync_completed_event_cb : NULL);
 417
 418   REPLY_MACRO (VL_API_NAT44_EI_HA_RESYNC_REPLY);
 419 }
 420
 421 static void
 422 vl_api_nat44_ei_del_user_t_handler (vl_api_nat44_ei_del_user_t *mp)
 423 {
 424   nat44_ei_main_t *nm = &nat44_ei_main;
 425   vl_api_nat44_ei_del_user_reply_t *rmp;
 426   ip4_address_t addr;
 427   int rv;
 428   memcpy (&addr.as_u8, mp->ip_address, 4);
 429   rv = nat44_ei_user_del (&addr, ntohl (mp->fib_index));
 430   REPLY_MACRO (VL_API_NAT44_EI_DEL_USER_REPLY);
 431 }
 432
 433 static void
 434 vl_api_nat44_ei_add_del_address_range_t_handler (
 435   vl_api_nat44_ei_add_del_address_range_t *mp)
 436 {
 437   nat44_ei_main_t *nm = &nat44_ei_main;
 438   vl_api_nat44_ei_add_del_address_range_reply_t *rmp;
 439   ip4_address_t this_addr;
 440   u8 is_add;
 441   u32 start_host_order, end_host_order;
 442   u32 vrf_id;
 443   int i, count;
 444   int rv = 0;
 445   u32 *tmp;
 446
 447   if (nm->static_mapping_only)
 448     {
 449       rv = VNET_API_ERROR_FEATURE_DISABLED;
 450       goto send_reply;
 451     }
 452
 453   is_add = mp->is_add;
 454
 455   tmp = (u32 *) mp->first_ip_address;
 456   start_host_order = clib_host_to_net_u32 (tmp[0]);
 457   tmp = (u32 *) mp->last_ip_address;
 458   end_host_order = clib_host_to_net_u32 (tmp[0]);
 459
 460   count = (end_host_order - start_host_order) + 1;
 461
 462   vrf_id = clib_host_to_net_u32 (mp->vrf_id);
 463
 464   if (count > 1024)
 465     nat44_ei_log_info ("%U - %U, %d addresses...", format_ip4_address,
 466                        mp->first_ip_address, format_ip4_address,
 467                        mp->last_ip_address, count);
 468
 469   memcpy (&this_addr.as_u8, mp->first_ip_address, 4);
 470
 471   for (i = 0; i < count; i++)
 472     {
 473       if (is_add)
 474         rv = nat44_ei_add_address (&this_addr, vrf_id);
 475       else
 476         rv = nat44_ei_del_address (this_addr, 0);
 477
 478       if (rv)
 479         goto send_reply;
 480
 481       if (nm->out2in_dpo)
 482         nat44_ei_add_del_address_dpo (this_addr, is_add);
 483
 484       increment_v4_address (&this_addr);
 485     }
 486
 487 send_reply:
 488   REPLY_MACRO (VL_API_NAT44_EI_ADD_DEL_ADDRESS_RANGE_REPLY);
 489 }
 490
 491 static void
 492 send_nat44_ei_address_details (nat44_ei_address_t *a,
 493                                vl_api_registration_t *reg, u32 context)
 494 {
 495   vl_api_nat44_ei_address_details_t *rmp;
 496   nat44_ei_main_t *nm = &nat44_ei_main;
 497
 498   rmp = vl_msg_api_alloc (sizeof (*rmp));
 499   clib_memset (rmp, 0, sizeof (*rmp));
 500   rmp->_vl_msg_id = ntohs (VL_API_NAT44_EI_ADDRESS_DETAILS + nm->msg_id_base);
 501   clib_memcpy (rmp->ip_address, &(a->addr), 4);
 502   if (a->fib_index != ~0)
 503     {
 504       fib_table_t *fib = fib_table_get (a->fib_index, FIB_PROTOCOL_IP4);
 505       rmp->vrf_id = ntohl (fib->ft_table_id);
 506     }
 507   else
 508     rmp->vrf_id = ~0;
 509   rmp->context = context;
 510
 511   vl_api_send_msg (reg, (u8 *) rmp);
 512 }
 513
 514 static void
 515 vl_api_nat44_ei_address_dump_t_handler (vl_api_nat44_ei_address_dump_t *mp)
 516 {
 517   vl_api_registration_t *reg;
 518   nat44_ei_main_t *nm = &nat44_ei_main;
 519   nat44_ei_address_t *a;
 520
 521   reg = vl_api_client_index_to_registration (mp->client_index);
 522   if (!reg)
 523     return;
 524
 525   vec_foreach (a, nm->addresses)
 526     {
 527       send_nat44_ei_address_details (a, reg, mp->context);
 528     }
 529 }
 530
 531 static void
 532 vl_api_nat44_ei_interface_add_del_feature_t_handler (
 533   vl_api_nat44_ei_interface_add_del_feature_t *mp)
 534 {
 535   nat44_ei_main_t *nm = &nat44_ei_main;
 536   vl_api_nat44_ei_interface_add_del_feature_reply_t *rmp;
 537   u32 sw_if_index = ntohl (mp->sw_if_index);
 538   int rv = 0;
 539
 540   VALIDATE_SW_IF_INDEX (mp);
 541
 542   if (mp->is_add)
 543     {
 544       rv =
 545         nat44_ei_add_interface (sw_if_index, mp->flags & NAT44_EI_IF_INSIDE);
 546     }
 547   else
 548     {
 549       rv =
 550         nat44_ei_del_interface (sw_if_index, mp->flags & NAT44_EI_IF_INSIDE);
 551     }
 552
 553   BAD_SW_IF_INDEX_LABEL;
 554   REPLY_MACRO (VL_API_NAT44_EI_INTERFACE_ADD_DEL_FEATURE_REPLY);
 555 }
 556
 557 static void
 558 send_nat44_ei_interface_details (nat44_ei_interface_t *i,
 559                                  vl_api_registration_t *reg, u32 context)
 560 {
 561   vl_api_nat44_ei_interface_details_t *rmp;
 562   nat44_ei_main_t *nm = &nat44_ei_main;
 563
 564   rmp = vl_msg_api_alloc (sizeof (*rmp));
 565   clib_memset (rmp, 0, sizeof (*rmp));
 566   rmp->_vl_msg_id =
 567     ntohs (VL_API_NAT44_EI_INTERFACE_DETAILS + nm->msg_id_base);
 568   rmp->sw_if_index = ntohl (i->sw_if_index);
 569
 570   if (nat44_ei_interface_is_inside (i))
 571     rmp->flags |= NAT44_EI_IF_INSIDE;
 572   if (nat44_ei_interface_is_outside (i))
 573     rmp->flags |= NAT44_EI_IF_OUTSIDE;
 574
 575   rmp->context = context;
 576
 577   vl_api_send_msg (reg, (u8 *) rmp);
 578 }
 579
 580 static void
 581 vl_api_nat44_ei_interface_dump_t_handler (vl_api_nat44_ei_interface_dump_t *mp)
 582 {
 583   vl_api_registration_t *reg;
 584   nat44_ei_main_t *nm = &nat44_ei_main;
 585   nat44_ei_interface_t *i;
 586
 587   reg = vl_api_client_index_to_registration (mp->client_index);
 588   if (!reg)
 589     return;
 590
 591   pool_foreach (i, nm->interfaces)
 592     {
 593       send_nat44_ei_interface_details (i, reg, mp->context);
 594     }
 595 }
 596
 597 static_always_inline int
 598 add_del_dummy_output_interface (u32 sw_if_index, u8 is_inside, u8 is_add)
 599 {
 600   nat44_ei_main_t *nm = &nat44_ei_main;
 601   nat44_ei_interface_t *i;
 602   int rv = 1;
 603
 604   pool_foreach (i, nm->output_feature_dummy_interfaces)
 605     {
 606       if (i->sw_if_index == sw_if_index)
 607         {
 608           if (!is_add)
 609             {
 610               pool_put (nm->output_feature_dummy_interfaces, i);
 611               rv = 0;
 612             }
 613           goto done;
 614         }
 615     }
 616
 617   if (is_add)
 618     {
 619       pool_get (nm->output_feature_dummy_interfaces, i);
 620       i->sw_if_index = sw_if_index;
 621
 622       if (is_inside)
 623         {
 624           i->flags |= NAT44_EI_INTERFACE_FLAG_IS_INSIDE;
 625         }
 626       else
 627         {
 628           i->flags |= NAT44_EI_INTERFACE_FLAG_IS_OUTSIDE;
 629         }
 630
 631       rv = 0;
 632     }
 633
 634 done:
 635   return rv;
 636 }
 637
 638 static void
 639 vl_api_nat44_ei_interface_add_del_output_feature_t_handler (
 640   vl_api_nat44_ei_interface_add_del_output_feature_t *mp)
 641 {
 642   vl_api_nat44_ei_interface_add_del_output_feature_reply_t *rmp;
 643   nat44_ei_main_t *nm = &nat44_ei_main;
 644   u32 sw_if_index;
 645   int rv = 0;
 646
 647   VALIDATE_SW_IF_INDEX (mp);
 648
 649   sw_if_index = ntohl (mp->sw_if_index);
 650
 651   // register all interfaces in the dummy structure
 652   rv = add_del_dummy_output_interface (
 653     sw_if_index, mp->flags & NAT44_EI_IF_INSIDE, mp->is_add);
 654
 655   if (!(mp->flags & NAT44_EI_IF_INSIDE))
 656     {
 657       if (mp->is_add)
 658         {
 659           rv = nat44_ei_add_output_interface (sw_if_index);
 660         }
 661       else
 662         {
 663           rv = nat44_ei_del_output_interface (sw_if_index);
 664         }
 665     }
 666
 667   BAD_SW_IF_INDEX_LABEL;
 668   REPLY_MACRO (VL_API_NAT44_EI_INTERFACE_ADD_DEL_OUTPUT_FEATURE_REPLY);
 669 }
 670
 671 static void
 672 send_nat44_ei_interface_output_feature_details (nat44_ei_interface_t *i,
 673                                                 vl_api_registration_t *reg,
 674                                                 u32 context)
 675 {
 676   vl_api_nat44_ei_interface_output_feature_details_t *rmp;
 677   nat44_ei_main_t *nm = &nat44_ei_main;
 678
 679   rmp = vl_msg_api_alloc (sizeof (*rmp));
 680   clib_memset (rmp, 0, sizeof (*rmp));
 681   rmp->_vl_msg_id =
 682     ntohs (VL_API_NAT44_EI_INTERFACE_OUTPUT_FEATURE_DETAILS + nm->msg_id_base);
 683   rmp->sw_if_index = ntohl (i->sw_if_index);
 684   rmp->context = context;
 685
 686   if (nat44_ei_interface_is_inside (i))
 687     {
 688       rmp->flags |= NAT44_EI_IF_INSIDE;
 689     }
 690
 691   vl_api_send_msg (reg, (u8 *) rmp);
 692 }
 693
 694 static void
 695 vl_api_nat44_ei_interface_output_feature_dump_t_handler (
 696   vl_api_nat44_ei_interface_output_feature_dump_t *mp)
 697 {
 698   vl_api_registration_t *reg;
 699   nat44_ei_main_t *nm = &nat44_ei_main;
 700   nat44_ei_interface_t *i;
 701
 702   reg = vl_api_client_index_to_registration (mp->client_index);
 703   if (!reg)
 704     return;
 705
 706   pool_foreach (i, nm->output_feature_dummy_interfaces)
 707     {
 708       send_nat44_ei_interface_output_feature_details (i, reg, mp->context);
 709     }
 710 }
 711
 712 static void
 713 vl_api_nat44_ei_add_del_output_interface_t_handler (
 714   vl_api_nat44_ei_add_del_output_interface_t *mp)
 715 {
 716   vl_api_nat44_ei_add_del_output_interface_reply_t *rmp;
 717   nat44_ei_main_t *nm = &nat44_ei_main;
 718   int rv = 0;
 719
 720   VALIDATE_SW_IF_INDEX_END (mp);
 721
 722   if (mp->is_add)
 723     {
 724       rv = nat44_ei_add_output_interface (mp->sw_if_index);
 725     }
 726   else
 727     {
 728       rv = nat44_ei_del_output_interface (mp->sw_if_index);
 729     }
 730
 731 bad_sw_if_index:
 732   REPLY_MACRO_END (VL_API_NAT44_EI_ADD_DEL_OUTPUT_INTERFACE_REPLY);
 733 }
 734
 735 #define vl_endianfun
 736 #include <nat/nat44-ei/nat44_ei.api.h>
 737 #undef vl_endianfun
 738 static void
 739 send_nat44_ei_output_interface_details (u32 index, vl_api_registration_t *rp,
 740                                         u32 context)
 741 {
 742   nat44_ei_main_t *nm = &nat44_ei_main;
 743   vl_api_nat44_ei_output_interface_details_t *rmp;
 744   nat44_ei_interface_t *i =
 745     pool_elt_at_index (nm->output_feature_interfaces, index);
 746
 747   /* Make sure every field is initiated (or don't skip the clib_memset()) */
 748   REPLY_MACRO_DETAILS4 (
 749     VL_API_NAT44_EI_OUTPUT_INTERFACE_DETAILS, rp, context, ({
 750       rmp->sw_if_index = i->sw_if_index;
 751
 752       /* Endian hack until apigen registers _details
 753        * endian functions */
 754       vl_api_nat44_ei_output_interface_details_t_endian (rmp,
 755                                                          1 /* to network */);
 756       rmp->_vl_msg_id = htons (rmp->_vl_msg_id);
 757       rmp->context = htonl (rmp->context);
 758     }));
 759 }
 760
 761 static void
 762 vl_api_nat44_ei_output_interface_get_t_handler (
 763   vl_api_nat44_ei_output_interface_get_t *mp)
 764 {
 765   vl_api_nat44_ei_output_interface_get_reply_t *rmp;
 766   nat44_ei_main_t *nm = &nat44_ei_main;
 767   i32 rv = 0;
 768
 769   if (pool_elts (nm->output_feature_interfaces) == 0)
 770     {
 771       REPLY_MACRO (VL_API_NAT44_EI_OUTPUT_INTERFACE_GET_REPLY);
 772       return;
 773     }
 774
 775   REPLY_AND_DETAILS_MACRO (
 776     VL_API_NAT44_EI_OUTPUT_INTERFACE_GET_REPLY, nm->output_feature_interfaces,
 777     ({ send_nat44_ei_output_interface_details (cursor, rp, mp->context); }));
 778 }
 779
 780 static void
 781 vl_api_nat44_ei_add_del_static_mapping_t_handler (
 782   vl_api_nat44_ei_add_del_static_mapping_t *mp)
 783 {
 784   vl_api_nat44_ei_add_del_static_mapping_reply_t *rmp;
 785
 786   nat44_ei_main_t *nm = &nat44_ei_main;
 787   int rv = 0;
 788
 789   ip4_address_t l_addr, e_addr, pool_addr = { 0 };
 790   u32 sw_if_index, flags = 0, vrf_id;
 791   u16 l_port = 0, e_port = 0;
 792   nat_protocol_t proto = 0;
 793   u8 *tag = 0;
 794
 795   memcpy (&l_addr.as_u8, mp->local_ip_address, 4);
 796
 797   if (mp->flags & NAT44_EI_ADDR_ONLY_MAPPING)
 798     {
 799       flags |= NAT44_EI_SM_FLAG_ADDR_ONLY;
 800     }
 801   else
 802     {
 803       l_port = mp->local_port;
 804       e_port = mp->external_port;
 805       proto = ip_proto_to_nat_proto (mp->protocol);
 806     }
 807
 808   sw_if_index = clib_net_to_host_u32 (mp->external_sw_if_index);
 809   if (sw_if_index != ~0)
 810     {
 811       flags |= NAT44_EI_SM_FLAG_SWITCH_ADDRESS;
 812     }
 813   else
 814     {
 815       memcpy (&e_addr.as_u8, mp->external_ip_address, 4);
 816     }
 817
 818   vrf_id = clib_net_to_host_u32 (mp->vrf_id);
 819
 820   if (mp->is_add)
 821     {
 822       mp->tag[sizeof (mp->tag) - 1] = 0;
 823       tag = format (0, "%s", mp->tag);
 824       vec_terminate_c_string (tag);
 825
 826       rv = nat44_ei_add_static_mapping (l_addr, e_addr, l_port, e_port, proto,
 827                                         vrf_id, sw_if_index, flags, pool_addr,
 828                                         tag);
 829       vec_free (tag);
 830     }
 831   else
 832     {
 833       rv = nat44_ei_del_static_mapping (l_addr, e_addr, l_port, e_port, proto,
 834                                         vrf_id, sw_if_index, flags);
 835     }
 836   REPLY_MACRO (VL_API_NAT44_EI_ADD_DEL_STATIC_MAPPING_REPLY);
 837 }
 838
 839 static void
 840 send_nat44_ei_static_mapping_details (nat44_ei_static_mapping_t *m,
 841                                       vl_api_registration_t *reg, u32 context)
 842 {
 843   vl_api_nat44_ei_static_mapping_details_t *rmp;
 844   nat44_ei_main_t *nm = &nat44_ei_main;
 845   u32 len = sizeof (*rmp);
 846
 847   rmp = vl_msg_api_alloc (len);
 848   clib_memset (rmp, 0, len);
 849   rmp->_vl_msg_id =
 850     ntohs (VL_API_NAT44_EI_STATIC_MAPPING_DETAILS + nm->msg_id_base);
 851
 852   clib_memcpy (rmp->local_ip_address, &(m->local_addr), 4);
 853   clib_memcpy (rmp->external_ip_address, &(m->external_addr), 4);
 854   rmp->external_sw_if_index = ~0;
 855   rmp->vrf_id = htonl (m->vrf_id);
 856   rmp->context = context;
 857
 858   if (is_sm_addr_only (m->flags))
 859     {
 860       rmp->flags |= NAT44_EI_ADDR_ONLY_MAPPING;
 861     }
 862   else
 863     {
 864       rmp->protocol = nat_proto_to_ip_proto (m->proto);
 865       rmp->external_port = m->external_port;
 866       rmp->local_port = m->local_port;
 867     }
 868
 869   if (m->tag)
 870     strncpy ((char *) rmp->tag, (char *) m->tag, vec_len (m->tag));
 871
 872   vl_api_send_msg (reg, (u8 *) rmp);
 873 }
 874
 875 static void
 876 send_nat44_ei_static_map_resolve_details (nat44_ei_static_map_resolve_t *m,
 877                                           vl_api_registration_t *reg,
 878                                           u32 context)
 879 {
 880   vl_api_nat44_ei_static_mapping_details_t *rmp;
 881   nat44_ei_main_t *nm = &nat44_ei_main;
 882
 883   rmp = vl_msg_api_alloc (sizeof (*rmp));
 884   clib_memset (rmp, 0, sizeof (*rmp));
 885   rmp->_vl_msg_id =
 886     ntohs (VL_API_NAT44_EI_STATIC_MAPPING_DETAILS + nm->msg_id_base);
 887   clib_memcpy (rmp->local_ip_address, &(m->l_addr), 4);
 888   rmp->external_sw_if_index = htonl (m->sw_if_index);
 889   rmp->vrf_id = htonl (m->vrf_id);
 890   rmp->context = context;
 891
 892   if (is_sm_addr_only (m->flags))
 893     {
 894       rmp->flags |= NAT44_EI_ADDR_ONLY_MAPPING;
 895     }
 896   else
 897     {
 898       rmp->protocol = nat_proto_to_ip_proto (m->proto);
 899       rmp->external_port = m->e_port;
 900       rmp->local_port = m->l_port;
 901     }
 902   if (m->tag)
 903     strncpy ((char *) rmp->tag, (char *) m->tag, vec_len (m->tag));
 904
 905   vl_api_send_msg (reg, (u8 *) rmp);
 906 }
 907
 908 static void
 909 vl_api_nat44_ei_static_mapping_dump_t_handler (
 910   vl_api_nat44_ei_static_mapping_dump_t *mp)
 911 {
 912   vl_api_registration_t *reg;
 913   nat44_ei_main_t *nm = &nat44_ei_main;
 914   nat44_ei_static_mapping_t *m;
 915   nat44_ei_static_map_resolve_t *rp;
 916   int j;
 917
 918   reg = vl_api_client_index_to_registration (mp->client_index);
 919   if (!reg)
 920     return;
 921
 922   pool_foreach (m, nm->static_mappings)
 923     {
 924       if (!is_sm_identity_nat (m->flags))
 925         send_nat44_ei_static_mapping_details (m, reg, mp->context);
 926     }
 927
 928   for (j = 0; j < vec_len (nm->to_resolve); j++)
 929     {
 930       rp = nm->to_resolve + j;
 931       if (!is_sm_identity_nat (rp->flags))
 932         send_nat44_ei_static_map_resolve_details (rp, reg, mp->context);
 933     }
 934 }
 935
 936 static void
 937 vl_api_nat44_ei_add_del_identity_mapping_t_handler (
 938   vl_api_nat44_ei_add_del_identity_mapping_t *mp)
 939 {
 940   vl_api_nat44_ei_add_del_identity_mapping_reply_t *rmp;
 941
 942   nat44_ei_main_t *nm = &nat44_ei_main;
 943   int rv = 0;
 944
 945   ip4_address_t addr, pool_addr = { 0 };
 946   u32 sw_if_index, flags, vrf_id;
 947   nat_protocol_t proto = 0;
 948   u16 port = 0;
 949   u8 *tag = 0;
 950
 951   flags = NAT44_EI_SM_FLAG_IDENTITY_NAT;
 952
 953   if (mp->flags & NAT44_EI_ADDR_ONLY_MAPPING)
 954     {
 955       flags |= NAT44_EI_SM_FLAG_ADDR_ONLY;
 956     }
 957   else
 958     {
 959       port = mp->port;
 960       proto = ip_proto_to_nat_proto (mp->protocol);
 961     }
 962
 963   sw_if_index = clib_net_to_host_u32 (mp->sw_if_index);
 964   if (sw_if_index != ~0)
 965     {
 966       flags |= NAT44_EI_SM_FLAG_SWITCH_ADDRESS;
 967     }
 968   else
 969     {
 970       memcpy (&addr.as_u8, mp->ip_address, 4);
 971     }
 972
 973   vrf_id = clib_net_to_host_u32 (mp->vrf_id);
 974
 975   if (mp->is_add)
 976     {
 977       mp->tag[sizeof (mp->tag) - 1] = 0;
 978       tag = format (0, "%s", mp->tag);
 979       vec_terminate_c_string (tag);
 980
 981       rv = nat44_ei_add_static_mapping (addr, addr, port, port, proto, vrf_id,
 982                                         sw_if_index, flags, pool_addr, tag);
 983       vec_free (tag);
 984     }
 985   else
 986     {
 987       rv = nat44_ei_del_static_mapping (addr, addr, port, port, proto, vrf_id,
 988                                         sw_if_index, flags);
 989     }
 990
 991   REPLY_MACRO (VL_API_NAT44_EI_ADD_DEL_IDENTITY_MAPPING_REPLY);
 992 }
 993
 994 static void
 995 send_nat44_ei_identity_mapping_details (nat44_ei_static_mapping_t *m,
 996                                         int index, vl_api_registration_t *reg,
 997                                         u32 context)
 998 {
 999   vl_api_nat44_ei_identity_mapping_details_t *rmp;
1000   nat44_ei_main_t *nm = &nat44_ei_main;
1001   nat44_ei_lb_addr_port_t *local = pool_elt_at_index (m->locals, index);
1002
1003   rmp = vl_msg_api_alloc (sizeof (*rmp));
1004   clib_memset (rmp, 0, sizeof (*rmp));
1005   rmp->_vl_msg_id =
1006     ntohs (VL_API_NAT44_EI_IDENTITY_MAPPING_DETAILS + nm->msg_id_base);
1007
1008   if (is_sm_addr_only (m->flags))
1009     rmp->flags |= NAT44_EI_ADDR_ONLY_MAPPING;
1010
1011   clib_memcpy (rmp->ip_address, &(m->local_addr), 4);
1012   rmp->port = m->local_port;
1013   rmp->sw_if_index = ~0;
1014   rmp->vrf_id = htonl (local->vrf_id);
1015   rmp->protocol = nat_proto_to_ip_proto (m->proto);
1016   rmp->context = context;
1017   if (m->tag)
1018     strncpy ((char *) rmp->tag, (char *) m->tag, vec_len (m->tag));
1019
1020   vl_api_send_msg (reg, (u8 *) rmp);
1021 }
1022
1023 static void
1024 send_nat44_ei_identity_map_resolve_details (nat44_ei_static_map_resolve_t *m,
1025                                             vl_api_registration_t *reg,
1026                                             u32 context)
1027 {
1028   vl_api_nat44_ei_identity_mapping_details_t *rmp;
1029   nat44_ei_main_t *nm = &nat44_ei_main;
1030   rmp = vl_msg_api_alloc (sizeof (*rmp));
1031   clib_memset (rmp, 0, sizeof (*rmp));
1032   rmp->_vl_msg_id =
1033     ntohs (VL_API_NAT44_EI_IDENTITY_MAPPING_DETAILS + nm->msg_id_base);
1034
1035   if (is_sm_addr_only (m->flags))
1036     rmp->flags = (vl_api_nat44_ei_config_flags_t) NAT44_EI_ADDR_ONLY_MAPPING;
1037
1038   rmp->port = m->l_port;
1039   rmp->sw_if_index = htonl (m->sw_if_index);
1040   rmp->vrf_id = htonl (m->vrf_id);
1041   rmp->protocol = nat_proto_to_ip_proto (m->proto);
1042   rmp->context = context;
1043   if (m->tag)
1044     strncpy ((char *) rmp->tag, (char *) m->tag, vec_len (m->tag));
1045
1046   vl_api_send_msg (reg, (u8 *) rmp);
1047 }
1048
1049 static void
1050 vl_api_nat44_ei_identity_mapping_dump_t_handler (
1051   vl_api_nat44_ei_identity_mapping_dump_t *mp)
1052 {
1053   vl_api_registration_t *reg;
1054   nat44_ei_main_t *nm = &nat44_ei_main;
1055   nat44_ei_static_mapping_t *m;
1056   nat44_ei_static_map_resolve_t *rp;
1057   int j;
1058
1059   reg = vl_api_client_index_to_registration (mp->client_index);
1060   if (!reg)
1061     return;
1062
1063   pool_foreach (m, nm->static_mappings)
1064     {
1065       if (is_sm_identity_nat (m->flags))
1066         {
1067           pool_foreach_index (j, m->locals)
1068             {
1069               send_nat44_ei_identity_mapping_details (m, j, reg, mp->context);
1070             }
1071         }
1072     }
1073
1074   for (j = 0; j < vec_len (nm->to_resolve); j++)
1075     {
1076       rp = nm->to_resolve + j;
1077       if (is_sm_identity_nat (rp->flags))
1078         send_nat44_ei_identity_map_resolve_details (rp, reg, mp->context);
1079     }
1080 }
1081
1082 static void
1083 vl_api_nat44_ei_add_del_interface_addr_t_handler (
1084   vl_api_nat44_ei_add_del_interface_addr_t *mp)
1085 {
1086   nat44_ei_main_t *nm = &nat44_ei_main;
1087   vl_api_nat44_ei_add_del_interface_addr_reply_t *rmp;
1088   u32 sw_if_index = ntohl (mp->sw_if_index);
1089   int rv = 0;
1090
1091   VALIDATE_SW_IF_INDEX (mp);
1092
1093   if (mp->is_add)
1094     {
1095       rv = nat44_ei_add_interface_address (sw_if_index);
1096     }
1097   else
1098     {
1099       rv = nat44_ei_del_interface_address (sw_if_index);
1100     }
1101
1102   BAD_SW_IF_INDEX_LABEL;
1103   REPLY_MACRO (VL_API_NAT44_EI_ADD_DEL_INTERFACE_ADDR_REPLY);
1104 }
1105
1106 static void
1107 send_nat44_ei_interface_addr_details (u32 sw_if_index,
1108                                       vl_api_registration_t *reg, u32 context)
1109 {
1110   vl_api_nat44_ei_interface_addr_details_t *rmp;
1111   nat44_ei_main_t *nm = &nat44_ei_main;
1112
1113   rmp = vl_msg_api_alloc (sizeof (*rmp));
1114   clib_memset (rmp, 0, sizeof (*rmp));
1115   rmp->_vl_msg_id =
1116     ntohs (VL_API_NAT44_EI_INTERFACE_ADDR_DETAILS + nm->msg_id_base);
1117   rmp->sw_if_index = ntohl (sw_if_index);
1118   rmp->context = context;
1119
1120   vl_api_send_msg (reg, (u8 *) rmp);
1121 }
1122
1123 static void
1124 vl_api_nat44_ei_interface_addr_dump_t_handler (
1125   vl_api_nat44_ei_interface_addr_dump_t *mp)
1126 {
1127   vl_api_registration_t *reg;
1128   nat44_ei_main_t *nm = &nat44_ei_main;
1129   u32 *i;
1130
1131   reg = vl_api_client_index_to_registration (mp->client_index);
1132   if (!reg)
1133     return;
1134
1135   vec_foreach (i, nm->auto_add_sw_if_indices)
1136     send_nat44_ei_interface_addr_details (*i, reg, mp->context);
1137 }
1138
1139 static void
1140 send_nat44_ei_user_details (nat44_ei_user_t *u, vl_api_registration_t *reg,
1141                             u32 context)
1142 {
1143   vl_api_nat44_ei_user_details_t *rmp;
1144   nat44_ei_main_t *nm = &nat44_ei_main;
1145   ip4_main_t *im = &ip4_main;
1146
1147   rmp = vl_msg_api_alloc (sizeof (*rmp));
1148   clib_memset (rmp, 0, sizeof (*rmp));
1149   rmp->_vl_msg_id = ntohs (VL_API_NAT44_EI_USER_DETAILS + nm->msg_id_base);
1150
1151   if (!pool_is_free_index (im->fibs, u->fib_index))
1152     {
1153       fib_table_t *fib = fib_table_get (u->fib_index, FIB_PROTOCOL_IP4);
1154       rmp->vrf_id = ntohl (fib->ft_table_id);
1155     }
1156
1157   clib_memcpy (rmp->ip_address, &(u->addr), 4);
1158   rmp->nsessions = ntohl (u->nsessions);
1159   rmp->nstaticsessions = ntohl (u->nstaticsessions);
1160   rmp->context = context;
1161
1162   vl_api_send_msg (reg, (u8 *) rmp);
1163 }
1164
1165 static void
1166 vl_api_nat44_ei_user_dump_t_handler (vl_api_nat44_ei_user_dump_t *mp)
1167 {
1168   vl_api_registration_t *reg;
1169   nat44_ei_main_t *nm = &nat44_ei_main;
1170   nat44_ei_main_per_thread_data_t *tnm;
1171   nat44_ei_user_t *u;
1172
1173   reg = vl_api_client_index_to_registration (mp->client_index);
1174   if (!reg)
1175     return;
1176
1177   vec_foreach (tnm, nm->per_thread_data)
1178     {
1179       pool_foreach (u, tnm->users)
1180         {
1181           send_nat44_ei_user_details (u, reg, mp->context);
1182         }
1183     }
1184 }
1185
1186 static void
1187 send_nat44_ei_user_session_details (nat44_ei_session_t *s,
1188                                     vl_api_registration_t *reg, u32 context)
1189 {
1190   vl_api_nat44_ei_user_session_details_t *rmp;
1191   nat44_ei_main_t *nm = &nat44_ei_main;
1192
1193   rmp = vl_msg_api_alloc (sizeof (*rmp));
1194   clib_memset (rmp, 0, sizeof (*rmp));
1195   rmp->_vl_msg_id =
1196     ntohs (VL_API_NAT44_EI_USER_SESSION_DETAILS + nm->msg_id_base);
1197   clib_memcpy (rmp->outside_ip_address, (&s->out2in.addr), 4);
1198   clib_memcpy (rmp->inside_ip_address, (&s->in2out.addr), 4);
1199
1200   if (nat44_ei_is_session_static (s))
1201     rmp->flags |= NAT44_EI_STATIC_MAPPING;
1202
1203   rmp->last_heard = clib_host_to_net_u64 ((u64) s->last_heard);
1204   rmp->total_bytes = clib_host_to_net_u64 (s->total_bytes);
1205   rmp->total_pkts = ntohl (s->total_pkts);
1206   rmp->context = context;
1207   if (nat44_ei_is_unk_proto_session (s))
1208     {
1209       rmp->outside_port = 0;
1210       rmp->inside_port = 0;
1211       rmp->protocol = ntohs (s->in2out.port);
1212     }
1213   else
1214     {
1215       rmp->outside_port = s->out2in.port;
1216       rmp->inside_port = s->in2out.port;
1217       rmp->protocol = ntohs (nat_proto_to_ip_proto (s->nat_proto));
1218     }
1219   vl_api_send_msg (reg, (u8 *) rmp);
1220 }
1221
1222 static void
1223 send_nat44_ei_user_session_v2_details (nat44_ei_session_t *s,
1224                                        vl_api_registration_t *reg, u32 context)
1225 {
1226   vl_api_nat44_ei_user_session_v2_details_t *rmp;
1227   nat44_ei_main_t *nm = &nat44_ei_main;
1228
1229   rmp = vl_msg_api_alloc (sizeof (*rmp));
1230   clib_memset (rmp, 0, sizeof (*rmp));
1231   rmp->_vl_msg_id =
1232     ntohs (VL_API_NAT44_EI_USER_SESSION_V2_DETAILS + nm->msg_id_base);
1233   clib_memcpy (rmp->outside_ip_address, (&s->out2in.addr), 4);
1234   clib_memcpy (rmp->inside_ip_address, (&s->in2out.addr), 4);
1235
1236   if (nat44_ei_is_session_static (s))
1237     rmp->flags |= NAT44_EI_STATIC_MAPPING;
1238
1239   rmp->last_heard = clib_host_to_net_u64 ((u64) s->last_heard);
1240   rmp->time_since_last_heard = clib_host_to_net_u64 (
1241     (u64) (vlib_time_now (vlib_get_main ()) - s->last_heard));
1242   rmp->total_bytes = clib_host_to_net_u64 (s->total_bytes);
1243   rmp->total_pkts = ntohl (s->total_pkts);
1244   rmp->context = context;
1245   if (nat44_ei_is_unk_proto_session (s))
1246     {
1247       rmp->outside_port = 0;
1248       rmp->inside_port = 0;
1249       rmp->protocol = ntohs (s->in2out.port);
1250     }
1251   else
1252     {
1253       rmp->outside_port = s->out2in.port;
1254       rmp->inside_port = s->in2out.port;
1255       rmp->protocol = ntohs (nat_proto_to_ip_proto (s->nat_proto));
1256     }
1257   vl_api_send_msg (reg, (u8 *) rmp);
1258 }
1259
1260 static void
1261 vl_api_nat44_ei_user_session_dump_t_handler (
1262   vl_api_nat44_ei_user_session_dump_t *mp)
1263 {
1264   vl_api_registration_t *reg;
1265   nat44_ei_main_t *nm = &nat44_ei_main;
1266   nat44_ei_main_per_thread_data_t *tnm;
1267   nat44_ei_session_t *s;
1268   clib_bihash_kv_8_8_t key, value;
1269   nat44_ei_user_key_t ukey;
1270   nat44_ei_user_t *u;
1271   u32 session_index, head_index, elt_index;
1272   dlist_elt_t *head, *elt;
1273   ip4_header_t ip;
1274
1275   reg = vl_api_client_index_to_registration (mp->client_index);
1276   if (!reg)
1277     return;
1278
1279   clib_memcpy (&ukey.addr, mp->ip_address, 4);
1280   ip.src_address.as_u32 = ukey.addr.as_u32;
1281   ukey.fib_index = fib_table_find (FIB_PROTOCOL_IP4, ntohl (mp->vrf_id));
1282   key.key = ukey.as_u64;
1283   if (nm->num_workers > 1)
1284     tnm = vec_elt_at_index (
1285       nm->per_thread_data,
1286       nat44_ei_get_in2out_worker_index (&ip, ukey.fib_index, 0));
1287   else
1288     tnm = vec_elt_at_index (nm->per_thread_data, nm->num_workers);
1289
1290   if (clib_bihash_search_8_8 (&tnm->user_hash, &key, &value))
1291     return;
1292   u = pool_elt_at_index (tnm->users, value.value);
1293   if (!u->nsessions && !u->nstaticsessions)
1294     return;
1295
1296   head_index = u->sessions_per_user_list_head_index;
1297   head = pool_elt_at_index (tnm->list_pool, head_index);
1298   elt_index = head->next;
1299   elt = pool_elt_at_index (tnm->list_pool, elt_index);
1300   session_index = elt->value;
1301   while (session_index != ~0)
1302     {
1303       s = pool_elt_at_index (tnm->sessions, session_index);
1304
1305       send_nat44_ei_user_session_details (s, reg, mp->context);
1306
1307       elt_index = elt->next;
1308       elt = pool_elt_at_index (tnm->list_pool, elt_index);
1309       session_index = elt->value;
1310     }
1311 }
1312
1313 static void
1314 vl_api_nat44_ei_user_session_v2_dump_t_handler (
1315   vl_api_nat44_ei_user_session_dump_t *mp)
1316 {
1317   vl_api_registration_t *reg;
1318   nat44_ei_main_t *nm = &nat44_ei_main;
1319   nat44_ei_main_per_thread_data_t *tnm;
1320   nat44_ei_session_t *s;
1321   clib_bihash_kv_8_8_t key, value;
1322   nat44_ei_user_key_t ukey;
1323   nat44_ei_user_t *u;
1324   u32 session_index, head_index, elt_index;
1325   dlist_elt_t *head, *elt;
1326   ip4_header_t ip;
1327
1328   reg = vl_api_client_index_to_registration (mp->client_index);
1329   if (!reg)
1330     return;
1331
1332   clib_memcpy (&ukey.addr, mp->ip_address, 4);
1333   ip.src_address.as_u32 = ukey.addr.as_u32;
1334   ukey.fib_index = fib_table_find (FIB_PROTOCOL_IP4, ntohl (mp->vrf_id));
1335   key.key = ukey.as_u64;
1336   if (nm->num_workers > 1)
1337     tnm = vec_elt_at_index (
1338       nm->per_thread_data,
1339       nat44_ei_get_in2out_worker_index (&ip, ukey.fib_index, 0));
1340   else
1341     tnm = vec_elt_at_index (nm->per_thread_data, nm->num_workers);
1342
1343   if (clib_bihash_search_8_8 (&tnm->user_hash, &key, &value))
1344     return;
1345   u = pool_elt_at_index (tnm->users, value.value);
1346   if (!u->nsessions && !u->nstaticsessions)
1347     return;
1348
1349   head_index = u->sessions_per_user_list_head_index;
1350   head = pool_elt_at_index (tnm->list_pool, head_index);
1351   elt_index = head->next;
1352   elt = pool_elt_at_index (tnm->list_pool, elt_index);
1353   session_index = elt->value;
1354   while (session_index != ~0)
1355     {
1356       s = pool_elt_at_index (tnm->sessions, session_index);
1357
1358       send_nat44_ei_user_session_v2_details (s, reg, mp->context);
1359
1360       elt_index = elt->next;
1361       elt = pool_elt_at_index (tnm->list_pool, elt_index);
1362       session_index = elt->value;
1363     }
1364 }
1365
1366 static void
1367 vl_api_nat44_ei_del_session_t_handler (vl_api_nat44_ei_del_session_t *mp)
1368 {
1369   nat44_ei_main_t *nm = &nat44_ei_main;
1370   vl_api_nat44_ei_del_session_reply_t *rmp;
1371   ip4_address_t addr, eh_addr;
1372   u16 port;
1373   u32 vrf_id;
1374   int rv = 0;
1375   u8 is_in;
1376   nat_protocol_t proto;
1377
1378   memcpy (&addr.as_u8, mp->address, 4);
1379   port = mp->port;
1380   vrf_id = clib_net_to_host_u32 (mp->vrf_id);
1381   proto = ip_proto_to_nat_proto (mp->protocol);
1382   memcpy (&eh_addr.as_u8, mp->ext_host_address, 4);
1383
1384   // is session inside ?
1385   is_in = mp->flags & NAT44_EI_IF_INSIDE;
1386
1387   rv = nat44_ei_del_session (nm, &addr, port, proto, vrf_id, is_in);
1388
1389   REPLY_MACRO (VL_API_NAT44_EI_DEL_SESSION_REPLY);
1390 }
1391
1392 static void
1393 vl_api_nat44_ei_forwarding_enable_disable_t_handler (
1394   vl_api_nat44_ei_forwarding_enable_disable_t *mp)
1395 {
1396   nat44_ei_main_t *nm = &nat44_ei_main;
1397   vl_api_nat44_ei_forwarding_enable_disable_reply_t *rmp;
1398   int rv = 0;
1399   u32 *ses_to_be_removed = 0, *ses_index;
1400   nat44_ei_main_per_thread_data_t *tnm;
1401   nat44_ei_session_t *s;
1402
1403   nm->forwarding_enabled = mp->enable != 0;
1404
1405   if (mp->enable == 0)
1406     {
1407       vec_foreach (tnm, nm->per_thread_data)
1408         {
1409           vec_foreach (ses_index, ses_to_be_removed)
1410             {
1411               s = pool_elt_at_index (tnm->sessions, ses_index[0]);
1412               nat44_ei_free_session_data (nm, s, tnm - nm->per_thread_data, 0);
1413               nat44_ei_delete_session (nm, s, tnm - nm->per_thread_data);
1414             }
1415
1416           vec_free (ses_to_be_removed);
1417         }
1418     }
1419
1420   REPLY_MACRO (VL_API_NAT44_EI_FORWARDING_ENABLE_DISABLE_REPLY);
1421 }
1422
1423 static void
1424 vl_api_nat44_ei_set_fq_options_t_handler (vl_api_nat44_ei_set_fq_options_t *mp)
1425 {
1426   nat44_ei_main_t *nm = &nat44_ei_main;
1427   vl_api_nat44_ei_set_fq_options_reply_t *rmp;
1428   int rv = 0;
1429   u32 frame_queue_nelts = ntohl (mp->frame_queue_nelts);
1430   rv = nat44_ei_set_frame_queue_nelts (frame_queue_nelts);
1431   REPLY_MACRO (VL_API_NAT44_EI_SET_FQ_OPTIONS_REPLY);
1432 }
1433
1434 static void
1435 vl_api_nat44_ei_show_fq_options_t_handler (
1436   vl_api_nat44_ei_show_fq_options_t *mp)
1437 {
1438   nat44_ei_main_t *nm = &nat44_ei_main;
1439   vl_api_nat44_ei_show_fq_options_reply_t *rmp;
1440   int rv = 0;
1441   /* clang-format off */
1442   REPLY_MACRO2_ZERO (VL_API_NAT44_EI_SHOW_FQ_OPTIONS_REPLY,
1443   ({
1444     rmp->frame_queue_nelts = htonl (nm->frame_queue_nelts);
1445   }));
1446   /* clang-format on */
1447 }
1448
1449 /* API definitions */
1450 #include <vnet/format_fns.h>
1451 #include <nat/nat44-ei/nat44_ei.api.c>
1452
1453 /* Set up the API message handling tables */
1454 clib_error_t *
1455 nat44_ei_api_hookup (vlib_main_t *vm)
1456 {
1457   nat44_ei_main_t *nm = &nat44_ei_main;
1458   nm->msg_id_base = setup_message_id_table ();
1459   return 0;
1460 }
1461
1462 /*
1463  * fd.io coding-style-patch-verification: ON
1464  *
1465  * Local Variables:
1466  * eval: (c-set-style "gnu")
1467  * End:
1468  */