Add getcwd shim

Author: pvdrz

src/helpers.rs

@@ -71,7 +71,7 @@ pub trait EvalContextExt<'mir, 'tcx: 'mir>: crate::MiriEvalContextExt<'mir, 'tcx
     /// Get the `Place` for a local
     fn local_place(&mut self, local: mir::Local) -> InterpResult<'tcx, PlaceTy<'tcx, Tag>> {
         let this = self.eval_context_mut();
-        let place = mir::Place { base: mir::PlaceBase::Local(local), projection: Box::new([]) };
+        let place = mir::Place { base: mir::PlaceBase::Local(local), projection: None };
         this.eval_place(&place)
     }
 

src/shims/env.rs

@@ -1,4 +1,5 @@
 use std::collections::HashMap;
+use std::env;
 
 use rustc::ty::layout::{Size};
 use rustc_mir::interpret::{Pointer, Memory};
@@ -21,7 +22,7 @@ impl EnvVars {
         excluded_env_vars.push("TERM".to_owned());
 
         if ecx.machine.communicate {
-            for (name, value) in std::env::vars() {
+            for (name, value) in env::vars() {
                 if !excluded_env_vars.contains(&name) {
                     let var_ptr = alloc_env_var(name.as_bytes(), value.as_bytes(), ecx.memory_mut());
                     ecx.machine.env_vars.map.insert(name.into_bytes(), var_ptr);
@@ -111,4 +112,31 @@ pub trait EvalContextExt<'mir, 'tcx: 'mir>: crate::MiriEvalContextExt<'mir, 'tcx
             Ok(-1)
         }
     }
+
+    fn getcwd(
+        &mut self,
+        buf_op: OpTy<'tcx, Tag>,
+        size_op: OpTy<'tcx, Tag>,
+    ) -> InterpResult<'tcx, Scalar<Tag>> {
+        let this = self.eval_context_mut();
+
+        let buf = this.read_scalar(buf_op)?.not_undef()?.assert_ptr();
+        let size = this.read_scalar(size_op)?.to_usize(&*this.tcx)?;
+        // If we cannot get the current directory, we return null
+        if let Ok(cwd) = env::current_dir() {
+            // It is not clear what happens with non-utf8 paths here
+            let mut bytes = cwd.display().to_string().into_bytes();
+            // If the buffer is smaller than the path, we return null
+            if bytes.len() as u64 <= size {
+                bytes.resize(size as usize, 0);
+                // Temporary pointer to write the path into miri's memory
+                let ptr = this.memory_mut().allocate_static_bytes(bytes.as_slice(), MiriMemoryKind::Env.into());
+                this.memory_mut().copy(ptr, buf, Size::from_bytes(size), true)?;
+                // Deallocate the temporary pointer
+                this.memory_mut().deallocate(ptr, None, MiriMemoryKind::Env.into())?;
+                return Ok(Scalar::Ptr(buf))
+            }
+        }
+        Ok(Scalar::ptr_null(&*this.tcx))
+    }
 }

src/shims/foreign_items.rs

@@ -436,6 +436,11 @@ pub trait EvalContextExt<'mir, 'tcx: 'mir>: crate::MiriEvalContextExt<'mir, 'tcx
                 this.write_scalar(Scalar::from_int(result, dest.layout.size), dest)?;
             }
 
+            "getcwd" => {
+                let result = this.getcwd(args[0], args[1])?;
+                this.write_scalar(result, dest)?;
+            }
+
             "write" => {
                 let fd = this.read_scalar(args[0])?.to_i32()?;
                 let buf = this.read_scalar(args[1])?.not_undef()?;