[PECOFF] Add a utility function to add DIR32 relocation.

llvm-svn: 202217

1 files changed, 13 insertions, 4 deletions

diff --git a/lld/lib/ReaderWriter/PECOFF/Pass.cpp b/lld/lib/ReaderWriter/PECOFF/Pass.cpp
index da6aa51544a..ed4880e4f4b 100644
--- a/lld/lib/ReaderWriter/PECOFF/Pass.cpp
+++ b/lld/lib/ReaderWriter/PECOFF/Pass.cpp
@@ -16,13 +16,22 @@
 namespace lld {
 namespace pecoff {
 
+static void addReloc(COFFBaseDefinedAtom *atom, const Atom *target,
+                     size_t offsetInAtom, Reference::KindValue relType) {
+  std::unique_ptr<COFFReference> ref(
+      new COFFReference(target, offsetInAtom, relType));
+  atom->addReference(std::move(ref));
+}
+
+void addDir32Reloc(COFFBaseDefinedAtom *atom, const Atom *target,
+                   size_t offsetInAtom) {
+  addReloc(atom, target, offsetInAtom, llvm::COFF::IMAGE_REL_I386_DIR32);
+}
+
 void addDir32NBReloc(COFFBaseDefinedAtom *atom, const Atom *target,
                      size_t offsetInAtom) {
-  std::unique_ptr<COFFReference> ref(new COFFReference(
-      target, offsetInAtom, llvm::COFF::IMAGE_REL_I386_DIR32NB));
-  atom->addReference(std::move(ref));
+  addReloc(atom, target, offsetInAtom, llvm::COFF::IMAGE_REL_I386_DIR32NB);
 }
 
 } // end namespace pecoff
 } // end namespace lld
-