1 /*
2  * Hauppauge HD PVR USB driver
3  *
4  * Copyright (C) 2001-2004 Greg Kroah-Hartman (greg@kroah.com)
5  * Copyright (C) 2008      Janne Grunau (j@jannau.net)
6  * Copyright (C) 2008      John Poet
7  *
8  *	This program is free software; you can redistribute it and/or
9  *	modify it under the terms of the GNU General Public License as
10  *	published by the Free Software Foundation, version 2.
11  *
12  */
13 
14 #include <linux/kernel.h>
15 #include <linux/errno.h>
16 #include <linux/init.h>
17 #include <linux/slab.h>
18 #include <linux/module.h>
19 #include <linux/uaccess.h>
20 #include <linux/atomic.h>
21 #include <linux/usb.h>
22 #include <linux/mutex.h>
23 #include <linux/i2c.h>
24 
25 #include <linux/videodev2.h>
26 #include <media/v4l2-dev.h>
27 #include <media/v4l2-common.h>
28 
29 #include "hdpvr.h"
30 
31 static int video_nr[HDPVR_MAX] = {[0 ... (HDPVR_MAX - 1)] = UNSET};
32 module_param_array(video_nr, int, NULL, 0);
33 MODULE_PARM_DESC(video_nr, "video device number (-1=Auto)");
34 
35 /* holds the number of currently registered devices */
36 static atomic_t dev_nr = ATOMIC_INIT(-1);
37 
38 int hdpvr_debug;
39 module_param(hdpvr_debug, int, S_IRUGO|S_IWUSR);
40 MODULE_PARM_DESC(hdpvr_debug, "enable debugging output");
41 
42 static uint default_video_input = HDPVR_VIDEO_INPUTS;
43 module_param(default_video_input, uint, S_IRUGO|S_IWUSR);
44 MODULE_PARM_DESC(default_video_input, "default video input: 0=Component / "
45 		 "1=S-Video / 2=Composite");
46 
47 static uint default_audio_input = HDPVR_AUDIO_INPUTS;
48 module_param(default_audio_input, uint, S_IRUGO|S_IWUSR);
49 MODULE_PARM_DESC(default_audio_input, "default audio input: 0=RCA back / "
50 		 "1=RCA front / 2=S/PDIF");
51 
52 static bool boost_audio;
53 module_param(boost_audio, bool, S_IRUGO|S_IWUSR);
54 MODULE_PARM_DESC(boost_audio, "boost the audio signal");
55 
56 
57 /* table of devices that work with this driver */
58 static struct usb_device_id hdpvr_table[] = {
59 	{ USB_DEVICE(HD_PVR_VENDOR_ID, HD_PVR_PRODUCT_ID) },
60 	{ USB_DEVICE(HD_PVR_VENDOR_ID, HD_PVR_PRODUCT_ID1) },
61 	{ USB_DEVICE(HD_PVR_VENDOR_ID, HD_PVR_PRODUCT_ID2) },
62 	{ USB_DEVICE(HD_PVR_VENDOR_ID, HD_PVR_PRODUCT_ID3) },
63 	{ USB_DEVICE(HD_PVR_VENDOR_ID, HD_PVR_PRODUCT_ID4) },
64 	{ }					/* Terminating entry */
65 };
66 MODULE_DEVICE_TABLE(usb, hdpvr_table);
67 
68 
69 void hdpvr_delete(struct hdpvr_device *dev)
70 {
71 	hdpvr_free_buffers(dev);
72 
73 	if (dev->video_dev)
74 		video_device_release(dev->video_dev);
75 
76 	usb_put_dev(dev->udev);
77 }
78 
79 static void challenge(u8 *bytes)
80 {
81 	u64 *i64P, tmp64;
82 	uint i, idx;
83 
84 	for (idx = 0; idx < 32; ++idx) {
85 
86 		if (idx & 0x3)
87 			bytes[(idx >> 3) + 3] = bytes[(idx >> 2) & 0x3];
88 
89 		switch (idx & 0x3) {
90 		case 0x3:
91 			bytes[2] += bytes[3] * 4 + bytes[4] + bytes[5];
92 			bytes[4] += bytes[(idx & 0x1) * 2] * 9 + 9;
93 			break;
94 		case 0x1:
95 			bytes[0] *= 8;
96 			bytes[0] += 7*idx + 4;
97 			bytes[6] += bytes[3] * 3;
98 			break;
99 		case 0x0:
100 			bytes[3 - (idx >> 3)] = bytes[idx >> 2];
101 			bytes[5] += bytes[6] * 3;
102 			for (i = 0; i < 3; i++)
103 				bytes[3] *= bytes[3] + 1;
104 			break;
105 		case 0x2:
106 			for (i = 0; i < 3; i++)
107 				bytes[1] *= bytes[6] + 1;
108 			for (i = 0; i < 3; i++) {
109 				i64P = (u64 *)bytes;
110 				tmp64 = le64_to_cpup(i64P);
111 				tmp64 <<= bytes[7] & 0x0f;
112 				*i64P += cpu_to_le64(tmp64);
113 			}
114 			break;
115 		}
116 	}
117 }
118 
119 /* try to init the device like the windows driver */
120 static int device_authorization(struct hdpvr_device *dev)
121 {
122 
123 	int ret, retval = -ENOMEM;
124 	char request_type = 0x38, rcv_request = 0x81;
125 	char *response;
126 #ifdef HDPVR_DEBUG
127 	size_t buf_size = 46;
128 	char *print_buf = kzalloc(5*buf_size+1, GFP_KERNEL);
129 	if (!print_buf) {
130 		v4l2_err(&dev->v4l2_dev, "Out of memory\n");
131 		return retval;
132 	}
133 #endif
134 
135 	mutex_lock(&dev->usbc_mutex);
136 	ret = usb_control_msg(dev->udev,
137 			      usb_rcvctrlpipe(dev->udev, 0),
138 			      rcv_request, 0x80 | request_type,
139 			      0x0400, 0x0003,
140 			      dev->usbc_buf, 46,
141 			      10000);
142 	if (ret != 46) {
143 		v4l2_err(&dev->v4l2_dev,
144 			 "unexpected answer of status request, len %d\n", ret);
145 		goto unlock;
146 	}
147 #ifdef HDPVR_DEBUG
148 	else {
149 		hex_dump_to_buffer(dev->usbc_buf, 46, 16, 1, print_buf,
150 				   5*buf_size+1, 0);
151 		v4l2_dbg(MSG_INFO, hdpvr_debug, &dev->v4l2_dev,
152 			 "Status request returned, len %d: %s\n",
153 			 ret, print_buf);
154 	}
155 #endif
156 
157 	dev->fw_ver = dev->usbc_buf[1];
158 
159 	v4l2_info(&dev->v4l2_dev, "firmware version 0x%x dated %s\n",
160 			  dev->fw_ver, &dev->usbc_buf[2]);
161 
162 	if (dev->fw_ver > 0x15) {
163 		dev->options.brightness	= 0x80;
164 		dev->options.contrast	= 0x40;
165 		dev->options.hue	= 0xf;
166 		dev->options.saturation	= 0x40;
167 		dev->options.sharpness	= 0x80;
168 	}
169 
170 	switch (dev->fw_ver) {
171 	case HDPVR_FIRMWARE_VERSION:
172 		dev->flags &= ~HDPVR_FLAG_AC3_CAP;
173 		break;
174 	case HDPVR_FIRMWARE_VERSION_AC3:
175 	case HDPVR_FIRMWARE_VERSION_0X12:
176 	case HDPVR_FIRMWARE_VERSION_0X15:
177 	case HDPVR_FIRMWARE_VERSION_0X1E:
178 		dev->flags |= HDPVR_FLAG_AC3_CAP;
179 		break;
180 	default:
181 		v4l2_info(&dev->v4l2_dev, "untested firmware, the driver might"
182 			  " not work.\n");
183 		if (dev->fw_ver >= HDPVR_FIRMWARE_VERSION_AC3)
184 			dev->flags |= HDPVR_FLAG_AC3_CAP;
185 		else
186 			dev->flags &= ~HDPVR_FLAG_AC3_CAP;
187 	}
188 
189 	response = dev->usbc_buf+38;
190 #ifdef HDPVR_DEBUG
191 	hex_dump_to_buffer(response, 8, 16, 1, print_buf, 5*buf_size+1, 0);
192 	v4l2_dbg(MSG_INFO, hdpvr_debug, &dev->v4l2_dev, "challenge: %s\n",
193 		 print_buf);
194 #endif
195 	challenge(response);
196 #ifdef HDPVR_DEBUG
197 	hex_dump_to_buffer(response, 8, 16, 1, print_buf, 5*buf_size+1, 0);
198 	v4l2_dbg(MSG_INFO, hdpvr_debug, &dev->v4l2_dev, " response: %s\n",
199 		 print_buf);
200 	kfree(print_buf);
201 #endif
202 
203 	msleep(100);
204 	ret = usb_control_msg(dev->udev,
205 			      usb_sndctrlpipe(dev->udev, 0),
206 			      0xd1, 0x00 | request_type,
207 			      0x0000, 0x0000,
208 			      response, 8,
209 			      10000);
210 	v4l2_dbg(MSG_INFO, hdpvr_debug, &dev->v4l2_dev,
211 		 "magic request returned %d\n", ret);
212 
213 	retval = ret != 8;
214 unlock:
215 	mutex_unlock(&dev->usbc_mutex);
216 	return retval;
217 }
218 
219 static int hdpvr_device_init(struct hdpvr_device *dev)
220 {
221 	int ret;
222 	u8 *buf;
223 	struct hdpvr_video_info *vidinf;
224 
225 	if (device_authorization(dev))
226 		return -EACCES;
227 
228 	/* default options for init */
229 	hdpvr_set_options(dev);
230 
231 	/* set filter options */
232 	mutex_lock(&dev->usbc_mutex);
233 	buf = dev->usbc_buf;
234 	buf[0] = 0x03; buf[1] = 0x03; buf[2] = 0x00; buf[3] = 0x00;
235 	ret = usb_control_msg(dev->udev,
236 			      usb_sndctrlpipe(dev->udev, 0),
237 			      0x01, 0x38,
238 			      CTRL_LOW_PASS_FILTER_VALUE, CTRL_DEFAULT_INDEX,
239 			      buf, 4,
240 			      1000);
241 	v4l2_dbg(MSG_INFO, hdpvr_debug, &dev->v4l2_dev,
242 		 "control request returned %d\n", ret);
243 	mutex_unlock(&dev->usbc_mutex);
244 
245 	vidinf = get_video_info(dev);
246 	if (!vidinf)
247 		v4l2_dbg(MSG_INFO, hdpvr_debug, &dev->v4l2_dev,
248 			"no valid video signal or device init failed\n");
249 	else
250 		kfree(vidinf);
251 
252 	/* enable fan and bling leds */
253 	mutex_lock(&dev->usbc_mutex);
254 	buf[0] = 0x1;
255 	ret = usb_control_msg(dev->udev,
256 			      usb_sndctrlpipe(dev->udev, 0),
257 			      0xd4, 0x38, 0, 0, buf, 1,
258 			      1000);
259 	v4l2_dbg(MSG_INFO, hdpvr_debug, &dev->v4l2_dev,
260 		 "control request returned %d\n", ret);
261 
262 	/* boost analog audio */
263 	buf[0] = boost_audio;
264 	ret = usb_control_msg(dev->udev,
265 			      usb_sndctrlpipe(dev->udev, 0),
266 			      0xd5, 0x38, 0, 0, buf, 1,
267 			      1000);
268 	v4l2_dbg(MSG_INFO, hdpvr_debug, &dev->v4l2_dev,
269 		 "control request returned %d\n", ret);
270 	mutex_unlock(&dev->usbc_mutex);
271 
272 	dev->status = STATUS_IDLE;
273 	return 0;
274 }
275 
276 static const struct hdpvr_options hdpvr_default_options = {
277 	.video_std	= HDPVR_60HZ,
278 	.video_input	= HDPVR_COMPONENT,
279 	.audio_input	= HDPVR_RCA_BACK,
280 	.bitrate	= 65, /* 6 mbps */
281 	.peak_bitrate	= 90, /* 9 mbps */
282 	.bitrate_mode	= HDPVR_CONSTANT,
283 	.gop_mode	= HDPVR_SIMPLE_IDR_GOP,
284 	.audio_codec	= V4L2_MPEG_AUDIO_ENCODING_AAC,
285 	/* original picture controls for firmware version <= 0x15 */
286 	/* updated in device_authorization() for newer firmware */
287 	.brightness	= 0x86,
288 	.contrast	= 0x80,
289 	.hue		= 0x80,
290 	.saturation	= 0x80,
291 	.sharpness	= 0x80,
292 };
293 
294 static int hdpvr_probe(struct usb_interface *interface,
295 		       const struct usb_device_id *id)
296 {
297 	struct hdpvr_device *dev;
298 	struct usb_host_interface *iface_desc;
299 	struct usb_endpoint_descriptor *endpoint;
300 	struct i2c_client *client;
301 	size_t buffer_size;
302 	int i;
303 	int retval = -ENOMEM;
304 
305 	/* allocate memory for our device state and initialize it */
306 	dev = kzalloc(sizeof(*dev), GFP_KERNEL);
307 	if (!dev) {
308 		dev_err(&interface->dev, "Out of memory\n");
309 		goto error;
310 	}
311 
312 	dev->workqueue = 0;
313 
314 	/* register v4l2_device early so it can be used for printks */
315 	if (v4l2_device_register(&interface->dev, &dev->v4l2_dev)) {
316 		dev_err(&interface->dev, "v4l2_device_register failed\n");
317 		goto error;
318 	}
319 
320 	mutex_init(&dev->io_mutex);
321 	mutex_init(&dev->i2c_mutex);
322 	mutex_init(&dev->usbc_mutex);
323 	dev->usbc_buf = kmalloc(64, GFP_KERNEL);
324 	if (!dev->usbc_buf) {
325 		v4l2_err(&dev->v4l2_dev, "Out of memory\n");
326 		goto error;
327 	}
328 
329 	init_waitqueue_head(&dev->wait_buffer);
330 	init_waitqueue_head(&dev->wait_data);
331 
332 	dev->workqueue = create_singlethread_workqueue("hdpvr_buffer");
333 	if (!dev->workqueue)
334 		goto error;
335 
336 	/* init video transfer queues */
337 	INIT_LIST_HEAD(&dev->free_buff_list);
338 	INIT_LIST_HEAD(&dev->rec_buff_list);
339 
340 	dev->options = hdpvr_default_options;
341 
342 	if (default_video_input < HDPVR_VIDEO_INPUTS)
343 		dev->options.video_input = default_video_input;
344 
345 	if (default_audio_input < HDPVR_AUDIO_INPUTS) {
346 		dev->options.audio_input = default_audio_input;
347 		if (default_audio_input == HDPVR_SPDIF)
348 			dev->options.audio_codec =
349 				V4L2_MPEG_AUDIO_ENCODING_AC3;
350 	}
351 
352 	dev->udev = usb_get_dev(interface_to_usbdev(interface));
353 
354 	/* set up the endpoint information */
355 	/* use only the first bulk-in and bulk-out endpoints */
356 	iface_desc = interface->cur_altsetting;
357 	for (i = 0; i < iface_desc->desc.bNumEndpoints; ++i) {
358 		endpoint = &iface_desc->endpoint[i].desc;
359 
360 		if (!dev->bulk_in_endpointAddr &&
361 		    usb_endpoint_is_bulk_in(endpoint)) {
362 			/* USB interface description is buggy, reported max
363 			 * packet size is 512 bytes, windows driver uses 8192 */
364 			buffer_size = 8192;
365 			dev->bulk_in_size = buffer_size;
366 			dev->bulk_in_endpointAddr = endpoint->bEndpointAddress;
367 		}
368 
369 	}
370 	if (!dev->bulk_in_endpointAddr) {
371 		v4l2_err(&dev->v4l2_dev, "Could not find bulk-in endpoint\n");
372 		goto error;
373 	}
374 
375 	/* init the device */
376 	if (hdpvr_device_init(dev)) {
377 		v4l2_err(&dev->v4l2_dev, "device init failed\n");
378 		goto error;
379 	}
380 
381 	mutex_lock(&dev->io_mutex);
382 	if (hdpvr_alloc_buffers(dev, NUM_BUFFERS)) {
383 		mutex_unlock(&dev->io_mutex);
384 		v4l2_err(&dev->v4l2_dev,
385 			 "allocating transfer buffers failed\n");
386 		goto error;
387 	}
388 	mutex_unlock(&dev->io_mutex);
389 
390 #if IS_ENABLED(CONFIG_I2C)
391 	retval = hdpvr_register_i2c_adapter(dev);
392 	if (retval < 0) {
393 		v4l2_err(&dev->v4l2_dev, "i2c adapter register failed\n");
394 		goto error;
395 	}
396 
397 	client = hdpvr_register_ir_rx_i2c(dev);
398 	if (!client) {
399 		v4l2_err(&dev->v4l2_dev, "i2c IR RX device register failed\n");
400 		retval = -ENODEV;
401 		goto reg_fail;
402 	}
403 
404 	client = hdpvr_register_ir_tx_i2c(dev);
405 	if (!client) {
406 		v4l2_err(&dev->v4l2_dev, "i2c IR TX device register failed\n");
407 		retval = -ENODEV;
408 		goto reg_fail;
409 	}
410 #endif
411 
412 	retval = hdpvr_register_videodev(dev, &interface->dev,
413 				    video_nr[atomic_inc_return(&dev_nr)]);
414 	if (retval < 0) {
415 		v4l2_err(&dev->v4l2_dev, "registering videodev failed\n");
416 		goto error;
417 	}
418 
419 	/* let the user know what node this device is now attached to */
420 	v4l2_info(&dev->v4l2_dev, "device now attached to %s\n",
421 		  video_device_node_name(dev->video_dev));
422 	return 0;
423 
424 reg_fail:
425 #if IS_ENABLED(CONFIG_I2C)
426 	i2c_del_adapter(&dev->i2c_adapter);
427 #endif
428 error:
429 	if (dev) {
430 		/* Destroy single thread */
431 		if (dev->workqueue)
432 			destroy_workqueue(dev->workqueue);
433 		/* this frees allocated memory */
434 		hdpvr_delete(dev);
435 	}
436 	return retval;
437 }
438 
439 static void hdpvr_disconnect(struct usb_interface *interface)
440 {
441 	struct hdpvr_device *dev = to_hdpvr_dev(usb_get_intfdata(interface));
442 
443 	v4l2_info(&dev->v4l2_dev, "device %s disconnected\n",
444 		  video_device_node_name(dev->video_dev));
445 	/* prevent more I/O from starting and stop any ongoing */
446 	mutex_lock(&dev->io_mutex);
447 	dev->status = STATUS_DISCONNECTED;
448 	wake_up_interruptible(&dev->wait_data);
449 	wake_up_interruptible(&dev->wait_buffer);
450 	mutex_unlock(&dev->io_mutex);
451 	v4l2_device_disconnect(&dev->v4l2_dev);
452 	msleep(100);
453 	flush_workqueue(dev->workqueue);
454 	mutex_lock(&dev->io_mutex);
455 	hdpvr_cancel_queue(dev);
456 	mutex_unlock(&dev->io_mutex);
457 #if IS_ENABLED(CONFIG_I2C)
458 	i2c_del_adapter(&dev->i2c_adapter);
459 #endif
460 	video_unregister_device(dev->video_dev);
461 	atomic_dec(&dev_nr);
462 }
463 
464 
465 static struct usb_driver hdpvr_usb_driver = {
466 	.name =		"hdpvr",
467 	.probe =	hdpvr_probe,
468 	.disconnect =	hdpvr_disconnect,
469 	.id_table =	hdpvr_table,
470 };
471 
472 module_usb_driver(hdpvr_usb_driver);
473 
474 MODULE_LICENSE("GPL");
475 MODULE_VERSION("0.2.1");
476 MODULE_AUTHOR("Janne Grunau");
477 MODULE_DESCRIPTION("Hauppauge HD PVR driver");
478