Chromium Code Reviews| 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) { |