2 use rustc::ty::layout::{Align, LayoutOf, Size};
3 use rustc::hir::def_id::DefId;
11 impl<'a, 'mir, 'tcx> EvalContextExt<'a, 'mir, 'tcx> for crate::MiriEvalContext<'a, 'mir, 'tcx> {}
12 pub trait EvalContextExt<'a, 'mir, 'tcx: 'a + 'mir>: crate::MiriEvalContextExt<'a, 'mir, 'tcx> {
15 instance: ty::Instance<'tcx>,
16 args: &[OpTy<'tcx, Tag>],
17 dest: Option<PlaceTy<'tcx, Tag>>,
18 ret: Option<mir::BasicBlock>,
19 ) -> EvalResult<'tcx, Option<&'mir mir::Mir<'tcx>>> {
20 let this = self.eval_context_mut();
21 trace!("eval_fn_call: {:#?}, {:?}", instance, dest.map(|place| *place));
23 // First, run the common hooks also supported by CTFE.
24 if this.hook_fn(instance, args, dest)? {
25 this.goto_block(ret)?;
28 // There are some more lang items we want to hook that CTFE does not hook (yet).
29 if this.tcx.lang_items().align_offset_fn() == Some(instance.def.def_id()) {
30 // FIXME: return a real value in case the target allocation has an
31 // alignment bigger than the one requested.
32 let n = u128::max_value();
33 let dest = dest.unwrap();
34 let n = this.truncate(n, dest.layout);
35 this.write_scalar(Scalar::from_uint(n, dest.layout.size), dest)?;
36 this.goto_block(ret)?;
40 // Try to see if we can do something about foreign items.
41 if this.tcx.is_foreign_item(instance.def_id()) {
42 // An external function that we cannot find MIR for, but we can still run enough
43 // of them to make miri viable.
44 this.emulate_foreign_item(instance.def_id(), args, dest, ret)?;
45 // `goto_block` already handled.
49 // Otherwise, load the MIR.
50 Ok(Some(this.load_mir(instance.def)?))
58 let this = self.eval_context_mut();
59 let tcx = &{this.tcx.tcx};
61 Scalar::from_int(0, this.pointer_size())
63 let align = this.tcx.data_layout.pointer_align.abi;
64 let ptr = this.memory_mut().allocate(Size::from_bytes(size), align, MiriMemoryKind::C.into());
66 // We just allocated this, the access cannot fail
68 .get_mut(ptr.alloc_id).unwrap()
69 .write_repeat(tcx, ptr, 0, Size::from_bytes(size)).unwrap();
78 ) -> EvalResult<'tcx> {
79 let this = self.eval_context_mut();
80 if !ptr.is_null_ptr(this) {
81 this.memory_mut().deallocate(
84 MiriMemoryKind::C.into(),
94 ) -> EvalResult<'tcx, Scalar<Tag>> {
95 let this = self.eval_context_mut();
96 let align = this.tcx.data_layout.pointer_align.abi;
97 if old_ptr.is_null_ptr(this) {
99 Ok(Scalar::from_int(0, this.pointer_size()))
101 let new_ptr = this.memory_mut().allocate(
102 Size::from_bytes(new_size),
104 MiriMemoryKind::C.into()
106 Ok(Scalar::Ptr(new_ptr))
109 let old_ptr = old_ptr.to_ptr()?;
110 let memory = this.memory_mut();
111 let old_size = Size::from_bytes(memory.get(old_ptr.alloc_id)?.bytes.len() as u64);
115 Some((old_size, align)),
116 MiriMemoryKind::C.into(),
118 Ok(Scalar::from_int(0, this.pointer_size()))
120 let new_ptr = memory.reallocate(
124 Size::from_bytes(new_size),
126 MiriMemoryKind::C.into(),
128 Ok(Scalar::Ptr(new_ptr))
133 /// Emulates calling a foreign item, failing if the item is not supported.
134 /// This function will handle `goto_block` if needed.
135 fn emulate_foreign_item(
138 args: &[OpTy<'tcx, Tag>],
139 dest: Option<PlaceTy<'tcx, Tag>>,
140 ret: Option<mir::BasicBlock>,
141 ) -> EvalResult<'tcx> {
142 let this = self.eval_context_mut();
143 let attrs = this.tcx.get_attrs(def_id);
144 let link_name = match attr::first_attr_value_str_by_name(&attrs, "link_name") {
145 Some(name) => name.as_str(),
146 None => this.tcx.item_name(def_id).as_str(),
148 // Strip linker suffixes (seen on 32-bit macOS).
149 let link_name = link_name.get().trim_end_matches("$UNIX2003");
150 let tcx = &{this.tcx.tcx};
152 // First: functions that could diverge.
154 "__rust_start_panic" | "panic_impl" => {
155 return err!(MachineError("the evaluated program panicked".to_string()));
157 _ => if dest.is_none() {
158 return err!(Unimplemented(
159 format!("can't call diverging foreign function: {}", link_name),
164 // Next: functions that assume a ret and dest.
165 let dest = dest.expect("we already checked for a dest");
166 let ret = ret.expect("dest is `Some` but ret is `None`");
169 let size = this.read_scalar(args[0])?.to_usize(this)?;
170 let res = this.malloc(size, /*zero_init:*/ false);
171 this.write_scalar(res, dest)?;
174 let items = this.read_scalar(args[0])?.to_usize(this)?;
175 let len = this.read_scalar(args[1])?.to_usize(this)?;
176 let size = items.checked_mul(len).ok_or_else(|| InterpError::Overflow(mir::BinOp::Mul))?;
177 let res = this.malloc(size, /*zero_init:*/ true);
178 this.write_scalar(res, dest)?;
180 "posix_memalign" => {
181 let ret = this.deref_operand(args[0])?;
182 let align = this.read_scalar(args[1])?.to_usize(this)?;
183 let size = this.read_scalar(args[2])?.to_usize(this)?;
184 // Align must be power of 2, and also at least ptr-sized (POSIX rules).
185 if !align.is_power_of_two() {
186 return err!(HeapAllocNonPowerOfTwoAlignment(align));
188 if align < this.pointer_size().bytes() {
189 return err!(MachineError(format!(
190 "posix_memalign: alignment must be at least the size of a pointer, but is {}",
195 this.write_null(ret.into())?;
197 let ptr = this.memory_mut().allocate(
198 Size::from_bytes(size),
199 Align::from_bytes(align).unwrap(),
200 MiriMemoryKind::C.into()
202 this.write_scalar(Scalar::Ptr(ptr), ret.into())?;
204 this.write_null(dest)?;
207 let ptr = this.read_scalar(args[0])?.not_undef()?;
211 let old_ptr = this.read_scalar(args[0])?.not_undef()?;
212 let new_size = this.read_scalar(args[1])?.to_usize(this)?;
213 let res = this.realloc(old_ptr, new_size)?;
214 this.write_scalar(res, dest)?;
218 let size = this.read_scalar(args[0])?.to_usize(this)?;
219 let align = this.read_scalar(args[1])?.to_usize(this)?;
221 return err!(HeapAllocZeroBytes);
223 if !align.is_power_of_two() {
224 return err!(HeapAllocNonPowerOfTwoAlignment(align));
226 let ptr = this.memory_mut()
228 Size::from_bytes(size),
229 Align::from_bytes(align).unwrap(),
230 MiriMemoryKind::Rust.into()
232 this.write_scalar(Scalar::Ptr(ptr), dest)?;
234 "__rust_alloc_zeroed" => {
235 let size = this.read_scalar(args[0])?.to_usize(this)?;
236 let align = this.read_scalar(args[1])?.to_usize(this)?;
238 return err!(HeapAllocZeroBytes);
240 if !align.is_power_of_two() {
241 return err!(HeapAllocNonPowerOfTwoAlignment(align));
243 let ptr = this.memory_mut()
245 Size::from_bytes(size),
246 Align::from_bytes(align).unwrap(),
247 MiriMemoryKind::Rust.into()
250 .get_mut(ptr.alloc_id)?
251 .write_repeat(tcx, ptr, 0, Size::from_bytes(size))?;
252 this.write_scalar(Scalar::Ptr(ptr), dest)?;
254 "__rust_dealloc" => {
255 let ptr = this.read_scalar(args[0])?.to_ptr()?;
256 let old_size = this.read_scalar(args[1])?.to_usize(this)?;
257 let align = this.read_scalar(args[2])?.to_usize(this)?;
259 return err!(HeapAllocZeroBytes);
261 if !align.is_power_of_two() {
262 return err!(HeapAllocNonPowerOfTwoAlignment(align));
264 this.memory_mut().deallocate(
266 Some((Size::from_bytes(old_size), Align::from_bytes(align).unwrap())),
267 MiriMemoryKind::Rust.into(),
270 "__rust_realloc" => {
271 let ptr = this.read_scalar(args[0])?.to_ptr()?;
272 let old_size = this.read_scalar(args[1])?.to_usize(this)?;
273 let align = this.read_scalar(args[2])?.to_usize(this)?;
274 let new_size = this.read_scalar(args[3])?.to_usize(this)?;
275 if old_size == 0 || new_size == 0 {
276 return err!(HeapAllocZeroBytes);
278 if !align.is_power_of_two() {
279 return err!(HeapAllocNonPowerOfTwoAlignment(align));
281 let new_ptr = this.memory_mut().reallocate(
283 Size::from_bytes(old_size),
284 Align::from_bytes(align).unwrap(),
285 Size::from_bytes(new_size),
286 Align::from_bytes(align).unwrap(),
287 MiriMemoryKind::Rust.into(),
289 this.write_scalar(Scalar::Ptr(new_ptr), dest)?;
293 let sys_getrandom = this.eval_path_scalar(&["libc", "SYS_getrandom"])?
294 .expect("Failed to get libc::SYS_getrandom")
297 // `libc::syscall(NR_GETRANDOM, buf.as_mut_ptr(), buf.len(), GRND_NONBLOCK)`
298 // is called if a `HashMap` is created the regular way (e.g. HashMap<K, V>).
299 match this.read_scalar(args[0])?.to_usize(this)? {
300 id if id == sys_getrandom => {
301 let ptr = this.read_scalar(args[1])?.to_ptr()?;
302 let len = this.read_scalar(args[2])?.to_usize(this)?;
304 // The only supported flags are GRND_RANDOM and GRND_NONBLOCK,
305 // neither of which have any effect on our current PRNG
306 let _flags = this.read_scalar(args[3])?.to_i32()?;
309 let data = gen_random(this, len as usize)?;
310 this.memory_mut().get_mut(ptr.alloc_id)?
311 .write_bytes(tcx, ptr, &data)?;
314 this.write_scalar(Scalar::from_uint(len, dest.layout.size), dest)?;
317 return err!(Unimplemented(
318 format!("miri does not support syscall ID {}", id),
325 let _handle = this.read_scalar(args[0])?;
326 let symbol = this.read_scalar(args[1])?.to_ptr()?;
327 let symbol_name = this.memory().get(symbol.alloc_id)?.read_c_str(tcx, symbol)?;
328 let err = format!("bad c unicode symbol: {:?}", symbol_name);
329 let symbol_name = ::std::str::from_utf8(symbol_name).unwrap_or(&err);
330 return err!(Unimplemented(format!(
331 "miri does not support dynamically loading libraries (requested symbol: {})",
336 "__rust_maybe_catch_panic" => {
337 // fn __rust_maybe_catch_panic(
340 // data_ptr: *mut usize,
341 // vtable_ptr: *mut usize,
343 // We abort on panic, so not much is going on here, but we still have to call the closure.
344 let f = this.read_scalar(args[0])?.to_ptr()?;
345 let data = this.read_scalar(args[1])?.not_undef()?;
346 let f_instance = this.memory().get_fn(f)?;
347 this.write_null(dest)?;
348 trace!("__rust_maybe_catch_panic: {:?}", f_instance);
350 // Now we make a function call.
351 // TODO: consider making this reusable? `InterpretCx::step` does something similar
352 // for the TLS destructors, and of course `eval_main`.
353 let mir = this.load_mir(f_instance.def)?;
354 let ret_place = MPlaceTy::dangling(this.layout_of(this.tcx.mk_unit())?, this).into();
355 this.push_stack_frame(
360 // Directly return to caller.
361 StackPopCleanup::Goto(Some(ret)),
363 let mut args = this.frame().mir.args_iter();
365 let arg_local = args.next().ok_or_else(||
366 InterpError::AbiViolation(
367 "Argument to __rust_maybe_catch_panic does not take enough arguments."
371 let arg_dest = this.eval_place(&mir::Place::Base(mir::PlaceBase::Local(arg_local)))?;
372 this.write_scalar(data, arg_dest)?;
374 assert!(args.next().is_none(), "__rust_maybe_catch_panic argument has more arguments than expected");
376 // We ourselves will return `0`, eventually (because we will not return if we paniced).
377 this.write_null(dest)?;
379 // Don't fall through, we do *not* want to `goto_block`!
384 let left = this.read_scalar(args[0])?.not_undef()?;
385 let right = this.read_scalar(args[1])?.not_undef()?;
386 let n = Size::from_bytes(this.read_scalar(args[2])?.to_usize(this)?);
389 let left_bytes = this.memory().read_bytes(left, n)?;
390 let right_bytes = this.memory().read_bytes(right, n)?;
392 use std::cmp::Ordering::*;
393 match left_bytes.cmp(right_bytes) {
401 Scalar::from_int(result, Size::from_bits(32)),
407 let ptr = this.read_scalar(args[0])?.not_undef()?;
408 let val = this.read_scalar(args[1])?.to_i32()? as u8;
409 let num = this.read_scalar(args[2])?.to_usize(this)?;
410 if let Some(idx) = this.memory().read_bytes(ptr, Size::from_bytes(num))?
411 .iter().rev().position(|&c| c == val)
413 let new_ptr = ptr.ptr_offset(Size::from_bytes(num - idx as u64 - 1), this)?;
414 this.write_scalar(new_ptr, dest)?;
416 this.write_null(dest)?;
421 let ptr = this.read_scalar(args[0])?.not_undef()?;
422 let val = this.read_scalar(args[1])?.to_i32()? as u8;
423 let num = this.read_scalar(args[2])?.to_usize(this)?;
426 .read_bytes(ptr, Size::from_bytes(num))?
428 .position(|&c| c == val);
429 if let Some(idx) = idx {
430 let new_ptr = ptr.ptr_offset(Size::from_bytes(idx as u64), this)?;
431 this.write_scalar(new_ptr, dest)?;
433 this.write_null(dest)?;
439 let name_ptr = this.read_scalar(args[0])?.to_ptr()?;
440 let name = this.memory().get(name_ptr.alloc_id)?.read_c_str(tcx, name_ptr)?;
441 match this.machine.env_vars.get(name) {
442 Some(&var) => Scalar::Ptr(var),
443 None => Scalar::ptr_null(&*this.tcx),
446 this.write_scalar(result, dest)?;
450 let mut success = None;
452 let name_ptr = this.read_scalar(args[0])?.not_undef()?;
453 if !name_ptr.is_null_ptr(this) {
454 let name_ptr = name_ptr.to_ptr()?;
457 .get(name_ptr.alloc_id)?
458 .read_c_str(tcx, name_ptr)?
460 if !name.is_empty() && !name.contains(&b'=') {
461 success = Some(this.machine.env_vars.remove(&name));
465 if let Some(old) = success {
466 if let Some(var) = old {
467 this.memory_mut().deallocate(var, None, MiriMemoryKind::Env.into())?;
469 this.write_null(dest)?;
471 this.write_scalar(Scalar::from_int(-1, dest.layout.size), dest)?;
478 let name_ptr = this.read_scalar(args[0])?.not_undef()?;
479 let value_ptr = this.read_scalar(args[1])?.to_ptr()?;
480 let value = this.memory().get(value_ptr.alloc_id)?.read_c_str(tcx, value_ptr)?;
481 if !name_ptr.is_null_ptr(this) {
482 let name_ptr = name_ptr.to_ptr()?;
483 let name = this.memory().get(name_ptr.alloc_id)?.read_c_str(tcx, name_ptr)?;
484 if !name.is_empty() && !name.contains(&b'=') {
485 new = Some((name.to_owned(), value.to_owned()));
489 if let Some((name, value)) = new {
490 // `+1` for the null terminator.
491 let value_copy = this.memory_mut().allocate(
492 Size::from_bytes((value.len() + 1) as u64),
493 Align::from_bytes(1).unwrap(),
494 MiriMemoryKind::Env.into(),
497 let alloc = this.memory_mut().get_mut(value_copy.alloc_id)?;
498 alloc.write_bytes(tcx, value_copy, &value)?;
499 let trailing_zero_ptr = value_copy.offset(
500 Size::from_bytes(value.len() as u64),
503 alloc.write_bytes(tcx, trailing_zero_ptr, &[0])?;
505 if let Some(var) = this.machine.env_vars.insert(
510 this.memory_mut().deallocate(var, None, MiriMemoryKind::Env.into())?;
512 this.write_null(dest)?;
514 this.write_scalar(Scalar::from_int(-1, dest.layout.size), dest)?;
519 let fd = this.read_scalar(args[0])?.to_i32()?;
520 let buf = this.read_scalar(args[1])?.not_undef()?;
521 let n = this.read_scalar(args[2])?.to_usize(&*this.tcx)?;
522 trace!("Called write({:?}, {:?}, {:?})", fd, buf, n);
523 let result = if fd == 1 || fd == 2 {
525 use std::io::{self, Write};
527 let buf_cont = this.memory().read_bytes(buf, Size::from_bytes(n))?;
528 // We need to flush to make sure this actually appears on the screen
529 let res = if fd == 1 {
530 // Stdout is buffered, flush to make sure it appears on the screen.
531 // This is the write() syscall of the interpreted program, we want it
532 // to correspond to a write() syscall on the host -- there is no good
533 // in adding extra buffering here.
534 let res = io::stdout().write(buf_cont);
535 io::stdout().flush().unwrap();
538 // No need to flush, stderr is not buffered.
539 io::stderr().write(buf_cont)
546 eprintln!("Miri: Ignored output to FD {}", fd);
547 // Pretend it all went well.
550 // Now, `result` is the value we return back to the program.
552 Scalar::from_int(result, dest.layout.size),
558 let ptr = this.read_scalar(args[0])?.to_ptr()?;
559 let n = this.memory().get(ptr.alloc_id)?.read_c_str(tcx, ptr)?.len();
560 this.write_scalar(Scalar::from_uint(n as u64, dest.layout.size), dest)?;
563 // Some things needed for `sys::thread` initialization to go through.
564 "signal" | "sigaction" | "sigaltstack" => {
565 this.write_scalar(Scalar::from_int(0, dest.layout.size), dest)?;
569 let name = this.read_scalar(args[0])?.to_i32()?;
571 trace!("sysconf() called with name {}", name);
572 // Cache the sysconf integers via Miri's global cache.
574 (&["libc", "_SC_PAGESIZE"], Scalar::from_int(4096, dest.layout.size)),
575 (&["libc", "_SC_GETPW_R_SIZE_MAX"], Scalar::from_int(-1, dest.layout.size)),
576 (&["libc", "_SC_NPROCESSORS_ONLN"], Scalar::from_int(1, dest.layout.size)),
578 let mut result = None;
579 for &(path, path_value) in paths {
580 if let Some(val) = this.eval_path_scalar(path)? {
581 let val = val.to_i32()?;
583 result = Some(path_value);
589 if let Some(result) = result {
590 this.write_scalar(result, dest)?;
592 return err!(Unimplemented(
593 format!("Unimplemented sysconf name: {}", name),
599 this.write_null(dest)?;
602 // Hook pthread calls that go to the thread-local storage memory subsystem.
603 "pthread_key_create" => {
604 let key_ptr = this.read_scalar(args[0])?.to_ptr()?;
606 // Extract the function type out of the signature (that seems easier than constructing it ourselves).
607 let dtor = match this.read_scalar(args[1])?.not_undef()? {
608 Scalar::Ptr(dtor_ptr) => Some(this.memory().get_fn(dtor_ptr)?),
609 Scalar::Bits { bits: 0, size } => {
610 assert_eq!(size as u64, this.memory().pointer_size().bytes());
613 Scalar::Bits { .. } => return err!(ReadBytesAsPointer),
616 // Figure out how large a pthread TLS key actually is.
617 // This is `libc::pthread_key_t`.
618 let key_type = args[0].layout.ty
620 .ok_or_else(|| InterpError::AbiViolation("wrong signature used for `pthread_key_create`: first argument must be a raw pointer.".to_owned()))?
622 let key_layout = this.layout_of(key_type)?;
624 // Create key and write it into the memory where `key_ptr` wants it.
625 let key = this.machine.tls.create_tls_key(dtor, tcx) as u128;
626 if key_layout.size.bits() < 128 && key >= (1u128 << key_layout.size.bits() as u128) {
627 return err!(OutOfTls);
630 this.memory().check_align(key_ptr.into(), key_layout.align.abi)?;
631 this.memory_mut().get_mut(key_ptr.alloc_id)?.write_scalar(
634 Scalar::from_uint(key, key_layout.size).into(),
638 // Return success (`0`).
639 this.write_null(dest)?;
641 "pthread_key_delete" => {
642 let key = this.read_scalar(args[0])?.to_bits(args[0].layout.size)?;
643 this.machine.tls.delete_tls_key(key)?;
644 // Return success (0)
645 this.write_null(dest)?;
647 "pthread_getspecific" => {
648 let key = this.read_scalar(args[0])?.to_bits(args[0].layout.size)?;
649 let ptr = this.machine.tls.load_tls(key)?;
650 this.write_scalar(ptr, dest)?;
652 "pthread_setspecific" => {
653 let key = this.read_scalar(args[0])?.to_bits(args[0].layout.size)?;
654 let new_ptr = this.read_scalar(args[1])?.not_undef()?;
655 this.machine.tls.store_tls(key, new_ptr)?;
657 // Return success (`0`).
658 this.write_null(dest)?;
661 // Determine stack base address.
662 "pthread_attr_init" | "pthread_attr_destroy" | "pthread_attr_get_np" |
663 "pthread_getattr_np" | "pthread_self" | "pthread_get_stacksize_np" => {
664 this.write_null(dest)?;
666 "pthread_attr_getstack" => {
667 // Second argument is where we are supposed to write the stack size.
668 let ptr = this.deref_operand(args[1])?;
670 let stack_addr = Scalar::from_int(0x80000, args[1].layout.size);
671 this.write_scalar(stack_addr, ptr.into())?;
672 // Return success (`0`).
673 this.write_null(dest)?;
675 "pthread_get_stackaddr_np" => {
677 let stack_addr = Scalar::from_int(0x80000, dest.layout.size);
678 this.write_scalar(stack_addr, dest)?;
681 // Stub out calls for condvar, mutex and rwlock, to just return `0`.
682 "pthread_mutexattr_init" | "pthread_mutexattr_settype" | "pthread_mutex_init" |
683 "pthread_mutexattr_destroy" | "pthread_mutex_lock" | "pthread_mutex_unlock" |
684 "pthread_mutex_destroy" | "pthread_rwlock_rdlock" | "pthread_rwlock_unlock" |
685 "pthread_rwlock_wrlock" | "pthread_rwlock_destroy" | "pthread_condattr_init" |
686 "pthread_condattr_setclock" | "pthread_cond_init" | "pthread_condattr_destroy" |
687 "pthread_cond_destroy" => {
688 this.write_null(dest)?;
691 // We don't support fork so we don't have to do anything for atfork.
692 "pthread_atfork" => {
693 this.write_null(dest)?;
697 // This is a horrible hack, but since the guard page mechanism calls mmap and expects a particular return value, we just give it that value.
698 let addr = this.read_scalar(args[0])?.not_undef()?;
699 this.write_scalar(addr, dest)?;
702 this.write_null(dest)?;
707 // FIXME: register the destructor.
710 this.write_scalar(Scalar::Ptr(this.machine.argc.unwrap()), dest)?;
713 this.write_scalar(Scalar::Ptr(this.machine.argv.unwrap()), dest)?;
715 "SecRandomCopyBytes" => {
716 let len = this.read_scalar(args[1])?.to_usize(this)?;
717 let ptr = this.read_scalar(args[2])?.to_ptr()?;
720 let data = gen_random(this, len as usize)?;
721 this.memory_mut().get_mut(ptr.alloc_id)?
722 .write_bytes(tcx, ptr, &data)?;
725 this.write_null(dest)?;
728 // Windows API stubs.
730 // DWORD = ULONG = u32
731 "GetProcessHeap" => {
732 // Just fake a HANDLE
733 this.write_scalar(Scalar::from_int(1, this.pointer_size()), dest)?;
736 let _handle = this.read_scalar(args[0])?.to_isize(this)?;
737 let flags = this.read_scalar(args[1])?.to_u32()?;
738 let size = this.read_scalar(args[2])?.to_usize(this)?;
739 let zero_init = (flags & 0x00000008) != 0; // HEAP_ZERO_MEMORY
740 let res = this.malloc(size, zero_init);
741 this.write_scalar(res, dest)?;
744 let _handle = this.read_scalar(args[0])?.to_isize(this)?;
745 let _flags = this.read_scalar(args[1])?.to_u32()?;
746 let ptr = this.read_scalar(args[2])?.not_undef()?;
750 let _handle = this.read_scalar(args[0])?.to_isize(this)?;
751 let _flags = this.read_scalar(args[1])?.to_u32()?;
752 let ptr = this.read_scalar(args[2])?.not_undef()?;
753 let size = this.read_scalar(args[3])?.to_usize(this)?;
754 let res = this.realloc(ptr, size)?;
755 this.write_scalar(res, dest)?;
759 let err = this.read_scalar(args[0])?.to_u32()?;
760 this.machine.last_error = err;
763 this.write_scalar(Scalar::from_uint(this.machine.last_error, Size::from_bits(32)), dest)?;
766 "AddVectoredExceptionHandler" => {
767 // Any non zero value works for the stdlib. This is just used for stack overflows anyway.
768 this.write_scalar(Scalar::from_int(1, dest.layout.size), dest)?;
770 "InitializeCriticalSection" |
771 "EnterCriticalSection" |
772 "LeaveCriticalSection" |
773 "DeleteCriticalSection" => {
774 // Nothing to do, not even a return value.
778 "TryEnterCriticalSection" |
779 "GetConsoleScreenBufferInfo" |
780 "SetConsoleTextAttribute" => {
781 // Pretend these do not exist / nothing happened, by returning zero.
782 this.write_null(dest)?;
785 let system_info = this.deref_operand(args[0])?;
786 let system_info_ptr = system_info.ptr.to_ptr()?;
787 // Initialize with `0`.
788 this.memory_mut().get_mut(system_info_ptr.alloc_id)?
789 .write_repeat(tcx, system_info_ptr, 0, system_info.layout.size)?;
790 // Set number of processors to `1`.
791 let dword_size = Size::from_bytes(4);
792 let offset = 2*dword_size + 3*tcx.pointer_size();
793 this.memory_mut().get_mut(system_info_ptr.alloc_id)?
796 system_info_ptr.offset(offset, tcx)?,
797 Scalar::from_int(1, dword_size).into(),
803 // This just creates a key; Windows does not natively support TLS destructors.
805 // Create key and return it.
806 let key = this.machine.tls.create_tls_key(None, tcx) as u128;
808 // Figure out how large a TLS key actually is. This is `c::DWORD`.
809 if dest.layout.size.bits() < 128
810 && key >= (1u128 << dest.layout.size.bits() as u128) {
811 return err!(OutOfTls);
813 this.write_scalar(Scalar::from_uint(key, dest.layout.size), dest)?;
816 let key = this.read_scalar(args[0])?.to_u32()? as u128;
817 let ptr = this.machine.tls.load_tls(key)?;
818 this.write_scalar(ptr, dest)?;
821 let key = this.read_scalar(args[0])?.to_u32()? as u128;
822 let new_ptr = this.read_scalar(args[1])?.not_undef()?;
823 this.machine.tls.store_tls(key, new_ptr)?;
825 // Return success (`1`).
826 this.write_scalar(Scalar::from_int(1, dest.layout.size), dest)?;
829 let which = this.read_scalar(args[0])?.to_i32()?;
830 // We just make this the identity function, so we know later in `WriteFile`
832 this.write_scalar(Scalar::from_int(which, this.pointer_size()), dest)?;
835 let handle = this.read_scalar(args[0])?.to_isize(this)?;
836 let buf = this.read_scalar(args[1])?.not_undef()?;
837 let n = this.read_scalar(args[2])?.to_u32()?;
838 let written_place = this.deref_operand(args[3])?;
839 // Spec says to always write `0` first.
840 this.write_null(written_place.into())?;
841 let written = if handle == -11 || handle == -12 {
843 use std::io::{self, Write};
845 let buf_cont = this.memory().read_bytes(buf, Size::from_bytes(u64::from(n)))?;
846 let res = if handle == -11 {
847 io::stdout().write(buf_cont)
849 io::stderr().write(buf_cont)
851 res.ok().map(|n| n as u32)
853 eprintln!("Miri: Ignored output to handle {}", handle);
854 // Pretend it all went well.
857 // If there was no error, write back how much was written.
858 if let Some(n) = written {
859 this.write_scalar(Scalar::from_uint(n, Size::from_bits(32)), written_place.into())?;
861 // Return whether this was a success.
863 Scalar::from_int(if written.is_some() { 1 } else { 0 }, dest.layout.size),
867 "GetConsoleMode" => {
868 // Everything is a pipe.
869 this.write_null(dest)?;
871 "GetEnvironmentVariableW" => {
872 // This is not the env var you are looking for.
873 this.machine.last_error = 203; // ERROR_ENVVAR_NOT_FOUND
874 this.write_null(dest)?;
876 "GetCommandLineW" => {
877 this.write_scalar(Scalar::Ptr(this.machine.cmd_line.unwrap()), dest)?;
879 // The actual name of 'RtlGenRandom'
880 "SystemFunction036" => {
881 let ptr = this.read_scalar(args[0])?.to_ptr()?;
882 let len = this.read_scalar(args[1])?.to_u32()?;
885 let data = gen_random(this, len as usize)?;
886 this.memory_mut().get_mut(ptr.alloc_id)?
887 .write_bytes(tcx, ptr, &data)?;
890 this.write_scalar(Scalar::from_bool(true), dest)?;
893 // We can't execute anything else.
895 return err!(Unimplemented(
896 format!("can't call foreign function: {}", link_name),
901 this.goto_block(Some(ret))?;
902 this.dump_place(*dest);
906 fn write_null(&mut self, dest: PlaceTy<'tcx, Tag>) -> EvalResult<'tcx> {
907 self.eval_context_mut().write_scalar(Scalar::from_int(0, dest.layout.size), dest)
910 /// Evaluates the scalar at the specified path. Returns Some(val)
911 /// if the path could be resolved, and None otherwise
912 fn eval_path_scalar(&mut self, path: &[&str]) -> EvalResult<'tcx, Option<ScalarMaybeUndef<Tag>>> {
913 let this = self.eval_context_mut();
914 if let Ok(instance) = this.resolve_path(path) {
919 let const_val = this.const_eval_raw(cid)?;
920 let const_val = this.read_scalar(const_val.into())?;
921 return Ok(Some(const_val));
927 fn gen_random<'a, 'mir, 'tcx>(
928 this: &mut MiriEvalContext<'a, 'mir, 'tcx>,
930 ) -> Result<Vec<u8>, EvalError<'tcx>> {
932 match &mut this.machine.rng {
934 let mut data = vec![0; len];
935 rng.fill_bytes(&mut data);
940 "miri does not support gathering system entropy in deterministic mode!
941 Use '-Zmiri-seed=<seed>' to enable random number generation.
942 WARNING: Miri does *not* generate cryptographically secure entropy -
943 do not use Miri to run any program that needs secure random number generation".to_owned(),