From 959bc7638aebb27fcdf7079daf0d0cafe845f56a Mon Sep 17 00:00:00 2001
From: Dominic Chen <d.c.ddcc@gmail.com>
Date: Tue, 2 Aug 2016 18:01:26 -0700
Subject: support pre-assigning indexes for functions that are called
 indirectly (#616)

This patch adds support for an ".indidx" primitive that pre-assigns table indexes for functions that are called indirectly. It is used by the upstream LLVM WebAssembly backend to support fine-grained control-flow integrity for indirect function calls by emitting instrumentation at each indirect call site to check that the destination index is within certain ranges that correspond to disjoint equivalence classes of indirect call targets. The reason that this primitive is necessary is because the layout of the table section isn't determined until the WebAssembly linker is executed, but indirect function to table index mappings need to be known when opt is executed to generate the correct range checking in the LLVM IR.
---
 src/wasm-linker.cpp | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

(limited to 'src/wasm-linker.cpp')

diff --git a/src/wasm-linker.cpp b/src/wasm-linker.cpp
index 628e24ecc..3ff90cc0a 100644
--- a/src/wasm-linker.cpp
+++ b/src/wasm-linker.cpp
@@ -112,9 +112,30 @@ void Linker::layout() {
   for (Name name : out.globls) exportFunction(name, false);
   for (Name name : out.initializerFunctions) exportFunction(name, true);
 
+  // Pre-assign the function indexes
+  for (auto& pair : out.indirectIndexes) {
+    if (functionIndexes.count(pair.first) != 0 ||
+        functionNames.count(pair.second) != 0) {
+      Fatal() << "Function " << pair.first << " already has an index " <<
+          functionIndexes[pair.first] << " while setting index " << pair.second;
+    }
+    if (debug) {
+      std::cerr << "pre-assigned function index: " << pair.first << ": "
+                << pair.second << '\n';
+    }
+    functionIndexes[pair.first] = pair.second;
+    functionNames[pair.second] = pair.first;
+  }
+
+  // Emit the pre-assigned function names in sorted order
+  for (const auto& P : functionNames) {
+    out.wasm.table.names.push_back(P.second);
+  }
+
   auto ensureFunctionIndex = [this](Name name) {
     if (functionIndexes.count(name) == 0) {
       functionIndexes[name] = out.wasm.table.names.size();
+      functionNames[functionIndexes[name]] = name;
       out.wasm.table.names.push_back(name);
       if (debug) {
         std::cerr << "function index: " << name << ": "
-- 
cgit v1.2.3