t_dir.c revision 1.3 
1/* $NetBSD: t_dir.c,v 1.3 2011/06/11 18:03:18 christos Exp $ */
2
3/*-
4 * Copyright (c) 2010 The NetBSD Foundation, Inc.
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 *    notice, this list of conditions and the following disclaimer in the
14 *    documentation and/or other materials provided with the distribution.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
17 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
18 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
19 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
20 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
26 * POSSIBILITY OF SUCH DAMAGE.
27 */
28
29#include <atf-c.h>
30
31#include <assert.h>
32#include <dirent.h>
33#include <err.h>
34#include <fcntl.h>
35#include <stdio.h>
36#include <stdlib.h>
37#include <string.h>
38#include <unistd.h>
39
40#include <sys/stat.h>
41
42ATF_TC(seekdir);
43
44ATF_TC_HEAD(seekdir, tc)
45{
46
47	atf_tc_set_md_var(tc, "descr",
48	    "Check telldir(3) and seekdir(3) for correct behavior (PR/24324)");
49}
50
51ATF_TC_BODY(seekdir, tc)
52{
53	DIR *dp;
54	char *wasname;
55	struct dirent *entry;
56	long here;
57
58	mkdir("t", 0755);
59	creat("t/a", 0600);
60	creat("t/b", 0600);
61	creat("t/c", 0600);
62
63	dp = opendir("t");
64	if ( dp == NULL)
65		atf_tc_fail("Could not open temp directory.");
66
67	/* skip two for . and .. */
68	entry = readdir(dp);
69	entry = readdir(dp);
70
71	/* get first entry */
72	entry = readdir(dp);
73	here = telldir(dp);
74
75	/* get second entry */
76	entry = readdir(dp);
77	wasname = strdup(entry->d_name);
78	if (wasname == NULL)
79		atf_tc_fail("cannot allocate memory");
80
81	/* get third entry */
82	entry = readdir(dp);
83
84	/* try to return to the position after the first entry */
85	seekdir(dp, here);
86	entry = readdir(dp);
87
88	if (entry == NULL)
89		atf_tc_fail("entry 1 not found");
90	if (strcmp(entry->d_name, wasname) != 0)
91		atf_tc_fail("1st seekdir found wrong name");
92
93	/* try again, and throw in a telldir() for good measure */
94	seekdir(dp, here);
95	here = telldir(dp);
96	entry = readdir(dp);
97
98	if (entry == NULL)
99		atf_tc_fail("entry 2 not found");
100	if (strcmp(entry->d_name, wasname) != 0)
101		atf_tc_fail("2nd seekdir found wrong name");
102
103	/* One more time, to make sure that telldir() doesn't affect result */
104	seekdir(dp, here);
105	entry = readdir(dp);
106
107	if (entry == NULL)
108		atf_tc_fail("entry 3 not found");
109	if (strcmp(entry->d_name, wasname) != 0)
110		atf_tc_fail("3rd seekdir found wrong name");
111
112	closedir(dp);
113}
114
115ATF_TC(telldir_leak);
116
117ATF_TC_HEAD(telldir_leak, tc)
118{
119
120	atf_tc_set_md_var(tc, "descr",
121	    "Check telldir(3) for memory leakage (PR/24324)");
122}
123
124ATF_TC_BODY(telldir_leak, tc)
125{
126	DIR *dp;
127	long loc;
128	char *memused;
129	int i;
130	int oktouse = 4096;
131
132	dp = opendir(".");
133	if (dp == NULL)
134		atf_tc_fail("Could not open current directory");
135
136	loc = telldir(dp);
137	memused = sbrk(0);
138	closedir(dp);
139
140	for (i = 0; i < 1000; i++) {
141		dp = opendir(".");
142		if (dp == NULL)
143			atf_tc_fail("Could not open current directory");
144
145		loc = telldir(dp);
146		closedir(dp);
147
148		if ((char *)sbrk(0) - memused > oktouse) {
149			(void)printf("Used %td extra bytes for %d telldir "
150			    "calls", ((char *)sbrk(0) - memused), i);
151			oktouse = (char *)sbrk(0) - memused;
152		}
153	}
154	if (oktouse > 4096) {
155		atf_tc_fail("Failure: leaked %d bytes", oktouse);
156	} else {
157		(void)printf("OK: used %td bytes\n", (char *)(sbrk(0))-memused);
158	}
159}
160
161ATF_TP_ADD_TCS(tp)
162{
163
164	ATF_TP_ADD_TC(tp, seekdir);
165	ATF_TP_ADD_TC(tp, telldir_leak);
166
167	return atf_no_error();
168}
169