safe_libc/src/stdio.rs


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

use crate as libc;
use crate::string;

use core::fmt;

pub struct OStream {
	file: *mut libc::FILE
}

#[inline]
pub unsafe fn stdout() -> OStream {
	OStream { file: libc::stdout }
}

#[inline]
pub unsafe fn stderr() -> OStream {
	OStream { file: libc::stderr }
}

impl OStream {
	#[inline]
	pub fn write(&mut self, b: &[u8]) {
		unsafe {
			libc::fwrite(
				b.as_ptr() as *const libc::c_void,
				1,
				b.len(),
				self.file,
			);
		}
	}

	#[inline]
	pub fn puts(&mut self, s: &string::CStr) {
		unsafe {
			libc::fputs(s.as_ptr(), self.file);
		}
	}
}

impl fmt::Write for OStream {
	#[inline]
	fn write_str(&mut self, s: &str) -> fmt::Result {
		self.write(s.as_bytes());
		Ok(())
	}
}