/* Module: io Basic input/output */ import result::result; import dvec::{dvec, extensions}; import libc::{c_int, c_long, c_uint, c_void, size_t, ssize_t}; import libc::consts::os::posix88::*; import libc::consts::os::extra::*; type fd_t = c_int; #[abi = "cdecl"] native mod rustrt { fn rust_get_stdin() -> *libc::FILE; fn rust_get_stdout() -> *libc::FILE; fn rust_get_stderr() -> *libc::FILE; } // Reading // FIXME This is all buffered. We might need an unbuffered variant as well // #2004 enum seek_style { seek_set, seek_end, seek_cur, } // The raw underlying reader iface. All readers must implement this. iface reader { // FIXME: Seekable really should be orthogonal. // #2004 fn read_bytes(uint) -> [u8]; fn read_byte() -> int; fn unread_byte(int); fn eof() -> bool; fn seek(int, seek_style); fn tell() -> uint; } // Generic utility functions defined on readers impl reader_util for reader { fn read_chars(n: uint) -> [char] { // returns the (consumed offset, n_req), appends characters to &chars fn chars_from_buf(buf: [u8], &chars: [char]) -> (uint, uint) { let mut i = 0u; while i < vec::len(buf) { let b0 = buf[i]; let w = str::utf8_char_width(b0); let end = i + w; i += 1u; assert (w > 0u); if w == 1u { chars += [ b0 as char ]; cont; } // can't satisfy this char with the existing data if end > vec::len(buf) { ret (i - 1u, end - vec::len(buf)); } let mut val = 0u; while i < end { let next = buf[i] as int; i += 1u; assert (next > -1); assert (next & 192 == 128); val <<= 6u; val += (next & 63) as uint; } // See str::char_at val += ((b0 << ((w + 1u) as u8)) as uint) << (w - 1u) * 6u - w - 1u; chars += [ val as char ]; } ret (i, 0u); } let mut buf: [u8] = []; let mut chars: [char] = []; // might need more bytes, but reading n will never over-read let mut nbread = n; while nbread > 0u { let data = self.read_bytes(nbread); if vec::len(data) == 0u { // eof - FIXME should we do something if // we're split in a unicode char? // #2004 break; } buf += data; let (offset, nbreq) = chars_from_buf(buf, chars); let ncreq = n - vec::len(chars); // again we either know we need a certain number of bytes // to complete a character, or we make sure we don't // over-read by reading 1-byte per char needed nbread = if ncreq > nbreq { ncreq } else { nbreq }; if nbread > 0u { buf = vec::slice(buf, offset, vec::len(buf)); } } chars } fn read_char() -> char { let c = self.read_chars(1u); if vec::len(c) == 0u { ret -1 as char; // FIXME will this stay valid? // #2004 } assert(vec::len(c) == 1u); ret c[0]; } fn read_line() -> str { let mut buf: [u8] = []; loop { let ch = self.read_byte(); if ch == -1 || ch == 10 { break; } buf += [ch as u8]; } str::from_bytes(buf) } fn read_c_str() -> str { let mut buf: [u8] = []; loop { let ch = self.read_byte(); if ch < 1 { break; } else { buf += [ch as u8]; } } str::from_bytes(buf) } // FIXME deal with eof? // #2004 fn read_le_uint(size: uint) -> uint { let mut val = 0u, pos = 0u, i = size; while i > 0u { val += (self.read_byte() as uint) << pos; pos += 8u; i -= 1u; } val } fn read_le_int(size: uint) -> int { let mut val = 0u, pos = 0u, i = size; while i > 0u { val += (self.read_byte() as uint) << pos; pos += 8u; i -= 1u; } val as int } fn read_be_uint(size: uint) -> uint { let mut val = 0u, i = size; while i > 0u { i -= 1u; val += (self.read_byte() as uint) << i * 8u; } val } fn read_whole_stream() -> [u8] { let mut buf: [u8] = []; while !self.eof() { buf += self.read_bytes(2048u); } buf } fn each_byte(it: fn(int) -> bool) { while !self.eof() { if !it(self.read_byte()) { break; } } } fn each_char(it: fn(char) -> bool) { while !self.eof() { if !it(self.read_char()) { break; } } } fn each_line(it: fn(str) -> bool) { while !self.eof() { if !it(self.read_line()) { break; } } } } // Reader implementations fn convert_whence(whence: seek_style) -> i32 { ret alt whence { seek_set { 0i32 } seek_cur { 1i32 } seek_end { 2i32 } }; } impl of reader for *libc::FILE { fn read_bytes(len: uint) -> [u8] unsafe { let mut buf : [mut u8] = [mut]; vec::reserve(buf, len); vec::as_mut_buf(buf) {|b| let read = libc::fread(b as *mut c_void, 1u as size_t, len as size_t, self); vec::unsafe::set_len(buf, read as uint); } ret vec::from_mut(buf); } fn read_byte() -> int { ret libc::fgetc(self) as int; } fn unread_byte(byte: int) { libc::ungetc(byte as c_int, self); } fn eof() -> bool { ret libc::feof(self) != 0 as c_int; } fn seek(offset: int, whence: seek_style) { assert libc::fseek(self, offset as c_long, convert_whence(whence)) == 0 as c_int; } fn tell() -> uint { ret libc::ftell(self) as uint; } } // A forwarding impl of reader that also holds on to a resource for the // duration of its lifetime. // FIXME there really should be a better way to do this // #2004 impl of reader for {base: T, cleanup: C} { fn read_bytes(len: uint) -> [u8] { self.base.read_bytes(len) } fn read_byte() -> int { self.base.read_byte() } fn unread_byte(byte: int) { self.base.unread_byte(byte); } fn eof() -> bool { self.base.eof() } fn seek(off: int, whence: seek_style) { self.base.seek(off, whence) } fn tell() -> uint { self.base.tell() } } resource FILE_res(f: *libc::FILE) { libc::fclose(f); } fn FILE_reader(f: *libc::FILE, cleanup: bool) -> reader { if cleanup { {base: f, cleanup: FILE_res(f)} as reader } else { f as reader } } // FIXME: this should either be an iface-less impl, a set of top-level // functions that take a reader, or a set of default methods on reader // (which can then be called reader) // #2004 fn stdin() -> reader { rustrt::rust_get_stdin() as reader } fn file_reader(path: str) -> result { let f = os::as_c_charp(path, {|pathbuf| os::as_c_charp("r", {|modebuf| libc::fopen(pathbuf, modebuf) }) }); ret if f as uint == 0u { result::err("error opening " + path) } else { result::ok(FILE_reader(f, true)) } } // Byte buffer readers // TODO: const u8, but this fails with rustboot. type byte_buf = {buf: [u8], mut pos: uint, len: uint}; impl of reader for byte_buf { fn read_bytes(len: uint) -> [u8] { let rest = self.len - self.pos; let mut to_read = len; if rest < to_read { to_read = rest; } let range = vec::slice(self.buf, self.pos, self.pos + to_read); self.pos += to_read; ret range; } fn read_byte() -> int { if self.pos == self.len { ret -1; } let b = self.buf[self.pos]; self.pos += 1u; ret b as int; } fn unread_byte(_byte: int) { #error("TODO: unread_byte"); fail; } fn eof() -> bool { self.pos == self.len } fn seek(offset: int, whence: seek_style) { let pos = self.pos; self.pos = seek_in_buf(offset, pos, self.len, whence); } fn tell() -> uint { self.pos } } fn bytes_reader(bytes: [u8]) -> reader { bytes_reader_between(bytes, 0u, vec::len(bytes)) } fn bytes_reader_between(bytes: [u8], start: uint, end: uint) -> reader { {buf: bytes, mut pos: start, len: end} as reader } fn with_bytes_reader(bytes: [u8], f: fn(reader) -> t) -> t { f(bytes_reader(bytes)) } fn with_bytes_reader_between(bytes: [u8], start: uint, end: uint, f: fn(reader) -> t) -> t { f(bytes_reader_between(bytes, start, end)) } fn str_reader(s: str) -> reader { bytes_reader(str::bytes(s)) } fn with_str_reader(s: str, f: fn(reader) -> T) -> T { str::as_bytes(s) { |bytes| with_bytes_reader_between(bytes, 0u, str::len(s), f) } } // Writing enum fileflag { append, create, truncate, no_flag, } // FIXME: Seekable really should be orthogonal. // FIXME: eventually u64 // #2004 iface writer { fn write([const u8]/&); fn seek(int, seek_style); fn tell() -> uint; fn flush() -> int; } impl of writer for {base: T, cleanup: C} { fn write(bs: [const u8]/&) { self.base.write(bs); } fn seek(off: int, style: seek_style) { self.base.seek(off, style); } fn tell() -> uint { self.base.tell() } fn flush() -> int { self.base.flush() } } impl of writer for *libc::FILE { fn write(v: [const u8]/&) unsafe { vec::unpack_const_slice(v) {|vbuf, len| let nout = libc::fwrite(vbuf as *c_void, len as size_t, 1u as size_t, self); if nout < 1 as size_t { #error("error writing buffer"); log(error, os::last_os_error()); fail; } } } fn seek(offset: int, whence: seek_style) { assert libc::fseek(self, offset as c_long, convert_whence(whence)) == 0 as c_int; } fn tell() -> uint { libc::ftell(self) as uint } fn flush() -> int { libc::fflush(self) as int } } fn FILE_writer(f: *libc::FILE, cleanup: bool) -> writer { if cleanup { {base: f, cleanup: FILE_res(f)} as writer } else { f as writer } } impl of writer for fd_t { fn write(v: [const u8]/&) unsafe { let mut count = 0u; vec::unpack_const_slice(v) {|vbuf, len| while count < len { let vb = ptr::const_offset(vbuf, count) as *c_void; let nout = libc::write(self, vb, len as size_t); if nout < 0 as ssize_t { #error("error writing buffer"); log(error, os::last_os_error()); fail; } count += nout as uint; } } } fn seek(_offset: int, _whence: seek_style) { #error("need 64-bit native calls for seek, sorry"); fail; } fn tell() -> uint { #error("need 64-bit native calls for tell, sorry"); fail; } fn flush() -> int { 0 } } resource fd_res(fd: fd_t) { libc::close(fd); } fn fd_writer(fd: fd_t, cleanup: bool) -> writer { if cleanup { {base: fd, cleanup: fd_res(fd)} as writer } else { fd as writer } } fn mk_file_writer(path: str, flags: [fileflag]) -> result { #[cfg(target_os = "win32")] fn wb() -> c_int { (O_WRONLY | O_BINARY) as c_int } #[cfg(target_os = "linux")] #[cfg(target_os = "macos")] #[cfg(target_os = "freebsd")] fn wb() -> c_int { O_WRONLY as c_int } let mut fflags: c_int = wb(); for vec::each(flags) {|f| alt f { append { fflags |= O_APPEND as c_int; } create { fflags |= O_CREAT as c_int; } truncate { fflags |= O_TRUNC as c_int; } no_flag { } } } let fd = os::as_c_charp(path) {|pathbuf| libc::open(pathbuf, fflags, (S_IRUSR | S_IWUSR) as c_int) }; if fd < (0 as c_int) { result::err(#fmt("error opening %s: %s", path, os::last_os_error())) } else { result::ok(fd_writer(fd, true)) } } fn u64_to_le_bytes(n: u64, size: uint, f: fn([u8]/&) -> T) -> T { assert size <= 8u; alt size { 1u { f([n as u8]/&) } 2u { f([n as u8, (n >> 8) as u8]/&) } 4u { f([n as u8, (n >> 8) as u8, (n >> 16) as u8, (n >> 24) as u8]/&) } 8u { f([n as u8, (n >> 8) as u8, (n >> 16) as u8, (n >> 24) as u8, (n >> 32) as u8, (n >> 40) as u8, (n >> 48) as u8, (n >> 56) as u8]/&) } _ { let mut bytes: [u8] = [], i = size, n = n; while i > 0u { bytes += [(n & 255_u64) as u8]; n >>= 8_u64; i -= 1u; } f(bytes) } } } fn u64_to_be_bytes(n: u64, size: uint, f: fn([u8]/&) -> T) -> T { assert size <= 8u; alt size { 1u { f([n as u8]/&) } 2u { f([(n >> 8) as u8, n as u8]/&) } 4u { f([(n >> 24) as u8, (n >> 16) as u8, (n >> 8) as u8, n as u8]/&) } 8u { f([(n >> 56) as u8, (n >> 48) as u8, (n >> 40) as u8, (n >> 32) as u8, (n >> 24) as u8, (n >> 16) as u8, (n >> 8) as u8, n as u8]/&) } _ { let mut bytes: [u8] = []; let mut i = size; while i > 0u { let shift = ((i - 1u) * 8u) as u64; bytes += [(n >> shift) as u8]; i -= 1u; } f(bytes) } } } fn u64_from_be_bytes(data: [u8], start: uint, size: uint) -> u64 { let mut sz = size; assert (sz <= 8u); let mut val = 0_u64; let mut pos = start; while sz > 0u { sz -= 1u; val += (data[pos] as u64) << ((sz * 8u) as u64); pos += 1u; } ret val; } impl writer_util for writer { fn write_char(ch: char) { if ch as uint < 128u { self.write([ch as u8]/&); } else { self.write_str(str::from_char(ch)); } } fn write_str(s: str/&) { str::byte_slice(s) {|v| self.write(v); } } fn write_line(s: str/&) { self.write_str(s); self.write_str("\n"/&); } fn write_int(n: int) { int::to_str_bytes(n, 10u) {|buf| self.write(buf) } } fn write_uint(n: uint) { uint::to_str_bytes(false, n, 10u) {|buf| self.write(buf) } } fn write_le_uint(n: uint, size: uint) { u64_to_le_bytes(n as u64, size) {|v| self.write(v); } } fn write_le_int(n: int, size: uint) { u64_to_le_bytes(n as u64, size) {|v| self.write(v); } } fn write_be_uint(n: uint, size: uint) { u64_to_be_bytes(n as u64, size) {|v| self.write(v); } } fn write_be_int(n: int, size: uint) { u64_to_be_bytes(n as u64, size) {|v| self.write(v); } } fn write_be_u64(n: u64) { u64_to_be_bytes(n, 8u) {|v| self.write(v); } } fn write_be_u32(n: u32) { u64_to_be_bytes(n as u64, 4u) {|v| self.write(v); } } fn write_be_u16(n: u16) { u64_to_be_bytes(n as u64, 2u) {|v| self.write(v); } } fn write_be_i64(n: i64) { u64_to_be_bytes(n as u64, 8u) {|v| self.write(v); } } fn write_be_i32(n: i32) { u64_to_be_bytes(n as u64, 4u) {|v| self.write(v); } } fn write_be_i16(n: i16) { u64_to_be_bytes(n as u64, 2u) {|v| self.write(v); } } fn write_le_u64(n: u64) { u64_to_le_bytes(n, 8u) {|v| self.write(v); } } fn write_le_u32(n: u32) { u64_to_le_bytes(n as u64, 4u) {|v| self.write(v); } } fn write_le_u16(n: u16) { u64_to_le_bytes(n as u64, 2u) {|v| self.write(v); } } fn write_le_i64(n: i64) { u64_to_le_bytes(n as u64, 8u) {|v| self.write(v); } } fn write_le_i32(n: i32) { u64_to_le_bytes(n as u64, 4u) {|v| self.write(v); } } fn write_le_i16(n: i16) { u64_to_le_bytes(n as u64, 2u) {|v| self.write(v); } } fn write_u8(n: u8) { self.write([n]/&) } } fn file_writer(path: str, flags: [fileflag]) -> result { result::chain(mk_file_writer(path, flags), { |w| result::ok(w)}) } // FIXME: fileflags // #2004 fn buffered_file_writer(path: str) -> result { let f = os::as_c_charp(path) {|pathbuf| os::as_c_charp("w") {|modebuf| libc::fopen(pathbuf, modebuf) } }; ret if f as uint == 0u { result::err("error opening " + path) } else { result::ok(FILE_writer(f, true)) } } // FIXME it would be great if this could be a const // FIXME why are these different from the way stdin() is implemented? // #2004 fn stdout() -> writer { fd_writer(libc::STDOUT_FILENO as c_int, false) } fn stderr() -> writer { fd_writer(libc::STDERR_FILENO as c_int, false) } fn print(s: str) { stdout().write_str(s); } fn println(s: str) { stdout().write_line(s); } type mem_buffer = @{buf: dvec, mut pos: uint}; impl of writer for mem_buffer { fn write(v: [const u8]/&) { // Fast path. let vlen = vec::len(v); let buf_len = self.buf.len(); if self.pos == buf_len { self.buf.push_all(v); self.pos += vlen; ret; } // FIXME #2004--use memcpy here? let mut pos = self.pos, vpos = 0u; while vpos < vlen && pos < buf_len { self.buf.set_elt(pos, v[vpos]); pos += 1u; vpos += 1u; } self.buf.push_slice(v, vpos, vlen); self.pos += vlen; } fn seek(offset: int, whence: seek_style) { let pos = self.pos; let len = self.buf.len(); self.pos = seek_in_buf(offset, pos, len, whence); } fn tell() -> uint { self.pos } fn flush() -> int { 0 } } fn mem_buffer() -> mem_buffer { @{buf: dvec(), mut pos: 0u} } fn mem_buffer_writer(b: mem_buffer) -> writer { b as writer } fn mem_buffer_buf(b: mem_buffer) -> [u8] { b.buf.get() } fn mem_buffer_str(b: mem_buffer) -> str { str::from_bytes(b.buf.get()) } fn with_str_writer(f: fn(writer)) -> str { let buf = mem_buffer(); let wr = mem_buffer_writer(buf); f(wr); io::mem_buffer_str(buf) } fn with_buf_writer(f: fn(writer)) -> [u8] { let buf = mem_buffer(); let wr = mem_buffer_writer(buf); f(wr); io::mem_buffer_buf(buf) } // Utility functions fn seek_in_buf(offset: int, pos: uint, len: uint, whence: seek_style) -> uint { let mut bpos = pos as int; let blen = len as int; alt whence { seek_set { bpos = offset; } seek_cur { bpos += offset; } seek_end { bpos = blen + offset; } } if bpos < 0 { bpos = 0; } else if bpos > blen { bpos = blen; } ret bpos as uint; } fn read_whole_file_str(file: str) -> result { result::chain(read_whole_file(file), { |bytes| result::ok(str::from_bytes(bytes)) }) } // FIXME implement this in a low-level way. Going through the abstractions is // pointless. // #2004 fn read_whole_file(file: str) -> result<[u8], str> { result::chain(file_reader(file), { |rdr| result::ok(rdr.read_whole_stream()) }) } // fsync related mod fsync { enum level { // whatever fsync does on that platform fsync, // fdatasync on linux, similiar or more on other platforms fdatasync, // full fsync // // You must additionally sync the parent directory as well! fullfsync, } // Resource of artifacts that need to fsync on destruction resource res(arg: arg) { alt arg.opt_level { option::none { } option::some(level) { // fail hard if not succesful assert(arg.fsync_fn(arg.val, level) != -1); } } } type arg = { val: t, opt_level: option, fsync_fn: fn@(t, level) -> int }; // fsync file after executing blk // FIXME find better way to create resources within lifetime of outer res // #2004 fn FILE_res_sync(&&file: FILE_res, opt_level: option, blk: fn(&&res<*libc::FILE>)) { blk(res({ val: *file, opt_level: opt_level, fsync_fn: fn@(&&file: *libc::FILE, l: level) -> int { ret os::fsync_fd(libc::fileno(file), l) as int; } })); } // fsync fd after executing blk fn fd_res_sync(&&fd: fd_res, opt_level: option, blk: fn(&&res)) { blk(res({ val: *fd, opt_level: opt_level, fsync_fn: fn@(&&fd: fd_t, l: level) -> int { ret os::fsync_fd(fd, l) as int; } })); } // Type of objects that may want to fsync iface t { fn fsync(l: level) -> int; } // Call o.fsync after executing blk fn obj_sync(&&o: t, opt_level: option, blk: fn(&&res)) { blk(res({ val: o, opt_level: opt_level, fsync_fn: fn@(&&o: t, l: level) -> int { ret o.fsync(l); } })); } } #[cfg(test)] mod tests { #[test] fn test_simple() { let tmpfile: str = "tmp/lib-io-test-simple.tmp"; log(debug, tmpfile); let frood: str = "A hoopy frood who really knows where his towel is."; log(debug, frood); { let out: io::writer = result::get( io::file_writer(tmpfile, [io::create, io::truncate])); out.write_str(frood); } let inp: io::reader = result::get(io::file_reader(tmpfile)); let frood2: str = inp.read_c_str(); log(debug, frood2); assert (str::eq(frood, frood2)); } #[test] fn test_readchars_empty() { let inp : io::reader = io::str_reader(""); let res : [char] = inp.read_chars(128u); assert(vec::len(res) == 0u); } #[test] fn test_readchars_wide() { let wide_test = "生锈的汤匙切肉汤hello生锈的汤匙切肉汤"; let ivals : [int] = [ 29983, 38152, 30340, 27748, 21273, 20999, 32905, 27748, 104, 101, 108, 108, 111, 29983, 38152, 30340, 27748, 21273, 20999, 32905, 27748]; fn check_read_ln(len : uint, s: str, ivals: [int]) { let inp : io::reader = io::str_reader(s); let res : [char] = inp.read_chars(len); if (len <= vec::len(ivals)) { assert(vec::len(res) == len); } assert(vec::slice(ivals, 0u, vec::len(res)) == vec::map(res, {|x| x as int})); } let mut i = 0u; while i < 8u { check_read_ln(i, wide_test, ivals); i += 1u; } // check a long read for good measure check_read_ln(128u, wide_test, ivals); } #[test] fn test_readchar() { let inp : io::reader = io::str_reader("生"); let res : char = inp.read_char(); assert(res as int == 29983); } #[test] fn test_readchar_empty() { let inp : io::reader = io::str_reader(""); let res : char = inp.read_char(); assert(res as int == -1); } #[test] fn file_reader_not_exist() { alt io::file_reader("not a file") { result::err(e) { assert e == "error opening not a file"; } result::ok(_) { fail; } } } #[test] fn file_writer_bad_name() { alt io::file_writer("?/?", []) { result::err(e) { assert str::starts_with(e, "error opening ?/?"); } result::ok(_) { fail; } } } #[test] fn buffered_file_writer_bad_name() { alt io::buffered_file_writer("?/?") { result::err(e) { assert e == "error opening ?/?"; } result::ok(_) { fail; } } } #[test] fn mem_buffer_overwrite() { let mbuf = mem_buffer(); mbuf.write([0u8, 1u8, 2u8, 3u8]); assert mem_buffer_buf(mbuf) == [0u8, 1u8, 2u8, 3u8]; mbuf.seek(-2, seek_cur); mbuf.write([4u8, 5u8, 6u8, 7u8]); assert mem_buffer_buf(mbuf) == [0u8, 1u8, 4u8, 5u8, 6u8, 7u8]; mbuf.seek(-2, seek_end); mbuf.write([8u8]); mbuf.seek(1, seek_set); mbuf.write([9u8]); assert mem_buffer_buf(mbuf) == [0u8, 9u8, 4u8, 5u8, 8u8, 7u8]; } } // // Local Variables: // mode: rust // fill-column: 78; // indent-tabs-mode: nil // c-basic-offset: 4 // buffer-file-coding-system: utf-8-unix // End: //