Index: llvm/trunk/include/llvm/Support/GraphWriter.h =================================================================== --- llvm/trunk/include/llvm/Support/GraphWriter.h +++ llvm/trunk/include/llvm/Support/GraphWriter.h @@ -27,6 +27,7 @@ #include "llvm/ADT/StringRef.h" #include "llvm/ADT/Twine.h" #include "llvm/Support/DOTGraphTraits.h" +#include "llvm/Support/FileSystem.h" #include "llvm/Support/raw_ostream.h" #include #include @@ -320,14 +321,32 @@ std::string createGraphFilename(const Twine &Name, int &FD); +/// Writes graph into a provided {@code Filename}. +/// If {@code Filename} is empty, generates a random one. +/// \return The resulting filename, or an empty string if writing +/// failed. template std::string WriteGraph(const GraphType &G, const Twine &Name, - bool ShortNames = false, const Twine &Title = "") { + bool ShortNames = false, + const Twine &Title = "", + std::string Filename = "") { int FD; // Windows can't always handle long paths, so limit the length of the name. std::string N = Name.str(); N = N.substr(0, std::min(N.size(), 140)); - std::string Filename = createGraphFilename(N, FD); + if (Filename.empty()) { + Filename = createGraphFilename(N, FD); + } else { + std::error_code EC = sys::fs::openFileForWrite(Filename, FD); + + // Writing over an existing file is not considered an error. + if (EC == std::errc::file_exists) { + errs() << "file exists, overwriting" << "\n"; + } else if (EC) { + errs() << "error writing into file" << "\n"; + return ""; + } + } raw_fd_ostream O(FD, /*shouldClose=*/ true); if (FD == -1) {