rust: BkeyValC

Now we have a rust-style enum for key types

Signed-off-by: Kent Overstreet <kent.overstreet@linux.dev>

1 files changed, 114 insertions, 0 deletions

diff --git a/rust-src/bch_bindgen/src/bkey.rs b/rust-src/bch_bindgen/src/bkey.rs
new file mode 100644
index 00000000..6735f298
--- /dev/null
+++ b/rust-src/bch_bindgen/src/bkey.rs
@@ -0,0 +1,114 @@
+#![allow(non_camel_case_types)]
+
+use crate::c;
+use crate::fs::Fs;
+use std::ffi::CStr;
+use std::fmt;
+use std::mem::transmute;
+
+pub struct BkeySC<'a> {
+    pub k:  &'a c::bkey,
+    pub v:  &'a c::bch_val,
+}
+
+pub enum BkeyValC<'a> {
+    deleted,
+    whiteout,
+    error,
+    cookie(&'a c::bch_cookie),
+    hash_whiteout(&'a c::bch_hash_whiteout),
+    btree_ptr(&'a c::bch_btree_ptr),
+    extent(&'a c::bch_extent),
+    reservation(&'a c::bch_reservation),
+    inode(&'a c::bch_inode),
+    inode_generation(&'a c::bch_inode_generation),
+    dirent(&'a c::bch_dirent),
+    xattr(&'a c::bch_xattr),
+    alloc(&'a c::bch_alloc),
+    quota(&'a c::bch_quota),
+    stripe(&'a c::bch_stripe),
+    reflink_p(&'a c::bch_reflink_p),
+    reflink_v(&'a c::bch_reflink_v),
+    inline_data(&'a c::bch_inline_data),
+    btree_ptr_v2(&'a c::bch_btree_ptr_v2),
+    indirect_inline_data(&'a c::bch_indirect_inline_data),
+    alloc_v2(&'a c::bch_alloc_v2),
+    subvolume(&'a c::bch_subvolume),
+    snapshot(&'a c::bch_snapshot),
+    inode_v2(&'a c::bch_inode_v2),
+    alloc_v3(&'a c::bch_alloc_v3),
+    set,
+    lru(&'a c::bch_lru),
+    alloc_v4(&'a c::bch_alloc_v4),
+    backpointer(&'a c::bch_backpointer),
+    inode_v3(&'a c::bch_inode_v3),
+    bucket_gens(&'a c::bch_bucket_gens),
+}
+
+impl<'a, 'b> BkeySC<'a> {
+    unsafe fn to_raw(&self) -> c::bkey_s_c {
+        c::bkey_s_c { k: self.k, v: self.v }
+    }
+
+    pub fn to_text(&'a self, fs: &'b Fs) -> BkeySCToText<'a, 'b> {
+        BkeySCToText { k: self, fs }
+    }
+
+    pub fn v(&'a self) -> BkeyValC {
+        let ty: c::bch_bkey_type = unsafe { transmute(self.k.type_ as u32) };
+
+        use c::bch_bkey_type::*;
+        use BkeyValC::*;
+        match ty {
+            KEY_TYPE_deleted                => deleted,
+            KEY_TYPE_whiteout               => whiteout,
+            KEY_TYPE_error                  => error,
+            KEY_TYPE_cookie                 => cookie(unsafe { transmute(self.v) }),
+            KEY_TYPE_hash_whiteout          => hash_whiteout(unsafe { transmute(self.v) }),
+            KEY_TYPE_btree_ptr              => btree_ptr(unsafe { transmute(self.v) }),
+            KEY_TYPE_extent                 => extent(unsafe { transmute(self.v) }),
+            KEY_TYPE_reservation            => reservation(unsafe { transmute(self.v) }),
+            KEY_TYPE_inode                  => inode(unsafe { transmute(self.v) }),
+            KEY_TYPE_inode_generation       => inode_generation(unsafe { transmute(self.v) }),
+            KEY_TYPE_dirent                 => dirent(unsafe { transmute(self.v) }),
+            KEY_TYPE_xattr                  => xattr(unsafe { transmute(self.v) }),
+            KEY_TYPE_alloc                  => alloc(unsafe { transmute(self.v) }),
+            KEY_TYPE_quota                  => quota(unsafe { transmute(self.v) }),
+            KEY_TYPE_stripe                 => stripe(unsafe { transmute(self.v) }),
+            KEY_TYPE_reflink_p              => reflink_p(unsafe { transmute(self.v) }),
+            KEY_TYPE_reflink_v              => reflink_v(unsafe { transmute(self.v) }),
+            KEY_TYPE_inline_data            => inline_data(unsafe { transmute(self.v) }),
+            KEY_TYPE_btree_ptr_v2           => btree_ptr_v2(unsafe { transmute(self.v) }),
+            KEY_TYPE_indirect_inline_data   => indirect_inline_data(unsafe { transmute(self.v) }),
+            KEY_TYPE_alloc_v2               => alloc_v2(unsafe { transmute(self.v) }),
+            KEY_TYPE_subvolume              => subvolume(unsafe { transmute(self.v) }),
+            KEY_TYPE_snapshot               => snapshot(unsafe { transmute(self.v) }),
+            KEY_TYPE_inode_v2               => inode_v2(unsafe { transmute(self.v) }),
+            KEY_TYPE_alloc_v3               => inode_v3(unsafe { transmute(self.v) }),
+            KEY_TYPE_set                    => set,
+            KEY_TYPE_lru                    => lru(unsafe { transmute(self.v) }),
+            KEY_TYPE_alloc_v4               => alloc_v4(unsafe { transmute(self.v) }),
+            KEY_TYPE_backpointer            => backpointer(unsafe { transmute(self.v) }),
+            KEY_TYPE_inode_v3               => inode_v3(unsafe { transmute(self.v) }),
+            KEY_TYPE_bucket_gens            => bucket_gens(unsafe { transmute(self.v) }),
+            KEY_TYPE_MAX                    => unreachable!(),
+        }
+    }
+}
+
+pub struct BkeySCToText<'a, 'b> {
+    k:  &'a BkeySC<'a>,
+    fs: &'b Fs,
+}
+
+impl<'a, 'b> fmt::Display for BkeySCToText<'a, 'b> {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        let mut buf = c::printbuf::new();
+
+        unsafe { c::bch2_bkey_val_to_text(&mut buf, self.fs.raw, self.k.to_raw()) };
+ 
+        let s = unsafe { CStr::from_ptr(buf.buf) };
+        let s = s.to_str().unwrap();
+        write!(f, "{}", s)
+    }
+}