src/shims/dlsym.rs

   1 use rustc::mir;
   2
   3 use crate::*;
   4
   5 #[derive(Debug, Copy, Clone)]
   6 pub enum Dlsym {
   7     GetEntropy,
   8 }
   9
  10 impl Dlsym {
  11     pub fn from_str(name: &str) -> Option<Dlsym> {
  12         use self::Dlsym::*;
  13         Some(match name {
  14             "getentropy" => GetEntropy,
  15             _ => return None,
  16         })
  17     }
  18 }
  19
  20 impl<'mir, 'tcx> EvalContextExt<'mir, 'tcx> for crate::MiriEvalContext<'mir, 'tcx> {}
  21 pub trait EvalContextExt<'mir, 'tcx: 'mir>: crate::MiriEvalContextExt<'mir, 'tcx> {
  22     fn call_dlsym(
  23         &mut self,
  24         dlsym: Dlsym,
  25         args: &[OpTy<'tcx, Tag>],
  26         dest: Option<PlaceTy<'tcx, Tag>>,
  27         ret: Option<mir::BasicBlock>,
  28     ) -> InterpResult<'tcx> {
  29         use self::Dlsym::*;
  30
  31         let this = self.eval_context_mut();
  32
  33         let dest = dest.expect("we don't support any diverging dlsym");
  34         let ret = ret.expect("dest is `Some` but ret is `None`");
  35
  36         match dlsym {
  37             GetEntropy => {
  38                 let ptr = this.read_scalar(args[0])?.not_undef()?;
  39                 let len = this.read_scalar(args[1])?.to_usize(this)?;
  40                 this.gen_random(len as usize, ptr)?;
  41                 this.write_null(dest)?;
  42             }
  43         }
  44
  45         this.goto_block(Some(ret))?;
  46         this.dump_place(*dest);
  47         Ok(())
  48     }
  49 }