[BOLT] Don't choke on injected functions' IO map

[rafauler]

Summary

AddressMap would fail lookup for injected functions and crash
BOLT. Fix that.

[jobnoorman]

LGTM

[maksfb]

bolt/test/X86/patch-entries.c

9–11

nit

bolt/lib/Core/BinaryFunction.cpp

for (BinaryBasicBlock *const BB : FF) {
       if (!BC.HasRelocations) {
         if (BB->isSplit())
           assert(FragmentBaseAddress == FF.getAddress());
         else
           assert(FragmentBaseAddress == getOutputAddress());
         (void)FragmentBaseAddress;
       }
 
-      const uint64_t BBAddress =
-          *BC.getIOAddressMap().lookup(BB->getInputOffset() + getAddress());
+      // Injected functions likely will fail lookup, as they have no
+      // input range. Just assign the BB the output address of the
+      // function.
+      auto MaybeBBAddress =
+          BC.getIOAddressMap().lookup(BB->getInputOffset() + getAddress());
+      const uint64_t BBAddress = MaybeBBAddress  ? *MaybeBBAddress
+                                 : BB->isSplit() ? FF.getAddress()
+                                                 : getOutputAddress();
       BB->setOutputStartAddress(BBAddress);
 
       if (PrevBB)
         PrevBB->setOutputEndAddress(BBAddress);
       PrevBB = BB;
     }
 
     PrevBB->setOutputEndAddress(PrevBB->isSplit()

bolt/test/X86/patch-entries.c

+// Checking crashes against injected binary functions created by patch
+// entries pass and debug info turned on. In these cases, we were
+// trying to fetch input to output maps on injected functions and
+// crashing.
+
+// REQUIRES: system-linux
+
+// RUN: %clang %cflags -no-pie -g %s -fuse-ld=lld -o %t.exe -Wl,-q
+// RUN: llvm-bolt -relocs %t.exe -o %t.out \
+// RUN:   -update-debug-sections \
+// RUN:  --force-patch
+
+#include <stdio.h>
+
+static void foo() { printf("foo\n"); }
+
+int main() {
+  foo();
+  return 0;
+}