src/vcl/vcom_socket.c

   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 #include <unistd.h>
  16 #include <stdio.h>
  17 #include <sys/uio.h>
  18 #include <limits.h>
  19 #define __need_IOV_MAX
  20 #include <bits/stdio_lim.h>
  21 #include <netinet/tcp.h>
  22
  23 #include <vppinfra/types.h>
  24 #include <vppinfra/hash.h>
  25 #include <vppinfra/pool.h>
  26
  27 #include <vcl/vcom_socket.h>
  28 #include <vcl/vcom_socket_wrapper.h>
  29 #include <vcl/vcom.h>
  30
  31 #include <vcl/vppcom.h>
  32
  33
  34 /*
  35  * VCOM_SOCKET Private definitions and functions.
  36  */
  37
  38 typedef struct vcom_socket_main_t_
  39 {
  40   u8 init;
  41
  42   /* vcom_socket pool */
  43   vcom_socket_t *vsockets;
  44
  45   /* Hash table for socketidx to fd mapping */
  46   uword *sockidx_by_fd;
  47
  48   /* vcom_epoll pool */
  49   vcom_epoll_t *vepolls;
  50
  51   /* Hash table for epollidx to epfd mapping */
  52   uword *epollidx_by_epfd;
  53
  54
  55   /* common epitem poll for all epfd */
  56   /* TBD: epitem poll per epfd */
  57   /* vcom_epitem pool */
  58   vcom_epitem_t *vepitems;
  59
  60   /* Hash table for epitemidx to epfdfd mapping */
  61   uword *epitemidx_by_epfdfd;
  62
  63   /* Hash table - key:epfd, value:vec of epitemidx */
  64   uword *epitemidxs_by_epfd;
  65   /* Hash table - key:fd, value:vec of epitemidx */
  66   uword *epitemidxs_by_fd;
  67
  68 } vcom_socket_main_t;
  69
  70 vcom_socket_main_t vcom_socket_main;
  71
  72
  73 static int
  74 vcom_socket_open_socket (int domain, int type, int protocol)
  75 {
  76   int rv = -1;
  77
  78   /* handle domains implemented by vpp */
  79   switch (domain)
  80     {
  81     case AF_INET:
  82     case AF_INET6:
  83       /* get socket type and
  84        * handle the socket types supported by vpp */
  85       switch (type & ~(SOCK_CLOEXEC | SOCK_NONBLOCK))
  86         {
  87         case SOCK_STREAM:
  88         case SOCK_DGRAM:
  89           /* the type argument serves a second purpose,
  90            * in addition to specifying a socket type,
  91            * it may include the bitwise OR of any of
  92            * SOCK_NONBLOCK and SOCK_CLOEXEC, to modify
  93            * the behavior of socket. */
  94           rv = libc_socket (domain, type, protocol);
  95           if (rv == -1)
  96             rv = -errno;
  97           break;
  98
  99         default:
 100           break;
 101         }
 102
 103       break;
 104
 105     default:
 106       break;
 107     }
 108
 109   return rv;
 110 }
 111
 112 static int
 113 vcom_socket_open_epoll (int flags)
 114 {
 115   int rv = -1;
 116
 117   if (flags < 0)
 118     {
 119       return -EINVAL;
 120     }
 121   if (flags && (flags & ~EPOLL_CLOEXEC))
 122     {
 123       return -EINVAL;
 124     }
 125
 126   /* flags can be either zero or EPOLL_CLOEXEC */
 127   rv = libc_epoll_create1 (flags);
 128   if (rv == -1)
 129     rv = -errno;
 130
 131   return rv;
 132 }
 133
 134 static int
 135 vcom_socket_close_socket (int fd)
 136 {
 137   int rv;
 138
 139   rv = libc_close (fd);
 140   if (rv == -1)
 141     rv = -errno;
 142
 143   return rv;
 144 }
 145
 146 static int
 147 vcom_socket_close_epoll (int epfd)
 148 {
 149   int rv;
 150
 151   rv = libc_close (epfd);
 152   if (rv == -1)
 153     rv = -errno;
 154
 155   return rv;
 156 }
 157
 158 /*
 159  * Public API functions
 160  */
 161
 162
 163 int
 164 vcom_socket_is_vcom_fd (int fd)
 165 {
 166   vcom_socket_main_t *vsm = &vcom_socket_main;
 167   uword *p;
 168   vcom_socket_t *vsock;
 169
 170   p = hash_get (vsm->sockidx_by_fd, fd);
 171
 172   if (p)
 173     {
 174       vsock = pool_elt_at_index (vsm->vsockets, p[0]);
 175       if (vsock && vsock->type == SOCKET_TYPE_VPPCOM_BOUND)
 176         return 1;
 177     }
 178   return 0;
 179 }
 180
 181 int
 182 vcom_socket_is_vcom_epfd (int epfd)
 183 {
 184   vcom_socket_main_t *vsm = &vcom_socket_main;
 185   uword *p;
 186   vcom_epoll_t *vepoll;
 187
 188   p = hash_get (vsm->epollidx_by_epfd, epfd);
 189
 190   if (p)
 191     {
 192       vepoll = pool_elt_at_index (vsm->vepolls, p[0]);
 193       if (vepoll && vepoll->type == EPOLL_TYPE_VPPCOM_BOUND)
 194         return 1;
 195     }
 196   return 0;
 197 }
 198
 199 static inline int
 200 vcom_socket_get_sid (int fd)
 201 {
 202   vcom_socket_main_t *vsm = &vcom_socket_main;
 203   uword *p;
 204   vcom_socket_t *vsock;
 205
 206   p = hash_get (vsm->sockidx_by_fd, fd);
 207
 208   if (p)
 209     {
 210       vsock = pool_elt_at_index (vsm->vsockets, p[0]);
 211       if (vsock && vsock->type == SOCKET_TYPE_VPPCOM_BOUND)
 212         return vsock->sid;
 213     }
 214   return INVALID_SESSION_ID;
 215 }
 216
 217 static inline int
 218 vcom_socket_get_vep_idx (int epfd)
 219 {
 220   vcom_socket_main_t *vsm = &vcom_socket_main;
 221   uword *p;
 222   vcom_epoll_t *vepoll;
 223
 224   p = hash_get (vsm->epollidx_by_epfd, epfd);
 225
 226   if (p)
 227     {
 228       vepoll = pool_elt_at_index (vsm->vepolls, p[0]);
 229       if (vepoll && vepoll->type == EPOLL_TYPE_VPPCOM_BOUND)
 230         return vepoll->vep_idx;
 231     }
 232   return INVALID_VEP_IDX;
 233 }
 234
 235 static inline int
 236 vcom_socket_get_sid_and_vsock (int fd, vcom_socket_t ** vsockp)
 237 {
 238   vcom_socket_main_t *vsm = &vcom_socket_main;
 239   uword *p;
 240   vcom_socket_t *vsock;
 241
 242   p = hash_get (vsm->sockidx_by_fd, fd);
 243
 244   if (p)
 245     {
 246       vsock = pool_elt_at_index (vsm->vsockets, p[0]);
 247       if (vsock && vsock->type == SOCKET_TYPE_VPPCOM_BOUND)
 248         {
 249           *vsockp = vsock;
 250           return vsock->sid;
 251         }
 252     }
 253   return INVALID_SESSION_ID;
 254 }
 255
 256 static inline int
 257 vcom_socket_get_vep_idx_and_vepoll (int epfd, vcom_epoll_t ** vepollp)
 258 {
 259   vcom_socket_main_t *vsm = &vcom_socket_main;
 260   uword *p;
 261   vcom_epoll_t *vepoll;
 262
 263   p = hash_get (vsm->epollidx_by_epfd, epfd);
 264
 265   if (p)
 266     {
 267       vepoll = pool_elt_at_index (vsm->vepolls, p[0]);
 268       if (vepoll && vepoll->type == EPOLL_TYPE_VPPCOM_BOUND)
 269         {
 270           *vepollp = vepoll;
 271           return vepoll->vep_idx;
 272         }
 273     }
 274   return INVALID_VEP_IDX;
 275 }
 276
 277
 278 static int
 279 vcom_socket_close_vepoll (int epfd)
 280 {
 281   int rv = -1;
 282   vcom_socket_main_t *vsm = &vcom_socket_main;
 283   uword *p;
 284   vcom_epoll_t *vepoll;
 285
 286   p = hash_get (vsm->epollidx_by_epfd, epfd);
 287   if (!p)
 288     return -EBADF;
 289
 290   vepoll = pool_elt_at_index (vsm->vepolls, p[0]);
 291   if (!vepoll)
 292     return -EBADF;
 293
 294   if (vepoll->type != EPOLL_TYPE_VPPCOM_BOUND)
 295     return -EINVAL;
 296
 297   if (vepoll->count)
 298     {
 299       if (!vepoll->close)
 300         {
 301           vepoll->close = 1;
 302           return 0;
 303         }
 304       else
 305         {
 306           return -EBADF;
 307         }
 308     }
 309
 310   /* count is zero */
 311   rv = vppcom_session_close (vepoll->vep_idx);
 312   rv = vcom_socket_close_epoll (vepoll->epfd);
 313
 314   vepoll_init (vepoll);
 315   hash_unset (vsm->epollidx_by_epfd, epfd);
 316   pool_put (vsm->vepolls, vepoll);
 317
 318   return rv;
 319 }
 320
 321 static int
 322 vcom_socket_close_vsock (int fd)
 323 {
 324   int rv = -1;
 325   vcom_socket_main_t *vsm = &vcom_socket_main;
 326   uword *p;
 327   vcom_socket_t *vsock;
 328
 329   vcom_epitem_t *vepitem;
 330
 331   i32 *vepitemidxs = 0;
 332   i32 *vepitemidxs_var = 0;
 333
 334   p = hash_get (vsm->sockidx_by_fd, fd);
 335   if (!p)
 336     return -EBADF;
 337
 338   vsock = pool_elt_at_index (vsm->vsockets, p[0]);
 339   if (!vsock)
 340     return -ENOTSOCK;
 341
 342   if (vsock->type != SOCKET_TYPE_VPPCOM_BOUND)
 343     return -EINVAL;
 344
 345   rv = vppcom_session_close (vsock->sid);
 346   rv = vcom_socket_close_socket (vsock->fd);
 347
 348   vsocket_init (vsock);
 349   hash_unset (vsm->sockidx_by_fd, fd);
 350   pool_put (vsm->vsockets, vsock);
 351
 352   /*
 353    * NOTE:
 354    * Before calling close(), user should remove
 355    * this fd from the epoll-set of all epoll instances,
 356    * otherwise resource(epitems) leaks ensues.
 357    */
 358
 359   /*
 360    * 00. close all epoll instances that are marked as "close"
 361    *     of which this fd is the "last" remaining member.
 362    * 01. epitems associated with this fd are intentionally
 363    *     not removed, see NOTE: above.
 364    * */
 365
 366   /* does this fd participate in epoll */
 367   p = hash_get (vsm->epitemidxs_by_fd, fd);
 368   if (p)
 369     {
 370       vepitemidxs = *(i32 **) p;
 371       vec_foreach (vepitemidxs_var, vepitemidxs)
 372       {
 373         vepitem = pool_elt_at_index (vsm->vepitems, vepitemidxs_var[0]);
 374         if (vepitem && vepitem->fd == fd &&
 375             vepitem->type == FD_TYPE_VCOM_SOCKET)
 376           {
 377             i32 vep_idx;
 378             vcom_epoll_t *vepoll;
 379             if ((vep_idx =
 380                  vcom_socket_get_vep_idx_and_vepoll (vepitem->epfd,
 381                                                      &vepoll)) !=
 382                 INVALID_VEP_IDX)
 383               {
 384                 if (vepoll->close)
 385                   {
 386                     if (vepoll->count == 1)
 387                       {
 388                         /*
 389                          * force count to zero and
 390                          * close this epoll instance
 391                          * */
 392                         vepoll->count = 0;
 393                         vcom_socket_close_vepoll (vepoll->epfd);
 394                       }
 395                     else
 396                       {
 397                         vepoll->count -= 1;
 398                       }
 399                   }
 400               }
 401           }
 402
 403       }
 404     }
 405
 406   return rv;
 407 }
 408
 409 int
 410 vcom_socket_close (int __fd)
 411 {
 412   int rv;
 413
 414   if (vcom_socket_is_vcom_fd (__fd))
 415     {
 416       rv = vcom_socket_close_vsock (__fd);
 417     }
 418   else if (vcom_socket_is_vcom_epfd (__fd))
 419     {
 420       rv = vcom_socket_close_vepoll (__fd);
 421     }
 422   else
 423     {
 424       rv = -EBADF;
 425     }
 426
 427   return rv;
 428 }
 429
 430 ssize_t
 431 vcom_socket_read (int __fd, void *__buf, size_t __nbytes)
 432 {
 433   int rv = -1;
 434   vcom_socket_main_t *vsm = &vcom_socket_main;
 435   uword *p;
 436   vcom_socket_t *vsock;
 437
 438   p = hash_get (vsm->sockidx_by_fd, __fd);
 439   if (!p)
 440     return -EBADF;
 441
 442   vsock = pool_elt_at_index (vsm->vsockets, p[0]);
 443   if (!vsock)
 444     return -ENOTSOCK;
 445
 446   if (vsock->type != SOCKET_TYPE_VPPCOM_BOUND)
 447     return -EINVAL;
 448
 449   if (!__buf)
 450     {
 451       return -EINVAL;
 452     }
 453
 454   rv = vcom_fcntl (__fd, F_GETFL, 0);
 455   if (rv < 0)
 456     {
 457       return rv;
 458
 459     }
 460
 461   /* is blocking */
 462   if (!(rv & O_NONBLOCK))
 463     {
 464       do
 465         {
 466           rv = vppcom_session_read (vsock->sid, __buf, __nbytes);
 467         }
 468       /* coverity[CONSTANT_EXPRESSION_RESULT] */
 469       while (rv == -EAGAIN || rv == -EWOULDBLOCK);
 470       return rv;
 471     }
 472   /* The file descriptor refers to a socket and has been
 473    * marked nonblocking(O_NONBLOCK) and the read would
 474    * block.
 475    * */
 476   /* is non blocking */
 477   rv = vppcom_session_read (vsock->sid, __buf, __nbytes);
 478   return rv;
 479 }
 480
 481 ssize_t
 482 vcom_socket_readv (int __fd, const struct iovec * __iov, int __iovcnt)
 483 {
 484   int rv;
 485   vcom_socket_main_t *vsm = &vcom_socket_main;
 486   uword *p;
 487   vcom_socket_t *vsock;
 488   ssize_t total = 0, len = 0;
 489   int i;
 490
 491   p = hash_get (vsm->sockidx_by_fd, __fd);
 492   if (!p)
 493     return -EBADF;
 494
 495   vsock = pool_elt_at_index (vsm->vsockets, p[0]);
 496   if (!vsock)
 497     return -ENOTSOCK;
 498
 499   if (vsock->type != SOCKET_TYPE_VPPCOM_BOUND)
 500     return -EINVAL;
 501
 502   if (__iov == 0 || __iovcnt == 0 || __iovcnt > IOV_MAX)
 503     return -EINVAL;
 504
 505   /* Sanity check */
 506   for (i = 0; i < __iovcnt; ++i)
 507     {
 508       if (SSIZE_MAX - len < __iov[i].iov_len)
 509         return -EINVAL;
 510       len += __iov[i].iov_len;
 511     }
 512
 513   rv = vcom_fcntl (__fd, F_GETFL, 0);
 514   if (rv < 0)
 515     {
 516       return rv;
 517     }
 518
 519   /* is blocking */
 520   if (!(rv & O_NONBLOCK))
 521     {
 522       do
 523         {
 524           for (i = 0; i < __iovcnt; ++i)
 525             {
 526               rv = vppcom_session_read (vsock->sid, __iov[i].iov_base,
 527                                         __iov[i].iov_len);
 528               if (rv < 0)
 529                 break;
 530               else
 531                 {
 532                   total += rv;
 533                   if (rv < __iov[i].iov_len)
 534                     /* Read less than buffer provided, no point to continue */
 535                     break;
 536                 }
 537             }
 538         }
 539       /* coverity[CONSTANT_EXPRESSION_RESULT] */
 540       while ((rv == -EAGAIN || rv == -EWOULDBLOCK) && total == 0);
 541       return total;
 542     }
 543
 544   /* is non blocking */
 545   for (i = 0; i < __iovcnt; ++i)
 546     {
 547       rv = vppcom_session_read (vsock->sid, __iov[i].iov_base,
 548                                 __iov[i].iov_len);
 549       if (rv < 0)
 550         {
 551           if (total > 0)
 552             break;
 553           else
 554             {
 555               errno = rv;
 556               return rv;
 557             }
 558         }
 559       else
 560         {
 561           total += rv;
 562           if (rv < __iov[i].iov_len)
 563             /* Read less than buffer provided, no point to continue */
 564             break;
 565         }
 566     }
 567   return total;
 568 }
 569
 570 ssize_t
 571 vcom_socket_write (int __fd, const void *__buf, size_t __n)
 572 {
 573   int rv = -1;
 574   vcom_socket_main_t *vsm = &vcom_socket_main;
 575   uword *p;
 576   vcom_socket_t *vsock;
 577
 578   if (!__buf)
 579     {
 580       return -EINVAL;
 581     }
 582
 583   p = hash_get (vsm->sockidx_by_fd, __fd);
 584   if (!p)
 585     return -EBADF;
 586
 587   vsock = pool_elt_at_index (vsm->vsockets, p[0]);
 588   if (!vsock)
 589     return -ENOTSOCK;
 590
 591   if (vsock->type != SOCKET_TYPE_VPPCOM_BOUND)
 592     return -EINVAL;
 593
 594   rv = vppcom_session_write (vsock->sid, (void *) __buf, __n);
 595   return rv;
 596 }
 597
 598 ssize_t
 599 vcom_socket_writev (int __fd, const struct iovec * __iov, int __iovcnt)
 600 {
 601   int rv = -1;
 602   ssize_t total = 0;
 603   vcom_socket_main_t *vsm = &vcom_socket_main;
 604   uword *p;
 605   vcom_socket_t *vsock;
 606   int i;
 607
 608   p = hash_get (vsm->sockidx_by_fd, __fd);
 609   if (!p)
 610     return -EBADF;
 611
 612   vsock = pool_elt_at_index (vsm->vsockets, p[0]);
 613   if (!vsock)
 614     return -ENOTSOCK;
 615
 616   if (vsock->type != SOCKET_TYPE_VPPCOM_BOUND)
 617     return -EINVAL;
 618
 619   if (__iov == 0 || __iovcnt == 0 || __iovcnt > IOV_MAX)
 620     return -EINVAL;
 621
 622   for (i = 0; i < __iovcnt; ++i)
 623     {
 624       rv = vppcom_session_write (vsock->sid, __iov[i].iov_base,
 625                                  __iov[i].iov_len);
 626       if (rv < 0)
 627         {
 628           if (total > 0)
 629             break;
 630           else
 631             return rv;
 632         }
 633       else
 634         total += rv;
 635     }
 636   return total;
 637 }
 638
 639 /*
 640  * RETURN:  0 - invalid cmd
 641  *          1 - cmd not handled by vcom and vppcom
 642  *          2 - cmd handled by vcom socket resource
 643  *          3 - cmd handled by vppcom
 644  * */
 645 /* TBD: incomplete list of cmd */
 646 static int
 647 vcom_socket_check_fcntl_cmd (int __cmd)
 648 {
 649   switch (__cmd)
 650     {
 651       /*cmd not handled by vcom and vppcom */
 652       /* Fallthrough */
 653     case F_DUPFD:
 654     case F_DUPFD_CLOEXEC:
 655       return 1;
 656
 657       /* cmd handled by vcom socket resource */
 658       /* Fallthrough */
 659     case F_GETFD:
 660     case F_SETFD:
 661     case F_GETLK:
 662     case F_SETLK:
 663     case F_SETLKW:
 664     case F_GETOWN:
 665     case F_SETOWN:
 666       return 2;
 667
 668       /* cmd handled by vcom and vppcom */
 669     case F_SETFL:
 670     case F_GETFL:
 671       return 3;
 672
 673       /* cmd not handled by vcom and vppcom */
 674     default:
 675       return 1;
 676     }
 677   return 0;
 678 }
 679
 680 static inline int
 681 vcom_session_fcntl_va (int __sid, int __cmd, va_list __ap)
 682 {
 683   int flags = va_arg (__ap, int);
 684   int rv = -EOPNOTSUPP;
 685   uint32_t size;
 686
 687   size = sizeof (flags);
 688   if (__cmd == F_SETFL)
 689     {
 690       rv = vppcom_session_attr (__sid, VPPCOM_ATTR_SET_FLAGS, &flags, &size);
 691     }
 692   else if (__cmd == F_GETFL)
 693     {
 694       rv = vppcom_session_attr (__sid, VPPCOM_ATTR_GET_FLAGS, &flags, &size);
 695       if (rv == VPPCOM_OK)
 696         rv = flags;
 697     }
 698
 699   return rv;
 700 }
 701
 702 int
 703 vcom_socket_fcntl_va (int __fd, int __cmd, va_list __ap)
 704 {
 705   int rv = -EBADF;
 706   vcom_socket_main_t *vsm = &vcom_socket_main;
 707   uword *p;
 708   vcom_socket_t *vsock;
 709
 710   p = hash_get (vsm->sockidx_by_fd, __fd);
 711   if (!p)
 712     return -EBADF;
 713
 714   vsock = pool_elt_at_index (vsm->vsockets, p[0]);
 715   if (!vsock)
 716     return -ENOTSOCK;
 717
 718   if (vsock->type != SOCKET_TYPE_VPPCOM_BOUND)
 719     return -EINVAL;
 720
 721   switch (vcom_socket_check_fcntl_cmd (__cmd))
 722     {
 723       /* invalid cmd */
 724     case 0:
 725       rv = -EBADF;
 726       break;
 727       /*cmd not handled by vcom and vppcom */
 728     case 1:
 729       rv = libc_vfcntl (vsock->fd, __cmd, __ap);
 730       break;
 731       /* cmd handled by vcom socket resource */
 732     case 2:
 733       rv = libc_vfcntl (vsock->fd, __cmd, __ap);
 734       break;
 735       /* cmd handled by vppcom */
 736     case 3:
 737       rv = vcom_session_fcntl_va (vsock->sid, __cmd, __ap);
 738       break;
 739
 740     default:
 741       rv = -EINVAL;
 742       break;
 743     }
 744
 745   return rv;
 746 }
 747
 748 /*
 749  * RETURN:  0 - invalid cmd
 750  *          1 - cmd not handled by vcom and vppcom
 751  *          2 - cmd handled by vcom socket resource
 752  *          3 - cmd handled by vppcom
 753  */
 754 static int
 755 vcom_socket_check_ioctl_cmd (unsigned long int __cmd)
 756 {
 757   int rc;
 758
 759   switch (__cmd)
 760     {
 761       /* cmd handled by vppcom */
 762     case FIONREAD:
 763       rc = 3;
 764       break;
 765
 766       /* cmd not handled by vcom and vppcom */
 767     default:
 768       rc = 1;
 769       break;
 770     }
 771   return rc;
 772 }
 773
 774 static inline int
 775 vcom_session_ioctl_va (int __sid, int __cmd, va_list __ap)
 776 {
 777   int rv;
 778
 779   if (__cmd == FIONREAD)
 780     rv = vppcom_session_attr (__sid, VPPCOM_ATTR_GET_NREAD, 0, 0);
 781   else
 782     rv = -EOPNOTSUPP;
 783   return rv;
 784 }
 785
 786 int
 787 vcom_socket_ioctl_va (int __fd, unsigned long int __cmd, va_list __ap)
 788 {
 789   int rv = -EBADF;
 790   vcom_socket_main_t *vsm = &vcom_socket_main;
 791   uword *p;
 792   vcom_socket_t *vsock;
 793
 794   p = hash_get (vsm->sockidx_by_fd, __fd);
 795   if (!p)
 796     return -EBADF;
 797
 798   vsock = pool_elt_at_index (vsm->vsockets, p[0]);
 799   if (!vsock)
 800     return -ENOTSOCK;
 801
 802   if (vsock->type != SOCKET_TYPE_VPPCOM_BOUND)
 803     return -EINVAL;
 804
 805   switch (vcom_socket_check_ioctl_cmd (__cmd))
 806     {
 807       /* Not supported cmd */
 808     case 0:
 809       rv = -EOPNOTSUPP;
 810       break;
 811
 812       /* cmd not handled by vcom and vppcom */
 813     case 1:
 814       rv = libc_vioctl (vsock->fd, __cmd, __ap);
 815       break;
 816
 817       /* cmd handled by vcom socket resource */
 818     case 2:
 819       rv = libc_vioctl (vsock->fd, __cmd, __ap);
 820       break;
 821
 822       /* cmd handled by vppcom */
 823     case 3:
 824       rv = vcom_session_ioctl_va (vsock->sid, __cmd, __ap);
 825       break;
 826
 827     default:
 828       rv = -EINVAL;
 829       break;
 830     }
 831
 832   return rv;
 833 }
 834
 835 static inline int
 836 vcom_socket_fds_2_sid_fds (
 837                             /* dest */
 838                             int *vcom_nsid_fds,
 839                             fd_set * __restrict vcom_rd_sid_fds,
 840                             fd_set * __restrict vcom_wr_sid_fds,
 841                             fd_set * __restrict vcom_ex_sid_fds,
 842                             /* src */
 843                             int vcom_nfds,
 844                             fd_set * __restrict vcom_readfds,
 845                             fd_set * __restrict vcom_writefds,
 846                             fd_set * __restrict vcom_exceptfds)
 847 {
 848   int rv = 0;
 849   int fd;
 850   int sid;
 851   /* invalid max_sid is -1 */
 852   int max_sid = -1;
 853   int nsid = 0;
 854
 855   /*
 856    *  set sid in sid sets corresponding to fd's in fd sets
 857    *  compute nsid and vcom_nsid_fds from sid sets
 858    */
 859
 860   for (fd = 0; fd < vcom_nfds; fd++)
 861     {
 862       /*
 863        * F fd set, src
 864        * S sid set, dest
 865        */
 866 #define _(S,F)                              \
 867       if ((F) && (S) && FD_ISSET (fd, (F))) \
 868         {                                   \
 869           sid = vcom_socket_get_sid (fd);   \
 870           if (sid != INVALID_SESSION_ID)    \
 871             {                               \
 872               FD_SET (sid, (S));            \
 873               if (sid > max_sid)            \
 874                 {                           \
 875                   max_sid = sid;            \
 876                 }                           \
 877               ++nsid;                       \
 878             }                               \
 879           else                              \
 880             {                               \
 881               rv = -EBADFD;                 \
 882               goto done;                    \
 883             }                               \
 884         }
 885
 886
 887       _(vcom_rd_sid_fds, vcom_readfds);
 888       _(vcom_wr_sid_fds, vcom_writefds);
 889       _(vcom_ex_sid_fds, vcom_exceptfds);
 890 #undef _
 891     }
 892
 893   *vcom_nsid_fds = max_sid != -1 ? max_sid + 1 : 0;
 894   rv = nsid;
 895
 896 done:
 897   return rv;
 898 }
 899
 900 /*
 901  * PRE: 00. sid sets were derived from fd sets
 902  *      01. sid sets were updated with sids that actually changed
 903  *          status
 904  *      02. fd sets still has watched fds
 905  *
 906  * This function will modify in place fd sets to indicate which fd's
 907  * actually changed status(inferred from sid sets)
 908  */
 909 static inline int
 910 vcom_socket_sid_fds_2_fds (
 911                             /* dest */
 912                             int *new_vcom_nfds,
 913                             int vcom_nfds,
 914                             fd_set * __restrict vcom_readfds,
 915                             fd_set * __restrict vcom_writefds,
 916                             fd_set * __restrict vcom_exceptfds,
 917                             /* src */
 918                             int vcom_nsid_fds,
 919                             fd_set * __restrict vcom_rd_sid_fds,
 920                             fd_set * __restrict vcom_wr_sid_fds,
 921                             fd_set * __restrict vcom_ex_sid_fds)
 922 {
 923   int rv = 0;
 924   int fd;
 925   int sid;
 926   /* invalid max_fd is -1 */
 927   int max_fd = -1;
 928   int nfd = 0;
 929
 930
 931   /*
 932    *  modify in place fd sets to indicate which fd's
 933    * actually changed status(inferred from sid sets)
 934    */
 935   for (fd = 0; fd < vcom_nfds; fd++)
 936     {
 937       /*
 938        * F fd set, dest
 939        * S sid set, src
 940        */
 941 #define _(S,F)                              \
 942       if ((F) && (S) && FD_ISSET (fd, (F))) \
 943         {                                   \
 944           sid = vcom_socket_get_sid (fd);   \
 945           if (sid != INVALID_SESSION_ID)    \
 946             {                               \
 947               if (!FD_ISSET (sid, (S)))     \
 948                 {                           \
 949                    FD_CLR(fd, (F));         \
 950                 }                           \
 951             }                               \
 952           else                              \
 953             {                               \
 954               rv = -EBADFD;                 \
 955               goto done;                    \
 956             }                               \
 957         }
 958
 959
 960       _(vcom_rd_sid_fds, vcom_readfds);
 961       _(vcom_wr_sid_fds, vcom_writefds);
 962       _(vcom_ex_sid_fds, vcom_exceptfds);
 963 #undef _
 964     }
 965
 966   /*
 967    *  compute nfd and new_vcom_nfds from fd sets
 968    */
 969   for (fd = 0; fd < vcom_nfds; fd++)
 970     {
 971
 972 #define _(F)                                \
 973       if ((F) && FD_ISSET (fd, (F)))        \
 974         {                                   \
 975           if (fd > max_fd)                  \
 976             {                               \
 977               max_fd = fd;                  \
 978             }                               \
 979           ++nfd;                            \
 980         }
 981
 982
 983       _(vcom_readfds);
 984       _(vcom_writefds);
 985       _(vcom_exceptfds);
 986 #undef _
 987
 988     }
 989
 990   *new_vcom_nfds = max_fd != -1 ? max_fd + 1 : 0;
 991   rv = nfd;
 992
 993 done:
 994   return rv;
 995 }
 996
 997 /*
 998  * PRE:
 999  * vom_socket_select is always called with
1000  * timeout->tv_sec and timeout->tv_usec set to zero.
1001  * hence vppcom_select return immediately.
1002  */
1003 /*
1004  * TBD: do{body;} while(timeout conditional); timeout loop
1005  */
1006 int
1007 vcom_socket_select (int vcom_nfds, fd_set * __restrict vcom_readfds,
1008                     fd_set * __restrict vcom_writefds,
1009                     fd_set * __restrict vcom_exceptfds,
1010                     struct timeval *__restrict timeout)
1011 {
1012   static unsigned long vcom_nsid_fds = 0;
1013   int vcom_nsid = 0;
1014   int rv = -EBADF;
1015   pid_t pid = getpid ();
1016
1017   int new_vcom_nfds = 0;
1018   int new_vcom_nfd = 0;
1019
1020   /* vcom sid fds */
1021   fd_set vcom_rd_sid_fds;
1022   fd_set vcom_wr_sid_fds;
1023   fd_set vcom_ex_sid_fds;
1024
1025   /* in seconds eg. 3.123456789 seconds */
1026   double time_to_wait = (double) 0;
1027
1028   /* validate inputs */
1029   if (vcom_nfds < 0)
1030     {
1031       return -EINVAL;
1032     }
1033
1034   /* convert timeval timeout to double time_to_wait */
1035   if (timeout)
1036     {
1037       if (timeout->tv_sec == 0 && timeout->tv_usec == 0)
1038         {
1039           /* polling: vppcom_select returns immediately */
1040           time_to_wait = (double) 0;
1041         }
1042       else
1043         {
1044           /*TBD:  use timeval api */
1045           time_to_wait = (double) timeout->tv_sec +
1046             (double) timeout->tv_usec / (double) 1000000 +
1047             (double) (timeout->tv_usec % 1000000) / (double) 1000000;
1048         }
1049     }
1050   else
1051     {
1052       /*
1053        * no timeout: vppcom_select can block indefinitely
1054        * waiting for a file descriptor to become ready
1055        * */
1056       /* set to a phantom value */
1057       time_to_wait = ~0;
1058     }
1059
1060   /* zero the sid_sets */
1061   /*
1062    * F fd set
1063    * S sid set
1064    */
1065 #define _(S,F)                          \
1066   if ((F))                              \
1067     {                                   \
1068       FD_ZERO ((S));                    \
1069     }
1070
1071
1072   _(&vcom_rd_sid_fds, vcom_readfds);
1073   _(&vcom_wr_sid_fds, vcom_writefds);
1074   _(&vcom_ex_sid_fds, vcom_exceptfds);
1075 #undef _
1076
1077   if (vcom_nfds == 0)
1078     {
1079       if (time_to_wait > 0)
1080         {
1081           if (VCOM_DEBUG > 0)
1082             fprintf (stderr,
1083                      "[%d] vcom_socket_select called to "
1084                      "emulate delay_ns()!\n", pid);
1085           rv = vppcom_select (0, NULL, NULL, NULL, time_to_wait);
1086         }
1087       else
1088         {
1089           fprintf (stderr, "[%d] vcom_socket_select called vcom_nfds = 0 "
1090                    "and invalid time_to_wait (%f)!\n", pid, time_to_wait);
1091         }
1092       return 0;
1093     }
1094
1095   /* populate read, write and except sid_sets */
1096   vcom_nsid = vcom_socket_fds_2_sid_fds (
1097                                           /* dest */
1098                                           vcom_readfds || vcom_writefds
1099                                           || vcom_exceptfds ? (int *)
1100                                           &vcom_nsid_fds : NULL,
1101                                           vcom_readfds ? &vcom_rd_sid_fds :
1102                                           NULL,
1103                                           vcom_writefds ? &vcom_wr_sid_fds :
1104                                           NULL,
1105                                           vcom_exceptfds ? &vcom_ex_sid_fds :
1106                                           NULL,
1107                                           /* src */
1108                                           vcom_nfds,
1109                                           vcom_readfds,
1110                                           vcom_writefds, vcom_exceptfds);
1111   if (vcom_nsid < 0)
1112     {
1113       return vcom_nsid;
1114     }
1115
1116   rv = vppcom_select (vcom_nsid_fds,
1117                       vcom_readfds ? (unsigned long *) &vcom_rd_sid_fds :
1118                       NULL,
1119                       vcom_writefds ? (unsigned long *) &vcom_wr_sid_fds :
1120                       NULL,
1121                       vcom_exceptfds ? (unsigned long *) &vcom_ex_sid_fds :
1122                       NULL, time_to_wait);
1123   if (VCOM_DEBUG > 2)
1124     fprintf (stderr, "[%d] called vppcom_select(): "
1125              "'%04d'='%04d'\n", pid, rv, (int) vcom_nsid_fds);
1126
1127   /* check if any file descriptors changed status */
1128   if (rv > 0)
1129     {
1130       /*
1131        * on exit, sets are modified in place to indicate which
1132        * file descriptors actually changed status
1133        * */
1134
1135       /*
1136        * comply with pre-condition
1137        * do not clear vcom fd sets befor calling
1138        * vcom_socket_sid_fds_2_fds
1139        */
1140       new_vcom_nfd = vcom_socket_sid_fds_2_fds (
1141                                                  /* dest */
1142                                                  &new_vcom_nfds,
1143                                                  vcom_nfds,
1144                                                  vcom_readfds,
1145                                                  vcom_writefds,
1146                                                  vcom_exceptfds,
1147                                                  /* src */
1148                                                  vcom_nsid_fds,
1149                                                  vcom_readfds ?
1150                                                  &vcom_rd_sid_fds : NULL,
1151                                                  vcom_writefds ?
1152                                                  &vcom_wr_sid_fds : NULL,
1153                                                  vcom_exceptfds ?
1154                                                  &vcom_ex_sid_fds : NULL);
1155       if (new_vcom_nfd < 0)
1156         {
1157           return new_vcom_nfd;
1158         }
1159       if (new_vcom_nfds < 0)
1160         {
1161           return -EINVAL;
1162         }
1163       rv = new_vcom_nfd;
1164     }
1165   return rv;
1166 }
1167
1168
1169 int
1170 vcom_socket_socket (int __domain, int __type, int __protocol)
1171 {
1172   int rv = -1;
1173   vcom_socket_main_t *vsm = &vcom_socket_main;
1174   vcom_socket_t *vsock;
1175
1176   i32 fd;
1177   i32 sid;
1178   i32 sockidx;
1179   u8 is_nonblocking = __type & SOCK_NONBLOCK ? 1 : 0;
1180   int type = __type & ~(SOCK_NONBLOCK | SOCK_CLOEXEC);
1181
1182   fd = vcom_socket_open_socket (__domain, __type, __protocol);
1183   if (fd < 0)
1184     {
1185       rv = fd;
1186       goto out;
1187     }
1188
1189   sid = vppcom_session_create (VPPCOM_VRF_DEFAULT,
1190                                (type == SOCK_DGRAM) ?
1191                                VPPCOM_PROTO_UDP : VPPCOM_PROTO_TCP,
1192                                is_nonblocking);
1193   if (sid < 0)
1194     {
1195       rv = sid;
1196       goto out_close_socket;
1197     }
1198
1199   pool_get (vsm->vsockets, vsock);
1200   vsocket_init (vsock);
1201
1202   sockidx = vsock - vsm->vsockets;
1203   hash_set (vsm->sockidx_by_fd, fd, sockidx);
1204
1205   vsocket_set (vsock, fd, sid, SOCKET_TYPE_VPPCOM_BOUND);
1206   return fd;
1207
1208 out_close_socket:
1209   vcom_socket_close_socket (fd);
1210 out:
1211   return rv;
1212 }
1213
1214 int
1215 vcom_socket_socketpair (int __domain, int __type, int __protocol,
1216                         int __fds[2])
1217 {
1218 /* TBD: */
1219   return 0;
1220 }
1221
1222 int
1223 vcom_socket_bind (int __fd, __CONST_SOCKADDR_ARG __addr, socklen_t __len)
1224 {
1225   int rv = -1;
1226   vcom_socket_main_t *vsm = &vcom_socket_main;
1227   uword *p;
1228   vcom_socket_t *vsock;
1229
1230   vppcom_endpt_t ep;
1231
1232   p = hash_get (vsm->sockidx_by_fd, __fd);
1233   if (!p)
1234     return -EBADF;
1235
1236   vsock = pool_elt_at_index (vsm->vsockets, p[0]);
1237   if (!vsock)
1238     return -ENOTSOCK;
1239
1240   if (vsock->type != SOCKET_TYPE_VPPCOM_BOUND)
1241     return -EINVAL;
1242
1243   if (!__addr)
1244     {
1245       return -EINVAL;
1246     }
1247
1248   ep.vrf = VPPCOM_VRF_DEFAULT;
1249   switch (__addr->sa_family)
1250     {
1251     case AF_INET:
1252       if (__len != sizeof (struct sockaddr_in))
1253         {
1254           return -EINVAL;
1255         }
1256       ep.is_ip4 = VPPCOM_IS_IP4;
1257       ep.ip = (u8 *) & ((const struct sockaddr_in *) __addr)->sin_addr;
1258       ep.port = (u16) ((const struct sockaddr_in *) __addr)->sin_port;
1259       break;
1260
1261     case AF_INET6:
1262       if (__len != sizeof (struct sockaddr_in6))
1263         {
1264           return -EINVAL;
1265         }
1266       ep.is_ip4 = VPPCOM_IS_IP6;
1267       ep.ip = (u8 *) & ((const struct sockaddr_in6 *) __addr)->sin6_addr;
1268       ep.port = (u16) ((const struct sockaddr_in6 *) __addr)->sin6_port;
1269       break;
1270
1271     default:
1272       return -1;
1273       break;
1274     }
1275
1276   rv = vppcom_session_bind (vsock->sid, &ep);
1277   return rv;
1278 }
1279
1280 static inline int
1281 vcom_session_getsockname (int sid, vppcom_endpt_t * ep)
1282 {
1283   int rv;
1284   uint32_t size = sizeof (*ep);
1285
1286   rv = vppcom_session_attr (sid, VPPCOM_ATTR_GET_LCL_ADDR, ep, &size);
1287   return rv;
1288 }
1289
1290 int
1291 vcom_socket_getsockname (int __fd, __SOCKADDR_ARG __addr,
1292                          socklen_t * __restrict __len)
1293 {
1294   int rv = -1;
1295   vcom_socket_main_t *vsm = &vcom_socket_main;
1296   uword *p;
1297   vcom_socket_t *vsock;
1298
1299
1300   p = hash_get (vsm->sockidx_by_fd, __fd);
1301   if (!p)
1302     return -EBADF;
1303
1304   vsock = pool_elt_at_index (vsm->vsockets, p[0]);
1305   if (!vsock)
1306     return -ENOTSOCK;
1307
1308   if (vsock->type != SOCKET_TYPE_VPPCOM_BOUND)
1309     return -EINVAL;
1310
1311   if (!__addr || !__len)
1312     return -EFAULT;
1313
1314   vppcom_endpt_t ep;
1315   ep.ip = (u8 *) & ((const struct sockaddr_in *) __addr)->sin_addr;
1316   rv = vcom_session_getsockname (vsock->sid, &ep);
1317   if (rv == 0)
1318     {
1319       if (ep.vrf == VPPCOM_VRF_DEFAULT)
1320         {
1321           __addr->sa_family = ep.is_ip4 == VPPCOM_IS_IP4 ? AF_INET : AF_INET6;
1322           switch (__addr->sa_family)
1323             {
1324             case AF_INET:
1325               ((struct sockaddr_in *) __addr)->sin_port = ep.port;
1326               *__len = sizeof (struct sockaddr_in);
1327               break;
1328
1329             case AF_INET6:
1330               ((struct sockaddr_in6 *) __addr)->sin6_port = ep.port;
1331               *__len = sizeof (struct sockaddr_in6);
1332               break;
1333
1334             default:
1335               break;
1336             }
1337         }
1338     }
1339
1340   return rv;
1341 }
1342
1343 int
1344 vcom_socket_connect (int __fd, __CONST_SOCKADDR_ARG __addr, socklen_t __len)
1345 {
1346   int rv = -1;
1347   vcom_socket_main_t *vsm = &vcom_socket_main;
1348   uword *p;
1349   vcom_socket_t *vsock;
1350
1351   vppcom_endpt_t ep;
1352
1353   p = hash_get (vsm->sockidx_by_fd, __fd);
1354   if (p)
1355     {
1356       vsock = pool_elt_at_index (vsm->vsockets, p[0]);
1357
1358       ep.vrf = VPPCOM_VRF_DEFAULT;
1359       switch (__addr->sa_family)
1360         {
1361         case AF_INET:
1362           ep.is_ip4 = VPPCOM_IS_IP4;
1363           ep.ip =
1364             (uint8_t *) & ((const struct sockaddr_in *) __addr)->sin_addr;
1365           ep.port =
1366             (uint16_t) ((const struct sockaddr_in *) __addr)->sin_port;
1367           break;
1368
1369         case AF_INET6:
1370           ep.is_ip4 = VPPCOM_IS_IP6;
1371           ep.ip =
1372             (uint8_t *) & ((const struct sockaddr_in6 *) __addr)->sin6_addr;
1373           ep.port =
1374             (uint16_t) ((const struct sockaddr_in6 *) __addr)->sin6_port;
1375           break;
1376
1377         default:
1378           return -1;
1379           break;
1380         }
1381
1382       rv = vppcom_session_connect (vsock->sid, &ep);
1383     }
1384   return rv;
1385 }
1386
1387 static inline int
1388 vcom_session_getpeername (int sid, vppcom_endpt_t * ep)
1389 {
1390   int rv;
1391   uint32_t size = sizeof (*ep);
1392
1393   rv = vppcom_session_attr (sid, VPPCOM_ATTR_GET_PEER_ADDR, ep, &size);
1394   return rv;
1395 }
1396
1397 int
1398 vcom_socket_getpeername (int __fd, __SOCKADDR_ARG __addr,
1399                          socklen_t * __restrict __len)
1400 {
1401   int rv = -1;
1402   vcom_socket_main_t *vsm = &vcom_socket_main;
1403   uword *p;
1404   vcom_socket_t *vsock;
1405
1406
1407   p = hash_get (vsm->sockidx_by_fd, __fd);
1408   if (!p)
1409     return -EBADF;
1410
1411   vsock = pool_elt_at_index (vsm->vsockets, p[0]);
1412   if (!vsock)
1413     return -ENOTSOCK;
1414
1415   if (vsock->type != SOCKET_TYPE_VPPCOM_BOUND)
1416     return -EINVAL;
1417
1418   if (!__addr || !__len)
1419     return -EFAULT;
1420
1421   vppcom_endpt_t ep;
1422   ep.ip = (u8 *) & ((const struct sockaddr_in *) __addr)->sin_addr;
1423   rv = vcom_session_getpeername (vsock->sid, &ep);
1424   if (rv == 0)
1425     {
1426       if (ep.vrf == VPPCOM_VRF_DEFAULT)
1427         {
1428           __addr->sa_family = ep.is_ip4 == VPPCOM_IS_IP4 ? AF_INET : AF_INET6;
1429           switch (__addr->sa_family)
1430             {
1431             case AF_INET:
1432               ((struct sockaddr_in *) __addr)->sin_port = ep.port;
1433               *__len = sizeof (struct sockaddr_in);
1434               break;
1435
1436             case AF_INET6:
1437               ((struct sockaddr_in6 *) __addr)->sin6_port = ep.port;
1438               *__len = sizeof (struct sockaddr_in6);
1439               break;
1440
1441             default:
1442               break;
1443             }
1444         }
1445     }
1446
1447   return rv;
1448 }
1449
1450 ssize_t
1451 vcom_socket_send (int __fd, const void *__buf, size_t __n, int __flags)
1452 {
1453   return vcom_socket_sendto (__fd, __buf, __n, __flags, NULL, 0);
1454 }
1455
1456 ssize_t
1457 vcom_socket_recv (int __fd, void *__buf, size_t __n, int __flags)
1458 {
1459   int rv = -1;
1460   rv = vcom_socket_recvfrom (__fd, __buf, __n, __flags, NULL, 0);
1461   return rv;
1462 }
1463
1464 /*
1465  * RETURN   1 if __fd is (SOCK_STREAM, SOCK_SEQPACKET),
1466  * 0 otherwise
1467  * */
1468 int
1469 vcom_socket_is_connection_mode_socket (int __fd)
1470 {
1471   int rv = -1;
1472   /* TBD define new vppcom api */
1473   vcom_socket_main_t *vsm = &vcom_socket_main;
1474   uword *p;
1475   vcom_socket_t *vsock;
1476
1477   int type;
1478   socklen_t optlen;
1479
1480   p = hash_get (vsm->sockidx_by_fd, __fd);
1481
1482   if (p)
1483     {
1484       vsock = pool_elt_at_index (vsm->vsockets, p[0]);
1485       if (vsock && vsock->type == SOCKET_TYPE_VPPCOM_BOUND)
1486         {
1487           optlen = sizeof (type);
1488           rv = libc_getsockopt (__fd, SOL_SOCKET, SO_TYPE, &type, &optlen);
1489           if (rv != 0)
1490             {
1491               return 0;
1492             }
1493           /* get socket type */
1494           switch (type & ~(SOCK_CLOEXEC | SOCK_NONBLOCK))
1495             {
1496             case SOCK_STREAM:
1497             case SOCK_SEQPACKET:
1498               return 1;
1499               break;
1500
1501             default:
1502               return 0;
1503               break;
1504             }
1505         }
1506     }
1507   return 0;
1508 }
1509
1510 static inline ssize_t
1511 vcom_session_sendto (int __sid, void *__buf, size_t __n,
1512                      int __flags, __CONST_SOCKADDR_ARG __addr,
1513                      socklen_t __addr_len)
1514 {
1515   int rv = -1;
1516   vppcom_endpt_t ep;
1517
1518   ep.vrf = VPPCOM_VRF_DEFAULT;
1519   switch (__addr->sa_family)
1520     {
1521     case AF_INET:
1522       ep.is_ip4 = VPPCOM_IS_IP4;
1523       ep.ip = (uint8_t *) & ((const struct sockaddr_in *) __addr)->sin_addr;
1524       ep.port = (uint16_t) ((const struct sockaddr_in *) __addr)->sin_port;
1525       break;
1526
1527     case AF_INET6:
1528       ep.is_ip4 = VPPCOM_IS_IP6;
1529       ep.ip = (uint8_t *) & ((const struct sockaddr_in6 *) __addr)->sin6_addr;
1530       ep.port = (uint16_t) ((const struct sockaddr_in6 *) __addr)->sin6_port;
1531       break;
1532
1533     default:
1534       return -1;
1535     }
1536
1537   rv = vppcom_session_sendto (__sid, __buf, __n, __flags, &ep);
1538
1539   return rv;
1540 }
1541
1542 ssize_t
1543 vcom_socket_sendto (int __fd, const void *__buf, size_t __n,
1544                     int __flags, __CONST_SOCKADDR_ARG __addr,
1545                     socklen_t __addr_len)
1546 {
1547   int rv = -1;
1548   vcom_socket_main_t *vsm = &vcom_socket_main;
1549   uword *p;
1550   vcom_socket_t *vsock;
1551
1552   if (!__buf)
1553     {
1554       return -EINVAL;
1555     }
1556
1557   p = hash_get (vsm->sockidx_by_fd, __fd);
1558   if (!p)
1559     return -EBADF;
1560
1561   vsock = pool_elt_at_index (vsm->vsockets, p[0]);
1562   if (!vsock)
1563     return -ENOTSOCK;
1564
1565   if (vsock->type != SOCKET_TYPE_VPPCOM_BOUND)
1566     {
1567       return -EINVAL;
1568     }
1569
1570   if (vcom_socket_is_connection_mode_socket (__fd))
1571     {
1572       /* ignore __addr and _addr_len */
1573       /* and EISCONN may be returned when they are not NULL and 0 */
1574       if ((__addr != NULL) || (__addr_len != 0))
1575         {
1576           return -EISCONN;
1577         }
1578     }
1579   else
1580     {
1581       if (!__addr)
1582         {
1583           return -EDESTADDRREQ;
1584         }
1585       /* not a vppcom supported address family */
1586       if (!((__addr->sa_family == AF_INET) ||
1587             (__addr->sa_family == AF_INET6)))
1588         {
1589           return -EINVAL;
1590         }
1591     }
1592
1593   rv = vcom_session_sendto (vsock->sid, (void *) __buf, (int) __n,
1594                             __flags, __addr, __addr_len);
1595   return rv;
1596 }
1597
1598 static inline ssize_t
1599 vcom_session_recvfrom (int __sid, void *__restrict __buf, size_t __n,
1600                        int __flags, __SOCKADDR_ARG __addr,
1601                        socklen_t * __restrict __addr_len)
1602 {
1603   int rv;
1604   vppcom_endpt_t ep;
1605
1606   if (__addr)
1607     {
1608       ep.ip = (u8 *) & ((const struct sockaddr_in *) __addr)->sin_addr;
1609       rv = vppcom_session_recvfrom (__sid, __buf, __n, __flags, &ep);
1610
1611       if (rv > 0)
1612         {
1613           if (ep.vrf == VPPCOM_VRF_DEFAULT)
1614             {
1615               __addr->sa_family =
1616                 ep.is_ip4 == VPPCOM_IS_IP4 ? AF_INET : AF_INET6;
1617               switch (__addr->sa_family)
1618                 {
1619                 case AF_INET:
1620                   ((struct sockaddr_in *) __addr)->sin_port = ep.port;
1621                   *__addr_len = sizeof (struct sockaddr_in);
1622                   break;
1623
1624                 case AF_INET6:
1625                   ((struct sockaddr_in6 *) __addr)->sin6_port = ep.port;
1626                   *__addr_len = sizeof (struct sockaddr_in6);
1627                   break;
1628
1629                 default:
1630                   rv = -1;
1631                   break;
1632                 }
1633             }
1634           else
1635             rv = -1;
1636         }
1637     }
1638   else
1639     rv = vppcom_session_recvfrom (__sid, __buf, __n, __flags, NULL);
1640
1641   return rv;
1642 }
1643
1644 ssize_t
1645 vcom_socket_recvfrom (int __fd, void *__restrict __buf, size_t __n,
1646                       int __flags, __SOCKADDR_ARG __addr,
1647                       socklen_t * __restrict __addr_len)
1648 {
1649   int rv = -1;
1650   vcom_socket_main_t *vsm = &vcom_socket_main;
1651   uword *p;
1652   vcom_socket_t *vsock;
1653
1654   if (__addr && !__addr_len)
1655     return -EINVAL;
1656
1657   p = hash_get (vsm->sockidx_by_fd, __fd);
1658   if (!p)
1659     return -EBADF;
1660
1661   vsock = pool_elt_at_index (vsm->vsockets, p[0]);
1662   if (!vsock)
1663     return -ENOTSOCK;
1664
1665   if (vsock->type != SOCKET_TYPE_VPPCOM_BOUND)
1666     {
1667       return -EINVAL;
1668     }
1669
1670   rv = vcom_session_recvfrom (vsock->sid, __buf, __n,
1671                               __flags, __addr, __addr_len);
1672   return rv;
1673 }
1674
1675 /* TBD: move it to vppcom */
1676 static inline ssize_t
1677 vcom_session_sendmsg (int __sid, const struct msghdr *__message, int __flags)
1678 {
1679   int rv = -1;
1680   /* rv = vppcom_session_write (__sid, (void *) __message->__buf,
1681      (int)__n); */
1682   return rv;
1683 }
1684
1685 ssize_t
1686 vcom_socket_sendmsg (int __fd, const struct msghdr * __message, int __flags)
1687 {
1688   int rv = -1;
1689   vcom_socket_main_t *vsm = &vcom_socket_main;
1690   uword *p;
1691   vcom_socket_t *vsock;
1692
1693   p = hash_get (vsm->sockidx_by_fd, __fd);
1694   if (!p)
1695     return -EBADF;
1696
1697   vsock = pool_elt_at_index (vsm->vsockets, p[0]);
1698   if (!vsock)
1699     return -ENOTSOCK;
1700
1701   if (vcom_socket_is_connection_mode_socket (__fd))
1702     {
1703       /* ignore __addr and _addr_len */
1704       /* and EISCONN may be returned when they are not NULL and 0 */
1705       if ((__message->msg_name != NULL) || (__message->msg_namelen != 0))
1706         {
1707           return -EISCONN;
1708         }
1709     }
1710   else
1711     {
1712       /* TBD: validate __message->msg_name and __message->msg_namelen
1713        * and return -EINVAL on validation error
1714        * */
1715       ;
1716     }
1717
1718   rv = vcom_session_sendmsg (vsock->sid, __message, __flags);
1719
1720   return rv;
1721 }
1722
1723 #ifdef __USE_GNU
1724 int
1725 vcom_socket_sendmmsg (int __fd, struct mmsghdr *__vmessages,
1726                       unsigned int __vlen, int __flags)
1727 {
1728
1729   /* TBD: define a new vppcom api */
1730   return 0;
1731 }
1732 #endif
1733
1734 /* TBD: move it to vppcom */
1735 static inline ssize_t
1736 vcom_session_recvmsg (int __sid, struct msghdr *__message, int __flags)
1737 {
1738   int rv = -1;
1739   /* rv = vppcom_session_read (__sid, (void *) __message->__buf,
1740      (int)__n); */
1741   rv = -EOPNOTSUPP;
1742   return rv;
1743 }
1744
1745 ssize_t
1746 vcom_socket_recvmsg (int __fd, struct msghdr * __message, int __flags)
1747 {
1748   int rv = -1;
1749   vcom_socket_main_t *vsm = &vcom_socket_main;
1750   uword *p;
1751   vcom_socket_t *vsock;
1752
1753   p = hash_get (vsm->sockidx_by_fd, __fd);
1754   if (!p)
1755     return -EBADF;
1756
1757   vsock = pool_elt_at_index (vsm->vsockets, p[0]);
1758   if (!vsock)
1759     return -ENOTSOCK;
1760
1761   if (vsock->type != SOCKET_TYPE_VPPCOM_BOUND)
1762     return -EINVAL;
1763
1764   if (!__message)
1765     {
1766       return -EINVAL;
1767     }
1768
1769   /* validate __flags */
1770
1771   rv = vcom_session_recvmsg (vsock->sid, __message, __flags);
1772   return rv;
1773 }
1774
1775 #ifdef __USE_GNU
1776 int
1777 vcom_socket_recvmmsg (int __fd, struct mmsghdr *__vmessages,
1778                       unsigned int __vlen, int __flags,
1779                       struct timespec *__tmo)
1780 {
1781   /* TBD: define a new vppcom api */
1782   return 0;
1783 }
1784 #endif
1785
1786 /* TBD: move it to vppcom */
1787 static inline int
1788 vcom_session_get_sockopt (int __sid, int __level, int __optname,
1789                           void *__restrict __optval,
1790                           socklen_t * __restrict __optlen)
1791 {
1792   int rv = 0;
1793
1794   /* 1. for socket level options that are NOT socket attributes
1795    *    and that has corresponding vpp options get from vppcom */
1796   switch (__level)
1797     {
1798     case SOL_SOCKET:
1799       switch (__optname)
1800         {
1801         case SO_ERROR:
1802           *(int *) __optval = 0;
1803           break;
1804         default:
1805           break;
1806         }
1807     default:
1808       break;
1809     }
1810   /* 2. unhandled options */
1811   return rv;
1812 }
1813
1814 int
1815 vcom_socket_getsockopt (int __fd, int __level, int __optname,
1816                         void *__restrict __optval,
1817                         socklen_t * __restrict __optlen)
1818 {
1819   int rv = -1;
1820   vcom_socket_main_t *vsm = &vcom_socket_main;
1821   uword *p;
1822   vcom_socket_t *vsock;
1823
1824   p = hash_get (vsm->sockidx_by_fd, __fd);
1825   if (!p)
1826     return -EBADF;
1827
1828   vsock = pool_elt_at_index (vsm->vsockets, p[0]);
1829   if (!vsock)
1830     return -ENOTSOCK;
1831
1832   if (vsock->type != SOCKET_TYPE_VPPCOM_BOUND)
1833     return -EINVAL;
1834
1835   if (!__optval && !__optlen)
1836     return -EFAULT;
1837
1838   switch (__level)
1839     {
1840     case SOL_SOCKET:
1841       switch (__optname)
1842         {
1843 /*
1844  *  1. for socket level options that are socket attributes,
1845  *     get from libc_getsockopt.
1846  *  2. for socket level options that are NOT socket
1847  *     attributes and that has corresponding vpp options
1848  *     get from vppcom.
1849  *  3. for socket level options unimplemented
1850  *     return -ENOPROTOOPT */
1851         case SO_DEBUG:
1852         case SO_DONTROUTE:
1853         case SO_BROADCAST:
1854         case SO_SNDBUF:
1855         case SO_RCVBUF:
1856         case SO_REUSEADDR:
1857         case SO_REUSEPORT:
1858         case SO_KEEPALIVE:
1859         case SO_TYPE:
1860         case SO_PROTOCOL:
1861         case SO_DOMAIN:
1862         case SO_OOBINLINE:
1863         case SO_NO_CHECK:
1864         case SO_PRIORITY:
1865         case SO_LINGER:
1866         case SO_BSDCOMPAT:
1867         case SO_TIMESTAMP:
1868         case SO_TIMESTAMPNS:
1869         case SO_TIMESTAMPING:
1870         case SO_RCVTIMEO:
1871         case SO_SNDTIMEO:
1872         case SO_RCVLOWAT:
1873         case SO_SNDLOWAT:
1874         case SO_PASSCRED:
1875         case SO_PEERCRED:
1876         case SO_PEERNAME:
1877         case SO_ACCEPTCONN:
1878         case SO_PASSSEC:
1879         case SO_PEERSEC:
1880         case SO_MARK:
1881         case SO_RXQ_OVFL:
1882         case SO_WIFI_STATUS:
1883         case SO_PEEK_OFF:
1884         case SO_NOFCS:
1885         case SO_BINDTODEVICE:
1886         case SO_GET_FILTER:
1887         case SO_LOCK_FILTER:
1888         case SO_BPF_EXTENSIONS:
1889         case SO_SELECT_ERR_QUEUE:
1890 #ifdef CONFIG_NET_RX_BUSY_POLL
1891         case SO_BUSY_POLL:
1892 #endif
1893         case SO_MAX_PACING_RATE:
1894 #ifdef SO_INCOMING_CPU
1895         case SO_INCOMING_CPU:
1896 #endif
1897           rv = libc_getsockopt (__fd, __level, __optname, __optval, __optlen);
1898           if (rv != 0)
1899             {
1900               rv = -errno;
1901               return rv;
1902             }
1903           break;
1904
1905         case SO_ERROR:
1906           rv = vcom_session_get_sockopt (vsock->sid, __level, __optname,
1907                                          __optval, __optlen);
1908           break;
1909
1910         default:
1911           /* We implement the SO_SNDLOWAT etc to not be settable
1912            * (1003.1g 7).
1913            */
1914           return -ENOPROTOOPT;
1915         }
1916
1917       break;
1918
1919     default:
1920       /* 1. handle options that are NOT socket level options,
1921        *    but have corresponding vpp otions. */
1922       rv = vcom_session_get_sockopt (vsock->sid, __level, __optname,
1923                                      __optval, __optlen);
1924       break;
1925     }
1926
1927   return rv;
1928 }
1929
1930 /* TBD: move it to vppcom */
1931 static inline int
1932 vcom_session_setsockopt (int __sid, int __level, int __optname,
1933                          const void *__optval, socklen_t __optlen)
1934 {
1935   int rv = -EOPNOTSUPP;
1936
1937   switch (__level)
1938     {
1939     case SOL_TCP:
1940       switch (__optname)
1941         {
1942         case TCP_KEEPIDLE:
1943           rv =
1944             vppcom_session_attr (__sid, VPPCOM_ATTR_SET_TCP_KEEPIDLE, 0, 0);
1945           break;
1946         case TCP_KEEPINTVL:
1947           rv =
1948             vppcom_session_attr (__sid, VPPCOM_ATTR_SET_TCP_KEEPINTVL, 0, 0);
1949           break;
1950         default:
1951           break;
1952         }
1953       break;
1954     case SOL_IPV6:
1955       switch (__optname)
1956         {
1957         case IPV6_V6ONLY:
1958           rv = vppcom_session_attr (__sid, VPPCOM_ATTR_SET_V6ONLY, 0, 0);
1959           break;
1960         default:
1961           break;
1962         }
1963       break;
1964     case SOL_SOCKET:
1965       switch (__optname)
1966         {
1967         case SO_KEEPALIVE:
1968           rv = vppcom_session_attr (__sid, VPPCOM_ATTR_SET_KEEPALIVE, 0, 0);
1969           break;
1970         case SO_REUSEADDR:
1971           rv = vppcom_session_attr (__sid, VPPCOM_ATTR_SET_REUSEADDR, 0, 0);
1972           break;
1973         case SO_BROADCAST:
1974           rv = vppcom_session_attr (__sid, VPPCOM_ATTR_SET_BROADCAST, 0, 0);
1975           break;
1976         default:
1977           break;
1978         }
1979       break;
1980     default:
1981       break;
1982     }
1983
1984   return rv;
1985 }
1986
1987 int
1988 vcom_socket_setsockopt (int __fd, int __level, int __optname,
1989                         const void *__optval, socklen_t __optlen)
1990 {
1991   int rv = -1;
1992   vcom_socket_main_t *vsm = &vcom_socket_main;
1993   uword *p;
1994   vcom_socket_t *vsock;
1995
1996   p = hash_get (vsm->sockidx_by_fd, __fd);
1997   if (!p)
1998     return -EBADF;
1999
2000   vsock = pool_elt_at_index (vsm->vsockets, p[0]);
2001   if (!vsock)
2002     return -ENOTSOCK;
2003
2004   if (vsock->type != SOCKET_TYPE_VPPCOM_BOUND)
2005     return -EINVAL;
2006
2007   /*
2008    *      Options without arguments
2009    */
2010
2011   if (__optname == SO_BINDTODEVICE)
2012     {
2013       rv = libc_setsockopt (__fd, __level, __optname, __optval, __optlen);
2014       if (rv != 0)
2015         {
2016           rv = -errno;
2017         }
2018       return rv;
2019     }
2020
2021   if (!__optval)
2022     return -EFAULT;
2023
2024   if (__optlen < sizeof (int))
2025     return -EINVAL;
2026
2027   switch (__level)
2028     {
2029     case SOL_IPV6:
2030       switch (__optname)
2031         {
2032         case IPV6_V6ONLY:
2033           rv = vcom_session_setsockopt (vsock->sid, __level, __optname,
2034                                         __optval, __optlen);
2035           break;
2036         default:
2037           return -EOPNOTSUPP;
2038         }
2039       break;
2040     case SOL_TCP:
2041       switch (__optname)
2042         {
2043         case TCP_NODELAY:
2044           return 0;
2045         case TCP_KEEPIDLE:
2046         case TCP_KEEPINTVL:
2047           rv = vcom_session_setsockopt (vsock->sid, __level, __optname,
2048                                         __optval, __optlen);
2049           break;
2050         default:
2051           return -EOPNOTSUPP;
2052         }
2053       break;
2054       /* handle options at socket level */
2055     case SOL_SOCKET:
2056       switch (__optname)
2057         {
2058         case SO_REUSEADDR:
2059         case SO_BROADCAST:
2060         case SO_KEEPALIVE:
2061           rv = vcom_session_setsockopt (vsock->sid, __level, __optname,
2062                                         __optval, __optlen);
2063           break;
2064
2065           /*
2066            * 1. for socket level options that are socket attributes,
2067            *    set it from libc_getsockopt
2068            * 2. for socket level options that are NOT socket
2069            *    attributes and that has corresponding vpp options
2070            *    set it from vppcom
2071            * 3. for socket level options unimplemented
2072            *    return -ENOPROTOOPT */
2073         case SO_DEBUG:
2074         case SO_DONTROUTE:
2075         case SO_SNDBUF:
2076         case SO_RCVBUF:
2077         case SO_REUSEPORT:
2078         case SO_TYPE:
2079         case SO_PROTOCOL:
2080         case SO_DOMAIN:
2081         case SO_ERROR:
2082         case SO_OOBINLINE:
2083         case SO_NO_CHECK:
2084         case SO_PRIORITY:
2085         case SO_LINGER:
2086         case SO_BSDCOMPAT:
2087         case SO_TIMESTAMP:
2088         case SO_TIMESTAMPNS:
2089         case SO_TIMESTAMPING:
2090         case SO_RCVTIMEO:
2091         case SO_SNDTIMEO:
2092         case SO_RCVLOWAT:
2093         case SO_SNDLOWAT:
2094         case SO_PASSCRED:
2095         case SO_PEERCRED:
2096         case SO_PEERNAME:
2097         case SO_ACCEPTCONN:
2098         case SO_PASSSEC:
2099         case SO_PEERSEC:
2100         case SO_MARK:
2101         case SO_RXQ_OVFL:
2102         case SO_WIFI_STATUS:
2103         case SO_PEEK_OFF:
2104         case SO_NOFCS:
2105           /*
2106            * SO_BINDTODEVICE already handled as
2107            * "Options without arguments" */
2108           /* case SO_BINDTODEVICE: */
2109         case SO_GET_FILTER:
2110         case SO_LOCK_FILTER:
2111         case SO_BPF_EXTENSIONS:
2112         case SO_SELECT_ERR_QUEUE:
2113 #ifdef CONFIG_NET_RX_BUSY_POLL
2114         case SO_BUSY_POLL:
2115 #endif
2116         case SO_MAX_PACING_RATE:
2117 #ifdef SO_INCOMING_CPU
2118         case SO_INCOMING_CPU:
2119 #endif
2120           rv = libc_setsockopt (__fd, __level, __optname, __optval, __optlen);
2121           if (rv != 0)
2122             {
2123               rv = -errno;
2124               return rv;
2125             }
2126           break;
2127
2128         default:
2129           /* We implement the SO_SNDLOWAT etc to not be settable
2130            * (1003.1g 7).
2131            */
2132           return -ENOPROTOOPT;
2133         }
2134
2135       break;
2136
2137     default:
2138       return -ENOPROTOOPT;
2139     }
2140
2141   return rv;
2142 }
2143
2144 int
2145 vcom_socket_listen (int __fd, int __n)
2146 {
2147   int rv = -1;
2148   vcom_socket_main_t *vsm = &vcom_socket_main;
2149   uword *p;
2150   vcom_socket_t *vsock;
2151
2152   p = hash_get (vsm->sockidx_by_fd, __fd);
2153   if (p)
2154     {
2155       vsock = pool_elt_at_index (vsm->vsockets, p[0]);
2156
2157       /* TBD vppcom to accept __n parameter */
2158       rv = vppcom_session_listen (vsock->sid, __n);
2159     }
2160
2161   return rv;
2162 }
2163
2164 static int
2165 vcom_socket_connected_socket (int __fd, int __sid,
2166                               int *__domain,
2167                               int *__type, int *__protocol, int flags)
2168 {
2169   int rv = -1;
2170   vcom_socket_main_t *vsm = &vcom_socket_main;
2171   vcom_socket_t *vsock;
2172
2173   i32 fd;
2174   i32 sockidx;
2175
2176   socklen_t optlen;
2177
2178   optlen = sizeof (*__domain);
2179   rv = libc_getsockopt (__fd, SOL_SOCKET, SO_DOMAIN, __domain, &optlen);
2180   if (rv != 0)
2181     {
2182       rv = -errno;
2183       goto out;
2184     }
2185
2186   optlen = sizeof (*__type);
2187   rv = libc_getsockopt (__fd, SOL_SOCKET, SO_TYPE, __type, &optlen);
2188   if (rv != 0)
2189     {
2190       rv = -errno;
2191       goto out;
2192     }
2193
2194   optlen = sizeof (*__protocol);
2195   rv = libc_getsockopt (__fd, SOL_SOCKET, SO_PROTOCOL, __protocol, &optlen);
2196   if (rv != 0)
2197     {
2198       rv = -errno;
2199       goto out;
2200     }
2201
2202   fd = vcom_socket_open_socket (*__domain, *__type | flags, *__protocol);
2203   if (fd < 0)
2204     {
2205       rv = fd;
2206       goto out;
2207     }
2208
2209   pool_get (vsm->vsockets, vsock);
2210   vsocket_init (vsock);
2211
2212   sockidx = vsock - vsm->vsockets;
2213   hash_set (vsm->sockidx_by_fd, fd, sockidx);
2214
2215   vsocket_set (vsock, fd, __sid, SOCKET_TYPE_VPPCOM_BOUND);
2216   return fd;
2217
2218 out:
2219   return rv;
2220 }
2221
2222 /* If flag is 0, then accept4() is the same as accept().
2223  * SOCK_NONBLOCK and SOCK_CLOEXEC can be bitwise ORed in flags
2224  */
2225 static int
2226 vcom_socket_accept_flags (int __fd, __SOCKADDR_ARG __addr,
2227                           socklen_t * __restrict __addr_len, int flags)
2228 {
2229   int rv = -1;
2230   vcom_socket_main_t *vsm = &vcom_socket_main;
2231   uword *p;
2232   vcom_socket_t *vsock;
2233
2234   int fd;
2235   int sid;
2236   int domain;
2237   int type;
2238   int protocol;
2239
2240   uint8_t addr8[sizeof (struct in6_addr)];
2241   vppcom_endpt_t ep;
2242
2243   ep.ip = addr8;
2244
2245   /* validate flags */
2246
2247   /*
2248    * for documentation
2249    *  switch (flags)
2250    *   {
2251    *   case 0:
2252    *   case SOCK_NONBLOCK:
2253    *   case SOCK_CLOEXEC:
2254    *   case SOCK_NONBLOCK | SOCK_CLOEXEC:
2255    *     break;
2256    *
2257    *   default:
2258    *     return -1;
2259    *   }
2260    */
2261   /* flags can be 0 or can be bitwise OR
2262    * of any of SOCK_NONBLOCK and SOCK_CLOEXEC */
2263
2264   if (!(!flags || (flags & (SOCK_NONBLOCK | SOCK_CLOEXEC))))
2265     {
2266       /* TBD: return proper error code */
2267       return -1;
2268     }
2269
2270   /* TBD: return proper error code */
2271
2272   if (!vcom_socket_is_connection_mode_socket (__fd))
2273     {
2274       return -EOPNOTSUPP;
2275     }
2276
2277   p = hash_get (vsm->sockidx_by_fd, __fd);
2278   if (p)
2279     {
2280       vsock = pool_elt_at_index (vsm->vsockets, p[0]);
2281
2282
2283       rv = vcom_fcntl (vsock->fd, F_GETFL, 0);
2284       if (rv < 0)
2285         {
2286           return rv;
2287         }
2288
2289       /* is blocking */
2290       if (!(rv & O_NONBLOCK))
2291         {
2292           /* socket is not marked as nonblocking
2293            * and no pending connections are present
2294            * on the queue, accept () blocks the caller
2295            * until a connection is present.
2296            */
2297           rv = vppcom_session_accept (vsock->sid, &ep,
2298                                       -1.0 /* wait forever */ );
2299         }
2300       else
2301         {
2302           /* The file descriptor refers to a socket and has been
2303            * marked nonblocking(O_NONBLOCK) and the accept would
2304            * block.
2305            * */
2306           /* is non blocking */
2307           rv = vppcom_session_accept (vsock->sid, &ep, 0);
2308           /* If the socket is marked nonblocking and
2309            * no pending connections are present on the
2310            * queue, accept fails with the error
2311            * EAGAIN or EWOULDBLOCK
2312            */
2313           if (rv == VPPCOM_ETIMEDOUT)
2314             {
2315               rv = VPPCOM_EAGAIN;
2316             }
2317         }
2318       if (rv < 0)
2319         {
2320           return rv;
2321         }
2322
2323       sid = rv;
2324
2325       /* create a new connected socket resource and set flags
2326        * on the new file descriptor.
2327        * update vsockets and sockidx_by_fd table
2328        * */
2329       fd = vcom_socket_connected_socket (__fd, sid,
2330                                          &domain, &type, &protocol, flags);
2331       if (fd < 0)
2332         {
2333           return fd;
2334         }
2335
2336       rv = fd;
2337
2338       /* TBD populate __addr and __addr_len */
2339       /* TBD: The returned address is truncated if the buffer
2340        * provided is too small, in this case, __addr_len will
2341        * return a value greater than was supplied to the call.*/
2342       if (__addr)
2343         {
2344           if (ep.is_cut_thru)
2345             {
2346               /* TBD populate __addr and __addr_len */
2347               switch (domain)
2348                 {
2349                 case AF_INET:
2350                   ((struct sockaddr_in *) __addr)->sin_family = AF_INET;
2351                   ((struct sockaddr_in *) __addr)->sin_port = ep.port;
2352                   memcpy (&((struct sockaddr_in *) __addr)->sin_addr,
2353                           addr8, sizeof (struct in_addr));
2354                   /* TBD: populate __addr_len */
2355                   if (__addr_len)
2356                     {
2357                       *__addr_len = sizeof (struct sockaddr_in);
2358                     }
2359                   break;
2360
2361                 case AF_INET6:
2362                   ((struct sockaddr_in6 *) __addr)->sin6_family = AF_INET6;
2363                   ((struct sockaddr_in6 *) __addr)->sin6_port = ep.port;
2364                   memcpy (((struct sockaddr_in6 *) __addr)->sin6_addr.
2365                           __in6_u.__u6_addr8, addr8,
2366                           sizeof (struct in6_addr));
2367                   /* TBD: populate __addr_len */
2368                   if (__addr_len)
2369                     {
2370                       *__addr_len = sizeof (struct sockaddr_in6);
2371                     }
2372                   break;
2373
2374                 default:
2375                   return -EAFNOSUPPORT;
2376                 }
2377             }
2378           else
2379             {
2380               switch (ep.is_ip4)
2381                 {
2382                 case VPPCOM_IS_IP4:
2383                   ((struct sockaddr_in *) __addr)->sin_family = AF_INET;
2384                   ((struct sockaddr_in *) __addr)->sin_port = ep.port;
2385                   memcpy (&((struct sockaddr_in *) __addr)->sin_addr,
2386                           addr8, sizeof (struct in_addr));
2387                   /* TBD: populate __addr_len */
2388                   if (__addr_len)
2389                     {
2390                       *__addr_len = sizeof (struct sockaddr_in);
2391                     }
2392                   break;
2393
2394                 case VPPCOM_IS_IP6:
2395                   ((struct sockaddr_in6 *) __addr)->sin6_family = AF_INET6;
2396                   ((struct sockaddr_in6 *) __addr)->sin6_port = ep.port;
2397                   memcpy (((struct sockaddr_in6 *) __addr)->sin6_addr.
2398                           __in6_u.__u6_addr8, addr8,
2399                           sizeof (struct in6_addr));
2400                   /* TBD: populate __addr_len */
2401                   if (__addr_len)
2402                     {
2403                       *__addr_len = sizeof (struct sockaddr_in6);
2404                     }
2405                   break;
2406
2407                 default:
2408                   return -EAFNOSUPPORT;
2409                 }
2410             }
2411         }
2412       else
2413         {
2414           /* when __addr is NULL, nothing is filled in,
2415            * in this case, __addr_len is not used,
2416            * and should also be null
2417            * */
2418           if (__addr_len)
2419             {
2420               /* TBD: return proper error code */
2421               return -1;
2422             }
2423         }
2424     }
2425
2426   return rv;
2427 }
2428
2429 int
2430 vcom_socket_accept (int __fd, __SOCKADDR_ARG __addr,
2431                     socklen_t * __restrict __addr_len)
2432 {
2433   /* set flags to 0 for accept() */
2434   return vcom_socket_accept_flags (__fd, __addr, __addr_len, 0);
2435 }
2436
2437 #ifdef __USE_GNU
2438 int
2439 vcom_socket_accept4 (int __fd, __SOCKADDR_ARG __addr,
2440                      socklen_t * __restrict __addr_len, int __flags)
2441 {
2442   /*  SOCK_NONBLOCK and SOCK_CLOEXEC can be bitwise ORed in flags */
2443   return vcom_socket_accept_flags (__fd, __addr, __addr_len, __flags);
2444 }
2445 #endif
2446
2447 /* TBD: move it to vppcom */
2448 static inline int
2449 vcom_session_shutdown (int __fd, int __how)
2450 {
2451   return 0;
2452 }
2453
2454 int
2455 vcom_socket_shutdown (int __fd, int __how)
2456 {
2457   int rv = -1;
2458   vcom_socket_main_t *vsm = &vcom_socket_main;
2459   uword *p;
2460   vcom_socket_t *vsock;
2461
2462   p = hash_get (vsm->sockidx_by_fd, __fd);
2463   if (p)
2464     {
2465       vsock = pool_elt_at_index (vsm->vsockets, p[0]);
2466       switch (__how)
2467         {
2468         case SHUT_RD:
2469         case SHUT_WR:
2470         case SHUT_RDWR:
2471           rv = vcom_session_shutdown (vsock->sid, __how);
2472           return rv;
2473           break;
2474
2475         default:
2476           return -EINVAL;
2477           break;
2478         }
2479     }
2480
2481   return rv;
2482 }
2483
2484 int
2485 vcom_socket_epoll_create1 (int __flags)
2486 {
2487   int rv = -1;
2488   vcom_socket_main_t *vsm = &vcom_socket_main;
2489   vcom_epoll_t *vepoll;
2490
2491   i32 epfd;
2492   i32 vep_idx;
2493   i32 epollidx;
2494
2495   epfd = vcom_socket_open_epoll (__flags);
2496   if (epfd < 0)
2497     {
2498       rv = epfd;
2499       goto out;
2500     }
2501
2502   vep_idx = vppcom_epoll_create ();
2503   if (vep_idx < 0)
2504     {
2505       rv = vep_idx;
2506       goto out_close_epoll;
2507     }
2508
2509   pool_get (vsm->vepolls, vepoll);
2510   vepoll_init (vepoll);
2511
2512   epollidx = vepoll - vsm->vepolls;
2513   hash_set (vsm->epollidx_by_epfd, epfd, epollidx);
2514
2515   vepoll_set (vepoll, epfd, vep_idx, EPOLL_TYPE_VPPCOM_BOUND, __flags, 0, 0);
2516
2517   return epfd;
2518
2519 out_close_epoll:
2520   vcom_socket_close_epoll (epfd);
2521 out:
2522   return rv;
2523 }
2524
2525 /*
2526  * PRE: vppcom_epoll_ctl() is successful
2527  * free_vepitem_on_del : 0 - no_pool_put, 1 - pool_put
2528  */
2529 int
2530 vcom_socket_ctl_vepitem (int __epfd, int __op, int __fd,
2531                          struct epoll_event *__event,
2532                          i32 vep_idx, vcom_epoll_t * vepoll,
2533                          i32 vfd_id, void *vfd, vcom_fd_type_t type,
2534                          int free_vepitem_on_del)
2535 {
2536   int rv = -1;
2537   vcom_socket_main_t *vsm = &vcom_socket_main;
2538   vcom_epitem_t *vepitem;
2539
2540   vcom_epitem_key_t epfdfd = {.epfd = __epfd,.fd = __fd };
2541   uword *p;
2542   i32 vepitemidx;
2543
2544   i32 *vepitemidxs = 0;
2545
2546   struct epoll_event revent = {.events = 0,.data.fd = INVALID_FD };
2547
2548   i32 vec_idx;
2549
2550   /* perform control operations on the epoll instance */
2551   switch (__op)
2552     {
2553     case EPOLL_CTL_ADD:
2554       /*
2555        * supplied file descriptor is already
2556        * registered with this epoll instance
2557        * */
2558       /* vepitem exists */
2559       p = hash_get (vsm->epitemidx_by_epfdfd, epfdfd.key);
2560       if (p)
2561         {
2562           rv = -EEXIST;
2563           goto out;
2564         }
2565
2566       /* add a new vepitem */
2567       pool_get (vsm->vepitems, vepitem);
2568       vepitem_init (vepitem);
2569
2570       vepitemidx = vepitem - vsm->vepitems;
2571       hash_set (vsm->epitemidx_by_epfdfd, epfdfd.key, vepitemidx);
2572       vepitem_set (vepitem, __epfd, __fd, __fd, __fd, type, *__event, revent);
2573
2574       /* update epitemidxs */
2575       /* by_epfd */
2576       p = hash_get (vsm->epitemidxs_by_epfd, __epfd);
2577       if (!p)                   /*  not exist */
2578         {
2579           vepitemidxs = 0;
2580           vec_add1 (vepitemidxs, vepitemidx);
2581           hash_set (vsm->epitemidxs_by_epfd, __epfd, vepitemidxs);
2582         }
2583       else                      /*  exists */
2584         {
2585           vepitemidxs = *(i32 **) p;
2586           vec_add1 (vepitemidxs, vepitemidx);
2587           hash_set3 (vsm->epitemidxs_by_epfd, __epfd, vepitemidxs, 0);
2588         }
2589       /* update epitemidxs */
2590       /* by_fd */
2591       p = hash_get (vsm->epitemidxs_by_fd, __fd);
2592       if (!p)                   /*  not exist */
2593         {
2594           vepitemidxs = 0;
2595           vec_add1 (vepitemidxs, vepitemidx);
2596           hash_set (vsm->epitemidxs_by_fd, __fd, vepitemidxs);
2597         }
2598       else                      /*  exists */
2599         {
2600           vepitemidxs = *(i32 **) p;
2601           vec_add1 (vepitemidxs, vepitemidx);
2602           hash_set3 (vsm->epitemidxs_by_fd, __fd, vepitemidxs, 0);
2603         }
2604
2605       /* increment vepoll fd count by 1 */
2606       vepoll->count += 1;
2607
2608       rv = 0;
2609       goto out;
2610       break;
2611
2612     case EPOLL_CTL_MOD:
2613       /*
2614        * supplied file descriptor is not
2615        * registered with this epoll instance
2616        * */
2617       /* vepitem not exist */
2618       p = hash_get (vsm->epitemidx_by_epfdfd, epfdfd.key);
2619       if (!p)
2620         {
2621           rv = -ENOENT;
2622           goto out;
2623         }
2624       vepitem = pool_elt_at_index (vsm->vepitems, p[0]);
2625       if (vepitem)
2626         {
2627           vepitem->event = *__event;
2628           vepitem->revent = revent;
2629         }
2630
2631       rv = 0;
2632       goto out;
2633       break;
2634
2635     case EPOLL_CTL_DEL:
2636       /*
2637        * supplied file descriptor is not
2638        * registered with this epoll instance
2639        * */
2640       /* vepitem not exist */
2641       p = hash_get (vsm->epitemidx_by_epfdfd, epfdfd.key);
2642       if (!p)
2643         {
2644           rv = -ENOENT;
2645           goto out;
2646         }
2647       vepitemidx = *(i32 *) p;
2648       hash_unset (vsm->epitemidx_by_epfdfd, epfdfd.key);
2649
2650       /* update epitemidxs */
2651       /* by_epfd */
2652       p = hash_get (vsm->epitemidxs_by_epfd, __epfd);
2653       if (!p)                   /*  not exist */
2654         {
2655           rv = -ENOENT;
2656           goto out;
2657         }
2658       else                      /*  exists */
2659         {
2660           vepitemidxs = *(i32 **) p;
2661           vec_idx = vec_search (vepitemidxs, vepitemidx);
2662           if (vec_idx != ~0)
2663             {
2664               vec_del1 (vepitemidxs, vec_idx);
2665               if (!vec_len (vepitemidxs))
2666                 {
2667                   vec_free (vepitemidxs);
2668                   hash_unset (vsm->epitemidxs_by_epfd, __epfd);
2669                 }
2670             }
2671         }
2672
2673       /* update epitemidxs */
2674       /* by_fd */
2675       p = hash_get (vsm->epitemidxs_by_fd, __fd);
2676       if (!p)                   /*  not exist */
2677         {
2678           rv = -ENOENT;
2679           goto out;
2680         }
2681       else                      /*  exists */
2682         {
2683           vepitemidxs = *(i32 **) p;
2684           vec_idx = vec_search (vepitemidxs, vepitemidx);
2685           if (vec_idx != ~0)
2686             {
2687               vec_del1 (vepitemidxs, vec_idx);
2688               if (!vec_len (vepitemidxs))
2689                 {
2690                   vec_free (vepitemidxs);
2691                   hash_unset (vsm->epitemidxs_by_fd, __fd);
2692                 }
2693             }
2694         }
2695
2696       /* pool put vepitem */
2697       vepitem = pool_elt_at_index (vsm->vepitems, vepitemidx);
2698       if (free_vepitem_on_del)
2699         {
2700           if (!vepitem)
2701             {
2702               rv = -ENOENT;
2703               goto out;
2704             }
2705           vepitem_init (vepitem);
2706           pool_put (vsm->vepitems, vepitem);
2707         }
2708       else
2709         {
2710           if (!vepitem)
2711             {
2712               vepitem_init (vepitem);
2713             }
2714         }
2715
2716       /* decrement vepoll fd count by 1 */
2717       vepoll->count -= 1;
2718
2719       rv = 0;
2720       goto out;
2721       break;
2722
2723     default:
2724       rv = -EINVAL;
2725       goto out;
2726       break;
2727     }
2728
2729 out:
2730   return rv;
2731 }
2732
2733 /*
2734  * PRE: 00. null pointer check on __event
2735  *      01. all other parameters are validated
2736  */
2737
2738 static int
2739 vcom_socket_epoll_ctl_internal (int __epfd, int __op, int __fd,
2740                                 struct epoll_event *__event,
2741                                 int free_vepitem_on_del)
2742 {
2743   int rv = -1;
2744
2745   /* vcom_socket_main_t *vsm = &vcom_socket_main; */
2746   vcom_epoll_t *vepoll;
2747
2748   /*__fd could could be vcom socket or vcom epoll or kernel fd */
2749   void *vfd;
2750   vcom_epoll_t *vfd_vepoll;
2751   vcom_socket_t *vfd_vsock;
2752
2753   i32 vep_idx;
2754   i32 vfd_id;
2755
2756   vcom_fd_type_t type = FD_TYPE_INVALID;
2757
2758   /* validate __event */
2759
2760   /* get vep_idx and vepoll */
2761   vep_idx = vcom_socket_get_vep_idx_and_vepoll (__epfd, &vepoll);
2762   if (vep_idx == INVALID_VEP_IDX)
2763     {
2764       return -EBADF;
2765     }
2766
2767   /* get vcom fd type, vfd_id and vfd */
2768   vfd_id = vcom_socket_get_sid_and_vsock (__fd, &vfd_vsock);
2769   if (vfd_id != INVALID_SESSION_ID)
2770     {
2771       type = FD_TYPE_VCOM_SOCKET;
2772       vfd = vfd_vsock;
2773     }
2774   else if ((vfd_id = vcom_socket_get_vep_idx_and_vepoll (__fd, &vfd_vepoll))
2775            != INVALID_VEP_IDX)
2776     {
2777       type = FD_TYPE_EPOLL;
2778       vfd = vfd_vepoll;
2779     }
2780   else
2781     {
2782       /* FD_TYPE_KERNEL not supported by epoll instance */
2783       type = FD_TYPE_INVALID;
2784       return -EBADF;
2785     }
2786
2787
2788   /* vepoll and vsock are now valid */
2789   rv = vppcom_epoll_ctl (vep_idx, __op, vfd_id, __event);
2790   if (rv < 0)
2791     {
2792       return rv;
2793     }
2794
2795   rv = vcom_socket_ctl_vepitem (__epfd, __op, __fd,
2796                                 __event,
2797                                 vep_idx, vepoll,
2798                                 vfd_id, vfd, type, free_vepitem_on_del);
2799   return rv;
2800 }
2801
2802 int
2803 vcom_socket_epoll_ctl (int __epfd, int __op, int __fd,
2804                        struct epoll_event *__event)
2805 {
2806   int rv = -1;
2807
2808   rv = vcom_socket_epoll_ctl_internal (__epfd, __op, __fd, __event, 1);
2809   return rv;
2810 }
2811
2812 static int
2813 vcom_socket_epoll_ctl1 (int __epfd, int __op, int __fd,
2814                         struct epoll_event *__event)
2815 {
2816   int rv = -1;
2817
2818   rv = vcom_socket_epoll_ctl_internal (__epfd, __op, __fd, __event, 0);
2819   return rv;
2820 }
2821
2822 int
2823 vcom_socket_epoll_pwait (int __epfd, struct epoll_event *__events,
2824                          int __maxevents, int __timeout,
2825                          const __sigset_t * __ss)
2826 {
2827   int rv = -EBADF;
2828
2829   /* in seconds eg. 3.123456789 seconds */
2830   double time_to_wait = (double) 0;
2831
2832   i32 vep_idx;
2833
2834   /* validate __event */
2835   if (!__events)
2836     {
2837       rv = -EFAULT;
2838       goto out;
2839     }
2840
2841   /* validate __timeout */
2842   if (__timeout > 0)
2843     {
2844       time_to_wait = (double) __timeout / (double) 1000;
2845     }
2846   else if (__timeout == 0)
2847     {
2848       time_to_wait = (double) 0;
2849     }
2850   else if (__timeout == -1)
2851     {
2852       time_to_wait = ~0;
2853     }
2854   else
2855     {
2856       rv = -EBADF;
2857       goto out;
2858     }
2859
2860   /* get vep_idx */
2861   vep_idx = vcom_socket_get_vep_idx (__epfd);
2862   if (vep_idx != INVALID_VEP_IDX)
2863     {
2864       rv = vppcom_epoll_wait (vep_idx, __events, __maxevents, time_to_wait);
2865     }
2866 out:
2867   return rv;
2868 }
2869
2870 static inline void
2871 vcom_pollfds_2_selectfds (
2872                            /* src */
2873                            struct pollfd *__fds, nfds_t __nfds,
2874                            /* dest */
2875                            int vcom_nfds,
2876                            fd_set * __restrict vcom_readfds,
2877                            fd_set * __restrict vcom_writefds,
2878                            fd_set * __restrict vcom_exceptfds)
2879 {
2880   nfds_t fds_idx = 0;
2881
2882   for (fds_idx = 0; fds_idx < __nfds; fds_idx++)
2883     {
2884       /* ignore negative fds */
2885       if (__fds[fds_idx].fd < 0)
2886         {
2887           continue;
2888         }
2889
2890       /* for POLLRDHUP, POLLERR, POLLHUP and  POLLNVAL */
2891       FD_SET (__fds[fds_idx].fd, vcom_exceptfds);
2892
2893       /* requested events */
2894       if (__fds[fds_idx].events)
2895         {
2896           if (__fds[fds_idx].events & POLLIN)
2897             {
2898               FD_SET (__fds[fds_idx].fd, vcom_readfds);
2899             }
2900           if (__fds[fds_idx].events & POLLPRI)
2901             {
2902               FD_SET (__fds[fds_idx].fd, vcom_readfds);
2903             }
2904           if (__fds[fds_idx].events & POLLOUT)
2905             {
2906               FD_SET (__fds[fds_idx].fd, vcom_writefds);
2907             }
2908 #if defined __USE_XOPEN || defined __USE_XOPEN2K8
2909           if (__fds[fds_idx].events & POLLRDNORM)
2910             {
2911               FD_SET (__fds[fds_idx].fd, vcom_readfds);
2912             }
2913           if (__fds[fds_idx].events & POLLRDBAND)
2914             {
2915               FD_SET (__fds[fds_idx].fd, vcom_readfds);
2916             }
2917           if (__fds[fds_idx].events & POLLWRNORM)
2918             {
2919               FD_SET (__fds[fds_idx].fd, vcom_writefds);
2920             }
2921           if (__fds[fds_idx].events & POLLWRBAND)
2922             {
2923               FD_SET (__fds[fds_idx].fd, vcom_writefds);
2924             }
2925 #endif
2926         }
2927     }                           /* for (fds_idx = 0; fds_idx < __nfds; fds_idx++) */
2928 }
2929
2930 static inline void
2931 vcom_selectfds_2_pollfds (
2932                            /* dest */
2933                            struct pollfd *__fds, nfds_t __nfds, int *nfd,
2934                            /* src */
2935                            int vcom_nfds,
2936                            fd_set * __restrict vcom_readfds,
2937                            fd_set * __restrict vcom_writefds,
2938                            fd_set * __restrict vcom_exceptfds)
2939 {
2940   nfds_t fds_idx = 0;
2941
2942
2943   for (fds_idx = 0; fds_idx < __nfds; fds_idx++)
2944     {
2945       /* ignore negative fds */
2946       if (__fds[fds_idx].fd < 0)
2947         {
2948           __fds[fds_idx].revents = 0;
2949         }
2950
2951       /* for POLLRDHUP, POLLERR, POLLHUP and  POLLNVAL */
2952       if (FD_ISSET (__fds[fds_idx].fd, vcom_exceptfds))
2953         {
2954           /*
2955            * TBD: for now any select exception
2956            *      is flagged as POLLERR
2957            * */
2958           __fds[fds_idx].revents |= POLLERR;
2959         }
2960
2961       /* requested events */
2962       if (__fds[fds_idx].events & POLLIN)
2963         {
2964           if (FD_ISSET (__fds[fds_idx].fd, vcom_readfds))
2965             {
2966               __fds[fds_idx].revents |= POLLIN;
2967             }
2968         }
2969       if (__fds[fds_idx].events & POLLPRI)
2970         {
2971           if (FD_ISSET (__fds[fds_idx].fd, vcom_readfds))
2972             {
2973               __fds[fds_idx].revents |= POLLIN;
2974             }
2975         }
2976       if (__fds[fds_idx].events & POLLOUT)
2977         {
2978           if (FD_ISSET (__fds[fds_idx].fd, vcom_writefds))
2979             {
2980               __fds[fds_idx].revents |= POLLOUT;
2981             }
2982         }
2983 #if defined __USE_XOPEN || defined __USE_XOPEN2K8
2984       if (__fds[fds_idx].events & POLLRDNORM)
2985         {
2986           if (FD_ISSET (__fds[fds_idx].fd, vcom_readfds))
2987             {
2988               __fds[fds_idx].revents |= POLLRDNORM;
2989             }
2990         }
2991       if (__fds[fds_idx].events & POLLRDBAND)
2992         {
2993           if (FD_ISSET (__fds[fds_idx].fd, vcom_readfds))
2994             {
2995               __fds[fds_idx].revents |= POLLRDBAND;
2996             }
2997         }
2998       if (__fds[fds_idx].events & POLLWRNORM)
2999         {
3000           if (FD_ISSET (__fds[fds_idx].fd, vcom_writefds))
3001             {
3002               __fds[fds_idx].revents |= POLLWRNORM;
3003             }
3004         }
3005       if (__fds[fds_idx].events & POLLWRBAND)
3006         {
3007           if (FD_ISSET (__fds[fds_idx].fd, vcom_writefds))
3008             {
3009               __fds[fds_idx].revents |= POLLWRBAND;
3010             }
3011         }
3012 #endif
3013     }                           /* for (fds_idx = 0; fds_idx < __nfds; fds_idx++) */
3014
3015   /*
3016    * nfd:
3017    * the number of structures which have nonzero revents  fields
3018    * (in  other  words,  those  descriptors  with events or
3019    * errors reported)
3020    * */
3021   *nfd = 0;
3022   for (fds_idx = 0; fds_idx < __nfds; fds_idx++)
3023     {
3024       /* ignore negative fds */
3025       if (__fds[fds_idx].fd < 0)
3026         {
3027           continue;
3028         }
3029
3030       if (__fds[fds_idx].revents)
3031         {
3032           (*nfd)++;
3033         }
3034     }
3035 }
3036
3037 /*
3038  * PRE: parameters are validated,
3039  *      vcom_socket_poll is always called with __timeout set to zero
3040  *      hence returns immediately
3041  *
3042  * ACTION: handle non negative validated vcom fds and ignore rest
3043  */
3044
3045 /*
3046  * implements vcom_socket_poll () interface
3047  *
3048  * internally uses vcom_socket_select ()
3049  * to realize the behavior
3050  * */
3051 int
3052 vcom_socket_poll_select_impl (struct pollfd *__fds, nfds_t __nfds,
3053                               int __timeout)
3054 {
3055   int rv;
3056   pid_t pid = getpid ();
3057
3058   nfds_t fds_idx = 0;
3059   int nfd = 0;
3060
3061   /* vcom */
3062   int vcom_nfds = 0;
3063   fd_set vcom_readfds;
3064   fd_set vcom_writefds;
3065   fd_set vcom_exceptfds;
3066   int vcom_nfd = -1;
3067   /* invalid max_vcom_fd is -1 */
3068   int max_vcom_fd = -1;
3069
3070   /* __timeout is zero to get ready events and return immediately */
3071   struct timeval tv = {.tv_sec = 0,.tv_usec = 0 };
3072
3073   /* validate __nfds from select perspective */
3074   if (__nfds > FD_SETSIZE)
3075     {
3076       rv = -EINVAL;
3077       goto poll_done;
3078     }
3079
3080   /* zero vcom fd sets */
3081   /*
3082    * V vcom fd set
3083    */
3084 #define _(V)      \
3085     FD_ZERO ((V))
3086
3087   _(&vcom_readfds);
3088   _(&vcom_writefds);
3089   _(&vcom_exceptfds);
3090 #undef _
3091
3092   vcom_nfds = 0;
3093   vcom_nfd = -1;
3094
3095
3096   for (fds_idx = 0; fds_idx < __nfds; fds_idx++)
3097     {
3098       /* ignore negative fds */
3099       if (__fds[fds_idx].fd < 0)
3100         {
3101           continue;
3102         }
3103
3104       /* non negative validated vcom fds */
3105       if (__fds[fds_idx].fd > FD_SETSIZE)
3106         {
3107           rv = -EINVAL;
3108           goto poll_done;
3109         }
3110
3111       /* max_vcom_fd and vcom_nfd */
3112       if (__fds[fds_idx].fd > max_vcom_fd)
3113         {
3114           /* requested events */
3115           if (__fds[fds_idx].events)
3116             {
3117               max_vcom_fd = __fds[fds_idx].fd;
3118             }
3119         }
3120       ++vcom_nfd;
3121     }
3122
3123   vcom_nfds = max_vcom_fd != -1 ? max_vcom_fd + 1 : 0;
3124
3125   if (!vcom_nfds)
3126     {
3127       rv = vcom_nfds;
3128       goto poll_done;
3129     }
3130
3131   vcom_pollfds_2_selectfds (
3132                              /* src */
3133                              __fds, __nfds,
3134                              /* dest */
3135                              vcom_nfds,
3136                              &vcom_readfds, &vcom_writefds, &vcom_exceptfds);
3137
3138   /* select on vcom fds */
3139   vcom_nfd = vcom_socket_select (vcom_nfds,
3140                                  &vcom_readfds,
3141                                  &vcom_writefds, &vcom_exceptfds, &tv);
3142   if (VCOM_DEBUG > 2)
3143     fprintf (stderr,
3144              "[%d] vcom_socket_select: "
3145              "'%04d'='%04d'\n", pid, vcom_nfd, vcom_nfds);
3146
3147   if (vcom_nfd < 0)
3148     {
3149       rv = vcom_nfd;
3150       goto poll_done;
3151     }
3152
3153   vcom_selectfds_2_pollfds (
3154                              /* dest */
3155                              __fds, __nfds, &nfd,
3156                              /* src */
3157                              vcom_nfds,
3158                              &vcom_readfds, &vcom_writefds, &vcom_exceptfds);
3159
3160   rv = nfd;
3161
3162 poll_done:
3163   return rv;
3164 }
3165
3166 /*
3167  * TBD: remove this static function once vppcom
3168  *      has an implementation in place
3169  *
3170  * ACTION:
3171  */
3172 static int
3173 vppcom_poll (struct pollfd *__fds, nfds_t __nfds, double time_to_wait)
3174 {
3175   return -EOPNOTSUPP;
3176 }
3177
3178 int
3179 vcom_socket_poll_vppcom_impl (struct pollfd *__fds, nfds_t __nfds,
3180                               int __timeout)
3181 {
3182   nfds_t fds_idx = 0;
3183
3184   /* in seconds eg. 3.123456789 seconds */
3185   double time_to_wait = (double) 0;
3186
3187   i32 sid;
3188   i32 vep_idx;
3189
3190   /* replace vcom fd with session idx */
3191   for (fds_idx = 0; fds_idx < __nfds; fds_idx++)
3192     {
3193       /* ignore negative fds */
3194       if (__fds[fds_idx].fd < 0)
3195         {
3196           continue;
3197         }
3198
3199       /* non negative validated vcom fds */
3200       sid = vcom_socket_get_sid (__fds[fds_idx].fd);
3201       if (sid != INVALID_SESSION_ID)
3202         {
3203           __fds[fds_idx].fd = sid;
3204         }
3205       else
3206         {
3207           /* get vep_idx */
3208           vep_idx = vcom_socket_get_vep_idx (__fds[fds_idx].fd);
3209           if (vep_idx != INVALID_VEP_IDX)
3210             {
3211               __fds[fds_idx].fd = vep_idx;
3212             }
3213           else
3214             {
3215               return -EBADF;
3216             }
3217         }
3218     }
3219
3220   /* validate __timeout */
3221   if (__timeout > 0)
3222     {
3223       time_to_wait = (double) __timeout / (double) 1000;
3224     }
3225   else if (__timeout == 0)
3226     {
3227       time_to_wait = (double) 0;
3228     }
3229   else
3230     {
3231       time_to_wait = ~0;
3232     }
3233
3234   return vppcom_poll (__fds, __nfds, time_to_wait);
3235 }
3236
3237 int
3238 vcom_socket_poll (struct pollfd *__fds, nfds_t __nfds, int __timeout)
3239 {
3240   /* select an implementation */
3241
3242   /* return vcom_socket_poll_vppcom_impl (__fds, __nfds, __timeout); */
3243   return vcom_socket_poll_select_impl (__fds, __nfds, __timeout);
3244 }
3245
3246 #ifdef __USE_GNU
3247 int
3248 vcom_socket_ppoll (struct pollfd *__fds, nfds_t __nfds,
3249                    const struct timespec *__timeout, const __sigset_t * __ss)
3250 {
3251   return -EOPNOTSUPP;
3252 }
3253 #endif
3254
3255 int
3256 vcom_socket_main_init (void)
3257 {
3258   vcom_socket_main_t *vsm = &vcom_socket_main;
3259
3260   if (VCOM_DEBUG > 0)
3261     printf ("vcom_socket_main_init\n");
3262
3263   if (!vsm->init)
3264     {
3265       /* TBD: define FD_MAXSIZE and use it here */
3266       pool_alloc (vsm->vsockets, FD_SETSIZE);
3267       vsm->sockidx_by_fd = hash_create (0, sizeof (i32));
3268
3269       pool_alloc (vsm->vepolls, FD_SETSIZE);
3270       vsm->epollidx_by_epfd = hash_create (0, sizeof (i32));
3271
3272       pool_alloc (vsm->vepitems, FD_SETSIZE);
3273       vsm->epitemidx_by_epfdfd = hash_create (0, sizeof (i32));
3274
3275       vsm->epitemidxs_by_epfd = hash_create (0, sizeof (i32 *));
3276       vsm->epitemidxs_by_fd = hash_create (0, sizeof (i32 *));
3277
3278       vsm->init = 1;
3279     }
3280
3281   return 0;
3282 }
3283
3284
3285 void
3286 vcom_socket_main_show (void)
3287 {
3288   vcom_socket_main_t *vsm = &vcom_socket_main;
3289   vcom_socket_t *vsock;
3290
3291   vcom_epoll_t *vepoll;
3292
3293   vcom_epitem_t *vepitem;
3294
3295   i32 epfd;
3296   i32 fd;
3297   i32 *vepitemidxs, *vepitemidxs_var;
3298
3299   if (vsm->init)
3300     {
3301       /* from active list of vsockets show vsock */
3302
3303       /* *INDENT-OFF* */
3304       pool_foreach (vsock, vsm->vsockets,
3305         ({
3306           printf(
3307                  "fd='%04d', sid='%08x',type='%-30s'\n",
3308                  vsock->fd, vsock->sid,
3309                  vcom_socket_type_str (vsock->type));
3310         }));
3311       /* *INDENT-ON* */
3312
3313       /* from active list of vepolls, show vepoll */
3314
3315       /* *INDENT-OFF* */
3316       pool_foreach (vepoll, vsm->vepolls,
3317         ({
3318           printf(
3319                  "epfd='%04d', vep_idx='%08x', "
3320                  "type='%-30s', "
3321                  "flags='%d', count='%d', close='%d'\n",
3322                  vepoll->epfd, vepoll->vep_idx,
3323                  vcom_socket_epoll_type_str (vepoll->type),
3324                  vepoll->flags, vepoll->count, vepoll->close);
3325         }));
3326       /* *INDENT-ON* */
3327
3328       /* from active list of vepitems, show vepitem */
3329
3330       /* *INDENT-OFF* */
3331       pool_foreach (vepitem, vsm->vepitems,
3332         ({
3333           printf(
3334                  "epfd='%04d', fd='%04d', "
3335                  "next_fd='%04d', prev_fd='%04d', "
3336                  "type='%-30s', "
3337                  "events='%04x', revents='%04x'\n",
3338                  vepitem->epfd, vepitem->fd,
3339                  vepitem->next_fd, vepitem->prev_fd,
3340                  vcom_socket_vcom_fd_type_str (vepitem->type),
3341                  vepitem->event.events, vepitem->revent.events);
3342         }));
3343
3344       /* *INDENT-ON* */
3345
3346       /* show epitemidxs for epfd */
3347       /* *INDENT-OFF* */
3348       hash_foreach (epfd, vepitemidxs,
3349                     vsm->epitemidxs_by_epfd,
3350       ({
3351         printf("\n[ '%04d': ", epfd);
3352         vec_foreach (vepitemidxs_var,vepitemidxs)
3353         {
3354           printf("'%04d' ", (int)vepitemidxs_var[0]);
3355         }
3356         printf("]\n");
3357       }));
3358       /* *INDENT-ON* */
3359
3360       /* show epitemidxs for fd */
3361       /* *INDENT-OFF* */
3362       hash_foreach (fd, vepitemidxs,
3363                     vsm->epitemidxs_by_fd,
3364       ({
3365         printf("\n{ '%04d': ", fd);
3366         vec_foreach (vepitemidxs_var,vepitemidxs)
3367         {
3368           printf("'%04d' ", (int)vepitemidxs_var[0]);
3369         }
3370         printf("}\n");
3371       }));
3372       /* *INDENT-ON* */
3373
3374     }
3375 }
3376
3377 void
3378 vcom_socket_main_destroy (void)
3379 {
3380   vcom_socket_main_t *vsm = &vcom_socket_main;
3381   vcom_socket_t *vsock;
3382
3383   vcom_epoll_t *vepoll;
3384
3385   vcom_epitem_t *vepitem;
3386
3387   i32 epfd;
3388   i32 fd;
3389   i32 *vepitemidxs;
3390
3391
3392   if (VCOM_DEBUG > 0)
3393     printf ("vcom_socket_main_destroy\n");
3394
3395   if (vsm->init)
3396     {
3397
3398       /*
3399        * from active list of vepitems,
3400        * remove all "vepitem" elements from the pool in a safe way
3401        * */
3402
3403       /* *INDENT-OFF* */
3404       pool_flush (vepitem, vsm->vepitems,
3405         ({
3406           if ((vepitem->type == FD_TYPE_EPOLL) ||
3407               (vepitem->type == FD_TYPE_VCOM_SOCKET))
3408           {
3409              vcom_socket_epoll_ctl1 (vepitem->epfd, EPOLL_CTL_DEL,
3410                                      vepitem->fd, NULL);
3411              vepitem_init (vepitem);
3412           }
3413         }));
3414       /* *INDENT-ON* */
3415
3416       pool_free (vsm->vepitems);
3417       hash_free (vsm->epitemidx_by_epfdfd);
3418
3419       /* free vepitemidxs for each epfd */
3420       /* *INDENT-OFF* */
3421       hash_foreach (epfd, vepitemidxs,
3422                     vsm->epitemidxs_by_epfd,
3423       ({
3424         vec_free (vepitemidxs);
3425       }));
3426       /* *INDENT-ON* */
3427       hash_free (vsm->epitemidxs_by_epfd);
3428
3429       /* free vepitemidxs for each fd */
3430       /* *INDENT-OFF* */
3431       hash_foreach (fd, vepitemidxs,
3432                     vsm->epitemidxs_by_fd,
3433       ({
3434         vec_free (vepitemidxs);
3435       }));
3436       /* *INDENT-ON* */
3437       hash_free (vsm->epitemidxs_by_fd);
3438
3439
3440       /*
3441        * from active list of vsockets,
3442        * close socket and vppcom session
3443        * */
3444
3445       /* *INDENT-OFF* */
3446       pool_foreach (vsock, vsm->vsockets,
3447         ({
3448           if (vsock->type == SOCKET_TYPE_VPPCOM_BOUND)
3449             {
3450               vppcom_session_close (vsock->sid);
3451               vcom_socket_close_socket (vsock->fd);
3452               vsocket_init (vsock);
3453             }
3454         }));
3455       /* *INDENT-ON* */
3456
3457       /*
3458        * return vsocket element to the pool
3459        * */
3460
3461       /* *INDENT-OFF* */
3462       pool_flush (vsock, vsm->vsockets,
3463         ({
3464           // vsocket_init(vsock);
3465           ;
3466         }));
3467       /* *INDENT-ON* */
3468
3469       pool_free (vsm->vsockets);
3470       hash_free (vsm->sockidx_by_fd);
3471
3472       /*
3473        * from active list of vepolls,
3474        * close epoll and vppcom_epoll
3475        * */
3476
3477       /* *INDENT-OFF* */
3478       pool_foreach (vepoll, vsm->vepolls,
3479         ({
3480           if (vepoll->type == EPOLL_TYPE_VPPCOM_BOUND)
3481             {
3482               vppcom_session_close (vepoll->vep_idx);
3483               vcom_socket_close_epoll (vepoll->epfd); /* TBD: */
3484               vepoll_init (vepoll);
3485             }
3486         }));
3487       /* *INDENT-ON* */
3488
3489       /*
3490        * return vepoll element to the pool
3491        * */
3492
3493       /* *INDENT-OFF* */
3494       pool_flush (vepoll, vsm->vepolls,
3495         ({
3496           // vepoll_init(vepoll);
3497           ;
3498         }));
3499       /* *INDENT-ON* */
3500
3501       pool_free (vsm->vepolls);
3502       hash_free (vsm->epollidx_by_epfd);
3503
3504       vsm->init = 0;
3505     }
3506 }
3507
3508
3509 /*
3510  * fd.io coding-style-patch-verification: ON
3511  *
3512  * Local Variables:
3513  * eval: (c-set-style "gnu")
3514  * End:
3515  */