[reactos] 02/02: [UBTRFS] Upgrade to 1.5

[Pierre Schweitzer]

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=86ee9b0cc20186afe9e2f…

commit 86ee9b0cc20186afe9e2f1c27e24c0e8da206c66
Author:     Pierre Schweitzer <pierre(a)reactos.org>
AuthorDate: Tue Nov 12 19:34:14 2019 +0100
Commit:     Pierre Schweitzer <pierre(a)reactos.org>
CommitDate: Tue Nov 12 19:35:43 2019 +0100

    [UBTRFS] Upgrade to 1.5
    
    CORE-16494
---
 dll/win32/ubtrfs/ubtrfs.rc        |   8 +--
 media/doc/README.FSD              |   2 +-
 sdk/lib/fslib/btrfslib/btrfslib.c | 108 ++++++++++++++++++++++++++++++--------
 3 files changed, 91 insertions(+), 27 deletions(-)

diff --git a/dll/win32/ubtrfs/ubtrfs.rc b/dll/win32/ubtrfs/ubtrfs.rc
index 1292452a5cd..1720597df5b 100644
--- a/dll/win32/ubtrfs/ubtrfs.rc
+++ b/dll/win32/ubtrfs/ubtrfs.rc
@@ -51,8 +51,8 @@ END
 //
 
 VS_VERSION_INFO VERSIONINFO
- FILEVERSION 1,4,0,0
- PRODUCTVERSION 1,4,0,0
+ FILEVERSION 1,5,0,0
+ PRODUCTVERSION 1,5,0,0
  FILEFLAGSMASK 0x17L
 #ifdef _DEBUG
  FILEFLAGS 0x1L
@@ -68,12 +68,12 @@ BEGIN
         BLOCK "080904b0"
         BEGIN
             VALUE "FileDescription", "Btrfs utility DLL"
-            VALUE "FileVersion", "1.4"
+            VALUE "FileVersion", "1.5"
             VALUE "InternalName", "ubtrfs"
             VALUE "LegalCopyright", "Copyright (c) Mark Harmstone
2016-19"
             VALUE "OriginalFilename", "ubtrfs.dll"
             VALUE "ProductName", "WinBtrfs"
-            VALUE "ProductVersion", "1.4"
+            VALUE "ProductVersion", "1.5"
         END
     END
     BLOCK "VarFileInfo"
diff --git a/media/doc/README.FSD b/media/doc/README.FSD
index 9ce6e3a7d29..ed4c991c0c3 100644
--- a/media/doc/README.FSD
+++ b/media/doc/README.FSD
@@ -5,7 +5,7 @@ The following FSD are shared with: https://github.com/maharmstone/btrfs.
 
 reactos/drivers/filesystems/btrfs           # Synced to 1.5
 reactos/dll/shellext/shellbtrfs             # Synced to 1.1
-reactos/sdk/lib/fslib/btrfslib              # Synced to 1.4
+reactos/sdk/lib/fslib/btrfslib              # Synced to 1.5
 
 The following FSD are shared with: http://www.ext2fsd.com/
 
diff --git a/sdk/lib/fslib/btrfslib/btrfslib.c b/sdk/lib/fslib/btrfslib/btrfslib.c
index ec9fe387037..254873089df 100644
--- a/sdk/lib/fslib/btrfslib/btrfslib.c
+++ b/sdk/lib/fslib/btrfslib/btrfslib.c
@@ -803,7 +803,7 @@ static NTSTATUS write_superblocks(HANDLE h, btrfs_dev* dev,
btrfs_root* chunk_ro
     sb->chunk_tree_addr = chunk_root->header.address;
     sb->total_bytes = dev->dev_item.num_bytes;
     sb->bytes_used = bytes_used;
-    sb->root_dir_objectid = 6;
+    sb->root_dir_objectid = BTRFS_ROOT_TREEDIR;
     sb->num_devices = 1;
     sb->sector_size = sector_size;
     sb->node_size = node_size;
@@ -836,14 +836,14 @@ static NTSTATUS write_superblocks(HANDLE h, btrfs_dev* dev,
btrfs_root* chunk_ro
         }
 
 #ifndef __REACTOS__
-        utf8len = WideCharToMultiByte(CP_UTF8, 0, label->Buffer, label->Length,
NULL, 0, NULL, NULL);
+        utf8len = WideCharToMultiByte(CP_UTF8, 0, label->Buffer, label->Length /
sizeof(WCHAR), NULL, 0, NULL, NULL);
 
         if (utf8len == 0 || utf8len > MAX_LABEL_SIZE) {
             free(sb);
             return STATUS_INVALID_VOLUME_LABEL;
         }
 
-        if (WideCharToMultiByte(CP_UTF8, 0, label->Buffer, label->Length,
sb->label, utf8len, NULL, NULL) == 0) {
+        if (WideCharToMultiByte(CP_UTF8, 0, label->Buffer, label->Length /
sizeof(WCHAR), sb->label, utf8len, NULL, NULL) == 0) {
             free(sb);
             return STATUS_INVALID_VOLUME_LABEL;
         }
@@ -929,9 +929,29 @@ GetSystemTimeAsFileTime(OUT PFILETIME lpFileTime)
 }
 #endif
 
+static void add_inode_ref(btrfs_root* r, uint64_t inode, uint64_t parent, uint64_t index,
const char* name) {
+    uint16_t name_len = (uint16_t)strlen(name);
+#ifndef __REACTOS__
+    INODE_REF* ir = malloc(offsetof(INODE_REF, name[0]) + name_len);
+#else
+    INODE_REF* ir = RtlAllocateHeap(RtlGetProcessHeap(), 0, offsetof(INODE_REF, name[0])
+ name_len);
+#endif
+
+    ir->index = 0;
+    ir->n = name_len;
+    memcpy(ir->name, name, name_len);
+
+    add_item(r, inode, TYPE_INODE_REF, parent, ir, (uint16_t)offsetof(INODE_REF, name[0])
+ ir->n);
+
+#ifndef __REACTOS__
+    free(ir);
+#else
+    RtlFreeHeap(RtlGetProcessHeap(), 0, ir);
+#endif
+}
+
 static void init_fs_tree(btrfs_root* r, uint32_t node_size) {
     INODE_ITEM ii;
-    INODE_REF* ir;
     FILETIME filetime;
     LARGE_INTEGER time;
 
@@ -951,24 +971,7 @@ static void init_fs_tree(btrfs_root* r, uint32_t node_size) {
 
     add_item(r, SUBVOL_ROOT_INODE, TYPE_INODE_ITEM, 0, &ii, sizeof(INODE_ITEM));
 
-#ifndef __REACTOS__
-    ir = malloc(sizeof(INODE_REF) - 1 + 2);
-#else
-    ir = RtlAllocateHeap(RtlGetProcessHeap(), 0, sizeof(INODE_REF) - 1 + 2);
-#endif
-
-    ir->index = 0;
-    ir->n = 2;
-    ir->name[0] = '.';
-    ir->name[1] = '.';
-
-    add_item(r, SUBVOL_ROOT_INODE, TYPE_INODE_REF, SUBVOL_ROOT_INODE, ir,
sizeof(INODE_REF) - 1 + ir->n);
-
-#ifndef __REACTOS__
-    free(ir);
-#else
-    RtlFreeHeap(RtlGetProcessHeap(), 0, ir);
-#endif
+    add_inode_ref(r, SUBVOL_ROOT_INODE, SUBVOL_ROOT_INODE, 0, "..");
 }
 
 static void add_block_group_items(LIST_ENTRY* chunks, btrfs_root* extent_root) {
@@ -1061,6 +1064,65 @@ static BOOL is_ssd(HANDLE h) {
     return FALSE;
 }
 
+static void add_dir_item(btrfs_root* root, uint64_t inode, uint32_t hash, uint64_t
key_objid, uint8_t key_type,
+                         uint64_t key_offset, uint64_t transid, uint8_t type, const char*
name) {
+    uint16_t name_len = (uint16_t)strlen(name);
+#ifndef __REACTOS__
+    DIR_ITEM* di = malloc(offsetof(DIR_ITEM, name[0]) + name_len);
+#else
+    DIR_ITEM* di = RtlAllocateHeap(RtlGetProcessHeap(), 0, offsetof(DIR_ITEM, name[0]) +
name_len);
+#endif
+
+    di->key.obj_id = key_objid;
+    di->key.obj_type = key_type;
+    di->key.offset = key_offset;
+    di->transid = transid;
+    di->m = 0;
+    di->n = name_len;
+    di->type = type;
+    memcpy(di->name, name, name_len);
+
+    add_item(root, inode, TYPE_DIR_ITEM, hash, di, (uint16_t)(offsetof(DIR_ITEM, name[0])
+ di->m + di->n));
+
+#ifndef __REACTOS__
+    free(di);
+#else
+    RtlFreeHeap(RtlGetProcessHeap(), 0, di);
+#endif
+}
+
+static void set_default_subvol(btrfs_root* root_root, uint32_t node_size) {
+    INODE_ITEM ii;
+    FILETIME filetime;
+    LARGE_INTEGER time;
+
+    static const char default_subvol[] = "default";
+    static const uint32_t default_hash = 0x8dbfc2d2;
+
+    add_inode_ref(root_root, BTRFS_ROOT_FSTREE, BTRFS_ROOT_TREEDIR, 0, default_subvol);
+
+    memset(&ii, 0, sizeof(INODE_ITEM));
+
+    ii.generation = 1;
+    ii.st_blocks = node_size;
+    ii.st_nlink = 1;
+    ii.st_mode = 040755;
+
+    GetSystemTimeAsFileTime(&filetime);
+    time.LowPart = filetime.dwLowDateTime;
+    time.HighPart = filetime.dwHighDateTime;
+
+    win_time_to_unix(time, &ii.st_atime);
+    ii.st_ctime = ii.st_mtime = ii.otime = ii.st_atime;
+
+    add_item(root_root, BTRFS_ROOT_TREEDIR, TYPE_INODE_ITEM, 0, &ii,
sizeof(INODE_ITEM));
+
+    add_inode_ref(root_root, BTRFS_ROOT_TREEDIR, BTRFS_ROOT_TREEDIR, 0, "..");
+
+    add_dir_item(root_root, BTRFS_ROOT_TREEDIR, default_hash, BTRFS_ROOT_FSTREE,
TYPE_ROOT_ITEM,
+                 0xffffffffffffffff, 0, BTRFS_TYPE_DIRECTORY, default_subvol);
+}
+
 static NTSTATUS write_btrfs(HANDLE h, uint64_t size, PUNICODE_STRING label, uint32_t
sector_size, uint32_t node_size, uint64_t incompat_flags) {
     NTSTATUS Status;
     LIST_ENTRY roots, chunks;
@@ -1121,6 +1183,8 @@ static NTSTATUS write_btrfs(HANDLE h, uint64_t size, PUNICODE_STRING
label, uint
 
     add_item(chunk_root, 1, TYPE_DEV_ITEM, dev.dev_item.dev_id, &dev.dev_item,
sizeof(DEV_ITEM));
 
+    set_default_subvol(root_root, node_size);
+
     init_fs_tree(fs_root, node_size);
     init_fs_tree(reloc_root, node_size);