[media] dvb: Add a new driver for az6007

/* DVB USB compliant Linux driver for the AzureWave 6017 USB2.0 DVB-S

 * receiver.

 * see Documentation/dvb/README.dvb-usb for more information

 */

#include "az6007.h"

#include "drxk.h"

#include "mt2063.h"

#include "dvb_ca_en50221.h"

/* debug */

int dvb_usb_az6007_debug;

module_param_named(debug,dvb_usb_az6007_debug, int, 0644);

MODULE_PARM_DESC(debug, "set debugging level (1=info,xfer=2,rc=4 (or-able))." DVB_USB_DEBUG_STATUS);

static int az6007_type =0;

module_param(az6007_type, int, 0644);

MODULE_PARM_DESC(az6007_type, "select delivery mode (0=DVB-T, 1=DVB-T");

//module_param_named(type, 6007_type, int, 0644);

//MODULE_PARM_DESC(type, "select delivery mode (0=DVB-T, 1=DVB-C)");

DVB_DEFINE_MOD_OPT_ADAPTER_NR(adapter_nr);

struct az6007_device_state {

	struct dvb_ca_en50221 ca;

	struct mutex ca_mutex;

	u8 power_state;

};

struct drxk3913_config az6007_drxk3913_config_DVBT = {	

	.demod_address = 0x52,

	.min_delay_ms = 100,

	.standard	= MTTUNEA_DVBT,

	.set_tuner = mt2063_setTune,

	.tuner_getlocked = mt2063_lockStatus,

	.tuner_MT2063_Open = tuner_MT2063_Open,

	.tuner_MT2063_SoftwareShutdown = tuner_MT2063_SoftwareShutdown,

	.tuner_MT2063_ClearPowerMaskBits = tuner_MT2063_ClearPowerMaskBits,

};

struct drxk3913_config az6007_drxk3913_config_DVBC = {	

	.demod_address = 0x52,

	.min_delay_ms = 100,

	.standard	= MTTUNEA_DVBC,

	.set_tuner = mt2063_setTune,

	.tuner_getlocked = mt2063_lockStatus,

	.tuner_MT2063_Open = tuner_MT2063_Open,

	.tuner_MT2063_SoftwareShutdown = tuner_MT2063_SoftwareShutdown,

	.tuner_MT2063_ClearPowerMaskBits = tuner_MT2063_ClearPowerMaskBits,

};

struct mt2063_config az6007_mt2063_config = {

	.tuner_address = 0xc0,

	.refclock = 36125000,

};

/* check for mutex FIXME */

int az6007_usb_in_op(struct dvb_usb_device *d, u8 req, u16 value, u16 index, u8 *b, int blen)

{

	int ret = -1;

		ret = usb_control_msg(d->udev,

			usb_rcvctrlpipe(d->udev,0),

			req,

			USB_TYPE_VENDOR | USB_DIR_IN,

			value,index,b,blen,

			5000);

	if (ret < 0) {

		warn("usb in operation failed. (%d)", ret);

		ret = -EIO;

	} else

		ret = 0;

	deb_xfer("in: req. %02x, val: %04x, ind: %04x, buffer: ",req,value,index);

	debug_dump(b,blen,deb_xfer);

	return ret;

}

static int az6007_usb_out_op(struct dvb_usb_device *d, u8 req, u16 value,

			     u16 index, u8 *b, int blen)

{

	int ret;

#if 0	

	int i=0, cyc=0, rem=0;

	cyc = blen/64;

	rem = blen%64;

#endif

	deb_xfer("out: req. %02x, val: %04x, ind: %04x, buffer: ",req,value,index);

	debug_dump(b,blen,deb_xfer);

#if 0

	if (blen>64)

	{

		for (i=0; i<cyc; i++)

		{

			if ((ret = usb_control_msg(d->udev,

				usb_sndctrlpipe(d->udev,0),

				req,

				USB_TYPE_VENDOR | USB_DIR_OUT,

				value,index+i*64,b+i*64,64,

				5000)) != 64) {

				warn("usb out operation failed. (%d)",ret);

				return -EIO;

			}	

		}

		if (rem>0)

		{

			if ((ret = usb_control_msg(d->udev,

				usb_sndctrlpipe(d->udev,0),

				req,

				USB_TYPE_VENDOR | USB_DIR_OUT,

				value,index+cyc*64,b+cyc*64,rem,

				5000)) != rem) {

				warn("usb out operation failed. (%d)",ret);

				return -EIO;

			}

		}

	}

	else

#endif		

	{

		if ((ret = usb_control_msg(d->udev,

				usb_sndctrlpipe(d->udev,0),

				req,

				USB_TYPE_VENDOR | USB_DIR_OUT,

				value,index,b,blen,

				5000)) != blen) {

			warn("usb out operation failed. (%d)",ret);

			return -EIO;

		}

	}

	return 0;

}

static int az6007_streaming_ctrl(struct dvb_usb_adapter *adap, int onoff)

{

	return 0;

}

/* keys for the enclosed remote control */

static struct dvb_usb_rc_key az6007_rc_keys[] = {

	{ 0x00, 0x01, KEY_1 },

	{ 0x00, 0x02, KEY_2 },

};

/* remote control stuff (does not work with my box) */

static int az6007_rc_query(struct dvb_usb_device *d, u32 *event, int *state)

{

	return 0;

#if 0

	u8 key[10];

	int i;

/* remove the following return to enabled remote querying */

	az6007_usb_in_op(d,READ_REMOTE_REQ,0,0,key,10);

	deb_rc("remote query key: %x %d\n",key[1],key[1]);

	if (key[1] == 0x44) {

		*state = REMOTE_NO_KEY_PRESSED;

		return 0;

	}

	for (i = 0; i < ARRAY_SIZE(az6007_rc_keys); i++)

		if (az6007_rc_keys[i].custom == key[1]) {

			*state = REMOTE_KEY_PRESSED;

			*event = az6007_rc_keys[i].event;

			break;

		}

	return 0;

#endif

}

/*

int az6007_power_ctrl(struct dvb_usb_device *d, int onoff)

{

	u8 v = onoff;

	return az6007_usb_out_op(d,0xBC,v,3,NULL,1);

}

*/

static int az6007_read_mac_addr(struct dvb_usb_device *d,u8 mac[6])

{

	az6007_usb_in_op(d, 0xb7, 6, 0, &mac[0], 6);

	return 0;

}

static int az6007_frontend_poweron(struct dvb_usb_adapter *adap)

{

	int ret;

	u8 req;

	u16 value;

	u16 index;

	int blen;

	info("az6007_frontend_poweron adap=%p adap->dev=%p\n", adap, adap->dev);

	req = 0xBC;

	value = 1;//power on

	index = 3;

	blen =0;

	if((ret = az6007_usb_out_op(adap->dev,req,value,index,NULL,blen)) != 0)

	{

		err("az6007_frontend_poweron failed!!!");

		 return -EIO;

	}

	msleep_interruptible(200);

	req = 0xBC;

	value = 0;//power on

	index = 3;

	blen =0;

	if((ret = az6007_usb_out_op(adap->dev,req,value,index,NULL,blen)) != 0)

	{

		err("az6007_frontend_poweron failed!!!");

		 return -EIO;

	}

	msleep_interruptible(200);

	req = 0xBC;

	value = 1;//power on

	index = 3;

	blen =0;

	if((ret = az6007_usb_out_op(adap->dev,req,value,index,NULL,blen)) != 0)

	{

		err("az6007_frontend_poweron failed!!!");

		 return -EIO;

	}

	info("az6007_frontend_poweron\n");

	return 0;

}

static int az6007_frontend_reset(struct dvb_usb_adapter *adap)

{

	int ret;

	u8 req;

	u16 value;

	u16 index;

	int blen;

	info("az6007_frontend_reset adap=%p adap->dev=%p\n", adap, adap->dev);

	//reset demodulator

	req = 0xC0;

	value = 1;//high

	index = 3;

	blen =0;

	if((ret = az6007_usb_out_op(adap->dev,req,value,index,NULL,blen)) != 0)

	{

		err("az6007_frontend_reset failed 1 !!!");

		   return -EIO;

	}

	req = 0xC0;

	value = 0;//low

	index = 3;

	blen =0;

	msleep_interruptible(200);

	if((ret = az6007_usb_out_op(adap->dev,req,value,index,NULL,blen)) != 0)

	{

		err("az6007_frontend_reset failed 2 !!!");

		   return -EIO;

	}

	msleep_interruptible(200);

	req = 0xC0;

	value = 1;//high

	index = 3;

	blen =0;

	if((ret = az6007_usb_out_op(adap->dev,req,value,index,NULL,blen)) != 0)

	{

		err("az6007_frontend_reset failed 3 !!!");

		   return -EIO;

	}

	msleep_interruptible(200);

	info("reset az6007 frontend\n");

	return 0;

}

static int az6007_led_on_off(struct usb_interface *intf, int onoff)

{

	int ret = -1;

	u8 req;

	u16 value;

	u16 index;

	int blen;

	//TS through

	req = 0xBC;

	value = onoff;

	index = 0;

	blen =0;

	ret = usb_control_msg(interface_to_usbdev(intf),

		usb_rcvctrlpipe(interface_to_usbdev(intf),0),

		req,

		USB_TYPE_VENDOR | USB_DIR_OUT,

		value,index,NULL,blen,

		2000);

	if (ret < 0) {

		warn("usb in operation failed. (%d)", ret);

		ret = -EIO;

	} else

		ret = 0;

	deb_xfer("in: req. %02x, val: %04x, ind: %04x, buffer: ",req,value,index);

	return ret;

}

static int az6007_frontend_tsbypass(struct dvb_usb_adapter *adap,int onoff)

{

	int ret;

	u8 req;

	u16 value;

	u16 index;

	int blen;

	//TS through

	req = 0xC7;

	value = onoff;

	index = 0;

	blen =0;

	if((ret = az6007_usb_out_op(adap->dev,req,value,index,NULL,blen)) != 0)

		   return -EIO;

	return 0;

}

static int az6007_frontend_attach(struct dvb_usb_adapter *adap)

{

	az6007_frontend_poweron(adap);

	az6007_frontend_reset(adap);

	info("az6007_frontend_attach\n");

	if (az6007_type == 0)

	{

		info("az6007_drxk3913_config_DVBT\n");

		adap->fe = drxk3913_attach(&az6007_drxk3913_config_DVBT, &adap->dev->i2c_adap);

	}

	else

	{

		info("az6007_drxk3913_config_DVBC\n");

		adap->fe = drxk3913_attach(&az6007_drxk3913_config_DVBC, &adap->dev->i2c_adap);

	}

	if (adap->fe) {		

		if (mt2063_attach(adap->fe, &az6007_mt2063_config, &adap->dev->i2c_adap)) {

			info("found STB6100 DVB-C/DVB-T frontend @0x%02x\n",az6007_mt2063_config.tuner_address);

			//vp6027_ci_init(adap);

		} else {

			adap->fe = NULL;

		}

	}

	else

	{

		adap->fe = NULL;	

		err("no front-end attached\n");

	}

	//az6007_frontend_tsbypass(adap,0);

	return 0;

}

static struct dvb_usb_device_properties az6007_properties;

static void

az6007_usb_disconnect(struct usb_interface *intf)

{

	dvb_usb_device_exit (intf);

}

/* I2C */

static int az6007_i2c_xfer(struct i2c_adapter *adap,struct i2c_msg msg[],int num)

{

	struct dvb_usb_device *d = i2c_get_adapdata(adap);

	int j=0,len=0;

	int ret=0;

	u16 index;

	u16 value;

	int length;

	u8 req;

	u8 data[512];

	if (mutex_lock_interruptible(&d->i2c_mutex) < 0)

		return -EAGAIN;

	if (num > 2)

		warn("more than 2 i2c messages at a time is not handled yet. TODO.");

	if (msg[0].addr == 0xc0) //MT2063

	{

		if (msg[0].flags != I2C_M_RD) //write

		{

			//printk("Tuner Tuner Write DevAddr=%02x  RegAddr=%d\n", msg[0].addr, msg[0].buf[0]);

			req = 0xBD;

			index = msg[0].buf[0];

			value = msg[0].addr | (1<<8);

			length = msg[0].len - 1;

			len = msg[0].len - 1;

			//printk("Tuner Tuner WriteDATA len=%d ", len);

			for(j=0;j<len;j++)

			{

				data[j] = msg[0].buf[j+1];

				//printk("data[%d]=%02x ", j, data[j]);

			}

			//printk("\n");

			ret = az6007_usb_out_op(d,req,value,index,data,length);

			//ret = az6007_usb_out_op(d,req,value,index,&(msg[0].buf[1]),length);

		}

		else //read

		{

			//printk("Tuner Tuner Read DevAddr=%02x RegAddr=%02x\n", msg[0].addr, msg[0].buf[0]);

			req = 0xB9;

			index = msg[0].buf[0];

			value = msg[0].addr + (1 << 8);

			length = msg[1].len + 6;

			ret = az6007_usb_in_op(d,req,value,index,data,length);

			len = msg[1].len;

			//printk("Tuner Tuner ReadDATA len=%d ", len);

			for (j=0; j<len; j++)

			{

				msg[1].buf[j] = data[j+5];	

				//printk("data[%d]=%02x ", j, data[j+5]);

			}

			//printk("\n");

		}

	}

	else

	{	//Demodulator

		if (msg[0].flags != I2C_M_RD) //write

		{

			//printk("Demodulator Write DevAddr=%02x  RegAddr=%d\n", msg[0].addr, msg[0].buf[0]);

			req = 0xBD;

			index = msg[0].buf[0];

			value = msg[0].addr | (1<<8);

			length = msg[0].len - 1;

			len = msg[0].len - 1;

			//printk("Demodulator WriteDATA len=%d ", len);			

			for(j=0;j<len;j++)

			{

				data[j] = msg[0].buf[j+1];

				//printk("data[%d]=%02x ", j, data[j]);

			}

			//printk("\n");			

			ret = az6007_usb_out_op(d,req,value,index,data,length);

		}

		else //read

		{

			//printk("Demodulator Read DevAddr=%02x RegAddr=%02x\n", msg[0].addr, msg[0].buf[0]);

			req = 0xB9;

			index = 0;

			value = msg[0].addr + (0 << 8);

			length = msg[0].len + 6;

			ret = az6007_usb_in_op(d,req,value,index,data,length);

			len = msg[0].len;

			//printk("Demodulator ReadDATA len=%d ", len);

			for (j=0; j<len; j++)

			{

				msg[0].buf[j] = data[j+5];	

				//printk("data[%d]=%02x ", j, data[j+5]);

			}

			//printk("\n");

		}

	}

	mutex_unlock(&d->i2c_mutex);

	return ret;

}

static u32 az6007_i2c_func(struct i2c_adapter *adapter)

{

	return I2C_FUNC_I2C;

}

static struct i2c_algorithm az6007_i2c_algo = {

	.master_xfer   = az6007_i2c_xfer,

	.functionality = az6007_i2c_func,

#ifdef NEED_ALGO_CONTROL

	.algo_control = dummy_algo_control,

#endif

};

int az6007_identify_state(struct usb_device *udev, struct dvb_usb_device_properties *props,

			struct dvb_usb_device_description **desc, int *cold)

{

	u8 b[16];

	s16 ret = usb_control_msg(udev, usb_rcvctrlpipe(udev,0),

		0xb7, USB_TYPE_VENDOR | USB_DIR_IN, 6, 0, b, 6, USB_CTRL_GET_TIMEOUT);

	info("FW GET_VERSION length: %d\n",ret);

	*cold = ret <= 0;

	info("cold: %d\n", *cold);

	return 0;

}

static int az6007_usb_probe(struct usb_interface *intf,

		const struct usb_device_id *id)

{

	az6007_led_on_off(intf, 0);	

	return dvb_usb_device_init(intf, &az6007_properties,

				   THIS_MODULE, NULL, adapter_nr);	

}

static struct usb_device_id az6007_usb_table [] = {

	    { USB_DEVICE(USB_VID_AZUREWAVE, USB_PID_AZUREWAVE_6007) },

	    { USB_DEVICE(USB_VID_TERRATEC, USB_PID_TERRATEC_H7) },

	    { 0 },

};

MODULE_DEVICE_TABLE(usb, az6007_usb_table);

static struct dvb_usb_device_properties az6007_properties = {

	.caps = DVB_USB_IS_AN_I2C_ADAPTER,

	.usb_ctrl = CYPRESS_FX2,

	//.download_firmware = az6007_download_firmware,

	.firmware            = "dvb-usb-az6007-03.fw",

	.no_reconnect        = 1,

	.size_of_priv     = sizeof(struct az6007_device_state),

	.identify_state		= az6007_identify_state,

	.num_adapters = 1,

	.adapter = {

		{

			//.caps             = DVB_USB_ADAP_RECEIVES_204_BYTE_TS,

			.streaming_ctrl   = az6007_streaming_ctrl,

			.frontend_attach  = az6007_frontend_attach,

			/* parameter for the MPEG2-data transfer */

			.stream = {

				.type = USB_BULK,

				.count = 10,

				.endpoint = 0x02,

				.u = {

					.bulk = {

						.buffersize = 4096,

					}

				}

			},

			.size_of_priv     = 0,//sizeof(struct az6007_state),

		}

	},

	//.power_ctrl       = az6007_power_ctrl,

	.read_mac_address = az6007_read_mac_addr,

	.rc_key_map       = az6007_rc_keys,

	.rc_key_map_size  = ARRAY_SIZE(az6007_rc_keys),

	.rc_interval      = 400,

	.rc_query         = az6007_rc_query,

	.i2c_algo         = &az6007_i2c_algo,

	.num_device_descs = 2,

	.devices = {

		{ .name = "AzureWave DTV StarBox DVB-T/C USB2.0 (az6007)",

		  .cold_ids = { &az6007_usb_table[0], NULL },

		  .warm_ids = { NULL },

		},

		{ .name = "TerraTec DTV StarBox DVB-T/C USB2.0 (az6007)",

		  .cold_ids = { &az6007_usb_table[1], NULL },

		  .warm_ids = { NULL },

		},

 		{ NULL },

	}

};

/* usb specific object needed to register this driver with the usb subsystem */

static struct usb_driver az6007_usb_driver = {

	.name		= "dvb_usb_az6007",

	.probe 		= az6007_usb_probe,

	.disconnect = dvb_usb_device_exit,

	//.disconnect 	= az6007_usb_disconnect,

	.id_table 	= az6007_usb_table,

};

/* module stuff */

static int __init az6007_usb_module_init(void)

{

	int result;

	info("henry :: az6007 usb module init");

	if ((result = usb_register(&az6007_usb_driver))) {

		err("usb_register failed. (%d)",result);

		return result;

	}

	return 0;

}

static void __exit az6007_usb_module_exit(void)

{

	/* deregister this driver from the USB subsystem */

	info("henry :: az6007 usb module exit");

	usb_deregister(&az6007_usb_driver);

}

module_init(az6007_usb_module_init);

module_exit(az6007_usb_module_exit);

MODULE_AUTHOR("Henry Wang <Henry.wang@AzureWave.com>");

MODULE_DESCRIPTION("Driver for AzureWave 6007 DVB-C/T USB2.0 and clones");

MODULE_VERSION("1.0");

MODULE_LICENSE("GPL");

#ifndef _DVB_USB_AZ6007_H_

#define _DVB_USB_AZ6007_H_

#define DVB_USB_LOG_PREFIX "az6007"

#include "dvb-usb.h"

extern int dvb_usb_az6007_debug;

#define deb_info(args...) dprintk(dvb_usb_az6007_debug,0x01,args)

#define deb_xfer(args...) dprintk(dvb_usb_az6007_debug,0x02,args)

#define deb_rc(args...)   dprintk(dvb_usb_az6007_debug,0x04,args)

#define deb_fe(args...)   dprintk(dvb_usb_az6007_debug,0x08,args)

extern int vp702x_usb_out_op(struct dvb_usb_device *d, u8 *o, int olen, u8 *i, int ilen, int msec);

extern int vp702x_usb_in_op(struct dvb_usb_device *d, u8 req, u16 value, u16 index, u8 *b, int blen);

#endif