Blame SOURCES/0005-fs-ntfs3-Add-attrib-operations.patch

Kmods SIG d83023
From be71b5cba2e6485e8959da7a9f9a44461a1bb074 Mon Sep 17 00:00:00 2001
Kmods SIG d83023
From: Konstantin Komarov <almaz.alexandrovich@paragon-software.com>
Kmods SIG d83023
Date: Fri, 13 Aug 2021 17:21:30 +0300
Kmods SIG d83023
Subject: [Backport be71b5cba2e6] src: Add attrib operations
Kmods SIG d83023
Kmods SIG d83023
This adds attrib operations
Kmods SIG d83023
Kmods SIG d83023
Signed-off-by: Konstantin Komarov <almaz.alexandrovich@paragon-software.com>
Kmods SIG d83023
---
Kmods SIG d83023
 src/attrib.c   | 2096 +++++++++++++++++++++++++++++++++++++++++++
Kmods SIG d83023
 src/attrlist.c |  456 ++++++++++
Kmods SIG d83023
 src/xattr.c    | 1128 +++++++++++++++++++++++
Kmods SIG d83023
 3 files changed, 3680 insertions(+)
Kmods SIG d83023
 create mode 100644 src/attrib.c
Kmods SIG d83023
 create mode 100644 src/attrlist.c
Kmods SIG d83023
 create mode 100644 src/xattr.c
Kmods SIG d83023
Kmods SIG d83023
diff --git a/src/attrib.c b/src/attrib.c
Kmods SIG d83023
new file mode 100644
Kmods SIG d83023
index 0000000000000000000000000000000000000000..046dc57f75f2169b47e7078c251f26b05ec69b42
Kmods SIG d83023
--- /dev/null
Kmods SIG d83023
+++ b/src/attrib.c
Kmods SIG d83023
@@ -0,0 +1,2096 @@
Kmods SIG d83023
+// SPDX-License-Identifier: GPL-2.0
Kmods SIG d83023
+/*
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * Copyright (C) 2019-2021 Paragon Software GmbH, All rights reserved.
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * TODO: merge attr_set_size/attr_data_get_block/attr_allocate_frame?
Kmods SIG d83023
+ */
Kmods SIG d83023
+
Kmods SIG d83023
+#include <linux/blkdev.h>
Kmods SIG d83023
+#include <linux/buffer_head.h>
Kmods SIG d83023
+#include <linux/fs.h>
Kmods SIG d83023
+#include <linux/hash.h>
Kmods SIG d83023
+#include <linux/nls.h>
Kmods SIG d83023
+#include <linux/ratelimit.h>
Kmods SIG d83023
+#include <linux/slab.h>
Kmods SIG d83023
+
Kmods SIG d83023
+#include "debug.h"
Kmods SIG d83023
+#include "ntfs.h"
Kmods SIG d83023
+#include "ntfs_fs.h"
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * You can set external NTFS_MIN_LOG2_OF_CLUMP/NTFS_MAX_LOG2_OF_CLUMP to manage
Kmods SIG d83023
+ * preallocate algorithm
Kmods SIG d83023
+ */
Kmods SIG d83023
+#ifndef NTFS_MIN_LOG2_OF_CLUMP
Kmods SIG d83023
+#define NTFS_MIN_LOG2_OF_CLUMP 16
Kmods SIG d83023
+#endif
Kmods SIG d83023
+
Kmods SIG d83023
+#ifndef NTFS_MAX_LOG2_OF_CLUMP
Kmods SIG d83023
+#define NTFS_MAX_LOG2_OF_CLUMP 26
Kmods SIG d83023
+#endif
Kmods SIG d83023
+
Kmods SIG d83023
+// 16M
Kmods SIG d83023
+#define NTFS_CLUMP_MIN (1 << (NTFS_MIN_LOG2_OF_CLUMP + 8))
Kmods SIG d83023
+// 16G
Kmods SIG d83023
+#define NTFS_CLUMP_MAX (1ull << (NTFS_MAX_LOG2_OF_CLUMP + 8))
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * get_pre_allocated
Kmods SIG d83023
+ *
Kmods SIG d83023
+ */
Kmods SIG d83023
+static inline u64 get_pre_allocated(u64 size)
Kmods SIG d83023
+{
Kmods SIG d83023
+	u32 clump;
Kmods SIG d83023
+	u8 align_shift;
Kmods SIG d83023
+	u64 ret;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (size <= NTFS_CLUMP_MIN) {
Kmods SIG d83023
+		clump = 1 << NTFS_MIN_LOG2_OF_CLUMP;
Kmods SIG d83023
+		align_shift = NTFS_MIN_LOG2_OF_CLUMP;
Kmods SIG d83023
+	} else if (size >= NTFS_CLUMP_MAX) {
Kmods SIG d83023
+		clump = 1 << NTFS_MAX_LOG2_OF_CLUMP;
Kmods SIG d83023
+		align_shift = NTFS_MAX_LOG2_OF_CLUMP;
Kmods SIG d83023
+	} else {
Kmods SIG d83023
+		align_shift = NTFS_MIN_LOG2_OF_CLUMP - 1 +
Kmods SIG d83023
+			      __ffs(size >> (8 + NTFS_MIN_LOG2_OF_CLUMP));
Kmods SIG d83023
+		clump = 1u << align_shift;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	ret = (((size + clump - 1) >> align_shift)) << align_shift;
Kmods SIG d83023
+
Kmods SIG d83023
+	return ret;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * attr_must_be_resident
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * returns true if attribute must be resident
Kmods SIG d83023
+ */
Kmods SIG d83023
+static inline bool attr_must_be_resident(struct ntfs_sb_info *sbi,
Kmods SIG d83023
+					 enum ATTR_TYPE type)
Kmods SIG d83023
+{
Kmods SIG d83023
+	const struct ATTR_DEF_ENTRY *de;
Kmods SIG d83023
+
Kmods SIG d83023
+	switch (type) {
Kmods SIG d83023
+	case ATTR_STD:
Kmods SIG d83023
+	case ATTR_NAME:
Kmods SIG d83023
+	case ATTR_ID:
Kmods SIG d83023
+	case ATTR_LABEL:
Kmods SIG d83023
+	case ATTR_VOL_INFO:
Kmods SIG d83023
+	case ATTR_ROOT:
Kmods SIG d83023
+	case ATTR_EA_INFO:
Kmods SIG d83023
+		return true;
Kmods SIG d83023
+	default:
Kmods SIG d83023
+		de = ntfs_query_def(sbi, type);
Kmods SIG d83023
+		if (de && (de->flags & NTFS_ATTR_MUST_BE_RESIDENT))
Kmods SIG d83023
+			return true;
Kmods SIG d83023
+		return false;
Kmods SIG d83023
+	}
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * attr_load_runs
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * load all runs stored in 'attr'
Kmods SIG d83023
+ */
Kmods SIG d83023
+int attr_load_runs(struct ATTRIB *attr, struct ntfs_inode *ni,
Kmods SIG d83023
+		   struct runs_tree *run, const CLST *vcn)
Kmods SIG d83023
+{
Kmods SIG d83023
+	int err;
Kmods SIG d83023
+	CLST svcn = le64_to_cpu(attr->nres.svcn);
Kmods SIG d83023
+	CLST evcn = le64_to_cpu(attr->nres.evcn);
Kmods SIG d83023
+	u32 asize;
Kmods SIG d83023
+	u16 run_off;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (svcn >= evcn + 1 || run_is_mapped_full(run, svcn, evcn))
Kmods SIG d83023
+		return 0;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (vcn && (evcn < *vcn || *vcn < svcn))
Kmods SIG d83023
+		return -EINVAL;
Kmods SIG d83023
+
Kmods SIG d83023
+	asize = le32_to_cpu(attr->size);
Kmods SIG d83023
+	run_off = le16_to_cpu(attr->nres.run_off);
Kmods SIG d83023
+	err = run_unpack_ex(run, ni->mi.sbi, ni->mi.rno, svcn, evcn,
Kmods SIG d83023
+			    vcn ? *vcn : svcn, Add2Ptr(attr, run_off),
Kmods SIG d83023
+			    asize - run_off);
Kmods SIG d83023
+	if (err < 0)
Kmods SIG d83023
+		return err;
Kmods SIG d83023
+
Kmods SIG d83023
+	return 0;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * int run_deallocate_ex
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * Deallocate clusters
Kmods SIG d83023
+ */
Kmods SIG d83023
+static int run_deallocate_ex(struct ntfs_sb_info *sbi, struct runs_tree *run,
Kmods SIG d83023
+			     CLST vcn, CLST len, CLST *done, bool trim)
Kmods SIG d83023
+{
Kmods SIG d83023
+	int err = 0;
Kmods SIG d83023
+	CLST vcn_next, vcn0 = vcn, lcn, clen, dn = 0;
Kmods SIG d83023
+	size_t idx;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!len)
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!run_lookup_entry(run, vcn, &lcn, &clen, &idx)) {
Kmods SIG d83023
+failed:
Kmods SIG d83023
+		run_truncate(run, vcn0);
Kmods SIG d83023
+		err = -EINVAL;
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	for (;;) {
Kmods SIG d83023
+		if (clen > len)
Kmods SIG d83023
+			clen = len;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (!clen) {
Kmods SIG d83023
+			err = -EINVAL;
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		if (lcn != SPARSE_LCN) {
Kmods SIG d83023
+			mark_as_free_ex(sbi, lcn, clen, trim);
Kmods SIG d83023
+			dn += clen;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		len -= clen;
Kmods SIG d83023
+		if (!len)
Kmods SIG d83023
+			break;
Kmods SIG d83023
+
Kmods SIG d83023
+		vcn_next = vcn + clen;
Kmods SIG d83023
+		if (!run_get_entry(run, ++idx, &vcn, &lcn, &clen) ||
Kmods SIG d83023
+		    vcn != vcn_next) {
Kmods SIG d83023
+			// save memory - don't load entire run
Kmods SIG d83023
+			goto failed;
Kmods SIG d83023
+		}
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+out:
Kmods SIG d83023
+	if (done)
Kmods SIG d83023
+		*done += dn;
Kmods SIG d83023
+
Kmods SIG d83023
+	return err;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * attr_allocate_clusters
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * find free space, mark it as used and store in 'run'
Kmods SIG d83023
+ */
Kmods SIG d83023
+int attr_allocate_clusters(struct ntfs_sb_info *sbi, struct runs_tree *run,
Kmods SIG d83023
+			   CLST vcn, CLST lcn, CLST len, CLST *pre_alloc,
Kmods SIG d83023
+			   enum ALLOCATE_OPT opt, CLST *alen, const size_t fr,
Kmods SIG d83023
+			   CLST *new_lcn)
Kmods SIG d83023
+{
Kmods SIG d83023
+	int err;
Kmods SIG d83023
+	CLST flen, vcn0 = vcn, pre = pre_alloc ? *pre_alloc : 0;
Kmods SIG d83023
+	struct wnd_bitmap *wnd = &sbi->used.bitmap;
Kmods SIG d83023
+	size_t cnt = run->count;
Kmods SIG d83023
+
Kmods SIG d83023
+	for (;;) {
Kmods SIG d83023
+		err = ntfs_look_for_free_space(sbi, lcn, len + pre, &lcn, &flen,
Kmods SIG d83023
+					       opt);
Kmods SIG d83023
+
Kmods SIG d83023
+		if (err == -ENOSPC && pre) {
Kmods SIG d83023
+			pre = 0;
Kmods SIG d83023
+			if (*pre_alloc)
Kmods SIG d83023
+				*pre_alloc = 0;
Kmods SIG d83023
+			continue;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		if (err)
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (new_lcn && vcn == vcn0)
Kmods SIG d83023
+			*new_lcn = lcn;
Kmods SIG d83023
+
Kmods SIG d83023
+		/* Add new fragment into run storage */
Kmods SIG d83023
+		if (!run_add_entry(run, vcn, lcn, flen, opt == ALLOCATE_MFT)) {
Kmods SIG d83023
+			down_write_nested(&wnd->rw_lock, BITMAP_MUTEX_CLUSTERS);
Kmods SIG d83023
+			wnd_set_free(wnd, lcn, flen);
Kmods SIG d83023
+			up_write(&wnd->rw_lock);
Kmods SIG d83023
+			err = -ENOMEM;
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		vcn += flen;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (flen >= len || opt == ALLOCATE_MFT ||
Kmods SIG d83023
+		    (fr && run->count - cnt >= fr)) {
Kmods SIG d83023
+			*alen = vcn - vcn0;
Kmods SIG d83023
+			return 0;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		len -= flen;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+out:
Kmods SIG d83023
+	/* undo */
Kmods SIG d83023
+	run_deallocate_ex(sbi, run, vcn0, vcn - vcn0, NULL, false);
Kmods SIG d83023
+	run_truncate(run, vcn0);
Kmods SIG d83023
+
Kmods SIG d83023
+	return err;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * if page is not NULL - it is already contains resident data
Kmods SIG d83023
+ * and locked (called from ni_write_frame)
Kmods SIG d83023
+ */
Kmods SIG d83023
+int attr_make_nonresident(struct ntfs_inode *ni, struct ATTRIB *attr,
Kmods SIG d83023
+			  struct ATTR_LIST_ENTRY *le, struct mft_inode *mi,
Kmods SIG d83023
+			  u64 new_size, struct runs_tree *run,
Kmods SIG d83023
+			  struct ATTRIB **ins_attr, struct page *page)
Kmods SIG d83023
+{
Kmods SIG d83023
+	struct ntfs_sb_info *sbi;
Kmods SIG d83023
+	struct ATTRIB *attr_s;
Kmods SIG d83023
+	struct MFT_REC *rec;
Kmods SIG d83023
+	u32 used, asize, rsize, aoff, align;
Kmods SIG d83023
+	bool is_data;
Kmods SIG d83023
+	CLST len, alen;
Kmods SIG d83023
+	char *next;
Kmods SIG d83023
+	int err;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (attr->non_res) {
Kmods SIG d83023
+		*ins_attr = attr;
Kmods SIG d83023
+		return 0;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	sbi = mi->sbi;
Kmods SIG d83023
+	rec = mi->mrec;
Kmods SIG d83023
+	attr_s = NULL;
Kmods SIG d83023
+	used = le32_to_cpu(rec->used);
Kmods SIG d83023
+	asize = le32_to_cpu(attr->size);
Kmods SIG d83023
+	next = Add2Ptr(attr, asize);
Kmods SIG d83023
+	aoff = PtrOffset(rec, attr);
Kmods SIG d83023
+	rsize = le32_to_cpu(attr->res.data_size);
Kmods SIG d83023
+	is_data = attr->type == ATTR_DATA && !attr->name_len;
Kmods SIG d83023
+
Kmods SIG d83023
+	align = sbi->cluster_size;
Kmods SIG d83023
+	if (is_attr_compressed(attr))
Kmods SIG d83023
+		align <<= COMPRESSION_UNIT;
Kmods SIG d83023
+	len = (rsize + align - 1) >> sbi->cluster_bits;
Kmods SIG d83023
+
Kmods SIG d83023
+	run_init(run);
Kmods SIG d83023
+
Kmods SIG d83023
+	/* make a copy of original attribute */
Kmods SIG d83023
+	attr_s = ntfs_memdup(attr, asize);
Kmods SIG d83023
+	if (!attr_s) {
Kmods SIG d83023
+		err = -ENOMEM;
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!len) {
Kmods SIG d83023
+		/* empty resident -> empty nonresident */
Kmods SIG d83023
+		alen = 0;
Kmods SIG d83023
+	} else {
Kmods SIG d83023
+		const char *data = resident_data(attr);
Kmods SIG d83023
+
Kmods SIG d83023
+		err = attr_allocate_clusters(sbi, run, 0, 0, len, NULL,
Kmods SIG d83023
+					     ALLOCATE_DEF, &alen, 0, NULL);
Kmods SIG d83023
+		if (err)
Kmods SIG d83023
+			goto out1;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (!rsize) {
Kmods SIG d83023
+			/* empty resident -> non empty nonresident */
Kmods SIG d83023
+		} else if (!is_data) {
Kmods SIG d83023
+			err = ntfs_sb_write_run(sbi, run, 0, data, rsize);
Kmods SIG d83023
+			if (err)
Kmods SIG d83023
+				goto out2;
Kmods SIG d83023
+		} else if (!page) {
Kmods SIG d83023
+			char *kaddr;
Kmods SIG d83023
+
Kmods SIG d83023
+			page = grab_cache_page(ni->vfs_inode.i_mapping, 0);
Kmods SIG d83023
+			if (!page) {
Kmods SIG d83023
+				err = -ENOMEM;
Kmods SIG d83023
+				goto out2;
Kmods SIG d83023
+			}
Kmods SIG d83023
+			kaddr = kmap_atomic(page);
Kmods SIG d83023
+			memcpy(kaddr, data, rsize);
Kmods SIG d83023
+			memset(kaddr + rsize, 0, PAGE_SIZE - rsize);
Kmods SIG d83023
+			kunmap_atomic(kaddr);
Kmods SIG d83023
+			flush_dcache_page(page);
Kmods SIG d83023
+			SetPageUptodate(page);
Kmods SIG d83023
+			set_page_dirty(page);
Kmods SIG d83023
+			unlock_page(page);
Kmods SIG d83023
+			put_page(page);
Kmods SIG d83023
+		}
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	/* remove original attribute */
Kmods SIG d83023
+	used -= asize;
Kmods SIG d83023
+	memmove(attr, Add2Ptr(attr, asize), used - aoff);
Kmods SIG d83023
+	rec->used = cpu_to_le32(used);
Kmods SIG d83023
+	mi->dirty = true;
Kmods SIG d83023
+	if (le)
Kmods SIG d83023
+		al_remove_le(ni, le);
Kmods SIG d83023
+
Kmods SIG d83023
+	err = ni_insert_nonresident(ni, attr_s->type, attr_name(attr_s),
Kmods SIG d83023
+				    attr_s->name_len, run, 0, alen,
Kmods SIG d83023
+				    attr_s->flags, &attr, NULL);
Kmods SIG d83023
+	if (err)
Kmods SIG d83023
+		goto out3;
Kmods SIG d83023
+
Kmods SIG d83023
+	ntfs_free(attr_s);
Kmods SIG d83023
+	attr->nres.data_size = cpu_to_le64(rsize);
Kmods SIG d83023
+	attr->nres.valid_size = attr->nres.data_size;
Kmods SIG d83023
+
Kmods SIG d83023
+	*ins_attr = attr;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (is_data)
Kmods SIG d83023
+		ni->ni_flags &= ~NI_FLAG_RESIDENT;
Kmods SIG d83023
+
Kmods SIG d83023
+	/* Resident attribute becomes non resident */
Kmods SIG d83023
+	return 0;
Kmods SIG d83023
+
Kmods SIG d83023
+out3:
Kmods SIG d83023
+	attr = Add2Ptr(rec, aoff);
Kmods SIG d83023
+	memmove(next, attr, used - aoff);
Kmods SIG d83023
+	memcpy(attr, attr_s, asize);
Kmods SIG d83023
+	rec->used = cpu_to_le32(used + asize);
Kmods SIG d83023
+	mi->dirty = true;
Kmods SIG d83023
+out2:
Kmods SIG d83023
+	/* undo: do not trim new allocated clusters */
Kmods SIG d83023
+	run_deallocate(sbi, run, false);
Kmods SIG d83023
+	run_close(run);
Kmods SIG d83023
+out1:
Kmods SIG d83023
+	ntfs_free(attr_s);
Kmods SIG d83023
+	/*reinsert le*/
Kmods SIG d83023
+out:
Kmods SIG d83023
+	return err;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * attr_set_size_res
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * helper for attr_set_size
Kmods SIG d83023
+ */
Kmods SIG d83023
+static int attr_set_size_res(struct ntfs_inode *ni, struct ATTRIB *attr,
Kmods SIG d83023
+			     struct ATTR_LIST_ENTRY *le, struct mft_inode *mi,
Kmods SIG d83023
+			     u64 new_size, struct runs_tree *run,
Kmods SIG d83023
+			     struct ATTRIB **ins_attr)
Kmods SIG d83023
+{
Kmods SIG d83023
+	struct ntfs_sb_info *sbi = mi->sbi;
Kmods SIG d83023
+	struct MFT_REC *rec = mi->mrec;
Kmods SIG d83023
+	u32 used = le32_to_cpu(rec->used);
Kmods SIG d83023
+	u32 asize = le32_to_cpu(attr->size);
Kmods SIG d83023
+	u32 aoff = PtrOffset(rec, attr);
Kmods SIG d83023
+	u32 rsize = le32_to_cpu(attr->res.data_size);
Kmods SIG d83023
+	u32 tail = used - aoff - asize;
Kmods SIG d83023
+	char *next = Add2Ptr(attr, asize);
Kmods SIG d83023
+	s64 dsize = QuadAlign(new_size) - QuadAlign(rsize);
Kmods SIG d83023
+
Kmods SIG d83023
+	if (dsize < 0) {
Kmods SIG d83023
+		memmove(next + dsize, next, tail);
Kmods SIG d83023
+	} else if (dsize > 0) {
Kmods SIG d83023
+		if (used + dsize > sbi->max_bytes_per_attr)
Kmods SIG d83023
+			return attr_make_nonresident(ni, attr, le, mi, new_size,
Kmods SIG d83023
+						     run, ins_attr, NULL);
Kmods SIG d83023
+
Kmods SIG d83023
+		memmove(next + dsize, next, tail);
Kmods SIG d83023
+		memset(next, 0, dsize);
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	if (new_size > rsize)
Kmods SIG d83023
+		memset(Add2Ptr(resident_data(attr), rsize), 0,
Kmods SIG d83023
+		       new_size - rsize);
Kmods SIG d83023
+
Kmods SIG d83023
+	rec->used = cpu_to_le32(used + dsize);
Kmods SIG d83023
+	attr->size = cpu_to_le32(asize + dsize);
Kmods SIG d83023
+	attr->res.data_size = cpu_to_le32(new_size);
Kmods SIG d83023
+	mi->dirty = true;
Kmods SIG d83023
+	*ins_attr = attr;
Kmods SIG d83023
+
Kmods SIG d83023
+	return 0;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * attr_set_size
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * change the size of attribute
Kmods SIG d83023
+ * Extend:
Kmods SIG d83023
+ *   - sparse/compressed: no allocated clusters
Kmods SIG d83023
+ *   - normal: append allocated and preallocated new clusters
Kmods SIG d83023
+ * Shrink:
Kmods SIG d83023
+ *   - no deallocate if keep_prealloc is set
Kmods SIG d83023
+ */
Kmods SIG d83023
+int attr_set_size(struct ntfs_inode *ni, enum ATTR_TYPE type,
Kmods SIG d83023
+		  const __le16 *name, u8 name_len, struct runs_tree *run,
Kmods SIG d83023
+		  u64 new_size, const u64 *new_valid, bool keep_prealloc,
Kmods SIG d83023
+		  struct ATTRIB **ret)
Kmods SIG d83023
+{
Kmods SIG d83023
+	int err = 0;
Kmods SIG d83023
+	struct ntfs_sb_info *sbi = ni->mi.sbi;
Kmods SIG d83023
+	u8 cluster_bits = sbi->cluster_bits;
Kmods SIG d83023
+	bool is_mft =
Kmods SIG d83023
+		ni->mi.rno == MFT_REC_MFT && type == ATTR_DATA && !name_len;
Kmods SIG d83023
+	u64 old_valid, old_size, old_alloc, new_alloc, new_alloc_tmp;
Kmods SIG d83023
+	struct ATTRIB *attr = NULL, *attr_b;
Kmods SIG d83023
+	struct ATTR_LIST_ENTRY *le, *le_b;
Kmods SIG d83023
+	struct mft_inode *mi, *mi_b;
Kmods SIG d83023
+	CLST alen, vcn, lcn, new_alen, old_alen, svcn, evcn;
Kmods SIG d83023
+	CLST next_svcn, pre_alloc = -1, done = 0;
Kmods SIG d83023
+	bool is_ext;
Kmods SIG d83023
+	u32 align;
Kmods SIG d83023
+	struct MFT_REC *rec;
Kmods SIG d83023
+
Kmods SIG d83023
+again:
Kmods SIG d83023
+	le_b = NULL;
Kmods SIG d83023
+	attr_b = ni_find_attr(ni, NULL, &le_b, type, name, name_len, NULL,
Kmods SIG d83023
+			      &mi_b);
Kmods SIG d83023
+	if (!attr_b) {
Kmods SIG d83023
+		err = -ENOENT;
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!attr_b->non_res) {
Kmods SIG d83023
+		err = attr_set_size_res(ni, attr_b, le_b, mi_b, new_size, run,
Kmods SIG d83023
+					&attr_b);
Kmods SIG d83023
+		if (err || !attr_b->non_res)
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+		/* layout of records may be changed, so do a full search */
Kmods SIG d83023
+		goto again;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	is_ext = is_attr_ext(attr_b);
Kmods SIG d83023
+
Kmods SIG d83023
+again_1:
Kmods SIG d83023
+	align = sbi->cluster_size;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (is_ext) {
Kmods SIG d83023
+		align <<= attr_b->nres.c_unit;
Kmods SIG d83023
+		if (is_attr_sparsed(attr_b))
Kmods SIG d83023
+			keep_prealloc = false;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	old_valid = le64_to_cpu(attr_b->nres.valid_size);
Kmods SIG d83023
+	old_size = le64_to_cpu(attr_b->nres.data_size);
Kmods SIG d83023
+	old_alloc = le64_to_cpu(attr_b->nres.alloc_size);
Kmods SIG d83023
+	old_alen = old_alloc >> cluster_bits;
Kmods SIG d83023
+
Kmods SIG d83023
+	new_alloc = (new_size + align - 1) & ~(u64)(align - 1);
Kmods SIG d83023
+	new_alen = new_alloc >> cluster_bits;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (keep_prealloc && is_ext)
Kmods SIG d83023
+		keep_prealloc = false;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (keep_prealloc && new_size < old_size) {
Kmods SIG d83023
+		attr_b->nres.data_size = cpu_to_le64(new_size);
Kmods SIG d83023
+		mi_b->dirty = true;
Kmods SIG d83023
+		goto ok;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	vcn = old_alen - 1;
Kmods SIG d83023
+
Kmods SIG d83023
+	svcn = le64_to_cpu(attr_b->nres.svcn);
Kmods SIG d83023
+	evcn = le64_to_cpu(attr_b->nres.evcn);
Kmods SIG d83023
+
Kmods SIG d83023
+	if (svcn <= vcn && vcn <= evcn) {
Kmods SIG d83023
+		attr = attr_b;
Kmods SIG d83023
+		le = le_b;
Kmods SIG d83023
+		mi = mi_b;
Kmods SIG d83023
+	} else if (!le_b) {
Kmods SIG d83023
+		err = -EINVAL;
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	} else {
Kmods SIG d83023
+		le = le_b;
Kmods SIG d83023
+		attr = ni_find_attr(ni, attr_b, &le, type, name, name_len, &vcn,
Kmods SIG d83023
+				    &mi);
Kmods SIG d83023
+		if (!attr) {
Kmods SIG d83023
+			err = -EINVAL;
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+next_le_1:
Kmods SIG d83023
+		svcn = le64_to_cpu(attr->nres.svcn);
Kmods SIG d83023
+		evcn = le64_to_cpu(attr->nres.evcn);
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+next_le:
Kmods SIG d83023
+	rec = mi->mrec;
Kmods SIG d83023
+
Kmods SIG d83023
+	err = attr_load_runs(attr, ni, run, NULL);
Kmods SIG d83023
+	if (err)
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (new_size > old_size) {
Kmods SIG d83023
+		CLST to_allocate;
Kmods SIG d83023
+		size_t free;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (new_alloc <= old_alloc) {
Kmods SIG d83023
+			attr_b->nres.data_size = cpu_to_le64(new_size);
Kmods SIG d83023
+			mi_b->dirty = true;
Kmods SIG d83023
+			goto ok;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		to_allocate = new_alen - old_alen;
Kmods SIG d83023
+add_alloc_in_same_attr_seg:
Kmods SIG d83023
+		lcn = 0;
Kmods SIG d83023
+		if (is_mft) {
Kmods SIG d83023
+			/* mft allocates clusters from mftzone */
Kmods SIG d83023
+			pre_alloc = 0;
Kmods SIG d83023
+		} else if (is_ext) {
Kmods SIG d83023
+			/* no preallocate for sparse/compress */
Kmods SIG d83023
+			pre_alloc = 0;
Kmods SIG d83023
+		} else if (pre_alloc == -1) {
Kmods SIG d83023
+			pre_alloc = 0;
Kmods SIG d83023
+			if (type == ATTR_DATA && !name_len &&
Kmods SIG d83023
+			    sbi->options.prealloc) {
Kmods SIG d83023
+				CLST new_alen2 = bytes_to_cluster(
Kmods SIG d83023
+					sbi, get_pre_allocated(new_size));
Kmods SIG d83023
+				pre_alloc = new_alen2 - new_alen;
Kmods SIG d83023
+			}
Kmods SIG d83023
+
Kmods SIG d83023
+			/* Get the last lcn to allocate from */
Kmods SIG d83023
+			if (old_alen &&
Kmods SIG d83023
+			    !run_lookup_entry(run, vcn, &lcn, NULL, NULL)) {
Kmods SIG d83023
+				lcn = SPARSE_LCN;
Kmods SIG d83023
+			}
Kmods SIG d83023
+
Kmods SIG d83023
+			if (lcn == SPARSE_LCN)
Kmods SIG d83023
+				lcn = 0;
Kmods SIG d83023
+			else if (lcn)
Kmods SIG d83023
+				lcn += 1;
Kmods SIG d83023
+
Kmods SIG d83023
+			free = wnd_zeroes(&sbi->used.bitmap);
Kmods SIG d83023
+			if (to_allocate > free) {
Kmods SIG d83023
+				err = -ENOSPC;
Kmods SIG d83023
+				goto out;
Kmods SIG d83023
+			}
Kmods SIG d83023
+
Kmods SIG d83023
+			if (pre_alloc && to_allocate + pre_alloc > free)
Kmods SIG d83023
+				pre_alloc = 0;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		vcn = old_alen;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (is_ext) {
Kmods SIG d83023
+			if (!run_add_entry(run, vcn, SPARSE_LCN, to_allocate,
Kmods SIG d83023
+					   false)) {
Kmods SIG d83023
+				err = -ENOMEM;
Kmods SIG d83023
+				goto out;
Kmods SIG d83023
+			}
Kmods SIG d83023
+			alen = to_allocate;
Kmods SIG d83023
+		} else {
Kmods SIG d83023
+			/* ~3 bytes per fragment */
Kmods SIG d83023
+			err = attr_allocate_clusters(
Kmods SIG d83023
+				sbi, run, vcn, lcn, to_allocate, &pre_alloc,
Kmods SIG d83023
+				is_mft ? ALLOCATE_MFT : 0, &alen,
Kmods SIG d83023
+				is_mft ? 0
Kmods SIG d83023
+				       : (sbi->record_size -
Kmods SIG d83023
+					  le32_to_cpu(rec->used) + 8) /
Kmods SIG d83023
+							 3 +
Kmods SIG d83023
+						 1,
Kmods SIG d83023
+				NULL);
Kmods SIG d83023
+			if (err)
Kmods SIG d83023
+				goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		done += alen;
Kmods SIG d83023
+		vcn += alen;
Kmods SIG d83023
+		if (to_allocate > alen)
Kmods SIG d83023
+			to_allocate -= alen;
Kmods SIG d83023
+		else
Kmods SIG d83023
+			to_allocate = 0;
Kmods SIG d83023
+
Kmods SIG d83023
+pack_runs:
Kmods SIG d83023
+		err = mi_pack_runs(mi, attr, run, vcn - svcn);
Kmods SIG d83023
+		if (err)
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+		next_svcn = le64_to_cpu(attr->nres.evcn) + 1;
Kmods SIG d83023
+		new_alloc_tmp = (u64)next_svcn << cluster_bits;
Kmods SIG d83023
+		attr_b->nres.alloc_size = cpu_to_le64(new_alloc_tmp);
Kmods SIG d83023
+		mi_b->dirty = true;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (next_svcn >= vcn && !to_allocate) {
Kmods SIG d83023
+			/* Normal way. update attribute and exit */
Kmods SIG d83023
+			attr_b->nres.data_size = cpu_to_le64(new_size);
Kmods SIG d83023
+			goto ok;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		/* at least two mft to avoid recursive loop*/
Kmods SIG d83023
+		if (is_mft && next_svcn == vcn &&
Kmods SIG d83023
+		    ((u64)done << sbi->cluster_bits) >= 2 * sbi->record_size) {
Kmods SIG d83023
+			new_size = new_alloc_tmp;
Kmods SIG d83023
+			attr_b->nres.data_size = attr_b->nres.alloc_size;
Kmods SIG d83023
+			goto ok;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		if (le32_to_cpu(rec->used) < sbi->record_size) {
Kmods SIG d83023
+			old_alen = next_svcn;
Kmods SIG d83023
+			evcn = old_alen - 1;
Kmods SIG d83023
+			goto add_alloc_in_same_attr_seg;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		attr_b->nres.data_size = attr_b->nres.alloc_size;
Kmods SIG d83023
+		if (new_alloc_tmp < old_valid)
Kmods SIG d83023
+			attr_b->nres.valid_size = attr_b->nres.data_size;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (type == ATTR_LIST) {
Kmods SIG d83023
+			err = ni_expand_list(ni);
Kmods SIG d83023
+			if (err)
Kmods SIG d83023
+				goto out;
Kmods SIG d83023
+			if (next_svcn < vcn)
Kmods SIG d83023
+				goto pack_runs;
Kmods SIG d83023
+
Kmods SIG d83023
+			/* layout of records is changed */
Kmods SIG d83023
+			goto again;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		if (!ni->attr_list.size) {
Kmods SIG d83023
+			err = ni_create_attr_list(ni);
Kmods SIG d83023
+			if (err)
Kmods SIG d83023
+				goto out;
Kmods SIG d83023
+			/* layout of records is changed */
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		if (next_svcn >= vcn) {
Kmods SIG d83023
+			/* this is mft data, repeat */
Kmods SIG d83023
+			goto again;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		/* insert new attribute segment */
Kmods SIG d83023
+		err = ni_insert_nonresident(ni, type, name, name_len, run,
Kmods SIG d83023
+					    next_svcn, vcn - next_svcn,
Kmods SIG d83023
+					    attr_b->flags, &attr, &mi);
Kmods SIG d83023
+		if (err)
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (!is_mft)
Kmods SIG d83023
+			run_truncate_head(run, evcn + 1);
Kmods SIG d83023
+
Kmods SIG d83023
+		svcn = le64_to_cpu(attr->nres.svcn);
Kmods SIG d83023
+		evcn = le64_to_cpu(attr->nres.evcn);
Kmods SIG d83023
+
Kmods SIG d83023
+		le_b = NULL;
Kmods SIG d83023
+		/* layout of records maybe changed */
Kmods SIG d83023
+		/* find base attribute to update*/
Kmods SIG d83023
+		attr_b = ni_find_attr(ni, NULL, &le_b, type, name, name_len,
Kmods SIG d83023
+				      NULL, &mi_b);
Kmods SIG d83023
+		if (!attr_b) {
Kmods SIG d83023
+			err = -ENOENT;
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		attr_b->nres.alloc_size = cpu_to_le64((u64)vcn << cluster_bits);
Kmods SIG d83023
+		attr_b->nres.data_size = attr_b->nres.alloc_size;
Kmods SIG d83023
+		attr_b->nres.valid_size = attr_b->nres.alloc_size;
Kmods SIG d83023
+		mi_b->dirty = true;
Kmods SIG d83023
+		goto again_1;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	if (new_size != old_size ||
Kmods SIG d83023
+	    (new_alloc != old_alloc && !keep_prealloc)) {
Kmods SIG d83023
+		vcn = max(svcn, new_alen);
Kmods SIG d83023
+		new_alloc_tmp = (u64)vcn << cluster_bits;
Kmods SIG d83023
+
Kmods SIG d83023
+		alen = 0;
Kmods SIG d83023
+		err = run_deallocate_ex(sbi, run, vcn, evcn - vcn + 1, &alen,
Kmods SIG d83023
+					true);
Kmods SIG d83023
+		if (err)
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+		run_truncate(run, vcn);
Kmods SIG d83023
+
Kmods SIG d83023
+		if (vcn > svcn) {
Kmods SIG d83023
+			err = mi_pack_runs(mi, attr, run, vcn - svcn);
Kmods SIG d83023
+			if (err)
Kmods SIG d83023
+				goto out;
Kmods SIG d83023
+		} else if (le && le->vcn) {
Kmods SIG d83023
+			u16 le_sz = le16_to_cpu(le->size);
Kmods SIG d83023
+
Kmods SIG d83023
+			/*
Kmods SIG d83023
+			 * NOTE: list entries for one attribute are always
Kmods SIG d83023
+			 * the same size. We deal with last entry (vcn==0)
Kmods SIG d83023
+			 * and it is not first in entries array
Kmods SIG d83023
+			 * (list entry for std attribute always first)
Kmods SIG d83023
+			 * So it is safe to step back
Kmods SIG d83023
+			 */
Kmods SIG d83023
+			mi_remove_attr(mi, attr);
Kmods SIG d83023
+
Kmods SIG d83023
+			if (!al_remove_le(ni, le)) {
Kmods SIG d83023
+				err = -EINVAL;
Kmods SIG d83023
+				goto out;
Kmods SIG d83023
+			}
Kmods SIG d83023
+
Kmods SIG d83023
+			le = (struct ATTR_LIST_ENTRY *)((u8 *)le - le_sz);
Kmods SIG d83023
+		} else {
Kmods SIG d83023
+			attr->nres.evcn = cpu_to_le64((u64)vcn - 1);
Kmods SIG d83023
+			mi->dirty = true;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		attr_b->nres.alloc_size = cpu_to_le64(new_alloc_tmp);
Kmods SIG d83023
+
Kmods SIG d83023
+		if (vcn == new_alen) {
Kmods SIG d83023
+			attr_b->nres.data_size = cpu_to_le64(new_size);
Kmods SIG d83023
+			if (new_size < old_valid)
Kmods SIG d83023
+				attr_b->nres.valid_size =
Kmods SIG d83023
+					attr_b->nres.data_size;
Kmods SIG d83023
+		} else {
Kmods SIG d83023
+			if (new_alloc_tmp <=
Kmods SIG d83023
+			    le64_to_cpu(attr_b->nres.data_size))
Kmods SIG d83023
+				attr_b->nres.data_size =
Kmods SIG d83023
+					attr_b->nres.alloc_size;
Kmods SIG d83023
+			if (new_alloc_tmp <
Kmods SIG d83023
+			    le64_to_cpu(attr_b->nres.valid_size))
Kmods SIG d83023
+				attr_b->nres.valid_size =
Kmods SIG d83023
+					attr_b->nres.alloc_size;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		if (is_ext)
Kmods SIG d83023
+			le64_sub_cpu(&attr_b->nres.total_size,
Kmods SIG d83023
+				     ((u64)alen << cluster_bits));
Kmods SIG d83023
+
Kmods SIG d83023
+		mi_b->dirty = true;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (new_alloc_tmp <= new_alloc)
Kmods SIG d83023
+			goto ok;
Kmods SIG d83023
+
Kmods SIG d83023
+		old_size = new_alloc_tmp;
Kmods SIG d83023
+		vcn = svcn - 1;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (le == le_b) {
Kmods SIG d83023
+			attr = attr_b;
Kmods SIG d83023
+			mi = mi_b;
Kmods SIG d83023
+			evcn = svcn - 1;
Kmods SIG d83023
+			svcn = 0;
Kmods SIG d83023
+			goto next_le;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		if (le->type != type || le->name_len != name_len ||
Kmods SIG d83023
+		    memcmp(le_name(le), name, name_len * sizeof(short))) {
Kmods SIG d83023
+			err = -EINVAL;
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		err = ni_load_mi(ni, le, &mi);
Kmods SIG d83023
+		if (err)
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+		attr = mi_find_attr(mi, NULL, type, name, name_len, &le->id);
Kmods SIG d83023
+		if (!attr) {
Kmods SIG d83023
+			err = -EINVAL;
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+		goto next_le_1;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+ok:
Kmods SIG d83023
+	if (new_valid) {
Kmods SIG d83023
+		__le64 valid = cpu_to_le64(min(*new_valid, new_size));
Kmods SIG d83023
+
Kmods SIG d83023
+		if (attr_b->nres.valid_size != valid) {
Kmods SIG d83023
+			attr_b->nres.valid_size = valid;
Kmods SIG d83023
+			mi_b->dirty = true;
Kmods SIG d83023
+		}
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+out:
Kmods SIG d83023
+	if (!err && attr_b && ret)
Kmods SIG d83023
+		*ret = attr_b;
Kmods SIG d83023
+
Kmods SIG d83023
+	/* update inode_set_bytes*/
Kmods SIG d83023
+	if (!err && ((type == ATTR_DATA && !name_len) ||
Kmods SIG d83023
+		     (type == ATTR_ALLOC && name == I30_NAME))) {
Kmods SIG d83023
+		bool dirty = false;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (ni->vfs_inode.i_size != new_size) {
Kmods SIG d83023
+			ni->vfs_inode.i_size = new_size;
Kmods SIG d83023
+			dirty = true;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		if (attr_b && attr_b->non_res) {
Kmods SIG d83023
+			new_alloc = le64_to_cpu(attr_b->nres.alloc_size);
Kmods SIG d83023
+			if (inode_get_bytes(&ni->vfs_inode) != new_alloc) {
Kmods SIG d83023
+				inode_set_bytes(&ni->vfs_inode, new_alloc);
Kmods SIG d83023
+				dirty = true;
Kmods SIG d83023
+			}
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		if (dirty) {
Kmods SIG d83023
+			ni->ni_flags |= NI_FLAG_UPDATE_PARENT;
Kmods SIG d83023
+			mark_inode_dirty(&ni->vfs_inode);
Kmods SIG d83023
+		}
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	return err;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+int attr_data_get_block(struct ntfs_inode *ni, CLST vcn, CLST clen, CLST *lcn,
Kmods SIG d83023
+			CLST *len, bool *new)
Kmods SIG d83023
+{
Kmods SIG d83023
+	int err = 0;
Kmods SIG d83023
+	struct runs_tree *run = &ni->file.run;
Kmods SIG d83023
+	struct ntfs_sb_info *sbi;
Kmods SIG d83023
+	u8 cluster_bits;
Kmods SIG d83023
+	struct ATTRIB *attr = NULL, *attr_b;
Kmods SIG d83023
+	struct ATTR_LIST_ENTRY *le, *le_b;
Kmods SIG d83023
+	struct mft_inode *mi, *mi_b;
Kmods SIG d83023
+	CLST hint, svcn, to_alloc, evcn1, next_svcn, asize, end;
Kmods SIG d83023
+	u64 total_size;
Kmods SIG d83023
+	u32 clst_per_frame;
Kmods SIG d83023
+	bool ok;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (new)
Kmods SIG d83023
+		*new = false;
Kmods SIG d83023
+
Kmods SIG d83023
+	down_read(&ni->file.run_lock);
Kmods SIG d83023
+	ok = run_lookup_entry(run, vcn, lcn, len, NULL);
Kmods SIG d83023
+	up_read(&ni->file.run_lock);
Kmods SIG d83023
+
Kmods SIG d83023
+	if (ok && (*lcn != SPARSE_LCN || !new)) {
Kmods SIG d83023
+		/* normal way */
Kmods SIG d83023
+		return 0;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!clen)
Kmods SIG d83023
+		clen = 1;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (ok && clen > *len)
Kmods SIG d83023
+		clen = *len;
Kmods SIG d83023
+
Kmods SIG d83023
+	sbi = ni->mi.sbi;
Kmods SIG d83023
+	cluster_bits = sbi->cluster_bits;
Kmods SIG d83023
+
Kmods SIG d83023
+	ni_lock(ni);
Kmods SIG d83023
+	down_write(&ni->file.run_lock);
Kmods SIG d83023
+
Kmods SIG d83023
+	le_b = NULL;
Kmods SIG d83023
+	attr_b = ni_find_attr(ni, NULL, &le_b, ATTR_DATA, NULL, 0, NULL, &mi_b);
Kmods SIG d83023
+	if (!attr_b) {
Kmods SIG d83023
+		err = -ENOENT;
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!attr_b->non_res) {
Kmods SIG d83023
+		*lcn = RESIDENT_LCN;
Kmods SIG d83023
+		*len = 1;
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	asize = le64_to_cpu(attr_b->nres.alloc_size) >> sbi->cluster_bits;
Kmods SIG d83023
+	if (vcn >= asize) {
Kmods SIG d83023
+		err = -EINVAL;
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	clst_per_frame = 1u << attr_b->nres.c_unit;
Kmods SIG d83023
+	to_alloc = (clen + clst_per_frame - 1) & ~(clst_per_frame - 1);
Kmods SIG d83023
+
Kmods SIG d83023
+	if (vcn + to_alloc > asize)
Kmods SIG d83023
+		to_alloc = asize - vcn;
Kmods SIG d83023
+
Kmods SIG d83023
+	svcn = le64_to_cpu(attr_b->nres.svcn);
Kmods SIG d83023
+	evcn1 = le64_to_cpu(attr_b->nres.evcn) + 1;
Kmods SIG d83023
+
Kmods SIG d83023
+	attr = attr_b;
Kmods SIG d83023
+	le = le_b;
Kmods SIG d83023
+	mi = mi_b;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (le_b && (vcn < svcn || evcn1 <= vcn)) {
Kmods SIG d83023
+		attr = ni_find_attr(ni, attr_b, &le, ATTR_DATA, NULL, 0, &vcn,
Kmods SIG d83023
+				    &mi);
Kmods SIG d83023
+		if (!attr) {
Kmods SIG d83023
+			err = -EINVAL;
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+		svcn = le64_to_cpu(attr->nres.svcn);
Kmods SIG d83023
+		evcn1 = le64_to_cpu(attr->nres.evcn) + 1;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	err = attr_load_runs(attr, ni, run, NULL);
Kmods SIG d83023
+	if (err)
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!ok) {
Kmods SIG d83023
+		ok = run_lookup_entry(run, vcn, lcn, len, NULL);
Kmods SIG d83023
+		if (ok && (*lcn != SPARSE_LCN || !new)) {
Kmods SIG d83023
+			/* normal way */
Kmods SIG d83023
+			err = 0;
Kmods SIG d83023
+			goto ok;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		if (!ok && !new) {
Kmods SIG d83023
+			*len = 0;
Kmods SIG d83023
+			err = 0;
Kmods SIG d83023
+			goto ok;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		if (ok && clen > *len) {
Kmods SIG d83023
+			clen = *len;
Kmods SIG d83023
+			to_alloc = (clen + clst_per_frame - 1) &
Kmods SIG d83023
+				   ~(clst_per_frame - 1);
Kmods SIG d83023
+		}
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!is_attr_ext(attr_b)) {
Kmods SIG d83023
+		err = -EINVAL;
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	/* Get the last lcn to allocate from */
Kmods SIG d83023
+	hint = 0;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (vcn > evcn1) {
Kmods SIG d83023
+		if (!run_add_entry(run, evcn1, SPARSE_LCN, vcn - evcn1,
Kmods SIG d83023
+				   false)) {
Kmods SIG d83023
+			err = -ENOMEM;
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+	} else if (vcn && !run_lookup_entry(run, vcn - 1, &hint, NULL, NULL)) {
Kmods SIG d83023
+		hint = -1;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	err = attr_allocate_clusters(
Kmods SIG d83023
+		sbi, run, vcn, hint + 1, to_alloc, NULL, 0, len,
Kmods SIG d83023
+		(sbi->record_size - le32_to_cpu(mi->mrec->used) + 8) / 3 + 1,
Kmods SIG d83023
+		lcn);
Kmods SIG d83023
+	if (err)
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	*new = true;
Kmods SIG d83023
+
Kmods SIG d83023
+	end = vcn + *len;
Kmods SIG d83023
+
Kmods SIG d83023
+	total_size = le64_to_cpu(attr_b->nres.total_size) +
Kmods SIG d83023
+		     ((u64)*len << cluster_bits);
Kmods SIG d83023
+
Kmods SIG d83023
+repack:
Kmods SIG d83023
+	err = mi_pack_runs(mi, attr, run, max(end, evcn1) - svcn);
Kmods SIG d83023
+	if (err)
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+	attr_b->nres.total_size = cpu_to_le64(total_size);
Kmods SIG d83023
+	inode_set_bytes(&ni->vfs_inode, total_size);
Kmods SIG d83023
+	ni->ni_flags |= NI_FLAG_UPDATE_PARENT;
Kmods SIG d83023
+
Kmods SIG d83023
+	mi_b->dirty = true;
Kmods SIG d83023
+	mark_inode_dirty(&ni->vfs_inode);
Kmods SIG d83023
+
Kmods SIG d83023
+	/* stored [vcn : next_svcn) from [vcn : end) */
Kmods SIG d83023
+	next_svcn = le64_to_cpu(attr->nres.evcn) + 1;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (end <= evcn1) {
Kmods SIG d83023
+		if (next_svcn == evcn1) {
Kmods SIG d83023
+			/* Normal way. update attribute and exit */
Kmods SIG d83023
+			goto ok;
Kmods SIG d83023
+		}
Kmods SIG d83023
+		/* add new segment [next_svcn : evcn1 - next_svcn )*/
Kmods SIG d83023
+		if (!ni->attr_list.size) {
Kmods SIG d83023
+			err = ni_create_attr_list(ni);
Kmods SIG d83023
+			if (err)
Kmods SIG d83023
+				goto out;
Kmods SIG d83023
+			/* layout of records is changed */
Kmods SIG d83023
+			le_b = NULL;
Kmods SIG d83023
+			attr_b = ni_find_attr(ni, NULL, &le_b, ATTR_DATA, NULL,
Kmods SIG d83023
+					      0, NULL, &mi_b);
Kmods SIG d83023
+			if (!attr_b) {
Kmods SIG d83023
+				err = -ENOENT;
Kmods SIG d83023
+				goto out;
Kmods SIG d83023
+			}
Kmods SIG d83023
+
Kmods SIG d83023
+			attr = attr_b;
Kmods SIG d83023
+			le = le_b;
Kmods SIG d83023
+			mi = mi_b;
Kmods SIG d83023
+			goto repack;
Kmods SIG d83023
+		}
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	svcn = evcn1;
Kmods SIG d83023
+
Kmods SIG d83023
+	/* Estimate next attribute */
Kmods SIG d83023
+	attr = ni_find_attr(ni, attr, &le, ATTR_DATA, NULL, 0, &svcn, &mi);
Kmods SIG d83023
+
Kmods SIG d83023
+	if (attr) {
Kmods SIG d83023
+		CLST alloc = bytes_to_cluster(
Kmods SIG d83023
+			sbi, le64_to_cpu(attr_b->nres.alloc_size));
Kmods SIG d83023
+		CLST evcn = le64_to_cpu(attr->nres.evcn);
Kmods SIG d83023
+
Kmods SIG d83023
+		if (end < next_svcn)
Kmods SIG d83023
+			end = next_svcn;
Kmods SIG d83023
+		while (end > evcn) {
Kmods SIG d83023
+			/* remove segment [svcn : evcn)*/
Kmods SIG d83023
+			mi_remove_attr(mi, attr);
Kmods SIG d83023
+
Kmods SIG d83023
+			if (!al_remove_le(ni, le)) {
Kmods SIG d83023
+				err = -EINVAL;
Kmods SIG d83023
+				goto out;
Kmods SIG d83023
+			}
Kmods SIG d83023
+
Kmods SIG d83023
+			if (evcn + 1 >= alloc) {
Kmods SIG d83023
+				/* last attribute segment */
Kmods SIG d83023
+				evcn1 = evcn + 1;
Kmods SIG d83023
+				goto ins_ext;
Kmods SIG d83023
+			}
Kmods SIG d83023
+
Kmods SIG d83023
+			if (ni_load_mi(ni, le, &mi)) {
Kmods SIG d83023
+				attr = NULL;
Kmods SIG d83023
+				goto out;
Kmods SIG d83023
+			}
Kmods SIG d83023
+
Kmods SIG d83023
+			attr = mi_find_attr(mi, NULL, ATTR_DATA, NULL, 0,
Kmods SIG d83023
+					    &le->id);
Kmods SIG d83023
+			if (!attr) {
Kmods SIG d83023
+				err = -EINVAL;
Kmods SIG d83023
+				goto out;
Kmods SIG d83023
+			}
Kmods SIG d83023
+			svcn = le64_to_cpu(attr->nres.svcn);
Kmods SIG d83023
+			evcn = le64_to_cpu(attr->nres.evcn);
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		if (end < svcn)
Kmods SIG d83023
+			end = svcn;
Kmods SIG d83023
+
Kmods SIG d83023
+		err = attr_load_runs(attr, ni, run, &end;;
Kmods SIG d83023
+		if (err)
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+		evcn1 = evcn + 1;
Kmods SIG d83023
+		attr->nres.svcn = cpu_to_le64(next_svcn);
Kmods SIG d83023
+		err = mi_pack_runs(mi, attr, run, evcn1 - next_svcn);
Kmods SIG d83023
+		if (err)
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+		le->vcn = cpu_to_le64(next_svcn);
Kmods SIG d83023
+		ni->attr_list.dirty = true;
Kmods SIG d83023
+		mi->dirty = true;
Kmods SIG d83023
+
Kmods SIG d83023
+		next_svcn = le64_to_cpu(attr->nres.evcn) + 1;
Kmods SIG d83023
+	}
Kmods SIG d83023
+ins_ext:
Kmods SIG d83023
+	if (evcn1 > next_svcn) {
Kmods SIG d83023
+		err = ni_insert_nonresident(ni, ATTR_DATA, NULL, 0, run,
Kmods SIG d83023
+					    next_svcn, evcn1 - next_svcn,
Kmods SIG d83023
+					    attr_b->flags, &attr, &mi);
Kmods SIG d83023
+		if (err)
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+ok:
Kmods SIG d83023
+	run_truncate_around(run, vcn);
Kmods SIG d83023
+out:
Kmods SIG d83023
+	up_write(&ni->file.run_lock);
Kmods SIG d83023
+	ni_unlock(ni);
Kmods SIG d83023
+
Kmods SIG d83023
+	return err;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+int attr_data_read_resident(struct ntfs_inode *ni, struct page *page)
Kmods SIG d83023
+{
Kmods SIG d83023
+	u64 vbo;
Kmods SIG d83023
+	struct ATTRIB *attr;
Kmods SIG d83023
+	u32 data_size;
Kmods SIG d83023
+
Kmods SIG d83023
+	attr = ni_find_attr(ni, NULL, NULL, ATTR_DATA, NULL, 0, NULL, NULL);
Kmods SIG d83023
+	if (!attr)
Kmods SIG d83023
+		return -EINVAL;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (attr->non_res)
Kmods SIG d83023
+		return E_NTFS_NONRESIDENT;
Kmods SIG d83023
+
Kmods SIG d83023
+	vbo = page->index << PAGE_SHIFT;
Kmods SIG d83023
+	data_size = le32_to_cpu(attr->res.data_size);
Kmods SIG d83023
+	if (vbo < data_size) {
Kmods SIG d83023
+		const char *data = resident_data(attr);
Kmods SIG d83023
+		char *kaddr = kmap_atomic(page);
Kmods SIG d83023
+		u32 use = data_size - vbo;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (use > PAGE_SIZE)
Kmods SIG d83023
+			use = PAGE_SIZE;
Kmods SIG d83023
+
Kmods SIG d83023
+		memcpy(kaddr, data + vbo, use);
Kmods SIG d83023
+		memset(kaddr + use, 0, PAGE_SIZE - use);
Kmods SIG d83023
+		kunmap_atomic(kaddr);
Kmods SIG d83023
+		flush_dcache_page(page);
Kmods SIG d83023
+		SetPageUptodate(page);
Kmods SIG d83023
+	} else if (!PageUptodate(page)) {
Kmods SIG d83023
+		zero_user_segment(page, 0, PAGE_SIZE);
Kmods SIG d83023
+		SetPageUptodate(page);
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	return 0;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+int attr_data_write_resident(struct ntfs_inode *ni, struct page *page)
Kmods SIG d83023
+{
Kmods SIG d83023
+	u64 vbo;
Kmods SIG d83023
+	struct mft_inode *mi;
Kmods SIG d83023
+	struct ATTRIB *attr;
Kmods SIG d83023
+	u32 data_size;
Kmods SIG d83023
+
Kmods SIG d83023
+	attr = ni_find_attr(ni, NULL, NULL, ATTR_DATA, NULL, 0, NULL, &mi);
Kmods SIG d83023
+	if (!attr)
Kmods SIG d83023
+		return -EINVAL;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (attr->non_res) {
Kmods SIG d83023
+		/*return special error code to check this case*/
Kmods SIG d83023
+		return E_NTFS_NONRESIDENT;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	vbo = page->index << PAGE_SHIFT;
Kmods SIG d83023
+	data_size = le32_to_cpu(attr->res.data_size);
Kmods SIG d83023
+	if (vbo < data_size) {
Kmods SIG d83023
+		char *data = resident_data(attr);
Kmods SIG d83023
+		char *kaddr = kmap_atomic(page);
Kmods SIG d83023
+		u32 use = data_size - vbo;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (use > PAGE_SIZE)
Kmods SIG d83023
+			use = PAGE_SIZE;
Kmods SIG d83023
+		memcpy(data + vbo, kaddr, use);
Kmods SIG d83023
+		kunmap_atomic(kaddr);
Kmods SIG d83023
+		mi->dirty = true;
Kmods SIG d83023
+	}
Kmods SIG d83023
+	ni->i_valid = data_size;
Kmods SIG d83023
+
Kmods SIG d83023
+	return 0;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * attr_load_runs_vcn
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * load runs with vcn
Kmods SIG d83023
+ */
Kmods SIG d83023
+int attr_load_runs_vcn(struct ntfs_inode *ni, enum ATTR_TYPE type,
Kmods SIG d83023
+		       const __le16 *name, u8 name_len, struct runs_tree *run,
Kmods SIG d83023
+		       CLST vcn)
Kmods SIG d83023
+{
Kmods SIG d83023
+	struct ATTRIB *attr;
Kmods SIG d83023
+	int err;
Kmods SIG d83023
+	CLST svcn, evcn;
Kmods SIG d83023
+	u16 ro;
Kmods SIG d83023
+
Kmods SIG d83023
+	attr = ni_find_attr(ni, NULL, NULL, type, name, name_len, &vcn, NULL);
Kmods SIG d83023
+	if (!attr)
Kmods SIG d83023
+		return -ENOENT;
Kmods SIG d83023
+
Kmods SIG d83023
+	svcn = le64_to_cpu(attr->nres.svcn);
Kmods SIG d83023
+	evcn = le64_to_cpu(attr->nres.evcn);
Kmods SIG d83023
+
Kmods SIG d83023
+	if (evcn < vcn || vcn < svcn)
Kmods SIG d83023
+		return -EINVAL;
Kmods SIG d83023
+
Kmods SIG d83023
+	ro = le16_to_cpu(attr->nres.run_off);
Kmods SIG d83023
+	err = run_unpack_ex(run, ni->mi.sbi, ni->mi.rno, svcn, evcn, svcn,
Kmods SIG d83023
+			    Add2Ptr(attr, ro), le32_to_cpu(attr->size) - ro);
Kmods SIG d83023
+	if (err < 0)
Kmods SIG d83023
+		return err;
Kmods SIG d83023
+	return 0;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * load runs for given range [from to)
Kmods SIG d83023
+ */
Kmods SIG d83023
+int attr_load_runs_range(struct ntfs_inode *ni, enum ATTR_TYPE type,
Kmods SIG d83023
+			 const __le16 *name, u8 name_len, struct runs_tree *run,
Kmods SIG d83023
+			 u64 from, u64 to)
Kmods SIG d83023
+{
Kmods SIG d83023
+	struct ntfs_sb_info *sbi = ni->mi.sbi;
Kmods SIG d83023
+	u8 cluster_bits = sbi->cluster_bits;
Kmods SIG d83023
+	CLST vcn = from >> cluster_bits;
Kmods SIG d83023
+	CLST vcn_last = (to - 1) >> cluster_bits;
Kmods SIG d83023
+	CLST lcn, clen;
Kmods SIG d83023
+	int err;
Kmods SIG d83023
+
Kmods SIG d83023
+	for (vcn = from >> cluster_bits; vcn <= vcn_last; vcn += clen) {
Kmods SIG d83023
+		if (!run_lookup_entry(run, vcn, &lcn, &clen, NULL)) {
Kmods SIG d83023
+			err = attr_load_runs_vcn(ni, type, name, name_len, run,
Kmods SIG d83023
+						 vcn);
Kmods SIG d83023
+			if (err)
Kmods SIG d83023
+				return err;
Kmods SIG d83023
+			clen = 0; /*next run_lookup_entry(vcn) must be success*/
Kmods SIG d83023
+		}
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	return 0;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+#ifdef CONFIG_NTFS3_LZX_XPRESS
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * attr_wof_frame_info
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * read header of xpress/lzx file to get info about frame
Kmods SIG d83023
+ */
Kmods SIG d83023
+int attr_wof_frame_info(struct ntfs_inode *ni, struct ATTRIB *attr,
Kmods SIG d83023
+			struct runs_tree *run, u64 frame, u64 frames,
Kmods SIG d83023
+			u8 frame_bits, u32 *ondisk_size, u64 *vbo_data)
Kmods SIG d83023
+{
Kmods SIG d83023
+	struct ntfs_sb_info *sbi = ni->mi.sbi;
Kmods SIG d83023
+	u64 vbo[2], off[2], wof_size;
Kmods SIG d83023
+	u32 voff;
Kmods SIG d83023
+	u8 bytes_per_off;
Kmods SIG d83023
+	char *addr;
Kmods SIG d83023
+	struct page *page;
Kmods SIG d83023
+	int i, err;
Kmods SIG d83023
+	__le32 *off32;
Kmods SIG d83023
+	__le64 *off64;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (ni->vfs_inode.i_size < 0x100000000ull) {
Kmods SIG d83023
+		/* file starts with array of 32 bit offsets */
Kmods SIG d83023
+		bytes_per_off = sizeof(__le32);
Kmods SIG d83023
+		vbo[1] = frame << 2;
Kmods SIG d83023
+		*vbo_data = frames << 2;
Kmods SIG d83023
+	} else {
Kmods SIG d83023
+		/* file starts with array of 64 bit offsets */
Kmods SIG d83023
+		bytes_per_off = sizeof(__le64);
Kmods SIG d83023
+		vbo[1] = frame << 3;
Kmods SIG d83023
+		*vbo_data = frames << 3;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	/*
Kmods SIG d83023
+	 * read 4/8 bytes at [vbo - 4(8)] == offset where compressed frame starts
Kmods SIG d83023
+	 * read 4/8 bytes at [vbo] == offset where compressed frame ends
Kmods SIG d83023
+	 */
Kmods SIG d83023
+	if (!attr->non_res) {
Kmods SIG d83023
+		if (vbo[1] + bytes_per_off > le32_to_cpu(attr->res.data_size)) {
Kmods SIG d83023
+			ntfs_inode_err(&ni->vfs_inode, "is corrupted");
Kmods SIG d83023
+			return -EINVAL;
Kmods SIG d83023
+		}
Kmods SIG d83023
+		addr = resident_data(attr);
Kmods SIG d83023
+
Kmods SIG d83023
+		if (bytes_per_off == sizeof(__le32)) {
Kmods SIG d83023
+			off32 = Add2Ptr(addr, vbo[1]);
Kmods SIG d83023
+			off[0] = vbo[1] ? le32_to_cpu(off32[-1]) : 0;
Kmods SIG d83023
+			off[1] = le32_to_cpu(off32[0]);
Kmods SIG d83023
+		} else {
Kmods SIG d83023
+			off64 = Add2Ptr(addr, vbo[1]);
Kmods SIG d83023
+			off[0] = vbo[1] ? le64_to_cpu(off64[-1]) : 0;
Kmods SIG d83023
+			off[1] = le64_to_cpu(off64[0]);
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		*vbo_data += off[0];
Kmods SIG d83023
+		*ondisk_size = off[1] - off[0];
Kmods SIG d83023
+		return 0;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	wof_size = le64_to_cpu(attr->nres.data_size);
Kmods SIG d83023
+	down_write(&ni->file.run_lock);
Kmods SIG d83023
+	page = ni->file.offs_page;
Kmods SIG d83023
+	if (!page) {
Kmods SIG d83023
+		page = alloc_page(GFP_KERNEL);
Kmods SIG d83023
+		if (!page) {
Kmods SIG d83023
+			err = -ENOMEM;
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+		page->index = -1;
Kmods SIG d83023
+		ni->file.offs_page = page;
Kmods SIG d83023
+	}
Kmods SIG d83023
+	lock_page(page);
Kmods SIG d83023
+	addr = page_address(page);
Kmods SIG d83023
+
Kmods SIG d83023
+	if (vbo[1]) {
Kmods SIG d83023
+		voff = vbo[1] & (PAGE_SIZE - 1);
Kmods SIG d83023
+		vbo[0] = vbo[1] - bytes_per_off;
Kmods SIG d83023
+		i = 0;
Kmods SIG d83023
+	} else {
Kmods SIG d83023
+		voff = 0;
Kmods SIG d83023
+		vbo[0] = 0;
Kmods SIG d83023
+		off[0] = 0;
Kmods SIG d83023
+		i = 1;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	do {
Kmods SIG d83023
+		pgoff_t index = vbo[i] >> PAGE_SHIFT;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (index != page->index) {
Kmods SIG d83023
+			u64 from = vbo[i] & ~(u64)(PAGE_SIZE - 1);
Kmods SIG d83023
+			u64 to = min(from + PAGE_SIZE, wof_size);
Kmods SIG d83023
+
Kmods SIG d83023
+			err = attr_load_runs_range(ni, ATTR_DATA, WOF_NAME,
Kmods SIG d83023
+						   ARRAY_SIZE(WOF_NAME), run,
Kmods SIG d83023
+						   from, to);
Kmods SIG d83023
+			if (err)
Kmods SIG d83023
+				goto out1;
Kmods SIG d83023
+
Kmods SIG d83023
+			err = ntfs_bio_pages(sbi, run, &page, 1, from,
Kmods SIG d83023
+					     to - from, REQ_OP_READ);
Kmods SIG d83023
+			if (err) {
Kmods SIG d83023
+				page->index = -1;
Kmods SIG d83023
+				goto out1;
Kmods SIG d83023
+			}
Kmods SIG d83023
+			page->index = index;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		if (i) {
Kmods SIG d83023
+			if (bytes_per_off == sizeof(__le32)) {
Kmods SIG d83023
+				off32 = Add2Ptr(addr, voff);
Kmods SIG d83023
+				off[1] = le32_to_cpu(*off32);
Kmods SIG d83023
+			} else {
Kmods SIG d83023
+				off64 = Add2Ptr(addr, voff);
Kmods SIG d83023
+				off[1] = le64_to_cpu(*off64);
Kmods SIG d83023
+			}
Kmods SIG d83023
+		} else if (!voff) {
Kmods SIG d83023
+			if (bytes_per_off == sizeof(__le32)) {
Kmods SIG d83023
+				off32 = Add2Ptr(addr, PAGE_SIZE - sizeof(u32));
Kmods SIG d83023
+				off[0] = le32_to_cpu(*off32);
Kmods SIG d83023
+			} else {
Kmods SIG d83023
+				off64 = Add2Ptr(addr, PAGE_SIZE - sizeof(u64));
Kmods SIG d83023
+				off[0] = le64_to_cpu(*off64);
Kmods SIG d83023
+			}
Kmods SIG d83023
+		} else {
Kmods SIG d83023
+			/* two values in one page*/
Kmods SIG d83023
+			if (bytes_per_off == sizeof(__le32)) {
Kmods SIG d83023
+				off32 = Add2Ptr(addr, voff);
Kmods SIG d83023
+				off[0] = le32_to_cpu(off32[-1]);
Kmods SIG d83023
+				off[1] = le32_to_cpu(off32[0]);
Kmods SIG d83023
+			} else {
Kmods SIG d83023
+				off64 = Add2Ptr(addr, voff);
Kmods SIG d83023
+				off[0] = le64_to_cpu(off64[-1]);
Kmods SIG d83023
+				off[1] = le64_to_cpu(off64[0]);
Kmods SIG d83023
+			}
Kmods SIG d83023
+			break;
Kmods SIG d83023
+		}
Kmods SIG d83023
+	} while (++i < 2);
Kmods SIG d83023
+
Kmods SIG d83023
+	*vbo_data += off[0];
Kmods SIG d83023
+	*ondisk_size = off[1] - off[0];
Kmods SIG d83023
+
Kmods SIG d83023
+out1:
Kmods SIG d83023
+	unlock_page(page);
Kmods SIG d83023
+out:
Kmods SIG d83023
+	up_write(&ni->file.run_lock);
Kmods SIG d83023
+	return err;
Kmods SIG d83023
+}
Kmods SIG d83023
+#endif
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * attr_is_frame_compressed
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * This function is used to detect compressed frame
Kmods SIG d83023
+ */
Kmods SIG d83023
+int attr_is_frame_compressed(struct ntfs_inode *ni, struct ATTRIB *attr,
Kmods SIG d83023
+			     CLST frame, CLST *clst_data)
Kmods SIG d83023
+{
Kmods SIG d83023
+	int err;
Kmods SIG d83023
+	u32 clst_frame;
Kmods SIG d83023
+	CLST clen, lcn, vcn, alen, slen, vcn_next;
Kmods SIG d83023
+	size_t idx;
Kmods SIG d83023
+	struct runs_tree *run;
Kmods SIG d83023
+
Kmods SIG d83023
+	*clst_data = 0;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!is_attr_compressed(attr))
Kmods SIG d83023
+		return 0;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!attr->non_res)
Kmods SIG d83023
+		return 0;
Kmods SIG d83023
+
Kmods SIG d83023
+	clst_frame = 1u << attr->nres.c_unit;
Kmods SIG d83023
+	vcn = frame * clst_frame;
Kmods SIG d83023
+	run = &ni->file.run;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!run_lookup_entry(run, vcn, &lcn, &clen, &idx)) {
Kmods SIG d83023
+		err = attr_load_runs_vcn(ni, attr->type, attr_name(attr),
Kmods SIG d83023
+					 attr->name_len, run, vcn);
Kmods SIG d83023
+		if (err)
Kmods SIG d83023
+			return err;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (!run_lookup_entry(run, vcn, &lcn, &clen, &idx))
Kmods SIG d83023
+			return -EINVAL;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	if (lcn == SPARSE_LCN) {
Kmods SIG d83023
+		/* sparsed frame */
Kmods SIG d83023
+		return 0;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	if (clen >= clst_frame) {
Kmods SIG d83023
+		/*
Kmods SIG d83023
+		 * The frame is not compressed 'cause
Kmods SIG d83023
+		 * it does not contain any sparse clusters
Kmods SIG d83023
+		 */
Kmods SIG d83023
+		*clst_data = clst_frame;
Kmods SIG d83023
+		return 0;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	alen = bytes_to_cluster(ni->mi.sbi, le64_to_cpu(attr->nres.alloc_size));
Kmods SIG d83023
+	slen = 0;
Kmods SIG d83023
+	*clst_data = clen;
Kmods SIG d83023
+
Kmods SIG d83023
+	/*
Kmods SIG d83023
+	 * The frame is compressed if *clst_data + slen >= clst_frame
Kmods SIG d83023
+	 * Check next fragments
Kmods SIG d83023
+	 */
Kmods SIG d83023
+	while ((vcn += clen) < alen) {
Kmods SIG d83023
+		vcn_next = vcn;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (!run_get_entry(run, ++idx, &vcn, &lcn, &clen) ||
Kmods SIG d83023
+		    vcn_next != vcn) {
Kmods SIG d83023
+			err = attr_load_runs_vcn(ni, attr->type,
Kmods SIG d83023
+						 attr_name(attr),
Kmods SIG d83023
+						 attr->name_len, run, vcn_next);
Kmods SIG d83023
+			if (err)
Kmods SIG d83023
+				return err;
Kmods SIG d83023
+			vcn = vcn_next;
Kmods SIG d83023
+
Kmods SIG d83023
+			if (!run_lookup_entry(run, vcn, &lcn, &clen, &idx))
Kmods SIG d83023
+				return -EINVAL;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		if (lcn == SPARSE_LCN) {
Kmods SIG d83023
+			slen += clen;
Kmods SIG d83023
+		} else {
Kmods SIG d83023
+			if (slen) {
Kmods SIG d83023
+				/*
Kmods SIG d83023
+				 * data_clusters + sparse_clusters =
Kmods SIG d83023
+				 * not enough for frame
Kmods SIG d83023
+				 */
Kmods SIG d83023
+				return -EINVAL;
Kmods SIG d83023
+			}
Kmods SIG d83023
+			*clst_data += clen;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		if (*clst_data + slen >= clst_frame) {
Kmods SIG d83023
+			if (!slen) {
Kmods SIG d83023
+				/*
Kmods SIG d83023
+				 * There is no sparsed clusters in this frame
Kmods SIG d83023
+				 * So it is not compressed
Kmods SIG d83023
+				 */
Kmods SIG d83023
+				*clst_data = clst_frame;
Kmods SIG d83023
+			} else {
Kmods SIG d83023
+				/*frame is compressed*/
Kmods SIG d83023
+			}
Kmods SIG d83023
+			break;
Kmods SIG d83023
+		}
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	return 0;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * attr_allocate_frame
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * allocate/free clusters for 'frame'
Kmods SIG d83023
+ * assumed: down_write(&ni->file.run_lock);
Kmods SIG d83023
+ */
Kmods SIG d83023
+int attr_allocate_frame(struct ntfs_inode *ni, CLST frame, size_t compr_size,
Kmods SIG d83023
+			u64 new_valid)
Kmods SIG d83023
+{
Kmods SIG d83023
+	int err = 0;
Kmods SIG d83023
+	struct runs_tree *run = &ni->file.run;
Kmods SIG d83023
+	struct ntfs_sb_info *sbi = ni->mi.sbi;
Kmods SIG d83023
+	struct ATTRIB *attr = NULL, *attr_b;
Kmods SIG d83023
+	struct ATTR_LIST_ENTRY *le, *le_b;
Kmods SIG d83023
+	struct mft_inode *mi, *mi_b;
Kmods SIG d83023
+	CLST svcn, evcn1, next_svcn, lcn, len;
Kmods SIG d83023
+	CLST vcn, end, clst_data;
Kmods SIG d83023
+	u64 total_size, valid_size, data_size;
Kmods SIG d83023
+
Kmods SIG d83023
+	le_b = NULL;
Kmods SIG d83023
+	attr_b = ni_find_attr(ni, NULL, &le_b, ATTR_DATA, NULL, 0, NULL, &mi_b);
Kmods SIG d83023
+	if (!attr_b)
Kmods SIG d83023
+		return -ENOENT;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!is_attr_ext(attr_b))
Kmods SIG d83023
+		return -EINVAL;
Kmods SIG d83023
+
Kmods SIG d83023
+	vcn = frame << NTFS_LZNT_CUNIT;
Kmods SIG d83023
+	total_size = le64_to_cpu(attr_b->nres.total_size);
Kmods SIG d83023
+
Kmods SIG d83023
+	svcn = le64_to_cpu(attr_b->nres.svcn);
Kmods SIG d83023
+	evcn1 = le64_to_cpu(attr_b->nres.evcn) + 1;
Kmods SIG d83023
+	data_size = le64_to_cpu(attr_b->nres.data_size);
Kmods SIG d83023
+
Kmods SIG d83023
+	if (svcn <= vcn && vcn < evcn1) {
Kmods SIG d83023
+		attr = attr_b;
Kmods SIG d83023
+		le = le_b;
Kmods SIG d83023
+		mi = mi_b;
Kmods SIG d83023
+	} else if (!le_b) {
Kmods SIG d83023
+		err = -EINVAL;
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	} else {
Kmods SIG d83023
+		le = le_b;
Kmods SIG d83023
+		attr = ni_find_attr(ni, attr_b, &le, ATTR_DATA, NULL, 0, &vcn,
Kmods SIG d83023
+				    &mi);
Kmods SIG d83023
+		if (!attr) {
Kmods SIG d83023
+			err = -EINVAL;
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+		svcn = le64_to_cpu(attr->nres.svcn);
Kmods SIG d83023
+		evcn1 = le64_to_cpu(attr->nres.evcn) + 1;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	err = attr_load_runs(attr, ni, run, NULL);
Kmods SIG d83023
+	if (err)
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+	err = attr_is_frame_compressed(ni, attr_b, frame, &clst_data);
Kmods SIG d83023
+	if (err)
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+	total_size -= (u64)clst_data << sbi->cluster_bits;
Kmods SIG d83023
+
Kmods SIG d83023
+	len = bytes_to_cluster(sbi, compr_size);
Kmods SIG d83023
+
Kmods SIG d83023
+	if (len == clst_data)
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (len < clst_data) {
Kmods SIG d83023
+		err = run_deallocate_ex(sbi, run, vcn + len, clst_data - len,
Kmods SIG d83023
+					NULL, true);
Kmods SIG d83023
+		if (err)
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (!run_add_entry(run, vcn + len, SPARSE_LCN, clst_data - len,
Kmods SIG d83023
+				   false)) {
Kmods SIG d83023
+			err = -ENOMEM;
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+		end = vcn + clst_data;
Kmods SIG d83023
+		/* run contains updated range [vcn + len : end) */
Kmods SIG d83023
+	} else {
Kmods SIG d83023
+		CLST alen, hint = 0;
Kmods SIG d83023
+		/* Get the last lcn to allocate from */
Kmods SIG d83023
+		if (vcn + clst_data &&
Kmods SIG d83023
+		    !run_lookup_entry(run, vcn + clst_data - 1, &hint, NULL,
Kmods SIG d83023
+				      NULL)) {
Kmods SIG d83023
+			hint = -1;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		err = attr_allocate_clusters(sbi, run, vcn + clst_data,
Kmods SIG d83023
+					     hint + 1, len - clst_data, NULL, 0,
Kmods SIG d83023
+					     &alen, 0, &lcn;;
Kmods SIG d83023
+		if (err)
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+		end = vcn + len;
Kmods SIG d83023
+		/* run contains updated range [vcn + clst_data : end) */
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	total_size += (u64)len << sbi->cluster_bits;
Kmods SIG d83023
+
Kmods SIG d83023
+repack:
Kmods SIG d83023
+	err = mi_pack_runs(mi, attr, run, max(end, evcn1) - svcn);
Kmods SIG d83023
+	if (err)
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+	attr_b->nres.total_size = cpu_to_le64(total_size);
Kmods SIG d83023
+	inode_set_bytes(&ni->vfs_inode, total_size);
Kmods SIG d83023
+
Kmods SIG d83023
+	mi_b->dirty = true;
Kmods SIG d83023
+	mark_inode_dirty(&ni->vfs_inode);
Kmods SIG d83023
+
Kmods SIG d83023
+	/* stored [vcn : next_svcn) from [vcn : end) */
Kmods SIG d83023
+	next_svcn = le64_to_cpu(attr->nres.evcn) + 1;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (end <= evcn1) {
Kmods SIG d83023
+		if (next_svcn == evcn1) {
Kmods SIG d83023
+			/* Normal way. update attribute and exit */
Kmods SIG d83023
+			goto ok;
Kmods SIG d83023
+		}
Kmods SIG d83023
+		/* add new segment [next_svcn : evcn1 - next_svcn )*/
Kmods SIG d83023
+		if (!ni->attr_list.size) {
Kmods SIG d83023
+			err = ni_create_attr_list(ni);
Kmods SIG d83023
+			if (err)
Kmods SIG d83023
+				goto out;
Kmods SIG d83023
+			/* layout of records is changed */
Kmods SIG d83023
+			le_b = NULL;
Kmods SIG d83023
+			attr_b = ni_find_attr(ni, NULL, &le_b, ATTR_DATA, NULL,
Kmods SIG d83023
+					      0, NULL, &mi_b);
Kmods SIG d83023
+			if (!attr_b) {
Kmods SIG d83023
+				err = -ENOENT;
Kmods SIG d83023
+				goto out;
Kmods SIG d83023
+			}
Kmods SIG d83023
+
Kmods SIG d83023
+			attr = attr_b;
Kmods SIG d83023
+			le = le_b;
Kmods SIG d83023
+			mi = mi_b;
Kmods SIG d83023
+			goto repack;
Kmods SIG d83023
+		}
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	svcn = evcn1;
Kmods SIG d83023
+
Kmods SIG d83023
+	/* Estimate next attribute */
Kmods SIG d83023
+	attr = ni_find_attr(ni, attr, &le, ATTR_DATA, NULL, 0, &svcn, &mi);
Kmods SIG d83023
+
Kmods SIG d83023
+	if (attr) {
Kmods SIG d83023
+		CLST alloc = bytes_to_cluster(
Kmods SIG d83023
+			sbi, le64_to_cpu(attr_b->nres.alloc_size));
Kmods SIG d83023
+		CLST evcn = le64_to_cpu(attr->nres.evcn);
Kmods SIG d83023
+
Kmods SIG d83023
+		if (end < next_svcn)
Kmods SIG d83023
+			end = next_svcn;
Kmods SIG d83023
+		while (end > evcn) {
Kmods SIG d83023
+			/* remove segment [svcn : evcn)*/
Kmods SIG d83023
+			mi_remove_attr(mi, attr);
Kmods SIG d83023
+
Kmods SIG d83023
+			if (!al_remove_le(ni, le)) {
Kmods SIG d83023
+				err = -EINVAL;
Kmods SIG d83023
+				goto out;
Kmods SIG d83023
+			}
Kmods SIG d83023
+
Kmods SIG d83023
+			if (evcn + 1 >= alloc) {
Kmods SIG d83023
+				/* last attribute segment */
Kmods SIG d83023
+				evcn1 = evcn + 1;
Kmods SIG d83023
+				goto ins_ext;
Kmods SIG d83023
+			}
Kmods SIG d83023
+
Kmods SIG d83023
+			if (ni_load_mi(ni, le, &mi)) {
Kmods SIG d83023
+				attr = NULL;
Kmods SIG d83023
+				goto out;
Kmods SIG d83023
+			}
Kmods SIG d83023
+
Kmods SIG d83023
+			attr = mi_find_attr(mi, NULL, ATTR_DATA, NULL, 0,
Kmods SIG d83023
+					    &le->id);
Kmods SIG d83023
+			if (!attr) {
Kmods SIG d83023
+				err = -EINVAL;
Kmods SIG d83023
+				goto out;
Kmods SIG d83023
+			}
Kmods SIG d83023
+			svcn = le64_to_cpu(attr->nres.svcn);
Kmods SIG d83023
+			evcn = le64_to_cpu(attr->nres.evcn);
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		if (end < svcn)
Kmods SIG d83023
+			end = svcn;
Kmods SIG d83023
+
Kmods SIG d83023
+		err = attr_load_runs(attr, ni, run, &end;;
Kmods SIG d83023
+		if (err)
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+		evcn1 = evcn + 1;
Kmods SIG d83023
+		attr->nres.svcn = cpu_to_le64(next_svcn);
Kmods SIG d83023
+		err = mi_pack_runs(mi, attr, run, evcn1 - next_svcn);
Kmods SIG d83023
+		if (err)
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+		le->vcn = cpu_to_le64(next_svcn);
Kmods SIG d83023
+		ni->attr_list.dirty = true;
Kmods SIG d83023
+		mi->dirty = true;
Kmods SIG d83023
+
Kmods SIG d83023
+		next_svcn = le64_to_cpu(attr->nres.evcn) + 1;
Kmods SIG d83023
+	}
Kmods SIG d83023
+ins_ext:
Kmods SIG d83023
+	if (evcn1 > next_svcn) {
Kmods SIG d83023
+		err = ni_insert_nonresident(ni, ATTR_DATA, NULL, 0, run,
Kmods SIG d83023
+					    next_svcn, evcn1 - next_svcn,
Kmods SIG d83023
+					    attr_b->flags, &attr, &mi);
Kmods SIG d83023
+		if (err)
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+ok:
Kmods SIG d83023
+	run_truncate_around(run, vcn);
Kmods SIG d83023
+out:
Kmods SIG d83023
+	if (new_valid > data_size)
Kmods SIG d83023
+		new_valid = data_size;
Kmods SIG d83023
+
Kmods SIG d83023
+	valid_size = le64_to_cpu(attr_b->nres.valid_size);
Kmods SIG d83023
+	if (new_valid != valid_size) {
Kmods SIG d83023
+		attr_b->nres.valid_size = cpu_to_le64(valid_size);
Kmods SIG d83023
+		mi_b->dirty = true;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	return err;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/* Collapse range in file */
Kmods SIG d83023
+int attr_collapse_range(struct ntfs_inode *ni, u64 vbo, u64 bytes)
Kmods SIG d83023
+{
Kmods SIG d83023
+	int err = 0;
Kmods SIG d83023
+	struct runs_tree *run = &ni->file.run;
Kmods SIG d83023
+	struct ntfs_sb_info *sbi = ni->mi.sbi;
Kmods SIG d83023
+	struct ATTRIB *attr = NULL, *attr_b;
Kmods SIG d83023
+	struct ATTR_LIST_ENTRY *le, *le_b;
Kmods SIG d83023
+	struct mft_inode *mi, *mi_b;
Kmods SIG d83023
+	CLST svcn, evcn1, len, dealloc, alen;
Kmods SIG d83023
+	CLST vcn, end;
Kmods SIG d83023
+	u64 valid_size, data_size, alloc_size, total_size;
Kmods SIG d83023
+	u32 mask;
Kmods SIG d83023
+	__le16 a_flags;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!bytes)
Kmods SIG d83023
+		return 0;
Kmods SIG d83023
+
Kmods SIG d83023
+	le_b = NULL;
Kmods SIG d83023
+	attr_b = ni_find_attr(ni, NULL, &le_b, ATTR_DATA, NULL, 0, NULL, &mi_b);
Kmods SIG d83023
+	if (!attr_b)
Kmods SIG d83023
+		return -ENOENT;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!attr_b->non_res) {
Kmods SIG d83023
+		/* Attribute is resident. Nothing to do? */
Kmods SIG d83023
+		return 0;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	data_size = le64_to_cpu(attr_b->nres.data_size);
Kmods SIG d83023
+	alloc_size = le64_to_cpu(attr_b->nres.alloc_size);
Kmods SIG d83023
+	a_flags = attr_b->flags;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (is_attr_ext(attr_b)) {
Kmods SIG d83023
+		total_size = le64_to_cpu(attr_b->nres.total_size);
Kmods SIG d83023
+		mask = (sbi->cluster_size << attr_b->nres.c_unit) - 1;
Kmods SIG d83023
+	} else {
Kmods SIG d83023
+		total_size = alloc_size;
Kmods SIG d83023
+		mask = sbi->cluster_mask;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	if ((vbo & mask) || (bytes & mask)) {
Kmods SIG d83023
+		/* allow to collapse only cluster aligned ranges */
Kmods SIG d83023
+		return -EINVAL;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	if (vbo > data_size)
Kmods SIG d83023
+		return -EINVAL;
Kmods SIG d83023
+
Kmods SIG d83023
+	down_write(&ni->file.run_lock);
Kmods SIG d83023
+
Kmods SIG d83023
+	if (vbo + bytes >= data_size) {
Kmods SIG d83023
+		u64 new_valid = min(ni->i_valid, vbo);
Kmods SIG d83023
+
Kmods SIG d83023
+		/* Simple truncate file at 'vbo' */
Kmods SIG d83023
+		truncate_setsize(&ni->vfs_inode, vbo);
Kmods SIG d83023
+		err = attr_set_size(ni, ATTR_DATA, NULL, 0, &ni->file.run, vbo,
Kmods SIG d83023
+				    &new_valid, true, NULL);
Kmods SIG d83023
+
Kmods SIG d83023
+		if (!err && new_valid < ni->i_valid)
Kmods SIG d83023
+			ni->i_valid = new_valid;
Kmods SIG d83023
+
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	/*
Kmods SIG d83023
+	 * Enumerate all attribute segments and collapse
Kmods SIG d83023
+	 */
Kmods SIG d83023
+	alen = alloc_size >> sbi->cluster_bits;
Kmods SIG d83023
+	vcn = vbo >> sbi->cluster_bits;
Kmods SIG d83023
+	len = bytes >> sbi->cluster_bits;
Kmods SIG d83023
+	end = vcn + len;
Kmods SIG d83023
+	dealloc = 0;
Kmods SIG d83023
+
Kmods SIG d83023
+	svcn = le64_to_cpu(attr_b->nres.svcn);
Kmods SIG d83023
+	evcn1 = le64_to_cpu(attr_b->nres.evcn) + 1;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (svcn <= vcn && vcn < evcn1) {
Kmods SIG d83023
+		attr = attr_b;
Kmods SIG d83023
+		le = le_b;
Kmods SIG d83023
+		mi = mi_b;
Kmods SIG d83023
+	} else if (!le_b) {
Kmods SIG d83023
+		err = -EINVAL;
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	} else {
Kmods SIG d83023
+		le = le_b;
Kmods SIG d83023
+		attr = ni_find_attr(ni, attr_b, &le, ATTR_DATA, NULL, 0, &vcn,
Kmods SIG d83023
+				    &mi);
Kmods SIG d83023
+		if (!attr) {
Kmods SIG d83023
+			err = -EINVAL;
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		svcn = le64_to_cpu(attr->nres.svcn);
Kmods SIG d83023
+		evcn1 = le64_to_cpu(attr->nres.evcn) + 1;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	for (;;) {
Kmods SIG d83023
+		if (svcn >= end) {
Kmods SIG d83023
+			/* shift vcn */
Kmods SIG d83023
+			attr->nres.svcn = cpu_to_le64(svcn - len);
Kmods SIG d83023
+			attr->nres.evcn = cpu_to_le64(evcn1 - 1 - len);
Kmods SIG d83023
+			if (le) {
Kmods SIG d83023
+				le->vcn = attr->nres.svcn;
Kmods SIG d83023
+				ni->attr_list.dirty = true;
Kmods SIG d83023
+			}
Kmods SIG d83023
+			mi->dirty = true;
Kmods SIG d83023
+		} else if (svcn < vcn || end < evcn1) {
Kmods SIG d83023
+			CLST vcn1, eat, next_svcn;
Kmods SIG d83023
+
Kmods SIG d83023
+			/* collapse a part of this attribute segment */
Kmods SIG d83023
+			err = attr_load_runs(attr, ni, run, &svcn);
Kmods SIG d83023
+			if (err)
Kmods SIG d83023
+				goto out;
Kmods SIG d83023
+			vcn1 = max(vcn, svcn);
Kmods SIG d83023
+			eat = min(end, evcn1) - vcn1;
Kmods SIG d83023
+
Kmods SIG d83023
+			err = run_deallocate_ex(sbi, run, vcn1, eat, &dealloc,
Kmods SIG d83023
+						true);
Kmods SIG d83023
+			if (err)
Kmods SIG d83023
+				goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+			if (!run_collapse_range(run, vcn1, eat)) {
Kmods SIG d83023
+				err = -ENOMEM;
Kmods SIG d83023
+				goto out;
Kmods SIG d83023
+			}
Kmods SIG d83023
+
Kmods SIG d83023
+			if (svcn >= vcn) {
Kmods SIG d83023
+				/* shift vcn */
Kmods SIG d83023
+				attr->nres.svcn = cpu_to_le64(vcn);
Kmods SIG d83023
+				if (le) {
Kmods SIG d83023
+					le->vcn = attr->nres.svcn;
Kmods SIG d83023
+					ni->attr_list.dirty = true;
Kmods SIG d83023
+				}
Kmods SIG d83023
+			}
Kmods SIG d83023
+
Kmods SIG d83023
+			err = mi_pack_runs(mi, attr, run, evcn1 - svcn - eat);
Kmods SIG d83023
+			if (err)
Kmods SIG d83023
+				goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+			next_svcn = le64_to_cpu(attr->nres.evcn) + 1;
Kmods SIG d83023
+			if (next_svcn + eat < evcn1) {
Kmods SIG d83023
+				err = ni_insert_nonresident(
Kmods SIG d83023
+					ni, ATTR_DATA, NULL, 0, run, next_svcn,
Kmods SIG d83023
+					evcn1 - eat - next_svcn, a_flags, &attr,
Kmods SIG d83023
+					&mi);
Kmods SIG d83023
+				if (err)
Kmods SIG d83023
+					goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+				/* layout of records maybe changed */
Kmods SIG d83023
+				attr_b = NULL;
Kmods SIG d83023
+				le = al_find_ex(ni, NULL, ATTR_DATA, NULL, 0,
Kmods SIG d83023
+						&next_svcn);
Kmods SIG d83023
+				if (!le) {
Kmods SIG d83023
+					err = -EINVAL;
Kmods SIG d83023
+					goto out;
Kmods SIG d83023
+				}
Kmods SIG d83023
+			}
Kmods SIG d83023
+
Kmods SIG d83023
+			/* free all allocated memory */
Kmods SIG d83023
+			run_truncate(run, 0);
Kmods SIG d83023
+		} else {
Kmods SIG d83023
+			u16 le_sz;
Kmods SIG d83023
+			u16 roff = le16_to_cpu(attr->nres.run_off);
Kmods SIG d83023
+
Kmods SIG d83023
+			/*run==1 means unpack and deallocate*/
Kmods SIG d83023
+			run_unpack_ex(RUN_DEALLOCATE, sbi, ni->mi.rno, svcn,
Kmods SIG d83023
+				      evcn1 - 1, svcn, Add2Ptr(attr, roff),
Kmods SIG d83023
+				      le32_to_cpu(attr->size) - roff);
Kmods SIG d83023
+
Kmods SIG d83023
+			/* delete this attribute segment */
Kmods SIG d83023
+			mi_remove_attr(mi, attr);
Kmods SIG d83023
+			if (!le)
Kmods SIG d83023
+				break;
Kmods SIG d83023
+
Kmods SIG d83023
+			le_sz = le16_to_cpu(le->size);
Kmods SIG d83023
+			if (!al_remove_le(ni, le)) {
Kmods SIG d83023
+				err = -EINVAL;
Kmods SIG d83023
+				goto out;
Kmods SIG d83023
+			}
Kmods SIG d83023
+
Kmods SIG d83023
+			if (evcn1 >= alen)
Kmods SIG d83023
+				break;
Kmods SIG d83023
+
Kmods SIG d83023
+			if (!svcn) {
Kmods SIG d83023
+				/* Load next record that contains this attribute */
Kmods SIG d83023
+				if (ni_load_mi(ni, le, &mi)) {
Kmods SIG d83023
+					err = -EINVAL;
Kmods SIG d83023
+					goto out;
Kmods SIG d83023
+				}
Kmods SIG d83023
+
Kmods SIG d83023
+				/* Look for required attribute */
Kmods SIG d83023
+				attr = mi_find_attr(mi, NULL, ATTR_DATA, NULL,
Kmods SIG d83023
+						    0, &le->id);
Kmods SIG d83023
+				if (!attr) {
Kmods SIG d83023
+					err = -EINVAL;
Kmods SIG d83023
+					goto out;
Kmods SIG d83023
+				}
Kmods SIG d83023
+				goto next_attr;
Kmods SIG d83023
+			}
Kmods SIG d83023
+			le = (struct ATTR_LIST_ENTRY *)((u8 *)le - le_sz);
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		if (evcn1 >= alen)
Kmods SIG d83023
+			break;
Kmods SIG d83023
+
Kmods SIG d83023
+		attr = ni_enum_attr_ex(ni, attr, &le, &mi);
Kmods SIG d83023
+		if (!attr) {
Kmods SIG d83023
+			err = -EINVAL;
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+next_attr:
Kmods SIG d83023
+		svcn = le64_to_cpu(attr->nres.svcn);
Kmods SIG d83023
+		evcn1 = le64_to_cpu(attr->nres.evcn) + 1;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!attr_b) {
Kmods SIG d83023
+		le_b = NULL;
Kmods SIG d83023
+		attr_b = ni_find_attr(ni, NULL, &le_b, ATTR_DATA, NULL, 0, NULL,
Kmods SIG d83023
+				      &mi_b);
Kmods SIG d83023
+		if (!attr_b) {
Kmods SIG d83023
+			err = -ENOENT;
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	data_size -= bytes;
Kmods SIG d83023
+	valid_size = ni->i_valid;
Kmods SIG d83023
+	if (vbo + bytes <= valid_size)
Kmods SIG d83023
+		valid_size -= bytes;
Kmods SIG d83023
+	else if (vbo < valid_size)
Kmods SIG d83023
+		valid_size = vbo;
Kmods SIG d83023
+
Kmods SIG d83023
+	attr_b->nres.alloc_size = cpu_to_le64(alloc_size - bytes);
Kmods SIG d83023
+	attr_b->nres.data_size = cpu_to_le64(data_size);
Kmods SIG d83023
+	attr_b->nres.valid_size = cpu_to_le64(min(valid_size, data_size));
Kmods SIG d83023
+	total_size -= (u64)dealloc << sbi->cluster_bits;
Kmods SIG d83023
+	if (is_attr_ext(attr_b))
Kmods SIG d83023
+		attr_b->nres.total_size = cpu_to_le64(total_size);
Kmods SIG d83023
+	mi_b->dirty = true;
Kmods SIG d83023
+
Kmods SIG d83023
+	/*update inode size*/
Kmods SIG d83023
+	ni->i_valid = valid_size;
Kmods SIG d83023
+	ni->vfs_inode.i_size = data_size;
Kmods SIG d83023
+	inode_set_bytes(&ni->vfs_inode, total_size);
Kmods SIG d83023
+	ni->ni_flags |= NI_FLAG_UPDATE_PARENT;
Kmods SIG d83023
+	mark_inode_dirty(&ni->vfs_inode);
Kmods SIG d83023
+
Kmods SIG d83023
+out:
Kmods SIG d83023
+	up_write(&ni->file.run_lock);
Kmods SIG d83023
+	if (err)
Kmods SIG d83023
+		make_bad_inode(&ni->vfs_inode);
Kmods SIG d83023
+
Kmods SIG d83023
+	return err;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/* not for normal files */
Kmods SIG d83023
+int attr_punch_hole(struct ntfs_inode *ni, u64 vbo, u64 bytes, u32 *frame_size)
Kmods SIG d83023
+{
Kmods SIG d83023
+	int err = 0;
Kmods SIG d83023
+	struct runs_tree *run = &ni->file.run;
Kmods SIG d83023
+	struct ntfs_sb_info *sbi = ni->mi.sbi;
Kmods SIG d83023
+	struct ATTRIB *attr = NULL, *attr_b;
Kmods SIG d83023
+	struct ATTR_LIST_ENTRY *le, *le_b;
Kmods SIG d83023
+	struct mft_inode *mi, *mi_b;
Kmods SIG d83023
+	CLST svcn, evcn1, vcn, len, end, alen, dealloc;
Kmods SIG d83023
+	u64 total_size, alloc_size;
Kmods SIG d83023
+	u32 mask;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!bytes)
Kmods SIG d83023
+		return 0;
Kmods SIG d83023
+
Kmods SIG d83023
+	le_b = NULL;
Kmods SIG d83023
+	attr_b = ni_find_attr(ni, NULL, &le_b, ATTR_DATA, NULL, 0, NULL, &mi_b);
Kmods SIG d83023
+	if (!attr_b)
Kmods SIG d83023
+		return -ENOENT;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!attr_b->non_res) {
Kmods SIG d83023
+		u32 data_size = le32_to_cpu(attr->res.data_size);
Kmods SIG d83023
+		u32 from, to;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (vbo > data_size)
Kmods SIG d83023
+			return 0;
Kmods SIG d83023
+
Kmods SIG d83023
+		from = vbo;
Kmods SIG d83023
+		to = (vbo + bytes) < data_size ? (vbo + bytes) : data_size;
Kmods SIG d83023
+		memset(Add2Ptr(resident_data(attr_b), from), 0, to - from);
Kmods SIG d83023
+		return 0;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!is_attr_ext(attr_b))
Kmods SIG d83023
+		return -EOPNOTSUPP;
Kmods SIG d83023
+
Kmods SIG d83023
+	alloc_size = le64_to_cpu(attr_b->nres.alloc_size);
Kmods SIG d83023
+	total_size = le64_to_cpu(attr_b->nres.total_size);
Kmods SIG d83023
+
Kmods SIG d83023
+	if (vbo >= alloc_size) {
Kmods SIG d83023
+		// NOTE: it is allowed
Kmods SIG d83023
+		return 0;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	mask = (sbi->cluster_size << attr_b->nres.c_unit) - 1;
Kmods SIG d83023
+
Kmods SIG d83023
+	bytes += vbo;
Kmods SIG d83023
+	if (bytes > alloc_size)
Kmods SIG d83023
+		bytes = alloc_size;
Kmods SIG d83023
+	bytes -= vbo;
Kmods SIG d83023
+
Kmods SIG d83023
+	if ((vbo & mask) || (bytes & mask)) {
Kmods SIG d83023
+		/* We have to zero a range(s)*/
Kmods SIG d83023
+		if (frame_size == NULL) {
Kmods SIG d83023
+			/* Caller insists range is aligned */
Kmods SIG d83023
+			return -EINVAL;
Kmods SIG d83023
+		}
Kmods SIG d83023
+		*frame_size = mask + 1;
Kmods SIG d83023
+		return E_NTFS_NOTALIGNED;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	down_write(&ni->file.run_lock);
Kmods SIG d83023
+	/*
Kmods SIG d83023
+	 * Enumerate all attribute segments and punch hole where necessary
Kmods SIG d83023
+	 */
Kmods SIG d83023
+	alen = alloc_size >> sbi->cluster_bits;
Kmods SIG d83023
+	vcn = vbo >> sbi->cluster_bits;
Kmods SIG d83023
+	len = bytes >> sbi->cluster_bits;
Kmods SIG d83023
+	end = vcn + len;
Kmods SIG d83023
+	dealloc = 0;
Kmods SIG d83023
+
Kmods SIG d83023
+	svcn = le64_to_cpu(attr_b->nres.svcn);
Kmods SIG d83023
+	evcn1 = le64_to_cpu(attr_b->nres.evcn) + 1;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (svcn <= vcn && vcn < evcn1) {
Kmods SIG d83023
+		attr = attr_b;
Kmods SIG d83023
+		le = le_b;
Kmods SIG d83023
+		mi = mi_b;
Kmods SIG d83023
+	} else if (!le_b) {
Kmods SIG d83023
+		err = -EINVAL;
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	} else {
Kmods SIG d83023
+		le = le_b;
Kmods SIG d83023
+		attr = ni_find_attr(ni, attr_b, &le, ATTR_DATA, NULL, 0, &vcn,
Kmods SIG d83023
+				    &mi);
Kmods SIG d83023
+		if (!attr) {
Kmods SIG d83023
+			err = -EINVAL;
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		svcn = le64_to_cpu(attr->nres.svcn);
Kmods SIG d83023
+		evcn1 = le64_to_cpu(attr->nres.evcn) + 1;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	while (svcn < end) {
Kmods SIG d83023
+		CLST vcn1, zero, dealloc2;
Kmods SIG d83023
+
Kmods SIG d83023
+		err = attr_load_runs(attr, ni, run, &svcn);
Kmods SIG d83023
+		if (err)
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		vcn1 = max(vcn, svcn);
Kmods SIG d83023
+		zero = min(end, evcn1) - vcn1;
Kmods SIG d83023
+
Kmods SIG d83023
+		dealloc2 = dealloc;
Kmods SIG d83023
+		err = run_deallocate_ex(sbi, run, vcn1, zero, &dealloc, true);
Kmods SIG d83023
+		if (err)
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (dealloc2 == dealloc) {
Kmods SIG d83023
+			/* looks like  the required range is already sparsed */
Kmods SIG d83023
+		} else {
Kmods SIG d83023
+			if (!run_add_entry(run, vcn1, SPARSE_LCN, zero,
Kmods SIG d83023
+					   false)) {
Kmods SIG d83023
+				err = -ENOMEM;
Kmods SIG d83023
+				goto out;
Kmods SIG d83023
+			}
Kmods SIG d83023
+
Kmods SIG d83023
+			err = mi_pack_runs(mi, attr, run, evcn1 - svcn);
Kmods SIG d83023
+			if (err)
Kmods SIG d83023
+				goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+		/* free all allocated memory */
Kmods SIG d83023
+		run_truncate(run, 0);
Kmods SIG d83023
+
Kmods SIG d83023
+		if (evcn1 >= alen)
Kmods SIG d83023
+			break;
Kmods SIG d83023
+
Kmods SIG d83023
+		attr = ni_enum_attr_ex(ni, attr, &le, &mi);
Kmods SIG d83023
+		if (!attr) {
Kmods SIG d83023
+			err = -EINVAL;
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		svcn = le64_to_cpu(attr->nres.svcn);
Kmods SIG d83023
+		evcn1 = le64_to_cpu(attr->nres.evcn) + 1;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	total_size -= (u64)dealloc << sbi->cluster_bits;
Kmods SIG d83023
+	attr_b->nres.total_size = cpu_to_le64(total_size);
Kmods SIG d83023
+	mi_b->dirty = true;
Kmods SIG d83023
+
Kmods SIG d83023
+	/*update inode size*/
Kmods SIG d83023
+	inode_set_bytes(&ni->vfs_inode, total_size);
Kmods SIG d83023
+	ni->ni_flags |= NI_FLAG_UPDATE_PARENT;
Kmods SIG d83023
+	mark_inode_dirty(&ni->vfs_inode);
Kmods SIG d83023
+
Kmods SIG d83023
+out:
Kmods SIG d83023
+	up_write(&ni->file.run_lock);
Kmods SIG d83023
+	if (err)
Kmods SIG d83023
+		make_bad_inode(&ni->vfs_inode);
Kmods SIG d83023
+
Kmods SIG d83023
+	return err;
Kmods SIG d83023
+}
Kmods SIG d83023
diff --git a/src/attrlist.c b/src/attrlist.c
Kmods SIG d83023
new file mode 100644
Kmods SIG d83023
index 0000000000000000000000000000000000000000..ea561361b576d5a8a1ec30c0017a5ac76f1a61df
Kmods SIG d83023
--- /dev/null
Kmods SIG d83023
+++ b/src/attrlist.c
Kmods SIG d83023
@@ -0,0 +1,456 @@
Kmods SIG d83023
+// SPDX-License-Identifier: GPL-2.0
Kmods SIG d83023
+/*
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * Copyright (C) 2019-2021 Paragon Software GmbH, All rights reserved.
Kmods SIG d83023
+ *
Kmods SIG d83023
+ */
Kmods SIG d83023
+
Kmods SIG d83023
+#include <linux/blkdev.h>
Kmods SIG d83023
+#include <linux/buffer_head.h>
Kmods SIG d83023
+#include <linux/fs.h>
Kmods SIG d83023
+#include <linux/nls.h>
Kmods SIG d83023
+
Kmods SIG d83023
+#include "debug.h"
Kmods SIG d83023
+#include "ntfs.h"
Kmods SIG d83023
+#include "ntfs_fs.h"
Kmods SIG d83023
+
Kmods SIG d83023
+/* Returns true if le is valid */
Kmods SIG d83023
+static inline bool al_is_valid_le(const struct ntfs_inode *ni,
Kmods SIG d83023
+				  struct ATTR_LIST_ENTRY *le)
Kmods SIG d83023
+{
Kmods SIG d83023
+	if (!le || !ni->attr_list.le || !ni->attr_list.size)
Kmods SIG d83023
+		return false;
Kmods SIG d83023
+
Kmods SIG d83023
+	return PtrOffset(ni->attr_list.le, le) + le16_to_cpu(le->size) <=
Kmods SIG d83023
+	       ni->attr_list.size;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+void al_destroy(struct ntfs_inode *ni)
Kmods SIG d83023
+{
Kmods SIG d83023
+	run_close(&ni->attr_list.run);
Kmods SIG d83023
+	ntfs_free(ni->attr_list.le);
Kmods SIG d83023
+	ni->attr_list.le = NULL;
Kmods SIG d83023
+	ni->attr_list.size = 0;
Kmods SIG d83023
+	ni->attr_list.dirty = false;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * ntfs_load_attr_list
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * This method makes sure that the ATTRIB list, if present,
Kmods SIG d83023
+ * has been properly set up.
Kmods SIG d83023
+ */
Kmods SIG d83023
+int ntfs_load_attr_list(struct ntfs_inode *ni, struct ATTRIB *attr)
Kmods SIG d83023
+{
Kmods SIG d83023
+	int err;
Kmods SIG d83023
+	size_t lsize;
Kmods SIG d83023
+	void *le = NULL;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (ni->attr_list.size)
Kmods SIG d83023
+		return 0;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!attr->non_res) {
Kmods SIG d83023
+		lsize = le32_to_cpu(attr->res.data_size);
Kmods SIG d83023
+		le = ntfs_malloc(al_aligned(lsize));
Kmods SIG d83023
+		if (!le) {
Kmods SIG d83023
+			err = -ENOMEM;
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+		memcpy(le, resident_data(attr), lsize);
Kmods SIG d83023
+	} else if (attr->nres.svcn) {
Kmods SIG d83023
+		err = -EINVAL;
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	} else {
Kmods SIG d83023
+		u16 run_off = le16_to_cpu(attr->nres.run_off);
Kmods SIG d83023
+
Kmods SIG d83023
+		lsize = le64_to_cpu(attr->nres.data_size);
Kmods SIG d83023
+
Kmods SIG d83023
+		run_init(&ni->attr_list.run);
Kmods SIG d83023
+
Kmods SIG d83023
+		err = run_unpack_ex(&ni->attr_list.run, ni->mi.sbi, ni->mi.rno,
Kmods SIG d83023
+				    0, le64_to_cpu(attr->nres.evcn), 0,
Kmods SIG d83023
+				    Add2Ptr(attr, run_off),
Kmods SIG d83023
+				    le32_to_cpu(attr->size) - run_off);
Kmods SIG d83023
+		if (err < 0)
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+		le = ntfs_malloc(al_aligned(lsize));
Kmods SIG d83023
+		if (!le) {
Kmods SIG d83023
+			err = -ENOMEM;
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		err = ntfs_read_run_nb(ni->mi.sbi, &ni->attr_list.run, 0, le,
Kmods SIG d83023
+				       lsize, NULL);
Kmods SIG d83023
+		if (err)
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	ni->attr_list.size = lsize;
Kmods SIG d83023
+	ni->attr_list.le = le;
Kmods SIG d83023
+
Kmods SIG d83023
+	return 0;
Kmods SIG d83023
+
Kmods SIG d83023
+out:
Kmods SIG d83023
+	ni->attr_list.le = le;
Kmods SIG d83023
+	al_destroy(ni);
Kmods SIG d83023
+
Kmods SIG d83023
+	return err;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * al_enumerate
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * Returns the next list 'le'
Kmods SIG d83023
+ * if 'le' is NULL then returns the first 'le'
Kmods SIG d83023
+ */
Kmods SIG d83023
+struct ATTR_LIST_ENTRY *al_enumerate(struct ntfs_inode *ni,
Kmods SIG d83023
+				     struct ATTR_LIST_ENTRY *le)
Kmods SIG d83023
+{
Kmods SIG d83023
+	size_t off;
Kmods SIG d83023
+	u16 sz;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!le) {
Kmods SIG d83023
+		le = ni->attr_list.le;
Kmods SIG d83023
+	} else {
Kmods SIG d83023
+		sz = le16_to_cpu(le->size);
Kmods SIG d83023
+		if (sz < sizeof(struct ATTR_LIST_ENTRY)) {
Kmods SIG d83023
+			/* Impossible 'cause we should not return such 'le' */
Kmods SIG d83023
+			return NULL;
Kmods SIG d83023
+		}
Kmods SIG d83023
+		le = Add2Ptr(le, sz);
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	/* Check boundary */
Kmods SIG d83023
+	off = PtrOffset(ni->attr_list.le, le);
Kmods SIG d83023
+	if (off + sizeof(struct ATTR_LIST_ENTRY) > ni->attr_list.size) {
Kmods SIG d83023
+		// The regular end of list
Kmods SIG d83023
+		return NULL;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	sz = le16_to_cpu(le->size);
Kmods SIG d83023
+
Kmods SIG d83023
+	/* Check 'le' for errors */
Kmods SIG d83023
+	if (sz < sizeof(struct ATTR_LIST_ENTRY) ||
Kmods SIG d83023
+	    off + sz > ni->attr_list.size ||
Kmods SIG d83023
+	    sz < le->name_off + le->name_len * sizeof(short)) {
Kmods SIG d83023
+		return NULL;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	return le;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * al_find_le
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * finds the first 'le' in the list which matches type, name and vcn
Kmods SIG d83023
+ * Returns NULL if not found
Kmods SIG d83023
+ */
Kmods SIG d83023
+struct ATTR_LIST_ENTRY *al_find_le(struct ntfs_inode *ni,
Kmods SIG d83023
+				   struct ATTR_LIST_ENTRY *le,
Kmods SIG d83023
+				   const struct ATTRIB *attr)
Kmods SIG d83023
+{
Kmods SIG d83023
+	CLST svcn = attr_svcn(attr);
Kmods SIG d83023
+
Kmods SIG d83023
+	return al_find_ex(ni, le, attr->type, attr_name(attr), attr->name_len,
Kmods SIG d83023
+			  &svcn);
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * al_find_ex
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * finds the first 'le' in the list which matches type, name and vcn
Kmods SIG d83023
+ * Returns NULL if not found
Kmods SIG d83023
+ */
Kmods SIG d83023
+struct ATTR_LIST_ENTRY *al_find_ex(struct ntfs_inode *ni,
Kmods SIG d83023
+				   struct ATTR_LIST_ENTRY *le,
Kmods SIG d83023
+				   enum ATTR_TYPE type, const __le16 *name,
Kmods SIG d83023
+				   u8 name_len, const CLST *vcn)
Kmods SIG d83023
+{
Kmods SIG d83023
+	struct ATTR_LIST_ENTRY *ret = NULL;
Kmods SIG d83023
+	u32 type_in = le32_to_cpu(type);
Kmods SIG d83023
+
Kmods SIG d83023
+	while ((le = al_enumerate(ni, le))) {
Kmods SIG d83023
+		u64 le_vcn;
Kmods SIG d83023
+		int diff = le32_to_cpu(le->type) - type_in;
Kmods SIG d83023
+
Kmods SIG d83023
+		/* List entries are sorted by type, name and vcn */
Kmods SIG d83023
+		if (diff < 0)
Kmods SIG d83023
+			continue;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (diff > 0)
Kmods SIG d83023
+			return ret;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (le->name_len != name_len)
Kmods SIG d83023
+			continue;
Kmods SIG d83023
+
Kmods SIG d83023
+		le_vcn = le64_to_cpu(le->vcn);
Kmods SIG d83023
+		if (!le_vcn) {
Kmods SIG d83023
+			/*
Kmods SIG d83023
+			 * compare entry names only for entry with vcn == 0
Kmods SIG d83023
+			 */
Kmods SIG d83023
+			diff = ntfs_cmp_names(le_name(le), name_len, name,
Kmods SIG d83023
+					      name_len, ni->mi.sbi->upcase,
Kmods SIG d83023
+					      true);
Kmods SIG d83023
+			if (diff < 0)
Kmods SIG d83023
+				continue;
Kmods SIG d83023
+
Kmods SIG d83023
+			if (diff > 0)
Kmods SIG d83023
+				return ret;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		if (!vcn)
Kmods SIG d83023
+			return le;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (*vcn == le_vcn)
Kmods SIG d83023
+			return le;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (*vcn < le_vcn)
Kmods SIG d83023
+			return ret;
Kmods SIG d83023
+
Kmods SIG d83023
+		ret = le;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	return ret;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * al_find_le_to_insert
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * finds the first list entry which matches type, name and vcn
Kmods SIG d83023
+ */
Kmods SIG d83023
+static struct ATTR_LIST_ENTRY *al_find_le_to_insert(struct ntfs_inode *ni,
Kmods SIG d83023
+						    enum ATTR_TYPE type,
Kmods SIG d83023
+						    const __le16 *name,
Kmods SIG d83023
+						    u8 name_len, CLST vcn)
Kmods SIG d83023
+{
Kmods SIG d83023
+	struct ATTR_LIST_ENTRY *le = NULL, *prev;
Kmods SIG d83023
+	u32 type_in = le32_to_cpu(type);
Kmods SIG d83023
+
Kmods SIG d83023
+	/* List entries are sorted by type, name, vcn */
Kmods SIG d83023
+	while ((le = al_enumerate(ni, prev = le))) {
Kmods SIG d83023
+		int diff = le32_to_cpu(le->type) - type_in;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (diff < 0)
Kmods SIG d83023
+			continue;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (diff > 0)
Kmods SIG d83023
+			return le;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (!le->vcn) {
Kmods SIG d83023
+			/*
Kmods SIG d83023
+			 * compare entry names only for entry with vcn == 0
Kmods SIG d83023
+			 */
Kmods SIG d83023
+			diff = ntfs_cmp_names(le_name(le), le->name_len, name,
Kmods SIG d83023
+					      name_len, ni->mi.sbi->upcase,
Kmods SIG d83023
+					      true);
Kmods SIG d83023
+			if (diff < 0)
Kmods SIG d83023
+				continue;
Kmods SIG d83023
+
Kmods SIG d83023
+			if (diff > 0)
Kmods SIG d83023
+				return le;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		if (le64_to_cpu(le->vcn) >= vcn)
Kmods SIG d83023
+			return le;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	return prev ? Add2Ptr(prev, le16_to_cpu(prev->size)) : ni->attr_list.le;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * al_add_le
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * adds an "attribute list entry" to the list.
Kmods SIG d83023
+ */
Kmods SIG d83023
+int al_add_le(struct ntfs_inode *ni, enum ATTR_TYPE type, const __le16 *name,
Kmods SIG d83023
+	      u8 name_len, CLST svcn, __le16 id, const struct MFT_REF *ref,
Kmods SIG d83023
+	      struct ATTR_LIST_ENTRY **new_le)
Kmods SIG d83023
+{
Kmods SIG d83023
+	int err;
Kmods SIG d83023
+	struct ATTRIB *attr;
Kmods SIG d83023
+	struct ATTR_LIST_ENTRY *le;
Kmods SIG d83023
+	size_t off;
Kmods SIG d83023
+	u16 sz;
Kmods SIG d83023
+	size_t asize, new_asize;
Kmods SIG d83023
+	u64 new_size;
Kmods SIG d83023
+	typeof(ni->attr_list) *al = &ni->attr_list;
Kmods SIG d83023
+
Kmods SIG d83023
+	/*
Kmods SIG d83023
+	 * Compute the size of the new 'le'
Kmods SIG d83023
+	 */
Kmods SIG d83023
+	sz = le_size(name_len);
Kmods SIG d83023
+	new_size = al->size + sz;
Kmods SIG d83023
+	asize = al_aligned(al->size);
Kmods SIG d83023
+	new_asize = al_aligned(new_size);
Kmods SIG d83023
+
Kmods SIG d83023
+	/* Scan forward to the point at which the new 'le' should be inserted. */
Kmods SIG d83023
+	le = al_find_le_to_insert(ni, type, name, name_len, svcn);
Kmods SIG d83023
+	off = PtrOffset(al->le, le);
Kmods SIG d83023
+
Kmods SIG d83023
+	if (new_size > asize) {
Kmods SIG d83023
+		void *ptr = ntfs_malloc(new_asize);
Kmods SIG d83023
+
Kmods SIG d83023
+		if (!ptr)
Kmods SIG d83023
+			return -ENOMEM;
Kmods SIG d83023
+
Kmods SIG d83023
+		memcpy(ptr, al->le, off);
Kmods SIG d83023
+		memcpy(Add2Ptr(ptr, off + sz), le, al->size - off);
Kmods SIG d83023
+		le = Add2Ptr(ptr, off);
Kmods SIG d83023
+		ntfs_free(al->le);
Kmods SIG d83023
+		al->le = ptr;
Kmods SIG d83023
+	} else {
Kmods SIG d83023
+		memmove(Add2Ptr(le, sz), le, al->size - off);
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	al->size = new_size;
Kmods SIG d83023
+
Kmods SIG d83023
+	le->type = type;
Kmods SIG d83023
+	le->size = cpu_to_le16(sz);
Kmods SIG d83023
+	le->name_len = name_len;
Kmods SIG d83023
+	le->name_off = offsetof(struct ATTR_LIST_ENTRY, name);
Kmods SIG d83023
+	le->vcn = cpu_to_le64(svcn);
Kmods SIG d83023
+	le->ref = *ref;
Kmods SIG d83023
+	le->id = id;
Kmods SIG d83023
+	memcpy(le->name, name, sizeof(short) * name_len);
Kmods SIG d83023
+
Kmods SIG d83023
+	al->dirty = true;
Kmods SIG d83023
+
Kmods SIG d83023
+	err = attr_set_size(ni, ATTR_LIST, NULL, 0, &al->run, new_size,
Kmods SIG d83023
+			    &new_size, true, &attr);
Kmods SIG d83023
+	if (err)
Kmods SIG d83023
+		return err;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (attr && attr->non_res) {
Kmods SIG d83023
+		err = ntfs_sb_write_run(ni->mi.sbi, &al->run, 0, al->le,
Kmods SIG d83023
+					al->size);
Kmods SIG d83023
+		if (err)
Kmods SIG d83023
+			return err;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	al->dirty = false;
Kmods SIG d83023
+	*new_le = le;
Kmods SIG d83023
+
Kmods SIG d83023
+	return 0;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * al_remove_le
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * removes 'le' from attribute list
Kmods SIG d83023
+ */
Kmods SIG d83023
+bool al_remove_le(struct ntfs_inode *ni, struct ATTR_LIST_ENTRY *le)
Kmods SIG d83023
+{
Kmods SIG d83023
+	u16 size;
Kmods SIG d83023
+	size_t off;
Kmods SIG d83023
+	typeof(ni->attr_list) *al = &ni->attr_list;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!al_is_valid_le(ni, le))
Kmods SIG d83023
+		return false;
Kmods SIG d83023
+
Kmods SIG d83023
+	/* Save on stack the size of 'le' */
Kmods SIG d83023
+	size = le16_to_cpu(le->size);
Kmods SIG d83023
+	off = PtrOffset(al->le, le);
Kmods SIG d83023
+
Kmods SIG d83023
+	memmove(le, Add2Ptr(le, size), al->size - (off + size));
Kmods SIG d83023
+
Kmods SIG d83023
+	al->size -= size;
Kmods SIG d83023
+	al->dirty = true;
Kmods SIG d83023
+
Kmods SIG d83023
+	return true;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * al_delete_le
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * deletes from the list the first 'le' which matches its parameters.
Kmods SIG d83023
+ */
Kmods SIG d83023
+bool al_delete_le(struct ntfs_inode *ni, enum ATTR_TYPE type, CLST vcn,
Kmods SIG d83023
+		  const __le16 *name, size_t name_len,
Kmods SIG d83023
+		  const struct MFT_REF *ref)
Kmods SIG d83023
+{
Kmods SIG d83023
+	u16 size;
Kmods SIG d83023
+	struct ATTR_LIST_ENTRY *le;
Kmods SIG d83023
+	size_t off;
Kmods SIG d83023
+	typeof(ni->attr_list) *al = &ni->attr_list;
Kmods SIG d83023
+
Kmods SIG d83023
+	/* Scan forward to the first 'le' that matches the input */
Kmods SIG d83023
+	le = al_find_ex(ni, NULL, type, name, name_len, &vcn;;
Kmods SIG d83023
+	if (!le)
Kmods SIG d83023
+		return false;
Kmods SIG d83023
+
Kmods SIG d83023
+	off = PtrOffset(al->le, le);
Kmods SIG d83023
+
Kmods SIG d83023
+next:
Kmods SIG d83023
+	if (off >= al->size)
Kmods SIG d83023
+		return false;
Kmods SIG d83023
+	if (le->type != type)
Kmods SIG d83023
+		return false;
Kmods SIG d83023
+	if (le->name_len != name_len)
Kmods SIG d83023
+		return false;
Kmods SIG d83023
+	if (name_len && ntfs_cmp_names(le_name(le), name_len, name, name_len,
Kmods SIG d83023
+				       ni->mi.sbi->upcase, true))
Kmods SIG d83023
+		return false;
Kmods SIG d83023
+	if (le64_to_cpu(le->vcn) != vcn)
Kmods SIG d83023
+		return false;
Kmods SIG d83023
+
Kmods SIG d83023
+	/*
Kmods SIG d83023
+	 * The caller specified a segment reference, so we have to
Kmods SIG d83023
+	 * scan through the matching entries until we find that segment
Kmods SIG d83023
+	 * reference or we run of matching entries.
Kmods SIG d83023
+	 */
Kmods SIG d83023
+	if (ref && memcmp(ref, &le->ref, sizeof(*ref))) {
Kmods SIG d83023
+		off += le16_to_cpu(le->size);
Kmods SIG d83023
+		le = Add2Ptr(al->le, off);
Kmods SIG d83023
+		goto next;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	/* Save on stack the size of 'le' */
Kmods SIG d83023
+	size = le16_to_cpu(le->size);
Kmods SIG d83023
+	/* Delete 'le'. */
Kmods SIG d83023
+	memmove(le, Add2Ptr(le, size), al->size - (off + size));
Kmods SIG d83023
+
Kmods SIG d83023
+	al->size -= size;
Kmods SIG d83023
+	al->dirty = true;
Kmods SIG d83023
+
Kmods SIG d83023
+	return true;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * al_update
Kmods SIG d83023
+ */
Kmods SIG d83023
+int al_update(struct ntfs_inode *ni)
Kmods SIG d83023
+{
Kmods SIG d83023
+	int err;
Kmods SIG d83023
+	struct ATTRIB *attr;
Kmods SIG d83023
+	typeof(ni->attr_list) *al = &ni->attr_list;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!al->dirty || !al->size)
Kmods SIG d83023
+		return 0;
Kmods SIG d83023
+
Kmods SIG d83023
+	/*
Kmods SIG d83023
+	 * attribute list increased on demand in al_add_le
Kmods SIG d83023
+	 * attribute list decreased here
Kmods SIG d83023
+	 */
Kmods SIG d83023
+	err = attr_set_size(ni, ATTR_LIST, NULL, 0, &al->run, al->size, NULL,
Kmods SIG d83023
+			    false, &attr);
Kmods SIG d83023
+	if (err)
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!attr->non_res) {
Kmods SIG d83023
+		memcpy(resident_data(attr), al->le, al->size);
Kmods SIG d83023
+	} else {
Kmods SIG d83023
+		err = ntfs_sb_write_run(ni->mi.sbi, &al->run, 0, al->le,
Kmods SIG d83023
+					al->size);
Kmods SIG d83023
+		if (err)
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+		attr->nres.valid_size = attr->nres.data_size;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	ni->mi.dirty = true;
Kmods SIG d83023
+	al->dirty = false;
Kmods SIG d83023
+
Kmods SIG d83023
+out:
Kmods SIG d83023
+	return err;
Kmods SIG d83023
+}
Kmods SIG d83023
diff --git a/src/xattr.c b/src/xattr.c
Kmods SIG d83023
new file mode 100644
Kmods SIG d83023
index 0000000000000000000000000000000000000000..98871c895e77b165bede3fbd1e50b91d5e07b106
Kmods SIG d83023
--- /dev/null
Kmods SIG d83023
+++ b/src/xattr.c
Kmods SIG d83023
@@ -0,0 +1,1128 @@
Kmods SIG d83023
+// SPDX-License-Identifier: GPL-2.0
Kmods SIG d83023
+/*
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * Copyright (C) 2019-2021 Paragon Software GmbH, All rights reserved.
Kmods SIG d83023
+ *
Kmods SIG d83023
+ */
Kmods SIG d83023
+
Kmods SIG d83023
+#include <linux/blkdev.h>
Kmods SIG d83023
+#include <linux/buffer_head.h>
Kmods SIG d83023
+#include <linux/fs.h>
Kmods SIG d83023
+#include <linux/nls.h>
Kmods SIG d83023
+#include <linux/posix_acl.h>
Kmods SIG d83023
+#include <linux/posix_acl_xattr.h>
Kmods SIG d83023
+#include <linux/xattr.h>
Kmods SIG d83023
+
Kmods SIG d83023
+#include "debug.h"
Kmods SIG d83023
+#include "ntfs.h"
Kmods SIG d83023
+#include "ntfs_fs.h"
Kmods SIG d83023
+
Kmods SIG d83023
+// clang-format off
Kmods SIG d83023
+#define SYSTEM_DOS_ATTRIB    "system.dos_attrib"
Kmods SIG d83023
+#define SYSTEM_NTFS_ATTRIB   "system.ntfs_attrib"
Kmods SIG d83023
+#define SYSTEM_NTFS_SECURITY "system.ntfs_security"
Kmods SIG d83023
+// clang-format on
Kmods SIG d83023
+
Kmods SIG d83023
+static inline size_t unpacked_ea_size(const struct EA_FULL *ea)
Kmods SIG d83023
+{
Kmods SIG d83023
+	return ea->size ? le32_to_cpu(ea->size)
Kmods SIG d83023
+			: DwordAlign(struct_size(
Kmods SIG d83023
+				  ea, name,
Kmods SIG d83023
+				  1 + ea->name_len + le16_to_cpu(ea->elength)));
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+static inline size_t packed_ea_size(const struct EA_FULL *ea)
Kmods SIG d83023
+{
Kmods SIG d83023
+	return struct_size(ea, name,
Kmods SIG d83023
+			   1 + ea->name_len + le16_to_cpu(ea->elength)) -
Kmods SIG d83023
+	       offsetof(struct EA_FULL, flags);
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * find_ea
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * assume there is at least one xattr in the list
Kmods SIG d83023
+ */
Kmods SIG d83023
+static inline bool find_ea(const struct EA_FULL *ea_all, u32 bytes,
Kmods SIG d83023
+			   const char *name, u8 name_len, u32 *off)
Kmods SIG d83023
+{
Kmods SIG d83023
+	*off = 0;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!ea_all || !bytes)
Kmods SIG d83023
+		return false;
Kmods SIG d83023
+
Kmods SIG d83023
+	for (;;) {
Kmods SIG d83023
+		const struct EA_FULL *ea = Add2Ptr(ea_all, *off);
Kmods SIG d83023
+		u32 next_off = *off + unpacked_ea_size(ea);
Kmods SIG d83023
+
Kmods SIG d83023
+		if (next_off > bytes)
Kmods SIG d83023
+			return false;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (ea->name_len == name_len &&
Kmods SIG d83023
+		    !memcmp(ea->name, name, name_len))
Kmods SIG d83023
+			return true;
Kmods SIG d83023
+
Kmods SIG d83023
+		*off = next_off;
Kmods SIG d83023
+		if (next_off >= bytes)
Kmods SIG d83023
+			return false;
Kmods SIG d83023
+	}
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * ntfs_read_ea
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * reads all extended attributes
Kmods SIG d83023
+ * ea - new allocated memory
Kmods SIG d83023
+ * info - pointer into resident data
Kmods SIG d83023
+ */
Kmods SIG d83023
+static int ntfs_read_ea(struct ntfs_inode *ni, struct EA_FULL **ea,
Kmods SIG d83023
+			size_t add_bytes, const struct EA_INFO **info)
Kmods SIG d83023
+{
Kmods SIG d83023
+	int err;
Kmods SIG d83023
+	struct ATTR_LIST_ENTRY *le = NULL;
Kmods SIG d83023
+	struct ATTRIB *attr_info, *attr_ea;
Kmods SIG d83023
+	void *ea_p;
Kmods SIG d83023
+	u32 size;
Kmods SIG d83023
+
Kmods SIG d83023
+	static_assert(le32_to_cpu(ATTR_EA_INFO) < le32_to_cpu(ATTR_EA));
Kmods SIG d83023
+
Kmods SIG d83023
+	*ea = NULL;
Kmods SIG d83023
+	*info = NULL;
Kmods SIG d83023
+
Kmods SIG d83023
+	attr_info =
Kmods SIG d83023
+		ni_find_attr(ni, NULL, &le, ATTR_EA_INFO, NULL, 0, NULL, NULL);
Kmods SIG d83023
+	attr_ea =
Kmods SIG d83023
+		ni_find_attr(ni, attr_info, &le, ATTR_EA, NULL, 0, NULL, NULL);
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!attr_ea || !attr_info)
Kmods SIG d83023
+		return 0;
Kmods SIG d83023
+
Kmods SIG d83023
+	*info = resident_data_ex(attr_info, sizeof(struct EA_INFO));
Kmods SIG d83023
+	if (!*info)
Kmods SIG d83023
+		return -EINVAL;
Kmods SIG d83023
+
Kmods SIG d83023
+	/* Check Ea limit */
Kmods SIG d83023
+	size = le32_to_cpu((*info)->size);
Kmods SIG d83023
+	if (size > ni->mi.sbi->ea_max_size)
Kmods SIG d83023
+		return -EFBIG;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (attr_size(attr_ea) > ni->mi.sbi->ea_max_size)
Kmods SIG d83023
+		return -EFBIG;
Kmods SIG d83023
+
Kmods SIG d83023
+	/* Allocate memory for packed Ea */
Kmods SIG d83023
+	ea_p = ntfs_malloc(size + add_bytes);
Kmods SIG d83023
+	if (!ea_p)
Kmods SIG d83023
+		return -ENOMEM;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (attr_ea->non_res) {
Kmods SIG d83023
+		struct runs_tree run;
Kmods SIG d83023
+
Kmods SIG d83023
+		run_init(&run;;
Kmods SIG d83023
+
Kmods SIG d83023
+		err = attr_load_runs(attr_ea, ni, &run, NULL);
Kmods SIG d83023
+		if (!err)
Kmods SIG d83023
+			err = ntfs_read_run_nb(ni->mi.sbi, &run, 0, ea_p, size,
Kmods SIG d83023
+					       NULL);
Kmods SIG d83023
+		run_close(&run;;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (err)
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+	} else {
Kmods SIG d83023
+		void *p = resident_data_ex(attr_ea, size);
Kmods SIG d83023
+
Kmods SIG d83023
+		if (!p) {
Kmods SIG d83023
+			err = -EINVAL;
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+		memcpy(ea_p, p, size);
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	memset(Add2Ptr(ea_p, size), 0, add_bytes);
Kmods SIG d83023
+	*ea = ea_p;
Kmods SIG d83023
+	return 0;
Kmods SIG d83023
+
Kmods SIG d83023
+out:
Kmods SIG d83023
+	ntfs_free(ea_p);
Kmods SIG d83023
+	*ea = NULL;
Kmods SIG d83023
+	return err;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * ntfs_list_ea
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * copy a list of xattrs names into the buffer
Kmods SIG d83023
+ * provided, or compute the buffer size required
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * Returns a negative error number on failure, or the number of bytes
Kmods SIG d83023
+ * used / required on success.
Kmods SIG d83023
+ */
Kmods SIG d83023
+static ssize_t ntfs_list_ea(struct ntfs_inode *ni, char *buffer,
Kmods SIG d83023
+			    size_t bytes_per_buffer)
Kmods SIG d83023
+{
Kmods SIG d83023
+	const struct EA_INFO *info;
Kmods SIG d83023
+	struct EA_FULL *ea_all = NULL;
Kmods SIG d83023
+	const struct EA_FULL *ea;
Kmods SIG d83023
+	u32 off, size;
Kmods SIG d83023
+	int err;
Kmods SIG d83023
+	size_t ret;
Kmods SIG d83023
+
Kmods SIG d83023
+	err = ntfs_read_ea(ni, &ea_all, 0, &info;;
Kmods SIG d83023
+	if (err)
Kmods SIG d83023
+		return err;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!info || !ea_all)
Kmods SIG d83023
+		return 0;
Kmods SIG d83023
+
Kmods SIG d83023
+	size = le32_to_cpu(info->size);
Kmods SIG d83023
+
Kmods SIG d83023
+	/* Enumerate all xattrs */
Kmods SIG d83023
+	for (ret = 0, off = 0; off < size; off += unpacked_ea_size(ea)) {
Kmods SIG d83023
+		ea = Add2Ptr(ea_all, off);
Kmods SIG d83023
+
Kmods SIG d83023
+		if (buffer) {
Kmods SIG d83023
+			if (ret + ea->name_len + 1 > bytes_per_buffer) {
Kmods SIG d83023
+				err = -ERANGE;
Kmods SIG d83023
+				goto out;
Kmods SIG d83023
+			}
Kmods SIG d83023
+
Kmods SIG d83023
+			memcpy(buffer + ret, ea->name, ea->name_len);
Kmods SIG d83023
+			buffer[ret + ea->name_len] = 0;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		ret += ea->name_len + 1;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+out:
Kmods SIG d83023
+	ntfs_free(ea_all);
Kmods SIG d83023
+	return err ? err : ret;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+static int ntfs_get_ea(struct inode *inode, const char *name, size_t name_len,
Kmods SIG d83023
+		       void *buffer, size_t size, size_t *required)
Kmods SIG d83023
+{
Kmods SIG d83023
+	struct ntfs_inode *ni = ntfs_i(inode);
Kmods SIG d83023
+	const struct EA_INFO *info;
Kmods SIG d83023
+	struct EA_FULL *ea_all = NULL;
Kmods SIG d83023
+	const struct EA_FULL *ea;
Kmods SIG d83023
+	u32 off, len;
Kmods SIG d83023
+	int err;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!(ni->ni_flags & NI_FLAG_EA))
Kmods SIG d83023
+		return -ENODATA;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!required)
Kmods SIG d83023
+		ni_lock(ni);
Kmods SIG d83023
+
Kmods SIG d83023
+	len = 0;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (name_len > 255) {
Kmods SIG d83023
+		err = -ENAMETOOLONG;
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	err = ntfs_read_ea(ni, &ea_all, 0, &info;;
Kmods SIG d83023
+	if (err)
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!info)
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+	/* Enumerate all xattrs */
Kmods SIG d83023
+	if (!find_ea(ea_all, le32_to_cpu(info->size), name, name_len, &off)) {
Kmods SIG d83023
+		err = -ENODATA;
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+	ea = Add2Ptr(ea_all, off);
Kmods SIG d83023
+
Kmods SIG d83023
+	len = le16_to_cpu(ea->elength);
Kmods SIG d83023
+	if (!buffer) {
Kmods SIG d83023
+		err = 0;
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	if (len > size) {
Kmods SIG d83023
+		err = -ERANGE;
Kmods SIG d83023
+		if (required)
Kmods SIG d83023
+			*required = len;
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	memcpy(buffer, ea->name + ea->name_len + 1, len);
Kmods SIG d83023
+	err = 0;
Kmods SIG d83023
+
Kmods SIG d83023
+out:
Kmods SIG d83023
+	ntfs_free(ea_all);
Kmods SIG d83023
+	if (!required)
Kmods SIG d83023
+		ni_unlock(ni);
Kmods SIG d83023
+
Kmods SIG d83023
+	return err ? err : len;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+static noinline int ntfs_set_ea(struct inode *inode, const char *name,
Kmods SIG d83023
+				size_t name_len, const void *value,
Kmods SIG d83023
+				size_t val_size, int flags, int locked)
Kmods SIG d83023
+{
Kmods SIG d83023
+	struct ntfs_inode *ni = ntfs_i(inode);
Kmods SIG d83023
+	struct ntfs_sb_info *sbi = ni->mi.sbi;
Kmods SIG d83023
+	int err;
Kmods SIG d83023
+	struct EA_INFO ea_info;
Kmods SIG d83023
+	const struct EA_INFO *info;
Kmods SIG d83023
+	struct EA_FULL *new_ea;
Kmods SIG d83023
+	struct EA_FULL *ea_all = NULL;
Kmods SIG d83023
+	size_t add, new_pack;
Kmods SIG d83023
+	u32 off, size;
Kmods SIG d83023
+	__le16 size_pack;
Kmods SIG d83023
+	struct ATTRIB *attr;
Kmods SIG d83023
+	struct ATTR_LIST_ENTRY *le;
Kmods SIG d83023
+	struct mft_inode *mi;
Kmods SIG d83023
+	struct runs_tree ea_run;
Kmods SIG d83023
+	u64 new_sz;
Kmods SIG d83023
+	void *p;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!locked)
Kmods SIG d83023
+		ni_lock(ni);
Kmods SIG d83023
+
Kmods SIG d83023
+	run_init(&ea_run);
Kmods SIG d83023
+
Kmods SIG d83023
+	if (name_len > 255) {
Kmods SIG d83023
+		err = -ENAMETOOLONG;
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	add = DwordAlign(struct_size(ea_all, name, 1 + name_len + val_size));
Kmods SIG d83023
+
Kmods SIG d83023
+	err = ntfs_read_ea(ni, &ea_all, add, &info;;
Kmods SIG d83023
+	if (err)
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!info) {
Kmods SIG d83023
+		memset(&ea_info, 0, sizeof(ea_info));
Kmods SIG d83023
+		size = 0;
Kmods SIG d83023
+		size_pack = 0;
Kmods SIG d83023
+	} else {
Kmods SIG d83023
+		memcpy(&ea_info, info, sizeof(ea_info));
Kmods SIG d83023
+		size = le32_to_cpu(ea_info.size);
Kmods SIG d83023
+		size_pack = ea_info.size_pack;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	if (info && find_ea(ea_all, size, name, name_len, &off)) {
Kmods SIG d83023
+		struct EA_FULL *ea;
Kmods SIG d83023
+		size_t ea_sz;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (flags & XATTR_CREATE) {
Kmods SIG d83023
+			err = -EEXIST;
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		ea = Add2Ptr(ea_all, off);
Kmods SIG d83023
+
Kmods SIG d83023
+		/*
Kmods SIG d83023
+		 * Check simple case when we try to insert xattr with the same value
Kmods SIG d83023
+		 * e.g. ntfs_save_wsl_perm
Kmods SIG d83023
+		 */
Kmods SIG d83023
+		if (val_size && le16_to_cpu(ea->elength) == val_size &&
Kmods SIG d83023
+		    !memcmp(ea->name + ea->name_len + 1, value, val_size)) {
Kmods SIG d83023
+			/* xattr already contains the required value */
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		/* Remove current xattr */
Kmods SIG d83023
+		if (ea->flags & FILE_NEED_EA)
Kmods SIG d83023
+			le16_add_cpu(&ea_info.count, -1);
Kmods SIG d83023
+
Kmods SIG d83023
+		ea_sz = unpacked_ea_size(ea);
Kmods SIG d83023
+
Kmods SIG d83023
+		le16_add_cpu(&ea_info.size_pack, 0 - packed_ea_size(ea));
Kmods SIG d83023
+
Kmods SIG d83023
+		memmove(ea, Add2Ptr(ea, ea_sz), size - off - ea_sz);
Kmods SIG d83023
+
Kmods SIG d83023
+		size -= ea_sz;
Kmods SIG d83023
+		memset(Add2Ptr(ea_all, size), 0, ea_sz);
Kmods SIG d83023
+
Kmods SIG d83023
+		ea_info.size = cpu_to_le32(size);
Kmods SIG d83023
+
Kmods SIG d83023
+		if ((flags & XATTR_REPLACE) && !val_size) {
Kmods SIG d83023
+			/* remove xattr */
Kmods SIG d83023
+			goto update_ea;
Kmods SIG d83023
+		}
Kmods SIG d83023
+	} else {
Kmods SIG d83023
+		if (flags & XATTR_REPLACE) {
Kmods SIG d83023
+			err = -ENODATA;
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		if (!ea_all) {
Kmods SIG d83023
+			ea_all = ntfs_zalloc(add);
Kmods SIG d83023
+			if (!ea_all) {
Kmods SIG d83023
+				err = -ENOMEM;
Kmods SIG d83023
+				goto out;
Kmods SIG d83023
+			}
Kmods SIG d83023
+		}
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	/* append new xattr */
Kmods SIG d83023
+	new_ea = Add2Ptr(ea_all, size);
Kmods SIG d83023
+	new_ea->size = cpu_to_le32(add);
Kmods SIG d83023
+	new_ea->flags = 0;
Kmods SIG d83023
+	new_ea->name_len = name_len;
Kmods SIG d83023
+	new_ea->elength = cpu_to_le16(val_size);
Kmods SIG d83023
+	memcpy(new_ea->name, name, name_len);
Kmods SIG d83023
+	new_ea->name[name_len] = 0;
Kmods SIG d83023
+	memcpy(new_ea->name + name_len + 1, value, val_size);
Kmods SIG d83023
+	new_pack = le16_to_cpu(ea_info.size_pack) + packed_ea_size(new_ea);
Kmods SIG d83023
+
Kmods SIG d83023
+	/* should fit into 16 bits */
Kmods SIG d83023
+	if (new_pack > 0xffff) {
Kmods SIG d83023
+		err = -EFBIG; // -EINVAL?
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+	ea_info.size_pack = cpu_to_le16(new_pack);
Kmods SIG d83023
+
Kmods SIG d83023
+	/* new size of ATTR_EA */
Kmods SIG d83023
+	size += add;
Kmods SIG d83023
+	if (size > sbi->ea_max_size) {
Kmods SIG d83023
+		err = -EFBIG; // -EINVAL?
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+	ea_info.size = cpu_to_le32(size);
Kmods SIG d83023
+
Kmods SIG d83023
+update_ea:
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!info) {
Kmods SIG d83023
+		/* Create xattr */
Kmods SIG d83023
+		if (!size) {
Kmods SIG d83023
+			err = 0;
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		err = ni_insert_resident(ni, sizeof(struct EA_INFO),
Kmods SIG d83023
+					 ATTR_EA_INFO, NULL, 0, NULL, NULL);
Kmods SIG d83023
+		if (err)
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+		err = ni_insert_resident(ni, 0, ATTR_EA, NULL, 0, NULL, NULL);
Kmods SIG d83023
+		if (err)
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	new_sz = size;
Kmods SIG d83023
+	err = attr_set_size(ni, ATTR_EA, NULL, 0, &ea_run, new_sz, &new_sz,
Kmods SIG d83023
+			    false, NULL);
Kmods SIG d83023
+	if (err)
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+	le = NULL;
Kmods SIG d83023
+	attr = ni_find_attr(ni, NULL, &le, ATTR_EA_INFO, NULL, 0, NULL, &mi);
Kmods SIG d83023
+	if (!attr) {
Kmods SIG d83023
+		err = -EINVAL;
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!size) {
Kmods SIG d83023
+		/* delete xattr, ATTR_EA_INFO */
Kmods SIG d83023
+		err = ni_remove_attr_le(ni, attr, le);
Kmods SIG d83023
+		if (err)
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+	} else {
Kmods SIG d83023
+		p = resident_data_ex(attr, sizeof(struct EA_INFO));
Kmods SIG d83023
+		if (!p) {
Kmods SIG d83023
+			err = -EINVAL;
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+		memcpy(p, &ea_info, sizeof(struct EA_INFO));
Kmods SIG d83023
+		mi->dirty = true;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	le = NULL;
Kmods SIG d83023
+	attr = ni_find_attr(ni, NULL, &le, ATTR_EA, NULL, 0, NULL, &mi);
Kmods SIG d83023
+	if (!attr) {
Kmods SIG d83023
+		err = -EINVAL;
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!size) {
Kmods SIG d83023
+		/* delete xattr, ATTR_EA */
Kmods SIG d83023
+		err = ni_remove_attr_le(ni, attr, le);
Kmods SIG d83023
+		if (err)
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+	} else if (attr->non_res) {
Kmods SIG d83023
+		err = ntfs_sb_write_run(sbi, &ea_run, 0, ea_all, size);
Kmods SIG d83023
+		if (err)
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+	} else {
Kmods SIG d83023
+		p = resident_data_ex(attr, size);
Kmods SIG d83023
+		if (!p) {
Kmods SIG d83023
+			err = -EINVAL;
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+		memcpy(p, ea_all, size);
Kmods SIG d83023
+		mi->dirty = true;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	/* Check if we delete the last xattr */
Kmods SIG d83023
+	if (size)
Kmods SIG d83023
+		ni->ni_flags |= NI_FLAG_EA;
Kmods SIG d83023
+	else
Kmods SIG d83023
+		ni->ni_flags &= ~NI_FLAG_EA;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (ea_info.size_pack != size_pack)
Kmods SIG d83023
+		ni->ni_flags |= NI_FLAG_UPDATE_PARENT;
Kmods SIG d83023
+	mark_inode_dirty(&ni->vfs_inode);
Kmods SIG d83023
+
Kmods SIG d83023
+out:
Kmods SIG d83023
+	if (!locked)
Kmods SIG d83023
+		ni_unlock(ni);
Kmods SIG d83023
+
Kmods SIG d83023
+	run_close(&ea_run);
Kmods SIG d83023
+	ntfs_free(ea_all);
Kmods SIG d83023
+
Kmods SIG d83023
+	return err;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+#ifdef CONFIG_NTFS3_FS_POSIX_ACL
Kmods SIG d83023
+static inline void ntfs_posix_acl_release(struct posix_acl *acl)
Kmods SIG d83023
+{
Kmods SIG d83023
+	if (acl && refcount_dec_and_test(&acl->a_refcount))
Kmods SIG d83023
+		kfree(acl);
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+static struct posix_acl *ntfs_get_acl_ex(struct user_namespace *mnt_userns,
Kmods SIG d83023
+					 struct inode *inode, int type,
Kmods SIG d83023
+					 int locked)
Kmods SIG d83023
+{
Kmods SIG d83023
+	struct ntfs_inode *ni = ntfs_i(inode);
Kmods SIG d83023
+	const char *name;
Kmods SIG d83023
+	size_t name_len;
Kmods SIG d83023
+	struct posix_acl *acl;
Kmods SIG d83023
+	size_t req;
Kmods SIG d83023
+	int err;
Kmods SIG d83023
+	void *buf;
Kmods SIG d83023
+
Kmods SIG d83023
+	/* allocate PATH_MAX bytes */
Kmods SIG d83023
+	buf = __getname();
Kmods SIG d83023
+	if (!buf)
Kmods SIG d83023
+		return ERR_PTR(-ENOMEM);
Kmods SIG d83023
+
Kmods SIG d83023
+	/* Possible values of 'type' was already checked above */
Kmods SIG d83023
+	if (type == ACL_TYPE_ACCESS) {
Kmods SIG d83023
+		name = XATTR_NAME_POSIX_ACL_ACCESS;
Kmods SIG d83023
+		name_len = sizeof(XATTR_NAME_POSIX_ACL_ACCESS) - 1;
Kmods SIG d83023
+	} else {
Kmods SIG d83023
+		name = XATTR_NAME_POSIX_ACL_DEFAULT;
Kmods SIG d83023
+		name_len = sizeof(XATTR_NAME_POSIX_ACL_DEFAULT) - 1;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!locked)
Kmods SIG d83023
+		ni_lock(ni);
Kmods SIG d83023
+
Kmods SIG d83023
+	err = ntfs_get_ea(inode, name, name_len, buf, PATH_MAX, &req;;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!locked)
Kmods SIG d83023
+		ni_unlock(ni);
Kmods SIG d83023
+
Kmods SIG d83023
+	/* Translate extended attribute to acl */
Kmods SIG d83023
+	if (err > 0) {
Kmods SIG d83023
+		acl = posix_acl_from_xattr(mnt_userns, buf, err);
Kmods SIG d83023
+		if (!IS_ERR(acl))
Kmods SIG d83023
+			set_cached_acl(inode, type, acl);
Kmods SIG d83023
+	} else {
Kmods SIG d83023
+		acl = err == -ENODATA ? NULL : ERR_PTR(err);
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	__putname(buf);
Kmods SIG d83023
+
Kmods SIG d83023
+	return acl;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * ntfs_get_acl
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * inode_operations::get_acl
Kmods SIG d83023
+ */
Kmods SIG d83023
+struct posix_acl *ntfs_get_acl(struct inode *inode, int type)
Kmods SIG d83023
+{
Kmods SIG d83023
+	/* TODO: init_user_ns? */
Kmods SIG d83023
+	return ntfs_get_acl_ex(&init_user_ns, inode, type, 0);
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+static noinline int ntfs_set_acl_ex(struct user_namespace *mnt_userns,
Kmods SIG d83023
+				    struct inode *inode, struct posix_acl *acl,
Kmods SIG d83023
+				    int type, int locked)
Kmods SIG d83023
+{
Kmods SIG d83023
+	const char *name;
Kmods SIG d83023
+	size_t size, name_len;
Kmods SIG d83023
+	void *value = NULL;
Kmods SIG d83023
+	int err = 0;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (S_ISLNK(inode->i_mode))
Kmods SIG d83023
+		return -EOPNOTSUPP;
Kmods SIG d83023
+
Kmods SIG d83023
+	switch (type) {
Kmods SIG d83023
+	case ACL_TYPE_ACCESS:
Kmods SIG d83023
+		if (acl) {
Kmods SIG d83023
+			umode_t mode = inode->i_mode;
Kmods SIG d83023
+
Kmods SIG d83023
+			err = posix_acl_equiv_mode(acl, &mode);
Kmods SIG d83023
+			if (err < 0)
Kmods SIG d83023
+				return err;
Kmods SIG d83023
+
Kmods SIG d83023
+			if (inode->i_mode != mode) {
Kmods SIG d83023
+				inode->i_mode = mode;
Kmods SIG d83023
+				mark_inode_dirty(inode);
Kmods SIG d83023
+			}
Kmods SIG d83023
+
Kmods SIG d83023
+			if (!err) {
Kmods SIG d83023
+				/*
Kmods SIG d83023
+				 * acl can be exactly represented in the
Kmods SIG d83023
+				 * traditional file mode permission bits
Kmods SIG d83023
+				 */
Kmods SIG d83023
+				acl = NULL;
Kmods SIG d83023
+			}
Kmods SIG d83023
+		}
Kmods SIG d83023
+		name = XATTR_NAME_POSIX_ACL_ACCESS;
Kmods SIG d83023
+		name_len = sizeof(XATTR_NAME_POSIX_ACL_ACCESS) - 1;
Kmods SIG d83023
+		break;
Kmods SIG d83023
+
Kmods SIG d83023
+	case ACL_TYPE_DEFAULT:
Kmods SIG d83023
+		if (!S_ISDIR(inode->i_mode))
Kmods SIG d83023
+			return acl ? -EACCES : 0;
Kmods SIG d83023
+		name = XATTR_NAME_POSIX_ACL_DEFAULT;
Kmods SIG d83023
+		name_len = sizeof(XATTR_NAME_POSIX_ACL_DEFAULT) - 1;
Kmods SIG d83023
+		break;
Kmods SIG d83023
+
Kmods SIG d83023
+	default:
Kmods SIG d83023
+		return -EINVAL;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!acl) {
Kmods SIG d83023
+		size = 0;
Kmods SIG d83023
+		value = NULL;
Kmods SIG d83023
+	} else {
Kmods SIG d83023
+		size = posix_acl_xattr_size(acl->a_count);
Kmods SIG d83023
+		value = ntfs_malloc(size);
Kmods SIG d83023
+		if (!value)
Kmods SIG d83023
+			return -ENOMEM;
Kmods SIG d83023
+
Kmods SIG d83023
+		err = posix_acl_to_xattr(mnt_userns, acl, value, size);
Kmods SIG d83023
+		if (err < 0)
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	err = ntfs_set_ea(inode, name, name_len, value, size,
Kmods SIG d83023
+			  acl ? 0 : XATTR_REPLACE, locked);
Kmods SIG d83023
+	if (!err)
Kmods SIG d83023
+		set_cached_acl(inode, type, acl);
Kmods SIG d83023
+
Kmods SIG d83023
+out:
Kmods SIG d83023
+	ntfs_free(value);
Kmods SIG d83023
+
Kmods SIG d83023
+	return err;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * ntfs_set_acl
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * inode_operations::set_acl
Kmods SIG d83023
+ */
Kmods SIG d83023
+int ntfs_set_acl(struct user_namespace *mnt_userns, struct inode *inode,
Kmods SIG d83023
+		 struct posix_acl *acl, int type)
Kmods SIG d83023
+{
Kmods SIG d83023
+	return ntfs_set_acl_ex(mnt_userns, inode, acl, type, 0);
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+static int ntfs_xattr_get_acl(struct user_namespace *mnt_userns,
Kmods SIG d83023
+			      struct inode *inode, int type, void *buffer,
Kmods SIG d83023
+			      size_t size)
Kmods SIG d83023
+{
Kmods SIG d83023
+	struct posix_acl *acl;
Kmods SIG d83023
+	int err;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!(inode->i_sb->s_flags & SB_POSIXACL))
Kmods SIG d83023
+		return -EOPNOTSUPP;
Kmods SIG d83023
+
Kmods SIG d83023
+	acl = ntfs_get_acl(inode, type);
Kmods SIG d83023
+	if (IS_ERR(acl))
Kmods SIG d83023
+		return PTR_ERR(acl);
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!acl)
Kmods SIG d83023
+		return -ENODATA;
Kmods SIG d83023
+
Kmods SIG d83023
+	err = posix_acl_to_xattr(mnt_userns, acl, buffer, size);
Kmods SIG d83023
+	ntfs_posix_acl_release(acl);
Kmods SIG d83023
+
Kmods SIG d83023
+	return err;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+static int ntfs_xattr_set_acl(struct user_namespace *mnt_userns,
Kmods SIG d83023
+			      struct inode *inode, int type, const void *value,
Kmods SIG d83023
+			      size_t size)
Kmods SIG d83023
+{
Kmods SIG d83023
+	struct posix_acl *acl;
Kmods SIG d83023
+	int err;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!(inode->i_sb->s_flags & SB_POSIXACL))
Kmods SIG d83023
+		return -EOPNOTSUPP;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!inode_owner_or_capable(mnt_userns, inode))
Kmods SIG d83023
+		return -EPERM;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!value) {
Kmods SIG d83023
+		acl = NULL;
Kmods SIG d83023
+	} else {
Kmods SIG d83023
+		acl = posix_acl_from_xattr(mnt_userns, value, size);
Kmods SIG d83023
+		if (IS_ERR(acl))
Kmods SIG d83023
+			return PTR_ERR(acl);
Kmods SIG d83023
+
Kmods SIG d83023
+		if (acl) {
Kmods SIG d83023
+			err = posix_acl_valid(mnt_userns, acl);
Kmods SIG d83023
+			if (err)
Kmods SIG d83023
+				goto release_and_out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	err = ntfs_set_acl(mnt_userns, inode, acl, type);
Kmods SIG d83023
+
Kmods SIG d83023
+release_and_out:
Kmods SIG d83023
+	ntfs_posix_acl_release(acl);
Kmods SIG d83023
+	return err;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * Initialize the ACLs of a new inode. Called from ntfs_create_inode.
Kmods SIG d83023
+ */
Kmods SIG d83023
+int ntfs_init_acl(struct user_namespace *mnt_userns, struct inode *inode,
Kmods SIG d83023
+		  struct inode *dir)
Kmods SIG d83023
+{
Kmods SIG d83023
+	struct posix_acl *default_acl, *acl;
Kmods SIG d83023
+	int err;
Kmods SIG d83023
+
Kmods SIG d83023
+	/*
Kmods SIG d83023
+	 * TODO refactoring lock
Kmods SIG d83023
+	 * ni_lock(dir) ... -> posix_acl_create(dir,...) -> ntfs_get_acl -> ni_lock(dir)
Kmods SIG d83023
+	 */
Kmods SIG d83023
+	inode->i_default_acl = NULL;
Kmods SIG d83023
+
Kmods SIG d83023
+	default_acl = ntfs_get_acl_ex(mnt_userns, dir, ACL_TYPE_DEFAULT, 1);
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!default_acl || default_acl == ERR_PTR(-EOPNOTSUPP)) {
Kmods SIG d83023
+		inode->i_mode &= ~current_umask();
Kmods SIG d83023
+		err = 0;
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	if (IS_ERR(default_acl)) {
Kmods SIG d83023
+		err = PTR_ERR(default_acl);
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	acl = default_acl;
Kmods SIG d83023
+	err = __posix_acl_create(&acl, GFP_NOFS, &inode->i_mode);
Kmods SIG d83023
+	if (err < 0)
Kmods SIG d83023
+		goto out1;
Kmods SIG d83023
+	if (!err) {
Kmods SIG d83023
+		posix_acl_release(acl);
Kmods SIG d83023
+		acl = NULL;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!S_ISDIR(inode->i_mode)) {
Kmods SIG d83023
+		posix_acl_release(default_acl);
Kmods SIG d83023
+		default_acl = NULL;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	if (default_acl)
Kmods SIG d83023
+		err = ntfs_set_acl_ex(mnt_userns, inode, default_acl,
Kmods SIG d83023
+				      ACL_TYPE_DEFAULT, 1);
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!acl)
Kmods SIG d83023
+		inode->i_acl = NULL;
Kmods SIG d83023
+	else if (!err)
Kmods SIG d83023
+		err = ntfs_set_acl_ex(mnt_userns, inode, acl, ACL_TYPE_ACCESS,
Kmods SIG d83023
+				      1);
Kmods SIG d83023
+
Kmods SIG d83023
+	posix_acl_release(acl);
Kmods SIG d83023
+out1:
Kmods SIG d83023
+	posix_acl_release(default_acl);
Kmods SIG d83023
+
Kmods SIG d83023
+out:
Kmods SIG d83023
+	return err;
Kmods SIG d83023
+}
Kmods SIG d83023
+#endif
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * ntfs_acl_chmod
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * helper for 'ntfs3_setattr'
Kmods SIG d83023
+ */
Kmods SIG d83023
+int ntfs_acl_chmod(struct user_namespace *mnt_userns, struct inode *inode)
Kmods SIG d83023
+{
Kmods SIG d83023
+	struct super_block *sb = inode->i_sb;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!(sb->s_flags & SB_POSIXACL))
Kmods SIG d83023
+		return 0;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (S_ISLNK(inode->i_mode))
Kmods SIG d83023
+		return -EOPNOTSUPP;
Kmods SIG d83023
+
Kmods SIG d83023
+	return posix_acl_chmod(mnt_userns, inode, inode->i_mode);
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * ntfs_permission
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * inode_operations::permission
Kmods SIG d83023
+ */
Kmods SIG d83023
+int ntfs_permission(struct user_namespace *mnt_userns, struct inode *inode,
Kmods SIG d83023
+		    int mask)
Kmods SIG d83023
+{
Kmods SIG d83023
+	if (ntfs_sb(inode->i_sb)->options.no_acs_rules) {
Kmods SIG d83023
+		/* "no access rules" mode - allow all changes */
Kmods SIG d83023
+		return 0;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	return generic_permission(mnt_userns, inode, mask);
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * ntfs_listxattr
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * inode_operations::listxattr
Kmods SIG d83023
+ */
Kmods SIG d83023
+ssize_t ntfs_listxattr(struct dentry *dentry, char *buffer, size_t size)
Kmods SIG d83023
+{
Kmods SIG d83023
+	struct inode *inode = d_inode(dentry);
Kmods SIG d83023
+	struct ntfs_inode *ni = ntfs_i(inode);
Kmods SIG d83023
+	ssize_t ret;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (!(ni->ni_flags & NI_FLAG_EA)) {
Kmods SIG d83023
+		/* no xattr in file */
Kmods SIG d83023
+		return 0;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	ni_lock(ni);
Kmods SIG d83023
+
Kmods SIG d83023
+	ret = ntfs_list_ea(ni, buffer, size);
Kmods SIG d83023
+
Kmods SIG d83023
+	ni_unlock(ni);
Kmods SIG d83023
+
Kmods SIG d83023
+	return ret;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+static int ntfs_getxattr(const struct xattr_handler *handler, struct dentry *de,
Kmods SIG d83023
+			 struct inode *inode, const char *name, void *buffer,
Kmods SIG d83023
+			 size_t size)
Kmods SIG d83023
+{
Kmods SIG d83023
+	int err;
Kmods SIG d83023
+	struct ntfs_inode *ni = ntfs_i(inode);
Kmods SIG d83023
+	size_t name_len = strlen(name);
Kmods SIG d83023
+
Kmods SIG d83023
+	/* Dispatch request */
Kmods SIG d83023
+	if (name_len == sizeof(SYSTEM_DOS_ATTRIB) - 1 &&
Kmods SIG d83023
+	    !memcmp(name, SYSTEM_DOS_ATTRIB, sizeof(SYSTEM_DOS_ATTRIB))) {
Kmods SIG d83023
+		/* system.dos_attrib */
Kmods SIG d83023
+		if (!buffer) {
Kmods SIG d83023
+			err = sizeof(u8);
Kmods SIG d83023
+		} else if (size < sizeof(u8)) {
Kmods SIG d83023
+			err = -ENODATA;
Kmods SIG d83023
+		} else {
Kmods SIG d83023
+			err = sizeof(u8);
Kmods SIG d83023
+			*(u8 *)buffer = le32_to_cpu(ni->std_fa);
Kmods SIG d83023
+		}
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	if (name_len == sizeof(SYSTEM_NTFS_ATTRIB) - 1 &&
Kmods SIG d83023
+	    !memcmp(name, SYSTEM_NTFS_ATTRIB, sizeof(SYSTEM_NTFS_ATTRIB))) {
Kmods SIG d83023
+		/* system.ntfs_attrib */
Kmods SIG d83023
+		if (!buffer) {
Kmods SIG d83023
+			err = sizeof(u32);
Kmods SIG d83023
+		} else if (size < sizeof(u32)) {
Kmods SIG d83023
+			err = -ENODATA;
Kmods SIG d83023
+		} else {
Kmods SIG d83023
+			err = sizeof(u32);
Kmods SIG d83023
+			*(u32 *)buffer = le32_to_cpu(ni->std_fa);
Kmods SIG d83023
+		}
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	if (name_len == sizeof(SYSTEM_NTFS_SECURITY) - 1 &&
Kmods SIG d83023
+	    !memcmp(name, SYSTEM_NTFS_SECURITY, sizeof(SYSTEM_NTFS_SECURITY))) {
Kmods SIG d83023
+		/* system.ntfs_security*/
Kmods SIG d83023
+		struct SECURITY_DESCRIPTOR_RELATIVE *sd = NULL;
Kmods SIG d83023
+		size_t sd_size = 0;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (!is_ntfs3(ni->mi.sbi)) {
Kmods SIG d83023
+			/* we should get nt4 security */
Kmods SIG d83023
+			err = -EINVAL;
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		} else if (le32_to_cpu(ni->std_security_id) <
Kmods SIG d83023
+			   SECURITY_ID_FIRST) {
Kmods SIG d83023
+			err = -ENOENT;
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		err = ntfs_get_security_by_id(ni->mi.sbi, ni->std_security_id,
Kmods SIG d83023
+					      &sd, &sd_size);
Kmods SIG d83023
+		if (err)
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (!is_sd_valid(sd, sd_size)) {
Kmods SIG d83023
+			ntfs_inode_warn(
Kmods SIG d83023
+				inode,
Kmods SIG d83023
+				"looks like you get incorrect security descriptor id=%u",
Kmods SIG d83023
+				ni->std_security_id);
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		if (!buffer) {
Kmods SIG d83023
+			err = sd_size;
Kmods SIG d83023
+		} else if (size < sd_size) {
Kmods SIG d83023
+			err = -ENODATA;
Kmods SIG d83023
+		} else {
Kmods SIG d83023
+			err = sd_size;
Kmods SIG d83023
+			memcpy(buffer, sd, sd_size);
Kmods SIG d83023
+		}
Kmods SIG d83023
+		ntfs_free(sd);
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+#ifdef CONFIG_NTFS3_FS_POSIX_ACL
Kmods SIG d83023
+	if ((name_len == sizeof(XATTR_NAME_POSIX_ACL_ACCESS) - 1 &&
Kmods SIG d83023
+	     !memcmp(name, XATTR_NAME_POSIX_ACL_ACCESS,
Kmods SIG d83023
+		     sizeof(XATTR_NAME_POSIX_ACL_ACCESS))) ||
Kmods SIG d83023
+	    (name_len == sizeof(XATTR_NAME_POSIX_ACL_DEFAULT) - 1 &&
Kmods SIG d83023
+	     !memcmp(name, XATTR_NAME_POSIX_ACL_DEFAULT,
Kmods SIG d83023
+		     sizeof(XATTR_NAME_POSIX_ACL_DEFAULT)))) {
Kmods SIG d83023
+		/* TODO: init_user_ns? */
Kmods SIG d83023
+		err = ntfs_xattr_get_acl(
Kmods SIG d83023
+			&init_user_ns, inode,
Kmods SIG d83023
+			name_len == sizeof(XATTR_NAME_POSIX_ACL_ACCESS) - 1
Kmods SIG d83023
+				? ACL_TYPE_ACCESS
Kmods SIG d83023
+				: ACL_TYPE_DEFAULT,
Kmods SIG d83023
+			buffer, size);
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+#endif
Kmods SIG d83023
+	/* deal with ntfs extended attribute */
Kmods SIG d83023
+	err = ntfs_get_ea(inode, name, name_len, buffer, size, NULL);
Kmods SIG d83023
+
Kmods SIG d83023
+out:
Kmods SIG d83023
+	return err;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * ntfs_setxattr
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * inode_operations::setxattr
Kmods SIG d83023
+ */
Kmods SIG d83023
+static noinline int ntfs_setxattr(const struct xattr_handler *handler,
Kmods SIG d83023
+				  struct user_namespace *mnt_userns,
Kmods SIG d83023
+				  struct dentry *de, struct inode *inode,
Kmods SIG d83023
+				  const char *name, const void *value,
Kmods SIG d83023
+				  size_t size, int flags)
Kmods SIG d83023
+{
Kmods SIG d83023
+	int err = -EINVAL;
Kmods SIG d83023
+	struct ntfs_inode *ni = ntfs_i(inode);
Kmods SIG d83023
+	size_t name_len = strlen(name);
Kmods SIG d83023
+	enum FILE_ATTRIBUTE new_fa;
Kmods SIG d83023
+
Kmods SIG d83023
+	/* Dispatch request */
Kmods SIG d83023
+	if (name_len == sizeof(SYSTEM_DOS_ATTRIB) - 1 &&
Kmods SIG d83023
+	    !memcmp(name, SYSTEM_DOS_ATTRIB, sizeof(SYSTEM_DOS_ATTRIB))) {
Kmods SIG d83023
+		if (sizeof(u8) != size)
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		new_fa = cpu_to_le32(*(u8 *)value);
Kmods SIG d83023
+		goto set_new_fa;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	if (name_len == sizeof(SYSTEM_NTFS_ATTRIB) - 1 &&
Kmods SIG d83023
+	    !memcmp(name, SYSTEM_NTFS_ATTRIB, sizeof(SYSTEM_NTFS_ATTRIB))) {
Kmods SIG d83023
+		if (size != sizeof(u32))
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		new_fa = cpu_to_le32(*(u32 *)value);
Kmods SIG d83023
+
Kmods SIG d83023
+		if (S_ISREG(inode->i_mode)) {
Kmods SIG d83023
+			/* Process compressed/sparsed in special way*/
Kmods SIG d83023
+			ni_lock(ni);
Kmods SIG d83023
+			err = ni_new_attr_flags(ni, new_fa);
Kmods SIG d83023
+			ni_unlock(ni);
Kmods SIG d83023
+			if (err)
Kmods SIG d83023
+				goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+set_new_fa:
Kmods SIG d83023
+		/*
Kmods SIG d83023
+		 * Thanks Mark Harmstone:
Kmods SIG d83023
+		 * keep directory bit consistency
Kmods SIG d83023
+		 */
Kmods SIG d83023
+		if (S_ISDIR(inode->i_mode))
Kmods SIG d83023
+			new_fa |= FILE_ATTRIBUTE_DIRECTORY;
Kmods SIG d83023
+		else
Kmods SIG d83023
+			new_fa &= ~FILE_ATTRIBUTE_DIRECTORY;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (ni->std_fa != new_fa) {
Kmods SIG d83023
+			ni->std_fa = new_fa;
Kmods SIG d83023
+			if (new_fa & FILE_ATTRIBUTE_READONLY)
Kmods SIG d83023
+				inode->i_mode &= ~0222;
Kmods SIG d83023
+			else
Kmods SIG d83023
+				inode->i_mode |= 0222;
Kmods SIG d83023
+			/* std attribute always in primary record */
Kmods SIG d83023
+			ni->mi.dirty = true;
Kmods SIG d83023
+			mark_inode_dirty(inode);
Kmods SIG d83023
+		}
Kmods SIG d83023
+		err = 0;
Kmods SIG d83023
+
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+	if (name_len == sizeof(SYSTEM_NTFS_SECURITY) - 1 &&
Kmods SIG d83023
+	    !memcmp(name, SYSTEM_NTFS_SECURITY, sizeof(SYSTEM_NTFS_SECURITY))) {
Kmods SIG d83023
+		/* system.ntfs_security*/
Kmods SIG d83023
+		__le32 security_id;
Kmods SIG d83023
+		bool inserted;
Kmods SIG d83023
+		struct ATTR_STD_INFO5 *std;
Kmods SIG d83023
+
Kmods SIG d83023
+		if (!is_ntfs3(ni->mi.sbi)) {
Kmods SIG d83023
+			/*
Kmods SIG d83023
+			 * we should replace ATTR_SECURE
Kmods SIG d83023
+			 * Skip this way cause it is nt4 feature
Kmods SIG d83023
+			 */
Kmods SIG d83023
+			err = -EINVAL;
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		if (!is_sd_valid(value, size)) {
Kmods SIG d83023
+			err = -EINVAL;
Kmods SIG d83023
+			ntfs_inode_warn(
Kmods SIG d83023
+				inode,
Kmods SIG d83023
+				"you try to set invalid security descriptor");
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+		}
Kmods SIG d83023
+
Kmods SIG d83023
+		err = ntfs_insert_security(ni->mi.sbi, value, size,
Kmods SIG d83023
+					   &security_id, &inserted);
Kmods SIG d83023
+		if (err)
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+		ni_lock(ni);
Kmods SIG d83023
+		std = ni_std5(ni);
Kmods SIG d83023
+		if (!std) {
Kmods SIG d83023
+			err = -EINVAL;
Kmods SIG d83023
+		} else if (std->security_id != security_id) {
Kmods SIG d83023
+			std->security_id = ni->std_security_id = security_id;
Kmods SIG d83023
+			/* std attribute always in primary record */
Kmods SIG d83023
+			ni->mi.dirty = true;
Kmods SIG d83023
+			mark_inode_dirty(&ni->vfs_inode);
Kmods SIG d83023
+		}
Kmods SIG d83023
+		ni_unlock(ni);
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+#ifdef CONFIG_NTFS3_FS_POSIX_ACL
Kmods SIG d83023
+	if ((name_len == sizeof(XATTR_NAME_POSIX_ACL_ACCESS) - 1 &&
Kmods SIG d83023
+	     !memcmp(name, XATTR_NAME_POSIX_ACL_ACCESS,
Kmods SIG d83023
+		     sizeof(XATTR_NAME_POSIX_ACL_ACCESS))) ||
Kmods SIG d83023
+	    (name_len == sizeof(XATTR_NAME_POSIX_ACL_DEFAULT) - 1 &&
Kmods SIG d83023
+	     !memcmp(name, XATTR_NAME_POSIX_ACL_DEFAULT,
Kmods SIG d83023
+		     sizeof(XATTR_NAME_POSIX_ACL_DEFAULT)))) {
Kmods SIG d83023
+		err = ntfs_xattr_set_acl(
Kmods SIG d83023
+			mnt_userns, inode,
Kmods SIG d83023
+			name_len == sizeof(XATTR_NAME_POSIX_ACL_ACCESS) - 1
Kmods SIG d83023
+				? ACL_TYPE_ACCESS
Kmods SIG d83023
+				: ACL_TYPE_DEFAULT,
Kmods SIG d83023
+			value, size);
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+#endif
Kmods SIG d83023
+	/* deal with ntfs extended attribute */
Kmods SIG d83023
+	err = ntfs_set_ea(inode, name, name_len, value, size, flags, 0);
Kmods SIG d83023
+
Kmods SIG d83023
+out:
Kmods SIG d83023
+	return err;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * ntfs_save_wsl_perm
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * save uid/gid/mode in xattr
Kmods SIG d83023
+ */
Kmods SIG d83023
+int ntfs_save_wsl_perm(struct inode *inode)
Kmods SIG d83023
+{
Kmods SIG d83023
+	int err;
Kmods SIG d83023
+	__le32 value;
Kmods SIG d83023
+
Kmods SIG d83023
+	value = cpu_to_le32(i_uid_read(inode));
Kmods SIG d83023
+	err = ntfs_set_ea(inode, "$LXUID", sizeof("$LXUID") - 1, &value,
Kmods SIG d83023
+			  sizeof(value), 0, 0);
Kmods SIG d83023
+	if (err)
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+	value = cpu_to_le32(i_gid_read(inode));
Kmods SIG d83023
+	err = ntfs_set_ea(inode, "$LXGID", sizeof("$LXGID") - 1, &value,
Kmods SIG d83023
+			  sizeof(value), 0, 0);
Kmods SIG d83023
+	if (err)
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+	value = cpu_to_le32(inode->i_mode);
Kmods SIG d83023
+	err = ntfs_set_ea(inode, "$LXMOD", sizeof("$LXMOD") - 1, &value,
Kmods SIG d83023
+			  sizeof(value), 0, 0);
Kmods SIG d83023
+	if (err)
Kmods SIG d83023
+		goto out;
Kmods SIG d83023
+
Kmods SIG d83023
+	if (S_ISCHR(inode->i_mode) || S_ISBLK(inode->i_mode)) {
Kmods SIG d83023
+		value = cpu_to_le32(inode->i_rdev);
Kmods SIG d83023
+		err = ntfs_set_ea(inode, "$LXDEV", sizeof("$LXDEV") - 1, &value,
Kmods SIG d83023
+				  sizeof(value), 0, 0);
Kmods SIG d83023
+		if (err)
Kmods SIG d83023
+			goto out;
Kmods SIG d83023
+	}
Kmods SIG d83023
+
Kmods SIG d83023
+out:
Kmods SIG d83023
+	/* In case of error should we delete all WSL xattr? */
Kmods SIG d83023
+	return err;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+/*
Kmods SIG d83023
+ * ntfs_get_wsl_perm
Kmods SIG d83023
+ *
Kmods SIG d83023
+ * get uid/gid/mode from xattr
Kmods SIG d83023
+ * it is called from ntfs_iget5->ntfs_read_mft
Kmods SIG d83023
+ */
Kmods SIG d83023
+void ntfs_get_wsl_perm(struct inode *inode)
Kmods SIG d83023
+{
Kmods SIG d83023
+	size_t sz;
Kmods SIG d83023
+	__le32 value[3];
Kmods SIG d83023
+
Kmods SIG d83023
+	if (ntfs_get_ea(inode, "$LXUID", sizeof("$LXUID") - 1, &value[0],
Kmods SIG d83023
+			sizeof(value[0]), &sz) == sizeof(value[0]) &&
Kmods SIG d83023
+	    ntfs_get_ea(inode, "$LXGID", sizeof("$LXGID") - 1, &value[1],
Kmods SIG d83023
+			sizeof(value[1]), &sz) == sizeof(value[1]) &&
Kmods SIG d83023
+	    ntfs_get_ea(inode, "$LXMOD", sizeof("$LXMOD") - 1, &value[2],
Kmods SIG d83023
+			sizeof(value[2]), &sz) == sizeof(value[2])) {
Kmods SIG d83023
+		i_uid_write(inode, (uid_t)le32_to_cpu(value[0]));
Kmods SIG d83023
+		i_gid_write(inode, (gid_t)le32_to_cpu(value[1]));
Kmods SIG d83023
+		inode->i_mode = le32_to_cpu(value[2]);
Kmods SIG d83023
+
Kmods SIG d83023
+		if (ntfs_get_ea(inode, "$LXDEV", sizeof("$$LXDEV") - 1,
Kmods SIG d83023
+				&value[0], sizeof(value),
Kmods SIG d83023
+				&sz) == sizeof(value[0])) {
Kmods SIG d83023
+			inode->i_rdev = le32_to_cpu(value[0]);
Kmods SIG d83023
+		}
Kmods SIG d83023
+	}
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+static bool ntfs_xattr_user_list(struct dentry *dentry)
Kmods SIG d83023
+{
Kmods SIG d83023
+	return true;
Kmods SIG d83023
+}
Kmods SIG d83023
+
Kmods SIG d83023
+// clang-format off
Kmods SIG d83023
+static const struct xattr_handler ntfs_xattr_handler = {
Kmods SIG d83023
+	.prefix	= "",
Kmods SIG d83023
+	.get	= ntfs_getxattr,
Kmods SIG d83023
+	.set	= ntfs_setxattr,
Kmods SIG d83023
+	.list	= ntfs_xattr_user_list,
Kmods SIG d83023
+};
Kmods SIG d83023
+
Kmods SIG d83023
+const struct xattr_handler *ntfs_xattr_handlers[] = {
Kmods SIG d83023
+	&ntfs_xattr_handler,
Kmods SIG d83023
+	NULL,
Kmods SIG d83023
+};
Kmods SIG d83023
+// clang-format on
Kmods SIG d83023
-- 
Kmods SIG d83023
2.31.1
Kmods SIG d83023