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     // Returns an error for unsupported symbols, and None if this symbol
  12     // should become a NULL pointer (pretend it does not exist).
  13     pub fn from_str(name: &str) -> InterpResult<'static, Option<Dlsym>> {
  14         use self::Dlsym::*;
  15         Ok(match name {
  16             "getentropy" => Some(GetEntropy),
  17             "__pthread_get_minstack" => None,
  18             _ =>
  19                 throw_unsup_format!("Unsupported dlsym: {}", name),
  20         })
  21     }
  22 }
  23
  24 impl<'mir, 'tcx> EvalContextExt<'mir, 'tcx> for crate::MiriEvalContext<'mir, 'tcx> {}
  25 pub trait EvalContextExt<'mir, 'tcx: 'mir>: crate::MiriEvalContextExt<'mir, 'tcx> {
  26     fn call_dlsym(
  27         &mut self,
  28         dlsym: Dlsym,
  29         args: &[OpTy<'tcx, Tag>],
  30         dest: Option<PlaceTy<'tcx, Tag>>,
  31         ret: Option<mir::BasicBlock>,
  32     ) -> InterpResult<'tcx> {
  33         use self::Dlsym::*;
  34
  35         let this = self.eval_context_mut();
  36
  37         let dest = dest.expect("we don't support any diverging dlsym");
  38         let ret = ret.expect("dest is `Some` but ret is `None`");
  39
  40         match dlsym {
  41             GetEntropy => {
  42                 let ptr = this.read_scalar(args[0])?.not_undef()?;
  43                 let len = this.read_scalar(args[1])?.to_usize(this)?;
  44                 this.gen_random(ptr, len as usize)?;
  45                 this.write_null(dest)?;
  46             }
  47         }
  48
  49         this.goto_block(Some(ret))?;
  50         this.dump_place(*dest);
  51         Ok(())
  52     }
  53 }