1 /*	$NetBSD: t_symlinkat.c,v 1.2 2013/03/17 04:46:06 jmmv Exp $ */
2 
3 /*-
4  * Copyright (c) 2012 The NetBSD Foundation, Inc.
5  * All rights reserved.
6  *
7  * This code is derived from software contributed to The NetBSD Foundation
8  * by Emmanuel Dreyfus.
9  *
10  * Redistribution and use in source and binary forms, with or without
11  * modification, are permitted provided that the following conditions
12  * are met:
13  * 1. Redistributions of source code must retain the above copyright
14  *    notice, this list of conditions and the following disclaimer.
15  * 2. Redistributions in binary form must reproduce the above copyright
16  *    notice, this list of conditions and the following disclaimer in the
17  *    documentation and/or other materials provided with the distribution.
18  *
19  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29  * POSSIBILITY OF SUCH DAMAGE.
30  */
31 #include <sys/cdefs.h>
32 __RCSID("$NetBSD: t_symlinkat.c,v 1.2 2013/03/17 04:46:06 jmmv Exp $");
33 
34 #include <atf-c.h>
35 #include <errno.h>
36 #include <fcntl.h>
37 #include <limits.h>
38 #include <paths.h>
39 #include <stdio.h>
40 #include <string.h>
41 #include <unistd.h>
42 #include <sys/param.h>
43 #include <sys/stat.h>
44 
45 #define ODIR "olddir"
46 #define NDIR "newdir"
47 #define FILE "olddir/old"
48 #define BASEFILE "old"
49 #define RELFILE "../olddir/old"
50 #define LINK "newdir/symlink"
51 #define BASELINK "symlink"
52 #define FILEERR "olddir/olderr"
53 
54 ATF_TC(symlinkat_fd);
55 ATF_TC_HEAD(symlinkat_fd, tc)
56 {
57 	atf_tc_set_md_var(tc, "descr", "See that symlinkat works with fd");
58 }
59 ATF_TC_BODY(symlinkat_fd, tc)
60 {
61 	int dfd, fd;
62 	struct stat ost, nst;
63 
64 	ATF_REQUIRE(mkdir(ODIR, 0755) == 0);
65 	ATF_REQUIRE(mkdir(NDIR, 0755) == 0);
66 	ATF_REQUIRE((fd = open(FILE, O_CREAT|O_RDWR, 0644)) != -1);
67 	ATF_REQUIRE(close(fd) != -1);
68 
69 	ATF_REQUIRE((dfd = open(NDIR, O_RDONLY, 0)) != -1);
70 	ATF_REQUIRE(symlinkat(RELFILE, dfd, BASELINK) == 0);
71 	ATF_REQUIRE(close(dfd) == 0);
72 
73 	ATF_REQUIRE(stat(FILE, &ost) == 0);
74 	ATF_REQUIRE(stat(LINK, &nst) == 0);
75 	ATF_REQUIRE(ost.st_ino == nst.st_ino);
76 }
77 
78 ATF_TC(symlinkat_fdcwd);
79 ATF_TC_HEAD(symlinkat_fdcwd, tc)
80 {
81 	atf_tc_set_md_var(tc, "descr",
82 			  "See that symlinkat works with fd as AT_FDCWD");
83 }
84 ATF_TC_BODY(symlinkat_fdcwd, tc)
85 {
86 	int fd;
87 	struct stat ost, nst;
88 
89 	ATF_REQUIRE(mkdir(ODIR, 0755) == 0);
90 	ATF_REQUIRE(mkdir(NDIR, 0755) == 0);
91 	ATF_REQUIRE((fd = open(FILE, O_CREAT|O_RDWR, 0644)) != -1);
92 	ATF_REQUIRE(close(fd) != -1);
93 
94 	ATF_REQUIRE(symlinkat(RELFILE, AT_FDCWD, LINK) == 0);
95 
96 	ATF_REQUIRE(stat(FILE, &ost) == 0);
97 	ATF_REQUIRE(stat(LINK, &nst) == 0);
98 	ATF_REQUIRE(ost.st_ino == nst.st_ino);
99 }
100 
101 ATF_TC(symlinkat_fdcwderr);
102 ATF_TC_HEAD(symlinkat_fdcwderr, tc)
103 {
104 	atf_tc_set_md_var(tc, "descr",
105 		  "See that symlinkat works with fd as AT_FDCWD and bad path");
106 }
107 ATF_TC_BODY(symlinkat_fdcwderr, tc)
108 {
109 	int fd;
110 	struct stat st;
111 
112 	ATF_REQUIRE(mkdir(ODIR, 0755) == 0);
113 	ATF_REQUIRE(mkdir(NDIR, 0755) == 0);
114 	ATF_REQUIRE((fd = open(FILE, O_CREAT|O_RDWR, 0644)) != -1);
115 	ATF_REQUIRE(close(fd) != -1);
116 
117 	ATF_REQUIRE(symlinkat(FILEERR, AT_FDCWD, LINK) == 0);
118 	ATF_REQUIRE(lstat(LINK, &st) == 0);
119 	ATF_REQUIRE(stat(LINK, &st) == -1);
120 	ATF_REQUIRE(errno == ENOENT);
121 
122 }
123 
124 ATF_TC(symlinkat_fderr);
125 ATF_TC_HEAD(symlinkat_fderr, tc)
126 {
127 	atf_tc_set_md_var(tc, "descr", "See that symlinkat fails with fd as -1");
128 }
129 ATF_TC_BODY(symlinkat_fderr, tc)
130 {
131 	int fd;
132 
133 	ATF_REQUIRE(mkdir(ODIR, 0755) == 0);
134 	ATF_REQUIRE(mkdir(NDIR, 0755) == 0);
135 	ATF_REQUIRE((fd = open(FILE, O_CREAT|O_RDWR, 0644)) != -1);
136 	ATF_REQUIRE(close(fd) != -1);
137 
138 	ATF_REQUIRE(symlinkat(RELFILE, -1, LINK) == -1);
139 }
140 
141 ATF_TP_ADD_TCS(tp)
142 {
143 
144 	ATF_TP_ADD_TC(tp, symlinkat_fd);
145 	ATF_TP_ADD_TC(tp, symlinkat_fdcwd);
146 	ATF_TP_ADD_TC(tp, symlinkat_fdcwderr);
147 	ATF_TP_ADD_TC(tp, symlinkat_fderr);
148 
149 	return atf_no_error();
150 }
151