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, |
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | 11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
12 // See the License for the specific language governing permissions and | 12 // See the License for the specific language governing permissions and |
13 // limitations under the License. | 13 // limitations under the License. |
14 | 14 |
15 #include "snapshot/mac/mach_o_image_reader.h" | 15 #include "snapshot/mac/mach_o_image_reader.h" |
16 | 16 |
17 #include <AvailabilityMacros.h> | 17 #include <AvailabilityMacros.h> |
18 #include <dlfcn.h> | 18 #include <dlfcn.h> |
19 #include <mach-o/dyld.h> | 19 #include <mach-o/dyld.h> |
20 #include <mach-o/dyld_images.h> | 20 #include <mach-o/dyld_images.h> |
21 #include <mach-o/getsect.h> | 21 #include <mach-o/getsect.h> |
22 #include <mach-o/ldsyms.h> | 22 #include <mach-o/ldsyms.h> |
23 #include <mach-o/loader.h> | 23 #include <mach-o/loader.h> |
24 #include <mach-o/nlist.h> | 24 #include <mach-o/nlist.h> |
25 #include <stdint.h> | 25 #include <stdint.h> |
26 | 26 |
27 #include "base/strings/stringprintf.h" | 27 #include "base/strings/stringprintf.h" |
28 #include "build/build_config.h" | 28 #include "build/build_config.h" |
| 29 #include "client/crashpad_info.h" |
29 #include "gtest/gtest.h" | 30 #include "gtest/gtest.h" |
30 #include "snapshot/mac/mach_o_image_segment_reader.h" | 31 #include "snapshot/mac/mach_o_image_segment_reader.h" |
31 #include "snapshot/mac/process_reader.h" | 32 #include "snapshot/mac/process_reader.h" |
32 #include "snapshot/mac/process_types.h" | 33 #include "snapshot/mac/process_types.h" |
33 #include "util/misc/uuid.h" | 34 #include "util/misc/uuid.h" |
34 #include "util/test/mac/dyld.h" | 35 #include "util/test/mac/dyld.h" |
35 | 36 |
36 // This file is responsible for testing MachOImageReader, | 37 // This file is responsible for testing MachOImageReader, |
37 // MachOImageSegmentReader, and MachOImageSymbolTableReader. | 38 // MachOImageSegmentReader, and MachOImageSymbolTableReader. |
38 | 39 |
(...skipping 486 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
525 ASSERT_NO_FATAL_FAILURE(ExpectSymbolTable(mh_execute_header, &image_reader)); | 526 ASSERT_NO_FATAL_FAILURE(ExpectSymbolTable(mh_execute_header, &image_reader)); |
526 } | 527 } |
527 | 528 |
528 TEST(MachOImageReader, Self_DyldImages) { | 529 TEST(MachOImageReader, Self_DyldImages) { |
529 ProcessReader process_reader; | 530 ProcessReader process_reader; |
530 ASSERT_TRUE(process_reader.Initialize(mach_task_self())); | 531 ASSERT_TRUE(process_reader.Initialize(mach_task_self())); |
531 | 532 |
532 uint32_t count = _dyld_image_count(); | 533 uint32_t count = _dyld_image_count(); |
533 ASSERT_GE(count, 1u); | 534 ASSERT_GE(count, 1u); |
534 | 535 |
| 536 size_t modules_with_crashpad_info = 0; |
| 537 |
535 for (uint32_t index = 0; index < count; ++index) { | 538 for (uint32_t index = 0; index < count; ++index) { |
536 const char* image_name = _dyld_get_image_name(index); | 539 const char* image_name = _dyld_get_image_name(index); |
537 SCOPED_TRACE(base::StringPrintf("index %u, image %s", index, image_name)); | 540 SCOPED_TRACE(base::StringPrintf("index %u, image %s", index, image_name)); |
538 | 541 |
539 // _dyld_get_image_header() is poorly-declared: it’s declared as returning | 542 // _dyld_get_image_header() is poorly-declared: it’s declared as returning |
540 // const mach_header* in both 32-bit and 64-bit environments, but in the | 543 // const mach_header* in both 32-bit and 64-bit environments, but in the |
541 // 64-bit environment, it should be const mach_header_64*. | 544 // 64-bit environment, it should be const mach_header_64*. |
542 const MachHeader* mach_header = | 545 const MachHeader* mach_header = |
543 reinterpret_cast<const MachHeader*>(_dyld_get_image_header(index)); | 546 reinterpret_cast<const MachHeader*>(_dyld_get_image_header(index)); |
544 mach_vm_address_t image_address = | 547 mach_vm_address_t image_address = |
545 reinterpret_cast<mach_vm_address_t>(mach_header); | 548 reinterpret_cast<mach_vm_address_t>(mach_header); |
546 | 549 |
547 MachOImageReader image_reader; | 550 MachOImageReader image_reader; |
548 ASSERT_TRUE( | 551 ASSERT_TRUE( |
549 image_reader.Initialize(&process_reader, image_address, image_name)); | 552 image_reader.Initialize(&process_reader, image_address, image_name)); |
550 | 553 |
551 uint32_t file_type = image_reader.FileType(); | 554 uint32_t file_type = image_reader.FileType(); |
552 if (index == 0) { | 555 if (index == 0) { |
553 EXPECT_EQ(implicit_cast<uint32_t>(MH_EXECUTE), file_type); | 556 EXPECT_EQ(implicit_cast<uint32_t>(MH_EXECUTE), file_type); |
554 } else { | 557 } else { |
555 EXPECT_TRUE(file_type == MH_DYLIB || file_type == MH_BUNDLE); | 558 EXPECT_TRUE(file_type == MH_DYLIB || file_type == MH_BUNDLE); |
556 } | 559 } |
557 | 560 |
558 intptr_t image_slide = _dyld_get_image_vmaddr_slide(index); | 561 intptr_t image_slide = _dyld_get_image_vmaddr_slide(index); |
559 ASSERT_NO_FATAL_FAILURE(ExpectMachImage( | 562 ASSERT_NO_FATAL_FAILURE(ExpectMachImage( |
560 mach_header, image_address, image_slide, &image_reader, false)); | 563 mach_header, image_address, image_slide, &image_reader, false)); |
561 | 564 |
562 ASSERT_NO_FATAL_FAILURE(ExpectSymbolTable(mach_header, &image_reader)); | 565 ASSERT_NO_FATAL_FAILURE(ExpectSymbolTable(mach_header, &image_reader)); |
| 566 |
| 567 process_types::CrashpadInfo crashpad_info; |
| 568 if (image_reader.GetCrashpadInfo(&crashpad_info)) { |
| 569 ++modules_with_crashpad_info; |
| 570 } |
563 } | 571 } |
564 | 572 |
| 573 EXPECT_GE(modules_with_crashpad_info, 1u); |
| 574 |
565 // Now that all of the modules have been verified, make sure that dyld itself | 575 // Now that all of the modules have been verified, make sure that dyld itself |
566 // can be read properly too. | 576 // can be read properly too. |
567 const struct dyld_all_image_infos* dyld_image_infos = | 577 const struct dyld_all_image_infos* dyld_image_infos = |
568 _dyld_get_all_image_infos(); | 578 _dyld_get_all_image_infos(); |
569 ASSERT_GE(dyld_image_infos->version, 1u); | 579 ASSERT_GE(dyld_image_infos->version, 1u); |
570 EXPECT_EQ(count, dyld_image_infos->infoArrayCount); | 580 EXPECT_EQ(count, dyld_image_infos->infoArrayCount); |
571 | 581 |
572 if (dyld_image_infos->version >= 2) { | 582 if (dyld_image_infos->version >= 2) { |
573 SCOPED_TRACE("dyld"); | 583 SCOPED_TRACE("dyld"); |
574 | 584 |
(...skipping 50 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
625 image_reader.UUID(&actual_uuid); | 635 image_reader.UUID(&actual_uuid); |
626 EXPECT_EQ(expected_uuid, actual_uuid); | 636 EXPECT_EQ(expected_uuid, actual_uuid); |
627 } | 637 } |
628 } | 638 } |
629 #endif | 639 #endif |
630 } | 640 } |
631 | 641 |
632 } // namespace | 642 } // namespace |
633 } // namespace test | 643 } // namespace test |
634 } // namespace crashpad | 644 } // namespace crashpad |
OLD | NEW |