git.sesse.net Git - bcachefs-tools-debian/blob - src/bcachefs.rs

   1 mod commands;
   2 mod key;
   3
   4 use std::ffi::CString;
   5
   6 use commands::cmd_completions::cmd_completions;
   7 use commands::cmd_list::cmd_list;
   8 use commands::cmd_mount::cmd_mount;
   9 use commands::logger::SimpleLogger;
  10 use bch_bindgen::c;
  11
  12 #[derive(Debug)]
  13 pub struct ErrnoError(pub errno::Errno);
  14 impl std::fmt::Display for ErrnoError {
  15     fn fmt(&self, f: &mut std::fmt::Formatter) -> Result<(), std::fmt::Error> {
  16         self.0.fmt(f)
  17     }
  18 }
  19
  20 impl std::error::Error for ErrnoError {}
  21
  22 fn handle_c_command(args: Vec<String>, symlink_cmd: Option<&str>) -> i32 {
  23     let mut argv: Vec<_> = args.clone();
  24
  25     let cmd = match symlink_cmd {
  26         Some(s) => s.to_string(),
  27         None => argv.remove(1),
  28     };
  29
  30     let argc: i32 = argv.len().try_into().unwrap();
  31
  32     let argv: Vec<_> = argv
  33         .iter()
  34         .map(|s| CString::new(s.as_str()).unwrap())
  35         .collect();
  36     let argv: Vec<_> = argv.iter().map(|s| s.as_ptr()).collect();
  37     let argv = argv.as_ptr() as *mut *mut i8;
  38
  39     // The C functions will mutate argv. It shouldn't be used after this block.
  40     unsafe {
  41         match cmd.as_str() {
  42             "--help" => {
  43                 c::bcachefs_usage();
  44                 0
  45             },
  46             "data" => c::data_cmds(argc, argv),
  47             "device" => c::device_cmds(argc, argv),
  48             "dump" => c::cmd_dump(argc, argv),
  49             "format" => c::cmd_format(argc, argv),
  50             "fs" => c::fs_cmds(argc, argv),
  51             "fsck" => c::cmd_fsck(argc, argv),
  52             "list_journal" => c::cmd_list_journal(argc, argv),
  53             "kill_btree_node" => c::cmd_kill_btree_node(argc, argv),
  54             "migrate" => c::cmd_migrate(argc, argv),
  55             "migrate-superblock" => c::cmd_migrate_superblock(argc, argv),
  56             "mkfs" => c::cmd_format(argc, argv),
  57             "remove-passphrase" => c::cmd_remove_passphrase(argc, argv),
  58             "reset-counters" => c::cmd_reset_counters(argc, argv),
  59             "set-option" => c::cmd_set_option(argc, argv),
  60             "set-passphrase" => c::cmd_set_passphrase(argc, argv),
  61             "setattr" => c::cmd_setattr(argc, argv),
  62             "show-super" => c::cmd_show_super(argc, argv),
  63             "subvolume" => c::subvolume_cmds(argc, argv),
  64             "unlock" => c::cmd_unlock(argc, argv),
  65             "version" => c::cmd_version(argc, argv),
  66
  67             #[cfg(fuse)]
  68             "fusemount" => c::cmd_fusemount(argc, argv),
  69
  70             _ => {
  71                 println!("Unknown command {}", cmd);
  72                 c::bcachefs_usage();
  73                 1
  74             }
  75         }
  76     }
  77 }
  78
  79 fn main() {
  80     let args: Vec<String> = std::env::args().collect();
  81
  82     let symlink_cmd: Option<&str> = if args[0].contains("mkfs") {
  83         Some("mkfs")
  84     } else if args[0].contains("fsck") {
  85         Some("fsck")
  86     } else if args[0].contains("mount.fuse") {
  87         Some("fusemount")
  88     } else if args[0].contains("mount") {
  89         Some("mount")
  90     } else {
  91         None
  92     };
  93
  94     if symlink_cmd.is_none() && args.len() < 2 {
  95         println!("missing command");
  96         unsafe { c::bcachefs_usage() };
  97         std::process::exit(1);
  98     }
  99
 100     unsafe { c::raid_init() };
 101
 102     log::set_boxed_logger(Box::new(SimpleLogger)).unwrap();
 103     log::set_max_level(log::LevelFilter::Warn);
 104
 105     let cmd = match symlink_cmd {
 106         Some(s) => s,
 107         None => args[1].as_str(),
 108     };
 109
 110     let ret = match cmd {
 111         "completions" => cmd_completions(args[1..].to_vec()),
 112         "list" => cmd_list(args[1..].to_vec()),
 113         "mount" => cmd_mount(args, symlink_cmd),
 114         _ => handle_c_command(args, symlink_cmd),
 115     };
 116
 117     if ret != 0 {
 118         std::process::exit(1);
 119     }
 120 }