rust/src/lib/_io.rs

58 lines
1.1 KiB
Rust
Raw Normal View History

type buf_reader = unsafe obj {
fn read() -> vec[u8];
2010-06-23 23:03:09 -05:00
};
fn default_bufsz() -> uint {
ret uint(4096);
}
fn new_buf() -> vec[u8] {
let vec[u8] v = vec();
let uint i = default_bufsz();
while (i > uint(0)) {
i -= uint(1);
v += vec(u8(0));
}
// FIXME (issue #93): should be:
// ret _vec.alloc[u8](default_bufsz());
ret v;
}
fn new_buf_reader(str s) -> buf_reader {
2010-06-23 23:03:09 -05:00
unsafe obj fd_buf_reader(int fd, mutable vec[u8] buf) {
fn read() -> vec[u8] {
// Ensure our buf is singly-referenced.
if (_vec.rustrt.refcount[u8](buf) != uint(1)) {
buf = new_buf();
}
auto len = _vec.len[u8](buf);
auto vbuf = _vec.buf[u8](buf);
auto count = os.libc.read(fd, vbuf, len);
2010-06-23 23:03:09 -05:00
if (count < 0) {
log "error filling buffer";
log sys.rustrt.last_os_error();
fail;
} else {
ret buf;
2010-06-23 23:03:09 -05:00
}
}
2010-06-23 23:03:09 -05:00
drop {
os.libc.close(fd);
}
}
auto fd = os.libc.open(_str.buf(s), 0);
if (fd < 0) {
log "error opening file";
log sys.rustrt.last_os_error();
fail;
}
ret fd_buf_reader(fd, new_buf());
2010-06-23 23:03:09 -05:00
}