1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
use std::io;
use std::cmp;
use winapi::*;
use kernel32::*;
#[derive(Debug)]
pub struct Handle(HANDLE);
unsafe impl Send for Handle {}
unsafe impl Sync for Handle {}
impl Handle {
pub fn new(handle: HANDLE) -> Handle {
Handle(handle)
}
pub fn raw(&self) -> HANDLE { self.0 }
pub fn into_raw(self) -> HANDLE {
use std::mem;
let ret = self.0;
mem::forget(self);
ret
}
pub fn write(&self, buf: &[u8]) -> io::Result<usize> {
let mut bytes = 0;
let len = cmp::min(buf.len(), <DWORD>::max_value() as usize) as DWORD;
try!(::cvt(unsafe {
WriteFile(self.0, buf.as_ptr() as *const _, len, &mut bytes,
0 as *mut _)
}));
Ok(bytes as usize)
}
pub fn read(&self, buf: &mut [u8]) -> io::Result<usize> {
let mut bytes = 0;
let len = cmp::min(buf.len(), <DWORD>::max_value() as usize) as DWORD;
try!(::cvt(unsafe {
ReadFile(self.0, buf.as_mut_ptr() as *mut _, len, &mut bytes,
0 as *mut _)
}));
Ok(bytes as usize)
}
pub unsafe fn read_overlapped(&self, buf: &mut [u8],
overlapped: *mut OVERLAPPED)
-> io::Result<Option<usize>> {
let len = cmp::min(buf.len(), <DWORD>::max_value() as usize) as DWORD;
let mut bytes = 0;
let res = ::cvt({
ReadFile(self.0,
buf.as_mut_ptr() as *mut _,
len,
&mut bytes,
overlapped)
});
match res {
Ok(_) => Ok(Some(bytes as usize)),
Err(ref e) if e.raw_os_error() == Some(ERROR_IO_PENDING as i32)
=> Ok(None),
Err(e) => Err(e),
}
}
pub unsafe fn write_overlapped(&self, buf: &[u8],
overlapped: *mut OVERLAPPED)
-> io::Result<Option<usize>> {
let len = cmp::min(buf.len(), <DWORD>::max_value() as usize) as DWORD;
let mut bytes = 0;
let res = ::cvt({
WriteFile(self.0,
buf.as_ptr() as *const _,
len,
&mut bytes,
overlapped)
});
match res {
Ok(_) => Ok(Some(bytes as usize)),
Err(ref e) if e.raw_os_error() == Some(ERROR_IO_PENDING as i32)
=> Ok(None),
Err(e) => Err(e),
}
}
}
impl Drop for Handle {
fn drop(&mut self) {
unsafe { CloseHandle(self.0) };
}
}
|
