chiark - git - mdw - mLib/blob - sys/t/fdpass-test.c

   1 /* -*-c-*-
   2  *
   3  * Test the file-descriptor passing code
   4  *
   5  * (c) 2018 Straylight/Edgeware
   6  */
   7
   8 /*----- Licensing notice --------------------------------------------------*
   9  *
  10  * This file is part of the mLib utilities library.
  11  *
  12  * mLib is free software: you can redistribute it and/or modify it under
  13  * the terms of the GNU Library General Public License as published by
  14  * the Free Software Foundation; either version 2 of the License, or (at
  15  * your option) any later version.
  16  *
  17  * mLib is distributed in the hope that it will be useful, but WITHOUT
  18  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  19  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Library General Public
  20  * License for more details.
  21  *
  22  * You should have received a copy of the GNU Library General Public
  23  * License along with mLib.  If not, write to the Free Software
  24  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307,
  25  * USA.
  26  */
  27
  28 /*----- Header files ------------------------------------------------------*/
  29
  30 #include <errno.h>
  31 #include <stdio.h>
  32 #include <stdlib.h>
  33 #include <string.h>
  34
  35 #include <sys/socket.h>
  36 #include <sys/un.h>
  37
  38 #include "fdpass.h"
  39 #include "quis.h"
  40 #include "report.h"
  41
  42 #undef sun
  43
  44 /*----- Main code ---------------------------------------------------------*/
  45
  46 static void set_unix_addr(struct sockaddr_un *sun, const char *path)
  47 {
  48   if (strlen(path) >= sizeof(sun->sun_path))
  49     die(2, "pathname `%s' too long", path);
  50   sun->sun_family = AF_UNIX;
  51   strcpy(sun->sun_path, path);
  52 }
  53
  54 const char sockmsg[] = "socket message", pipemsg[] = "pipe message";
  55
  56 int main(int argc, char *argv[])
  57 {
  58   struct sockaddr_un sun;
  59   char buf[16];
  60   int lsk, sk, fd, pfd[2];
  61   socklen_t len;
  62   ssize_t n;
  63
  64   ego(argv[0]);
  65
  66   if (argc == 3 && strcmp(argv[1], "server") == 0) {
  67     set_unix_addr(&sun, argv[2]);
  68     lsk = socket(PF_UNIX, SOCK_STREAM, 0);
  69     if (lsk < 0) die(2, "socket: %s", strerror(errno));
  70     if (bind(lsk, (struct sockaddr *)&sun, sizeof(sun)))
  71       die(2, "bind: %s", strerror(errno));
  72     if (listen(lsk, 5)) die(2, "listen: %s", strerror(errno));
  73     len = sizeof(sun); sk = accept(lsk, (struct sockaddr *)&sun, &len);
  74     if (sk < 0) die(2, "accept: %s", strerror(errno));
  75     close(lsk);
  76     n = fdpass_recv(sk, &fd, buf, sizeof(buf));
  77     if (n < 0) die(2, "fdrecv: %s", strerror(errno));
  78     close(sk);
  79     if (fd == -1) die(2, "no fd found");
  80     if (n != sizeof(sockmsg) || strcmp(buf, sockmsg) != 0)
  81       die(2, "socket message mismatch (found `%.*s')", (int)n, buf);
  82     n = read(fd, buf, sizeof(buf));
  83     if (n < 0) die(2, "read: %s", strerror(errno));
  84     if (n != sizeof(pipemsg) || strcmp(buf, pipemsg) != 0)
  85       die(2, "pipe message mismatch (found `%.*s')", (int)n, buf);
  86     close(fd);
  87   } else if (argc == 3 && strcmp(argv[1], "client") == 0) {
  88     set_unix_addr(&sun, argv[2]);
  89     if (pipe(pfd)) die(2, "pipe: %s", strerror(errno));
  90     sk = socket(PF_UNIX, SOCK_STREAM, 0);
  91     if (sk < 0) die(2, "socket: %s", strerror(errno));
  92     if (connect(sk, (struct sockaddr *)&sun, sizeof(sun)))
  93       die(2, "connect: %s", strerror(errno));
  94     if (fdpass_send(sk, pfd[0], sockmsg, sizeof(sockmsg)) < 0)
  95       die(2, "fdsend: %s", strerror(errno));
  96     close(pfd[0]);
  97     close(sk);
  98     if (write(pfd[1], pipemsg, sizeof(pipemsg)) < 0)
  99       die(2, "write: %s", strerror(errno));
 100     close(pfd[1]);
 101   } else {
 102     pquis(stderr, "usage: $ client|server SOCK\n");
 103     exit(2);
 104   }
 105   return (0);
 106 }
 107
 108 /*----- That's all, folks -------------------------------------------------*/