// IO functions.
//===----------------------------------------------------------------------===//
#include "FuzzerInternal.h"
+#include <iterator>
#include <fstream>
#include <dirent.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <unistd.h>
+#include <cstdio>
+
namespace fuzzer {
-std::vector<std::string> ListFilesInDir(const std::string &Dir) {
+static long GetEpoch(const std::string &Path) {
+ struct stat St;
+ if (stat(Path.c_str(), &St)) return 0;
+ return St.st_mtime;
+}
+
+static std::vector<std::string> ListFilesInDir(const std::string &Dir,
+ long *Epoch) {
std::vector<std::string> V;
+ if (Epoch) {
+ auto E = GetEpoch(Dir.c_str());
+ if (*Epoch >= E) return V;
+ *Epoch = E;
+ }
DIR *D = opendir(Dir.c_str());
if (!D) return V;
while (auto E = readdir(D)) {
std::istreambuf_iterator<char>());
}
+std::string FileToString(const std::string &Path) {
+ std::ifstream T(Path);
+ return std::string((std::istreambuf_iterator<char>(T)),
+ std::istreambuf_iterator<char>());
+}
+
+void CopyFileToErr(const std::string &Path) {
+ Printf("%s", FileToString(Path).c_str());
+}
+
void WriteToFile(const Unit &U, const std::string &Path) {
std::ofstream OF(Path);
OF.write((const char*)U.data(), U.size());
}
-void ReadDirToVectorOfUnits(const char *Path, std::vector<Unit> *V) {
- for (auto &X : ListFilesInDir(Path))
- V->push_back(FileToVector(std::string(Path) + "/" + X));
+void ReadDirToVectorOfUnits(const char *Path, std::vector<Unit> *V,
+ long *Epoch) {
+ long E = Epoch ? *Epoch : 0;
+ for (auto &X : ListFilesInDir(Path, Epoch)) {
+ auto FilePath = DirPlusFile(Path, X);
+ if (Epoch && GetEpoch(FilePath) < E) continue;
+ V->push_back(FileToVector(FilePath));
+ }
+}
+
+std::string DirPlusFile(const std::string &DirPath,
+ const std::string &FileName) {
+ return DirPath + "/" + FileName;
+}
+
+void PrintFileAsBase64(const std::string &Path) {
+ std::string Cmd = "base64 -w 0 < " + Path + "; echo";
+ ExecuteCommand(Cmd);
+}
+
+void Printf(const char *Fmt, ...) {
+ va_list ap;
+ va_start(ap, Fmt);
+ vfprintf(stderr, Fmt, ap);
+ va_end(ap);
}
} // namespace fuzzer