Index: tools/clang/rewrite_to_chrome_style/RewriteToChromeStyle.cpp |
diff --git a/tools/clang/rewrite_to_chrome_style/RewriteToChromeStyle.cpp b/tools/clang/rewrite_to_chrome_style/RewriteToChromeStyle.cpp |
index 98b4c9453d376aa24450476a1991068f6e7e1443..f4fe3743d6170d941ba8f91bc9f5e6fa3feeaa0c 100644 |
--- a/tools/clang/rewrite_to_chrome_style/RewriteToChromeStyle.cpp |
+++ b/tools/clang/rewrite_to_chrome_style/RewriteToChromeStyle.cpp |
@@ -20,6 +20,13 @@ |
#include <unordered_map> |
#include <unordered_set> |
+#if defined(_WIN32) |
+#include <windows.h> |
+#else |
+#include <sys/file.h> |
+#include <unistd.h> |
+#endif |
+ |
#include "clang/AST/ASTContext.h" |
#include "clang/ASTMatchers/ASTMatchFinder.h" |
#include "clang/ASTMatchers/ASTMatchers.h" |
@@ -511,6 +518,17 @@ int main(int argc, const char* argv[]) { |
if (result != 0) |
return result; |
+#if defined(_WIN32) |
+ OFSTRUCT lockfd_out; |
+ HFILE lockfd = OpenFile("rewrite-sym.lock", &lockfd_out, OF_CREATE); |
+ OVERLAPPED lock_out = {} LockFileEx(lockfd, LOCKFILE_EXCLUSIVE_LOCK, 0, 1, 0, |
dcheng
2016/01/23 00:10:07
Missing a ; here I think.
danakj
2016/01/23 00:22:38
oops :) ya
|
+ &lock_out); |
+#else |
+ int lockfd = open("rewrite-sym.lock", O_RDWR | O_CREAT, 0666); |
+ while (flock(lockfd, LOCK_EX)) { // :D |
+ } |
+#endif |
+ |
std::ofstream replacement_db_file("rewrite-sym.txt", |
std::ios_base::out | std::ios_base::app); |
for (const auto& p : field_decl_rewriter.replacement_names()) |
@@ -523,6 +541,14 @@ int main(int argc, const char* argv[]) { |
replacement_db_file << "fun:" << p.first << ":" << p.second << "\n"; |
replacement_db_file.close(); |
+#if defined(_WIN32) |
+ UnlockFile(lockfd, 0, 0, 1, 0); |
+ CloseHandle(lockfd); |
+#else |
+ flock(lockfd, LOCK_UN); |
+ close(lockfd); |
+#endif |
+ |
// Serialization format is documented in tools/clang/scripts/run_tool.py |
llvm::outs() << "==== BEGIN EDITS ====\n"; |
for (const auto& r : replacements) { |