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.
22 #include <sys/resource.h>
23 #include <netinet/tcp.h>
25 #include <vcl/ldp_socket_wrapper.h>
29 #include <vcl/vcl_locked.h>
30 #include <vppinfra/time.h>
31 #include <vppinfra/bitmap.h>
32 #include <vppinfra/lock.h>
33 #include <vppinfra/pool.h>
34 #include <vppinfra/hash.h>
36 #define HAVE_CONSTRUCTOR_ATTRIBUTE
37 #ifdef HAVE_CONSTRUCTOR_ATTRIBUTE
38 #define CONSTRUCTOR_ATTRIBUTE \
39 __attribute__ ((constructor))
41 #define CONSTRUCTOR_ATTRIBUTE
42 #endif /* HAVE_CONSTRUCTOR_ATTRIBUTE */
44 #define HAVE_DESTRUCTOR_ATTRIBUTE
45 #ifdef HAVE_DESTRUCTOR_ATTRIBUTE
46 #define DESTRUCTOR_ATTRIBUTE \
47 __attribute__ ((destructor))
49 #define DESTRUCTOR_ATTRIBUTE
52 #define LDP_MAX_NWORKERS 32
54 typedef struct ldp_worker_ctx_
57 clib_time_t clib_time;
62 clib_bitmap_t *rd_bitmap;
63 clib_bitmap_t *wr_bitmap;
64 clib_bitmap_t *ex_bitmap;
65 clib_bitmap_t *si_rd_bitmap;
66 clib_bitmap_t *si_wr_bitmap;
67 clib_bitmap_t *si_ex_bitmap;
68 clib_bitmap_t *libc_rd_bitmap;
69 clib_bitmap_t *libc_wr_bitmap;
70 clib_bitmap_t *libc_ex_bitmap;
76 struct pollfd *libc_poll;
88 /* clib_bitmap_t, fd_mask and vcl_si_set are used interchangeably. Make sure
89 * they are the same size */
90 STATIC_ASSERT (sizeof (clib_bitmap_t) == sizeof (fd_mask),
91 "ldp bitmap size mismatch");
92 STATIC_ASSERT (sizeof (vcl_si_set) == sizeof (fd_mask),
93 "ldp bitmap size mismatch");
97 ldp_worker_ctx_t *workers;
99 char app_name[LDP_APP_NAME_MAX];
105 /** vcl needs next epoll_create to go to libc_epoll */
106 u8 vcl_needs_real_epoll;
109 #define LDP_DEBUG ldp->debug
111 #define LDBG(_lvl, _fmt, _args...) \
112 if (ldp->debug > _lvl) \
114 int errno_saved = errno; \
115 fprintf (stderr, "ldp<%d>: " _fmt "\n", getpid(), ##_args); \
116 errno = errno_saved; \
119 static ldp_main_t ldp_main = {
120 .vlsh_bit_val = (1 << LDP_SID_BIT_MIN),
121 .vlsh_bit_mask = (1 << LDP_SID_BIT_MIN) - 1,
122 .debug = LDP_DEBUG_INIT,
123 .transparent_tls = 0,
126 static ldp_main_t *ldp = &ldp_main;
128 static inline ldp_worker_ctx_t *
129 ldp_worker_get_current (void)
131 return (ldp->workers + vppcom_worker_index ());
135 * RETURN: 0 on success or -1 on error.
138 ldp_set_app_name (char *app_name)
140 snprintf (ldp->app_name, LDP_APP_NAME_MAX,
141 "ldp-%d-%s", getpid (), app_name);
147 if (ldp->app_name[0] == '\0')
148 ldp_set_app_name ("app");
150 return ldp->app_name;
154 ldp_vlsh_to_fd (vls_handle_t vlsh)
156 return (vlsh + ldp->vlsh_bit_val);
159 static inline vls_handle_t
160 ldp_fd_to_vlsh (int fd)
162 if (fd < ldp->vlsh_bit_val)
163 return VLS_INVALID_HANDLE;
165 return (fd - ldp->vlsh_bit_val);
169 ldp_alloc_workers (void)
173 pool_alloc (ldp->workers, LDP_MAX_NWORKERS);
179 ldp_worker_ctx_t *ldpw;
182 if (PREDICT_TRUE (ldp->init))
186 ldp->vcl_needs_real_epoll = 1;
187 rv = vls_app_create (ldp_get_app_name ());
190 ldp->vcl_needs_real_epoll = 0;
191 if (rv == VPPCOM_EEXIST)
193 LDBG (2, "\nERROR: ldp_init: vppcom_app_create()"
194 " failed! rv = %d (%s)\n", rv, vppcom_retval_str (rv));
198 ldp->vcl_needs_real_epoll = 0;
199 ldp_alloc_workers ();
200 ldpw = ldp_worker_get_current ();
202 char *env_var_str = getenv (LDP_ENV_DEBUG);
206 if (sscanf (env_var_str, "%u", &tmp) != 1)
207 clib_warning ("LDP<%d>: WARNING: Invalid LDP debug level specified in"
208 " the env var " LDP_ENV_DEBUG " (%s)!", getpid (),
213 LDBG (0, "configured LDP debug level (%u) from env var "
214 LDP_ENV_DEBUG "!", ldp->debug);
218 env_var_str = getenv (LDP_ENV_APP_NAME);
221 ldp_set_app_name (env_var_str);
222 LDBG (0, "configured LDP app name (%s) from the env var "
223 LDP_ENV_APP_NAME "!", ldp->app_name);
226 env_var_str = getenv (LDP_ENV_SID_BIT);
230 if (sscanf (env_var_str, "%u", &sb) != 1)
232 LDBG (0, "WARNING: Invalid LDP sid bit specified in the env var "
233 LDP_ENV_SID_BIT " (%s)! sid bit value %d (0x%x)", env_var_str,
234 ldp->vlsh_bit_val, ldp->vlsh_bit_val);
236 else if (sb < LDP_SID_BIT_MIN)
238 ldp->vlsh_bit_val = (1 << LDP_SID_BIT_MIN);
239 ldp->vlsh_bit_mask = ldp->vlsh_bit_val - 1;
241 LDBG (0, "WARNING: LDP sid bit (%u) specified in the env var "
242 LDP_ENV_SID_BIT " (%s) is too small. Using LDP_SID_BIT_MIN"
243 " (%d)! sid bit value %d (0x%x)", sb, env_var_str,
244 LDP_SID_BIT_MIN, ldp->vlsh_bit_val, ldp->vlsh_bit_val);
246 else if (sb > LDP_SID_BIT_MAX)
248 ldp->vlsh_bit_val = (1 << LDP_SID_BIT_MAX);
249 ldp->vlsh_bit_mask = ldp->vlsh_bit_val - 1;
251 LDBG (0, "WARNING: LDP sid bit (%u) specified in the env var "
252 LDP_ENV_SID_BIT " (%s) is too big. Using LDP_SID_BIT_MAX"
253 " (%d)! sid bit value %d (0x%x)", sb, env_var_str,
254 LDP_SID_BIT_MAX, ldp->vlsh_bit_val, ldp->vlsh_bit_val);
258 ldp->vlsh_bit_val = (1 << sb);
259 ldp->vlsh_bit_mask = ldp->vlsh_bit_val - 1;
261 LDBG (0, "configured LDP sid bit (%u) from "
262 LDP_ENV_SID_BIT "! sid bit value %d (0x%x)", sb,
263 ldp->vlsh_bit_val, ldp->vlsh_bit_val);
266 /* Make sure there are enough bits in the fd set for vcl sessions */
267 if (ldp->vlsh_bit_val > FD_SETSIZE / 2)
269 LDBG (0, "ERROR: LDP vlsh bit value %d > FD_SETSIZE/2 %d!",
270 ldp->vlsh_bit_val, FD_SETSIZE / 2);
275 env_var_str = getenv (LDP_ENV_TLS_TRANS);
278 ldp->transparent_tls = 1;
282 pool_foreach (ldpw, ldp->workers) {
283 clib_memset (&ldpw->clib_time, 0, sizeof (ldpw->clib_time));
287 LDBG (0, "LDP initialization: done!");
298 if ((errno = -ldp_init ()))
301 vlsh = ldp_fd_to_vlsh (fd);
302 if (vlsh != VLS_INVALID_HANDLE)
304 epfd = vls_attr (vlsh, VPPCOM_ATTR_GET_LIBC_EPFD, 0, 0);
307 LDBG (0, "fd %d: calling libc_close: epfd %u", fd, epfd);
309 rv = libc_close (epfd);
312 u32 size = sizeof (epfd);
315 (void) vls_attr (vlsh, VPPCOM_ATTR_SET_LIBC_EPFD, &epfd, &size);
318 else if (PREDICT_FALSE (epfd < 0))
325 LDBG (0, "fd %d: calling vls_close: vlsh %u", fd, vlsh);
327 rv = vls_close (vlsh);
336 LDBG (0, "fd %d: calling libc_close", fd);
337 rv = libc_close (fd);
345 read (int fd, void *buf, size_t nbytes)
350 if ((errno = -ldp_init ()))
353 vlsh = ldp_fd_to_vlsh (fd);
354 if (vlsh != VLS_INVALID_HANDLE)
356 size = vls_read (vlsh, buf, nbytes);
365 size = libc_read (fd, buf, nbytes);
372 readv (int fd, const struct iovec * iov, int iovcnt)
374 int rv = 0, i, total = 0;
378 if ((errno = -ldp_init ()))
381 vlsh = ldp_fd_to_vlsh (fd);
382 if (vlsh != VLS_INVALID_HANDLE)
384 for (i = 0; i < iovcnt; ++i)
386 rv = vls_read (vlsh, iov[i].iov_base, iov[i].iov_len);
392 if (rv < iov[i].iov_len)
396 if (rv < 0 && total == 0)
406 size = libc_readv (fd, iov, iovcnt);
413 write (int fd, const void *buf, size_t nbytes)
418 if ((errno = -ldp_init ()))
421 vlsh = ldp_fd_to_vlsh (fd);
422 if (vlsh != VLS_INVALID_HANDLE)
424 size = vls_write_msg (vlsh, (void *) buf, nbytes);
433 size = libc_write (fd, buf, nbytes);
440 writev (int fd, const struct iovec * iov, int iovcnt)
442 ssize_t size = 0, total = 0;
446 if ((errno = -ldp_init ()))
449 vlsh = ldp_fd_to_vlsh (fd);
450 if (vlsh != VLS_INVALID_HANDLE)
452 for (i = 0; i < iovcnt; ++i)
454 rv = vls_write_msg (vlsh, iov[i].iov_base, iov[i].iov_len);
460 if (rv < iov[i].iov_len)
465 if (rv < 0 && total == 0)
475 size = libc_writev (fd, iov, iovcnt);
482 fcntl_internal (int fd, int cmd, va_list ap)
487 vlsh = ldp_fd_to_vlsh (fd);
488 LDBG (0, "fd %u vlsh %d, cmd %u", fd, vlsh, cmd);
489 if (vlsh != VLS_INVALID_HANDLE)
491 int flags = va_arg (ap, int);
494 size = sizeof (flags);
499 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_FLAGS, &flags, &size);
503 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_FLAGS, &flags, &size);
508 /* TODO handle this */
509 LDBG (0, "F_SETFD ignored flags %u", flags);
525 rv = libc_vfcntl64 (fd, cmd, ap);
527 rv = libc_vfcntl (fd, cmd, ap);
535 fcntl (int fd, int cmd, ...)
540 if ((errno = -ldp_init ()))
544 rv = fcntl_internal (fd, cmd, ap);
551 fcntl64 (int fd, int cmd, ...)
556 if ((errno = -ldp_init ()))
560 rv = fcntl_internal (fd, cmd, ap);
566 ioctl (int fd, unsigned long int cmd, ...)
572 if ((errno = -ldp_init ()))
577 vlsh = ldp_fd_to_vlsh (fd);
578 if (vlsh != VLS_INVALID_HANDLE)
583 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_NREAD, 0, 0);
588 u32 flags = va_arg (ap, int) ? O_NONBLOCK : 0;
589 u32 size = sizeof (flags);
591 /* TBD: When VPPCOM_ATTR_[GS]ET_FLAGS supports flags other than
592 * non-blocking, the flags should be read here and merged
595 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_FLAGS, &flags, &size);
611 rv = libc_vioctl (fd, cmd, ap);
619 ldp_select_init_maps (fd_set * __restrict original,
620 clib_bitmap_t ** resultb, clib_bitmap_t ** libcb,
621 clib_bitmap_t ** vclb, int nfds, u32 minbits,
622 u32 n_bytes, uword * si_bits, uword * libc_bits)
624 uword si_bits_set, libc_bits_set;
628 clib_bitmap_validate (*vclb, minbits);
629 clib_bitmap_validate (*libcb, minbits);
630 clib_bitmap_validate (*resultb, minbits);
631 clib_memcpy_fast (*resultb, original, n_bytes);
632 memset (original, 0, n_bytes);
635 clib_bitmap_foreach (fd, *resultb) {
638 vlsh = ldp_fd_to_vlsh (fd);
639 if (vlsh == VLS_INVALID_HANDLE)
640 clib_bitmap_set_no_check (*libcb, fd, 1);
642 *vclb = clib_bitmap_set (*vclb, vlsh_to_session_index (vlsh), 1);
646 si_bits_set = clib_bitmap_last_set (*vclb) + 1;
647 *si_bits = (si_bits_set > *si_bits) ? si_bits_set : *si_bits;
648 clib_bitmap_validate (*resultb, *si_bits);
650 libc_bits_set = clib_bitmap_last_set (*libcb) + 1;
651 *libc_bits = (libc_bits_set > *libc_bits) ? libc_bits_set : *libc_bits;
655 ldp_select_vcl_map_to_libc (clib_bitmap_t * vclb, fd_set * __restrict libcb)
665 clib_bitmap_foreach (si, vclb) {
666 vlsh = vls_session_index_to_vlsh (si);
667 ASSERT (vlsh != VLS_INVALID_HANDLE);
668 fd = ldp_vlsh_to_fd (vlsh);
669 if (PREDICT_FALSE (fd < 0))
682 ldp_select_libc_map_merge (clib_bitmap_t * result, fd_set * __restrict libcb)
690 clib_bitmap_foreach (fd, result)
691 FD_SET ((int)fd, libcb);
696 ldp_pselect (int nfds, fd_set * __restrict readfds,
697 fd_set * __restrict writefds,
698 fd_set * __restrict exceptfds,
699 const struct timespec *__restrict timeout,
700 const __sigset_t * __restrict sigmask)
702 u32 minbits = clib_max (nfds, BITS (uword)), n_bytes;
703 ldp_worker_ctx_t *ldpw = ldp_worker_get_current ();
704 struct timespec libc_tspec = { 0 };
705 f64 time_out, vcl_timeout = 0;
706 uword si_bits, libc_bits;
707 int rv, bits_set = 0;
715 if (PREDICT_FALSE (ldpw->clib_time.init_cpu_time == 0))
716 clib_time_init (&ldpw->clib_time);
720 time_out = (timeout->tv_sec == 0 && timeout->tv_nsec == 0) ?
721 (f64) 0 : (f64) timeout->tv_sec + (f64) timeout->tv_nsec / (f64) 1e9;
723 /* select as fine grained sleep */
726 time_out += clib_time_now (&ldpw->clib_time);
727 while (clib_time_now (&ldpw->clib_time) < time_out)
740 if (nfds <= ldp->vlsh_bit_val)
742 rv = libc_pselect (nfds, readfds, writefds, exceptfds,
747 si_bits = libc_bits = 0;
748 n_bytes = nfds / 8 + ((nfds % 8) ? 1 : 0);
751 ldp_select_init_maps (readfds, &ldpw->rd_bitmap, &ldpw->libc_rd_bitmap,
752 &ldpw->si_rd_bitmap, nfds, minbits, n_bytes,
753 &si_bits, &libc_bits);
755 ldp_select_init_maps (writefds, &ldpw->wr_bitmap,
756 &ldpw->libc_wr_bitmap, &ldpw->si_wr_bitmap, nfds,
757 minbits, n_bytes, &si_bits, &libc_bits);
759 ldp_select_init_maps (exceptfds, &ldpw->ex_bitmap,
760 &ldpw->libc_ex_bitmap, &ldpw->si_ex_bitmap, nfds,
761 minbits, n_bytes, &si_bits, &libc_bits);
763 if (PREDICT_FALSE (!si_bits && !libc_bits))
771 libc_tspec = timeout ? *timeout : libc_tspec;
778 clib_memcpy_fast (ldpw->rd_bitmap, ldpw->si_rd_bitmap,
779 vec_len (ldpw->si_rd_bitmap) *
780 sizeof (clib_bitmap_t));
782 clib_memcpy_fast (ldpw->wr_bitmap, ldpw->si_wr_bitmap,
783 vec_len (ldpw->si_wr_bitmap) *
784 sizeof (clib_bitmap_t));
786 clib_memcpy_fast (ldpw->ex_bitmap, ldpw->si_ex_bitmap,
787 vec_len (ldpw->si_ex_bitmap) *
788 sizeof (clib_bitmap_t));
790 rv = vls_select (si_bits, readfds ? ldpw->rd_bitmap : NULL,
791 writefds ? ldpw->wr_bitmap : NULL,
792 exceptfds ? ldpw->ex_bitmap : NULL, vcl_timeout);
801 if (ldp_select_vcl_map_to_libc (ldpw->rd_bitmap, readfds))
807 if (ldp_select_vcl_map_to_libc (ldpw->wr_bitmap, writefds))
813 if (ldp_select_vcl_map_to_libc (ldpw->ex_bitmap, exceptfds))
824 clib_memcpy_fast (ldpw->rd_bitmap, ldpw->libc_rd_bitmap,
825 vec_len (ldpw->libc_rd_bitmap) *
826 sizeof (clib_bitmap_t));
828 clib_memcpy_fast (ldpw->wr_bitmap, ldpw->libc_wr_bitmap,
829 vec_len (ldpw->libc_wr_bitmap) *
830 sizeof (clib_bitmap_t));
832 clib_memcpy_fast (ldpw->ex_bitmap, ldpw->libc_ex_bitmap,
833 vec_len (ldpw->libc_ex_bitmap) *
834 sizeof (clib_bitmap_t));
836 rv = libc_pselect (libc_bits,
837 readfds ? (fd_set *) ldpw->rd_bitmap : NULL,
838 writefds ? (fd_set *) ldpw->wr_bitmap : NULL,
839 exceptfds ? (fd_set *) ldpw->ex_bitmap : NULL,
840 &libc_tspec, sigmask);
843 ldp_select_libc_map_merge (ldpw->rd_bitmap, readfds);
844 ldp_select_libc_map_merge (ldpw->wr_bitmap, writefds);
845 ldp_select_libc_map_merge (ldpw->ex_bitmap, exceptfds);
856 while ((time_out == -1) || (clib_time_now (&ldpw->clib_time) < time_out));
860 /* TBD: set timeout to amount of time left */
861 clib_bitmap_zero (ldpw->rd_bitmap);
862 clib_bitmap_zero (ldpw->si_rd_bitmap);
863 clib_bitmap_zero (ldpw->libc_rd_bitmap);
864 clib_bitmap_zero (ldpw->wr_bitmap);
865 clib_bitmap_zero (ldpw->si_wr_bitmap);
866 clib_bitmap_zero (ldpw->libc_wr_bitmap);
867 clib_bitmap_zero (ldpw->ex_bitmap);
868 clib_bitmap_zero (ldpw->si_ex_bitmap);
869 clib_bitmap_zero (ldpw->libc_ex_bitmap);
875 select (int nfds, fd_set * __restrict readfds,
876 fd_set * __restrict writefds,
877 fd_set * __restrict exceptfds, struct timeval *__restrict timeout)
879 struct timespec tspec;
883 tspec.tv_sec = timeout->tv_sec;
884 tspec.tv_nsec = timeout->tv_usec * 1000;
886 return ldp_pselect (nfds, readfds, writefds, exceptfds,
887 timeout ? &tspec : NULL, NULL);
892 pselect (int nfds, fd_set * __restrict readfds,
893 fd_set * __restrict writefds,
894 fd_set * __restrict exceptfds,
895 const struct timespec *__restrict timeout,
896 const __sigset_t * __restrict sigmask)
898 return ldp_pselect (nfds, readfds, writefds, exceptfds, timeout, 0);
902 /* If transparent TLS mode is turned on, then ldp will load key and cert.
905 load_tls_cert (vls_handle_t vlsh)
907 char *env_var_str = getenv (LDP_ENV_TLS_CERT);
915 fp = fopen (env_var_str, "r");
918 LDBG (0, "ERROR: failed to open cert file %s \n", env_var_str);
921 cert_size = fread (inbuf, sizeof (char), sizeof (inbuf), fp);
923 vppcom_session_tls_add_cert (vlsh_to_session_index (vlsh), tls_cert,
929 LDBG (0, "ERROR: failed to read LDP environment %s\n",
937 load_tls_key (vls_handle_t vlsh)
939 char *env_var_str = getenv (LDP_ENV_TLS_KEY);
947 fp = fopen (env_var_str, "r");
950 LDBG (0, "ERROR: failed to open key file %s \n", env_var_str);
953 key_size = fread (inbuf, sizeof (char), sizeof (inbuf), fp);
955 vppcom_session_tls_add_key (vlsh_to_session_index (vlsh), tls_key,
961 LDBG (0, "ERROR: failed to read LDP environment %s\n", LDP_ENV_TLS_KEY);
968 socket (int domain, int type, int protocol)
970 int rv, sock_type = type & ~(SOCK_CLOEXEC | SOCK_NONBLOCK);
971 u8 is_nonblocking = type & SOCK_NONBLOCK ? 1 : 0;
974 if ((errno = -ldp_init ()))
977 if (((domain == AF_INET) || (domain == AF_INET6)) &&
978 ((sock_type == SOCK_STREAM) || (sock_type == SOCK_DGRAM)))
981 if (ldp->transparent_tls)
983 proto = VPPCOM_PROTO_TLS;
986 proto = ((sock_type == SOCK_DGRAM) ?
987 VPPCOM_PROTO_UDP : VPPCOM_PROTO_TCP);
989 LDBG (0, "calling vls_create: proto %u (%s), is_nonblocking %u",
990 proto, vppcom_proto_str (proto), is_nonblocking);
992 vlsh = vls_create (proto, is_nonblocking);
1000 if (ldp->transparent_tls)
1002 if (load_tls_cert (vlsh) < 0 || load_tls_key (vlsh) < 0)
1007 rv = ldp_vlsh_to_fd (vlsh);
1012 LDBG (0, "calling libc_socket");
1013 rv = libc_socket (domain, type, protocol);
1020 * Create two new sockets, of type TYPE in domain DOMAIN and using
1021 * protocol PROTOCOL, which are connected to each other, and put file
1022 * descriptors for them in FDS[0] and FDS[1]. If PROTOCOL is zero,
1023 * one will be chosen automatically.
1024 * Returns 0 on success, -1 for errors.
1027 socketpair (int domain, int type, int protocol, int fds[2])
1029 int rv, sock_type = type & ~(SOCK_CLOEXEC | SOCK_NONBLOCK);
1031 if ((errno = -ldp_init ()))
1034 if (((domain == AF_INET) || (domain == AF_INET6)) &&
1035 ((sock_type == SOCK_STREAM) || (sock_type == SOCK_DGRAM)))
1037 LDBG (0, "LDP-TBD");
1043 LDBG (1, "calling libc_socketpair");
1044 rv = libc_socketpair (domain, type, protocol, fds);
1051 bind (int fd, __CONST_SOCKADDR_ARG addr, socklen_t len)
1056 if ((errno = -ldp_init ()))
1059 vlsh = ldp_fd_to_vlsh (fd);
1060 if (vlsh != VLS_INVALID_HANDLE)
1064 switch (addr->sa_family)
1067 if (len != sizeof (struct sockaddr_in))
1069 LDBG (0, "ERROR: fd %d: vlsh %u: Invalid AF_INET addr len %u!",
1075 ep.is_ip4 = VPPCOM_IS_IP4;
1076 ep.ip = (u8 *) & ((const struct sockaddr_in *) addr)->sin_addr;
1077 ep.port = (u16) ((const struct sockaddr_in *) addr)->sin_port;
1081 if (len != sizeof (struct sockaddr_in6))
1083 LDBG (0, "ERROR: fd %d: vlsh %u: Invalid AF_INET6 addr len %u!",
1089 ep.is_ip4 = VPPCOM_IS_IP6;
1090 ep.ip = (u8 *) & ((const struct sockaddr_in6 *) addr)->sin6_addr;
1091 ep.port = (u16) ((const struct sockaddr_in6 *) addr)->sin6_port;
1095 LDBG (0, "ERROR: fd %d: vlsh %u: Unsupported address family %u!",
1096 fd, vlsh, addr->sa_family);
1097 errno = EAFNOSUPPORT;
1101 LDBG (0, "fd %d: calling vls_bind: vlsh %u, addr %p, len %u", fd, vlsh,
1104 rv = vls_bind (vlsh, &ep);
1105 if (rv != VPPCOM_OK)
1113 LDBG (0, "fd %d: calling libc_bind: addr %p, len %u", fd, addr, len);
1114 rv = libc_bind (fd, addr, len);
1118 LDBG (1, "fd %d: returning %d", fd, rv);
1124 ldp_copy_ep_to_sockaddr (__SOCKADDR_ARG addr, socklen_t * __restrict len,
1125 vppcom_endpt_t * ep)
1128 int sa_len, copy_len;
1130 if ((errno = -ldp_init ()))
1133 if (addr && len && ep)
1135 addr->sa_family = (ep->is_ip4 == VPPCOM_IS_IP4) ? AF_INET : AF_INET6;
1136 switch (addr->sa_family)
1139 ((struct sockaddr_in *) addr)->sin_port = ep->port;
1140 if (*len > sizeof (struct sockaddr_in))
1141 *len = sizeof (struct sockaddr_in);
1142 sa_len = sizeof (struct sockaddr_in) - sizeof (struct in_addr);
1143 copy_len = *len - sa_len;
1145 memcpy (&((struct sockaddr_in *) addr)->sin_addr, ep->ip,
1150 ((struct sockaddr_in6 *) addr)->sin6_port = ep->port;
1151 if (*len > sizeof (struct sockaddr_in6))
1152 *len = sizeof (struct sockaddr_in6);
1153 sa_len = sizeof (struct sockaddr_in6) - sizeof (struct in6_addr);
1154 copy_len = *len - sa_len;
1156 memcpy (((struct sockaddr_in6 *) addr)->sin6_addr.
1157 __in6_u.__u6_addr8, ep->ip, copy_len);
1170 getsockname (int fd, __SOCKADDR_ARG addr, socklen_t * __restrict len)
1175 if ((errno = -ldp_init ()))
1178 vlsh = ldp_fd_to_vlsh (fd);
1179 if (vlsh != VLS_INVALID_HANDLE)
1182 u8 addr_buf[sizeof (struct in6_addr)];
1183 u32 size = sizeof (ep);
1187 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_LCL_ADDR, &ep, &size);
1188 if (rv != VPPCOM_OK)
1195 rv = ldp_copy_ep_to_sockaddr (addr, len, &ep);
1196 if (rv != VPPCOM_OK)
1205 rv = libc_getsockname (fd, addr, len);
1212 connect (int fd, __CONST_SOCKADDR_ARG addr, socklen_t len)
1217 if ((errno = -ldp_init ()))
1222 LDBG (0, "ERROR: fd %d: NULL addr, len %u", fd, len);
1228 vlsh = ldp_fd_to_vlsh (fd);
1229 if (vlsh != VLS_INVALID_HANDLE)
1233 switch (addr->sa_family)
1236 if (len != sizeof (struct sockaddr_in))
1238 LDBG (0, "fd %d: ERROR vlsh %u: Invalid AF_INET addr len %u!",
1244 ep.is_ip4 = VPPCOM_IS_IP4;
1245 ep.ip = (u8 *) & ((const struct sockaddr_in *) addr)->sin_addr;
1246 ep.port = (u16) ((const struct sockaddr_in *) addr)->sin_port;
1250 if (len != sizeof (struct sockaddr_in6))
1252 LDBG (0, "fd %d: ERROR vlsh %u: Invalid AF_INET6 addr len %u!",
1258 ep.is_ip4 = VPPCOM_IS_IP6;
1259 ep.ip = (u8 *) & ((const struct sockaddr_in6 *) addr)->sin6_addr;
1260 ep.port = (u16) ((const struct sockaddr_in6 *) addr)->sin6_port;
1264 LDBG (0, "fd %d: ERROR vlsh %u: Unsupported address family %u!",
1265 fd, vlsh, addr->sa_family);
1266 errno = EAFNOSUPPORT;
1270 LDBG (0, "fd %d: calling vls_connect(): vlsh %u addr %p len %u", fd,
1273 rv = vls_connect (vlsh, &ep);
1274 if (rv != VPPCOM_OK)
1282 LDBG (0, "fd %d: calling libc_connect(): addr %p, len %u",
1285 rv = libc_connect (fd, addr, len);
1289 LDBG (1, "fd %d: returning %d (0x%x)", fd, rv, rv);
1294 getpeername (int fd, __SOCKADDR_ARG addr, socklen_t * __restrict len)
1299 if ((errno = -ldp_init ()))
1302 vlsh = ldp_fd_to_vlsh (fd);
1303 if (vlsh != VLS_INVALID_HANDLE)
1306 u8 addr_buf[sizeof (struct in6_addr)];
1307 u32 size = sizeof (ep);
1310 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_PEER_ADDR, &ep, &size);
1311 if (rv != VPPCOM_OK)
1318 rv = ldp_copy_ep_to_sockaddr (addr, len, &ep);
1319 if (rv != VPPCOM_OK)
1328 rv = libc_getpeername (fd, addr, len);
1335 send (int fd, const void *buf, size_t n, int flags)
1337 vls_handle_t vlsh = ldp_fd_to_vlsh (fd);
1340 if ((errno = -ldp_init ()))
1343 if (vlsh != VLS_INVALID_HANDLE)
1345 size = vls_sendto (vlsh, (void *) buf, n, flags, NULL);
1346 if (size < VPPCOM_OK)
1354 size = libc_send (fd, buf, n, flags);
1361 sendfile (int out_fd, int in_fd, off_t * offset, size_t len)
1363 ldp_worker_ctx_t *ldpw = ldp_worker_get_current ();
1367 if ((errno = -ldp_init ()))
1370 vlsh = ldp_fd_to_vlsh (out_fd);
1371 if (vlsh != VLS_INVALID_HANDLE)
1374 ssize_t results = 0;
1375 size_t n_bytes_left = len;
1376 size_t bytes_to_read;
1379 u32 flags, flags_len = sizeof (flags);
1381 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_FLAGS, &flags, &flags_len);
1382 if (PREDICT_FALSE (rv != VPPCOM_OK))
1384 LDBG (0, "ERROR: out fd %d: vls_attr: vlsh %u, returned %d (%s)!",
1385 out_fd, vlsh, rv, vppcom_retval_str (rv));
1387 vec_reset_length (ldpw->io_buffer);
1395 off_t off = lseek (in_fd, *offset, SEEK_SET);
1396 if (PREDICT_FALSE (off == -1))
1402 ASSERT (off == *offset);
1407 size = vls_attr (vlsh, VPPCOM_ATTR_GET_NWRITE, 0, 0);
1410 LDBG (0, "ERROR: fd %d: vls_attr: vlsh %u returned %ld (%s)!",
1411 out_fd, vlsh, size, vppcom_retval_str (size));
1412 vec_reset_length (ldpw->io_buffer);
1418 bytes_to_read = size;
1419 if (bytes_to_read == 0)
1421 if (flags & O_NONBLOCK)
1430 bytes_to_read = clib_min (n_bytes_left, bytes_to_read);
1431 vec_validate (ldpw->io_buffer, bytes_to_read);
1432 nbytes = libc_read (in_fd, ldpw->io_buffer, bytes_to_read);
1437 vec_reset_length (ldpw->io_buffer);
1444 size = vls_write (vlsh, ldpw->io_buffer, nbytes);
1447 if (size == VPPCOM_EAGAIN)
1449 if (flags & O_NONBLOCK)
1460 vec_reset_length (ldpw->io_buffer);
1469 ASSERT (n_bytes_left >= nbytes);
1470 n_bytes_left = n_bytes_left - nbytes;
1472 while (n_bytes_left > 0);
1475 vec_reset_length (ldpw->io_buffer);
1478 off_t off = lseek (in_fd, *offset, SEEK_SET);
1479 if (PREDICT_FALSE (off == -1))
1485 ASSERT (off == *offset);
1486 *offset += results + 1;
1498 size = libc_sendfile (out_fd, in_fd, offset, len);
1506 sendfile64 (int out_fd, int in_fd, off_t * offset, size_t len)
1508 return sendfile (out_fd, in_fd, offset, len);
1512 recv (int fd, void *buf, size_t n, int flags)
1517 if ((errno = -ldp_init ()))
1520 vlsh = ldp_fd_to_vlsh (fd);
1521 if (vlsh != VLS_INVALID_HANDLE)
1523 size = vls_recvfrom (vlsh, buf, n, flags, NULL);
1532 size = libc_recv (fd, buf, n, flags);
1539 ldp_vls_sendo (vls_handle_t vlsh, const void *buf, size_t n, int flags,
1540 __CONST_SOCKADDR_ARG addr, socklen_t addr_len)
1542 vppcom_endpt_t *ep = 0;
1548 switch (addr->sa_family)
1551 ep->is_ip4 = VPPCOM_IS_IP4;
1553 (uint8_t *) & ((const struct sockaddr_in *) addr)->sin_addr;
1554 ep->port = (uint16_t) ((const struct sockaddr_in *) addr)->sin_port;
1558 ep->is_ip4 = VPPCOM_IS_IP6;
1560 (uint8_t *) & ((const struct sockaddr_in6 *) addr)->sin6_addr;
1562 (uint16_t) ((const struct sockaddr_in6 *) addr)->sin6_port;
1566 return EAFNOSUPPORT;
1570 return vls_sendto (vlsh, (void *) buf, n, flags, ep);
1574 ldp_vls_recvfrom (vls_handle_t vlsh, void *__restrict buf, size_t n,
1575 int flags, __SOCKADDR_ARG addr,
1576 socklen_t * __restrict addr_len)
1578 u8 src_addr[sizeof (struct sockaddr_in6)];
1586 size = vls_recvfrom (vlsh, buf, n, flags, &ep);
1590 rv = ldp_copy_ep_to_sockaddr (addr, addr_len, &ep);
1596 size = vls_recvfrom (vlsh, buf, n, flags, NULL);
1602 sendto (int fd, const void *buf, size_t n, int flags,
1603 __CONST_SOCKADDR_ARG addr, socklen_t addr_len)
1608 if ((errno = -ldp_init ()))
1611 vlsh = ldp_fd_to_vlsh (fd);
1612 if (vlsh != INVALID_SESSION_ID)
1614 size = ldp_vls_sendo (vlsh, buf, n, flags, addr, addr_len);
1623 size = libc_sendto (fd, buf, n, flags, addr, addr_len);
1630 recvfrom (int fd, void *__restrict buf, size_t n, int flags,
1631 __SOCKADDR_ARG addr, socklen_t * __restrict addr_len)
1636 if ((errno = -ldp_init ()))
1639 vlsh = ldp_fd_to_vlsh (fd);
1640 if (vlsh != VLS_INVALID_HANDLE)
1642 size = ldp_vls_recvfrom (vlsh, buf, n, flags, addr, addr_len);
1651 size = libc_recvfrom (fd, buf, n, flags, addr, addr_len);
1658 sendmsg (int fd, const struct msghdr * msg, int flags)
1663 if ((errno = -ldp_init ()))
1666 vlsh = ldp_fd_to_vlsh (fd);
1667 if (vlsh != VLS_INVALID_HANDLE)
1669 struct iovec *iov = msg->msg_iov;
1673 for (i = 0; i < msg->msg_iovlen; ++i)
1675 rv = ldp_vls_sendo (vlsh, iov[i].iov_base, iov[i].iov_len, flags,
1676 msg->msg_name, msg->msg_namelen);
1682 if (rv < iov[i].iov_len)
1687 if (rv < 0 && total == 0)
1697 size = libc_sendmsg (fd, msg, flags);
1705 sendmmsg (int fd, struct mmsghdr *vmessages, unsigned int vlen, int flags)
1708 const char *func_str;
1709 u32 sh = ldp_fd_to_vlsh (fd);
1711 if ((errno = -ldp_init ()))
1714 if (sh != INVALID_SESSION_ID)
1716 clib_warning ("LDP<%d>: LDP-TBD", getpid ());
1722 func_str = "libc_sendmmsg";
1725 clib_warning ("LDP<%d>: fd %d (0x%x): calling %s(): "
1726 "vmessages %p, vlen %u, flags 0x%x",
1727 getpid (), fd, fd, func_str, vmessages, vlen, flags);
1729 size = libc_sendmmsg (fd, vmessages, vlen, flags);
1736 int errno_val = errno;
1738 clib_warning ("LDP<%d>: ERROR: fd %d (0x%x): %s() failed! "
1739 "rv %d, errno = %d", getpid (), fd, fd,
1740 func_str, size, errno_val);
1744 clib_warning ("LDP<%d>: fd %d (0x%x): returning %d (0x%x)",
1745 getpid (), fd, fd, size, size);
1752 recvmsg (int fd, struct msghdr * msg, int flags)
1757 if ((errno = -ldp_init ()))
1760 vlsh = ldp_fd_to_vlsh (fd);
1761 if (vlsh != VLS_INVALID_HANDLE)
1763 struct iovec *iov = msg->msg_iov;
1764 ssize_t max_deq, total = 0;
1767 max_deq = vls_attr (vlsh, VPPCOM_ATTR_GET_NREAD, 0, 0);
1771 for (i = 0; i < msg->msg_iovlen; i++)
1773 rv = ldp_vls_recvfrom (vlsh, iov[i].iov_base, iov[i].iov_len, flags,
1774 (i == 0 ? msg->msg_name : NULL),
1775 (i == 0 ? &msg->msg_namelen : NULL));
1781 if (rv < iov[i].iov_len)
1784 if (total >= max_deq)
1788 if (rv < 0 && total == 0)
1798 size = libc_recvmsg (fd, msg, flags);
1806 recvmmsg (int fd, struct mmsghdr *vmessages,
1807 unsigned int vlen, int flags, struct timespec *tmo)
1810 const char *func_str;
1811 u32 sh = ldp_fd_to_vlsh (fd);
1813 if ((errno = -ldp_init ()))
1816 if (sh != INVALID_SESSION_ID)
1818 clib_warning ("LDP<%d>: LDP-TBD", getpid ());
1824 func_str = "libc_recvmmsg";
1827 clib_warning ("LDP<%d>: fd %d (0x%x): calling %s(): "
1828 "vmessages %p, vlen %u, flags 0x%x, tmo %p",
1829 getpid (), fd, fd, func_str, vmessages, vlen,
1832 size = libc_recvmmsg (fd, vmessages, vlen, flags, tmo);
1839 int errno_val = errno;
1841 clib_warning ("LDP<%d>: ERROR: fd %d (0x%x): %s() failed! "
1842 "rv %d, errno = %d", getpid (), fd, fd,
1843 func_str, size, errno_val);
1847 clib_warning ("LDP<%d>: fd %d (0x%x): returning %d (0x%x)",
1848 getpid (), fd, fd, size, size);
1855 getsockopt (int fd, int level, int optname,
1856 void *__restrict optval, socklen_t * __restrict optlen)
1861 if ((errno = -ldp_init ()))
1864 vlsh = ldp_fd_to_vlsh (fd);
1865 if (vlsh != VLS_INVALID_HANDLE)
1875 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_TCP_NODELAY,
1879 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_TCP_USER_MSS,
1883 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_TCP_KEEPIDLE,
1887 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_TCP_KEEPINTVL,
1891 if (optval && optlen && (*optlen == sizeof (struct tcp_info)))
1893 LDBG (1, "fd %d: vlsh %u SOL_TCP, TCP_INFO, optval %p, "
1894 "optlen %d: #LDP-NOP#", fd, vlsh, optval, *optlen);
1895 memset (optval, 0, *optlen);
1901 case TCP_CONGESTION:
1902 *optlen = strlen ("cubic");
1903 strncpy (optval, "cubic", *optlen + 1);
1907 LDBG (0, "ERROR: fd %d: getsockopt SOL_TCP: sid %u, "
1908 "optname %d unsupported!", fd, vlsh, optname);
1916 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_V6ONLY, optval, optlen);
1919 LDBG (0, "ERROR: fd %d: getsockopt SOL_IPV6: vlsh %u "
1920 "optname %d unsupported!", fd, vlsh, optname);
1928 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_LISTEN, optval, optlen);
1931 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_KEEPALIVE, optval, optlen);
1934 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_PROTOCOL, optval, optlen);
1935 *(int *) optval = *(int *) optval ? SOCK_DGRAM : SOCK_STREAM;
1938 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_TX_FIFO_LEN,
1942 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_RX_FIFO_LEN,
1946 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_REUSEADDR, optval, optlen);
1949 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_BROADCAST, optval, optlen);
1952 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_ERROR, optval, optlen);
1955 LDBG (0, "ERROR: fd %d: getsockopt SOL_SOCKET: vlsh %u "
1956 "optname %d unsupported!", fd, vlsh, optname);
1964 if (rv != VPPCOM_OK)
1972 rv = libc_getsockopt (fd, level, optname, optval, optlen);
1979 setsockopt (int fd, int level, int optname,
1980 const void *optval, socklen_t optlen)
1985 if ((errno = -ldp_init ()))
1988 vlsh = ldp_fd_to_vlsh (fd);
1989 if (vlsh != VLS_INVALID_HANDLE)
1999 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_TCP_NODELAY,
2000 (void *) optval, &optlen);
2003 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_TCP_USER_MSS,
2004 (void *) optval, &optlen);
2007 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_TCP_KEEPIDLE,
2008 (void *) optval, &optlen);
2011 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_TCP_KEEPINTVL,
2012 (void *) optval, &optlen);
2014 case TCP_CONGESTION:
2020 LDBG (0, "ERROR: fd %d: setsockopt() SOL_TCP: vlsh %u"
2021 "optname %d unsupported!", fd, vlsh, optname);
2029 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_V6ONLY,
2030 (void *) optval, &optlen);
2033 LDBG (0, "ERROR: fd %d: setsockopt SOL_IPV6: vlsh %u"
2034 "optname %d unsupported!", fd, vlsh, optname);
2042 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_KEEPALIVE,
2043 (void *) optval, &optlen);
2046 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_REUSEADDR,
2047 (void *) optval, &optlen);
2050 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_BROADCAST,
2051 (void *) optval, &optlen);
2054 LDBG (0, "ERROR: fd %d: setsockopt SOL_SOCKET: vlsh %u "
2055 "optname %d unsupported!", fd, vlsh, optname);
2063 if (rv != VPPCOM_OK)
2071 rv = libc_setsockopt (fd, level, optname, optval, optlen);
2078 listen (int fd, int n)
2083 if ((errno = -ldp_init ()))
2086 vlsh = ldp_fd_to_vlsh (fd);
2087 if (vlsh != VLS_INVALID_HANDLE)
2089 LDBG (0, "fd %d: calling vls_listen: vlsh %u, n %d", fd, vlsh, n);
2091 rv = vls_listen (vlsh, n);
2092 if (rv != VPPCOM_OK)
2100 LDBG (0, "fd %d: calling libc_listen(): n %d", fd, n);
2101 rv = libc_listen (fd, n);
2104 LDBG (1, "fd %d: returning %d", fd, rv);
2109 ldp_accept4 (int listen_fd, __SOCKADDR_ARG addr,
2110 socklen_t * __restrict addr_len, int flags)
2112 vls_handle_t listen_vlsh, accept_vlsh;
2115 if ((errno = -ldp_init ()))
2118 listen_vlsh = ldp_fd_to_vlsh (listen_fd);
2119 if (listen_vlsh != VLS_INVALID_HANDLE)
2122 u8 src_addr[sizeof (struct sockaddr_in6)];
2123 memset (&ep, 0, sizeof (ep));
2126 LDBG (0, "listen fd %d: calling vppcom_session_accept: listen sid %u,"
2127 " ep %p, flags 0x%x", listen_fd, listen_vlsh, &ep, flags);
2129 accept_vlsh = vls_accept (listen_vlsh, &ep, flags);
2130 if (accept_vlsh < 0)
2132 errno = -accept_vlsh;
2137 rv = ldp_copy_ep_to_sockaddr (addr, addr_len, &ep);
2138 if (rv != VPPCOM_OK)
2140 (void) vls_close (accept_vlsh);
2146 rv = ldp_vlsh_to_fd (accept_vlsh);
2152 LDBG (0, "listen fd %d: calling libc_accept4(): addr %p, addr_len %p,"
2153 " flags 0x%x", listen_fd, addr, addr_len, flags);
2155 rv = libc_accept4 (listen_fd, addr, addr_len, flags);
2158 LDBG (1, "listen fd %d: accept returning %d", listen_fd, rv);
2164 accept4 (int fd, __SOCKADDR_ARG addr, socklen_t * __restrict addr_len,
2167 return ldp_accept4 (fd, addr, addr_len, flags);
2171 accept (int fd, __SOCKADDR_ARG addr, socklen_t * __restrict addr_len)
2173 return ldp_accept4 (fd, addr, addr_len, 0);
2177 shutdown (int fd, int how)
2181 u32 flags_len = sizeof (flags);
2183 if ((errno = -ldp_init ()))
2186 vlsh = ldp_fd_to_vlsh (fd);
2187 if (vlsh != VLS_INVALID_HANDLE)
2189 LDBG (0, "called shutdown: fd %u vlsh %u how %d", fd, vlsh, how);
2191 if (vls_attr (vlsh, VPPCOM_ATTR_SET_SHUT, &how, &flags_len))
2197 if (vls_attr (vlsh, VPPCOM_ATTR_GET_SHUT, &flags, &flags_len))
2203 if (flags == SHUT_RDWR)
2208 LDBG (0, "fd %d: calling libc_shutdown: how %d", fd, how);
2209 rv = libc_shutdown (fd, how);
2216 epoll_create1 (int flags)
2218 ldp_worker_ctx_t *ldpw = ldp_worker_get_current ();
2222 if ((errno = -ldp_init ()))
2225 if (ldp->vcl_needs_real_epoll || vls_use_real_epoll ())
2227 /* Make sure workers have been allocated */
2230 ldp_alloc_workers ();
2231 ldpw = ldp_worker_get_current ();
2233 rv = libc_epoll_create1 (flags);
2234 ldp->vcl_needs_real_epoll = 0;
2235 ldpw->vcl_mq_epfd = rv;
2236 LDBG (0, "created vcl epfd %u", rv);
2240 vlsh = vls_epoll_create ();
2241 if (PREDICT_FALSE (vlsh == VLS_INVALID_HANDLE))
2248 rv = ldp_vlsh_to_fd (vlsh);
2250 LDBG (0, "epoll_create epfd %u vlsh %u", rv, vlsh);
2255 epoll_create (int size)
2257 return epoll_create1 (0);
2261 epoll_ctl (int epfd, int op, int fd, struct epoll_event *event)
2263 vls_handle_t vep_vlsh, vlsh;
2266 if ((errno = -ldp_init ()))
2269 vep_vlsh = ldp_fd_to_vlsh (epfd);
2270 if (PREDICT_FALSE (vep_vlsh == VLS_INVALID_HANDLE))
2272 /* The LDP epoll_create1 always creates VCL epfd's.
2273 * The app should never have a kernel base epoll fd unless it
2274 * was acquired outside of the LD_PRELOAD process context.
2275 * In any case, if we get one, punt it to libc_epoll_ctl.
2277 LDBG (1, "epfd %d: calling libc_epoll_ctl: op %d, fd %d"
2278 " event %p", epfd, op, fd, event);
2280 rv = libc_epoll_ctl (epfd, op, fd, event);
2284 vlsh = ldp_fd_to_vlsh (fd);
2286 LDBG (0, "epfd %d ep_vlsh %d, fd %u vlsh %d, op %u", epfd, vep_vlsh, fd,
2289 if (vlsh != VLS_INVALID_HANDLE)
2291 LDBG (1, "epfd %d: calling vls_epoll_ctl: ep_vlsh %d op %d, vlsh %u,"
2292 " event %p", epfd, vep_vlsh, op, vlsh, event);
2294 rv = vls_epoll_ctl (vep_vlsh, op, vlsh, event);
2295 if (rv != VPPCOM_OK)
2304 u32 size = sizeof (epfd);
2306 libc_epfd = vls_attr (vep_vlsh, VPPCOM_ATTR_GET_LIBC_EPFD, 0, 0);
2309 LDBG (1, "epfd %d, vep_vlsh %d calling libc_epoll_create1: "
2310 "EPOLL_CLOEXEC", epfd, vep_vlsh);
2312 libc_epfd = libc_epoll_create1 (EPOLL_CLOEXEC);
2319 rv = vls_attr (vep_vlsh, VPPCOM_ATTR_SET_LIBC_EPFD, &libc_epfd,
2328 else if (PREDICT_FALSE (libc_epfd < 0))
2335 LDBG (1, "epfd %d: calling libc_epoll_ctl: libc_epfd %d, op %d, fd %d,"
2336 " event %p", epfd, libc_epfd, op, fd, event);
2338 rv = libc_epoll_ctl (libc_epfd, op, fd, event);
2346 ldp_epoll_pwait (int epfd, struct epoll_event *events, int maxevents,
2347 int timeout, const sigset_t * sigmask)
2349 ldp_worker_ctx_t *ldpw = ldp_worker_get_current ();
2350 double time_to_wait = (double) 0, max_time;
2351 int libc_epfd, rv = 0;
2352 vls_handle_t ep_vlsh;
2354 if ((errno = -ldp_init ()))
2357 if (PREDICT_FALSE (!events || (timeout < -1)))
2363 if (epfd == ldpw->vcl_mq_epfd)
2364 return libc_epoll_pwait (epfd, events, maxevents, timeout, sigmask);
2366 ep_vlsh = ldp_fd_to_vlsh (epfd);
2367 if (PREDICT_FALSE (ep_vlsh == VLS_INVALID_HANDLE))
2369 LDBG (0, "epfd %d: bad ep_vlsh %d!", epfd, ep_vlsh);
2374 if (PREDICT_FALSE (ldpw->clib_time.init_cpu_time == 0))
2375 clib_time_init (&ldpw->clib_time);
2376 time_to_wait = ((timeout >= 0) ? (double) timeout / 1000 : 0);
2377 max_time = clib_time_now (&ldpw->clib_time) + time_to_wait;
2379 libc_epfd = vls_attr (ep_vlsh, VPPCOM_ATTR_GET_LIBC_EPFD, 0, 0);
2380 if (PREDICT_FALSE (libc_epfd < 0))
2387 LDBG (2, "epfd %d: vep_idx %d, libc_epfd %d, events %p, maxevents %d, "
2388 "timeout %d, sigmask %p: time_to_wait %.02f", epfd, ep_vlsh,
2389 libc_epfd, events, maxevents, timeout, sigmask, time_to_wait);
2392 if (!ldpw->epoll_wait_vcl)
2394 rv = vls_epoll_wait (ep_vlsh, events, maxevents, 0);
2397 ldpw->epoll_wait_vcl = 1;
2408 ldpw->epoll_wait_vcl = 0;
2412 rv = libc_epoll_pwait (libc_epfd, events, maxevents, 0, sigmask);
2417 while ((timeout == -1) || (clib_time_now (&ldpw->clib_time) < max_time));
2424 ldp_epoll_pwait_eventfd (int epfd, struct epoll_event *events,
2425 int maxevents, int timeout, const sigset_t * sigmask)
2427 ldp_worker_ctx_t *ldpw;
2428 int libc_epfd, rv = 0, num_ev;
2429 vls_handle_t ep_vlsh;
2431 if ((errno = -ldp_init ()))
2434 if (PREDICT_FALSE (!events || (timeout < -1)))
2440 /* Make sure the vcl worker is valid. Could be that epoll fd was created on
2441 * one thread but it is now used on another */
2442 if (PREDICT_FALSE (vppcom_worker_index () == ~0))
2443 vls_register_vcl_worker ();
2445 ldpw = ldp_worker_get_current ();
2446 if (epfd == ldpw->vcl_mq_epfd)
2447 return libc_epoll_pwait (epfd, events, maxevents, timeout, sigmask);
2449 ep_vlsh = ldp_fd_to_vlsh (epfd);
2450 if (PREDICT_FALSE (ep_vlsh == VLS_INVALID_HANDLE))
2452 LDBG (0, "epfd %d: bad ep_vlsh %d!", epfd, ep_vlsh);
2457 libc_epfd = vls_attr (ep_vlsh, VPPCOM_ATTR_GET_LIBC_EPFD, 0, 0);
2458 if (PREDICT_FALSE (!libc_epfd))
2460 u32 size = sizeof (epfd);
2462 LDBG (1, "epfd %d, vep_vlsh %d calling libc_epoll_create1: "
2463 "EPOLL_CLOEXEC", epfd, ep_vlsh);
2464 libc_epfd = libc_epoll_create1 (EPOLL_CLOEXEC);
2471 rv = vls_attr (ep_vlsh, VPPCOM_ATTR_SET_LIBC_EPFD, &libc_epfd, &size);
2479 if (PREDICT_FALSE (libc_epfd <= 0))
2486 if (PREDICT_FALSE (!ldpw->mq_epfd_added))
2488 struct epoll_event e = { 0 };
2490 e.data.fd = ldpw->vcl_mq_epfd;
2491 if (libc_epoll_ctl (libc_epfd, EPOLL_CTL_ADD, ldpw->vcl_mq_epfd, &e) <
2494 LDBG (0, "epfd %d, add libc mq epoll fd %d to libc epoll fd %d",
2495 epfd, ldpw->vcl_mq_epfd, libc_epfd);
2499 ldpw->mq_epfd_added = 1;
2502 rv = vls_epoll_wait (ep_vlsh, events, maxevents, 0);
2505 else if (PREDICT_FALSE (rv < 0))
2512 rv = libc_epoll_pwait (libc_epfd, events, maxevents, timeout, sigmask);
2515 for (int i = 0; i < rv; i++)
2517 if (events[i].data.fd == ldpw->vcl_mq_epfd)
2519 /* We should remove mq epoll fd from events. */
2523 events[i].events = events[rv].events;
2524 events[i].data.u64 = events[rv].data.u64;
2526 num_ev = vls_epoll_wait (ep_vlsh, &events[rv], maxevents - rv, 0);
2527 if (PREDICT_TRUE (num_ev > 0))
2538 epoll_pwait (int epfd, struct epoll_event *events,
2539 int maxevents, int timeout, const sigset_t * sigmask)
2541 if (vls_use_eventfd ())
2542 return ldp_epoll_pwait_eventfd (epfd, events, maxevents, timeout,
2545 return ldp_epoll_pwait (epfd, events, maxevents, timeout, sigmask);
2549 epoll_wait (int epfd, struct epoll_event *events, int maxevents, int timeout)
2551 if (vls_use_eventfd ())
2552 return ldp_epoll_pwait_eventfd (epfd, events, maxevents, timeout, NULL);
2554 return ldp_epoll_pwait (epfd, events, maxevents, timeout, NULL);
2558 poll (struct pollfd *fds, nfds_t nfds, int timeout)
2560 ldp_worker_ctx_t *ldpw = ldp_worker_get_current ();
2561 int rv, i, n_revents = 0;
2566 LDBG (3, "fds %p, nfds %ld, timeout %d", fds, nfds, timeout);
2568 if (PREDICT_FALSE (ldpw->clib_time.init_cpu_time == 0))
2569 clib_time_init (&ldpw->clib_time);
2571 max_time = (timeout >= 0) ? (f64) timeout / 1000 : 0;
2572 max_time += clib_time_now (&ldpw->clib_time);
2574 for (i = 0; i < nfds; i++)
2579 vlsh = ldp_fd_to_vlsh (fds[i].fd);
2580 if (vlsh != VLS_INVALID_HANDLE)
2582 fds[i].fd = -fds[i].fd;
2583 vec_add2 (ldpw->vcl_poll, vp, 1);
2585 vp->sh = vlsh_to_sh (vlsh);
2586 vp->events = fds[i].events;
2587 #ifdef __USE_XOPEN2K
2588 if (fds[i].events & POLLRDNORM)
2589 vp->events |= POLLIN;
2590 if (fds[i].events & POLLWRNORM)
2591 vp->events |= POLLOUT;
2593 vp->revents = fds[i].revents;
2597 vec_add1 (ldpw->libc_poll, fds[i]);
2598 vec_add1 (ldpw->libc_poll_idxs, i);
2604 if (vec_len (ldpw->vcl_poll))
2606 rv = vppcom_poll (ldpw->vcl_poll, vec_len (ldpw->vcl_poll), 0);
2617 if (vec_len (ldpw->libc_poll))
2619 rv = libc_poll (ldpw->libc_poll, vec_len (ldpw->libc_poll), 0);
2632 while ((timeout < 0) || (clib_time_now (&ldpw->clib_time) < max_time));
2636 vec_foreach (vp, ldpw->vcl_poll)
2638 fds[vp->fds_ndx].fd = -fds[vp->fds_ndx].fd;
2639 fds[vp->fds_ndx].revents = vp->revents;
2640 #ifdef __USE_XOPEN2K
2641 if ((fds[vp->fds_ndx].revents & POLLIN) &&
2642 (fds[vp->fds_ndx].events & POLLRDNORM))
2643 fds[vp->fds_ndx].revents |= POLLRDNORM;
2644 if ((fds[vp->fds_ndx].revents & POLLOUT) &&
2645 (fds[vp->fds_ndx].events & POLLWRNORM))
2646 fds[vp->fds_ndx].revents |= POLLWRNORM;
2649 vec_reset_length (ldpw->vcl_poll);
2651 for (i = 0; i < vec_len (ldpw->libc_poll); i++)
2653 fds[ldpw->libc_poll_idxs[i]].revents = ldpw->libc_poll[i].revents;
2655 vec_reset_length (ldpw->libc_poll_idxs);
2656 vec_reset_length (ldpw->libc_poll);
2663 ppoll (struct pollfd *fds, nfds_t nfds,
2664 const struct timespec *timeout, const sigset_t * sigmask)
2666 if ((errno = -ldp_init ()))
2669 clib_warning ("LDP<%d>: LDP-TBD", getpid ());
2677 void CONSTRUCTOR_ATTRIBUTE ldp_constructor (void);
2679 void DESTRUCTOR_ATTRIBUTE ldp_destructor (void);
2682 * This function is called when the library is loaded
2685 ldp_constructor (void)
2687 swrap_constructor ();
2688 if (ldp_init () != 0)
2690 fprintf (stderr, "\nLDP<%d>: ERROR: ldp_constructor: failed!\n",
2694 else if (LDP_DEBUG > 0)
2695 clib_warning ("LDP<%d>: LDP constructor: done!\n", getpid ());
2699 * This function is called when the library is unloaded
2702 ldp_destructor (void)
2705 swrap_destructor ();
2710 /* Don't use clib_warning() here because that calls writev()
2711 * which will call ldp_init().
2714 fprintf (stderr, "%s:%d: LDP<%d>: LDP destructor: done!\n",
2715 __func__, __LINE__, getpid ());
2720 * fd.io coding-style-patch-verification: ON
2723 * eval: (c-set-style "gnu")