io/tst-mknodat.c

   1 #include <dirent.h>
   2 #include <fcntl.h>
   3 #include <stdio.h>
   4 #include <stdlib.h>
   5 #include <string.h>
   6 #include <unistd.h>
   7
   8
   9 static void prepare (void);
  10 #define PREPARE(argc, argv) prepare ()
  11
  12 static int do_test (void);
  13 #define TEST_FUNCTION do_test ()
  14
  15 #include "../test-skeleton.c"
  16
  17 static int dir_fd;
  18
  19 static void
  20 prepare (void)
  21 {
  22   size_t test_dir_len = strlen (test_dir);
  23   static const char dir_name[] = "/tst-mknodat.XXXXXX";
  24
  25   size_t dirbuflen = test_dir_len + sizeof (dir_name);
  26   char *dirbuf = malloc (dirbuflen);
  27   if (dirbuf == NULL)
  28     {
  29       puts ("out of memory");
  30       exit (1);
  31     }
  32
  33   snprintf (dirbuf, dirbuflen, "%s%s", test_dir, dir_name);
  34   if (mkdtemp (dirbuf) == NULL)
  35     {
  36       puts ("cannot create temporary directory");
  37       exit (1);
  38     }
  39
  40   add_temp_file (dirbuf);
  41
  42   dir_fd = open (dirbuf, O_RDONLY | O_DIRECTORY);
  43   if (dir_fd == -1)
  44     {
  45       puts ("cannot open directory");
  46       exit (1);
  47     }
  48 }
  49
  50
  51 static int
  52 do_test (void)
  53 {
  54   /* fdopendir takes over the descriptor, make a copy.  */
  55   int dupfd = dup (dir_fd);
  56   if (dupfd == -1)
  57     {
  58       puts ("dup failed");
  59       return 1;
  60     }
  61   if (lseek (dupfd, 0, SEEK_SET) != 0)
  62     {
  63       puts ("1st lseek failed");
  64       return 1;
  65     }
  66
  67   /* The directory should be empty safe the . and .. files.  */
  68   DIR *dir = fdopendir (dupfd);
  69   if (dir == NULL)
  70     {
  71       puts ("fdopendir failed");
  72       return 1;
  73     }
  74   struct dirent64 *d;
  75   while ((d = readdir64 (dir)) != NULL)
  76     if (strcmp (d->d_name, ".") != 0 && strcmp (d->d_name, "..") != 0)
  77       {
  78         printf ("temp directory contains file \"%s\"\n", d->d_name);
  79         return 1;
  80       }
  81   closedir (dir);
  82
  83   /* Create a new directory.  */
  84   int e = mknodat (dir_fd, "some-sock", 0777 | S_IFSOCK, 0);
  85   if (e == -1)
  86     {
  87       if (errno == ENOSYS)
  88         {
  89           puts ("*at functions not supported");
  90           return 0;
  91         }
  92
  93       puts ("socket creation failed");
  94       return 1;
  95     }
  96
  97   struct stat64 st1;
  98   if (fstatat64 (dir_fd, "some-sock", &st1, 0) != 0)
  99     {
 100       puts ("fstat64 failed");
 101       return 1;
 102     }
 103   if (!S_ISSOCK (st1.st_mode))
 104     {
 105       puts ("mknodat did not create a Unix domain socket");
 106       return 1;
 107     }
 108
 109   dupfd = dup (dir_fd);
 110   if (dupfd == -1)
 111     {
 112       puts ("dup failed");
 113       return 1;
 114     }
 115   if (lseek (dupfd, 0, SEEK_SET) != 0)
 116     {
 117       puts ("1st lseek failed");
 118       return 1;
 119     }
 120
 121   dir = fdopendir (dupfd);
 122   if (dir == NULL)
 123     {
 124       puts ("2nd fdopendir failed");
 125       return 1;
 126     }
 127   bool has_some_sock = false;
 128   while ((d = readdir64 (dir)) != NULL)
 129     if (strcmp (d->d_name, "some-sock") == 0)
 130       {
 131         has_some_sock = true;
 132 #ifdef _DIRENT_HAVE_D_TYPE
 133         if (d->d_type != DT_UNKNOWN && d->d_type != DT_SOCK)
 134           {
 135             puts ("d_type for some-sock wrong");
 136             return 1;
 137           }
 138 #endif
 139       }
 140     else if (strcmp (d->d_name, ".") != 0 && strcmp (d->d_name, "..") != 0)
 141       {
 142         printf ("temp directory contains file \"%s\"\n", d->d_name);
 143         return 1;
 144       }
 145   closedir (dir);
 146
 147   if (!has_some_sock)
 148     {
 149       puts ("some-sock not in directory list");
 150       return 1;
 151     }
 152
 153   if (unlinkat (dir_fd, "some-sock", 0) != 0)
 154     {
 155       puts ("unlinkat failed");
 156       return 1;
 157     }
 158
 159   close (dir_fd);
 160
 161   return 0;
 162 }