RGBCube/uutils-coreutils
1
Fork 0
mirror of https://github.com/RGBCube/uutils-coreutils synced 2025-07-30 04:27:45 +00:00
Code Issues Activity

Merge pull request #542 from kwantam/master

Browse source 
update cat, cat tests to new API
This commit is contained in:
Heather 2015-04-25 16:28:05 +03:00
commit b81a751ebd
2 changed files with 78 additions and 55 deletions
Download patch file Download diff file Expand all files Collapse all files

94
src/cat/cat.rs
View file

@ -1,5 +1,5 @@
#![crate_name = "cat"]
#![feature(collections, core, old_io, old_path, rustc_private)]
#![feature(collections, rustc_private)]
#![feature(box_syntax, unsafe_destructor)]
@ -15,11 +15,15 @@
/* last synced with: cat (GNU coreutils) 8.13 */
extern crate getopts;
extern crate libc;
use std::old_io::{print, File};
use std::old_io::stdio::{stdout_raw, stdin_raw, stderr};
use std::old_io::{IoResult};
use std::ptr::{copy_nonoverlapping_memory};
use std::fs::File;
use std::io::{stdout, stdin, stderr, Write, Read};
use std::io::Result;
use std::intrinsics::{copy_nonoverlapping};
use libc::consts::os::posix88::STDIN_FILENO;
use libc::funcs::posix88::unistd::isatty;
use libc::types::os::arch::c95::c_int;
pub fn uumain(args: Vec<String>) -> i32 {
let program = &args[0];
@ -48,7 +52,7 @@ pub fn uumain(args: Vec<String>) -> i32 {
println!("Usage:");
println!(" {0} [OPTION]... [FILE]...", program);
println!("");
print(&getopts::usage("Concatenate FILE(s), or standard input, to \
print!("{}", &getopts::usage("Concatenate FILE(s), or standard input, to \
standard output.", &opts)[..]);
println!("");
println!("With no FILE, or when FILE is -, read standard input.");
@ -99,13 +103,13 @@ fn write_lines(files: Vec<String>, number: NumberingMode, squeeze_blank: bool,
let mut in_buf = [0; 1024 * 31];
let mut out_buf = [0; 1024 * 64];
let mut writer = UnsafeWriter::new(out_buf.as_mut_slice(), stdout_raw());
let mut writer = UnsafeWriter::new(&mut out_buf[..], stdout());
let mut at_line_start = true;
while let Ok(n) = reader.read(&mut in_buf) {
if n == 0 { break }
let in_buf = &in_buf[..n];
let mut buf_pos = range(0, n);
let mut buf_pos = 0..n;
loop {
writer.possibly_flush();
let pos = match buf_pos.next() {
@ -119,9 +123,9 @@ fn write_lines(files: Vec<String>, number: NumberingMode, squeeze_blank: bool,
line_counter += 1;
}
if show_ends {
writer.write_u8('$' as u8).unwrap();
writer.write_all(&['$' as u8]).unwrap();
}
writer.write_u8('\n' as u8).unwrap();
writer.write_all(&['\n' as u8]).unwrap();
if interactive {
writer.flush().unwrap();
}
@ -137,13 +141,13 @@ fn write_lines(files: Vec<String>, number: NumberingMode, squeeze_blank: bool,
Some(p) => {
writer.write_all(&in_buf[pos..pos + p]).unwrap();
if show_ends {
writer.write_u8('$' as u8).unwrap();
writer.write_all(&['$' as u8]).unwrap();
}
writer.write_u8('\n' as u8).unwrap();
writer.write_all(&['\n' as u8]).unwrap();
if interactive {
writer.flush().unwrap();
}
buf_pos = range(pos + p + 1, n);
buf_pos = pos + p + 1..n;
at_line_start = true;
},
None => {
@ -165,11 +169,11 @@ fn write_bytes(files: Vec<String>, number: NumberingMode, squeeze_blank: bool,
for (mut reader, interactive) in files.iter().filter_map(|p| open(&p[..])) {
// Flush all 1024 iterations.
let mut flush_counter = range(0usize, 1024);
let mut flush_counter = 0usize..1024;
let mut in_buf = [0; 1024 * 32];
let mut out_buf = [0; 1024 * 64];
let mut writer = UnsafeWriter::new(out_buf.as_mut_slice(), stdout_raw());
let mut writer = UnsafeWriter::new(&mut out_buf[..], stdout());
let mut at_line_start = true;
while let Ok(n) = reader.read(&mut in_buf) {
if n == 0 { break }
@ -177,7 +181,7 @@ fn write_bytes(files: Vec<String>, number: NumberingMode, squeeze_blank: bool,
for &byte in in_buf[..n].iter() {
if flush_counter.next().is_none() {
writer.possibly_flush();
flush_counter = range(0usize, 1024);
flush_counter = 0usize..1024;
}
if byte == '\n' as u8 {
if !at_line_start || !squeeze_blank {
@ -186,9 +190,9 @@ fn write_bytes(files: Vec<String>, number: NumberingMode, squeeze_blank: bool,
line_counter += 1;
}
if show_ends {
writer.write_u8('$' as u8).unwrap();
writer.write_all(&['$' as u8]).unwrap();
}
writer.write_u8('\n' as u8).unwrap();
writer.write_all(&['\n' as u8]).unwrap();
if interactive {
writer.flush().unwrap();
}
@ -205,14 +209,14 @@ fn write_bytes(files: Vec<String>, number: NumberingMode, squeeze_blank: bool,
// this by having the whole loop inside show_nonprint.
if byte == '\t' as u8 {
if show_tabs {
writer.write_str("^I")
writer.write_all("^I".as_bytes())
} else {
writer.write_u8(byte)
writer.write_all(&[byte])
}
} else if show_nonprint {
let byte = match byte {
128 ... 255 => {
writer.write_str("M-").unwrap();
writer.write_all("M-".as_bytes()).unwrap();
byte - 128
},
_ => byte,
@ -220,10 +224,10 @@ fn write_bytes(files: Vec<String>, number: NumberingMode, squeeze_blank: bool,
match byte {
0 ... 31 => writer.write_all(&['^' as u8, byte + 64]),
127 => writer.write_all(&['^' as u8, byte - 64]),
_ => writer.write_u8(byte),
_ => writer.write_all(&[byte]),
}
} else {
writer.write_u8(byte)
writer.write_all(&[byte])
}.unwrap();
}
}
@ -231,7 +235,7 @@ fn write_bytes(files: Vec<String>, number: NumberingMode, squeeze_blank: bool,
}
fn write_fast(files: Vec<String>) {
let mut writer = stdout_raw();
let mut writer = stdout();
let mut in_buf = [0; 1024 * 64];
for (mut reader, _) in files.iter().filter_map(|p| open(&p[..])) {
@ -255,15 +259,15 @@ fn exec(files: Vec<String>, number: NumberingMode, show_nonprint: bool,
}
}
fn open(path: &str) -> Option<(Box<Reader>, bool)> {
fn open(path: &str) -> Option<(Box<Read>, bool)> {
if path == "-" {
let stdin = stdin_raw();
let interactive = stdin.isatty();
return Some((box stdin as Box<Reader>, interactive));
let stdin = stdin();
let interactive = unsafe { isatty(STDIN_FILENO) } != 0 as c_int;
return Some((box stdin as Box<Read>, interactive));
}
match File::open(&std::old_path::Path::new(path)) {
Ok(f) => Some((box f as Box<Reader>, false)),
match File::open(path) {
Ok(f) => Some((box f as Box<Read>, false)),
Err(e) => {
(writeln!(&mut stderr(), "cat: {0}: {1}", path, e.to_string())).unwrap();
None
@ -271,14 +275,14 @@ fn open(path: &str) -> Option<(Box<Reader>, bool)> {
}
}
struct UnsafeWriter<'a, W> {
struct UnsafeWriter<'a, W: Write> {
inner: W,
buf: &'a mut [u8],
pos: usize,
threshold: usize,
}
impl<'a, W: Writer> UnsafeWriter<'a, W> {
impl<'a, W: Write> UnsafeWriter<'a, W> {
fn new(buf: &'a mut [u8], inner: W) -> UnsafeWriter<'a, W> {
let threshold = buf.len()/2;
UnsafeWriter {
@ -289,11 +293,14 @@ impl<'a, W: Writer> UnsafeWriter<'a, W> {
}
}
fn flush_buf(&mut self) -> IoResult<()> {
fn flush_buf(&mut self) -> Result<()> {
if self.pos != 0 {
let ret = self.inner.write_all(&self.buf[..self.pos]);
let ret = self.inner.write(&self.buf[..self.pos]);
self.pos = 0;
ret
match ret {
Ok(_) => Ok(()),
Err(e) => Err(e),
}
} else {
Ok(())
}
@ -312,26 +319,27 @@ fn fail() -> ! {
panic!("assertion failed");
}
impl<'a, W: Writer> Writer for UnsafeWriter<'a, W> {
fn write_all(&mut self, buf: &[u8]) -> IoResult<()> {
impl<'a, W: Write> Write for UnsafeWriter<'a, W> {
fn write(&mut self, buf: &[u8]) -> Result<usize> {
let dst = &mut self.buf[self.pos..];
if buf.len() > dst.len() {
let len = buf.len();
if len > dst.len() {
fail();
}
unsafe {
copy_nonoverlapping_memory(dst.as_mut_ptr(), buf.as_ptr(), buf.len())
copy_nonoverlapping(buf.as_ptr(), dst.as_mut_ptr(), len)
}
self.pos += buf.len();
Ok(())
self.pos += len;
Ok(len)
}
fn flush(&mut self) -> IoResult<()> {
fn flush(&mut self) -> Result<()> {
self.flush_buf().and_then(|()| self.inner.flush())
}
}
#[unsafe_destructor]
impl<'a, W: Writer> Drop for UnsafeWriter<'a, W> {
impl<'a, W: Write> Drop for UnsafeWriter<'a, W> {
fn drop(&mut self) {
let _ = self.flush_buf();
}

39
test/cat.rs
View file

@ -1,6 +1,6 @@
#![allow(unstable)]
use std::old_io::process::Command;
use std::io::Write;
use std::process::Command;
use std::process::Stdio;
use std::str;
static PROGNAME: &'static str = "./cat";
@ -17,29 +17,44 @@ fn test_output_multi_files_print_all_chars() {
Err(err) => panic!("{}", err),
};
let out = str::from_utf8(po.output.as_slice()).unwrap();
let out = str::from_utf8(&po.stdout[..]).unwrap();
assert_eq!(out,
" 1\tabcde$\n 2\tfghij$\n 3\tklmno$\n 4\tpqrst$\n 5\tuvwxyz$\n 6\t^@^A^B^C^D^E^F^G^H^I$\n 7\t^K^L^M^N^O^P^Q^R^S^T^U^V^W^X^Y^Z^[^\\^]^^^_ !\"#$%&\'()*+,-./0123456789:;<=>?@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^_`abcdefghijklmnopqrstuvwxyz{|}~^?M-^@M-^AM-^BM-^CM-^DM-^EM-^FM-^GM-^HM-^IM-^JM-^KM-^LM-^MM-^NM-^OM-^PM-^QM-^RM-^SM-^TM-^UM-^VM-^WM-^XM-^YM-^ZM-^[M-^\\M-^]M-^^M-^_M- M-!M-\"M-#M-$M-%M-&M-\'M-(M-)M-*M-+M-,M--M-.M-/M-0M-1M-2M-3M-4M-5M-6M-7M-8M-9M-:M-;M-<M-=M->M-?M-@M-AM-BM-CM-DM-EM-FM-GM-HM-IM-JM-KM-LM-MM-NM-OM-PM-QM-RM-SM-TM-UM-VM-WM-XM-YM-ZM-[M-\\M-]M-^M-_M-`M-aM-bM-cM-dM-eM-fM-gM-hM-iM-jM-kM-lM-mM-nM-oM-pM-qM-rM-sM-tM-uM-vM-wM-xM-yM-zM-{M-|M-}M-~M-^?");
}
#[test]
fn test_stdin_squeeze() {
let mut process= Command::new(PROGNAME).arg("-A").spawn().unwrap();
let mut process = Command::new(PROGNAME)
.arg("-A")
.stdin(Stdio::piped())
.stdout(Stdio::piped())
.spawn()
.unwrap_or_else(|e| panic!("{}", e));
process.stdin.take().unwrap().write_all(b"\x00\x01\x02").unwrap();
let po = process.wait_with_output().unwrap();
let out = str::from_utf8(po.output.as_slice()).unwrap();
process.stdin.take().unwrap_or_else(|| panic!("Could not grab child process stdin"))
.write_all("\x00\x01\x02".as_bytes()).unwrap_or_else(|e| panic!("{}", e));
let po = process.wait_with_output().unwrap_or_else(|e| panic!("{}", e));
let out = str::from_utf8(&po.stdout[..]).unwrap_or_else(|e| panic!("{}", e));
assert_eq!(out, "^@^A^B");
}
#[test]
fn test_stdin_number_non_blank() {
let mut process = Command::new(PROGNAME).arg("-b").arg("-").spawn().unwrap();
let mut process = Command::new(PROGNAME)
.arg("-b")
.arg("-")
.stdin(Stdio::piped())
.stdout(Stdio::piped())
.spawn()
.unwrap_or_else(|e| panic!("{}", e));
process.stdin.take().unwrap().write_all(b"\na\nb\n\n\nc").unwrap();
let po = process.wait_with_output().unwrap();
let out = str::from_utf8(po.output.as_slice()).unwrap();
process.stdin.take().unwrap_or_else(|| panic!("Could not grab child process stdin"))
.write_all("\na\nb\n\n\nc".as_bytes()).unwrap_or_else(|e| panic!("{}", e));
let po = process.wait_with_output().unwrap_or_else(|e| panic!("{}", e));
let out = str::from_utf8(&po.stdout[..]).unwrap_or_else(|e| panic!("{}", e));
assert_eq!(out, "\n 1\ta\n 2\tb\n\n\n 3\tc");
}