/*      $NetBSD: oss3_mixer.c,v 1.1 2021/06/08 18:43:54 nia Exp $       */

/*-
 * Copyright (c) 1997-2021 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * 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/audioio.h>
#include <sys/stat.h>
#include <errno.h>
#include "internal.h"

/* If the NetBSD mixer device should have more than NETBSD_MAXDEVS devices
 * some will not be available to OSS applications */
#define NETBSD_MAXDEVS 64

struct audiodevinfo {
        int done;
        dev_t dev;
        int16_t devmap[SOUND_MIXER_NRDEVICES],
                rdevmap[NETBSD_MAXDEVS];
        char names[NETBSD_MAXDEVS][MAX_AUDIO_DEV_LEN];
        int enum2opaque[NETBSD_MAXDEVS];
        u_long devmask, recmask, stereomask;
        u_long caps;
        int source;
};

static struct audiodevinfo *getdevinfo(int);
static int opaque_to_enum(struct audiodevinfo *, audio_mixer_name_t *, int);
static int enum_to_ord(struct audiodevinfo *, int);
static int enum_to_mask(struct audiodevinfo *, int);

oss_private int
_oss3_mixer_ioctl(int fd, unsigned long com, void *argp)
{
        struct audiodevinfo *di;
        struct mixer_info *omi;
        struct audio_device adev;
        mixer_ctrl_t mc;
        u_long idat, n;
        int i;
        int retval;
        int l, r, error, e;

        idat = 0;
        di = getdevinfo(fd);
        if (di == 0)
                return -1;

        switch (com) {
        case OSS_GETVERSION:
                idat = SOUND_VERSION;
                break;
        case SOUND_MIXER_INFO:
        case SOUND_OLD_MIXER_INFO:
                error = ioctl(fd, AUDIO_GETDEV, &adev);
                if (error)
                        return (error);
                omi = argp;
                if (com == SOUND_MIXER_INFO)
                        omi->modify_counter = 1;
                strlcpy(omi->id, adev.name, sizeof omi->id);
                strlcpy(omi->name, adev.name, sizeof omi->name);
                return 0;
        case SOUND_MIXER_READ_RECSRC:
                if (di->source == -1) {
                        errno = EINVAL;
                        return -1;
                }
                mc.dev = di->source;
                if (di->caps & SOUND_CAP_EXCL_INPUT) {
                        mc.type = AUDIO_MIXER_ENUM;
                        retval = ioctl(fd, AUDIO_MIXER_READ, &mc);
                        if (retval < 0)
                                return retval;
                        e = opaque_to_enum(di, NULL, mc.un.ord);
                        if (e >= 0)
                                idat = 1 << di->rdevmap[e];
                } else {
                        mc.type = AUDIO_MIXER_SET;
                        retval = ioctl(fd, AUDIO_MIXER_READ, &mc);
                        if (retval < 0)
                                return retval;
                        e = opaque_to_enum(di, NULL, mc.un.mask);
                        if (e >= 0)
                                idat = 1 << di->rdevmap[e];
                }
                break;
        case SOUND_MIXER_READ_DEVMASK:
                idat = di->devmask;
                break;
        case SOUND_MIXER_READ_RECMASK:
                idat = di->recmask;
                break;
        case SOUND_MIXER_READ_STEREODEVS:
                idat = di->stereomask;
                break;
        case SOUND_MIXER_READ_CAPS:
                idat = di->caps;
                break;
        case SOUND_MIXER_WRITE_RECSRC:
        case SOUND_MIXER_WRITE_R_RECSRC:
                if (di->source == -1) {
                        errno = EINVAL;
                        return -1;
                }
                mc.dev = di->source;
                idat = INTARG;
                if (di->caps & SOUND_CAP_EXCL_INPUT) {
                        mc.type = AUDIO_MIXER_ENUM;
                        for(i = 0; i < SOUND_MIXER_NRDEVICES; i++)
                                if (idat & (1 << i))
                                        break;
                        if (i >= SOUND_MIXER_NRDEVICES ||
                            di->devmap[i] == -1) {
                                errno = EINVAL;
                                return -1;
                        }
                        mc.un.ord = enum_to_ord(di, di->devmap[i]);
                } else {
                        mc.type = AUDIO_MIXER_SET;
                        mc.un.mask = 0;
                        for(i = 0; i < SOUND_MIXER_NRDEVICES; i++) {
                                if (idat & (1 << i)) {
                                        if (di->devmap[i] == -1) {
                                                errno = EINVAL;
                                                return -1;
                                        }
                                        mc.un.mask |=
                                            enum_to_mask(di, di->devmap[i]);
                                }
                        }
                }
                return ioctl(fd, AUDIO_MIXER_WRITE, &mc);
        default:
                if (MIXER_READ(SOUND_MIXER_FIRST) <= com &&
                    com < MIXER_READ(SOUND_MIXER_NRDEVICES)) {
                        n = GET_DEV(com);
                        if (di->devmap[n] == -1) {
                                errno = EINVAL;
                                return -1;
                        }
                        mc.dev = di->devmap[n];
                        mc.type = AUDIO_MIXER_VALUE;
                    doread:
                        mc.un.value.num_channels =
                            di->stereomask & (1 << (u_int)n) ? 2 : 1;
                        retval = ioctl(fd, AUDIO_MIXER_READ, &mc);
                        if (retval < 0)
                                return retval;
                        if (mc.type != AUDIO_MIXER_VALUE) {
                                errno = EINVAL;
                                return -1;
                        }
                        if (mc.un.value.num_channels != 2) {
                                l = r =
                                    mc.un.value.level[AUDIO_MIXER_LEVEL_MONO];
                        } else {
                                l = mc.un.value.level[AUDIO_MIXER_LEVEL_LEFT];
                                r = mc.un.value.level[AUDIO_MIXER_LEVEL_RIGHT];
                        }
                        idat = TO_OSSVOL(l) | (TO_OSSVOL(r) << 8);
                        break;
                } else if ((MIXER_WRITE_R(SOUND_MIXER_FIRST) <= com &&
                           com < MIXER_WRITE_R(SOUND_MIXER_NRDEVICES)) ||
                           (MIXER_WRITE(SOUND_MIXER_FIRST) <= com &&
                           com < MIXER_WRITE(SOUND_MIXER_NRDEVICES))) {
                        n = GET_DEV(com);
                        if (di->devmap[n] == -1) {
                                errno = EINVAL;
                                return -1;
                        }
                        idat = INTARG;
                        l = FROM_OSSVOL((u_int)idat & 0xff);
                        r = FROM_OSSVOL(((u_int)idat >> 8) & 0xff);
                        mc.dev = di->devmap[n];
                        mc.type = AUDIO_MIXER_VALUE;
                        if (di->stereomask & (1 << (u_int)n)) {
                                mc.un.value.num_channels = 2;
                                mc.un.value.level[AUDIO_MIXER_LEVEL_LEFT] = l;
                                mc.un.value.level[AUDIO_MIXER_LEVEL_RIGHT] = r;
                        } else {
                                mc.un.value.num_channels = 1;
                                mc.un.value.level[AUDIO_MIXER_LEVEL_MONO] =
                                    (l + r) / 2;
                        }
                        retval = ioctl(fd, AUDIO_MIXER_WRITE, &mc);
                        if (retval < 0)
                                return retval;
                        if (MIXER_WRITE(SOUND_MIXER_FIRST) <= com &&
                           com < MIXER_WRITE(SOUND_MIXER_NRDEVICES))
                                return 0;
                        goto doread;
                } else {
                        errno = EINVAL;
                        return -1;
                }
        }
        INTARG = (int)idat;
        return 0;
}

/*
 * Collect the audio device information to allow faster
 * emulation of the OSSv3 mixer ioctls.  Cache the information
 * to eliminate the overhead of repeating all the ioctls needed
 * to collect the information.
 */
static struct audiodevinfo *
getdevinfo(int fd)
{
        mixer_devinfo_t mi;
        int i, j, e;
        static struct {
                const char *name;
                int code;
        } *dp, devs[] = {
                { AudioNmicrophone,     SOUND_MIXER_MIC },
                { AudioNline,           SOUND_MIXER_LINE },
                { AudioNcd,             SOUND_MIXER_CD },
                { AudioNdac,            SOUND_MIXER_PCM },
                { AudioNaux,            SOUND_MIXER_LINE1 },
                { AudioNrecord,         SOUND_MIXER_IMIX },
                { AudioNmaster,         SOUND_MIXER_VOLUME },
                { AudioNtreble,         SOUND_MIXER_TREBLE },
                { AudioNbass,           SOUND_MIXER_BASS },
                { AudioNspeaker,        SOUND_MIXER_SPEAKER },
/*              { AudioNheadphone,      ?? },*/
                { AudioNoutput,         SOUND_MIXER_OGAIN },
                { AudioNinput,          SOUND_MIXER_IGAIN },
/*              { AudioNmaster,         SOUND_MIXER_SPEAKER },*/
/*              { AudioNstereo,         ?? },*/
/*              { AudioNmono,           ?? },*/
                { AudioNfmsynth,        SOUND_MIXER_SYNTH },
/*              { AudioNwave,           SOUND_MIXER_PCM },*/
                { AudioNmidi,           SOUND_MIXER_SYNTH },
/*              { AudioNmixerout,       ?? },*/
                { 0, -1 }
        };
        static struct audiodevinfo devcache = { .done = 0 };
        struct audiodevinfo *di = &devcache;
        struct stat sb;
        size_t mlen, dlen;

        /* Figure out what device it is so we can check if the
         * cached data is valid.
         */
        if (fstat(fd, &sb) < 0)
                return 0;
        if (di->done && di->dev == sb.st_dev)
                return di;

        di->done = 1;
        di->dev = sb.st_dev;
        di->devmask = 0;
        di->recmask = 0;
        di->stereomask = 0;
        di->source = ~0;
        di->caps = 0;
        for(i = 0; i < SOUND_MIXER_NRDEVICES; i++)
                di->devmap[i] = -1;
        for(i = 0; i < NETBSD_MAXDEVS; i++) {
                di->rdevmap[i] = -1;
                di->names[i][0] = '\0';
                di->enum2opaque[i] = -1;
        }
        for(i = 0; i < NETBSD_MAXDEVS; i++) {
                mi.index = i;
                if (ioctl(fd, AUDIO_MIXER_DEVINFO, &mi) < 0)
                        break;
                switch(mi.type) {
                case AUDIO_MIXER_VALUE:
                        for(dp = devs; dp->name; dp++) {
                                if (strcmp(dp->name, mi.label.name) == 0)
                                        break;
                                dlen = strlen(dp->name);
                                mlen = strlen(mi.label.name);
                                if (dlen < mlen
                                    && mi.label.name[mlen-dlen-1] == '.'
                                    && strcmp(dp->name,
                                    mi.label.name + mlen - dlen) == 0)
                                        break;
                        }
                        if (dp->code >= 0) {
                                di->devmap[dp->code] = i;
                                di->rdevmap[i] = dp->code;
                                di->devmask |= 1 << dp->code;
                                if (mi.un.v.num_channels == 2)
                                        di->stereomask |= 1 << dp->code;
                                strlcpy(di->names[i], mi.label.name,
                                        sizeof di->names[i]);
                        }
                        break;
                }
        }
        for(i = 0; i < NETBSD_MAXDEVS; i++) {
                mi.index = i;
                if (ioctl(fd, AUDIO_MIXER_DEVINFO, &mi) < 0)
                        break;
                if (strcmp(mi.label.name, AudioNsource) != 0)
                        continue;
                di->source = i;
                switch(mi.type) {
                case AUDIO_MIXER_ENUM:
                        for(j = 0; j < mi.un.e.num_mem; j++) {
                                e = opaque_to_enum(di,
                                                   &mi.un.e.member[j].label,
                                                   mi.un.e.member[j].ord);
                                if (e >= 0)
                                        di->recmask |= 1 << di->rdevmap[e];
                        }
                        di->caps = SOUND_CAP_EXCL_INPUT;
                        break;
                case AUDIO_MIXER_SET:
                        for(j = 0; j < mi.un.s.num_mem; j++) {
                                e = opaque_to_enum(di,
                                                   &mi.un.s.member[j].label,
                                                   mi.un.s.member[j].mask);
                                if (e >= 0)
                                        di->recmask |= 1 << di->rdevmap[e];
                        }
                        break;
                }
        }
        return di;
}

static int
opaque_to_enum(struct audiodevinfo *di, audio_mixer_name_t *label, int opq)
{
        int i, o;

        for (i = 0; i < NETBSD_MAXDEVS; i++) {
                o = di->enum2opaque[i];
                if (o == opq)
                        break;
                if (o == -1 && label != NULL &&
                    !strncmp(di->names[i], label->name, sizeof di->names[i])) {
                        di->enum2opaque[i] = opq;
                        break;
                }
        }
        if (i >= NETBSD_MAXDEVS)
                i = -1;
        /*printf("opq_to_enum %s %d -> %d\n", label->name, opq, i);*/
        return (i);
}

static int
enum_to_ord(struct audiodevinfo *di, int enm)
{
        if (enm >= NETBSD_MAXDEVS)
                return (-1);

        /*printf("enum_to_ord %d -> %d\n", enm, di->enum2opaque[enm]);*/
        return (di->enum2opaque[enm]);
}

static int
enum_to_mask(struct audiodevinfo *di, int enm)
{
        int m;
        if (enm >= NETBSD_MAXDEVS)
                return (0);

        m = di->enum2opaque[enm];
        if (m == -1)
                m = 0;
        /*printf("enum_to_mask %d -> %d\n", enm, di->enum2opaque[enm]);*/
        return (m);
}