Create sanitizer passes in a separate function

1 files changed, 26 insertions, 20 deletions

diff --git a/src/librustc_codegen_llvm/back/write.rs b/src/librustc_codegen_llvm/back/write.rs
index b664c78f658..6cc7b0b4b1e 100644
--- a/src/librustc_codegen_llvm/back/write.rs
+++ b/src/librustc_codegen_llvm/back/write.rs
@@ -363,26 +363,7 @@ pub(crate) unsafe fn optimize(cgcx: &CodegenContext<LlvmCodegenBackend>,
                 }
             }
 
-            if let Some(sanitizer) = &config.sanitizer {
-                let recover = config.sanitizer_recover.contains(sanitizer);
-                match sanitizer {
-                    Sanitizer::Address => {
-                        extra_passes.push(llvm::LLVMRustCreateAddressSanitizerFunctionPass(
-                                recover));
-                        extra_passes.push(llvm::LLVMRustCreateModuleAddressSanitizerPass(
-                                recover));
-                    }
-                    Sanitizer::Memory => {
-                        let track_origins = config.sanitizer_memory_track_origins as c_int;
-                        extra_passes.push(llvm::LLVMRustCreateMemorySanitizerPass(
-                                track_origins, recover));
-                    }
-                    Sanitizer::Thread => {
-                        extra_passes.push(llvm::LLVMRustCreateThreadSanitizerPass());
-                    }
-                    _ => {}
-                }
-            }
+            add_sanitizer_passes(config, &mut extra_passes);
 
             for pass_name in &cgcx.plugin_passes {
                 if let Some(pass) = find_pass(pass_name) {
@@ -469,6 +450,31 @@ pub(crate) unsafe fn optimize(cgcx: &CodegenContext<LlvmCodegenBackend>,
     Ok(())
 }
 
+unsafe fn add_sanitizer_passes(config: &ModuleConfig,
+                               passes: &mut Vec<&'static mut llvm::Pass>) {
+
+    let sanitizer = match &config.sanitizer {
+        None => return,
+        Some(s) => s,
+    };
+
+    let recover = config.sanitizer_recover.contains(sanitizer);
+    match sanitizer {
+        Sanitizer::Address => {
+            passes.push(llvm::LLVMRustCreateAddressSanitizerFunctionPass(recover));
+            passes.push(llvm::LLVMRustCreateModuleAddressSanitizerPass(recover));
+        }
+        Sanitizer::Memory => {
+            let track_origins = config.sanitizer_memory_track_origins as c_int;
+            passes.push(llvm::LLVMRustCreateMemorySanitizerPass(track_origins, recover));
+        }
+        Sanitizer::Thread => {
+            passes.push(llvm::LLVMRustCreateThreadSanitizerPass());
+        }
+        Sanitizer::Leak => {}
+    }
+}
+
 pub(crate) unsafe fn codegen(cgcx: &CodegenContext<LlvmCodegenBackend>,
                   diag_handler: &Handler,
                   module: ModuleCodegen<ModuleLlvm>,