mirror of
https://github.com/RGBCube/uutils-coreutils
synced 2025-07-28 03:27:44 +00:00
Heather
commit
2e44b5b7ee
4 changed files with 304 additions and 1 deletions
2
Makefile
2
Makefile
|
|
@ -24,6 +24,7 @@ PROGS := \
|
|||
seq \
|
||||
tee \
|
||||
true \
|
||||
truncate \
|
||||
users \
|
||||
wc \
|
||||
whoami \
|
||||
|
|
@ -41,6 +42,7 @@ TEST_PROGS := \
|
|||
cat \
|
||||
mkdir \
|
||||
seq \
|
||||
truncate \
|
||||
|
||||
TEST ?= $(TEST_PROGS)
|
||||
|
||||
|
|
|
|||
|
|
@ -155,7 +155,6 @@ To do
|
|||
- timeout
|
||||
- touch
|
||||
- tr
|
||||
- truncate
|
||||
- tsort
|
||||
- uname-arch
|
||||
- uname-uname
|
||||
|
|
|
|||
40
truncate/test.rs
Normal file
40
truncate/test.rs
Normal file
|
|
@ -0,0 +1,40 @@
|
|||
use std::{run, io};
|
||||
|
||||
static PROG: &'static str = "build/truncate";
|
||||
static TESTNAME: &'static str = "THISISARANDOMFILENAME";
|
||||
|
||||
fn make_file() -> io::File {
|
||||
while Path::new(TESTNAME).exists() { io::timer::sleep(1000); }
|
||||
match io::File::create(&Path::new(TESTNAME)) {
|
||||
Ok(f) => f,
|
||||
Err(_) => fail!()
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_increase_file_size() {
|
||||
let mut file = make_file();
|
||||
if !run::process_status(PROG, [~"-s", ~"+5K", TESTNAME.to_owned()]).unwrap().success() {
|
||||
fail!();
|
||||
}
|
||||
file.seek(0, io::SeekEnd);
|
||||
if file.tell().unwrap() != 5 * 1024 {
|
||||
fail!();
|
||||
}
|
||||
io::fs::unlink(&Path::new(TESTNAME));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_decrease_file_size() {
|
||||
let mut file = make_file();
|
||||
file.write(bytes!("1234567890"));
|
||||
if !run::process_status(PROG, [~"--size=-4", TESTNAME.to_owned()]).unwrap().success() {
|
||||
fail!();
|
||||
}
|
||||
file.seek(0, io::SeekEnd);
|
||||
if file.tell().unwrap() != 6 {
|
||||
println!("{}", file.tell());
|
||||
fail!();
|
||||
}
|
||||
io::fs::unlink(&Path::new(TESTNAME));
|
||||
}
|
||||
262
truncate/truncate.rs
Normal file
262
truncate/truncate.rs
Normal file
|
|
@ -0,0 +1,262 @@
|
|||
#[crate_id(name="truncate", vers="1.0.0", author="Arcterus")];
|
||||
|
||||
/*
|
||||
* This file is part of the uutils coreutils package.
|
||||
*
|
||||
* (c) Arcterus <arcterus@mail.com>
|
||||
*
|
||||
* For the full copyright and license information, please view the LICENSE
|
||||
* file that was distributed with this source code.
|
||||
*/
|
||||
|
||||
#[feature(macro_rules)];
|
||||
|
||||
extern mod extra;
|
||||
|
||||
use std::io::{stderr, File, Open, ReadWrite, Writer, SeekEnd, SeekSet};
|
||||
use std::os;
|
||||
use std::u64;
|
||||
use extra::getopts::groups;
|
||||
|
||||
macro_rules! get_file_size(
|
||||
($file:ident, $action:expr) => ({
|
||||
match $file.seek(0, SeekEnd) {
|
||||
Ok(_) => {}
|
||||
Err(f) => {
|
||||
writeln!(&mut stderr() as &mut Writer, "{}", f.to_str());
|
||||
os::set_exit_status(1);
|
||||
$action
|
||||
}
|
||||
}
|
||||
let size = match $file.tell() {
|
||||
Ok(m) => m,
|
||||
Err(f) => {
|
||||
writeln!(&mut stderr() as &mut Writer, "{}", f.to_str());
|
||||
os::set_exit_status(1);
|
||||
$action
|
||||
}
|
||||
};
|
||||
match $file.seek(0, SeekSet) {
|
||||
Ok(_) => {}
|
||||
Err(f) => {
|
||||
writeln!(&mut stderr() as &mut Writer, "{}", f.to_str());
|
||||
os::set_exit_status(1);
|
||||
$action
|
||||
}
|
||||
}
|
||||
size
|
||||
})
|
||||
)
|
||||
|
||||
#[deriving(Eq)]
|
||||
enum TruncateMode {
|
||||
Reference,
|
||||
Extend,
|
||||
Reduce,
|
||||
AtMost,
|
||||
AtLeast,
|
||||
RoundDown,
|
||||
RoundUp
|
||||
}
|
||||
|
||||
fn main() {
|
||||
let args = os::args();
|
||||
let program = args[0].clone();
|
||||
|
||||
let opts = ~[
|
||||
groups::optflag("c", "no-create", "do not create files that do not exist"),
|
||||
groups::optflag("o", "io-blocks", "treat SIZE as the number of I/O blocks of the file rather than bytes (NOT IMPLEMENTED)"),
|
||||
groups::optopt("r", "reference", "base the size of each file on the size of RFILE", "RFILE"),
|
||||
groups::optopt("s", "size", "set or adjust the size of each file according to SIZE, which is in bytes unless --io-blocks is specified", "SIZE"),
|
||||
groups::optflag("h", "help", "display this help and exit"),
|
||||
groups::optflag("V", "version", "output version information and exit")
|
||||
];
|
||||
let matches = match groups::getopts(args.tail(), opts) {
|
||||
Ok(m) => m,
|
||||
Err(f) => {
|
||||
writeln!(&mut stderr() as &mut Writer, "{}", f.to_err_msg());
|
||||
os::set_exit_status(1);
|
||||
return
|
||||
}
|
||||
};
|
||||
|
||||
if matches.opt_present("help") {
|
||||
println!("truncate 1.0.0");
|
||||
println!("");
|
||||
println!("Usage:");
|
||||
println!(" {0:s} [OPTION]... FILE...", program);
|
||||
println!("");
|
||||
print!("{}", groups::usage("Shrink or extend the size of each file to the specified size.", opts));
|
||||
print!("
|
||||
SIZE is an integer with an optional prefix and optional unit.
|
||||
The available units (K, M, G, T, P, E, Z, and Y) use the following format:
|
||||
'KB' => 1000 (kilobytes)
|
||||
'K' => 1024 (kibibytes)
|
||||
'MB' => 1000*1000 (megabytes)
|
||||
'M' => 1024*1024 (mebibytes)
|
||||
'GB' => 1000*1000*1000 (gigabytes)
|
||||
'G' => 1024*1024*1024 (gibibytes)
|
||||
SIZE may also be prefixed by one of the following to adjust the size of each
|
||||
file based on its current size:
|
||||
'+' => extend by
|
||||
'-' => reduce by
|
||||
'<' => at most
|
||||
'>' => at least
|
||||
'/' => round down to multiple of
|
||||
'%' => round up to multiple of
|
||||
");
|
||||
} else if matches.opt_present("version") {
|
||||
println!("truncate 1.0.0");
|
||||
} else if matches.free.is_empty() {
|
||||
writeln!(&mut stderr() as &mut Writer, "Missing an argument");
|
||||
writeln!(&mut stderr() as &mut Writer,
|
||||
"For help, try '{0:s} --help'", program);
|
||||
os::set_exit_status(1);
|
||||
} else {
|
||||
let no_create = matches.opt_present("no-create");
|
||||
let io_blocks = matches.opt_present("io-blocks");
|
||||
let reference = matches.opt_str("reference");
|
||||
let size = matches.opt_str("size");
|
||||
if reference.is_none() && size.is_none() {
|
||||
writeln!(&mut stderr() as &mut Writer, "You must specify either --reference or --size.");
|
||||
os::set_exit_status(1);
|
||||
} else {
|
||||
truncate(no_create, io_blocks, reference, size, matches.free);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn truncate(no_create: bool, io_blocks: bool, reference: Option<~str>, size: Option<~str>, filenames: ~[~str]) {
|
||||
let (refsize, mode) = match reference {
|
||||
Some(rfilename) => {
|
||||
let mut rfile = match File::open(&Path::new(rfilename.clone())) {
|
||||
Ok(m) => m,
|
||||
Err(f) => {
|
||||
writeln!(&mut stderr() as &mut Writer, "{}", f.to_str());
|
||||
os::set_exit_status(1);
|
||||
return
|
||||
}
|
||||
};
|
||||
(get_file_size!(rfile, return), Reference)
|
||||
}
|
||||
None => {
|
||||
match parse_size(size.unwrap()) {
|
||||
Ok(szpair) => szpair,
|
||||
Err(()) => return
|
||||
}
|
||||
}
|
||||
};
|
||||
for filename in filenames.iter() {
|
||||
let filename: &str = *filename;
|
||||
let path = Path::new(filename);
|
||||
if path.exists() || !no_create {
|
||||
match File::open_mode(&path, Open, ReadWrite) {
|
||||
Ok(mut file) => {
|
||||
let fsize = get_file_size!(file, continue);
|
||||
let tsize = match mode {
|
||||
Reference => refsize,
|
||||
Extend => fsize + refsize,
|
||||
Reduce => fsize - refsize,
|
||||
AtMost => if fsize > refsize { refsize } else { fsize },
|
||||
AtLeast => if fsize < refsize { refsize } else { fsize },
|
||||
RoundDown => fsize - fsize % refsize,
|
||||
RoundUp => fsize + fsize % refsize
|
||||
};
|
||||
match file.truncate(tsize as i64) {
|
||||
Ok(_) => {}
|
||||
Err(f) => {
|
||||
writeln!(&mut stderr() as &mut Writer,
|
||||
"{}", f.to_str());
|
||||
os::set_exit_status(1);
|
||||
}
|
||||
}
|
||||
}
|
||||
Err(f) => {
|
||||
writeln!(&mut stderr() as &mut Writer, "{}", f.to_str());
|
||||
os::set_exit_status(1);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn parse_size(size: ~str) -> Result<(u64, TruncateMode), ()> {
|
||||
let mut err = false;
|
||||
let mode = match size.char_at(0) {
|
||||
'+' => Extend,
|
||||
'-' => Reduce,
|
||||
'<' => AtMost,
|
||||
'>' => AtLeast,
|
||||
'/' => RoundDown,
|
||||
'*' => RoundUp,
|
||||
_ => Reference /* assume that the size is just a number */
|
||||
};
|
||||
let bytes = {
|
||||
let mut slice =
|
||||
if mode == Reference {
|
||||
let size: &str = size;
|
||||
size
|
||||
} else {
|
||||
size.slice_from(1)
|
||||
};
|
||||
if slice.char_at(slice.len() - 1).is_alphabetic() {
|
||||
slice = slice.slice_to(slice.len() - 1);
|
||||
if slice.len() > 0 && slice.char_at(slice.len() - 1).is_alphabetic() {
|
||||
slice = slice.slice_to(slice.len() - 1);
|
||||
}
|
||||
}
|
||||
slice
|
||||
}.bytes().to_owned_vec();
|
||||
let mut number = match u64::parse_bytes(bytes, 10) {
|
||||
Some(num) => num,
|
||||
None => {
|
||||
writeln!(&mut stderr() as &mut Writer,
|
||||
"'{}' is not a valid number.", size);
|
||||
os::set_exit_status(1);
|
||||
err = true;
|
||||
0
|
||||
}
|
||||
};
|
||||
if !err && size.char_at(size.len() - 1).is_alphabetic() {
|
||||
number *= match size.char_at(size.len() - 1).to_ascii().to_upper().to_char() {
|
||||
'B' => match size.char_at(size.len() - 2).to_ascii().to_upper().to_char() {
|
||||
'K' => 1000,
|
||||
'M' => 1000 * 1000,
|
||||
'G' => 1000 * 1000 * 1000,
|
||||
'T' => 1000 * 1000 * 1000 * 1000,
|
||||
'P' => 1000 * 1000 * 1000 * 1000 * 1000,
|
||||
'E' => 1000 * 1000 * 1000 * 1000 * 1000 * 1000,
|
||||
'Z' => 1000 * 1000 * 1000 * 1000 * 1000 * 1000 * 1000,
|
||||
'Y' => 1000 * 1000 * 1000 * 1000 * 1000 * 1000 * 1000 * 1000,
|
||||
letter => {
|
||||
writeln!(&mut stderr() as &mut Writer,
|
||||
"'{}B' is not a valid suffix.", letter);
|
||||
os::set_exit_status(1);
|
||||
err = true;
|
||||
1
|
||||
}
|
||||
},
|
||||
'K' => 1024,
|
||||
'M' => 1024 * 1024,
|
||||
'G' => 1024 * 1024 * 1024,
|
||||
'T' => 1024 * 1024 * 1024 * 1024,
|
||||
'P' => 1024 * 1024 * 1024 * 1024 * 1024,
|
||||
'E' => 1024 * 1024 * 1024 * 1024 * 1024 * 1024,
|
||||
'Z' => 1024 * 1024 * 1024 * 1024 * 1024 * 1024 * 1024,
|
||||
'Y' => 1024 * 1024 * 1024 * 1024 * 1024 * 1024 * 1024 * 1024,
|
||||
letter => {
|
||||
writeln!(&mut stderr() as &mut Writer,
|
||||
"'{}' is not a valid suffix.", letter);
|
||||
os::set_exit_status(1);
|
||||
err = true;
|
||||
1
|
||||
}
|
||||
};
|
||||
}
|
||||
if err {
|
||||
Err(())
|
||||
} else {
|
||||
Ok((number, mode))
|
||||
}
|
||||
}
|
||||
|
||||
Loading…
Add table
Add a link
Reference in a new issue