| OLD | NEW |
| 1 // Copyright 2014 The Crashpad Authors. All rights reserved. | 1 // Copyright 2014 The Crashpad Authors. All rights reserved. |
| 2 // | 2 // |
| 3 // Licensed under the Apache License, Version 2.0 (the "License"); | 3 // Licensed under the Apache License, Version 2.0 (the "License"); |
| 4 // you may not use this file except in compliance with the License. | 4 // you may not use this file except in compliance with the License. |
| 5 // You may obtain a copy of the License at | 5 // You may obtain a copy of the License at |
| 6 // | 6 // |
| 7 // http://www.apache.org/licenses/LICENSE-2.0 | 7 // http://www.apache.org/licenses/LICENSE-2.0 |
| 8 // | 8 // |
| 9 // Unless required by applicable law or agreed to in writing, software | 9 // Unless required by applicable law or agreed to in writing, software |
| 10 // distributed under the License is distributed on an "AS IS" BASIS, | 10 // distributed under the License is distributed on an "AS IS" BASIS, |
| (...skipping 13 matching lines...) Expand all Loading... |
| 24 #include <string> | 24 #include <string> |
| 25 #include <vector> | 25 #include <vector> |
| 26 | 26 |
| 27 #include "base/logging.h" | 27 #include "base/logging.h" |
| 28 #include "base/mac/scoped_mach_port.h" | 28 #include "base/mac/scoped_mach_port.h" |
| 29 #include "base/posix/eintr_wrapper.h" | 29 #include "base/posix/eintr_wrapper.h" |
| 30 #include "base/strings/stringprintf.h" | 30 #include "base/strings/stringprintf.h" |
| 31 #include "build/build_config.h" | 31 #include "build/build_config.h" |
| 32 #include "gtest/gtest.h" | 32 #include "gtest/gtest.h" |
| 33 #include "snapshot/mac/mach_o_image_reader.h" | 33 #include "snapshot/mac/mach_o_image_reader.h" |
| 34 #include "util/file/fd_io.h" | 34 #include "util/file/file_io.h" |
| 35 #include "util/mach/mach_extensions.h" | 35 #include "util/mach/mach_extensions.h" |
| 36 #include "util/stdlib/pointer_container.h" | 36 #include "util/stdlib/pointer_container.h" |
| 37 #include "util/synchronization/semaphore.h" | 37 #include "util/synchronization/semaphore.h" |
| 38 #include "util/test/errors.h" | 38 #include "util/test/errors.h" |
| 39 #include "util/test/mac/dyld.h" | 39 #include "util/test/mac/dyld.h" |
| 40 #include "util/test/mac/mach_errors.h" | 40 #include "util/test/mac/mach_errors.h" |
| 41 #include "util/test/mac/mach_multiprocess.h" | 41 #include "util/test/mac/mach_multiprocess.h" |
| 42 | 42 |
| 43 namespace crashpad { | 43 namespace crashpad { |
| 44 namespace test { | 44 namespace test { |
| (...skipping 39 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 84 #else | 84 #else |
| 85 EXPECT_TRUE(process_reader.Is64Bit()); | 85 EXPECT_TRUE(process_reader.Is64Bit()); |
| 86 #endif | 86 #endif |
| 87 | 87 |
| 88 EXPECT_EQ(getpid(), process_reader.ParentProcessID()); | 88 EXPECT_EQ(getpid(), process_reader.ParentProcessID()); |
| 89 EXPECT_EQ(ChildPID(), process_reader.ProcessID()); | 89 EXPECT_EQ(ChildPID(), process_reader.ProcessID()); |
| 90 | 90 |
| 91 int read_fd = ReadPipeFD(); | 91 int read_fd = ReadPipeFD(); |
| 92 | 92 |
| 93 mach_vm_address_t address; | 93 mach_vm_address_t address; |
| 94 CheckedReadFD(read_fd, &address, sizeof(address)); | 94 CheckedReadFile(read_fd, &address, sizeof(address)); |
| 95 | 95 |
| 96 std::string read_string; | 96 std::string read_string; |
| 97 ASSERT_TRUE(process_reader.Memory()->ReadCString(address, &read_string)); | 97 ASSERT_TRUE(process_reader.Memory()->ReadCString(address, &read_string)); |
| 98 EXPECT_EQ(kTestMemory, read_string); | 98 EXPECT_EQ(kTestMemory, read_string); |
| 99 } | 99 } |
| 100 | 100 |
| 101 void MachMultiprocessChild() override { | 101 void MachMultiprocessChild() override { |
| 102 int write_fd = WritePipeFD(); | 102 int write_fd = WritePipeFD(); |
| 103 | 103 |
| 104 mach_vm_address_t address = | 104 mach_vm_address_t address = |
| 105 reinterpret_cast<mach_vm_address_t>(kTestMemory); | 105 reinterpret_cast<mach_vm_address_t>(kTestMemory); |
| 106 CheckedWriteFD(write_fd, &address, sizeof(address)); | 106 CheckedWriteFile(write_fd, &address, sizeof(address)); |
| 107 | 107 |
| 108 // Wait for the parent to signal that it’s OK to exit by closing its end of | 108 // Wait for the parent to signal that it’s OK to exit by closing its end of |
| 109 // the pipe. | 109 // the pipe. |
| 110 CheckedReadFDAtEOF(ReadPipeFD()); | 110 CheckedReadFileAtEOF(ReadPipeFD()); |
| 111 } | 111 } |
| 112 | 112 |
| 113 DISALLOW_COPY_AND_ASSIGN(ProcessReaderChild); | 113 DISALLOW_COPY_AND_ASSIGN(ProcessReaderChild); |
| 114 }; | 114 }; |
| 115 | 115 |
| 116 TEST(ProcessReader, ChildBasic) { | 116 TEST(ProcessReader, ChildBasic) { |
| 117 ProcessReaderChild process_reader_child; | 117 ProcessReaderChild process_reader_child; |
| 118 process_reader_child.Run(); | 118 process_reader_child.Run(); |
| 119 } | 119 } |
| 120 | 120 |
| (...skipping 306 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 427 int read_fd = ReadPipeFD(); | 427 int read_fd = ReadPipeFD(); |
| 428 | 428 |
| 429 // Build a map of all expected threads, keyed by each thread’s ID, and with | 429 // Build a map of all expected threads, keyed by each thread’s ID, and with |
| 430 // addresses that should lie somewhere within each thread’s stack as values. | 430 // addresses that should lie somewhere within each thread’s stack as values. |
| 431 // These IDs and addresses all come from the child process via the pipe. | 431 // These IDs and addresses all come from the child process via the pipe. |
| 432 ThreadMap thread_map; | 432 ThreadMap thread_map; |
| 433 for (size_t thread_index = 0; | 433 for (size_t thread_index = 0; |
| 434 thread_index < thread_count_ + 1; | 434 thread_index < thread_count_ + 1; |
| 435 ++thread_index) { | 435 ++thread_index) { |
| 436 uint64_t thread_id; | 436 uint64_t thread_id; |
| 437 CheckedReadFD(read_fd, &thread_id, sizeof(thread_id)); | 437 CheckedReadFile(read_fd, &thread_id, sizeof(thread_id)); |
| 438 | 438 |
| 439 TestThreadPool::ThreadExpectation expectation; | 439 TestThreadPool::ThreadExpectation expectation; |
| 440 CheckedReadFD(read_fd, | 440 CheckedReadFile(read_fd, |
| 441 &expectation.stack_address, | 441 &expectation.stack_address, |
| 442 sizeof(expectation.stack_address)); | 442 sizeof(expectation.stack_address)); |
| 443 CheckedReadFD(read_fd, | 443 CheckedReadFile(read_fd, |
| 444 &expectation.suspend_count, | 444 &expectation.suspend_count, |
| 445 sizeof(expectation.suspend_count)); | 445 sizeof(expectation.suspend_count)); |
| 446 | 446 |
| 447 // There can’t be any duplicate thread IDs. | 447 // There can’t be any duplicate thread IDs. |
| 448 EXPECT_EQ(0u, thread_map.count(thread_id)); | 448 EXPECT_EQ(0u, thread_map.count(thread_id)); |
| 449 | 449 |
| 450 thread_map[thread_id] = expectation; | 450 thread_map[thread_id] = expectation; |
| 451 } | 451 } |
| 452 | 452 |
| 453 const std::vector<ProcessReader::Thread>& threads = process_reader.Threads()
; | 453 const std::vector<ProcessReader::Thread>& threads = process_reader.Threads()
; |
| 454 | 454 |
| 455 // The child shouldn’t have any threads other than its main thread and the | 455 // The child shouldn’t have any threads other than its main thread and the |
| 456 // ones it created in its pool, so pass false for |tolerate_extra_threads|. | 456 // ones it created in its pool, so pass false for |tolerate_extra_threads|. |
| 457 ExpectSeveralThreads(&thread_map, threads, false); | 457 ExpectSeveralThreads(&thread_map, threads, false); |
| 458 } | 458 } |
| 459 | 459 |
| 460 void MachMultiprocessChild() override { | 460 void MachMultiprocessChild() override { |
| 461 TestThreadPool thread_pool; | 461 TestThreadPool thread_pool; |
| 462 ASSERT_NO_FATAL_FAILURE(thread_pool.StartThreads(thread_count_)); | 462 ASSERT_NO_FATAL_FAILURE(thread_pool.StartThreads(thread_count_)); |
| 463 | 463 |
| 464 int write_fd = WritePipeFD(); | 464 int write_fd = WritePipeFD(); |
| 465 | 465 |
| 466 // This thread isn’t part of the thread pool, but the parent will be able | 466 // This thread isn’t part of the thread pool, but the parent will be able |
| 467 // to inspect it. Write an entry for it. | 467 // to inspect it. Write an entry for it. |
| 468 uint64_t thread_id = PthreadToThreadID(pthread_self()); | 468 uint64_t thread_id = PthreadToThreadID(pthread_self()); |
| 469 | 469 |
| 470 CheckedWriteFD(write_fd, &thread_id, sizeof(thread_id)); | 470 CheckedWriteFile(write_fd, &thread_id, sizeof(thread_id)); |
| 471 | 471 |
| 472 TestThreadPool::ThreadExpectation expectation; | 472 TestThreadPool::ThreadExpectation expectation; |
| 473 expectation.stack_address = reinterpret_cast<mach_vm_address_t>(&thread_id); | 473 expectation.stack_address = reinterpret_cast<mach_vm_address_t>(&thread_id); |
| 474 expectation.suspend_count = 0; | 474 expectation.suspend_count = 0; |
| 475 | 475 |
| 476 CheckedWriteFD(write_fd, | 476 CheckedWriteFile(write_fd, |
| 477 &expectation.stack_address, | 477 &expectation.stack_address, |
| 478 sizeof(expectation.stack_address)); | 478 sizeof(expectation.stack_address)); |
| 479 CheckedWriteFD(write_fd, | 479 CheckedWriteFile(write_fd, |
| 480 &expectation.suspend_count, | 480 &expectation.suspend_count, |
| 481 sizeof(expectation.suspend_count)); | 481 sizeof(expectation.suspend_count)); |
| 482 | 482 |
| 483 // Write an entry for everything in the thread pool. | 483 // Write an entry for everything in the thread pool. |
| 484 for (size_t thread_index = 0; | 484 for (size_t thread_index = 0; |
| 485 thread_index < thread_count_; | 485 thread_index < thread_count_; |
| 486 ++thread_index) { | 486 ++thread_index) { |
| 487 uint64_t thread_id = | 487 uint64_t thread_id = |
| 488 thread_pool.GetThreadInfo(thread_index, &expectation); | 488 thread_pool.GetThreadInfo(thread_index, &expectation); |
| 489 | 489 |
| 490 CheckedWriteFD(write_fd, &thread_id, sizeof(thread_id)); | 490 CheckedWriteFile(write_fd, &thread_id, sizeof(thread_id)); |
| 491 CheckedWriteFD(write_fd, | 491 CheckedWriteFile(write_fd, |
| 492 &expectation.stack_address, | 492 &expectation.stack_address, |
| 493 sizeof(expectation.stack_address)); | 493 sizeof(expectation.stack_address)); |
| 494 CheckedWriteFD(write_fd, | 494 CheckedWriteFile(write_fd, |
| 495 &expectation.suspend_count, | 495 &expectation.suspend_count, |
| 496 sizeof(expectation.suspend_count)); | 496 sizeof(expectation.suspend_count)); |
| 497 } | 497 } |
| 498 | 498 |
| 499 // Wait for the parent to signal that it’s OK to exit by closing its end of | 499 // Wait for the parent to signal that it’s OK to exit by closing its end of |
| 500 // the pipe. | 500 // the pipe. |
| 501 CheckedReadFDAtEOF(ReadPipeFD()); | 501 CheckedReadFileAtEOF(ReadPipeFD()); |
| 502 } | 502 } |
| 503 | 503 |
| 504 size_t thread_count_; | 504 size_t thread_count_; |
| 505 | 505 |
| 506 DISALLOW_COPY_AND_ASSIGN(ProcessReaderThreadedChild); | 506 DISALLOW_COPY_AND_ASSIGN(ProcessReaderThreadedChild); |
| 507 }; | 507 }; |
| 508 | 508 |
| 509 TEST(ProcessReader, ChildOneThread) { | 509 TEST(ProcessReader, ChildOneThread) { |
| 510 // The main thread plus zero child threads equals one thread. | 510 // The main thread plus zero child threads equals one thread. |
| 511 const size_t kChildThreads = 0; | 511 const size_t kChildThreads = 0; |
| (...skipping 78 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 590 const std::vector<ProcessReader::Module>& modules = | 590 const std::vector<ProcessReader::Module>& modules = |
| 591 process_reader.Modules(); | 591 process_reader.Modules(); |
| 592 | 592 |
| 593 // There needs to be at least an entry for the main executable, for a dylib, | 593 // There needs to be at least an entry for the main executable, for a dylib, |
| 594 // and for dyld. | 594 // and for dyld. |
| 595 ASSERT_GE(modules.size(), 3u); | 595 ASSERT_GE(modules.size(), 3u); |
| 596 | 596 |
| 597 int read_fd = ReadPipeFD(); | 597 int read_fd = ReadPipeFD(); |
| 598 | 598 |
| 599 uint32_t expect_modules; | 599 uint32_t expect_modules; |
| 600 CheckedReadFD(read_fd, &expect_modules, sizeof(expect_modules)); | 600 CheckedReadFile(read_fd, &expect_modules, sizeof(expect_modules)); |
| 601 | 601 |
| 602 ASSERT_EQ(expect_modules, modules.size()); | 602 ASSERT_EQ(expect_modules, modules.size()); |
| 603 | 603 |
| 604 for (size_t index = 0; index < modules.size(); ++index) { | 604 for (size_t index = 0; index < modules.size(); ++index) { |
| 605 SCOPED_TRACE(base::StringPrintf( | 605 SCOPED_TRACE(base::StringPrintf( |
| 606 "index %zu, name %s", index, modules[index].name.c_str())); | 606 "index %zu, name %s", index, modules[index].name.c_str())); |
| 607 | 607 |
| 608 uint32_t expect_name_length; | 608 uint32_t expect_name_length; |
| 609 CheckedReadFD( | 609 CheckedReadFile( |
| 610 read_fd, &expect_name_length, sizeof(expect_name_length)); | 610 read_fd, &expect_name_length, sizeof(expect_name_length)); |
| 611 | 611 |
| 612 // The NUL terminator is not read. | 612 // The NUL terminator is not read. |
| 613 std::string expect_name(expect_name_length, '\0'); | 613 std::string expect_name(expect_name_length, '\0'); |
| 614 CheckedReadFD(read_fd, &expect_name[0], expect_name_length); | 614 CheckedReadFile(read_fd, &expect_name[0], expect_name_length); |
| 615 EXPECT_EQ(expect_name, modules[index].name); | 615 EXPECT_EQ(expect_name, modules[index].name); |
| 616 | 616 |
| 617 mach_vm_address_t expect_address; | 617 mach_vm_address_t expect_address; |
| 618 CheckedReadFD(read_fd, &expect_address, sizeof(expect_address)); | 618 CheckedReadFile(read_fd, &expect_address, sizeof(expect_address)); |
| 619 EXPECT_EQ(expect_address, modules[index].reader->Address()); | 619 EXPECT_EQ(expect_address, modules[index].reader->Address()); |
| 620 | 620 |
| 621 if (index == 0 || index == modules.size() - 1) { | 621 if (index == 0 || index == modules.size() - 1) { |
| 622 // dyld didn’t load the main executable or itself, so it couldn’t record | 622 // dyld didn’t load the main executable or itself, so it couldn’t record |
| 623 // these timestamps, and they are reported as 0. | 623 // these timestamps, and they are reported as 0. |
| 624 EXPECT_EQ(0, modules[index].timestamp); | 624 EXPECT_EQ(0, modules[index].timestamp); |
| 625 } else { | 625 } else { |
| 626 // Hope that the module didn’t change on disk. | 626 // Hope that the module didn’t change on disk. |
| 627 struct stat stat_buf; | 627 struct stat stat_buf; |
| 628 int rv = stat(expect_name.c_str(), &stat_buf); | 628 int rv = stat(expect_name.c_str(), &stat_buf); |
| (...skipping 12 matching lines...) Expand all Loading... |
| 641 const struct dyld_all_image_infos* dyld_image_infos = | 641 const struct dyld_all_image_infos* dyld_image_infos = |
| 642 _dyld_get_all_image_infos(); | 642 _dyld_get_all_image_infos(); |
| 643 | 643 |
| 644 uint32_t write_image_count = dyld_image_count; | 644 uint32_t write_image_count = dyld_image_count; |
| 645 if (dyld_image_infos->version >= 2) { | 645 if (dyld_image_infos->version >= 2) { |
| 646 // dyld_image_count doesn’t include an entry for dyld itself, but one will | 646 // dyld_image_count doesn’t include an entry for dyld itself, but one will |
| 647 // be written. | 647 // be written. |
| 648 ++write_image_count; | 648 ++write_image_count; |
| 649 } | 649 } |
| 650 | 650 |
| 651 CheckedWriteFD(write_fd, &write_image_count, sizeof(write_image_count)); | 651 CheckedWriteFile(write_fd, &write_image_count, sizeof(write_image_count)); |
| 652 | 652 |
| 653 for (size_t index = 0; index < write_image_count; ++index) { | 653 for (size_t index = 0; index < write_image_count; ++index) { |
| 654 const char* dyld_image_name; | 654 const char* dyld_image_name; |
| 655 mach_vm_address_t dyld_image_address; | 655 mach_vm_address_t dyld_image_address; |
| 656 | 656 |
| 657 if (index < dyld_image_count) { | 657 if (index < dyld_image_count) { |
| 658 dyld_image_name = _dyld_get_image_name(index); | 658 dyld_image_name = _dyld_get_image_name(index); |
| 659 dyld_image_address = | 659 dyld_image_address = |
| 660 reinterpret_cast<mach_vm_address_t>(_dyld_get_image_header(index)); | 660 reinterpret_cast<mach_vm_address_t>(_dyld_get_image_header(index)); |
| 661 } else { | 661 } else { |
| 662 dyld_image_name = "/usr/lib/dyld"; | 662 dyld_image_name = "/usr/lib/dyld"; |
| 663 dyld_image_address = reinterpret_cast<mach_vm_address_t>( | 663 dyld_image_address = reinterpret_cast<mach_vm_address_t>( |
| 664 dyld_image_infos->dyldImageLoadAddress); | 664 dyld_image_infos->dyldImageLoadAddress); |
| 665 } | 665 } |
| 666 | 666 |
| 667 uint32_t dyld_image_name_length = strlen(dyld_image_name); | 667 uint32_t dyld_image_name_length = strlen(dyld_image_name); |
| 668 CheckedWriteFD( | 668 CheckedWriteFile( |
| 669 write_fd, &dyld_image_name_length, sizeof(dyld_image_name_length)); | 669 write_fd, &dyld_image_name_length, sizeof(dyld_image_name_length)); |
| 670 | 670 |
| 671 // The NUL terminator is not written. | 671 // The NUL terminator is not written. |
| 672 CheckedWriteFD(write_fd, dyld_image_name, dyld_image_name_length); | 672 CheckedWriteFile(write_fd, dyld_image_name, dyld_image_name_length); |
| 673 | 673 |
| 674 CheckedWriteFD(write_fd, &dyld_image_address, sizeof(dyld_image_address)); | 674 CheckedWriteFile( |
| 675 write_fd, &dyld_image_address, sizeof(dyld_image_address)); |
| 675 } | 676 } |
| 676 | 677 |
| 677 // Wait for the parent to signal that it’s OK to exit by closing its end of | 678 // Wait for the parent to signal that it’s OK to exit by closing its end of |
| 678 // the pipe. | 679 // the pipe. |
| 679 CheckedReadFDAtEOF(ReadPipeFD()); | 680 CheckedReadFileAtEOF(ReadPipeFD()); |
| 680 } | 681 } |
| 681 | 682 |
| 682 DISALLOW_COPY_AND_ASSIGN(ProcessReaderModulesChild); | 683 DISALLOW_COPY_AND_ASSIGN(ProcessReaderModulesChild); |
| 683 }; | 684 }; |
| 684 | 685 |
| 685 TEST(ProcessReader, ChildModules) { | 686 TEST(ProcessReader, ChildModules) { |
| 686 ProcessReaderModulesChild process_reader_modules_child; | 687 ProcessReaderModulesChild process_reader_modules_child; |
| 687 process_reader_modules_child.Run(); | 688 process_reader_modules_child.Run(); |
| 688 } | 689 } |
| 689 | 690 |
| 690 } // namespace | 691 } // namespace |
| 691 } // namespace test | 692 } // namespace test |
| 692 } // namespace crashpad | 693 } // namespace crashpad |
| OLD | NEW |