mykola/filepwn
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

almost got working it on linux, directory permissions needs to be fixed

Browse source
This commit is contained in:
mykola2312 2024-03-18 02:40:38 +02:00
parent 1df268d7f2
commit 7fc4e37a30
1 changed files with 30 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

50
src/main.rs
View file

@ -1,8 +1,10 @@
use clap::Parser; use clap::Parser;
use std::fs::{self, DirEntry}; use std::fs;
use std::fs::{File, DirEntry};
use std::collections::HashMap; use std::collections::HashMap;
use std::num::ParseIntError; use std::num::ParseIntError;
use std::path::Path; use std::path::Path;
use std::os::unix::fs::{chown, PermissionsExt};
#[derive(Parser, Debug)] #[derive(Parser, Debug)]
#[command(version, about)] #[command(version, about)]
@ -56,15 +58,6 @@ fn parse_etc_file(path: &str) -> Result<HashMap<String, u32>, ParserError> {
.collect() .collect()
} }
#[cfg(debug_assertions)]
const PASSWD_PATH: &str = "passwd";
#[cfg(not(debug_assertions))]
const PASSWD_PATH: &str = "/etc/passwd";
#[cfg(debug_assertions)]
const GROUP_PATH: &str = "group";
#[cfg(not(debug_assertions))]
const GROUP_PATH: &str = "/etc/group";
fn traverse_filesystem(path: &Path) -> (Vec<String>, Vec<String>){ fn traverse_filesystem(path: &Path) -> (Vec<String>, Vec<String>){
let mut files: Vec<String> = Vec::new(); let mut files: Vec<String> = Vec::new();
let mut directories: Vec<String> = Vec::new(); let mut directories: Vec<String> = Vec::new();
@ -121,16 +114,25 @@ fn traverse_filesystem(path: &Path) -> (Vec<String>, Vec<String>){
(files, directories) (files, directories)
} }
fn set_permissions(path: &str, mode: u32) -> Result<(), std::io::Error> {
let file = File::create(path)?;
let metadata = file.metadata()?;
let mut permissions = metadata.permissions();
permissions.set_mode(mode);
Ok(())
}
fn main() { fn main() {
let args = Args::parse(); let args = Args::parse();
let users = parse_etc_file(PASSWD_PATH) let users = parse_etc_file("/etc/passwd")
.expect("failed to parse passwd file"); .expect("failed to parse passwd file");
let groups = parse_etc_file(GROUP_PATH) let groups = parse_etc_file("/etc/group")
.expect("failed to parse group file"); .expect("failed to parse group file");
let uid = users.get(&args.user).expect("user not found"); let uid = *users.get(&args.user).expect("user not found");
let gid = groups.get(&args.group).expect("group not found"); let gid = *groups.get(&args.group).expect("group not found");
let file_permissions = u32::from_str_radix(args.file_permissions.as_str(), 8) let file_permissions = u32::from_str_radix(args.file_permissions.as_str(), 8)
.expect("file permissions must be an octal number"); .expect("file permissions must be an octal number");
@ -141,15 +143,23 @@ fn main() {
return; return;
} }
dbg!(file_permissions, directory_permissions);
let (files, directories) = traverse_filesystem(Path::new(&args.path)); let (files, directories) = traverse_filesystem(Path::new(&args.path));
for file in files { for file in &files {
// do the file permissions if let Err(e) = set_permissions(&file, file_permissions) {
eprintln!("failed to set {file} permissions: {e}");
}
if let Err(e) = chown(file, Some(uid), Some(gid)) {
eprintln!("failed to set {file} ownership: {e}");
}
} }
for directory in directories { for directory in &directories {
// do the directory permissions if let Err(e) = set_permissions(&directory, directory_permissions) {
eprintln!("failed to set {directory} permissions: {e}");
}
if let Err(e) = chown(directory, Some(uid), Some(gid)) {
eprintln!("failed to set {directory} ownership: {e}");
}
} }
println!("All done!") println!("All done!")