2 * Copyright (c) 2016-2019 Cisco and/or its affiliates.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at:
7 * http://www.apache.org/licenses/LICENSE-2.0
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.
16 #ifdef HAVE_GNU_SOURCE
27 #include <sys/resource.h>
28 #include <netinet/tcp.h>
29 #include <linux/udp.h>
31 #include <vcl/ldp_socket_wrapper.h>
35 #include <vcl/vcl_locked.h>
36 #include <vppinfra/time.h>
37 #include <vppinfra/bitmap.h>
38 #include <vppinfra/lock.h>
39 #include <vppinfra/pool.h>
40 #include <vppinfra/hash.h>
42 #define HAVE_CONSTRUCTOR_ATTRIBUTE
43 #ifdef HAVE_CONSTRUCTOR_ATTRIBUTE
44 #define CONSTRUCTOR_ATTRIBUTE \
45 __attribute__ ((constructor))
47 #define CONSTRUCTOR_ATTRIBUTE
48 #endif /* HAVE_CONSTRUCTOR_ATTRIBUTE */
50 #define HAVE_DESTRUCTOR_ATTRIBUTE
51 #ifdef HAVE_DESTRUCTOR_ATTRIBUTE
52 #define DESTRUCTOR_ATTRIBUTE \
53 __attribute__ ((destructor))
55 #define DESTRUCTOR_ATTRIBUTE
58 #define LDP_MAX_NWORKERS 32
60 #ifdef HAVE_GNU_SOURCE
61 #define SOCKADDR_GET_SA(__addr) __addr.__sockaddr__;
63 #define SOCKADDR_GET_SA(__addr) _addr;
66 typedef struct ldp_worker_ctx_
69 clib_time_t clib_time;
74 clib_bitmap_t *rd_bitmap;
75 clib_bitmap_t *wr_bitmap;
76 clib_bitmap_t *ex_bitmap;
77 clib_bitmap_t *si_rd_bitmap;
78 clib_bitmap_t *si_wr_bitmap;
79 clib_bitmap_t *si_ex_bitmap;
80 clib_bitmap_t *libc_rd_bitmap;
81 clib_bitmap_t *libc_wr_bitmap;
82 clib_bitmap_t *libc_ex_bitmap;
88 struct pollfd *libc_poll;
100 /* clib_bitmap_t, fd_mask and vcl_si_set are used interchangeably. Make sure
101 * they are the same size */
102 STATIC_ASSERT (sizeof (clib_bitmap_t) == sizeof (fd_mask),
103 "ldp bitmap size mismatch");
104 STATIC_ASSERT (sizeof (vcl_si_set) == sizeof (fd_mask),
105 "ldp bitmap size mismatch");
109 ldp_worker_ctx_t *workers;
111 char app_name[LDP_APP_NAME_MAX];
116 /** vcl needs next epoll_create to go to libc_epoll */
117 u8 vcl_needs_real_epoll;
120 * crypto state used only for testing
126 #define LDP_DEBUG ldp->debug
128 #define LDBG(_lvl, _fmt, _args...) \
129 if (ldp->debug > _lvl) \
131 int errno_saved = errno; \
132 fprintf (stderr, "ldp<%d>: " _fmt "\n", getpid(), ##_args); \
133 errno = errno_saved; \
136 static ldp_main_t ldp_main = {
137 .vlsh_bit_val = (1 << LDP_SID_BIT_MIN),
138 .vlsh_bit_mask = (1 << LDP_SID_BIT_MIN) - 1,
139 .debug = LDP_DEBUG_INIT,
140 .transparent_tls = 0,
144 static ldp_main_t *ldp = &ldp_main;
146 static inline ldp_worker_ctx_t *
147 ldp_worker_get_current (void)
149 return (ldp->workers + vppcom_worker_index ());
153 * RETURN: 0 on success or -1 on error.
156 ldp_set_app_name (char *app_name)
158 snprintf (ldp->app_name, LDP_APP_NAME_MAX,
159 "ldp-%d-%s", getpid (), app_name);
165 if (ldp->app_name[0] == '\0')
166 ldp_set_app_name ("app");
168 return ldp->app_name;
172 ldp_vlsh_to_fd (vls_handle_t vlsh)
174 return (vlsh + ldp->vlsh_bit_val);
177 static inline vls_handle_t
178 ldp_fd_to_vlsh (int fd)
180 if (fd < ldp->vlsh_bit_val)
181 return VLS_INVALID_HANDLE;
183 return (fd - ldp->vlsh_bit_val);
187 ldp_alloc_workers (void)
191 pool_alloc (ldp->workers, LDP_MAX_NWORKERS);
197 ldp_worker_ctx_t *ldpw;
203 ldp->vcl_needs_real_epoll = 1;
204 rv = vls_app_create (ldp_get_app_name ());
207 ldp->vcl_needs_real_epoll = 0;
208 if (rv == VPPCOM_EEXIST)
210 LDBG (2, "\nERROR: ldp_init: vppcom_app_create()"
211 " failed! rv = %d (%s)\n", rv, vppcom_retval_str (rv));
215 ldp->vcl_needs_real_epoll = 0;
216 ldp_alloc_workers ();
217 ldpw = ldp_worker_get_current ();
219 char *env_var_str = getenv (LDP_ENV_DEBUG);
223 if (sscanf (env_var_str, "%u", &tmp) != 1)
224 clib_warning ("LDP<%d>: WARNING: Invalid LDP debug level specified in"
225 " the env var " LDP_ENV_DEBUG " (%s)!", getpid (),
230 LDBG (0, "configured LDP debug level (%u) from env var "
231 LDP_ENV_DEBUG "!", ldp->debug);
235 env_var_str = getenv (LDP_ENV_APP_NAME);
238 ldp_set_app_name (env_var_str);
239 LDBG (0, "configured LDP app name (%s) from the env var "
240 LDP_ENV_APP_NAME "!", ldp->app_name);
243 env_var_str = getenv (LDP_ENV_SID_BIT);
247 if (sscanf (env_var_str, "%u", &sb) != 1)
249 LDBG (0, "WARNING: Invalid LDP sid bit specified in the env var "
250 LDP_ENV_SID_BIT " (%s)! sid bit value %d (0x%x)", env_var_str,
251 ldp->vlsh_bit_val, ldp->vlsh_bit_val);
253 else if (sb < LDP_SID_BIT_MIN)
255 ldp->vlsh_bit_val = (1 << LDP_SID_BIT_MIN);
256 ldp->vlsh_bit_mask = ldp->vlsh_bit_val - 1;
258 LDBG (0, "WARNING: LDP sid bit (%u) specified in the env var "
259 LDP_ENV_SID_BIT " (%s) is too small. Using LDP_SID_BIT_MIN"
260 " (%d)! sid bit value %d (0x%x)", sb, env_var_str,
261 LDP_SID_BIT_MIN, ldp->vlsh_bit_val, ldp->vlsh_bit_val);
263 else if (sb > LDP_SID_BIT_MAX)
265 ldp->vlsh_bit_val = (1 << LDP_SID_BIT_MAX);
266 ldp->vlsh_bit_mask = ldp->vlsh_bit_val - 1;
268 LDBG (0, "WARNING: LDP sid bit (%u) specified in the env var "
269 LDP_ENV_SID_BIT " (%s) is too big. Using LDP_SID_BIT_MAX"
270 " (%d)! sid bit value %d (0x%x)", sb, env_var_str,
271 LDP_SID_BIT_MAX, ldp->vlsh_bit_val, ldp->vlsh_bit_val);
275 ldp->vlsh_bit_val = (1 << sb);
276 ldp->vlsh_bit_mask = ldp->vlsh_bit_val - 1;
278 LDBG (0, "configured LDP sid bit (%u) from "
279 LDP_ENV_SID_BIT "! sid bit value %d (0x%x)", sb,
280 ldp->vlsh_bit_val, ldp->vlsh_bit_val);
283 /* Make sure there are enough bits in the fd set for vcl sessions */
284 if (ldp->vlsh_bit_val > FD_SETSIZE / 2)
286 /* Only valid for select/pselect, so just WARNING and not exit */
288 "WARNING: LDP vlsh bit value %d > FD_SETSIZE/2 %d, "
289 "select/pselect not supported now!",
290 ldp->vlsh_bit_val, FD_SETSIZE / 2);
293 env_var_str = getenv (LDP_ENV_TLS_TRANS);
296 ldp->transparent_tls = 1;
300 pool_foreach (ldpw, ldp->workers) {
301 clib_memset (&ldpw->clib_time, 0, sizeof (ldpw->clib_time));
305 LDBG (0, "LDP initialization: done!");
310 #define ldp_init_check() \
311 if (PREDICT_FALSE (!ldp->init)) \
313 if ((errno = -ldp_init ())) \
325 vlsh = ldp_fd_to_vlsh (fd);
326 if (vlsh != VLS_INVALID_HANDLE)
328 epfd = vls_attr (vlsh, VPPCOM_ATTR_GET_LIBC_EPFD, 0, 0);
331 LDBG (0, "fd %d: calling libc_close: epfd %u", fd, epfd);
333 rv = libc_close (epfd);
336 u32 size = sizeof (epfd);
339 (void) vls_attr (vlsh, VPPCOM_ATTR_SET_LIBC_EPFD, &epfd, &size);
342 else if (PREDICT_FALSE (epfd < 0))
349 LDBG (0, "fd %d: calling vls_close: vlsh %u", fd, vlsh);
351 rv = vls_close (vlsh);
360 LDBG (0, "fd %d: calling libc_close", fd);
361 rv = libc_close (fd);
369 read (int fd, void *buf, size_t nbytes)
376 vlsh = ldp_fd_to_vlsh (fd);
377 if (vlsh != VLS_INVALID_HANDLE)
379 size = vls_read (vlsh, buf, nbytes);
388 size = libc_read (fd, buf, nbytes);
395 readv (int fd, const struct iovec * iov, int iovcnt)
397 int rv = 0, i, total = 0;
403 vlsh = ldp_fd_to_vlsh (fd);
404 if (vlsh != VLS_INVALID_HANDLE)
406 for (i = 0; i < iovcnt; ++i)
408 rv = vls_read (vlsh, iov[i].iov_base, iov[i].iov_len);
414 if (rv < iov[i].iov_len)
418 if (rv < 0 && total == 0)
428 size = libc_readv (fd, iov, iovcnt);
435 write (int fd, const void *buf, size_t nbytes)
442 vlsh = ldp_fd_to_vlsh (fd);
443 if (vlsh != VLS_INVALID_HANDLE)
445 size = vls_write_msg (vlsh, (void *) buf, nbytes);
454 size = libc_write (fd, buf, nbytes);
461 writev (int fd, const struct iovec * iov, int iovcnt)
463 ssize_t size = 0, total = 0;
469 vlsh = ldp_fd_to_vlsh (fd);
470 if (vlsh != VLS_INVALID_HANDLE)
472 for (i = 0; i < iovcnt; ++i)
474 rv = vls_write_msg (vlsh, iov[i].iov_base, iov[i].iov_len);
480 if (rv < iov[i].iov_len)
485 if (rv < 0 && total == 0)
495 size = libc_writev (fd, iov, iovcnt);
502 fcntl_internal (int fd, int cmd, va_list ap)
507 vlsh = ldp_fd_to_vlsh (fd);
508 LDBG (0, "fd %u vlsh %d, cmd %u", fd, vlsh, cmd);
509 if (vlsh != VLS_INVALID_HANDLE)
511 int flags = va_arg (ap, int);
514 size = sizeof (flags);
519 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_FLAGS, &flags, &size);
523 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_FLAGS, &flags, &size);
528 /* TODO handle this */
529 LDBG (0, "F_SETFD ignored flags %u", flags);
545 rv = libc_vfcntl64 (fd, cmd, ap);
547 rv = libc_vfcntl (fd, cmd, ap);
555 fcntl (int fd, int cmd, ...)
563 rv = fcntl_internal (fd, cmd, ap);
570 fcntl64 (int fd, int cmd, ...)
578 rv = fcntl_internal (fd, cmd, ap);
584 ioctl (int fd, unsigned long int cmd, ...)
594 vlsh = ldp_fd_to_vlsh (fd);
595 if (vlsh != VLS_INVALID_HANDLE)
600 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_NREAD, 0, 0);
605 u32 flags = va_arg (ap, int) ? O_NONBLOCK : 0;
606 u32 size = sizeof (flags);
608 /* TBD: When VPPCOM_ATTR_[GS]ET_FLAGS supports flags other than
609 * non-blocking, the flags should be read here and merged
612 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_FLAGS, &flags, &size);
628 rv = libc_vioctl (fd, cmd, ap);
636 ldp_select_init_maps (fd_set * __restrict original,
637 clib_bitmap_t ** resultb, clib_bitmap_t ** libcb,
638 clib_bitmap_t ** vclb, int nfds, u32 minbits,
639 u32 n_bytes, uword * si_bits, uword * libc_bits)
641 uword si_bits_set, libc_bits_set;
645 clib_bitmap_validate (*vclb, minbits);
646 clib_bitmap_validate (*libcb, minbits);
647 clib_bitmap_validate (*resultb, minbits);
648 clib_memcpy_fast (*resultb, original, n_bytes);
649 memset (original, 0, n_bytes);
652 clib_bitmap_foreach (fd, *resultb) {
655 vlsh = ldp_fd_to_vlsh (fd);
656 if (vlsh == VLS_INVALID_HANDLE)
657 clib_bitmap_set_no_check (*libcb, fd, 1);
659 *vclb = clib_bitmap_set (*vclb, vlsh_to_session_index (vlsh), 1);
663 si_bits_set = clib_bitmap_last_set (*vclb) + 1;
664 *si_bits = (si_bits_set > *si_bits) ? si_bits_set : *si_bits;
665 clib_bitmap_validate (*resultb, *si_bits);
667 libc_bits_set = clib_bitmap_last_set (*libcb) + 1;
668 *libc_bits = (libc_bits_set > *libc_bits) ? libc_bits_set : *libc_bits;
672 ldp_select_vcl_map_to_libc (clib_bitmap_t * vclb, fd_set * __restrict libcb)
682 clib_bitmap_foreach (si, vclb) {
683 vlsh = vls_session_index_to_vlsh (si);
684 ASSERT (vlsh != VLS_INVALID_HANDLE);
685 fd = ldp_vlsh_to_fd (vlsh);
686 if (PREDICT_FALSE (fd < 0))
699 ldp_select_libc_map_merge (clib_bitmap_t * result, fd_set * __restrict libcb)
707 clib_bitmap_foreach (fd, result)
708 FD_SET ((int)fd, libcb);
713 ldp_pselect (int nfds, fd_set * __restrict readfds,
714 fd_set * __restrict writefds,
715 fd_set * __restrict exceptfds,
716 const struct timespec *__restrict timeout,
717 const __sigset_t * __restrict sigmask)
719 u32 minbits = clib_max (nfds, BITS (uword)), n_bytes;
720 ldp_worker_ctx_t *ldpw = ldp_worker_get_current ();
721 struct timespec libc_tspec = { 0 };
722 f64 time_out, vcl_timeout = 0;
723 uword si_bits, libc_bits;
724 int rv, bits_set = 0;
732 if (PREDICT_FALSE (ldpw->clib_time.init_cpu_time == 0))
733 clib_time_init (&ldpw->clib_time);
737 time_out = (timeout->tv_sec == 0 && timeout->tv_nsec == 0) ?
738 (f64) 0 : (f64) timeout->tv_sec + (f64) timeout->tv_nsec / (f64) 1e9;
740 time_out += clib_time_now (&ldpw->clib_time);
742 /* select as fine grained sleep */
745 while (clib_time_now (&ldpw->clib_time) < time_out)
758 if (nfds <= ldp->vlsh_bit_val)
760 rv = libc_pselect (nfds, readfds, writefds, exceptfds,
765 si_bits = libc_bits = 0;
766 n_bytes = nfds / 8 + ((nfds % 8) ? 1 : 0);
769 ldp_select_init_maps (readfds, &ldpw->rd_bitmap, &ldpw->libc_rd_bitmap,
770 &ldpw->si_rd_bitmap, nfds, minbits, n_bytes,
771 &si_bits, &libc_bits);
773 ldp_select_init_maps (writefds, &ldpw->wr_bitmap,
774 &ldpw->libc_wr_bitmap, &ldpw->si_wr_bitmap, nfds,
775 minbits, n_bytes, &si_bits, &libc_bits);
777 ldp_select_init_maps (exceptfds, &ldpw->ex_bitmap,
778 &ldpw->libc_ex_bitmap, &ldpw->si_ex_bitmap, nfds,
779 minbits, n_bytes, &si_bits, &libc_bits);
781 if (PREDICT_FALSE (!si_bits && !libc_bits))
789 libc_tspec = timeout ? *timeout : libc_tspec;
796 clib_memcpy_fast (ldpw->rd_bitmap, ldpw->si_rd_bitmap,
797 vec_len (ldpw->si_rd_bitmap) *
798 sizeof (clib_bitmap_t));
800 clib_memcpy_fast (ldpw->wr_bitmap, ldpw->si_wr_bitmap,
801 vec_len (ldpw->si_wr_bitmap) *
802 sizeof (clib_bitmap_t));
804 clib_memcpy_fast (ldpw->ex_bitmap, ldpw->si_ex_bitmap,
805 vec_len (ldpw->si_ex_bitmap) *
806 sizeof (clib_bitmap_t));
808 rv = vls_select (si_bits, readfds ? ldpw->rd_bitmap : NULL,
809 writefds ? ldpw->wr_bitmap : NULL,
810 exceptfds ? ldpw->ex_bitmap : NULL, vcl_timeout);
819 if (ldp_select_vcl_map_to_libc (ldpw->rd_bitmap, readfds))
825 if (ldp_select_vcl_map_to_libc (ldpw->wr_bitmap, writefds))
831 if (ldp_select_vcl_map_to_libc (ldpw->ex_bitmap, exceptfds))
842 clib_memcpy_fast (ldpw->rd_bitmap, ldpw->libc_rd_bitmap,
843 vec_len (ldpw->libc_rd_bitmap) *
844 sizeof (clib_bitmap_t));
846 clib_memcpy_fast (ldpw->wr_bitmap, ldpw->libc_wr_bitmap,
847 vec_len (ldpw->libc_wr_bitmap) *
848 sizeof (clib_bitmap_t));
850 clib_memcpy_fast (ldpw->ex_bitmap, ldpw->libc_ex_bitmap,
851 vec_len (ldpw->libc_ex_bitmap) *
852 sizeof (clib_bitmap_t));
854 rv = libc_pselect (libc_bits,
855 readfds ? (fd_set *) ldpw->rd_bitmap : NULL,
856 writefds ? (fd_set *) ldpw->wr_bitmap : NULL,
857 exceptfds ? (fd_set *) ldpw->ex_bitmap : NULL,
858 &libc_tspec, sigmask);
861 ldp_select_libc_map_merge (ldpw->rd_bitmap, readfds);
862 ldp_select_libc_map_merge (ldpw->wr_bitmap, writefds);
863 ldp_select_libc_map_merge (ldpw->ex_bitmap, exceptfds);
874 while ((time_out == -1) || (clib_time_now (&ldpw->clib_time) < time_out));
878 /* TBD: set timeout to amount of time left */
879 clib_bitmap_zero (ldpw->rd_bitmap);
880 clib_bitmap_zero (ldpw->si_rd_bitmap);
881 clib_bitmap_zero (ldpw->libc_rd_bitmap);
882 clib_bitmap_zero (ldpw->wr_bitmap);
883 clib_bitmap_zero (ldpw->si_wr_bitmap);
884 clib_bitmap_zero (ldpw->libc_wr_bitmap);
885 clib_bitmap_zero (ldpw->ex_bitmap);
886 clib_bitmap_zero (ldpw->si_ex_bitmap);
887 clib_bitmap_zero (ldpw->libc_ex_bitmap);
893 select (int nfds, fd_set * __restrict readfds,
894 fd_set * __restrict writefds,
895 fd_set * __restrict exceptfds, struct timeval *__restrict timeout)
897 struct timespec tspec;
901 tspec.tv_sec = timeout->tv_sec;
902 tspec.tv_nsec = timeout->tv_usec * 1000;
904 return ldp_pselect (nfds, readfds, writefds, exceptfds,
905 timeout ? &tspec : NULL, NULL);
910 pselect (int nfds, fd_set * __restrict readfds,
911 fd_set * __restrict writefds,
912 fd_set * __restrict exceptfds,
913 const struct timespec *__restrict timeout,
914 const __sigset_t * __restrict sigmask)
916 return ldp_pselect (nfds, readfds, writefds, exceptfds, timeout, 0);
920 /* If transparent TLS mode is turned on, then ldp will load key and cert.
923 load_cert_key_pair (void)
925 char *cert_str = getenv (LDP_ENV_TLS_CERT);
926 char *key_str = getenv (LDP_ENV_TLS_KEY);
927 char cert_buf[4096], key_buf[4096];
928 int cert_size, key_size;
929 vppcom_cert_key_pair_t crypto;
933 if (!cert_str || !key_str)
935 LDBG (0, "ERROR: failed to read LDP environment %s\n",
940 fp = fopen (cert_str, "r");
943 LDBG (0, "ERROR: failed to open cert file %s \n", cert_str);
946 cert_size = fread (cert_buf, sizeof (char), sizeof (cert_buf), fp);
949 fp = fopen (key_str, "r");
952 LDBG (0, "ERROR: failed to open key file %s \n", key_str);
955 key_size = fread (key_buf, sizeof (char), sizeof (key_buf), fp);
958 crypto.cert = cert_buf;
959 crypto.key = key_buf;
960 crypto.cert_len = cert_size;
961 crypto.key_len = key_size;
962 ckp_index = vppcom_add_cert_key_pair (&crypto);
965 LDBG (0, "ERROR: failed to add cert key pair\n");
969 ldp->ckpair_index = ckp_index;
975 assign_cert_key_pair (vls_handle_t vlsh)
979 if (ldp->ckpair_index == ~0 && load_cert_key_pair () < 0)
982 ckp_len = sizeof (ldp->ckpair_index);
983 return vls_attr (vlsh, VPPCOM_ATTR_SET_CKPAIR, &ldp->ckpair_index, &ckp_len);
987 socket (int domain, int type, int protocol)
989 int rv, sock_type = type & ~(SOCK_CLOEXEC | SOCK_NONBLOCK);
990 u8 is_nonblocking = type & SOCK_NONBLOCK ? 1 : 0;
995 if (((domain == AF_INET) || (domain == AF_INET6)) &&
996 ((sock_type == SOCK_STREAM) || (sock_type == SOCK_DGRAM)))
999 if (ldp->transparent_tls)
1001 proto = VPPCOM_PROTO_TLS;
1004 proto = ((sock_type == SOCK_DGRAM) ?
1005 VPPCOM_PROTO_UDP : VPPCOM_PROTO_TCP);
1007 LDBG (0, "calling vls_create: proto %u (%s), is_nonblocking %u",
1008 proto, vppcom_proto_str (proto), is_nonblocking);
1010 vlsh = vls_create (proto, is_nonblocking);
1018 if (ldp->transparent_tls)
1020 if (assign_cert_key_pair (vlsh) < 0)
1023 rv = ldp_vlsh_to_fd (vlsh);
1028 LDBG (0, "calling libc_socket");
1029 rv = libc_socket (domain, type, protocol);
1036 * Create two new sockets, of type TYPE in domain DOMAIN and using
1037 * protocol PROTOCOL, which are connected to each other, and put file
1038 * descriptors for them in FDS[0] and FDS[1]. If PROTOCOL is zero,
1039 * one will be chosen automatically.
1040 * Returns 0 on success, -1 for errors.
1043 socketpair (int domain, int type, int protocol, int fds[2])
1045 int rv, sock_type = type & ~(SOCK_CLOEXEC | SOCK_NONBLOCK);
1049 if (((domain == AF_INET) || (domain == AF_INET6)) &&
1050 ((sock_type == SOCK_STREAM) || (sock_type == SOCK_DGRAM)))
1052 LDBG (0, "LDP-TBD");
1058 LDBG (1, "calling libc_socketpair");
1059 rv = libc_socketpair (domain, type, protocol, fds);
1066 bind (int fd, __CONST_SOCKADDR_ARG _addr, socklen_t len)
1068 const struct sockaddr *addr = SOCKADDR_GET_SA (_addr);
1074 vlsh = ldp_fd_to_vlsh (fd);
1075 if (vlsh != VLS_INVALID_HANDLE)
1079 switch (addr->sa_family)
1082 if (len != sizeof (struct sockaddr_in))
1084 LDBG (0, "ERROR: fd %d: vlsh %u: Invalid AF_INET addr len %u!",
1090 ep.is_ip4 = VPPCOM_IS_IP4;
1091 ep.ip = (u8 *) & ((const struct sockaddr_in *) addr)->sin_addr;
1092 ep.port = (u16) ((const struct sockaddr_in *) addr)->sin_port;
1096 if (len != sizeof (struct sockaddr_in6))
1098 LDBG (0, "ERROR: fd %d: vlsh %u: Invalid AF_INET6 addr len %u!",
1104 ep.is_ip4 = VPPCOM_IS_IP6;
1105 ep.ip = (u8 *) & ((const struct sockaddr_in6 *) addr)->sin6_addr;
1106 ep.port = (u16) ((const struct sockaddr_in6 *) addr)->sin6_port;
1110 LDBG (0, "ERROR: fd %d: vlsh %u: Unsupported address family %u!",
1111 fd, vlsh, addr->sa_family);
1112 errno = EAFNOSUPPORT;
1116 LDBG (0, "fd %d: calling vls_bind: vlsh %u, addr %p, len %u", fd, vlsh,
1119 rv = vls_bind (vlsh, &ep);
1120 if (rv != VPPCOM_OK)
1128 LDBG (0, "fd %d: calling libc_bind: addr %p, len %u", fd, addr, len);
1129 rv = libc_bind (fd, addr, len);
1133 LDBG (1, "fd %d: returning %d", fd, rv);
1139 ldp_copy_ep_to_sockaddr (struct sockaddr *addr, socklen_t *__restrict len,
1142 int rv = 0, sa_len, copy_len;
1146 if (addr && len && ep)
1148 addr->sa_family = (ep->is_ip4 == VPPCOM_IS_IP4) ? AF_INET : AF_INET6;
1149 switch (addr->sa_family)
1152 ((struct sockaddr_in *) addr)->sin_port = ep->port;
1153 if (*len > sizeof (struct sockaddr_in))
1154 *len = sizeof (struct sockaddr_in);
1155 sa_len = sizeof (struct sockaddr_in) - sizeof (struct in_addr);
1156 copy_len = *len - sa_len;
1158 memcpy (&((struct sockaddr_in *) addr)->sin_addr, ep->ip,
1163 ((struct sockaddr_in6 *) addr)->sin6_port = ep->port;
1164 if (*len > sizeof (struct sockaddr_in6))
1165 *len = sizeof (struct sockaddr_in6);
1166 sa_len = sizeof (struct sockaddr_in6) - sizeof (struct in6_addr);
1167 copy_len = *len - sa_len;
1169 memcpy (((struct sockaddr_in6 *) addr)->sin6_addr.
1170 __in6_u.__u6_addr8, ep->ip, copy_len);
1183 getsockname (int fd, __SOCKADDR_ARG _addr, socklen_t *__restrict len)
1185 struct sockaddr *addr = SOCKADDR_GET_SA (_addr);
1191 vlsh = ldp_fd_to_vlsh (fd);
1192 if (vlsh != VLS_INVALID_HANDLE)
1195 u8 addr_buf[sizeof (struct in6_addr)];
1196 u32 size = sizeof (ep);
1200 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_LCL_ADDR, &ep, &size);
1201 if (rv != VPPCOM_OK)
1208 rv = ldp_copy_ep_to_sockaddr (addr, len, &ep);
1209 if (rv != VPPCOM_OK)
1218 rv = libc_getsockname (fd, _addr, len);
1225 connect (int fd, __CONST_SOCKADDR_ARG _addr, socklen_t len)
1227 const struct sockaddr *addr = SOCKADDR_GET_SA (_addr);
1235 LDBG (0, "ERROR: fd %d: NULL addr, len %u", fd, len);
1241 vlsh = ldp_fd_to_vlsh (fd);
1242 if (vlsh != VLS_INVALID_HANDLE)
1246 switch (addr->sa_family)
1249 if (len != sizeof (struct sockaddr_in))
1251 LDBG (0, "fd %d: ERROR vlsh %u: Invalid AF_INET addr len %u!",
1257 ep.is_ip4 = VPPCOM_IS_IP4;
1258 ep.ip = (u8 *) & ((const struct sockaddr_in *) addr)->sin_addr;
1259 ep.port = (u16) ((const struct sockaddr_in *) addr)->sin_port;
1263 if (len != sizeof (struct sockaddr_in6))
1265 LDBG (0, "fd %d: ERROR vlsh %u: Invalid AF_INET6 addr len %u!",
1271 ep.is_ip4 = VPPCOM_IS_IP6;
1272 ep.ip = (u8 *) & ((const struct sockaddr_in6 *) addr)->sin6_addr;
1273 ep.port = (u16) ((const struct sockaddr_in6 *) addr)->sin6_port;
1277 LDBG (0, "fd %d: ERROR vlsh %u: Unsupported address family %u!",
1278 fd, vlsh, addr->sa_family);
1279 errno = EAFNOSUPPORT;
1283 LDBG (0, "fd %d: calling vls_connect(): vlsh %u addr %p len %u", fd,
1286 rv = vls_connect (vlsh, &ep);
1287 if (rv != VPPCOM_OK)
1295 LDBG (0, "fd %d: calling libc_connect(): addr %p, len %u",
1298 rv = libc_connect (fd, addr, len);
1302 LDBG (1, "fd %d: returning %d (0x%x)", fd, rv, rv);
1307 getpeername (int fd, __SOCKADDR_ARG _addr, socklen_t *__restrict len)
1309 struct sockaddr *addr = SOCKADDR_GET_SA (_addr);
1315 vlsh = ldp_fd_to_vlsh (fd);
1316 if (vlsh != VLS_INVALID_HANDLE)
1319 u8 addr_buf[sizeof (struct in6_addr)];
1320 u32 size = sizeof (ep);
1323 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_PEER_ADDR, &ep, &size);
1324 if (rv != VPPCOM_OK)
1331 rv = ldp_copy_ep_to_sockaddr (addr, len, &ep);
1332 if (rv != VPPCOM_OK)
1341 rv = libc_getpeername (fd, addr, len);
1348 send (int fd, const void *buf, size_t n, int flags)
1350 vls_handle_t vlsh = ldp_fd_to_vlsh (fd);
1355 if (vlsh != VLS_INVALID_HANDLE)
1357 size = vls_sendto (vlsh, (void *) buf, n, flags, NULL);
1358 if (size < VPPCOM_OK)
1366 size = libc_send (fd, buf, n, flags);
1373 sendfile (int out_fd, int in_fd, off_t * offset, size_t len)
1375 ldp_worker_ctx_t *ldpw = ldp_worker_get_current ();
1381 vlsh = ldp_fd_to_vlsh (out_fd);
1382 if (vlsh != VLS_INVALID_HANDLE)
1385 ssize_t results = 0;
1386 size_t n_bytes_left = len;
1387 size_t bytes_to_read;
1390 u32 flags, flags_len = sizeof (flags);
1392 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_FLAGS, &flags, &flags_len);
1393 if (PREDICT_FALSE (rv != VPPCOM_OK))
1395 LDBG (0, "ERROR: out fd %d: vls_attr: vlsh %u, returned %d (%s)!",
1396 out_fd, vlsh, rv, vppcom_retval_str (rv));
1398 vec_reset_length (ldpw->io_buffer);
1406 off_t off = lseek (in_fd, *offset, SEEK_SET);
1407 if (PREDICT_FALSE (off == -1))
1413 ASSERT (off == *offset);
1418 size = vls_attr (vlsh, VPPCOM_ATTR_GET_NWRITE, 0, 0);
1421 LDBG (0, "ERROR: fd %d: vls_attr: vlsh %u returned %ld (%s)!",
1422 out_fd, vlsh, size, vppcom_retval_str (size));
1423 vec_reset_length (ldpw->io_buffer);
1429 bytes_to_read = size;
1430 if (bytes_to_read == 0)
1432 if (flags & O_NONBLOCK)
1441 bytes_to_read = clib_min (n_bytes_left, bytes_to_read);
1442 vec_validate (ldpw->io_buffer, bytes_to_read);
1443 nbytes = libc_read (in_fd, ldpw->io_buffer, bytes_to_read);
1448 vec_reset_length (ldpw->io_buffer);
1455 size = vls_write (vlsh, ldpw->io_buffer, nbytes);
1458 if (size == VPPCOM_EAGAIN)
1460 if (flags & O_NONBLOCK)
1471 vec_reset_length (ldpw->io_buffer);
1480 ASSERT (n_bytes_left >= nbytes);
1481 n_bytes_left = n_bytes_left - nbytes;
1483 while (n_bytes_left > 0);
1486 vec_reset_length (ldpw->io_buffer);
1489 off_t off = lseek (in_fd, *offset, SEEK_SET);
1490 if (PREDICT_FALSE (off == -1))
1496 ASSERT (off == *offset);
1497 *offset += results + 1;
1509 size = libc_sendfile (out_fd, in_fd, offset, len);
1517 sendfile64 (int out_fd, int in_fd, off_t * offset, size_t len)
1519 return sendfile (out_fd, in_fd, offset, len);
1523 recv (int fd, void *buf, size_t n, int flags)
1530 vlsh = ldp_fd_to_vlsh (fd);
1531 if (vlsh != VLS_INVALID_HANDLE)
1533 size = vls_recvfrom (vlsh, buf, n, flags, NULL);
1542 size = libc_recv (fd, buf, n, flags);
1549 __recv_chk (int fd, void *buf, size_t n, size_t buflen, int flags)
1554 return recv (fd, buf, n, flags);
1558 ldp_vls_sendo (vls_handle_t vlsh, const void *buf, size_t n,
1559 vppcom_endpt_tlv_t *ep_tlv, int flags,
1560 __CONST_SOCKADDR_ARG _addr, socklen_t addr_len)
1562 const struct sockaddr *addr = SOCKADDR_GET_SA (_addr);
1563 vppcom_endpt_t *ep = 0;
1568 _ep.app_data = *ep_tlv;
1574 switch (addr->sa_family)
1577 ep->is_ip4 = VPPCOM_IS_IP4;
1579 (uint8_t *) & ((const struct sockaddr_in *) addr)->sin_addr;
1580 ep->port = (uint16_t) ((const struct sockaddr_in *) addr)->sin_port;
1584 ep->is_ip4 = VPPCOM_IS_IP6;
1586 (uint8_t *) & ((const struct sockaddr_in6 *) addr)->sin6_addr;
1588 (uint16_t) ((const struct sockaddr_in6 *) addr)->sin6_port;
1592 return EAFNOSUPPORT;
1596 return vls_sendto (vlsh, (void *) buf, n, flags, ep);
1600 ldp_vls_recvfrom (vls_handle_t vlsh, void *__restrict buf, size_t n, int flags,
1601 __SOCKADDR_ARG _addr, socklen_t *__restrict addr_len)
1603 u8 src_addr[sizeof (struct sockaddr_in6)];
1604 struct sockaddr *addr = SOCKADDR_GET_SA (_addr);
1612 size = vls_recvfrom (vlsh, buf, n, flags, &ep);
1616 rv = ldp_copy_ep_to_sockaddr (addr, addr_len, &ep);
1622 size = vls_recvfrom (vlsh, buf, n, flags, NULL);
1628 sendto (int fd, const void *buf, size_t n, int flags,
1629 __CONST_SOCKADDR_ARG _addr, socklen_t addr_len)
1631 const struct sockaddr *addr = SOCKADDR_GET_SA (_addr);
1637 vlsh = ldp_fd_to_vlsh (fd);
1638 if (vlsh != VLS_INVALID_HANDLE)
1640 size = ldp_vls_sendo (vlsh, buf, n, NULL, flags, addr, addr_len);
1649 size = libc_sendto (fd, buf, n, flags, addr, addr_len);
1656 recvfrom (int fd, void *__restrict buf, size_t n, int flags,
1657 __SOCKADDR_ARG addr, socklen_t * __restrict addr_len)
1664 vlsh = ldp_fd_to_vlsh (fd);
1665 if (vlsh != VLS_INVALID_HANDLE)
1667 size = ldp_vls_recvfrom (vlsh, buf, n, flags, addr, addr_len);
1676 size = libc_recvfrom (fd, buf, n, flags, addr, addr_len);
1683 sendmsg (int fd, const struct msghdr * msg, int flags)
1690 vlsh = ldp_fd_to_vlsh (fd);
1691 if (vlsh != VLS_INVALID_HANDLE)
1693 struct iovec *iov = msg->msg_iov;
1696 struct cmsghdr *cmsg;
1698 vppcom_endpt_tlv_t _app_data;
1699 vppcom_endpt_tlv_t *p_app_data = NULL;
1701 cmsg = CMSG_FIRSTHDR (msg);
1702 if (cmsg && cmsg->cmsg_type == UDP_SEGMENT)
1704 p_app_data = &_app_data;
1705 valp = (void *) CMSG_DATA (cmsg);
1706 p_app_data->data_type = VCL_UDP_SEGMENT;
1707 p_app_data->data_len = sizeof (*valp);
1708 p_app_data->value = *valp;
1711 for (i = 0; i < msg->msg_iovlen; ++i)
1714 ldp_vls_sendo (vlsh, iov[i].iov_base, iov[i].iov_len, p_app_data,
1715 flags, msg->msg_name, msg->msg_namelen);
1721 if (rv < iov[i].iov_len)
1726 if (rv < 0 && total == 0)
1736 size = libc_sendmsg (fd, msg, flags);
1744 sendmmsg (int fd, struct mmsghdr *vmessages, unsigned int vlen, int flags)
1747 const char *func_str;
1748 u32 sh = ldp_fd_to_vlsh (fd);
1752 if (sh != VLS_INVALID_HANDLE)
1754 clib_warning ("LDP<%d>: LDP-TBD", getpid ());
1760 func_str = "libc_sendmmsg";
1763 clib_warning ("LDP<%d>: fd %d (0x%x): calling %s(): "
1764 "vmessages %p, vlen %u, flags 0x%x",
1765 getpid (), fd, fd, func_str, vmessages, vlen, flags);
1767 size = libc_sendmmsg (fd, vmessages, vlen, flags);
1774 int errno_val = errno;
1776 clib_warning ("LDP<%d>: ERROR: fd %d (0x%x): %s() failed! "
1777 "rv %d, errno = %d", getpid (), fd, fd,
1778 func_str, size, errno_val);
1782 clib_warning ("LDP<%d>: fd %d (0x%x): returning %d (0x%x)",
1783 getpid (), fd, fd, size, size);
1790 recvmsg (int fd, struct msghdr * msg, int flags)
1797 vlsh = ldp_fd_to_vlsh (fd);
1798 if (vlsh != VLS_INVALID_HANDLE)
1800 struct iovec *iov = msg->msg_iov;
1801 ssize_t max_deq, total = 0;
1804 max_deq = vls_attr (vlsh, VPPCOM_ATTR_GET_NREAD, 0, 0);
1808 for (i = 0; i < msg->msg_iovlen; i++)
1810 rv = ldp_vls_recvfrom (vlsh, iov[i].iov_base, iov[i].iov_len, flags,
1811 (i == 0 ? msg->msg_name : NULL),
1812 (i == 0 ? &msg->msg_namelen : NULL));
1818 if (rv < iov[i].iov_len)
1821 if (total >= max_deq)
1825 if (rv < 0 && total == 0)
1835 size = libc_recvmsg (fd, msg, flags);
1843 recvmmsg (int fd, struct mmsghdr *vmessages,
1844 unsigned int vlen, int flags, struct timespec *tmo)
1846 ldp_worker_ctx_t *ldpw = ldp_worker_get_current ();
1851 sh = ldp_fd_to_vlsh (fd);
1853 if (sh != VLS_INVALID_HANDLE)
1860 if (PREDICT_FALSE (ldpw->clib_time.init_cpu_time == 0))
1861 clib_time_init (&ldpw->clib_time);
1864 time_out = (f64) tmo->tv_sec + (f64) tmo->tv_nsec / (f64) 1e9;
1865 time_out += clib_time_now (&ldpw->clib_time);
1869 time_out = (f64) ~0;
1872 while (nvecs < vlen)
1874 mh = &vmessages[nvecs];
1875 rv = recvmsg (fd, &mh->msg_hdr, flags);
1883 if (!time_out || clib_time_now (&ldpw->clib_time) >= time_out)
1889 return nvecs > 0 ? nvecs : rv;
1893 return libc_recvmmsg (fd, vmessages, vlen, flags, tmo);
1899 getsockopt (int fd, int level, int optname,
1900 void *__restrict optval, socklen_t * __restrict optlen)
1907 vlsh = ldp_fd_to_vlsh (fd);
1908 if (vlsh != VLS_INVALID_HANDLE)
1918 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_TCP_NODELAY,
1922 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_TCP_USER_MSS,
1926 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_TCP_KEEPIDLE,
1930 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_TCP_KEEPINTVL,
1934 if (optval && optlen && (*optlen == sizeof (struct tcp_info)))
1936 LDBG (1, "fd %d: vlsh %u SOL_TCP, TCP_INFO, optval %p, "
1937 "optlen %d: #LDP-NOP#", fd, vlsh, optval, *optlen);
1938 memset (optval, 0, *optlen);
1944 case TCP_CONGESTION:
1945 *optlen = strlen ("cubic");
1946 strncpy (optval, "cubic", *optlen + 1);
1950 LDBG (0, "ERROR: fd %d: getsockopt SOL_TCP: sid %u, "
1951 "optname %d unsupported!", fd, vlsh, optname);
1959 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_V6ONLY, optval, optlen);
1962 LDBG (0, "ERROR: fd %d: getsockopt SOL_IPV6: vlsh %u "
1963 "optname %d unsupported!", fd, vlsh, optname);
1971 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_LISTEN, optval, optlen);
1974 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_KEEPALIVE, optval, optlen);
1977 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_PROTOCOL, optval, optlen);
1978 *(int *) optval = *(int *) optval ? SOCK_DGRAM : SOCK_STREAM;
1981 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_TX_FIFO_LEN,
1985 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_RX_FIFO_LEN,
1989 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_REUSEADDR, optval, optlen);
1992 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_REUSEPORT, optval, optlen);
1995 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_BROADCAST, optval, optlen);
1998 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_DOMAIN, optval, optlen);
2001 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_ERROR, optval, optlen);
2003 case SO_BINDTODEVICE:
2007 LDBG (0, "ERROR: fd %d: getsockopt SOL_SOCKET: vlsh %u "
2008 "optname %d unsupported!", fd, vlsh, optname);
2016 if (rv != VPPCOM_OK)
2024 rv = libc_getsockopt (fd, level, optname, optval, optlen);
2031 setsockopt (int fd, int level, int optname,
2032 const void *optval, socklen_t optlen)
2039 vlsh = ldp_fd_to_vlsh (fd);
2040 if (vlsh != VLS_INVALID_HANDLE)
2050 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_TCP_NODELAY,
2051 (void *) optval, &optlen);
2054 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_TCP_USER_MSS,
2055 (void *) optval, &optlen);
2058 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_TCP_KEEPIDLE,
2059 (void *) optval, &optlen);
2062 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_TCP_KEEPINTVL,
2063 (void *) optval, &optlen);
2065 case TCP_CONGESTION:
2071 LDBG (0, "ERROR: fd %d: setsockopt() SOL_TCP: vlsh %u"
2072 "optname %d unsupported!", fd, vlsh, optname);
2080 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_V6ONLY,
2081 (void *) optval, &optlen);
2084 LDBG (0, "ERROR: fd %d: setsockopt SOL_IPV6: vlsh %u"
2085 "optname %d unsupported!", fd, vlsh, optname);
2093 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_KEEPALIVE,
2094 (void *) optval, &optlen);
2097 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_REUSEADDR,
2098 (void *) optval, &optlen);
2101 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_REUSEPORT, (void *) optval,
2105 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_BROADCAST,
2106 (void *) optval, &optlen);
2112 LDBG (0, "ERROR: fd %d: setsockopt SOL_SOCKET: vlsh %u "
2113 "optname %d unsupported!", fd, vlsh, optname);
2121 if (rv != VPPCOM_OK)
2129 rv = libc_setsockopt (fd, level, optname, optval, optlen);
2136 listen (int fd, int n)
2143 vlsh = ldp_fd_to_vlsh (fd);
2144 if (vlsh != VLS_INVALID_HANDLE)
2146 LDBG (0, "fd %d: calling vls_listen: vlsh %u, n %d", fd, vlsh, n);
2148 rv = vls_listen (vlsh, n);
2149 if (rv != VPPCOM_OK)
2157 LDBG (0, "fd %d: calling libc_listen(): n %d", fd, n);
2158 rv = libc_listen (fd, n);
2161 LDBG (1, "fd %d: returning %d", fd, rv);
2166 ldp_accept4 (int listen_fd, __SOCKADDR_ARG _addr,
2167 socklen_t *__restrict addr_len, int flags)
2169 struct sockaddr *addr = SOCKADDR_GET_SA (_addr);
2170 vls_handle_t listen_vlsh, accept_vlsh;
2175 listen_vlsh = ldp_fd_to_vlsh (listen_fd);
2176 if (listen_vlsh != VLS_INVALID_HANDLE)
2179 u8 src_addr[sizeof (struct sockaddr_in6)];
2180 memset (&ep, 0, sizeof (ep));
2183 LDBG (0, "listen fd %d: calling vppcom_session_accept: listen sid %u,"
2184 " ep %p, flags 0x%x", listen_fd, listen_vlsh, &ep, flags);
2186 accept_vlsh = vls_accept (listen_vlsh, &ep, flags);
2187 if (accept_vlsh < 0)
2189 errno = -accept_vlsh;
2194 rv = ldp_copy_ep_to_sockaddr (addr, addr_len, &ep);
2195 if (rv != VPPCOM_OK)
2197 (void) vls_close (accept_vlsh);
2203 rv = ldp_vlsh_to_fd (accept_vlsh);
2209 LDBG (0, "listen fd %d: calling libc_accept4(): addr %p, addr_len %p,"
2210 " flags 0x%x", listen_fd, addr, addr_len, flags);
2212 rv = libc_accept4 (listen_fd, addr, addr_len, flags);
2215 LDBG (1, "listen fd %d: accept returning %d", listen_fd, rv);
2221 accept4 (int fd, __SOCKADDR_ARG addr, socklen_t * __restrict addr_len,
2224 return ldp_accept4 (fd, addr, addr_len, flags);
2228 accept (int fd, __SOCKADDR_ARG addr, socklen_t * __restrict addr_len)
2230 return ldp_accept4 (fd, addr, addr_len, 0);
2234 shutdown (int fd, int how)
2241 vlsh = ldp_fd_to_vlsh (fd);
2242 if (vlsh != VLS_INVALID_HANDLE)
2244 LDBG (0, "called shutdown: fd %u vlsh %u how %d", fd, vlsh, how);
2245 rv = vls_shutdown (vlsh, how);
2249 LDBG (0, "fd %d: calling libc_shutdown: how %d", fd, how);
2250 rv = libc_shutdown (fd, how);
2257 epoll_create1 (int flags)
2259 ldp_worker_ctx_t *ldpw = ldp_worker_get_current ();
2265 if (ldp->vcl_needs_real_epoll || vls_use_real_epoll ())
2267 /* Make sure workers have been allocated */
2270 ldp_alloc_workers ();
2271 ldpw = ldp_worker_get_current ();
2273 rv = libc_epoll_create1 (flags);
2274 ldp->vcl_needs_real_epoll = 0;
2275 ldpw->vcl_mq_epfd = rv;
2276 LDBG (0, "created vcl epfd %u", rv);
2280 vlsh = vls_epoll_create ();
2281 if (PREDICT_FALSE (vlsh == VLS_INVALID_HANDLE))
2288 rv = ldp_vlsh_to_fd (vlsh);
2290 LDBG (0, "epoll_create epfd %u vlsh %u", rv, vlsh);
2295 epoll_create (int size)
2297 return epoll_create1 (0);
2301 epoll_ctl (int epfd, int op, int fd, struct epoll_event *event)
2303 vls_handle_t vep_vlsh, vlsh;
2308 vep_vlsh = ldp_fd_to_vlsh (epfd);
2309 if (PREDICT_FALSE (vep_vlsh == VLS_INVALID_HANDLE))
2311 /* The LDP epoll_create1 always creates VCL epfd's.
2312 * The app should never have a kernel base epoll fd unless it
2313 * was acquired outside of the LD_PRELOAD process context.
2314 * In any case, if we get one, punt it to libc_epoll_ctl.
2316 LDBG (1, "epfd %d: calling libc_epoll_ctl: op %d, fd %d"
2317 " event %p", epfd, op, fd, event);
2319 rv = libc_epoll_ctl (epfd, op, fd, event);
2323 vlsh = ldp_fd_to_vlsh (fd);
2325 LDBG (0, "epfd %d ep_vlsh %d, fd %u vlsh %d, op %u", epfd, vep_vlsh, fd,
2328 if (vlsh != VLS_INVALID_HANDLE)
2330 LDBG (1, "epfd %d: calling vls_epoll_ctl: ep_vlsh %d op %d, vlsh %u,"
2331 " event %p", epfd, vep_vlsh, op, vlsh, event);
2333 rv = vls_epoll_ctl (vep_vlsh, op, vlsh, event);
2334 if (rv != VPPCOM_OK)
2343 u32 size = sizeof (epfd);
2345 libc_epfd = vls_attr (vep_vlsh, VPPCOM_ATTR_GET_LIBC_EPFD, 0, 0);
2348 LDBG (1, "epfd %d, vep_vlsh %d calling libc_epoll_create1: "
2349 "EPOLL_CLOEXEC", epfd, vep_vlsh);
2351 libc_epfd = libc_epoll_create1 (EPOLL_CLOEXEC);
2358 rv = vls_attr (vep_vlsh, VPPCOM_ATTR_SET_LIBC_EPFD, &libc_epfd,
2367 else if (PREDICT_FALSE (libc_epfd < 0))
2374 LDBG (1, "epfd %d: calling libc_epoll_ctl: libc_epfd %d, op %d, fd %d,"
2375 " event %p", epfd, libc_epfd, op, fd, event);
2377 rv = libc_epoll_ctl (libc_epfd, op, fd, event);
2385 ldp_epoll_pwait (int epfd, struct epoll_event *events, int maxevents,
2386 int timeout, const sigset_t * sigmask)
2388 ldp_worker_ctx_t *ldpw;
2389 double time_to_wait = (double) 0, max_time;
2390 int libc_epfd, rv = 0;
2391 vls_handle_t ep_vlsh;
2395 if (PREDICT_FALSE (!events || (timeout < -1)))
2401 if (PREDICT_FALSE (vppcom_worker_index () == ~0))
2402 vls_register_vcl_worker ();
2404 ldpw = ldp_worker_get_current ();
2405 if (epfd == ldpw->vcl_mq_epfd)
2406 return libc_epoll_pwait (epfd, events, maxevents, timeout, sigmask);
2408 ep_vlsh = ldp_fd_to_vlsh (epfd);
2409 if (PREDICT_FALSE (ep_vlsh == VLS_INVALID_HANDLE))
2411 LDBG (0, "epfd %d: bad ep_vlsh %d!", epfd, ep_vlsh);
2416 if (PREDICT_FALSE (ldpw->clib_time.init_cpu_time == 0))
2417 clib_time_init (&ldpw->clib_time);
2418 time_to_wait = ((timeout >= 0) ? (double) timeout / 1000 : 0);
2419 max_time = clib_time_now (&ldpw->clib_time) + time_to_wait;
2421 libc_epfd = vls_attr (ep_vlsh, VPPCOM_ATTR_GET_LIBC_EPFD, 0, 0);
2422 if (PREDICT_FALSE (libc_epfd < 0))
2429 LDBG (2, "epfd %d: vep_idx %d, libc_epfd %d, events %p, maxevents %d, "
2430 "timeout %d, sigmask %p: time_to_wait %.02f", epfd, ep_vlsh,
2431 libc_epfd, events, maxevents, timeout, sigmask, time_to_wait);
2434 if (!ldpw->epoll_wait_vcl)
2436 rv = vls_epoll_wait (ep_vlsh, events, maxevents, 0);
2439 ldpw->epoll_wait_vcl = 1;
2450 ldpw->epoll_wait_vcl = 0;
2454 rv = libc_epoll_pwait (libc_epfd, events, maxevents, 0, sigmask);
2459 while ((timeout == -1) || (clib_time_now (&ldpw->clib_time) < max_time));
2466 ldp_epoll_pwait_eventfd (int epfd, struct epoll_event *events,
2467 int maxevents, int timeout, const sigset_t * sigmask)
2469 ldp_worker_ctx_t *ldpw;
2470 int libc_epfd, rv = 0, num_ev;
2471 vls_handle_t ep_vlsh;
2475 if (PREDICT_FALSE (!events || (timeout < -1)))
2481 /* Make sure the vcl worker is valid. Could be that epoll fd was created on
2482 * one thread but it is now used on another */
2483 if (PREDICT_FALSE (vppcom_worker_index () == ~0))
2484 vls_register_vcl_worker ();
2486 ldpw = ldp_worker_get_current ();
2487 if (epfd == ldpw->vcl_mq_epfd)
2488 return libc_epoll_pwait (epfd, events, maxevents, timeout, sigmask);
2490 ep_vlsh = ldp_fd_to_vlsh (epfd);
2491 if (PREDICT_FALSE (ep_vlsh == VLS_INVALID_HANDLE))
2493 LDBG (0, "epfd %d: bad ep_vlsh %d!", epfd, ep_vlsh);
2498 libc_epfd = vls_attr (ep_vlsh, VPPCOM_ATTR_GET_LIBC_EPFD, 0, 0);
2499 if (PREDICT_FALSE (!libc_epfd))
2501 u32 size = sizeof (epfd);
2503 LDBG (1, "epfd %d, vep_vlsh %d calling libc_epoll_create1: "
2504 "EPOLL_CLOEXEC", epfd, ep_vlsh);
2505 libc_epfd = libc_epoll_create1 (EPOLL_CLOEXEC);
2512 rv = vls_attr (ep_vlsh, VPPCOM_ATTR_SET_LIBC_EPFD, &libc_epfd, &size);
2520 if (PREDICT_FALSE (libc_epfd <= 0))
2527 if (PREDICT_FALSE (!ldpw->mq_epfd_added))
2529 struct epoll_event e = { 0 };
2531 e.data.fd = ldpw->vcl_mq_epfd;
2532 if (libc_epoll_ctl (libc_epfd, EPOLL_CTL_ADD, ldpw->vcl_mq_epfd, &e) <
2535 LDBG (0, "epfd %d, add libc mq epoll fd %d to libc epoll fd %d",
2536 epfd, ldpw->vcl_mq_epfd, libc_epfd);
2540 ldpw->mq_epfd_added = 1;
2543 /* Request to only drain unhandled to prevent libc_epoll_wait starved */
2544 rv = vls_epoll_wait (ep_vlsh, events, maxevents, -2);
2547 else if (PREDICT_FALSE (rv < 0))
2554 rv = libc_epoll_pwait (libc_epfd, events, maxevents, timeout, sigmask);
2557 for (int i = 0; i < rv; i++)
2559 if (events[i].data.fd == ldpw->vcl_mq_epfd)
2561 /* We should remove mq epoll fd from events. */
2565 events[i].events = events[rv].events;
2566 events[i].data.u64 = events[rv].data.u64;
2568 num_ev = vls_epoll_wait (ep_vlsh, &events[rv], maxevents - rv, 0);
2569 if (PREDICT_TRUE (num_ev > 0))
2580 epoll_pwait (int epfd, struct epoll_event *events,
2581 int maxevents, int timeout, const sigset_t * sigmask)
2583 if (vls_use_eventfd ())
2584 return ldp_epoll_pwait_eventfd (epfd, events, maxevents, timeout,
2587 return ldp_epoll_pwait (epfd, events, maxevents, timeout, sigmask);
2591 epoll_wait (int epfd, struct epoll_event *events, int maxevents, int timeout)
2593 if (vls_use_eventfd ())
2594 return ldp_epoll_pwait_eventfd (epfd, events, maxevents, timeout, NULL);
2596 return ldp_epoll_pwait (epfd, events, maxevents, timeout, NULL);
2600 poll (struct pollfd *fds, nfds_t nfds, int timeout)
2602 ldp_worker_ctx_t *ldpw = ldp_worker_get_current ();
2603 int rv, i, n_revents = 0;
2608 LDBG (3, "fds %p, nfds %ld, timeout %d", fds, nfds, timeout);
2610 if (PREDICT_FALSE (ldpw->clib_time.init_cpu_time == 0))
2611 clib_time_init (&ldpw->clib_time);
2613 max_time = (timeout >= 0) ? (f64) timeout / 1000 : 0;
2614 max_time += clib_time_now (&ldpw->clib_time);
2616 for (i = 0; i < nfds; i++)
2621 vlsh = ldp_fd_to_vlsh (fds[i].fd);
2622 if (vlsh != VLS_INVALID_HANDLE)
2624 fds[i].fd = -fds[i].fd;
2625 vec_add2 (ldpw->vcl_poll, vp, 1);
2627 vp->sh = vlsh_to_sh (vlsh);
2628 vp->events = fds[i].events;
2629 #ifdef __USE_XOPEN2K
2630 if (fds[i].events & POLLRDNORM)
2631 vp->events |= POLLIN;
2632 if (fds[i].events & POLLWRNORM)
2633 vp->events |= POLLOUT;
2635 vp->revents = fds[i].revents;
2639 vec_add1 (ldpw->libc_poll, fds[i]);
2640 vec_add1 (ldpw->libc_poll_idxs, i);
2646 if (vec_len (ldpw->vcl_poll))
2648 rv = vppcom_poll (ldpw->vcl_poll, vec_len (ldpw->vcl_poll), 0);
2659 if (vec_len (ldpw->libc_poll))
2661 rv = libc_poll (ldpw->libc_poll, vec_len (ldpw->libc_poll), 0);
2674 while ((timeout < 0) || (clib_time_now (&ldpw->clib_time) < max_time));
2678 vec_foreach (vp, ldpw->vcl_poll)
2680 fds[vp->fds_ndx].fd = -fds[vp->fds_ndx].fd;
2681 fds[vp->fds_ndx].revents = vp->revents;
2682 #ifdef __USE_XOPEN2K
2683 if ((fds[vp->fds_ndx].revents & POLLIN) &&
2684 (fds[vp->fds_ndx].events & POLLRDNORM))
2685 fds[vp->fds_ndx].revents |= POLLRDNORM;
2686 if ((fds[vp->fds_ndx].revents & POLLOUT) &&
2687 (fds[vp->fds_ndx].events & POLLWRNORM))
2688 fds[vp->fds_ndx].revents |= POLLWRNORM;
2691 vec_reset_length (ldpw->vcl_poll);
2693 for (i = 0; i < vec_len (ldpw->libc_poll); i++)
2695 fds[ldpw->libc_poll_idxs[i]].revents = ldpw->libc_poll[i].revents;
2697 vec_reset_length (ldpw->libc_poll_idxs);
2698 vec_reset_length (ldpw->libc_poll);
2705 ppoll (struct pollfd *fds, nfds_t nfds,
2706 const struct timespec *timeout, const sigset_t * sigmask)
2710 clib_warning ("LDP<%d>: LDP-TBD", getpid ());
2718 void CONSTRUCTOR_ATTRIBUTE ldp_constructor (void);
2720 void DESTRUCTOR_ATTRIBUTE ldp_destructor (void);
2723 * This function is called when the library is loaded
2726 ldp_constructor (void)
2728 swrap_constructor ();
2729 if (ldp_init () != 0)
2731 fprintf (stderr, "\nLDP<%d>: ERROR: ldp_constructor: failed!\n",
2735 else if (LDP_DEBUG > 0)
2736 clib_warning ("LDP<%d>: LDP constructor: done!\n", getpid ());
2740 * This function is called when the library is unloaded
2743 ldp_destructor (void)
2746 swrap_destructor ();
2751 /* Don't use clib_warning() here because that calls writev()
2752 * which will call ldp_init().
2755 fprintf (stderr, "%s:%d: LDP<%d>: LDP destructor: done!\n",
2756 __func__, __LINE__, getpid ());
2761 * fd.io coding-style-patch-verification: ON
2764 * eval: (c-set-style "gnu")