Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(231)

Unified Diff: util/file/file_io_test.cc

Issue 1001673002: Add Locking calls to file_io.h plus implementations and test (Closed) Base URL: https://chromium.googlesource.com/crashpad/crashpad@master
Patch Set: . Created 5 years, 9 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: util/file/file_io_test.cc
diff --git a/util/file/file_io_test.cc b/util/file/file_io_test.cc
new file mode 100644
index 0000000000000000000000000000000000000000..4f0218f511247d9f3177057cdf3b7cdbbafd1d94
--- /dev/null
+++ b/util/file/file_io_test.cc
@@ -0,0 +1,148 @@
+// Copyright 2015 The Crashpad Authors. All rights reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#include "util/file/file_io.h"
+
+#include "base/files/file_path.h"
+#include "gtest/gtest.h"
+#include "util/test/scoped_temp_dir.h"
+
+namespace crashpad {
+namespace test {
+namespace {
+
+TEST(FileIO, FileShareMode) {
+ ScopedTempDir temp_dir;
+ base::FilePath shared_file =
+ temp_dir.path().Append(FILE_PATH_LITERAL("shared_file"));
+ auto handle1 = ScopedFileHandle(LoggingOpenFileForWrite(
+ shared_file, FileWriteMode::kReuseOrCreate, FilePermissions::kOwnerOnly));
Mark Mentovai 2015/03/12 04:47:42 The first one should use kCreateOrFail. Same on li
scottmg 2015/03/19 22:06:20 Done.
+ ASSERT_NE(handle1, kInvalidFileHandle);
+ auto handle2 = ScopedFileHandle(LoggingOpenFileForWrite(
+ shared_file, FileWriteMode::kReuseOrCreate, FilePermissions::kOwnerOnly));
+ EXPECT_NE(handle2, kInvalidFileHandle);
+}
+
+TEST(FileIO, MultipleSharedLocks) {
+ ScopedTempDir temp_dir;
+ base::FilePath shared_file =
+ temp_dir.path().Append(FILE_PATH_LITERAL("file_to_lock"));
+
+ auto handle1 = ScopedFileHandle(LoggingOpenFileForWrite(
+ shared_file, FileWriteMode::kReuseOrCreate, FilePermissions::kOwnerOnly));
Mark Mentovai 2015/03/12 04:47:42 ASSERT_NE(handle1, kInvalidFileHandle) after this,
scottmg 2015/03/19 22:06:20 Done.
+ EXPECT_TRUE(LoggingLockFile(handle1.get(), FileLocking::kShared));
+
+ auto handle2 = ScopedFileHandle(LoggingOpenFileForWrite(
+ shared_file, FileWriteMode::kReuseOrCreate, FilePermissions::kOwnerOnly));
+ EXPECT_TRUE(LoggingLockFile(handle2.get(), FileLocking::kShared));
+
+ EXPECT_TRUE(LoggingUnlockFile(handle1.get()));
+ EXPECT_TRUE(LoggingUnlockFile(handle2.get()));
+}
+
+struct ThreadMainInfo {
+#if defined(OS_POSIX)
Mark Mentovai 2015/03/12 04:47:42 OK for now. Robert and I were talking about needin
Robert Sesek 2015/03/12 15:28:03 I think it's probably time to create this abstract
scottmg 2015/03/19 22:06:20 Added some version of this to util/test/thread.h.
+ pthread_t pthread;
+#elif defined(OS_WIN)
+ HANDLE thread;
+#endif
+ FileHandle file;
Mark Mentovai 2015/03/12 04:47:42 ScopedFileHandle?
scottmg 2015/03/19 22:06:20 Done.
+ int iterations;
+ int* shared_counter;
+};
+
+#if defined(OS_POSIX)
+void*
+#elif defined(OS_WIN)
+DWORD WINAPI
+#endif // OS_POSIX
+ThreadMain(void* argument) {
+ ThreadMainInfo* info = reinterpret_cast<ThreadMainInfo*>(argument);
+ for (int i = 0; i < info->iterations; ++i) {
+ EXPECT_TRUE(LoggingLockFile(info->file, FileLocking::kExclusive));
Mark Mentovai 2015/03/12 04:47:42 You haven’t done anything to test that an exclusiv
scottmg 2015/03/19 22:06:20 Done.
+ *info->shared_counter += 1;
Mark Mentovai 2015/03/12 04:47:42 For the case where the the other threads will be u
scottmg 2015/03/19 22:06:20 Done.
+ EXPECT_TRUE(LoggingUnlockFile(info->file));
+ }
+
+#if defined(OS_POSIX)
+ return nullptr;
+#elif defined(OS_WIN)
+ return 0;
+#endif // OS_POSIX
+}
+
+void StartThread(ThreadMainInfo* info) {
+#if defined(OS_POSIX)
+ int rv = pthread_create(&info->pthread, nullptr, ThreadMain, info);
+ ASSERT_EQ(0, rv) << "pthread_create";
Mark Mentovai 2015/03/12 04:47:42 From util/test/errors.h, use ErrnoMessage, like
scottmg 2015/03/19 22:06:19 Done.
+#elif defined(OS_WIN)
+ info->thread = CreateThread(nullptr, 0, ThreadMain, info, 0, nullptr);
+ ASSERT_NE(nullptr, info->thread) << "CreateThread";
+#endif // OS_POSIX
+}
+
+void JoinThread(ThreadMainInfo* info) {
+#if defined(OS_POSIX)
+ int rv = pthread_join(info->pthread, nullptr);
+ EXPECT_EQ(0, rv) << "pthread_join";
+#elif defined(OS_WIN)
+ DWORD result = WaitForSingleObject(info->thread, INFINITE);
+ EXPECT_EQ(WAIT_OBJECT_0, result) << "WaitForSingleObject";
+#endif // OS_POSIX
+}
+
+TEST(FileIO, ExclusiveExcludes) {
+ ScopedTempDir temp_dir;
+ base::FilePath shared_file =
+ temp_dir.path().Append(FILE_PATH_LITERAL("file_to_lock"));
+
+ auto initial = ScopedFileHandle(LoggingOpenFileForWrite(
+ shared_file, FileWriteMode::kReuseOrCreate, FilePermissions::kOwnerOnly));
+ ASSERT_TRUE(LoggingLockFile(initial.get(), FileLocking::kExclusive));
+
+ int shared_counter = 0;
+
+ const int kThreads = 10;
+ ThreadMainInfo info[kThreads];
+ int iterations = 0;
Mark Mentovai 2015/03/12 04:47:42 expected_iterations makes it clearer. If shared_co
scottmg 2015/03/19 22:06:20 Done.
+ for (int index = 0; index < kThreads; ++index) {
Mark Mentovai 2015/03/12 04:47:42 #include "base/basictypes.h" and use arraysize(inf
scottmg 2015/03/19 22:06:19 Done.
+ info[index].file = LoggingOpenFileForWrite(shared_file,
+ FileWriteMode::kReuseOrCreate,
+ FilePermissions::kOwnerOnly);
+ ASSERT_NE(info[index].file, kInvalidFileHandle);
+ info[index].iterations = index * 10;
+ info[index].shared_counter = &shared_counter;
+ iterations += info[index].iterations;
+
+ ASSERT_NO_FATAL_FAILURE(StartThread(&info[index]));
+ }
+
+ EXPECT_EQ(0, shared_counter);
+
+ ASSERT_TRUE(LoggingUnlockFile(initial.get()));
+
+ for (int index = 0; index < kThreads; ++index) {
+ JoinThread(&info[index]);
+ }
+
+ EXPECT_EQ(shared_counter, iterations);
Mark Mentovai 2015/03/12 04:47:42 These arguments are swapped.
scottmg 2015/03/19 22:06:20 Done.
+
+ for (int index = 0; index < kThreads; ++index) {
+ EXPECT_TRUE(LoggingCloseFile(info[index].file));
Mark Mentovai 2015/03/12 04:47:42 This could have happened in ThreadMain() instead o
scottmg 2015/03/19 22:06:20 Removed via ScopedFileHandle.
+ }
+}
+
+} // namespace
+} // namespace test
+} // namespace crashpad

Powered by Google App Engine
This is Rietveld 408576698