| // Copyright 2014 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. |
| |
| use io::prelude::*; |
| use os::windows::prelude::*; |
| |
| use ffi::OsString; |
| use fmt; |
| use io::{self, Error, SeekFrom}; |
| use mem; |
| use path::{Path, PathBuf}; |
| use ptr; |
| use slice; |
| use sync::Arc; |
| use sys::handle::Handle; |
| use sys::{c, cvt}; |
| use sys_common::FromInner; |
| |
| use super::to_u16s; |
| |
| pub struct File { handle: Handle } |
| |
| #[derive(Clone)] |
| pub struct FileAttr { |
| data: c::WIN32_FILE_ATTRIBUTE_DATA, |
| reparse_tag: c::DWORD, |
| } |
| |
| #[derive(Copy, Clone, PartialEq, Eq, Hash)] |
| pub enum FileType { |
| Dir, File, Symlink, ReparsePoint, MountPoint, |
| } |
| |
| pub struct ReadDir { |
| handle: FindNextFileHandle, |
| root: Arc<PathBuf>, |
| first: Option<c::WIN32_FIND_DATAW>, |
| } |
| |
| struct FindNextFileHandle(c::HANDLE); |
| |
| unsafe impl Send for FindNextFileHandle {} |
| unsafe impl Sync for FindNextFileHandle {} |
| |
| pub struct DirEntry { |
| root: Arc<PathBuf>, |
| data: c::WIN32_FIND_DATAW, |
| } |
| |
| #[derive(Clone, Default)] |
| pub struct OpenOptions { |
| create: bool, |
| append: bool, |
| read: bool, |
| write: bool, |
| truncate: bool, |
| desired_access: Option<c::DWORD>, |
| share_mode: Option<c::DWORD>, |
| creation_disposition: Option<c::DWORD>, |
| flags_and_attributes: Option<c::DWORD>, |
| security_attributes: usize, // *mut T doesn't have a Default impl |
| } |
| |
| #[derive(Clone, PartialEq, Eq, Debug)] |
| pub struct FilePermissions { attrs: c::DWORD } |
| |
| pub struct DirBuilder; |
| |
| impl Iterator for ReadDir { |
| type Item = io::Result<DirEntry>; |
| fn next(&mut self) -> Option<io::Result<DirEntry>> { |
| if let Some(first) = self.first.take() { |
| if let Some(e) = DirEntry::new(&self.root, &first) { |
| return Some(Ok(e)); |
| } |
| } |
| unsafe { |
| let mut wfd = mem::zeroed(); |
| loop { |
| if c::FindNextFileW(self.handle.0, &mut wfd) == 0 { |
| if c::GetLastError() == c::ERROR_NO_MORE_FILES { |
| return None |
| } else { |
| return Some(Err(Error::last_os_error())) |
| } |
| } |
| if let Some(e) = DirEntry::new(&self.root, &wfd) { |
| return Some(Ok(e)) |
| } |
| } |
| } |
| } |
| } |
| |
| impl Drop for FindNextFileHandle { |
| fn drop(&mut self) { |
| let r = unsafe { c::FindClose(self.0) }; |
| debug_assert!(r != 0); |
| } |
| } |
| |
| impl DirEntry { |
| fn new(root: &Arc<PathBuf>, wfd: &c::WIN32_FIND_DATAW) -> Option<DirEntry> { |
| match &wfd.cFileName[0..3] { |
| // check for '.' and '..' |
| [46, 0, ..] | |
| [46, 46, 0, ..] => return None, |
| _ => {} |
| } |
| |
| Some(DirEntry { |
| root: root.clone(), |
| data: *wfd, |
| }) |
| } |
| |
| pub fn path(&self) -> PathBuf { |
| self.root.join(&self.file_name()) |
| } |
| |
| pub fn file_name(&self) -> OsString { |
| let filename = super::truncate_utf16_at_nul(&self.data.cFileName); |
| OsString::from_wide(filename) |
| } |
| |
| pub fn file_type(&self) -> io::Result<FileType> { |
| Ok(FileType::new(self.data.dwFileAttributes, |
| /* reparse_tag = */ self.data.dwReserved0)) |
| } |
| |
| pub fn metadata(&self) -> io::Result<FileAttr> { |
| Ok(FileAttr { |
| data: c::WIN32_FILE_ATTRIBUTE_DATA { |
| dwFileAttributes: self.data.dwFileAttributes, |
| ftCreationTime: self.data.ftCreationTime, |
| ftLastAccessTime: self.data.ftLastAccessTime, |
| ftLastWriteTime: self.data.ftLastWriteTime, |
| nFileSizeHigh: self.data.nFileSizeHigh, |
| nFileSizeLow: self.data.nFileSizeLow, |
| }, |
| reparse_tag: self.data.dwReserved0, |
| }) |
| } |
| } |
| |
| impl OpenOptions { |
| pub fn new() -> OpenOptions { Default::default() } |
| pub fn read(&mut self, read: bool) { self.read = read; } |
| pub fn write(&mut self, write: bool) { self.write = write; } |
| pub fn append(&mut self, append: bool) { self.append = append; } |
| pub fn create(&mut self, create: bool) { self.create = create; } |
| pub fn truncate(&mut self, truncate: bool) { self.truncate = truncate; } |
| pub fn creation_disposition(&mut self, val: u32) { |
| self.creation_disposition = Some(val); |
| } |
| pub fn flags_and_attributes(&mut self, val: u32) { |
| self.flags_and_attributes = Some(val); |
| } |
| pub fn desired_access(&mut self, val: u32) { |
| self.desired_access = Some(val); |
| } |
| pub fn share_mode(&mut self, val: u32) { |
| self.share_mode = Some(val); |
| } |
| pub fn security_attributes(&mut self, attrs: c::LPSECURITY_ATTRIBUTES) { |
| self.security_attributes = attrs as usize; |
| } |
| |
| fn get_desired_access(&self) -> c::DWORD { |
| self.desired_access.unwrap_or({ |
| let mut base = if self.read {c::FILE_GENERIC_READ} else {0} | |
| if self.write {c::FILE_GENERIC_WRITE} else {0}; |
| if self.append { |
| base &= !c::FILE_WRITE_DATA; |
| base |= c::FILE_APPEND_DATA; |
| } |
| base |
| }) |
| } |
| |
| fn get_share_mode(&self) -> c::DWORD { |
| // libuv has a good comment about this, but the basic idea is that |
| // we try to emulate unix semantics by enabling all sharing by |
| // allowing things such as deleting a file while it's still open. |
| self.share_mode.unwrap_or(c::FILE_SHARE_READ | |
| c::FILE_SHARE_WRITE | |
| c::FILE_SHARE_DELETE) |
| } |
| |
| fn get_creation_disposition(&self) -> c::DWORD { |
| self.creation_disposition.unwrap_or({ |
| match (self.create, self.truncate) { |
| (true, true) => c::CREATE_ALWAYS, |
| (true, false) => c::OPEN_ALWAYS, |
| (false, false) => c::OPEN_EXISTING, |
| (false, true) => { |
| if self.write && !self.append { |
| c::CREATE_ALWAYS |
| } else { |
| c::TRUNCATE_EXISTING |
| } |
| } |
| } |
| }) |
| } |
| |
| fn get_flags_and_attributes(&self) -> c::DWORD { |
| self.flags_and_attributes.unwrap_or(c::FILE_ATTRIBUTE_NORMAL) |
| } |
| } |
| |
| impl File { |
| fn open_reparse_point(path: &Path, write: bool) -> io::Result<File> { |
| let mut opts = OpenOptions::new(); |
| opts.read(!write); |
| opts.write(write); |
| opts.flags_and_attributes(c::FILE_FLAG_OPEN_REPARSE_POINT | |
| c::FILE_FLAG_BACKUP_SEMANTICS); |
| File::open(path, &opts) |
| } |
| |
| pub fn open(path: &Path, opts: &OpenOptions) -> io::Result<File> { |
| let path = try!(to_u16s(path)); |
| let handle = unsafe { |
| c::CreateFileW(path.as_ptr(), |
| opts.get_desired_access(), |
| opts.get_share_mode(), |
| opts.security_attributes as *mut _, |
| opts.get_creation_disposition(), |
| opts.get_flags_and_attributes(), |
| ptr::null_mut()) |
| }; |
| if handle == c::INVALID_HANDLE_VALUE { |
| Err(Error::last_os_error()) |
| } else { |
| Ok(File { handle: Handle::new(handle) }) |
| } |
| } |
| |
| pub fn fsync(&self) -> io::Result<()> { |
| try!(cvt(unsafe { c::FlushFileBuffers(self.handle.raw()) })); |
| Ok(()) |
| } |
| |
| pub fn datasync(&self) -> io::Result<()> { self.fsync() } |
| |
| pub fn truncate(&self, size: u64) -> io::Result<()> { |
| let mut info = c::FILE_END_OF_FILE_INFO { |
| EndOfFile: size as c::LARGE_INTEGER, |
| }; |
| let size = mem::size_of_val(&info); |
| try!(cvt(unsafe { |
| c::SetFileInformationByHandle(self.handle.raw(), |
| c::FileEndOfFileInfo, |
| &mut info as *mut _ as *mut _, |
| size as c::DWORD) |
| })); |
| Ok(()) |
| } |
| |
| pub fn file_attr(&self) -> io::Result<FileAttr> { |
| unsafe { |
| let mut info: c::BY_HANDLE_FILE_INFORMATION = mem::zeroed(); |
| try!(cvt(c::GetFileInformationByHandle(self.handle.raw(), |
| &mut info))); |
| let mut attr = FileAttr { |
| data: c::WIN32_FILE_ATTRIBUTE_DATA { |
| dwFileAttributes: info.dwFileAttributes, |
| ftCreationTime: info.ftCreationTime, |
| ftLastAccessTime: info.ftLastAccessTime, |
| ftLastWriteTime: info.ftLastWriteTime, |
| nFileSizeHigh: info.nFileSizeHigh, |
| nFileSizeLow: info.nFileSizeLow, |
| }, |
| reparse_tag: 0, |
| }; |
| if attr.is_reparse_point() { |
| let mut b = [0; c::MAXIMUM_REPARSE_DATA_BUFFER_SIZE]; |
| if let Ok((_, buf)) = self.reparse_point(&mut b) { |
| attr.reparse_tag = buf.ReparseTag; |
| } |
| } |
| Ok(attr) |
| } |
| } |
| |
| pub fn read(&self, buf: &mut [u8]) -> io::Result<usize> { |
| self.handle.read(buf) |
| } |
| |
| pub fn write(&self, buf: &[u8]) -> io::Result<usize> { |
| self.handle.write(buf) |
| } |
| |
| pub fn flush(&self) -> io::Result<()> { Ok(()) } |
| |
| pub fn seek(&self, pos: SeekFrom) -> io::Result<u64> { |
| let (whence, pos) = match pos { |
| SeekFrom::Start(n) => (c::FILE_BEGIN, n as i64), |
| SeekFrom::End(n) => (c::FILE_END, n), |
| SeekFrom::Current(n) => (c::FILE_CURRENT, n), |
| }; |
| let pos = pos as c::LARGE_INTEGER; |
| let mut newpos = 0; |
| try!(cvt(unsafe { |
| c::SetFilePointerEx(self.handle.raw(), pos, |
| &mut newpos, whence) |
| })); |
| Ok(newpos as u64) |
| } |
| |
| pub fn handle(&self) -> &Handle { &self.handle } |
| |
| pub fn into_handle(self) -> Handle { self.handle } |
| |
| fn reparse_point<'a>(&self, |
| space: &'a mut [u8; c::MAXIMUM_REPARSE_DATA_BUFFER_SIZE]) |
| -> io::Result<(c::DWORD, &'a c::REPARSE_DATA_BUFFER)> { |
| unsafe { |
| let mut bytes = 0; |
| try!(cvt({ |
| c::DeviceIoControl(self.handle.raw(), |
| c::FSCTL_GET_REPARSE_POINT, |
| ptr::null_mut(), |
| 0, |
| space.as_mut_ptr() as *mut _, |
| space.len() as c::DWORD, |
| &mut bytes, |
| ptr::null_mut()) |
| })); |
| Ok((bytes, &*(space.as_ptr() as *const c::REPARSE_DATA_BUFFER))) |
| } |
| } |
| |
| fn readlink(&self) -> io::Result<PathBuf> { |
| let mut space = [0u8; c::MAXIMUM_REPARSE_DATA_BUFFER_SIZE]; |
| let (_bytes, buf) = try!(self.reparse_point(&mut space)); |
| if buf.ReparseTag != c::IO_REPARSE_TAG_SYMLINK { |
| return Err(io::Error::new(io::ErrorKind::Other, "not a symlink")) |
| } |
| |
| unsafe { |
| let info: *const c::SYMBOLIC_LINK_REPARSE_BUFFER = |
| &buf.rest as *const _ as *const _; |
| let path_buffer = &(*info).PathBuffer as *const _ as *const u16; |
| let subst_off = (*info).SubstituteNameOffset / 2; |
| let subst_ptr = path_buffer.offset(subst_off as isize); |
| let subst_len = (*info).SubstituteNameLength / 2; |
| let subst = slice::from_raw_parts(subst_ptr, subst_len as usize); |
| |
| Ok(PathBuf::from(OsString::from_wide(subst))) |
| } |
| } |
| } |
| |
| impl FromInner<c::HANDLE> for File { |
| fn from_inner(handle: c::HANDLE) -> File { |
| File { handle: Handle::new(handle) } |
| } |
| } |
| |
| impl fmt::Debug for File { |
| fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { |
| // FIXME(#24570): add more info here (e.g. mode) |
| let mut b = f.debug_struct("File"); |
| b.field("handle", &self.handle.raw()); |
| if let Ok(path) = get_path(&self) { |
| b.field("path", &path); |
| } |
| b.finish() |
| } |
| } |
| |
| impl FileAttr { |
| pub fn size(&self) -> u64 { |
| ((self.data.nFileSizeHigh as u64) << 32) | (self.data.nFileSizeLow as u64) |
| } |
| |
| pub fn perm(&self) -> FilePermissions { |
| FilePermissions { attrs: self.data.dwFileAttributes } |
| } |
| |
| pub fn attrs(&self) -> u32 { self.data.dwFileAttributes as u32 } |
| |
| pub fn file_type(&self) -> FileType { |
| FileType::new(self.data.dwFileAttributes, self.reparse_tag) |
| } |
| |
| pub fn created(&self) -> u64 { self.to_u64(&self.data.ftCreationTime) } |
| pub fn accessed(&self) -> u64 { self.to_u64(&self.data.ftLastAccessTime) } |
| pub fn modified(&self) -> u64 { self.to_u64(&self.data.ftLastWriteTime) } |
| |
| fn to_u64(&self, ft: &c::FILETIME) -> u64 { |
| (ft.dwLowDateTime as u64) | ((ft.dwHighDateTime as u64) << 32) |
| } |
| |
| fn is_reparse_point(&self) -> bool { |
| self.data.dwFileAttributes & c::FILE_ATTRIBUTE_REPARSE_POINT != 0 |
| } |
| } |
| |
| impl FilePermissions { |
| pub fn readonly(&self) -> bool { |
| self.attrs & c::FILE_ATTRIBUTE_READONLY != 0 |
| } |
| |
| pub fn set_readonly(&mut self, readonly: bool) { |
| if readonly { |
| self.attrs |= c::FILE_ATTRIBUTE_READONLY; |
| } else { |
| self.attrs &= !c::FILE_ATTRIBUTE_READONLY; |
| } |
| } |
| } |
| |
| impl FileType { |
| fn new(attrs: c::DWORD, reparse_tag: c::DWORD) -> FileType { |
| if attrs & c::FILE_ATTRIBUTE_REPARSE_POINT != 0 { |
| match reparse_tag { |
| c::IO_REPARSE_TAG_SYMLINK => FileType::Symlink, |
| c::IO_REPARSE_TAG_MOUNT_POINT => FileType::MountPoint, |
| _ => FileType::ReparsePoint, |
| } |
| } else if attrs & c::FILE_ATTRIBUTE_DIRECTORY != 0 { |
| FileType::Dir |
| } else { |
| FileType::File |
| } |
| } |
| |
| pub fn is_dir(&self) -> bool { *self == FileType::Dir } |
| pub fn is_file(&self) -> bool { *self == FileType::File } |
| pub fn is_symlink(&self) -> bool { |
| *self == FileType::Symlink || *self == FileType::MountPoint |
| } |
| } |
| |
| impl DirBuilder { |
| pub fn new() -> DirBuilder { DirBuilder } |
| |
| pub fn mkdir(&self, p: &Path) -> io::Result<()> { |
| let p = try!(to_u16s(p)); |
| try!(cvt(unsafe { |
| c::CreateDirectoryW(p.as_ptr(), ptr::null_mut()) |
| })); |
| Ok(()) |
| } |
| } |
| |
| pub fn readdir(p: &Path) -> io::Result<ReadDir> { |
| let root = p.to_path_buf(); |
| let star = p.join("*"); |
| let path = try!(to_u16s(&star)); |
| |
| unsafe { |
| let mut wfd = mem::zeroed(); |
| let find_handle = c::FindFirstFileW(path.as_ptr(), &mut wfd); |
| if find_handle != c::INVALID_HANDLE_VALUE { |
| Ok(ReadDir { |
| handle: FindNextFileHandle(find_handle), |
| root: Arc::new(root), |
| first: Some(wfd), |
| }) |
| } else { |
| Err(Error::last_os_error()) |
| } |
| } |
| } |
| |
| pub fn unlink(p: &Path) -> io::Result<()> { |
| let p_u16s = try!(to_u16s(p)); |
| try!(cvt(unsafe { c::DeleteFileW(p_u16s.as_ptr()) })); |
| Ok(()) |
| } |
| |
| pub fn rename(old: &Path, new: &Path) -> io::Result<()> { |
| let old = try!(to_u16s(old)); |
| let new = try!(to_u16s(new)); |
| try!(cvt(unsafe { |
| c::MoveFileExW(old.as_ptr(), new.as_ptr(), c::MOVEFILE_REPLACE_EXISTING) |
| })); |
| Ok(()) |
| } |
| |
| pub fn rmdir(p: &Path) -> io::Result<()> { |
| let p = try!(to_u16s(p)); |
| try!(cvt(unsafe { c::RemoveDirectoryW(p.as_ptr()) })); |
| Ok(()) |
| } |
| |
| pub fn readlink(p: &Path) -> io::Result<PathBuf> { |
| let file = try!(File::open_reparse_point(p, false)); |
| file.readlink() |
| } |
| |
| pub fn symlink(src: &Path, dst: &Path) -> io::Result<()> { |
| symlink_inner(src, dst, false) |
| } |
| |
| pub fn symlink_inner(src: &Path, dst: &Path, dir: bool) -> io::Result<()> { |
| let src = try!(to_u16s(src)); |
| let dst = try!(to_u16s(dst)); |
| let flags = if dir { c::SYMBOLIC_LINK_FLAG_DIRECTORY } else { 0 }; |
| try!(cvt(unsafe { |
| c::CreateSymbolicLinkW(dst.as_ptr(), src.as_ptr(), flags) as c::BOOL |
| })); |
| Ok(()) |
| } |
| |
| pub fn link(src: &Path, dst: &Path) -> io::Result<()> { |
| let src = try!(to_u16s(src)); |
| let dst = try!(to_u16s(dst)); |
| try!(cvt(unsafe { |
| c::CreateHardLinkW(dst.as_ptr(), src.as_ptr(), ptr::null_mut()) |
| })); |
| Ok(()) |
| } |
| |
| pub fn stat(p: &Path) -> io::Result<FileAttr> { |
| let attr = try!(lstat(p)); |
| |
| // If this is a reparse point, then we need to reopen the file to get the |
| // actual destination. We also pass the FILE_FLAG_BACKUP_SEMANTICS flag to |
| // ensure that we can open directories (this path may be a directory |
| // junction). Once the file is opened we ask the opened handle what its |
| // metadata information is. |
| if attr.is_reparse_point() { |
| let mut opts = OpenOptions::new(); |
| opts.flags_and_attributes(c::FILE_FLAG_BACKUP_SEMANTICS); |
| let file = try!(File::open(p, &opts)); |
| file.file_attr() |
| } else { |
| Ok(attr) |
| } |
| } |
| |
| pub fn lstat(p: &Path) -> io::Result<FileAttr> { |
| let u16s = try!(to_u16s(p)); |
| unsafe { |
| let mut attr: FileAttr = mem::zeroed(); |
| try!(cvt(c::GetFileAttributesExW(u16s.as_ptr(), |
| c::GetFileExInfoStandard, |
| &mut attr.data as *mut _ as *mut _))); |
| if attr.is_reparse_point() { |
| attr.reparse_tag = File::open_reparse_point(p, false).and_then(|f| { |
| let mut b = [0; c::MAXIMUM_REPARSE_DATA_BUFFER_SIZE]; |
| f.reparse_point(&mut b).map(|(_, b)| b.ReparseTag) |
| }).unwrap_or(0); |
| } |
| Ok(attr) |
| } |
| } |
| |
| pub fn set_perm(p: &Path, perm: FilePermissions) -> io::Result<()> { |
| let p = try!(to_u16s(p)); |
| unsafe { |
| try!(cvt(c::SetFileAttributesW(p.as_ptr(), perm.attrs))); |
| Ok(()) |
| } |
| } |
| |
| fn get_path(f: &File) -> io::Result<PathBuf> { |
| super::fill_utf16_buf(|buf, sz| unsafe { |
| c::GetFinalPathNameByHandleW(f.handle.raw(), buf, sz, |
| c::VOLUME_NAME_DOS) |
| }, |buf| { |
| PathBuf::from(OsString::from_wide(buf)) |
| }) |
| } |
| |
| pub fn canonicalize(p: &Path) -> io::Result<PathBuf> { |
| let mut opts = OpenOptions::new(); |
| opts.read(true); |
| // This flag is so we can open directories too |
| opts.flags_and_attributes(c::FILE_FLAG_BACKUP_SEMANTICS); |
| let f = try!(File::open(p, &opts)); |
| get_path(&f) |
| } |
| |
| pub fn copy(from: &Path, to: &Path) -> io::Result<u64> { |
| unsafe extern "system" fn callback( |
| _TotalFileSize: c::LARGE_INTEGER, |
| TotalBytesTransferred: c::LARGE_INTEGER, |
| _StreamSize: c::LARGE_INTEGER, |
| _StreamBytesTransferred: c::LARGE_INTEGER, |
| _dwStreamNumber: c::DWORD, |
| _dwCallbackReason: c::DWORD, |
| _hSourceFile: c::HANDLE, |
| _hDestinationFile: c::HANDLE, |
| lpData: c::LPVOID, |
| ) -> c::DWORD { |
| *(lpData as *mut i64) = TotalBytesTransferred; |
| c::PROGRESS_CONTINUE |
| } |
| let pfrom = try!(to_u16s(from)); |
| let pto = try!(to_u16s(to)); |
| let mut size = 0i64; |
| try!(cvt(unsafe { |
| c::CopyFileExW(pfrom.as_ptr(), pto.as_ptr(), Some(callback), |
| &mut size as *mut _ as *mut _, ptr::null_mut(), 0) |
| })); |
| Ok(size as u64) |
| } |
| |
| #[test] |
| fn directory_junctions_are_directories() { |
| use ffi::OsStr; |
| use env; |
| use rand::{self, StdRng, Rng}; |
| use vec::Vec; |
| |
| macro_rules! t { |
| ($e:expr) => (match $e { |
| Ok(e) => e, |
| Err(e) => panic!("{} failed with: {}", stringify!($e), e), |
| }) |
| } |
| |
| let d = DirBuilder::new(); |
| let p = env::temp_dir(); |
| let mut r = rand::thread_rng(); |
| let ret = p.join(&format!("rust-{}", r.next_u32())); |
| let foo = ret.join("foo"); |
| let bar = ret.join("bar"); |
| t!(d.mkdir(&ret)); |
| t!(d.mkdir(&foo)); |
| t!(d.mkdir(&bar)); |
| |
| t!(create_junction(&bar, &foo)); |
| let metadata = stat(&bar); |
| t!(delete_junction(&bar)); |
| |
| t!(rmdir(&foo)); |
| t!(rmdir(&bar)); |
| t!(rmdir(&ret)); |
| |
| let metadata = t!(metadata); |
| assert!(metadata.file_type().is_dir()); |
| |
| // Creating a directory junction on windows involves dealing with reparse |
| // points and the DeviceIoControl function, and this code is a skeleton of |
| // what can be found here: |
| // |
| // http://www.flexhex.com/docs/articles/hard-links.phtml |
| fn create_junction(src: &Path, dst: &Path) -> io::Result<()> { |
| let f = try!(opendir(src, true)); |
| let h = f.handle().raw(); |
| |
| unsafe { |
| let mut data = [0u8; c::MAXIMUM_REPARSE_DATA_BUFFER_SIZE]; |
| let mut db = data.as_mut_ptr() |
| as *mut c::REPARSE_MOUNTPOINT_DATA_BUFFER; |
| let mut buf = &mut (*db).ReparseTarget as *mut _; |
| let mut i = 0; |
| let v = br"\??\"; |
| let v = v.iter().map(|x| *x as u16); |
| for c in v.chain(dst.as_os_str().encode_wide()) { |
| *buf.offset(i) = c; |
| i += 1; |
| } |
| *buf.offset(i) = 0; |
| i += 1; |
| (*db).ReparseTag = c::IO_REPARSE_TAG_MOUNT_POINT; |
| (*db).ReparseTargetMaximumLength = (i * 2) as c::WORD; |
| (*db).ReparseTargetLength = ((i - 1) * 2) as c::WORD; |
| (*db).ReparseDataLength = |
| (*db).ReparseTargetLength as c::DWORD + 12; |
| |
| let mut ret = 0; |
| cvt(c::DeviceIoControl(h as *mut _, |
| c::FSCTL_SET_REPARSE_POINT, |
| data.as_ptr() as *mut _, |
| (*db).ReparseDataLength + 8, |
| ptr::null_mut(), 0, |
| &mut ret, |
| ptr::null_mut())).map(|_| ()) |
| } |
| } |
| |
| fn opendir(p: &Path, write: bool) -> io::Result<File> { |
| unsafe { |
| let mut token = ptr::null_mut(); |
| let mut tp: c::TOKEN_PRIVILEGES = mem::zeroed(); |
| try!(cvt(c::OpenProcessToken(c::GetCurrentProcess(), |
| c::TOKEN_ADJUST_PRIVILEGES, |
| &mut token))); |
| let name: &OsStr = if write { |
| "SeRestorePrivilege".as_ref() |
| } else { |
| "SeBackupPrivilege".as_ref() |
| }; |
| let name = name.encode_wide().chain(Some(0)).collect::<Vec<_>>(); |
| try!(cvt(c::LookupPrivilegeValueW(ptr::null(), |
| name.as_ptr(), |
| &mut tp.Privileges[0].Luid))); |
| tp.PrivilegeCount = 1; |
| tp.Privileges[0].Attributes = c::SE_PRIVILEGE_ENABLED; |
| let size = mem::size_of::<c::TOKEN_PRIVILEGES>() as c::DWORD; |
| try!(cvt(c::AdjustTokenPrivileges(token, c::FALSE, &mut tp, size, |
| ptr::null_mut(), ptr::null_mut()))); |
| try!(cvt(c::CloseHandle(token))); |
| |
| File::open_reparse_point(p, write) |
| } |
| } |
| |
| fn delete_junction(p: &Path) -> io::Result<()> { |
| unsafe { |
| let f = try!(opendir(p, true)); |
| let h = f.handle().raw(); |
| let mut data = [0u8; c::MAXIMUM_REPARSE_DATA_BUFFER_SIZE]; |
| let mut db = data.as_mut_ptr() |
| as *mut c::REPARSE_MOUNTPOINT_DATA_BUFFER; |
| (*db).ReparseTag = c::IO_REPARSE_TAG_MOUNT_POINT; |
| let mut bytes = 0; |
| cvt(c::DeviceIoControl(h as *mut _, |
| c::FSCTL_DELETE_REPARSE_POINT, |
| data.as_ptr() as *mut _, |
| (*db).ReparseDataLength + 8, |
| ptr::null_mut(), 0, |
| &mut bytes, |
| ptr::null_mut())).map(|_| ()) |
| } |
| } |
| } |