47 lines
1.5 KiB
Rust
47 lines
1.5 KiB
Rust
// Copyright 2015 The Rust Project Developers. See the COPYRIGHT
|
|
// file at the top-level directory of this distribution and at
|
|
// http://rust-lang.org/COPYRIGHT.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
|
|
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
|
|
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
|
|
// option. This file may not be copied, modified, or distributed
|
|
// except according to those terms.
|
|
|
|
//! This pass removes the unwind branch of all the terminators when the no-landing-pads option is
|
|
//! specified.
|
|
|
|
use rustc::ty::TyCtxt;
|
|
use rustc::mir::*;
|
|
use rustc::mir::visit::MutVisitor;
|
|
use transform::{MirPass, MirSource};
|
|
|
|
pub struct NoLandingPads;
|
|
|
|
impl MirPass for NoLandingPads {
|
|
fn run_pass<'a, 'tcx>(&self,
|
|
tcx: TyCtxt<'a, 'tcx, 'tcx>,
|
|
_: MirSource,
|
|
mir: &mut Mir<'tcx>) {
|
|
no_landing_pads(tcx, mir)
|
|
}
|
|
}
|
|
|
|
pub fn no_landing_pads<'a, 'tcx>(tcx: TyCtxt<'a, 'tcx, 'tcx>, mir: &mut Mir<'tcx>) {
|
|
if tcx.sess.no_landing_pads() {
|
|
NoLandingPads.visit_mir(mir);
|
|
}
|
|
}
|
|
|
|
impl<'tcx> MutVisitor<'tcx> for NoLandingPads {
|
|
fn visit_terminator(&mut self,
|
|
bb: BasicBlock,
|
|
terminator: &mut Terminator<'tcx>,
|
|
location: Location) {
|
|
if let Some(unwind) = terminator.kind.unwind_mut() {
|
|
unwind.take();
|
|
}
|
|
self.super_terminator(bb, terminator, location);
|
|
}
|
|
}
|