1 /* 2 * Copyright (c) 2015-2016 Dmitry V. Levin <ldv (at) altlinux.org> 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 3. The name of the author may not be used to endorse or promote products 14 * derived from this software without specific prior written permission. 15 * 16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 17 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 18 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 19 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 20 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 21 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 22 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 23 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 24 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 25 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 26 */ 27 28 #include "tests.h" 29 30 #include <assert.h> 31 #include <errno.h> 32 #include <string.h> 33 #include <unistd.h> 34 #include <sys/socket.h> 35 #include <asm/unistd.h> 36 37 #ifndef __NR_send 38 # define __NR_send -1 39 #endif 40 #define SC_send 9 41 42 #ifndef __NR_recv 43 # define __NR_recv -1 44 #endif 45 #define SC_recv 10 46 47 static int 48 sys_send(int sockfd, const void *buf, size_t len, int flags) 49 { 50 int rc = socketcall(__NR_send, SC_send, 51 sockfd, (long) buf, len, flags, 0); 52 if (rc < 0 && ENOSYS == errno) 53 perror_msg_and_skip("send"); 54 return rc; 55 } 56 57 static int 58 sys_recv(int sockfd, const void *buf, size_t len, int flags) 59 { 60 int rc = socketcall(__NR_recv, SC_recv, 61 sockfd, (long) buf, len, flags, 0); 62 if (rc < 0 && ENOSYS == errno) 63 perror_msg_and_skip("recv"); 64 return rc; 65 } 66 67 static void 68 transpose(char *str, const size_t len) 69 { 70 size_t i; 71 72 for (i = 0; i < len / 2; ++i) { 73 char c = str[i]; 74 str[i] = str[len - 1 - i]; 75 str[len - 1 - i] = c; 76 } 77 } 78 79 int 80 main(int ac, char **av) 81 { 82 assert(ac == 2); 83 const size_t len = strlen(av[1]); 84 assert(len); 85 char *const buf0 = tail_alloc(len); 86 char *const buf1 = tail_alloc(len); 87 memcpy(buf0, av[1], len); 88 89 (void) close(0); 90 (void) close(1); 91 92 int sv[2]; 93 if (socketpair(AF_UNIX, SOCK_STREAM, 0, sv)) 94 perror_msg_and_skip("socketpair"); 95 96 assert(sys_send(0, buf0, len, MSG_DONTROUTE) == (int) len); 97 assert(sys_recv(1, buf1, len, MSG_WAITALL) == (int) len); 98 99 transpose(buf1, len); 100 assert(sys_send(1, buf1, len, MSG_DONTROUTE) == (int) len); 101 if (close(1)) 102 perror_msg_and_fail("close(1)"); 103 104 assert(sys_recv(0, buf0, len, MSG_WAITALL) == (int) len); 105 if (close(0)) 106 perror_msg_and_fail("close(0)"); 107 assert(sys_recv(0, NULL, len, MSG_DONTWAIT) == -1); 108 109 return 0; 110 } 111
