/*
 * LGPL HEADER START
 *
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * (C) Copyright 2014 Commissariat a l'energie atomique et aux energies
 *     alternatives
 *
 * Copyright (c) 2017, Intel Corporation.
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the GNU Lesser General Public License
 * (LGPL) version 2.1 or (at your discretion) any later version.
 * (LGPL) version 2.1 accompanies this distribution, and is available at
 * http://www.gnu.org/licenses/lgpl-2.1.html
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * LGPL HEADER END
 */
/*
 * lustre/utils/liblustreapi_lease.c
 *
 * lustreapi library for file leases
 *
 * Author: Henri Doreau <henri.doreau@cea.fr>
 */

#include <unistd.h>
#include <fcntl.h>
#include <sys/ioctl.h>
#include <errno.h>
#include <lustre/lustreapi.h>
#include "lustreapi_internal.h"

static inline const char *lease_mode2str(enum ll_lease_mode mode)
{
	switch (mode) {
	case LL_LEASE_WRLCK: return "WRITE";
	case LL_LEASE_RDLCK: return "READ";
	case LL_LEASE_UNLCK: return "UNLOCK";
	}
	return "???";
}

/**
 * Extend lease set support.
 *
 * \param fd	File to set lease on.
 * \param data	ll_ioc_lease data.
 *
 * For setting lease lock, it will return zero for success. For unlock, it will
 * return the lock type it owned for succuess.
 *
 * \retval >= 0 on success.
 * \retval -errno on error.
 */
int llapi_lease_set(int fd, const struct ll_ioc_lease *data)
{
	int rc;

	rc = ioctl(fd, LL_IOC_SET_LEASE, data);
	if (rc < 0) {
		rc = -errno;

		llapi_error(LLAPI_MSG_ERROR, rc, "cannot get %s lease, ext %x",
			    lease_mode2str(data->lil_mode), data->lil_flags);
	}
	return rc;
}

/**
 * Acquire a lease on an open file.
 *
 * \param fd    File to get the lease on.
 * \param mode  Lease mode, either LL_LEASE_RDLCK or LL_LEASE_WRLCK.
 *
 * \see llapi_lease_release().
 *
 * \retval >= 0 on success.
 * \retval -errno on error.
 */
int llapi_lease_acquire(int fd, enum ll_lease_mode mode)
{
	struct ll_ioc_lease data = { .lil_mode = mode };
	int rc;

	if (mode != LL_LEASE_RDLCK && mode != LL_LEASE_WRLCK)
		return -EINVAL;

	rc = llapi_lease_set(fd, &data);
	if (rc == -ENOTTY) {
		rc = ioctl(fd, LL_IOC_SET_LEASE_OLD, mode);
		if (rc < 0)
			rc = -errno;
	}

	return rc;
}

/**
 * Release a lease.
 *
 * \param fd    File to remove the lease from.
 *
 * \retval type of the lease that was removed (LL_LEASE_READ or LL_LEASE_WRITE).
 * \retval 0 if no lease was present.
 * \retval -errno on error.
 */
int llapi_lease_release(int fd)
{
	struct ll_ioc_lease data = { .lil_mode = LL_LEASE_UNLCK };

	return llapi_lease_set(fd, &data);
}

/**
 * Release a lease with intent operation. This API will release the lease
 * and execute the intent operation atomically.
 *
 * \param fd    File to remove the lease from.
 *
 * \retval type of the lease that was removed (LL_LEASE_READ or LL_LEASE_WRITE).
 * \retval 0 if no lease was present.
 * \retval -EBUSY lease broken, intent operation not executed.
 * \retval -errno on error.
 */
int llapi_lease_release_intent(int fd, struct ll_ioc_lease *data)
{
	if (data->lil_mode != LL_LEASE_UNLCK)
		return -EINVAL;

	return llapi_lease_set(fd, data);
}

/**
 * Check if a lease is still set on a file.
 *
 * \param fd    File to check the lease on.
 *
 * \retval lease type if present (LL_LEASE_READ or LL_LEASE_WRITE).
 * \retval 0 if no lease is present.
 * \retval -errno on error.
 */
int llapi_lease_check(int fd)
{
	int rc;

	rc = ioctl(fd, LL_IOC_GET_LEASE);
	if (rc < 0) {
		rc = -errno;
		llapi_error(LLAPI_MSG_ERROR, rc, "cannot check lease");
	}
	return rc;
}

/**
 * XXX: This is an obsoleted API - do not use it any more.
 */
int llapi_lease_get(int fd, int mode)
{
	int rc;

	if (mode != LL_LEASE_RDLCK && mode != LL_LEASE_WRLCK)
		return -EINVAL;

	rc = ioctl(fd, LL_IOC_SET_LEASE_OLD, mode);
	if (rc < 0)
		rc = -errno;

	return rc;
}

/**
 * XXX: This is an obsoleted API - do not use it any more.
 */
int llapi_lease_put(int fd)
{
	int rc;

	rc = ioctl(fd, LL_IOC_SET_LEASE_OLD, LL_LEASE_UNLCK);
	if (rc < 0)
		rc = -errno;

	return rc;
}