1 /*
2 * Copyright (C) 2013, 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 "dictionary/utils/file_utils.h"
18
19 #include <cstdio>
20 #include <cstring>
21 #include <dirent.h>
22 #include <fcntl.h>
23 #include <libgen.h>
24 #include <sys/types.h>
25 #include <sys/stat.h>
26 #include <unistd.h>
27
28 namespace latinime {
29
30 // Returns -1 on error.
getFileSize(const char * const filePath)31 /* static */ int FileUtils::getFileSize(const char *const filePath) {
32 const int fd = open(filePath, O_RDONLY);
33 if (fd == -1) {
34 return -1;
35 }
36 struct stat statBuf;
37 if (fstat(fd, &statBuf) != 0) {
38 close(fd);
39 return -1;
40 }
41 close(fd);
42 return static_cast<int>(statBuf.st_size);
43 }
44
existsDir(const char * const dirPath)45 /* static */ bool FileUtils::existsDir(const char *const dirPath) {
46 DIR *const dir = opendir(dirPath);
47 if (dir == NULL) {
48 return false;
49 }
50 closedir(dir);
51 return true;
52 }
53
54 // Remove a directory and all files in the directory.
removeDirAndFiles(const char * const dirPath)55 /* static */ bool FileUtils::removeDirAndFiles(const char *const dirPath) {
56 return removeDirAndFiles(dirPath, 5 /* maxTries */);
57 }
58
59 // Remove a directory and all files in the directory, trying up to maxTimes.
removeDirAndFiles(const char * const dirPath,const int maxTries)60 /* static */ bool FileUtils::removeDirAndFiles(const char *const dirPath, const int maxTries) {
61 DIR *const dir = opendir(dirPath);
62 if (dir == NULL) {
63 AKLOGE("Cannot open dir %s.", dirPath);
64 return true;
65 }
66 struct dirent *dirent;
67 while ((dirent = readdir(dir)) != NULL) {
68 if (dirent->d_type == DT_DIR) {
69 continue;
70 }
71 if (strcmp(dirent->d_name, ".") == 0 || strcmp(dirent->d_name, "..") == 0) {
72 continue;
73 }
74 const int filePathBufSize = getFilePathBufSize(dirPath, dirent->d_name);
75 char filePath[filePathBufSize];
76 getFilePath(dirPath, dirent->d_name, filePathBufSize, filePath);
77 if (remove(filePath) != 0) {
78 AKLOGE("Cannot remove file %s.", filePath);
79 closedir(dir);
80 return false;
81 }
82 }
83 closedir(dir);
84 if (remove(dirPath) != 0) {
85 if (maxTries > 0) {
86 // On NFS, deleting files sometimes creates new files. I'm not sure what the
87 // correct way of dealing with this is, but for the time being, this seems to work.
88 removeDirAndFiles(dirPath, maxTries - 1);
89 } else {
90 AKLOGE("Cannot remove directory %s.", dirPath);
91 return false;
92 }
93 }
94 return true;
95 }
96
getFilePathWithSuffixBufSize(const char * const filePath,const char * const suffix)97 /* static */ int FileUtils::getFilePathWithSuffixBufSize(const char *const filePath,
98 const char *const suffix) {
99 return strlen(filePath) + strlen(suffix) + 1 /* terminator */;
100 }
101
getFilePathWithSuffix(const char * const filePath,const char * const suffix,const int filePathBufSize,char * const outFilePath)102 /* static */ void FileUtils::getFilePathWithSuffix(const char *const filePath,
103 const char *const suffix, const int filePathBufSize, char *const outFilePath) {
104 snprintf(outFilePath, filePathBufSize, "%s%s", filePath, suffix);
105 }
106
getFilePathBufSize(const char * const dirPath,const char * const fileName)107 /* static */ int FileUtils::getFilePathBufSize(const char *const dirPath,
108 const char *const fileName) {
109 return strlen(dirPath) + 1 /* '/' */ + strlen(fileName) + 1 /* terminator */;
110 }
111
getFilePath(const char * const dirPath,const char * const fileName,const int filePathBufSize,char * const outFilePath)112 /* static */ void FileUtils::getFilePath(const char *const dirPath, const char *const fileName,
113 const int filePathBufSize, char *const outFilePath) {
114 snprintf(outFilePath, filePathBufSize, "%s/%s", dirPath, fileName);
115 }
116
getFilePathWithoutSuffix(const char * const filePath,const char * const suffix,const int outDirPathBufSize,char * const outDirPath)117 /* static */ bool FileUtils::getFilePathWithoutSuffix(const char *const filePath,
118 const char *const suffix, const int outDirPathBufSize, char *const outDirPath) {
119 const int filePathLength = strlen(filePath);
120 const int suffixLength = strlen(suffix);
121 if (filePathLength <= suffixLength) {
122 AKLOGE("File path length (%s:%d) is shorter that suffix length (%s:%d).",
123 filePath, filePathLength, suffix, suffixLength);
124 return false;
125 }
126 const int resultFilePathLength = filePathLength - suffixLength;
127 if (outDirPathBufSize <= resultFilePathLength) {
128 AKLOGE("outDirPathBufSize is too small. filePath: %s, suffix: %s, outDirPathBufSize: %d",
129 filePath, suffix, outDirPathBufSize);
130 return false;
131 }
132 if (strncmp(filePath + resultFilePathLength, suffix, suffixLength) != 0) {
133 AKLOGE("File Path %s does not have %s as a suffix", filePath, suffix);
134 return false;
135 }
136 snprintf(outDirPath, resultFilePathLength + 1 /* terminator */, "%s", filePath);
137 return true;
138 }
139
getDirPath(const char * const filePath,const int outDirPathBufSize,char * const outDirPath)140 /* static */ void FileUtils::getDirPath(const char *const filePath, const int outDirPathBufSize,
141 char *const outDirPath) {
142 for (int i = strlen(filePath) - 1; i >= 0; --i) {
143 if (filePath[i] == '/') {
144 if (i >= outDirPathBufSize) {
145 AKLOGE("outDirPathBufSize is too small. filePath: %s, outDirPathBufSize: %d",
146 filePath, outDirPathBufSize);
147 ASSERT(false);
148 return;
149 }
150 snprintf(outDirPath, i + 1 /* terminator */, "%s", filePath);
151 return;
152 }
153 }
154 }
155
getBasename(const char * const filePath,const int outNameBufSize,char * const outName)156 /* static */ void FileUtils::getBasename(const char *const filePath,
157 const int outNameBufSize, char *const outName) {
158 const int filePathBufSize = strlen(filePath) + 1 /* terminator */;
159 char filePathBuf[filePathBufSize];
160 snprintf(filePathBuf, filePathBufSize, "%s", filePath);
161 const char *const baseName = basename(filePathBuf);
162 const int baseNameLength = strlen(baseName);
163 if (baseNameLength >= outNameBufSize) {
164 AKLOGE("outNameBufSize is too small. filePath: %s, outNameBufSize: %d",
165 filePath, outNameBufSize);
166 return;
167 }
168 snprintf(outName, baseNameLength + 1 /* terminator */, "%s", baseName);
169 }
170
171 } // namespace latinime
172