1 /*
2 * Copyright (C) 2014 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 #include <ftw.h>
18
19 #include <stdio.h>
20 #include <stdlib.h>
21 #include <sys/stat.h>
22 #include <sys/types.h>
23 #include <unistd.h>
24
25 #include "TemporaryFile.h"
26
27 #include <gtest/gtest.h>
28
MakeTree(const char * root)29 static void MakeTree(const char* root) {
30 char path[PATH_MAX];
31
32 snprintf(path, sizeof(path), "%s/dir", root);
33 ASSERT_EQ(0, mkdir(path, 0755)) << path;
34 snprintf(path, sizeof(path), "%s/dir/sub", root);
35 ASSERT_EQ(0, mkdir(path, 0555)) << path;
36 snprintf(path, sizeof(path), "%s/unreadable-dir", root);
37 ASSERT_EQ(0, mkdir(path, 0000)) << path;
38
39 snprintf(path, sizeof(path), "%s/dangler", root);
40 ASSERT_EQ(0, symlink("/does-not-exist", path));
41 snprintf(path, sizeof(path), "%s/symlink", root);
42 ASSERT_EQ(0, symlink("sub2", path));
43
44 int fd;
45 snprintf(path, sizeof(path), "%s/regular", root);
46 ASSERT_NE(-1, fd = open(path, O_CREAT|O_TRUNC, 0666));
47 ASSERT_EQ(0, close(fd));
48 }
49
sanity_check_ftw(const char * fpath,const struct stat * sb,int tflag)50 void sanity_check_ftw(const char* fpath, const struct stat* sb, int tflag) {
51 ASSERT_TRUE(fpath != NULL);
52 ASSERT_TRUE(sb != NULL);
53
54 if (S_ISDIR(sb->st_mode)) {
55 EXPECT_TRUE(tflag == FTW_D || tflag == FTW_DNR || tflag == FTW_DP) << fpath;
56 } else if (S_ISLNK(sb->st_mode)) {
57 EXPECT_EQ(FTW_SL, tflag) << fpath;
58 } else {
59 EXPECT_EQ(FTW_F, tflag) << fpath;
60 }
61 }
62
sanity_check_nftw(const char * fpath,const struct stat * sb,int tflag,struct FTW * ftwbuf)63 void sanity_check_nftw(const char* fpath, const struct stat* sb, int tflag, struct FTW* ftwbuf) {
64 sanity_check_ftw(fpath, sb, tflag);
65 ASSERT_EQ('/', fpath[ftwbuf->base - 1]) << fpath;
66 }
67
check_ftw(const char * fpath,const struct stat * sb,int tflag)68 int check_ftw(const char* fpath, const struct stat* sb, int tflag) {
69 sanity_check_ftw(fpath, sb, tflag);
70 return 0;
71 }
72
check_ftw64(const char * fpath,const struct stat64 * sb,int tflag)73 int check_ftw64(const char* fpath, const struct stat64* sb, int tflag) {
74 sanity_check_ftw(fpath, reinterpret_cast<const struct stat*>(sb), tflag);
75 return 0;
76 }
77
check_nftw(const char * fpath,const struct stat * sb,int tflag,struct FTW * ftwbuf)78 int check_nftw(const char* fpath, const struct stat* sb, int tflag, struct FTW* ftwbuf) {
79 sanity_check_nftw(fpath, sb, tflag, ftwbuf);
80 return 0;
81 }
82
check_nftw64(const char * fpath,const struct stat64 * sb,int tflag,struct FTW * ftwbuf)83 int check_nftw64(const char* fpath, const struct stat64* sb, int tflag, struct FTW* ftwbuf) {
84 sanity_check_nftw(fpath, reinterpret_cast<const struct stat*>(sb), tflag, ftwbuf);
85 return 0;
86 }
87
TEST(ftw,ftw)88 TEST(ftw, ftw) {
89 TemporaryDir root;
90 MakeTree(root.dirname);
91 ASSERT_EQ(0, ftw(root.dirname, check_ftw, 128));
92 }
93
TEST(ftw,ftw64)94 TEST(ftw, ftw64) {
95 TemporaryDir root;
96 MakeTree(root.dirname);
97 ASSERT_EQ(0, ftw64(root.dirname, check_ftw64, 128));
98 }
99
TEST(ftw,nftw)100 TEST(ftw, nftw) {
101 TemporaryDir root;
102 MakeTree(root.dirname);
103 ASSERT_EQ(0, nftw(root.dirname, check_nftw, 128, 0));
104 }
105
TEST(ftw,nftw64)106 TEST(ftw, nftw64) {
107 TemporaryDir root;
108 MakeTree(root.dirname);
109 ASSERT_EQ(0, nftw64(root.dirname, check_nftw64, 128, 0));
110 }
111