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 <netinet/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;
67 #define UDP_SEGMENT 103
70 #ifndef SO_ORIGINAL_DST
71 /* from <linux/netfilter_ipv4.h> */
72 #define SO_ORIGINAL_DST 80
74 typedef struct ldp_worker_ctx_
77 clib_time_t clib_time;
82 clib_bitmap_t *rd_bitmap;
83 clib_bitmap_t *wr_bitmap;
84 clib_bitmap_t *ex_bitmap;
85 clib_bitmap_t *si_rd_bitmap;
86 clib_bitmap_t *si_wr_bitmap;
87 clib_bitmap_t *si_ex_bitmap;
88 clib_bitmap_t *libc_rd_bitmap;
89 clib_bitmap_t *libc_wr_bitmap;
90 clib_bitmap_t *libc_ex_bitmap;
96 struct pollfd *libc_poll;
108 /* clib_bitmap_t, fd_mask and vcl_si_set are used interchangeably. Make sure
109 * they are the same size */
110 STATIC_ASSERT (sizeof (clib_bitmap_t) == sizeof (fd_mask),
111 "ldp bitmap size mismatch");
112 STATIC_ASSERT (sizeof (vcl_si_set) == sizeof (fd_mask),
113 "ldp bitmap size mismatch");
117 ldp_worker_ctx_t *workers;
119 char app_name[LDP_APP_NAME_MAX];
124 /** vcl needs next epoll_create to go to libc_epoll */
125 u8 vcl_needs_real_epoll;
128 * crypto state used only for testing
134 #define LDP_DEBUG ldp->debug
136 #define LDBG(_lvl, _fmt, _args...) \
137 if (ldp->debug > _lvl) \
139 int errno_saved = errno; \
140 fprintf (stderr, "ldp<%d>: " _fmt "\n", getpid(), ##_args); \
141 errno = errno_saved; \
144 static ldp_main_t ldp_main = {
145 .vlsh_bit_val = (1 << LDP_SID_BIT_MIN),
146 .vlsh_bit_mask = (1 << LDP_SID_BIT_MIN) - 1,
147 .debug = LDP_DEBUG_INIT,
148 .transparent_tls = 0,
152 static ldp_main_t *ldp = &ldp_main;
154 static inline ldp_worker_ctx_t *
155 ldp_worker_get_current (void)
157 return (ldp->workers + vppcom_worker_index ());
161 * RETURN: 0 on success or -1 on error.
164 ldp_set_app_name (char *app_name)
166 snprintf (ldp->app_name, LDP_APP_NAME_MAX, "%s-ldp-%d", app_name, getpid ());
172 if (ldp->app_name[0] == '\0')
173 ldp_set_app_name (program_invocation_short_name);
175 return ldp->app_name;
179 ldp_vlsh_to_fd (vls_handle_t vlsh)
181 return (vlsh + ldp->vlsh_bit_val);
184 static inline vls_handle_t
185 ldp_fd_to_vlsh (int fd)
187 if (fd < ldp->vlsh_bit_val)
188 return VLS_INVALID_HANDLE;
190 return (fd - ldp->vlsh_bit_val);
194 ldp_alloc_workers (void)
198 pool_alloc (ldp->workers, LDP_MAX_NWORKERS);
204 char *env_var_str = getenv (LDP_ENV_DEBUG);
208 if (sscanf (env_var_str, "%u", &tmp) != 1)
209 clib_warning ("LDP<%d>: WARNING: Invalid LDP debug level specified in"
210 " the env var " LDP_ENV_DEBUG " (%s)!", getpid (),
215 LDBG (0, "configured LDP debug level (%u) from env var "
216 LDP_ENV_DEBUG "!", ldp->debug);
220 env_var_str = getenv (LDP_ENV_APP_NAME);
223 ldp_set_app_name (env_var_str);
224 LDBG (0, "configured LDP app name (%s) from the env var "
225 LDP_ENV_APP_NAME "!", ldp->app_name);
228 env_var_str = getenv (LDP_ENV_SID_BIT);
232 if (sscanf (env_var_str, "%u", &sb) != 1)
234 LDBG (0, "WARNING: Invalid LDP sid bit specified in the env var "
235 LDP_ENV_SID_BIT " (%s)! sid bit value %d (0x%x)", env_var_str,
236 ldp->vlsh_bit_val, ldp->vlsh_bit_val);
238 else if (sb < LDP_SID_BIT_MIN)
240 ldp->vlsh_bit_val = (1 << LDP_SID_BIT_MIN);
241 ldp->vlsh_bit_mask = ldp->vlsh_bit_val - 1;
243 LDBG (0, "WARNING: LDP sid bit (%u) specified in the env var "
244 LDP_ENV_SID_BIT " (%s) is too small. Using LDP_SID_BIT_MIN"
245 " (%d)! sid bit value %d (0x%x)", sb, env_var_str,
246 LDP_SID_BIT_MIN, ldp->vlsh_bit_val, ldp->vlsh_bit_val);
248 else if (sb > LDP_SID_BIT_MAX)
250 ldp->vlsh_bit_val = (1 << LDP_SID_BIT_MAX);
251 ldp->vlsh_bit_mask = ldp->vlsh_bit_val - 1;
253 LDBG (0, "WARNING: LDP sid bit (%u) specified in the env var "
254 LDP_ENV_SID_BIT " (%s) is too big. Using LDP_SID_BIT_MAX"
255 " (%d)! sid bit value %d (0x%x)", sb, env_var_str,
256 LDP_SID_BIT_MAX, ldp->vlsh_bit_val, ldp->vlsh_bit_val);
260 ldp->vlsh_bit_val = (1 << sb);
261 ldp->vlsh_bit_mask = ldp->vlsh_bit_val - 1;
263 LDBG (0, "configured LDP sid bit (%u) from "
264 LDP_ENV_SID_BIT "! sid bit value %d (0x%x)", sb,
265 ldp->vlsh_bit_val, ldp->vlsh_bit_val);
268 /* Make sure there are enough bits in the fd set for vcl sessions */
269 if (ldp->vlsh_bit_val > FD_SETSIZE / 2)
271 /* Only valid for select/pselect, so just WARNING and not exit */
273 "WARNING: LDP vlsh bit value %d > FD_SETSIZE/2 %d, "
274 "select/pselect not supported now!",
275 ldp->vlsh_bit_val, FD_SETSIZE / 2);
278 env_var_str = getenv (LDP_ENV_TLS_TRANS);
281 ldp->transparent_tls = 1;
288 ldp_worker_ctx_t *ldpw;
295 ldp->vcl_needs_real_epoll = 1;
296 rv = vls_app_create (ldp_get_app_name ());
299 ldp->vcl_needs_real_epoll = 0;
300 if (rv == VPPCOM_EEXIST)
303 "\nERROR: ldp_init: vppcom_app_create()"
304 " failed! rv = %d (%s)\n",
305 rv, vppcom_retval_str (rv));
309 ldp->vcl_needs_real_epoll = 0;
310 ldp_alloc_workers ();
311 ldpw = ldp_worker_get_current ();
313 pool_foreach (ldpw, ldp->workers) {
314 clib_memset (&ldpw->clib_time, 0, sizeof (ldpw->clib_time));
317 LDBG (0, "LDP initialization: done!");
322 #define ldp_init_check() \
323 if (PREDICT_FALSE (!ldp->init)) \
325 if ((errno = -ldp_init ())) \
337 vlsh = ldp_fd_to_vlsh (fd);
338 if (vlsh != VLS_INVALID_HANDLE)
340 epfd = vls_attr (vlsh, VPPCOM_ATTR_GET_LIBC_EPFD, 0, 0);
343 ldp_worker_ctx_t *ldpw = ldp_worker_get_current ();
344 u32 size = sizeof (epfd);
346 LDBG (0, "fd %d: calling libc_close: epfd %u", fd, epfd);
349 ldpw->mq_epfd_added = 0;
352 (void) vls_attr (vlsh, VPPCOM_ATTR_SET_LIBC_EPFD, &epfd, &size);
354 else if (PREDICT_FALSE (epfd < 0))
361 LDBG (0, "fd %d: calling vls_close: vlsh %u", fd, vlsh);
363 rv = vls_close (vlsh);
372 LDBG (0, "fd %d: calling libc_close", fd);
373 rv = libc_close (fd);
381 read (int fd, void *buf, size_t nbytes)
388 vlsh = ldp_fd_to_vlsh (fd);
389 if (vlsh != VLS_INVALID_HANDLE)
391 size = vls_read (vlsh, buf, nbytes);
400 size = libc_read (fd, buf, nbytes);
407 readv (int fd, const struct iovec * iov, int iovcnt)
409 int rv = 0, i, total = 0;
415 vlsh = ldp_fd_to_vlsh (fd);
416 if (vlsh != VLS_INVALID_HANDLE)
418 for (i = 0; i < iovcnt; ++i)
420 rv = vls_read (vlsh, iov[i].iov_base, iov[i].iov_len);
426 if (rv < iov[i].iov_len)
430 if (rv < 0 && total == 0)
440 size = libc_readv (fd, iov, iovcnt);
447 write (int fd, const void *buf, size_t nbytes)
454 vlsh = ldp_fd_to_vlsh (fd);
455 if (vlsh != VLS_INVALID_HANDLE)
457 size = vls_write_msg (vlsh, (void *) buf, nbytes);
466 size = libc_write (fd, buf, nbytes);
473 writev (int fd, const struct iovec * iov, int iovcnt)
475 ssize_t size = 0, total = 0;
481 vlsh = ldp_fd_to_vlsh (fd);
482 if (vlsh != VLS_INVALID_HANDLE)
484 for (i = 0; i < iovcnt; ++i)
486 rv = vls_write_msg (vlsh, iov[i].iov_base, iov[i].iov_len);
492 if (rv < iov[i].iov_len)
497 if (rv < 0 && total == 0)
507 size = libc_writev (fd, iov, iovcnt);
514 fcntl_internal (int fd, int cmd, va_list ap)
519 vlsh = ldp_fd_to_vlsh (fd);
520 LDBG (0, "fd %u vlsh %d, cmd %u", fd, vlsh, cmd);
521 if (vlsh != VLS_INVALID_HANDLE)
523 int flags = va_arg (ap, int);
526 size = sizeof (flags);
531 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_FLAGS, &flags, &size);
535 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_FLAGS, &flags, &size);
540 /* TODO handle this */
541 LDBG (0, "F_SETFD ignored flags %u", flags);
557 rv = libc_vfcntl64 (fd, cmd, ap);
559 rv = libc_vfcntl (fd, cmd, ap);
567 fcntl (int fd, int cmd, ...)
575 rv = fcntl_internal (fd, cmd, ap);
582 fcntl64 (int fd, int cmd, ...)
590 rv = fcntl_internal (fd, cmd, ap);
596 ioctl (int fd, unsigned long int cmd, ...)
606 vlsh = ldp_fd_to_vlsh (fd);
607 if (vlsh != VLS_INVALID_HANDLE)
612 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_NREAD, 0, 0);
617 u32 flags = *(va_arg (ap, int *)) ? O_NONBLOCK : 0;
618 u32 size = sizeof (flags);
620 /* TBD: When VPPCOM_ATTR_[GS]ET_FLAGS supports flags other than
621 * non-blocking, the flags should be read here and merged
624 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_FLAGS, &flags, &size);
640 rv = libc_vioctl (fd, cmd, ap);
648 ldp_select_init_maps (fd_set * __restrict original,
649 clib_bitmap_t ** resultb, clib_bitmap_t ** libcb,
650 clib_bitmap_t ** vclb, int nfds, u32 minbits,
651 u32 n_bytes, uword * si_bits, uword * libc_bits)
653 uword si_bits_set, libc_bits_set;
657 clib_bitmap_validate (*vclb, minbits);
658 clib_bitmap_validate (*libcb, minbits);
659 clib_bitmap_validate (*resultb, minbits);
660 clib_memcpy_fast (*resultb, original, n_bytes);
661 memset (original, 0, n_bytes);
664 clib_bitmap_foreach (fd, *resultb) {
667 vlsh = ldp_fd_to_vlsh (fd);
668 if (vlsh == VLS_INVALID_HANDLE)
669 clib_bitmap_set_no_check (*libcb, fd, 1);
671 *vclb = clib_bitmap_set (*vclb, vlsh_to_session_index (vlsh), 1);
675 si_bits_set = clib_bitmap_last_set (*vclb) + 1;
676 *si_bits = (si_bits_set > *si_bits) ? si_bits_set : *si_bits;
677 clib_bitmap_validate (*resultb, *si_bits);
679 libc_bits_set = clib_bitmap_last_set (*libcb) + 1;
680 *libc_bits = (libc_bits_set > *libc_bits) ? libc_bits_set : *libc_bits;
684 ldp_select_vcl_map_to_libc (clib_bitmap_t * vclb, fd_set * __restrict libcb)
694 clib_bitmap_foreach (si, vclb) {
695 vlsh = vls_session_index_to_vlsh (si);
696 ASSERT (vlsh != VLS_INVALID_HANDLE);
697 fd = ldp_vlsh_to_fd (vlsh);
698 if (PREDICT_FALSE (fd < 0))
711 ldp_select_libc_map_merge (clib_bitmap_t * result, fd_set * __restrict libcb)
719 clib_bitmap_foreach (fd, result)
720 FD_SET ((int)fd, libcb);
725 ldp_pselect (int nfds, fd_set * __restrict readfds,
726 fd_set * __restrict writefds,
727 fd_set * __restrict exceptfds,
728 const struct timespec *__restrict timeout,
729 const __sigset_t * __restrict sigmask)
731 u32 minbits = clib_max (nfds, BITS (uword)), n_bytes;
732 ldp_worker_ctx_t *ldpw = ldp_worker_get_current ();
733 struct timespec libc_tspec = { 0 };
734 f64 time_out, vcl_timeout = 0;
735 uword si_bits, libc_bits;
736 int rv, bits_set = 0;
744 if (PREDICT_FALSE (ldpw->clib_time.init_cpu_time == 0))
745 clib_time_init (&ldpw->clib_time);
749 time_out = (timeout->tv_sec == 0 && timeout->tv_nsec == 0) ?
750 (f64) 0 : (f64) timeout->tv_sec + (f64) timeout->tv_nsec / (f64) 1e9;
752 time_out += clib_time_now (&ldpw->clib_time);
754 /* select as fine grained sleep */
757 while (clib_time_now (&ldpw->clib_time) < time_out)
770 if (nfds <= ldp->vlsh_bit_val)
772 rv = libc_pselect (nfds, readfds, writefds, exceptfds,
777 si_bits = libc_bits = 0;
778 n_bytes = nfds / 8 + ((nfds % 8) ? 1 : 0);
781 ldp_select_init_maps (readfds, &ldpw->rd_bitmap, &ldpw->libc_rd_bitmap,
782 &ldpw->si_rd_bitmap, nfds, minbits, n_bytes,
783 &si_bits, &libc_bits);
785 ldp_select_init_maps (writefds, &ldpw->wr_bitmap,
786 &ldpw->libc_wr_bitmap, &ldpw->si_wr_bitmap, nfds,
787 minbits, n_bytes, &si_bits, &libc_bits);
789 ldp_select_init_maps (exceptfds, &ldpw->ex_bitmap,
790 &ldpw->libc_ex_bitmap, &ldpw->si_ex_bitmap, nfds,
791 minbits, n_bytes, &si_bits, &libc_bits);
793 if (PREDICT_FALSE (!si_bits && !libc_bits))
801 libc_tspec = timeout ? *timeout : libc_tspec;
808 clib_memcpy_fast (ldpw->rd_bitmap, ldpw->si_rd_bitmap,
809 vec_len (ldpw->si_rd_bitmap) *
810 sizeof (clib_bitmap_t));
812 clib_memcpy_fast (ldpw->wr_bitmap, ldpw->si_wr_bitmap,
813 vec_len (ldpw->si_wr_bitmap) *
814 sizeof (clib_bitmap_t));
816 clib_memcpy_fast (ldpw->ex_bitmap, ldpw->si_ex_bitmap,
817 vec_len (ldpw->si_ex_bitmap) *
818 sizeof (clib_bitmap_t));
820 rv = vls_select (si_bits, readfds ? ldpw->rd_bitmap : NULL,
821 writefds ? ldpw->wr_bitmap : NULL,
822 exceptfds ? ldpw->ex_bitmap : NULL, vcl_timeout);
831 if (ldp_select_vcl_map_to_libc (ldpw->rd_bitmap, readfds))
837 if (ldp_select_vcl_map_to_libc (ldpw->wr_bitmap, writefds))
843 if (ldp_select_vcl_map_to_libc (ldpw->ex_bitmap, exceptfds))
854 clib_memcpy_fast (ldpw->rd_bitmap, ldpw->libc_rd_bitmap,
855 vec_len (ldpw->libc_rd_bitmap) *
856 sizeof (clib_bitmap_t));
858 clib_memcpy_fast (ldpw->wr_bitmap, ldpw->libc_wr_bitmap,
859 vec_len (ldpw->libc_wr_bitmap) *
860 sizeof (clib_bitmap_t));
862 clib_memcpy_fast (ldpw->ex_bitmap, ldpw->libc_ex_bitmap,
863 vec_len (ldpw->libc_ex_bitmap) *
864 sizeof (clib_bitmap_t));
866 rv = libc_pselect (libc_bits,
867 readfds ? (fd_set *) ldpw->rd_bitmap : NULL,
868 writefds ? (fd_set *) ldpw->wr_bitmap : NULL,
869 exceptfds ? (fd_set *) ldpw->ex_bitmap : NULL,
870 &libc_tspec, sigmask);
873 ldp_select_libc_map_merge (ldpw->rd_bitmap, readfds);
874 ldp_select_libc_map_merge (ldpw->wr_bitmap, writefds);
875 ldp_select_libc_map_merge (ldpw->ex_bitmap, exceptfds);
886 while ((time_out == -1) || (clib_time_now (&ldpw->clib_time) < time_out));
890 /* TBD: set timeout to amount of time left */
891 clib_bitmap_zero (ldpw->rd_bitmap);
892 clib_bitmap_zero (ldpw->si_rd_bitmap);
893 clib_bitmap_zero (ldpw->libc_rd_bitmap);
894 clib_bitmap_zero (ldpw->wr_bitmap);
895 clib_bitmap_zero (ldpw->si_wr_bitmap);
896 clib_bitmap_zero (ldpw->libc_wr_bitmap);
897 clib_bitmap_zero (ldpw->ex_bitmap);
898 clib_bitmap_zero (ldpw->si_ex_bitmap);
899 clib_bitmap_zero (ldpw->libc_ex_bitmap);
905 select (int nfds, fd_set * __restrict readfds,
906 fd_set * __restrict writefds,
907 fd_set * __restrict exceptfds, struct timeval *__restrict timeout)
909 struct timespec tspec;
913 tspec.tv_sec = timeout->tv_sec;
914 tspec.tv_nsec = timeout->tv_usec * 1000;
916 return ldp_pselect (nfds, readfds, writefds, exceptfds,
917 timeout ? &tspec : NULL, NULL);
922 pselect (int nfds, fd_set * __restrict readfds,
923 fd_set * __restrict writefds,
924 fd_set * __restrict exceptfds,
925 const struct timespec *__restrict timeout,
926 const __sigset_t * __restrict sigmask)
928 return ldp_pselect (nfds, readfds, writefds, exceptfds, timeout, 0);
932 /* If transparent TLS mode is turned on, then ldp will load key and cert.
935 load_cert_key_pair (void)
937 char *cert_str = getenv (LDP_ENV_TLS_CERT);
938 char *key_str = getenv (LDP_ENV_TLS_KEY);
939 char cert_buf[4096], key_buf[4096];
940 int cert_size, key_size;
941 vppcom_cert_key_pair_t crypto;
945 if (!cert_str || !key_str)
947 LDBG (0, "ERROR: failed to read LDP environment %s\n",
952 fp = fopen (cert_str, "r");
955 LDBG (0, "ERROR: failed to open cert file %s \n", cert_str);
958 cert_size = fread (cert_buf, sizeof (char), sizeof (cert_buf), fp);
961 fp = fopen (key_str, "r");
964 LDBG (0, "ERROR: failed to open key file %s \n", key_str);
967 key_size = fread (key_buf, sizeof (char), sizeof (key_buf), fp);
970 crypto.cert = cert_buf;
971 crypto.key = key_buf;
972 crypto.cert_len = cert_size;
973 crypto.key_len = key_size;
974 ckp_index = vppcom_add_cert_key_pair (&crypto);
977 LDBG (0, "ERROR: failed to add cert key pair\n");
981 ldp->ckpair_index = ckp_index;
987 assign_cert_key_pair (vls_handle_t vlsh)
991 if (ldp->ckpair_index == ~0 && load_cert_key_pair () < 0)
994 ckp_len = sizeof (ldp->ckpair_index);
995 return vls_attr (vlsh, VPPCOM_ATTR_SET_CKPAIR, &ldp->ckpair_index, &ckp_len);
999 socket (int domain, int type, int protocol)
1001 int rv, sock_type = type & ~(SOCK_CLOEXEC | SOCK_NONBLOCK);
1002 u8 is_nonblocking = type & SOCK_NONBLOCK ? 1 : 0;
1007 if (((domain == AF_INET) || (domain == AF_INET6)) &&
1008 ((sock_type == SOCK_STREAM) || (sock_type == SOCK_DGRAM)))
1011 if (ldp->transparent_tls)
1013 proto = VPPCOM_PROTO_TLS;
1016 proto = ((sock_type == SOCK_DGRAM) ?
1017 VPPCOM_PROTO_UDP : VPPCOM_PROTO_TCP);
1019 LDBG (0, "calling vls_create: proto %u (%s), is_nonblocking %u",
1020 proto, vppcom_proto_str (proto), is_nonblocking);
1022 vlsh = vls_create (proto, is_nonblocking);
1030 if (ldp->transparent_tls)
1032 if (assign_cert_key_pair (vlsh) < 0)
1035 rv = ldp_vlsh_to_fd (vlsh);
1040 LDBG (0, "calling libc_socket");
1041 rv = libc_socket (domain, type, protocol);
1048 * Create two new sockets, of type TYPE in domain DOMAIN and using
1049 * protocol PROTOCOL, which are connected to each other, and put file
1050 * descriptors for them in FDS[0] and FDS[1]. If PROTOCOL is zero,
1051 * one will be chosen automatically.
1052 * Returns 0 on success, -1 for errors.
1055 socketpair (int domain, int type, int protocol, int fds[2])
1057 int rv, sock_type = type & ~(SOCK_CLOEXEC | SOCK_NONBLOCK);
1061 if (((domain == AF_INET) || (domain == AF_INET6)) &&
1062 ((sock_type == SOCK_STREAM) || (sock_type == SOCK_DGRAM)))
1064 LDBG (0, "LDP-TBD");
1070 LDBG (1, "calling libc_socketpair");
1071 rv = libc_socketpair (domain, type, protocol, fds);
1078 bind (int fd, __CONST_SOCKADDR_ARG _addr, socklen_t len)
1080 const struct sockaddr *addr = SOCKADDR_GET_SA (_addr);
1086 vlsh = ldp_fd_to_vlsh (fd);
1087 if (vlsh != VLS_INVALID_HANDLE)
1091 switch (addr->sa_family)
1094 if (len != sizeof (struct sockaddr_in))
1096 LDBG (0, "ERROR: fd %d: vlsh %u: Invalid AF_INET addr len %u!",
1102 ep.is_ip4 = VPPCOM_IS_IP4;
1103 ep.ip = (u8 *) & ((const struct sockaddr_in *) addr)->sin_addr;
1104 ep.port = (u16) ((const struct sockaddr_in *) addr)->sin_port;
1108 if (len != sizeof (struct sockaddr_in6))
1110 LDBG (0, "ERROR: fd %d: vlsh %u: Invalid AF_INET6 addr len %u!",
1116 ep.is_ip4 = VPPCOM_IS_IP6;
1117 ep.ip = (u8 *) & ((const struct sockaddr_in6 *) addr)->sin6_addr;
1118 ep.port = (u16) ((const struct sockaddr_in6 *) addr)->sin6_port;
1122 LDBG (0, "ERROR: fd %d: vlsh %u: Unsupported address family %u!",
1123 fd, vlsh, addr->sa_family);
1124 errno = EAFNOSUPPORT;
1128 LDBG (0, "fd %d: calling vls_bind: vlsh %u, addr %p, len %u", fd, vlsh,
1131 rv = vls_bind (vlsh, &ep);
1132 if (rv != VPPCOM_OK)
1140 LDBG (0, "fd %d: calling libc_bind: addr %p, len %u", fd, addr, len);
1141 rv = libc_bind (fd, addr, len);
1145 LDBG (1, "fd %d: returning %d", fd, rv);
1151 ldp_copy_ep_to_sockaddr (struct sockaddr *addr, socklen_t *__restrict len,
1154 int rv = 0, sa_len, copy_len;
1158 if (addr && len && ep)
1160 addr->sa_family = (ep->is_ip4 == VPPCOM_IS_IP4) ? AF_INET : AF_INET6;
1161 switch (addr->sa_family)
1164 ((struct sockaddr_in *) addr)->sin_port = ep->port;
1165 if (*len > sizeof (struct sockaddr_in))
1166 *len = sizeof (struct sockaddr_in);
1167 sa_len = sizeof (struct sockaddr_in) - sizeof (struct in_addr);
1168 copy_len = *len - sa_len;
1170 memcpy (&((struct sockaddr_in *) addr)->sin_addr, ep->ip,
1175 ((struct sockaddr_in6 *) addr)->sin6_port = ep->port;
1176 if (*len > sizeof (struct sockaddr_in6))
1177 *len = sizeof (struct sockaddr_in6);
1178 sa_len = sizeof (struct sockaddr_in6) - sizeof (struct in6_addr);
1179 copy_len = *len - sa_len;
1181 memcpy (((struct sockaddr_in6 *) addr)->sin6_addr.
1182 __in6_u.__u6_addr8, ep->ip, copy_len);
1195 getsockname (int fd, __SOCKADDR_ARG _addr, socklen_t *__restrict len)
1197 struct sockaddr *addr = SOCKADDR_GET_SA (_addr);
1203 vlsh = ldp_fd_to_vlsh (fd);
1204 if (vlsh != VLS_INVALID_HANDLE)
1207 u8 addr_buf[sizeof (struct in6_addr)];
1208 u32 size = sizeof (ep);
1212 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_LCL_ADDR, &ep, &size);
1213 if (rv != VPPCOM_OK)
1220 rv = ldp_copy_ep_to_sockaddr (addr, len, &ep);
1221 if (rv != VPPCOM_OK)
1230 rv = libc_getsockname (fd, _addr, len);
1237 connect (int fd, __CONST_SOCKADDR_ARG _addr, socklen_t len)
1239 const struct sockaddr *addr = SOCKADDR_GET_SA (_addr);
1247 LDBG (0, "ERROR: fd %d: NULL addr, len %u", fd, len);
1253 vlsh = ldp_fd_to_vlsh (fd);
1254 if (vlsh != VLS_INVALID_HANDLE)
1258 switch (addr->sa_family)
1261 if (len != sizeof (struct sockaddr_in))
1263 LDBG (0, "fd %d: ERROR vlsh %u: Invalid AF_INET addr len %u!",
1269 ep.is_ip4 = VPPCOM_IS_IP4;
1270 ep.ip = (u8 *) & ((const struct sockaddr_in *) addr)->sin_addr;
1271 ep.port = (u16) ((const struct sockaddr_in *) addr)->sin_port;
1275 if (len != sizeof (struct sockaddr_in6))
1277 LDBG (0, "fd %d: ERROR vlsh %u: Invalid AF_INET6 addr len %u!",
1283 ep.is_ip4 = VPPCOM_IS_IP6;
1284 ep.ip = (u8 *) & ((const struct sockaddr_in6 *) addr)->sin6_addr;
1285 ep.port = (u16) ((const struct sockaddr_in6 *) addr)->sin6_port;
1289 LDBG (0, "fd %d: ERROR vlsh %u: Unsupported address family %u!",
1290 fd, vlsh, addr->sa_family);
1291 errno = EAFNOSUPPORT;
1295 LDBG (0, "fd %d: calling vls_connect(): vlsh %u addr %p len %u", fd,
1298 rv = vls_connect (vlsh, &ep);
1299 if (rv != VPPCOM_OK)
1307 LDBG (0, "fd %d: calling libc_connect(): addr %p, len %u",
1310 rv = libc_connect (fd, addr, len);
1314 LDBG (1, "fd %d: returning %d (0x%x)", fd, rv, rv);
1319 getpeername (int fd, __SOCKADDR_ARG _addr, socklen_t *__restrict len)
1321 struct sockaddr *addr = SOCKADDR_GET_SA (_addr);
1327 vlsh = ldp_fd_to_vlsh (fd);
1328 if (vlsh != VLS_INVALID_HANDLE)
1331 u8 addr_buf[sizeof (struct in6_addr)];
1332 u32 size = sizeof (ep);
1335 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_PEER_ADDR, &ep, &size);
1336 if (rv != VPPCOM_OK)
1343 rv = ldp_copy_ep_to_sockaddr (addr, len, &ep);
1344 if (rv != VPPCOM_OK)
1353 rv = libc_getpeername (fd, addr, len);
1360 send (int fd, const void *buf, size_t n, int flags)
1362 vls_handle_t vlsh = ldp_fd_to_vlsh (fd);
1367 if (vlsh != VLS_INVALID_HANDLE)
1369 size = vls_sendto (vlsh, (void *) buf, n, flags, NULL);
1370 if (size < VPPCOM_OK)
1378 size = libc_send (fd, buf, n, flags);
1385 sendfile (int out_fd, int in_fd, off_t * offset, size_t len)
1387 ldp_worker_ctx_t *ldpw = ldp_worker_get_current ();
1393 vlsh = ldp_fd_to_vlsh (out_fd);
1394 if (vlsh != VLS_INVALID_HANDLE)
1397 ssize_t results = 0;
1398 size_t n_bytes_left = len;
1399 size_t bytes_to_read;
1402 u32 flags, flags_len = sizeof (flags);
1404 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_FLAGS, &flags, &flags_len);
1405 if (PREDICT_FALSE (rv != VPPCOM_OK))
1407 LDBG (0, "ERROR: out fd %d: vls_attr: vlsh %u, returned %d (%s)!",
1408 out_fd, vlsh, rv, vppcom_retval_str (rv));
1410 vec_reset_length (ldpw->io_buffer);
1418 off_t off = lseek (in_fd, *offset, SEEK_SET);
1419 if (PREDICT_FALSE (off == -1))
1425 ASSERT (off == *offset);
1430 size = vls_attr (vlsh, VPPCOM_ATTR_GET_NWRITE, 0, 0);
1433 LDBG (0, "ERROR: fd %d: vls_attr: vlsh %u returned %ld (%s)!",
1434 out_fd, vlsh, size, vppcom_retval_str (size));
1435 vec_reset_length (ldpw->io_buffer);
1441 bytes_to_read = size;
1442 if (bytes_to_read == 0)
1444 if (flags & O_NONBLOCK)
1453 bytes_to_read = clib_min (n_bytes_left, bytes_to_read);
1454 vec_validate (ldpw->io_buffer, bytes_to_read);
1455 nbytes = libc_read (in_fd, ldpw->io_buffer, bytes_to_read);
1460 vec_reset_length (ldpw->io_buffer);
1467 size = vls_write (vlsh, ldpw->io_buffer, nbytes);
1470 if (size == VPPCOM_EAGAIN)
1472 if (flags & O_NONBLOCK)
1483 vec_reset_length (ldpw->io_buffer);
1492 ASSERT (n_bytes_left >= nbytes);
1493 n_bytes_left = n_bytes_left - nbytes;
1495 while (n_bytes_left > 0);
1498 vec_reset_length (ldpw->io_buffer);
1501 off_t off = lseek (in_fd, *offset, SEEK_SET);
1502 if (PREDICT_FALSE (off == -1))
1508 ASSERT (off == *offset);
1509 *offset += results + 1;
1521 size = libc_sendfile (out_fd, in_fd, offset, len);
1529 sendfile64 (int out_fd, int in_fd, off_t * offset, size_t len)
1531 return sendfile (out_fd, in_fd, offset, len);
1535 recv (int fd, void *buf, size_t n, int flags)
1542 vlsh = ldp_fd_to_vlsh (fd);
1543 if (vlsh != VLS_INVALID_HANDLE)
1545 size = vls_recvfrom (vlsh, buf, n, flags, NULL);
1554 size = libc_recv (fd, buf, n, flags);
1561 __recv_chk (int fd, void *buf, size_t n, size_t buflen, int flags)
1566 return recv (fd, buf, n, flags);
1570 ldp_vls_sendo (vls_handle_t vlsh, const void *buf, size_t n,
1571 vppcom_endpt_tlv_t *app_tlvs, int flags,
1572 __CONST_SOCKADDR_ARG _addr, socklen_t addr_len)
1574 const struct sockaddr *addr = SOCKADDR_GET_SA (_addr);
1575 vppcom_endpt_t *ep = 0;
1578 _ep.app_tlvs = app_tlvs;
1583 switch (addr->sa_family)
1586 ep->is_ip4 = VPPCOM_IS_IP4;
1588 (uint8_t *) & ((const struct sockaddr_in *) addr)->sin_addr;
1589 ep->port = (uint16_t) ((const struct sockaddr_in *) addr)->sin_port;
1593 ep->is_ip4 = VPPCOM_IS_IP6;
1595 (uint8_t *) & ((const struct sockaddr_in6 *) addr)->sin6_addr;
1597 (uint16_t) ((const struct sockaddr_in6 *) addr)->sin6_port;
1601 return EAFNOSUPPORT;
1605 return vls_sendto (vlsh, (void *) buf, n, flags, ep);
1609 ldp_vls_recvfrom (vls_handle_t vlsh, void *__restrict buf, size_t n, int flags,
1610 __SOCKADDR_ARG _addr, socklen_t *__restrict addr_len)
1612 u8 src_addr[sizeof (struct sockaddr_in6)];
1613 struct sockaddr *addr = SOCKADDR_GET_SA (_addr);
1621 size = vls_recvfrom (vlsh, buf, n, flags, &ep);
1625 rv = ldp_copy_ep_to_sockaddr (addr, addr_len, &ep);
1631 size = vls_recvfrom (vlsh, buf, n, flags, NULL);
1637 sendto (int fd, const void *buf, size_t n, int flags,
1638 __CONST_SOCKADDR_ARG _addr, socklen_t addr_len)
1640 const struct sockaddr *addr = SOCKADDR_GET_SA (_addr);
1646 vlsh = ldp_fd_to_vlsh (fd);
1647 if (vlsh != VLS_INVALID_HANDLE)
1649 size = ldp_vls_sendo (vlsh, buf, n, NULL, flags, addr, addr_len);
1658 size = libc_sendto (fd, buf, n, flags, addr, addr_len);
1665 recvfrom (int fd, void *__restrict buf, size_t n, int flags,
1666 __SOCKADDR_ARG addr, socklen_t * __restrict addr_len)
1673 vlsh = ldp_fd_to_vlsh (fd);
1674 if (vlsh != VLS_INVALID_HANDLE)
1676 size = ldp_vls_recvfrom (vlsh, buf, n, flags, addr, addr_len);
1685 size = libc_recvfrom (fd, buf, n, flags, addr, addr_len);
1692 ldp_parse_cmsg (vls_handle_t vlsh, const struct msghdr *msg,
1693 vppcom_endpt_tlv_t **app_tlvs)
1695 uint8_t *ad, *at = (uint8_t *) *app_tlvs;
1696 vppcom_endpt_tlv_t *adh;
1697 struct in_pktinfo *pi;
1698 struct cmsghdr *cmsg;
1700 cmsg = CMSG_FIRSTHDR (msg);
1702 while (cmsg != NULL)
1704 switch (cmsg->cmsg_level)
1707 switch (cmsg->cmsg_type)
1710 vec_add2 (at, adh, sizeof (*adh));
1711 adh->data_type = VCL_UDP_SEGMENT;
1712 adh->data_len = sizeof (uint16_t);
1713 vec_add2 (at, ad, sizeof (uint16_t));
1714 *(uint16_t *) ad = *(uint16_t *) CMSG_DATA (cmsg);
1717 LDBG (1, "SOL_UDP cmsg_type %u not supported", cmsg->cmsg_type);
1722 switch (cmsg->cmsg_type)
1725 vec_add2 (at, adh, sizeof (*adh));
1726 adh->data_type = VCL_IP_PKTINFO;
1727 adh->data_len = sizeof (struct in_addr);
1728 vec_add2 (at, ad, sizeof (struct in_addr));
1729 pi = (void *) CMSG_DATA (cmsg);
1730 clib_memcpy_fast (ad, &pi->ipi_spec_dst,
1731 sizeof (struct in_addr));
1734 LDBG (1, "SOL_IP cmsg_type %u not supported", cmsg->cmsg_type);
1739 LDBG (1, "cmsg_level %u not supported", cmsg->cmsg_level);
1742 cmsg = CMSG_NXTHDR ((struct msghdr *) msg, cmsg);
1744 *app_tlvs = (vppcom_endpt_tlv_t *) at;
1749 ldp_make_cmsg (vls_handle_t vlsh, struct msghdr *msg)
1751 u32 optval, optlen = sizeof (optval);
1752 struct cmsghdr *cmsg;
1754 cmsg = CMSG_FIRSTHDR (msg);
1755 memset (cmsg, 0, sizeof (*cmsg));
1757 if (!vls_attr (vlsh, VPPCOM_ATTR_GET_IP_PKTINFO, (void *) &optval, &optlen))
1763 u8 addr_buf[sizeof (struct in_addr)];
1764 u32 size = sizeof (ep);
1768 if (!vls_attr (vlsh, VPPCOM_ATTR_GET_LCL_ADDR, &ep, &size))
1770 struct in_pktinfo pi = {};
1772 clib_memcpy (&pi.ipi_addr, ep.ip, sizeof (struct in_addr));
1773 cmsg->cmsg_level = SOL_IP;
1774 cmsg->cmsg_type = IP_PKTINFO;
1775 cmsg->cmsg_len = CMSG_LEN (sizeof (pi));
1776 clib_memcpy (CMSG_DATA (cmsg), &pi, sizeof (pi));
1784 sendmsg (int fd, const struct msghdr * msg, int flags)
1791 vlsh = ldp_fd_to_vlsh (fd);
1792 if (vlsh != VLS_INVALID_HANDLE)
1794 vppcom_endpt_tlv_t *app_tlvs = 0;
1795 struct iovec *iov = msg->msg_iov;
1799 ldp_parse_cmsg (vlsh, msg, &app_tlvs);
1801 for (i = 0; i < msg->msg_iovlen; ++i)
1803 rv = ldp_vls_sendo (vlsh, iov[i].iov_base, iov[i].iov_len, app_tlvs,
1804 flags, msg->msg_name, msg->msg_namelen);
1810 if (rv < iov[i].iov_len)
1815 vec_free (app_tlvs);
1817 if (rv < 0 && total == 0)
1827 size = libc_sendmsg (fd, msg, flags);
1835 sendmmsg (int fd, struct mmsghdr *vmessages, unsigned int vlen, int flags)
1838 const char *func_str;
1839 u32 sh = ldp_fd_to_vlsh (fd);
1843 if (sh != VLS_INVALID_HANDLE)
1845 clib_warning ("LDP<%d>: LDP-TBD", getpid ());
1851 func_str = "libc_sendmmsg";
1854 clib_warning ("LDP<%d>: fd %d (0x%x): calling %s(): "
1855 "vmessages %p, vlen %u, flags 0x%x",
1856 getpid (), fd, fd, func_str, vmessages, vlen, flags);
1858 size = libc_sendmmsg (fd, vmessages, vlen, flags);
1865 int errno_val = errno;
1867 clib_warning ("LDP<%d>: ERROR: fd %d (0x%x): %s() failed! "
1868 "rv %d, errno = %d", getpid (), fd, fd,
1869 func_str, size, errno_val);
1873 clib_warning ("LDP<%d>: fd %d (0x%x): returning %d (0x%x)",
1874 getpid (), fd, fd, size, size);
1881 recvmsg (int fd, struct msghdr * msg, int flags)
1888 vlsh = ldp_fd_to_vlsh (fd);
1889 if (vlsh != VLS_INVALID_HANDLE)
1891 struct iovec *iov = msg->msg_iov;
1892 ssize_t max_deq, total = 0;
1895 max_deq = vls_attr (vlsh, VPPCOM_ATTR_GET_NREAD, 0, 0);
1899 for (i = 0; i < msg->msg_iovlen; i++)
1901 rv = ldp_vls_recvfrom (vlsh, iov[i].iov_base, iov[i].iov_len, flags,
1902 (i == 0 ? msg->msg_name : NULL),
1903 (i == 0 ? &msg->msg_namelen : NULL));
1909 if (rv < iov[i].iov_len)
1912 if (total >= max_deq)
1916 if (rv < 0 && total == 0)
1923 if (msg->msg_controllen)
1924 ldp_make_cmsg (vlsh, msg);
1930 size = libc_recvmsg (fd, msg, flags);
1938 recvmmsg (int fd, struct mmsghdr *vmessages,
1939 unsigned int vlen, int flags, struct timespec *tmo)
1941 ldp_worker_ctx_t *ldpw = ldp_worker_get_current ();
1946 sh = ldp_fd_to_vlsh (fd);
1948 if (sh != VLS_INVALID_HANDLE)
1955 if (PREDICT_FALSE (ldpw->clib_time.init_cpu_time == 0))
1956 clib_time_init (&ldpw->clib_time);
1959 time_out = (f64) tmo->tv_sec + (f64) tmo->tv_nsec / (f64) 1e9;
1960 time_out += clib_time_now (&ldpw->clib_time);
1964 time_out = (f64) ~0;
1967 while (nvecs < vlen)
1969 mh = &vmessages[nvecs];
1970 rv = recvmsg (fd, &mh->msg_hdr, flags);
1978 if (!time_out || clib_time_now (&ldpw->clib_time) >= time_out)
1984 return nvecs > 0 ? nvecs : rv;
1988 return libc_recvmmsg (fd, vmessages, vlen, flags, tmo);
1994 getsockopt (int fd, int level, int optname,
1995 void *__restrict optval, socklen_t * __restrict optlen)
2002 vlsh = ldp_fd_to_vlsh (fd);
2003 if (vlsh != VLS_INVALID_HANDLE)
2013 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_TCP_NODELAY,
2017 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_TCP_USER_MSS,
2021 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_TCP_KEEPIDLE,
2025 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_TCP_KEEPINTVL,
2029 if (optval && optlen && (*optlen == sizeof (struct tcp_info)))
2031 LDBG (1, "fd %d: vlsh %u SOL_TCP, TCP_INFO, optval %p, "
2032 "optlen %d: #LDP-NOP#", fd, vlsh, optval, *optlen);
2033 memset (optval, 0, *optlen);
2039 case TCP_CONGESTION:
2040 *optlen = strlen ("cubic");
2041 strncpy (optval, "cubic", *optlen + 1);
2045 LDBG (0, "ERROR: fd %d: getsockopt SOL_TCP: sid %u, "
2046 "optname %d unsupported!", fd, vlsh, optname);
2053 case SO_ORIGINAL_DST:
2055 vls_attr (vlsh, VPPCOM_ATTR_GET_ORIGINAL_DST, optval, optlen);
2059 "ERROR: fd %d: getsockopt SOL_IP: vlsh %u "
2060 "optname %d unsupported!",
2069 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_V6ONLY, optval, optlen);
2072 LDBG (0, "ERROR: fd %d: getsockopt SOL_IPV6: vlsh %u "
2073 "optname %d unsupported!", fd, vlsh, optname);
2081 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_LISTEN, optval, optlen);
2084 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_KEEPALIVE, optval, optlen);
2087 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_PROTOCOL, optval, optlen);
2088 *(int *) optval = *(int *) optval ? SOCK_DGRAM : SOCK_STREAM;
2091 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_TX_FIFO_LEN,
2095 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_RX_FIFO_LEN,
2099 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_REUSEADDR, optval, optlen);
2102 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_REUSEPORT, optval, optlen);
2105 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_BROADCAST, optval, optlen);
2108 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_DOMAIN, optval, optlen);
2111 rv = vls_attr (vlsh, VPPCOM_ATTR_GET_ERROR, optval, optlen);
2113 case SO_BINDTODEVICE:
2117 LDBG (0, "ERROR: fd %d: getsockopt SOL_SOCKET: vlsh %u "
2118 "optname %d unsupported!", fd, vlsh, optname);
2126 if (rv != VPPCOM_OK)
2134 rv = libc_getsockopt (fd, level, optname, optval, optlen);
2141 setsockopt (int fd, int level, int optname,
2142 const void *optval, socklen_t optlen)
2149 vlsh = ldp_fd_to_vlsh (fd);
2150 if (vlsh != VLS_INVALID_HANDLE)
2160 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_TCP_NODELAY,
2161 (void *) optval, &optlen);
2164 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_TCP_USER_MSS,
2165 (void *) optval, &optlen);
2168 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_TCP_KEEPIDLE,
2169 (void *) optval, &optlen);
2172 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_TCP_KEEPINTVL,
2173 (void *) optval, &optlen);
2175 case TCP_CONGESTION:
2181 LDBG (0, "ERROR: fd %d: setsockopt() SOL_TCP: vlsh %u"
2182 "optname %d unsupported!", fd, vlsh, optname);
2190 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_V6ONLY,
2191 (void *) optval, &optlen);
2194 LDBG (0, "ERROR: fd %d: setsockopt SOL_IPV6: vlsh %u"
2195 "optname %d unsupported!", fd, vlsh, optname);
2203 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_KEEPALIVE,
2204 (void *) optval, &optlen);
2207 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_REUSEADDR,
2208 (void *) optval, &optlen);
2211 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_REUSEPORT, (void *) optval,
2215 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_BROADCAST,
2216 (void *) optval, &optlen);
2222 LDBG (0, "ERROR: fd %d: setsockopt SOL_SOCKET: vlsh %u "
2223 "optname %d unsupported!", fd, vlsh, optname);
2231 rv = vls_attr (vlsh, VPPCOM_ATTR_SET_IP_PKTINFO, (void *) optval,
2236 "ERROR: fd %d: setsockopt SOL_IP: vlsh %u optname %d"
2246 if (rv != VPPCOM_OK)
2254 rv = libc_setsockopt (fd, level, optname, optval, optlen);
2261 listen (int fd, int n)
2268 vlsh = ldp_fd_to_vlsh (fd);
2269 if (vlsh != VLS_INVALID_HANDLE)
2271 LDBG (0, "fd %d: calling vls_listen: vlsh %u, n %d", fd, vlsh, n);
2273 rv = vls_listen (vlsh, n);
2274 if (rv != VPPCOM_OK)
2282 LDBG (0, "fd %d: calling libc_listen(): n %d", fd, n);
2283 rv = libc_listen (fd, n);
2286 LDBG (1, "fd %d: returning %d", fd, rv);
2291 ldp_accept4 (int listen_fd, __SOCKADDR_ARG _addr,
2292 socklen_t *__restrict addr_len, int flags)
2294 struct sockaddr *addr = SOCKADDR_GET_SA (_addr);
2295 vls_handle_t listen_vlsh, accept_vlsh;
2300 listen_vlsh = ldp_fd_to_vlsh (listen_fd);
2301 if (listen_vlsh != VLS_INVALID_HANDLE)
2304 u8 src_addr[sizeof (struct sockaddr_in6)];
2305 memset (&ep, 0, sizeof (ep));
2308 LDBG (0, "listen fd %d: calling vppcom_session_accept: listen sid %u,"
2309 " ep %p, flags 0x%x", listen_fd, listen_vlsh, &ep, flags);
2311 accept_vlsh = vls_accept (listen_vlsh, &ep, flags);
2312 if (accept_vlsh < 0)
2314 errno = -accept_vlsh;
2319 rv = ldp_copy_ep_to_sockaddr (addr, addr_len, &ep);
2320 if (rv != VPPCOM_OK)
2322 (void) vls_close (accept_vlsh);
2328 rv = ldp_vlsh_to_fd (accept_vlsh);
2334 LDBG (0, "listen fd %d: calling libc_accept4(): addr %p, addr_len %p,"
2335 " flags 0x%x", listen_fd, addr, addr_len, flags);
2337 rv = libc_accept4 (listen_fd, addr, addr_len, flags);
2340 LDBG (1, "listen fd %d: accept returning %d", listen_fd, rv);
2346 accept4 (int fd, __SOCKADDR_ARG addr, socklen_t * __restrict addr_len,
2349 return ldp_accept4 (fd, addr, addr_len, flags);
2353 accept (int fd, __SOCKADDR_ARG addr, socklen_t * __restrict addr_len)
2355 return ldp_accept4 (fd, addr, addr_len, 0);
2359 shutdown (int fd, int how)
2366 vlsh = ldp_fd_to_vlsh (fd);
2367 if (vlsh != VLS_INVALID_HANDLE)
2369 LDBG (0, "called shutdown: fd %u vlsh %u how %d", fd, vlsh, how);
2370 rv = vls_shutdown (vlsh, how);
2374 LDBG (0, "fd %d: calling libc_shutdown: how %d", fd, how);
2375 rv = libc_shutdown (fd, how);
2382 epoll_create1 (int flags)
2384 ldp_worker_ctx_t *ldpw = ldp_worker_get_current ();
2390 if (ldp->vcl_needs_real_epoll || vls_use_real_epoll ())
2392 /* Make sure workers have been allocated */
2395 ldp_alloc_workers ();
2396 ldpw = ldp_worker_get_current ();
2398 rv = libc_epoll_create1 (flags);
2399 ldp->vcl_needs_real_epoll = 0;
2400 ldpw->vcl_mq_epfd = rv;
2401 LDBG (0, "created vcl epfd %u", rv);
2405 vlsh = vls_epoll_create ();
2406 if (PREDICT_FALSE (vlsh == VLS_INVALID_HANDLE))
2413 rv = ldp_vlsh_to_fd (vlsh);
2415 LDBG (0, "epoll_create epfd %u vlsh %u", rv, vlsh);
2420 epoll_create (int size)
2422 return epoll_create1 (0);
2426 epoll_ctl (int epfd, int op, int fd, struct epoll_event *event)
2428 vls_handle_t vep_vlsh, vlsh;
2433 vep_vlsh = ldp_fd_to_vlsh (epfd);
2434 if (PREDICT_FALSE (vep_vlsh == VLS_INVALID_HANDLE))
2436 /* The LDP epoll_create1 always creates VCL epfd's.
2437 * The app should never have a kernel base epoll fd unless it
2438 * was acquired outside of the LD_PRELOAD process context.
2439 * In any case, if we get one, punt it to libc_epoll_ctl.
2442 "epfd %d: calling libc_epoll_ctl: op %d, fd %d"
2444 epfd, op, fd, event ? event->events : 0);
2446 rv = libc_epoll_ctl (epfd, op, fd, event);
2450 vlsh = ldp_fd_to_vlsh (fd);
2452 LDBG (0, "epfd %d ep_vlsh %d, fd %u vlsh %d, op %u", epfd, vep_vlsh, fd,
2455 if (vlsh != VLS_INVALID_HANDLE)
2458 "epfd %d: calling vls_epoll_ctl: ep_vlsh %d op %d, vlsh %u,"
2460 epfd, vep_vlsh, op, vlsh, event ? event->events : 0);
2462 rv = vls_epoll_ctl (vep_vlsh, op, vlsh, event);
2463 if (rv != VPPCOM_OK)
2472 u32 size = sizeof (epfd);
2474 libc_epfd = vls_attr (vep_vlsh, VPPCOM_ATTR_GET_LIBC_EPFD, 0, 0);
2477 LDBG (1, "epfd %d, vep_vlsh %d calling libc_epoll_create1: "
2478 "EPOLL_CLOEXEC", epfd, vep_vlsh);
2480 libc_epfd = libc_epoll_create1 (EPOLL_CLOEXEC);
2487 rv = vls_attr (vep_vlsh, VPPCOM_ATTR_SET_LIBC_EPFD, &libc_epfd,
2496 else if (PREDICT_FALSE (libc_epfd < 0))
2503 LDBG (1, "epfd %d: calling libc_epoll_ctl: libc_epfd %d, op %d, fd %d,"
2504 " event %p", epfd, libc_epfd, op, fd, event);
2506 rv = libc_epoll_ctl (libc_epfd, op, fd, event);
2514 ldp_epoll_pwait (int epfd, struct epoll_event *events, int maxevents,
2515 int timeout, const sigset_t * sigmask)
2517 ldp_worker_ctx_t *ldpw;
2518 double time_to_wait = (double) 0, max_time;
2519 int libc_epfd, rv = 0;
2520 vls_handle_t ep_vlsh;
2524 if (PREDICT_FALSE (!events || (timeout < -1)))
2530 if (PREDICT_FALSE (vppcom_worker_index () == ~0))
2531 vls_register_vcl_worker ();
2533 ldpw = ldp_worker_get_current ();
2534 if (epfd == ldpw->vcl_mq_epfd)
2535 return libc_epoll_pwait (epfd, events, maxevents, timeout, sigmask);
2537 ep_vlsh = ldp_fd_to_vlsh (epfd);
2538 if (PREDICT_FALSE (ep_vlsh == VLS_INVALID_HANDLE))
2540 LDBG (0, "epfd %d: bad ep_vlsh %d!", epfd, ep_vlsh);
2545 if (PREDICT_FALSE (ldpw->clib_time.init_cpu_time == 0))
2546 clib_time_init (&ldpw->clib_time);
2547 time_to_wait = ((timeout >= 0) ? (double) timeout / 1000 : 0);
2548 max_time = clib_time_now (&ldpw->clib_time) + time_to_wait;
2550 libc_epfd = vls_attr (ep_vlsh, VPPCOM_ATTR_GET_LIBC_EPFD, 0, 0);
2551 if (PREDICT_FALSE (libc_epfd < 0))
2558 LDBG (2, "epfd %d: vep_idx %d, libc_epfd %d, events %p, maxevents %d, "
2559 "timeout %d, sigmask %p: time_to_wait %.02f", epfd, ep_vlsh,
2560 libc_epfd, events, maxevents, timeout, sigmask, time_to_wait);
2563 if (!ldpw->epoll_wait_vcl)
2565 rv = vls_epoll_wait (ep_vlsh, events, maxevents, 0);
2568 ldpw->epoll_wait_vcl = 1;
2579 ldpw->epoll_wait_vcl = 0;
2583 rv = libc_epoll_pwait (libc_epfd, events, maxevents, 0, sigmask);
2588 while ((timeout == -1) || (clib_time_now (&ldpw->clib_time) < max_time));
2595 ldp_epoll_pwait_eventfd (int epfd, struct epoll_event *events,
2596 int maxevents, int timeout, const sigset_t * sigmask)
2598 int libc_epfd, rv = 0, num_ev, libc_num_ev, vcl_wups = 0;
2599 struct epoll_event *libc_evts;
2600 ldp_worker_ctx_t *ldpw;
2601 vls_handle_t ep_vlsh;
2605 if (PREDICT_FALSE (!events || (timeout < -1)))
2611 /* Make sure the vcl worker is valid. Could be that epoll fd was created on
2612 * one thread but it is now used on another */
2613 if (PREDICT_FALSE (vppcom_worker_index () == ~0))
2614 vls_register_vcl_worker ();
2616 ldpw = ldp_worker_get_current ();
2617 if (epfd == ldpw->vcl_mq_epfd)
2618 return libc_epoll_pwait (epfd, events, maxevents, timeout, sigmask);
2620 ep_vlsh = ldp_fd_to_vlsh (epfd);
2621 if (PREDICT_FALSE (ep_vlsh == VLS_INVALID_HANDLE))
2623 LDBG (0, "epfd %d: bad ep_vlsh %d!", epfd, ep_vlsh);
2628 libc_epfd = vls_attr (ep_vlsh, VPPCOM_ATTR_GET_LIBC_EPFD, 0, 0);
2629 if (PREDICT_FALSE (!libc_epfd))
2631 u32 size = sizeof (epfd);
2633 LDBG (1, "epfd %d, vep_vlsh %d calling libc_epoll_create1: "
2634 "EPOLL_CLOEXEC", epfd, ep_vlsh);
2635 libc_epfd = libc_epoll_create1 (EPOLL_CLOEXEC);
2642 rv = vls_attr (ep_vlsh, VPPCOM_ATTR_SET_LIBC_EPFD, &libc_epfd, &size);
2650 if (PREDICT_FALSE (libc_epfd <= 0))
2657 if (PREDICT_FALSE (!ldpw->mq_epfd_added))
2659 struct epoll_event e = { 0 };
2661 e.data.fd = ldpw->vcl_mq_epfd;
2662 if (libc_epoll_ctl (libc_epfd, EPOLL_CTL_ADD, ldpw->vcl_mq_epfd, &e) <
2665 LDBG (0, "epfd %d, add libc mq epoll fd %d to libc epoll fd %d",
2666 epfd, ldpw->vcl_mq_epfd, libc_epfd);
2670 ldpw->mq_epfd_added = 1;
2673 /* Request to only drain unhandled to prevent libc_epoll_wait starved */
2674 rv = vls_epoll_wait (ep_vlsh, events, maxevents, -2);
2678 if (rv >= maxevents)
2681 else if (PREDICT_FALSE (rv < 0))
2690 libc_evts = &events[rv];
2692 libc_epoll_pwait (libc_epfd, libc_evts, maxevents - rv, timeout, sigmask);
2693 if (libc_num_ev <= 0)
2695 rv = rv >= 0 ? rv : -1;
2699 for (int i = 0; i < libc_num_ev; i++)
2701 if (libc_evts[i].data.fd == ldpw->vcl_mq_epfd)
2703 /* We should remove mq epoll fd from events. */
2705 if (i != libc_num_ev)
2707 libc_evts[i].events = libc_evts[libc_num_ev].events;
2708 libc_evts[i].data.u64 = libc_evts[libc_num_ev].data.u64;
2710 num_ev = vls_epoll_wait (ep_vlsh, &libc_evts[libc_num_ev],
2711 maxevents - libc_num_ev, 0);
2712 if (PREDICT_TRUE (num_ev > 0))
2714 /* Woken up by vcl but no events generated. Accept it once */
2715 if (rv == 0 && libc_num_ev == 0 && timeout && vcl_wups++ < 1)
2728 epoll_pwait (int epfd, struct epoll_event *events,
2729 int maxevents, int timeout, const sigset_t * sigmask)
2731 if (vls_use_eventfd ())
2732 return ldp_epoll_pwait_eventfd (epfd, events, maxevents, timeout,
2735 return ldp_epoll_pwait (epfd, events, maxevents, timeout, sigmask);
2739 epoll_wait (int epfd, struct epoll_event *events, int maxevents, int timeout)
2741 if (vls_use_eventfd ())
2742 return ldp_epoll_pwait_eventfd (epfd, events, maxevents, timeout, NULL);
2744 return ldp_epoll_pwait (epfd, events, maxevents, timeout, NULL);
2748 poll (struct pollfd *fds, nfds_t nfds, int timeout)
2750 ldp_worker_ctx_t *ldpw = ldp_worker_get_current ();
2751 int rv, i, n_revents = 0;
2756 LDBG (3, "fds %p, nfds %ld, timeout %d", fds, nfds, timeout);
2758 if (PREDICT_FALSE (ldpw->clib_time.init_cpu_time == 0))
2759 clib_time_init (&ldpw->clib_time);
2761 max_time = (timeout >= 0) ? (f64) timeout / 1000 : 0;
2762 max_time += clib_time_now (&ldpw->clib_time);
2764 for (i = 0; i < nfds; i++)
2769 vlsh = ldp_fd_to_vlsh (fds[i].fd);
2770 if (vlsh != VLS_INVALID_HANDLE)
2772 fds[i].fd = -fds[i].fd;
2773 vec_add2 (ldpw->vcl_poll, vp, 1);
2775 vp->sh = vlsh_to_sh (vlsh);
2776 vp->events = fds[i].events;
2777 #ifdef __USE_XOPEN2K
2778 if (fds[i].events & POLLRDNORM)
2779 vp->events |= POLLIN;
2780 if (fds[i].events & POLLWRNORM)
2781 vp->events |= POLLOUT;
2783 vp->revents = fds[i].revents;
2787 vec_add1 (ldpw->libc_poll, fds[i]);
2788 vec_add1 (ldpw->libc_poll_idxs, i);
2794 if (vec_len (ldpw->vcl_poll))
2796 rv = vppcom_poll (ldpw->vcl_poll, vec_len (ldpw->vcl_poll), 0);
2807 if (vec_len (ldpw->libc_poll))
2809 rv = libc_poll (ldpw->libc_poll, vec_len (ldpw->libc_poll), 0);
2822 while ((timeout < 0) || (clib_time_now (&ldpw->clib_time) < max_time));
2826 vec_foreach (vp, ldpw->vcl_poll)
2828 fds[vp->fds_ndx].fd = -fds[vp->fds_ndx].fd;
2829 fds[vp->fds_ndx].revents = vp->revents;
2830 #ifdef __USE_XOPEN2K
2831 if ((fds[vp->fds_ndx].revents & POLLIN) &&
2832 (fds[vp->fds_ndx].events & POLLRDNORM))
2833 fds[vp->fds_ndx].revents |= POLLRDNORM;
2834 if ((fds[vp->fds_ndx].revents & POLLOUT) &&
2835 (fds[vp->fds_ndx].events & POLLWRNORM))
2836 fds[vp->fds_ndx].revents |= POLLWRNORM;
2839 vec_reset_length (ldpw->vcl_poll);
2841 for (i = 0; i < vec_len (ldpw->libc_poll); i++)
2843 fds[ldpw->libc_poll_idxs[i]].revents = ldpw->libc_poll[i].revents;
2845 vec_reset_length (ldpw->libc_poll_idxs);
2846 vec_reset_length (ldpw->libc_poll);
2853 ppoll (struct pollfd *fds, nfds_t nfds,
2854 const struct timespec *timeout, const sigset_t * sigmask)
2858 clib_warning ("LDP<%d>: LDP-TBD", getpid ());
2866 void CONSTRUCTOR_ATTRIBUTE ldp_constructor (void);
2868 void DESTRUCTOR_ATTRIBUTE ldp_destructor (void);
2871 * This function is called when the library is loaded
2874 ldp_constructor (void)
2876 swrap_constructor ();
2877 if (ldp_init () != 0)
2879 fprintf (stderr, "\nLDP<%d>: ERROR: ldp_constructor: failed!\n",
2883 else if (LDP_DEBUG > 0)
2884 clib_warning ("LDP<%d>: LDP constructor: done!\n", getpid ());
2888 * This function is called when the library is unloaded
2891 ldp_destructor (void)
2894 swrap_destructor ();
2899 /* Don't use clib_warning() here because that calls writev()
2900 * which will call ldp_init().
2903 fprintf (stderr, "%s:%d: LDP<%d>: LDP destructor: done!\n",
2904 __func__, __LINE__, getpid ());
2909 * fd.io coding-style-patch-verification: ON
2912 * eval: (c-set-style "gnu")