]>
Commit | Line | Data |
---|---|---|
1e59de90 TL |
1 | //===- FuzzerIOPosix.cpp - IO utils for Posix. ----------------------------===// |
2 | // | |
3 | // The LLVM Compiler Infrastructure | |
4 | // | |
5 | // This file is distributed under the University of Illinois Open Source | |
6 | // License. See LICENSE.TXT for details. | |
7 | // | |
8 | //===----------------------------------------------------------------------===// | |
9 | // IO functions implementation using Posix API. | |
10 | //===----------------------------------------------------------------------===// | |
11 | #include "FuzzerDefs.h" | |
12 | #if LIBFUZZER_POSIX | |
13 | ||
14 | #include "FuzzerExtFunctions.h" | |
15 | #include "FuzzerIO.h" | |
16 | #include <cstdarg> | |
17 | #include <cstdio> | |
18 | #include <dirent.h> | |
19 | #include <fstream> | |
20 | #include <iterator> | |
21 | #include <libgen.h> | |
22 | #include <sys/stat.h> | |
23 | #include <sys/types.h> | |
24 | #include <unistd.h> | |
25 | ||
26 | namespace fuzzer { | |
27 | ||
28 | bool IsFile(const std::string &Path) { | |
29 | struct stat St; | |
30 | if (stat(Path.c_str(), &St)) | |
31 | return false; | |
32 | return S_ISREG(St.st_mode); | |
33 | } | |
34 | ||
35 | void ListFilesInDirRecursive(const std::string &Dir, long *Epoch, | |
36 | std::vector<std::string> *V, bool TopDir) { | |
37 | auto E = GetEpoch(Dir); | |
38 | if (Epoch) | |
39 | if (E && *Epoch >= E) return; | |
40 | ||
41 | DIR *D = opendir(Dir.c_str()); | |
42 | if (!D) { | |
43 | Printf("No such directory: %s; exiting\n", Dir.c_str()); | |
44 | exit(1); | |
45 | } | |
46 | while (auto E = readdir(D)) { | |
47 | std::string Path = DirPlusFile(Dir, E->d_name); | |
48 | if (E->d_type == DT_REG || E->d_type == DT_LNK) | |
49 | V->push_back(Path); | |
50 | else if (E->d_type == DT_DIR && *E->d_name != '.') | |
51 | ListFilesInDirRecursive(Path, Epoch, V, false); | |
52 | } | |
53 | closedir(D); | |
54 | if (Epoch && TopDir) | |
55 | *Epoch = E; | |
56 | } | |
57 | ||
58 | char GetSeparator() { | |
59 | return '/'; | |
60 | } | |
61 | ||
62 | FILE* OpenFile(int Fd, const char* Mode) { | |
63 | return fdopen(Fd, Mode); | |
64 | } | |
65 | ||
66 | int CloseFile(int fd) { | |
67 | return close(fd); | |
68 | } | |
69 | ||
70 | int DuplicateFile(int Fd) { | |
71 | return dup(Fd); | |
72 | } | |
73 | ||
74 | void RemoveFile(const std::string &Path) { | |
75 | unlink(Path.c_str()); | |
76 | } | |
77 | ||
78 | std::string DirName(const std::string &FileName) { | |
79 | char *Tmp = new char[FileName.size() + 1]; | |
80 | memcpy(Tmp, FileName.c_str(), FileName.size() + 1); | |
81 | std::string Res = dirname(Tmp); | |
82 | delete [] Tmp; | |
83 | return Res; | |
84 | } | |
85 | ||
86 | } // namespace fuzzer | |
87 | ||
88 | #endif // LIBFUZZER_POSIX |