// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * NTFS kernel quota ($Quota) handling.
 *
 * Copyright (c) 2004 Anton Altaparmakov
 */

#include "index.h"
#include "quota.h"
#include "debug.h"
#include "ntfs.h"

/*
 * ntfs_mark_quotas_out_of_date - mark the quotas out of date on an ntfs volume
 * @vol:	ntfs volume on which to mark the quotas out of date
 *
 * Mark the quotas out of date on the ntfs volume @vol and return 'true' on
 * success and 'false' on error.
 */
bool ntfs_mark_quotas_out_of_date(struct ntfs_volume *vol)
{
	struct ntfs_index_context *ictx;
	struct quota_control_entry *qce;
	const __le32 qid = QUOTA_DEFAULTS_ID;
	int err;

	ntfs_debug("Entering.");
	if (NVolQuotaOutOfDate(vol))
		goto done;
	if (!vol->quota_ino || !vol->quota_q_ino) {
		ntfs_error(vol->sb, "Quota inodes are not open.");
		return false;
	}
	inode_lock(vol->quota_q_ino);
	ictx = ntfs_index_ctx_get(NTFS_I(vol->quota_q_ino), I30, 4);
	if (!ictx) {
		ntfs_error(vol->sb, "Failed to get index context.");
		goto err_out;
	}
	err = ntfs_index_lookup(&qid, sizeof(qid), ictx);
	if (err) {
		if (err == -ENOENT)
			ntfs_error(vol->sb, "Quota defaults entry is not present.");
		else
			ntfs_error(vol->sb, "Lookup of quota defaults entry failed.");
		goto err_out;
	}
	if (ictx->data_len < offsetof(struct quota_control_entry, sid)) {
		ntfs_error(vol->sb, "Quota defaults entry size is invalid.  Run chkdsk.");
		goto err_out;
	}
	qce = (struct quota_control_entry *)ictx->data;
	if (le32_to_cpu(qce->version) != QUOTA_VERSION) {
		ntfs_error(vol->sb,
			"Quota defaults entry version 0x%x is not supported.",
			le32_to_cpu(qce->version));
		goto err_out;
	}
	ntfs_debug("Quota defaults flags = 0x%x.", le32_to_cpu(qce->flags));
	/* If quotas are already marked out of date, no need to do anything. */
	if (qce->flags & QUOTA_FLAG_OUT_OF_DATE)
		goto set_done;
	/*
	 * If quota tracking is neither requested, nor enabled and there are no
	 * pending deletes, no need to mark the quotas out of date.
	 */
	if (!(qce->flags & (QUOTA_FLAG_TRACKING_ENABLED |
			QUOTA_FLAG_TRACKING_REQUESTED |
			QUOTA_FLAG_PENDING_DELETES)))
		goto set_done;
	/*
	 * Set the QUOTA_FLAG_OUT_OF_DATE bit thus marking quotas out of date.
	 * This is verified on WinXP to be sufficient to cause windows to
	 * rescan the volume on boot and update all quota entries.
	 */
	qce->flags |= QUOTA_FLAG_OUT_OF_DATE;
	/* Ensure the modified flags are written to disk. */
	ntfs_index_entry_mark_dirty(ictx);
set_done:
	ntfs_index_ctx_put(ictx);
	inode_unlock(vol->quota_q_ino);
	/*
	 * We set the flag so we do not try to mark the quotas out of date
	 * again on remount.
	 */
	NVolSetQuotaOutOfDate(vol);
done:
	ntfs_debug("Done.");
	return true;
err_out:
	if (ictx)
		ntfs_index_ctx_put(ictx);
	inode_unlock(vol->quota_q_ino);
	return false;
}