busycrate 0.1.0

BusyBox but in Rust
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

/* This Source Code Form is subject to the terms of the Mozilla Public
 *  * License, v. 2.0. If a copy of the MPL was not distributed with this
 *   * file, You can obtain one at https://mozilla.org/MPL/2.0/. */

use std::path::Path;
use nix::fcntl::OFlag;
use nix::sys::stat::Mode;

pub struct Args<'a> {
    pub paths: Vec<&'a Path>,
}

pub fn main(args: Args) -> i32 {
    if args.paths.is_empty() {
        eprintln!("missing file operand");
        eprintln!("Try 'touch --help' for more information");
        return crate::EXIT_CODE_INVALID_USAGE;
    }

    let mut status = 0;
    for fpath in args.paths {
        let flags = OFlag::O_CREAT | OFlag::O_WRONLY;
        let mode = Mode::S_IRUSR | Mode::S_IWUSR; // user can read+write
        let mode = mode | Mode::S_IRGRP | Mode::S_IWGRP; // group can read+write
        let mode = mode | Mode::S_IROTH | Mode::S_IWOTH; // others can read+write

        let fd = match nix::fcntl::open(fpath, flags, mode) {
            Ok(fd) => fd,
            Err(e) => {
                eprintln!("Unable to create file '{}': {}", fpath.display(), e);
                status = crate::EXIT_CODE_UNKNOWN_ERR;
                continue
            }
        };

        if let Err(e) = nix::unistd::close(fd) {
            eprintln!("Error closing file '{}': {}", fpath.display(), e);
            // don't set the status code here since, really, the fd will be closed regardless of
            // any error. The stderr message is here just to let the user know that _something_
            // happened. If we were writing anything to the file, there would be potential for
            // dataloss, but we aren't, so we don't care.
        }
    }
    return status;
}