1 /* 2 * Copyright 2016 Jeremy Allison 3 * 4 * Permission is hereby granted, free of charge, to any person obtaining a 5 * copy of this software and associated documentation files (the "Software"), 6 * to deal in the Software without restriction, including without limitation 7 * the rights to use, copy, modify, merge, publish, distribute, sublicense, 8 * and/or sell copies of the Software, and to permit persons to whom the 9 * Software is furnished to do so, subject to the following conditions: 10 * 11 * The above copyright notice and this permission notice shall be included 12 * in all copies or substantial portions of the Software. 13 * 14 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 15 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 16 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 17 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 18 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING 19 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER 20 * DEALINGS IN THE SOFTWARE. 21 */ 22 23 /* 24 * Copyright 2019 Joyent, Inc. 25 */ 26 27 #include <sys/param.h> 28 #include <sys/types.h> 29 #include <sys/stat.h> 30 #include <sys/socket.h> 31 #include <sys/un.h> 32 #include <stdio.h> 33 #include <unistd.h> 34 #include <string.h> 35 #include <errno.h> 36 #include <stdint.h> 37 #include <stdlib.h> 38 #include <pthread.h> 39 40 extern const char *__progname; 41 42 static void * 43 server(void *varg) 44 { 45 int ret; 46 int sock = (int)varg; 47 unsigned int i; 48 49 for (i = 0; i < 5; i++) { 50 struct iovec iov; 51 struct msghdr msg; 52 uint8_t buf[4096]; 53 54 iov = (struct iovec) { 55 .iov_base = buf, 56 .iov_len = sizeof (buf) 57 }; 58 59 msg = (struct msghdr) { 60 .msg_iov = &iov, 61 .msg_iovlen = 1, 62 }; 63 64 ret = recvmsg(sock, &msg, 0); 65 if (ret == -1) { 66 fprintf(stderr, "server - recvmsg fail %s\n", 67 strerror(errno)); 68 exit(1); 69 } 70 71 printf("SERVER:%s\n", (char *)msg.msg_iov->iov_base); 72 fflush(stdout); 73 } 74 75 exit(0); 76 } 77 78 static void * 79 listener(void *varg) 80 { 81 struct sockaddr_un *addr = varg; 82 int ret; 83 int lsock; 84 int asock; 85 86 /* Child. */ 87 lsock = socket(AF_UNIX, SOCK_STREAM, 0); 88 if (lsock == -1) { 89 fprintf(stderr, "listener - socket fail %s\n", strerror(errno)); 90 exit(1); 91 } 92 93 ret = bind(lsock, (struct sockaddr *)addr, sizeof (*addr)); 94 if (ret == -1) { 95 fprintf(stderr, "listener - bind fail %s\n", strerror(errno)); 96 exit(1); 97 } 98 99 ret = listen(lsock, 5); 100 if (ret == -1) { 101 fprintf(stderr, "listener - listen fail %s\n", strerror(errno)); 102 exit(1); 103 } 104 105 for (;;) { 106 asock = accept(lsock, NULL, 0); 107 if (asock == -1) { 108 fprintf(stderr, "listener - accept fail %s\n", 109 strerror(errno)); 110 exit(1); 111 } 112 113 /* start worker */ 114 ret = pthread_create(NULL, NULL, server, (void *)asock); 115 if (ret == -1) { 116 fprintf(stderr, "%s - thread create fail %s\n", 117 __progname, strerror(errno)); 118 exit(1); 119 } 120 } 121 } 122 123 /* 124 * This should be a place only root is allowed to write. 125 * The test will create and destroy this directory. 126 */ 127 char testdir[100] = "/var/run/os-tests-sockfs"; 128 struct sockaddr_un addr; 129 int test_uid = UID_NOBODY; 130 131 int 132 main(int argc, char **argv) 133 { 134 int ret; 135 int sock; 136 unsigned int i; 137 138 if (argc > 1) { 139 ret = strlcpy(testdir, argv[1], sizeof (testdir)); 140 if (ret >= sizeof (testdir)) { 141 fprintf(stderr, "%s: too long\n", argv[1]); 142 exit(1); 143 } 144 } 145 146 addr.sun_family = AF_UNIX; 147 (void) sprintf(addr.sun_path, "%s/s", testdir); 148 149 if (mkdir(testdir, 0700) != 0) { 150 switch (errno) { 151 case EEXIST: 152 case EISDIR: 153 break; 154 default: 155 perror(testdir); 156 exit(1); 157 } 158 } 159 (void) unlink(addr.sun_path); 160 161 /* Set up the listener. */ 162 ret = pthread_create(NULL, NULL, listener, &addr); 163 if (ret == -1) { 164 fprintf(stderr, "%s - thread create fail %s\n", 165 argv[0], strerror(errno)); 166 exit(1); 167 } 168 169 sleep(1); 170 171 /* Create and connect the socket endpoint. */ 172 173 sock = socket(AF_UNIX, SOCK_STREAM, 0); 174 if (sock == -1) { 175 fprintf(stderr, "%s - socket fail %s\n", 176 argv[0], strerror(errno)); 177 exit(1); 178 } 179 180 ret = connect(sock, (struct sockaddr *)&addr, sizeof (addr)); 181 if (ret == -1) { 182 fprintf(stderr, "%s - connect fail %s\n", 183 argv[0], strerror(errno)); 184 exit(1); 185 } 186 187 /* Send some messages */ 188 for (i = 0; i < 5; i++) { 189 struct iovec iov; 190 struct msghdr msg; 191 uint8_t buf[4096]; 192 193 memcpy(buf, "TEST0", sizeof ("TEST0")); 194 buf[4] = '0' + i; 195 196 printf("CLIENT:%s\n", buf); 197 198 iov = (struct iovec) { 199 .iov_base = buf, 200 .iov_len = sizeof (buf), 201 }; 202 203 msg = (struct msghdr) { 204 .msg_iov = &iov, 205 .msg_iovlen = 1, 206 }; 207 208 ret = sendmsg(sock, &msg, 0); 209 210 if (ret == -1) { 211 fprintf(stderr, "%s - sendmsg fail %s\n", 212 argv[0], strerror(errno)); 213 exit(1); 214 } 215 216 fflush(stdout); 217 sleep(1); 218 } 219 220 close(sock); 221 return (0); 222 } 223