/*
 * FireDTV driver (formerly known as FireSAT)
 *
 * Copyright (C) 2004 Andreas Monitzer <andy@monitzer.com>
 * Copyright (C) 2008 Henrik Kurelid <henrik@kurelid.se>
 *
 *	This program is free software; you can redistribute it and/or
 *	modify it under the terms of the GNU General Public License as
 *	published by the Free Software Foundation; either version 2 of
 *	the License, or (at your option) any later version.
 */

#include <linux/dvb/ca.h>
#include <linux/fs.h>
#include <linux/module.h>

#include <dvbdev.h>

#include "avc_api.h"
#include "firesat.h"
#include "firesat-ci.h"

static int firesat_ca_ready(ANTENNA_INPUT_INFO *info)
{
	return info->CaInitializationStatus == 1 &&
	       info->CaErrorFlag == 0 &&
	       info->CaDvbFlag == 1 &&
	       info->CaModulePresentStatus == 1;
}

static int firesat_get_ca_flags(ANTENNA_INPUT_INFO *info)
{
	int flags = 0;

	if (info->CaModulePresentStatus == 1)
		flags |= CA_CI_MODULE_PRESENT;
	if (info->CaInitializationStatus == 1 &&
	    info->CaErrorFlag == 0 &&
	    info->CaDvbFlag == 1)
		flags |= CA_CI_MODULE_READY;
	return flags;
}

static int firesat_ca_reset(struct firesat *firesat)
{
	return avc_ca_reset(firesat) ? -EFAULT : 0;
}

static int firesat_ca_get_caps(void *arg)
{
	struct ca_caps *cap = arg;

	cap->slot_num = 1;
	cap->slot_type = CA_CI;
	cap->descr_num = 1;
	cap->descr_type = CA_ECD;
	return 0;
}

static int firesat_ca_get_slot_info(struct firesat *firesat, void *arg)
{
	ANTENNA_INPUT_INFO info;
	struct ca_slot_info *slot = arg;

	if (avc_tuner_status(firesat, &info))
		return -EFAULT;

	if (slot->num != 0)
		return -EFAULT;

	slot->type = CA_CI;
	slot->flags = firesat_get_ca_flags(&info);
	return 0;
}

static int firesat_ca_app_info(struct firesat *firesat, void *arg)
{
	struct ca_msg *reply = arg;

	return
	    avc_ca_app_info(firesat, reply->msg, &reply->length) ? -EFAULT : 0;
}

static int firesat_ca_info(struct firesat *firesat, void *arg)
{
	struct ca_msg *reply = arg;

	return avc_ca_info(firesat, reply->msg, &reply->length) ? -EFAULT : 0;
}

static int firesat_ca_get_mmi(struct firesat *firesat, void *arg)
{
	struct ca_msg *reply = arg;

	return
	    avc_ca_get_mmi(firesat, reply->msg, &reply->length) ? -EFAULT : 0;
}

static int firesat_ca_get_msg(struct firesat *firesat, void *arg)
{
	ANTENNA_INPUT_INFO info;
	int err;

	switch (firesat->ca_last_command) {
	case TAG_APP_INFO_ENQUIRY:
		err = firesat_ca_app_info(firesat, arg);
		break;
	case TAG_CA_INFO_ENQUIRY:
		err = firesat_ca_info(firesat, arg);
		break;
	default:
		if (avc_tuner_status(firesat, &info))
			err = -EFAULT;
		else if (info.CaMmi == 1)
			err = firesat_ca_get_mmi(firesat, arg);
		else {
			printk(KERN_INFO "%s: Unhandled message 0x%08X\n",
			       __func__, firesat->ca_last_command);
			err = -EFAULT;
		}
	}
	firesat->ca_last_command = 0;
	return err;
}

static int firesat_ca_pmt(struct firesat *firesat, void *arg)
{
	struct ca_msg *msg = arg;
	int data_pos;

	if (msg->msg[3] & 0x80)
		data_pos = (msg->msg[4] && 0x7F) + 4;
	else
		data_pos = 4;

	return avc_ca_pmt(firesat, &msg->msg[data_pos],
			  msg->length - data_pos) ? -EFAULT : 0;
}

static int firesat_ca_send_msg(struct firesat *firesat, void *arg)
{
	struct ca_msg *msg = arg;
	int err;

	/* Do we need a semaphore for this? */
	firesat->ca_last_command =
		(msg->msg[0] << 16) + (msg->msg[1] << 8) + msg->msg[2];
	switch (firesat->ca_last_command) {
	case TAG_CA_PMT:
		err = firesat_ca_pmt(firesat, arg);
		break;
	case TAG_APP_INFO_ENQUIRY:
		/* handled in ca_get_msg */
		err = 0;
		break;
	case TAG_CA_INFO_ENQUIRY:
		/* handled in ca_get_msg */
		err = 0;
		break;
	case TAG_ENTER_MENU:
		err = avc_ca_enter_menu(firesat);
		break;
	default:
		printk(KERN_ERR "%s: Unhandled unknown message 0x%08X\n",
		       __func__, firesat->ca_last_command);
		err = -EFAULT;
	}
	return err;
}

static int firesat_ca_ioctl(struct inode *inode, struct file *file,
			    unsigned int cmd, void *arg)
{
	struct dvb_device *dvbdev = file->private_data;
	struct firesat *firesat = dvbdev->priv;
	ANTENNA_INPUT_INFO info;
	int err;

	switch(cmd) {
	case CA_RESET:
		err = firesat_ca_reset(firesat);
		break;
	case CA_GET_CAP:
		err = firesat_ca_get_caps(arg);
		break;
	case CA_GET_SLOT_INFO:
		err = firesat_ca_get_slot_info(firesat, arg);
		break;
	case CA_GET_MSG:
		err = firesat_ca_get_msg(firesat, arg);
		break;
	case CA_SEND_MSG:
		err = firesat_ca_send_msg(firesat, arg);
		break;
	default:
		printk(KERN_INFO "%s: Unhandled ioctl, command: %u\n",__func__,
		       cmd);
		err = -EOPNOTSUPP;
	}

	/* FIXME Is this necessary? */
	avc_tuner_status(firesat, &info);

	return err;
}

static unsigned int firesat_ca_io_poll(struct file *file, poll_table *wait)
{
	return POLLIN;
}

static struct file_operations firesat_ca_fops = {
	.owner		= THIS_MODULE,
	.ioctl		= dvb_generic_ioctl,
	.open		= dvb_generic_open,
	.release	= dvb_generic_release,
	.poll		= firesat_ca_io_poll,
};

static struct dvb_device firesat_ca = {
	.users		= 1,
	.readers	= 1,
	.writers	= 1,
	.fops		= &firesat_ca_fops,
	.kernel_ioctl	= firesat_ca_ioctl,
};

int firesat_ca_register(struct firesat *firesat)
{
	ANTENNA_INPUT_INFO info;
	int err;

	if (avc_tuner_status(firesat, &info))
		return -EINVAL;

	if (!firesat_ca_ready(&info))
		return -EFAULT;

	err = dvb_register_device(&firesat->adapter, &firesat->cadev,
				  &firesat_ca, firesat, DVB_DEVICE_CA);

	if (info.CaApplicationInfo == 0)
		printk(KERN_ERR "%s: CaApplicationInfo is not set.\n",
		       __func__);
	if (info.CaDateTimeRequest == 1)
		avc_ca_get_time_date(firesat, &firesat->ca_time_interval);

	return err;
}

void firesat_ca_release(struct firesat *firesat)
{
	if (firesat->cadev)
		dvb_unregister_device(firesat->cadev);
}