/*
 * GPL HEADER START
 *
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 only,
 * as published by the Free Software Foundation.
 *
 * This program 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
 * General Public License version 2 for more details (a copy is included
 * in the LICENSE file that accompanied this code).
 *
 * You should have received a copy of the GNU General Public License
 * version 2 along with this program; If not, see
 * http://www.gnu.org/licenses/gpl-2.0.html
 *
 * GPL HEADER END
 */
/*
 * Copyright (c) 2007, 2010, Oracle and/or its affiliates. All rights reserved.
 * Use is subject to license terms.
 *
 * Copyright (c) 2014, Intel Corporation.
 */
/*
 * This file is part of Lustre, http://www.lustre.org/
 *
 * lustre/ptlrpc/sec_lproc.c
 *
 * Author: Eric Mei <ericm@clusterfs.com>
 */

#define DEBUG_SUBSYSTEM S_SEC

#include <libcfs/libcfs.h>
#include <linux/crypto.h>

#include <obd.h>
#include <obd_class.h>
#include <obd_support.h>
#include <lustre_net.h>
#include <lustre_import.h>
#include <lustre_dlm.h>
#include <lustre_sec.h>

#include "ptlrpc_internal.h"

static char *sec_flags2str(unsigned long flags, char *buf, int bufsize)
{
	buf[0] = '\0';

	if (flags & PTLRPC_SEC_FL_REVERSE)
		strlcat(buf, "reverse,", bufsize);
	if (flags & PTLRPC_SEC_FL_ROOTONLY)
		strlcat(buf, "rootonly,", bufsize);
	if (flags & PTLRPC_SEC_FL_UDESC)
		strlcat(buf, "udesc,", bufsize);
	if (flags & PTLRPC_SEC_FL_BULK)
		strlcat(buf, "bulk,", bufsize);
	if (buf[0] == '\0')
		strlcat(buf, "-,", bufsize);

	return buf;
}

static int sptlrpc_info_lprocfs_seq_show(struct seq_file *seq, void *v)
{
	struct obd_device *obd = seq->private;
	struct client_obd *cli = &obd->u.cli;
	struct ptlrpc_sec *sec = NULL;
	char               str[32];

	LASSERT(strcmp(obd->obd_type->typ_name, LUSTRE_OSC_NAME) == 0 ||
		strcmp(obd->obd_type->typ_name, LUSTRE_MDC_NAME) == 0 ||
		strcmp(obd->obd_type->typ_name, LUSTRE_MGC_NAME) == 0 ||
		strcmp(obd->obd_type->typ_name, LUSTRE_LWP_NAME) == 0 ||
		strcmp(obd->obd_type->typ_name, LUSTRE_OSP_NAME) == 0);

        if (cli->cl_import)
                sec = sptlrpc_import_sec_ref(cli->cl_import);
        if (sec == NULL)
                goto out;

        sec_flags2str(sec->ps_flvr.sf_flags, str, sizeof(str));

	seq_printf(seq, "rpc flavor:	%s\n",
		   sptlrpc_flavor2name_base(sec->ps_flvr.sf_rpc));
	seq_printf(seq, "bulk flavor:	%s\n",
		   sptlrpc_flavor2name_bulk(&sec->ps_flvr, str, sizeof(str)));
	seq_printf(seq, "flags:		%s\n",
		   sec_flags2str(sec->ps_flvr.sf_flags, str, sizeof(str)));
	seq_printf(seq, "id:		%d\n", sec->ps_id);
	seq_printf(seq, "refcount:	%d\n",
		   atomic_read(&sec->ps_refcount));
	seq_printf(seq, "nctx:	%d\n", atomic_read(&sec->ps_nctx));
	seq_printf(seq, "gc internal	%lld\n", sec->ps_gc_interval);
	seq_printf(seq, "gc next	%lld\n",
		   sec->ps_gc_interval ?
		   (s64)(sec->ps_gc_next - ktime_get_real_seconds()) : 0ll);

	sptlrpc_sec_put(sec);
out:
        return 0;
}

LDEBUGFS_SEQ_FOPS_RO(sptlrpc_info_lprocfs);

static int sptlrpc_ctxs_lprocfs_seq_show(struct seq_file *seq, void *v)
{
	struct obd_device *obd = seq->private;
	struct client_obd *cli = &obd->u.cli;
	struct ptlrpc_sec *sec = NULL;

	LASSERT(strcmp(obd->obd_type->typ_name, LUSTRE_OSC_NAME) == 0 ||
		strcmp(obd->obd_type->typ_name, LUSTRE_MDC_NAME) == 0 ||
		strcmp(obd->obd_type->typ_name, LUSTRE_MGC_NAME) == 0 ||
		strcmp(obd->obd_type->typ_name, LUSTRE_LWP_NAME) == 0 ||
		strcmp(obd->obd_type->typ_name, LUSTRE_OSP_NAME) == 0);

        if (cli->cl_import)
                sec = sptlrpc_import_sec_ref(cli->cl_import);
        if (sec == NULL)
                goto out;

        if (sec->ps_policy->sp_cops->display)
                sec->ps_policy->sp_cops->display(sec, seq);

        sptlrpc_sec_put(sec);
out:
        return 0;
}

LDEBUGFS_SEQ_FOPS_RO(sptlrpc_ctxs_lprocfs);

#if LUSTRE_VERSION_CODE < OBD_OCD_VERSION(2, 16, 53, 0)
static ssize_t sepol_seq_write_old(struct obd_device *obd,
				   const char __user *buffer,
				   size_t count)
{
	struct client_obd *cli = &obd->u.cli;
	struct obd_import *imp = cli->cl_import;
	struct sepol_downcall_data_old *param;
	int size = sizeof(*param);
	__u16 len;
	int rc = 0;

	if (count < size) {
		rc = -EINVAL;
		CERROR("%s: invalid data count = %lu, size = %d: rc = %d\n",
		       obd->obd_name, (unsigned long) count, size, rc);
		return rc;
	}

	OBD_ALLOC(param, size);
	if (param == NULL)
		return -ENOMEM;

	if (copy_from_user(param, buffer, size)) {
		rc = -EFAULT;
		CERROR("%s: bad sepol data: rc = %d\n", obd->obd_name, rc);
		GOTO(out, rc);
	}

	if (param->sdd_magic != SEPOL_DOWNCALL_MAGIC_OLD) {
		rc = -EINVAL;
		CERROR("%s: sepol downcall bad params: rc = %d\n",
		       obd->obd_name, rc);
		GOTO(out, rc);
	}

	if (param->sdd_sepol_len == 0 ||
	    param->sdd_sepol_len >= sizeof(imp->imp_sec->ps_sepol)) {
		rc = -EINVAL;
		CERROR("%s: invalid sepol data returned: rc = %d\n",
		       obd->obd_name, rc);
		GOTO(out, rc);
	}
	len = param->sdd_sepol_len; /* save sdd_sepol_len */
	OBD_FREE(param, size);
	size = offsetof(struct sepol_downcall_data_old,
			sdd_sepol[len]);

	if (count < size) {
		rc = -EINVAL;
		CERROR("%s: invalid sepol count = %lu, size = %d: rc = %d\n",
		       obd->obd_name, (unsigned long) count, size, rc);
		return rc;
	}

	/* alloc again with real size */
	OBD_ALLOC(param, size);
	if (param == NULL)
		return -ENOMEM;

	if (copy_from_user(param, buffer, size)) {
		rc = -EFAULT;
		CERROR("%s: cannot copy sepol data: rc = %d\n",
		       obd->obd_name, rc);
		GOTO(out, rc);
	}

	spin_lock(&imp->imp_sec->ps_lock);
	snprintf(imp->imp_sec->ps_sepol, param->sdd_sepol_len + 1, "%s",
		 param->sdd_sepol);
	imp->imp_sec->ps_sepol_mtime = ktime_set(param->sdd_sepol_mtime, 0);
	spin_unlock(&imp->imp_sec->ps_lock);

out:
	if (param != NULL)
		OBD_FREE(param, size);

	return rc ? rc : count;
}
#endif

static ssize_t
ldebugfs_sptlrpc_sepol_seq_write(struct file *file, const char __user *buffer,
				 size_t count, void *data)
{
	struct seq_file	*seq = file->private_data;
	struct obd_device *obd = seq->private;
	struct client_obd *cli = &obd->u.cli;
	struct obd_import *imp = cli->cl_import;
	struct sepol_downcall_data *param;
	__u32 magic;
	int size = sizeof(magic);
	__u16 len;
	int rc = 0;

	if (count < size) {
		rc = -EINVAL;
		CERROR("%s: invalid buffer count = %lu, size = %d: rc = %d\n",
		       obd->obd_name, (unsigned long) count, size, rc);
		return rc;
	}

	if (copy_from_user(&magic, buffer, size)) {
		rc = -EFAULT;
		CERROR("%s: bad sepol magic: rc = %d\n", obd->obd_name, rc);
		return rc;
	}

	if (magic != SEPOL_DOWNCALL_MAGIC) {
#if LUSTRE_VERSION_CODE < OBD_OCD_VERSION(2, 16, 53, 0)
		if (magic == SEPOL_DOWNCALL_MAGIC_OLD) {
			return sepol_seq_write_old(obd, buffer, count);
		}
#endif
		rc = -EINVAL;
		CERROR("%s: sepol downcall bad magic '%#08x': rc = %d\n",
		       obd->obd_name, magic, rc);
		return rc;
	}

	size = sizeof(*param);
	if (count < size) {
		rc = -EINVAL;
		CERROR("%s: invalid data count = %lu, size = %d: rc = %d\n",
		       obd->obd_name, (unsigned long) count, size, rc);
		return rc;
	}

	OBD_ALLOC(param, size);
	if (param == NULL)
		return -ENOMEM;

	if (copy_from_user(param, buffer, size)) {
		rc = -EFAULT;
		CERROR("%s: bad sepol data: rc = %d\n", obd->obd_name, rc);
		GOTO(out, rc);
	}

	if (param->sdd_sepol_len == 0 ||
	    param->sdd_sepol_len >= sizeof(imp->imp_sec->ps_sepol)) {
		rc = -EINVAL;
		CERROR("%s: invalid sepol data returned: rc = %d\n",
		       obd->obd_name, rc);
		GOTO(out, rc);
	}
	len = param->sdd_sepol_len; /* save sdd_sepol_len */
	OBD_FREE(param, size);
	size = offsetof(struct sepol_downcall_data,
			sdd_sepol[len]);

	/* alloc again with real size */
	OBD_ALLOC(param, size);
	if (param == NULL)
		return -ENOMEM;

	if (copy_from_user(param, buffer, size)) {
		rc = -EFAULT;
		CERROR("%s: cannot copy sepol data: rc = %d\n",
		       obd->obd_name, rc);
		GOTO(out, rc);
	}

	spin_lock(&imp->imp_sec->ps_lock);
	snprintf(imp->imp_sec->ps_sepol, param->sdd_sepol_len + 1, "%s",
		 param->sdd_sepol);
	imp->imp_sec->ps_sepol_mtime = ktime_set(param->sdd_sepol_mtime, 0);
	spin_unlock(&imp->imp_sec->ps_lock);

out:
	if (param != NULL)
		OBD_FREE(param, size);

	return rc ? rc : count;
}
LDEBUGFS_FOPS_WR_ONLY(srpc, sptlrpc_sepol);

int sptlrpc_lprocfs_cliobd_attach(struct obd_device *obd)
{
	if (strcmp(obd->obd_type->typ_name, LUSTRE_OSC_NAME) != 0 &&
	    strcmp(obd->obd_type->typ_name, LUSTRE_MDC_NAME) != 0 &&
	    strcmp(obd->obd_type->typ_name, LUSTRE_MGC_NAME) != 0 &&
	    strcmp(obd->obd_type->typ_name, LUSTRE_LWP_NAME) != 0 &&
	    strcmp(obd->obd_type->typ_name, LUSTRE_OSP_NAME) != 0) {
		CERROR("can't register lproc for obd type %s\n",
		       obd->obd_type->typ_name);
		return -EINVAL;
	}

	debugfs_create_file("srpc_info", 0444, obd->obd_debugfs_entry, obd,
			    &sptlrpc_info_lprocfs_fops);

	debugfs_create_file("srpc_contexts", 0444, obd->obd_debugfs_entry, obd,
			    &sptlrpc_ctxs_lprocfs_fops);

	debugfs_create_file("srpc_sepol", 0200, obd->obd_debugfs_entry, obd,
			    &srpc_sptlrpc_sepol_fops);

	return 0;
}
EXPORT_SYMBOL(sptlrpc_lprocfs_cliobd_attach);

LDEBUGFS_SEQ_FOPS_RO(sptlrpc_proc_enc_pool);

static struct ldebugfs_vars sptlrpc_lprocfs_vars[] = {
	{ .name	=	"encrypt_page_pools",
	  .fops	=	&sptlrpc_proc_enc_pool_fops	},
	{ NULL }
};

struct dentry *sptlrpc_debugfs_dir;
EXPORT_SYMBOL(sptlrpc_debugfs_dir);

struct proc_dir_entry *sptlrpc_lprocfs_dir;
EXPORT_SYMBOL(sptlrpc_lprocfs_dir);

int sptlrpc_lproc_init(void)
{
	int rc = 0;

	LASSERT(sptlrpc_debugfs_dir == NULL);

	sptlrpc_debugfs_dir = debugfs_create_dir("sptlrpc",
						 debugfs_lustre_root);
	ldebugfs_add_vars(sptlrpc_debugfs_dir, sptlrpc_lprocfs_vars, NULL);

	sptlrpc_lprocfs_dir = lprocfs_register("sptlrpc", proc_lustre_root,
					       NULL, NULL);
	if (IS_ERR_OR_NULL(sptlrpc_lprocfs_dir)) {
		rc = PTR_ERR(sptlrpc_lprocfs_dir);
		rc = sptlrpc_lprocfs_dir ? PTR_ERR(sptlrpc_lprocfs_dir)
			: -ENOMEM;
		sptlrpc_lprocfs_dir = NULL;
	}
	return rc;
}

void sptlrpc_lproc_fini(void)
{
	debugfs_remove_recursive(sptlrpc_debugfs_dir);
	sptlrpc_debugfs_dir = NULL;

	if (!IS_ERR_OR_NULL(sptlrpc_lprocfs_dir))
		lprocfs_remove(&sptlrpc_lprocfs_dir);
}
