Skip to content

Commit aec47ba

Browse files
committed
Add option for user
1 parent 65fdff7 commit aec47ba

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

src/main.rs

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -31,18 +31,21 @@ struct Args {
3131
/// Interval to check audit logs in milliseconds
3232
#[clap(short, long, default_value_t = 1500)]
3333
pub interval: u64,
34+
35+
/// The user to watch the home directory of
36+
#[clap(short, long, default_value_t = get_user())]
37+
pub user: String
3438
}
3539

3640
fn main() {
3741
let args = Args::parse();
38-
let user = get_user();
39-
let user_env = UserEnvironment::from_user(&user).expect("Failed to get user env");
40-
let std_dirs = StandardDirectories::new(&user, &user_env);
42+
let user_env = UserEnvironment::from_user(&args.user).expect("Failed to get user env");
43+
let std_dirs = StandardDirectories::new(&args.user, &user_env);
4144
println!("Watching {}", std_dirs.home());
4245
let excluded_dirs= get_excluded_directories(&std_dirs, &args.excluded_dirs);
4346
let excluded_dirs_str = excluded_dirs.join(", ");
4447
println!("Excluding {excluded_dirs_str}");
45-
let db_file = create_db_file(&std_dirs, &user);
48+
let db_file = create_db_file(&std_dirs, &args.user);
4649
let conn = open_db(&db_file);
4750

4851
let auditd_rules = file_audit::generate_audit_rules(std_dirs.home(), &excluded_dirs);

0 commit comments

Comments
 (0)