/* * Copyright (c) 2017-2018 Cisco and/or its affiliates. * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at: * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ #ifndef __sock_test_h__ #define __sock_test_h__ #include <netdb.h> #include <errno.h> #include <stdlib.h> #include <string.h> #include <vcl/vcl_test.h> #define SOCK_TEST_AF_UNIX_FILENAME "/tmp/ldp_server_af_unix_socket" #define SOCK_TEST_MIXED_EPOLL_DATA "Hello, world! (over an AF_UNIX socket)" #define SOCK_TEST_AF_UNIX_ACCEPT_DATA 0xaf0000af #define SOCK_TEST_AF_UNIX_FD_MASK 0x00af0000 #define SOCK_TEST_BANNER_STRING \ "============================================\n" static inline int sock_test_read (int fd, uint8_t *buf, uint32_t nbytes, vcl_test_stats_t *stats) { int rx_bytes, errno_val; do { if (stats) stats->rx_xacts++; rx_bytes = read (fd, buf, nbytes); if (stats) { if ((rx_bytes == 0) || ((rx_bytes < 0) && ((errno == EAGAIN) || (errno == EWOULDBLOCK)))) stats->rx_eagain++; else if (rx_bytes < nbytes) stats->rx_incomp++; } } while ((rx_bytes == 0) || ((rx_bytes < 0) && ((errno == EAGAIN) || (errno == EWOULDBLOCK)))); if (rx_bytes < 0) { errno_val = errno; perror ("ERROR in sock_test_read()"); fprintf (stderr, "SOCK_TEST: ERROR: socket read " "failed (errno = %d)!\n", errno_val); errno = errno_val; } else if (stats) stats->rx_bytes += rx_bytes; return (rx_bytes); } static inline int sock_test_write (int fd, uint8_t *buf, uint32_t nbytes, vcl_test_stats_t *stats, uint32_t verbose) { int tx_bytes = 0; int nbytes_left = nbytes; int rv, errno_val; do { if (stats) stats->tx_xacts++; rv = write (fd, buf, nbytes_left); if (rv < 0) { if ((errno == EAGAIN) || (errno == EWOULDBLOCK)) { if (stats) stats->tx_eagain++; continue; } else break; } tx_bytes += rv; if (tx_bytes != nbytes) { nbytes_left = nbytes_left - rv; if (stats) stats->tx_incomp++; if (verbose) { printf ("SOCK_TEST: WARNING: bytes written (%d) " "!= bytes to write (%d)!\n", tx_bytes, nbytes); } } } while (tx_bytes != nbytes); if (tx_bytes < 0) { errno_val = errno; perror ("ERROR in sock_test_write()"); fprintf (stderr, "SOCK_TEST: ERROR: socket write failed " "(errno = %d)!\n", errno_val); } else if (stats) stats->tx_bytes += tx_bytes; return (tx_bytes); } #endif /* __sock_test_h__ */