/*      $NetBSD: compat_50_quota.c,v 1.4 2022/09/21 07:15:24 dholland Exp $ */

/*-
 * Copyright (c) 2008 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * This code is derived from software contributed to The NetBSD Foundation
 * by Christos Zoulas.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */
#include <sys/cdefs.h>
__KERNEL_RCSID(0, "$NetBSD: compat_50_quota.c,v 1.4 2022/09/21 07:15:24 dholland Exp $");

#if defined(_KERNEL_OPT)
#include "opt_compat_netbsd.h"
#endif

#include <sys/module.h>
#include <sys/namei.h>
#include <sys/param.h>
#include <sys/quota.h>
#include <sys/quotactl.h>
#include <sys/systm.h>
#include <sys/syscall.h>
#include <sys/syscallvar.h>
#include <sys/syscallargs.h>
#include <sys/vfs_syscalls.h>
#include <sys/vnode.h>

#include <ufs/ufs/quota1.h>

static const struct syscall_package vfs_syscalls_50_quota_syscalls[] = {
        { SYS_compat_50_quotactl, 0, (sy_call_t *)compat_50_sys_quotactl },
        { 0, 0, NULL }
};

/* ARGSUSED */
int   
compat_50_sys_quotactl(struct lwp *l, const struct compat_50_sys_quotactl_args *uap, register_t *retval)
{
        /* {
                syscallarg(const char *) path;
                syscallarg(int) cmd;
                syscallarg(int) uid;
                syscallarg(void *) arg; 
        } */
        struct vnode *vp;
        struct mount *mp;
        int q1cmd;
        int idtype;
        char *qfile;
        struct dqblk dqblk;
        struct quotakey key;
        struct quotaval blocks, files;
        struct quotastat qstat;
        int error;

        error = namei_simple_user(SCARG(uap, path),
                                NSM_FOLLOW_TRYEMULROOT, &vp);
        if (error != 0)
                return (error);       

        mp = vp->v_mount;
        q1cmd = SCARG(uap, cmd);
        idtype = quota_idtype_from_ufs(q1cmd & SUBCMDMASK);
        if (idtype == -1) {
                return EINVAL;
        }

        switch ((q1cmd & ~SUBCMDMASK) >> SUBCMDSHIFT) {
        case Q_QUOTAON:
                qfile = PNBUF_GET();
                error = copyinstr(SCARG(uap, arg), qfile, PATH_MAX, NULL);
                if (error != 0) {
                        PNBUF_PUT(qfile);
                        break;
                }

                error = vfs_quotactl_quotaon(mp, idtype, qfile);

                PNBUF_PUT(qfile);
                break;

        case Q_QUOTAOFF:
                error = vfs_quotactl_quotaoff(mp, idtype);
                break;

        case Q_GETQUOTA:
                key.qk_idtype = idtype;
                key.qk_id = SCARG(uap, uid);

                key.qk_objtype = QUOTA_OBJTYPE_BLOCKS;
                error = vfs_quotactl_get(mp, &key, &blocks);
                if (error) {
                        break;
                }

                key.qk_objtype = QUOTA_OBJTYPE_FILES;
                error = vfs_quotactl_get(mp, &key, &files);
                if (error) {
                        break;
                }

                quotavals_to_dqblk(&blocks, &files, &dqblk);
                error = copyout(&dqblk, SCARG(uap, arg), sizeof(dqblk));
                break;
                
        case Q_SETQUOTA:
                error = copyin(SCARG(uap, arg), &dqblk, sizeof(dqblk));
                if (error) {
                        break;
                }
                dqblk_to_quotavals(&dqblk, &blocks, &files);

                key.qk_idtype = idtype;
                key.qk_id = SCARG(uap, uid);

                key.qk_objtype = QUOTA_OBJTYPE_BLOCKS;
                error = vfs_quotactl_put(mp, &key, &blocks);
                if (error) {
                        break;
                }

                key.qk_objtype = QUOTA_OBJTYPE_FILES;
                error = vfs_quotactl_put(mp, &key, &files);
                break;
                
        case Q_SYNC:
                /*
                 * not supported but used only to see if quota is supported,
                 * emulate with stat
                 *
                 * XXX should probably be supported
                 */
                (void)idtype; /* not used */

                error = vfs_quotactl_stat(mp, &qstat);
                break;

        case Q_SETUSE:
        default:
                error = EOPNOTSUPP;
                break;
        }

        vrele(vp);
        return error;
}

MODULE(MODULE_CLASS_EXEC, compat_50_quota, "compat_50,ufs");

static int
compat_50_quota_modcmd(modcmd_t cmd, void *arg)
{

        switch (cmd) {
        case MODULE_CMD_INIT:
                return syscall_establish(NULL, vfs_syscalls_50_quota_syscalls);
        case MODULE_CMD_FINI:
                return syscall_disestablish(NULL, vfs_syscalls_50_quota_syscalls);
        default:
                return ENOTTY;
        }
}