// IO functions.
//===----------------------------------------------------------------------===//
#include "FuzzerInternal.h"
-#include <iostream>
#include <iterator>
#include <fstream>
#include <dirent.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
+#include <cstdarg>
+#include <cstdio>
namespace fuzzer {
static long GetEpoch(const std::string &Path) {
struct stat St;
- if (stat(Path.c_str(), &St)) return 0;
+ if (stat(Path.c_str(), &St))
+ return 0; // Can't stat, be conservative.
return St.st_mtime;
}
long *Epoch) {
std::vector<std::string> V;
if (Epoch) {
- auto E = GetEpoch(Dir.c_str());
+ auto E = GetEpoch(Dir);
if (*Epoch >= E) return V;
*Epoch = E;
}
DIR *D = opendir(Dir.c_str());
- if (!D) return V;
+ if (!D) {
+ Printf("No such directory: %s; exiting\n", Dir.c_str());
+ exit(1);
+ }
while (auto E = readdir(D)) {
if (E->d_type == DT_REG || E->d_type == DT_LNK)
V.push_back(E->d_name);
Unit FileToVector(const std::string &Path) {
std::ifstream T(Path);
+ if (!T) {
+ Printf("No such directory: %s; exiting\n", Path.c_str());
+ exit(1);
+ }
return Unit((std::istreambuf_iterator<char>(T)),
std::istreambuf_iterator<char>());
}
}
void CopyFileToErr(const std::string &Path) {
- std::ifstream T(Path);
- std::copy(std::istreambuf_iterator<char>(T), std::istreambuf_iterator<char>(),
- std::ostream_iterator<char>(std::cerr, ""));
+ 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());
+ // Use raw C interface because this function may be called from a sig handler.
+ FILE *Out = fopen(Path.c_str(), "w");
+ if (!Out) return;
+ fwrite(U.data(), sizeof(U[0]), U.size(), Out);
+ fclose(Out);
}
void ReadDirToVectorOfUnits(const char *Path, std::vector<Unit> *V,
return DirPath + "/" + FileName;
}
-void PrintFileAsBase64(const std::string &Path) {
- std::string Cmd = "base64 -w 0 < " + Path + "; echo";
- system(Cmd.c_str());
+void Printf(const char *Fmt, ...) {
+ va_list ap;
+ va_start(ap, Fmt);
+ vfprintf(stderr, Fmt, ap);
+ va_end(ap);
}
} // namespace fuzzer