Home | History | Annotate | Download | only in tests
      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 #include <asm/unistd.h>
     30 
     31 #ifdef __NR_getdents64
     32 
     33 # include <assert.h>
     34 # include <dirent.h>
     35 # include <fcntl.h>
     36 # include <inttypes.h>
     37 # include <stddef.h>
     38 # include <stdio.h>
     39 # include <sys/stat.h>
     40 # include <unistd.h>
     41 
     42 static const char fname[] =
     43 	"A\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\n"
     44 	"A\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\n"
     45 	"A\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\n"
     46 	"A\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\n"
     47 	"A\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\n"
     48 	"A\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\n"
     49 	"A\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\n"
     50 	"A\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nA\nZ";
     51 static const char qname[] =
     52 	"A\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\n"
     53 	"A\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\n"
     54 	"A\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\n"
     55 	"A\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\n"
     56 	"A\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\n"
     57 	"A\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\n"
     58 	"A\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\n"
     59 	"A\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nA\\nZ";
     60 
     61 typedef struct {
     62 		uint64_t d_ino;
     63 		uint64_t d_off;
     64 		unsigned short d_reclen;
     65 		unsigned char d_type;
     66 		char d_name[256];
     67 } kernel_dirent64;
     68 
     69 static char buf[8192];
     70 
     71 static const char *
     72 str_d_type(const unsigned char d_type)
     73 {
     74 	switch (d_type) {
     75 		case DT_DIR:
     76 			return "DT_DIR";
     77 		case DT_REG:
     78 			return "DT_REG";
     79 		default:
     80 			return "DT_UNKNOWN";
     81 	}
     82 }
     83 static void
     84 print_dirent(const kernel_dirent64 *d)
     85 {
     86 	const unsigned int d_name_offset = offsetof(kernel_dirent64, d_name);
     87 	int d_name_len = d->d_reclen - d_name_offset;
     88 	assert(d_name_len > 0);
     89 
     90 	printf("{d_ino=%" PRIu64 ", d_off=%" PRId64
     91 	       ", d_reclen=%u, d_type=%s, d_name=",
     92 	       d->d_ino, d->d_off, d->d_reclen, str_d_type(d->d_type));
     93 
     94 	if (d->d_name[0] == '.')
     95 		printf("\"%.*s\"}", d_name_len, d->d_name);
     96 	else
     97 		printf("\"%s\"}", qname);
     98 }
     99 
    100 int
    101 main(int ac, const char **av)
    102 {
    103 	char *dname;
    104 
    105 	assert(ac == 1);
    106 	assert(asprintf(&dname, "%s.test.tmp.dir", av[0]) > 0);
    107 	assert(!mkdir(dname, 0700));
    108 	assert(!chdir(dname));
    109 	(void) close(0);
    110 	assert(!creat(fname, 0600));
    111 	assert(!close(0));
    112 	assert(!open(".", O_RDONLY | O_DIRECTORY));
    113 
    114 	unsigned long count = (unsigned long) 0xfacefeeddeadbeefULL;
    115 	long rc = syscall(__NR_getdents64, (long) 0xdefacedffffffffULL, NULL,
    116 			  count);
    117 	printf("getdents64(-1, NULL, %u) = %ld %s (%m)\n",
    118 	       (unsigned) count, rc, errno2name());
    119 
    120 	count = (unsigned long) 0xfacefeed00000000ULL | sizeof(buf);
    121 	while ((rc = syscall(__NR_getdents64, 0, buf, count))) {
    122 		kernel_dirent64 *d;
    123 		long i;
    124 
    125 		if (rc < 0)
    126 			perror_msg_and_skip("getdents64");
    127 		printf("getdents64(0, [");
    128 		for (i = 0; i < rc; i += d->d_reclen) {
    129 			d = (kernel_dirent64 *) &buf[i];
    130 			if (i)
    131 				printf(", ");
    132 			print_dirent(d);
    133 		}
    134 		printf("], %u) = %ld\n", (unsigned) count, rc);
    135 	}
    136 	printf("getdents64(0, [], %u) = 0\n", (unsigned) count);
    137 	puts("+++ exited with 0 +++");
    138 	assert(!unlink(fname));
    139 	assert(!chdir(".."));
    140 	assert(!rmdir(dname));
    141 
    142 	return 0;
    143 }
    144 
    145 #else
    146 
    147 SKIP_MAIN_UNDEFINED("__NR_getdents64")
    148 
    149 #endif
    150