Extend relocation packing to cover arm64.

tools/relocation_packer/src/elf_file_unittest.cc

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "elf_file.h"
 
 #include <limits.h>
 #include <stdio.h>
 #include <unistd.h>
 #include <string>
 #include <vector>
 #include "debug.h"
+#include "elf_traits.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 // Macro stringification.
 // https://gcc.gnu.org/onlinedocs/cpp/Stringification.html
 #define XSTR(S) STR(S)
 #define STR(S) #S
 
 namespace {
 
 void GetDataFilePath(const char* name, std::string* path) {
@@ skipping 40 matching lines @@
   } while (bytes > 0);
 
   ASSERT_EQ(0, fclose(testfile));
   ASSERT_EQ(0, fseek(temporary, 0, SEEK_SET));
   ASSERT_EQ(0, lseek(fileno(temporary), 0, SEEK_SET));
 
   *stream = temporary;
 }
 
 void OpenRelocsTestFiles(FILE** relocs_so, FILE** packed_relocs_so) {
-  OpenRelocsTestFile("elf_file_unittest_relocs.so", relocs_so);
-  OpenRelocsTestFile("elf_file_unittest_relocs_packed.so", packed_relocs_so);
+  const char* arch = NULL;
+  if (ELF::kMachine == EM_ARM) {
+    arch = "arm32";
+  } else if (ELF::kMachine == EM_AARCH64) {
+    arch = "arm64";
+  }
+  ASSERT_FALSE(arch == NULL);
+
+  const std::string base = std::string("elf_file_unittest_relocs_") + arch;
+  const std::string relocs = base + ".so";
+  const std::string packed_relocs = base + "_packed.so";
+
+  OpenRelocsTestFile(relocs.c_str(), relocs_so);
+  OpenRelocsTestFile(packed_relocs.c_str(), packed_relocs_so);
 }
 
 void CloseRelocsTestFile(FILE* temporary) {
   fclose(temporary);
 }
 
 void CloseRelocsTestFiles(FILE* relocs_so, FILE* packed_relocs_so) {
   CloseRelocsTestFile(relocs_so);
   CloseRelocsTestFile(packed_relocs_so);
 }
@@ skipping 57 matching lines @@
   EXPECT_FALSE(elf_file.PackRelocations());
 
   // Unpack golden relocations, and check files are now identical.
   EXPECT_TRUE(elf_file.UnpackRelocations());
   CheckFileContentsEqual(packed_relocs_so, relocs_so);
 
   CloseRelocsTestFiles(relocs_so, packed_relocs_so);
 }
 
 }  // namespace relocation_packer