crashpad/snapshot/win/crashpad_snapshot_test_image_reader.cc
Scott Graham 9104d53d1c Fix thread startup in snapshot reader test
The use of InterlockedCompareExchange() was very wrong. Improved error
checking coming in another CL from mark@, see linked bug for discussion.

Bug: crashpad:160
Change-Id: Id230af6f37c6cdce807dd4d8aba9d33e9bdeffd0
Reviewed-on: https://chromium-review.googlesource.com/459230
Commit-Queue: Scott Graham <scottmg@chromium.org>
Reviewed-by: Mark Mentovai <mark@chromium.org>
2017-03-24 19:32:04 +00:00

88 lines
2.8 KiB
C++

// 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 <windows.h>
#include "base/logging.h"
#include "client/crashpad_info.h"
#include "util/file/file_io.h"
#include "util/win/scoped_handle.h"
DWORD WINAPI LotsOfReferencesThreadProc(void* param) {
LONG* count = reinterpret_cast<LONG*>(param);
// Allocate a bunch of pointers to things on the stack.
int* pointers[1000];
for (int i = 0; i < arraysize(pointers); ++i) {
pointers[i] = new int[2048];
}
InterlockedIncrement(count);
Sleep(INFINITE);
return 0;
}
int wmain(int argc, wchar_t* argv[]) {
CHECK_EQ(argc, 2);
crashpad::ScopedKernelHANDLE done(CreateEvent(nullptr, true, false, argv[1]));
PCHECK(LoadLibrary(L"crashpad_snapshot_test_image_reader_module.dll"))
<< "LoadLibrary";
// Create threads with lots of stack pointers to memory. This is used to
// verify the cap on pointed-to memory.
LONG thread_ready_count = 0;
crashpad::ScopedKernelHANDLE threads[100];
for (int i = 0; i < arraysize(threads); ++i) {
threads[i].reset(CreateThread(nullptr,
0,
&LotsOfReferencesThreadProc,
reinterpret_cast<void*>(&thread_ready_count),
0,
nullptr));
if (!threads[i].is_valid()) {
PLOG(ERROR) << "CreateThread";
return 1;
}
}
for (;;) {
if (InterlockedCompareExchange(&thread_ready_count,
thread_ready_count,
arraysize(threads)) == arraysize(threads)) {
// All threads have allocated their references.
break;
}
Sleep(10);
}
crashpad::CrashpadInfo* crashpad_info =
crashpad::CrashpadInfo::GetCrashpadInfo();
crashpad_info->set_gather_indirectly_referenced_memory(
crashpad::TriState::kEnabled, 100000);
// Tell the parent process we're ready to proceed.
HANDLE out = GetStdHandle(STD_OUTPUT_HANDLE);
PCHECK(out != INVALID_HANDLE_VALUE) << "GetStdHandle";
char c = ' ';
crashpad::CheckedWriteFile(out, &c, sizeof(c));
// Parent process says we can exit.
CHECK_EQ(WAIT_OBJECT_0, WaitForSingleObject(done.get(), INFINITE));
return 0;
}