Merge branch 'upstream-linus' of git://git.kernel.org/pub/scm/linux/kernel/git/mfashe...
[linux-2.6] / drivers / media / video / ivtv / ivtv-fileops.c
1 /*
2     file operation functions
3     Copyright (C) 2003-2004  Kevin Thayer <nufan_wfk at yahoo.com>
4     Copyright (C) 2004  Chris Kennedy <c@groovy.org>
5     Copyright (C) 2005-2007  Hans Verkuil <hverkuil@xs4all.nl>
6
7     This program is free software; you can redistribute it and/or modify
8     it under the terms of the GNU General Public License as published by
9     the Free Software Foundation; either version 2 of the License, or
10     (at your option) any later version.
11
12     This program is distributed in the hope that it will be useful,
13     but WITHOUT ANY WARRANTY; without even the implied warranty of
14     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15     GNU General Public License for more details.
16
17     You should have received a copy of the GNU General Public License
18     along with this program; if not, write to the Free Software
19     Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
20  */
21
22 #include "ivtv-driver.h"
23 #include "ivtv-fileops.h"
24 #include "ivtv-i2c.h"
25 #include "ivtv-queue.h"
26 #include "ivtv-udma.h"
27 #include "ivtv-irq.h"
28 #include "ivtv-vbi.h"
29 #include "ivtv-mailbox.h"
30 #include "ivtv-routing.h"
31 #include "ivtv-streams.h"
32 #include "ivtv-yuv.h"
33 #include "ivtv-ioctl.h"
34 #include "ivtv-cards.h"
35 #include <media/saa7115.h>
36
37 /* This function tries to claim the stream for a specific file descriptor.
38    If no one else is using this stream then the stream is claimed and
39    associated VBI streams are also automatically claimed.
40    Possible error returns: -EBUSY if someone else has claimed
41    the stream or 0 on success. */
42 static int ivtv_claim_stream(struct ivtv_open_id *id, int type)
43 {
44         struct ivtv *itv = id->itv;
45         struct ivtv_stream *s = &itv->streams[type];
46         struct ivtv_stream *s_vbi;
47         int vbi_type;
48
49         if (test_and_set_bit(IVTV_F_S_CLAIMED, &s->s_flags)) {
50                 /* someone already claimed this stream */
51                 if (s->id == id->open_id) {
52                         /* yes, this file descriptor did. So that's OK. */
53                         return 0;
54                 }
55                 if (s->id == -1 && (type == IVTV_DEC_STREAM_TYPE_VBI ||
56                                          type == IVTV_ENC_STREAM_TYPE_VBI)) {
57                         /* VBI is handled already internally, now also assign
58                            the file descriptor to this stream for external
59                            reading of the stream. */
60                         s->id = id->open_id;
61                         IVTV_DEBUG_INFO("Start Read VBI\n");
62                         return 0;
63                 }
64                 /* someone else is using this stream already */
65                 IVTV_DEBUG_INFO("Stream %d is busy\n", type);
66                 return -EBUSY;
67         }
68         s->id = id->open_id;
69         if (type == IVTV_DEC_STREAM_TYPE_VBI) {
70                 /* Enable reinsertion interrupt */
71                 ivtv_clear_irq_mask(itv, IVTV_IRQ_DEC_VBI_RE_INSERT);
72         }
73
74         /* IVTV_DEC_STREAM_TYPE_MPG needs to claim IVTV_DEC_STREAM_TYPE_VBI,
75            IVTV_ENC_STREAM_TYPE_MPG needs to claim IVTV_ENC_STREAM_TYPE_VBI
76            (provided VBI insertion is on and sliced VBI is selected), for all
77            other streams we're done */
78         if (type == IVTV_DEC_STREAM_TYPE_MPG) {
79                 vbi_type = IVTV_DEC_STREAM_TYPE_VBI;
80         } else if (type == IVTV_ENC_STREAM_TYPE_MPG &&
81                    itv->vbi.insert_mpeg && !ivtv_raw_vbi(itv)) {
82                 vbi_type = IVTV_ENC_STREAM_TYPE_VBI;
83         } else {
84                 return 0;
85         }
86         s_vbi = &itv->streams[vbi_type];
87
88         if (!test_and_set_bit(IVTV_F_S_CLAIMED, &s_vbi->s_flags)) {
89                 /* Enable reinsertion interrupt */
90                 if (vbi_type == IVTV_DEC_STREAM_TYPE_VBI)
91                         ivtv_clear_irq_mask(itv, IVTV_IRQ_DEC_VBI_RE_INSERT);
92         }
93         /* mark that it is used internally */
94         set_bit(IVTV_F_S_INTERNAL_USE, &s_vbi->s_flags);
95         return 0;
96 }
97
98 /* This function releases a previously claimed stream. It will take into
99    account associated VBI streams. */
100 void ivtv_release_stream(struct ivtv_stream *s)
101 {
102         struct ivtv *itv = s->itv;
103         struct ivtv_stream *s_vbi;
104
105         s->id = -1;
106         if ((s->type == IVTV_DEC_STREAM_TYPE_VBI || s->type == IVTV_ENC_STREAM_TYPE_VBI) &&
107                 test_bit(IVTV_F_S_INTERNAL_USE, &s->s_flags)) {
108                 /* this stream is still in use internally */
109                 return;
110         }
111         if (!test_and_clear_bit(IVTV_F_S_CLAIMED, &s->s_flags)) {
112                 IVTV_DEBUG_WARN("Release stream %s not in use!\n", s->name);
113                 return;
114         }
115
116         ivtv_flush_queues(s);
117
118         /* disable reinsertion interrupt */
119         if (s->type == IVTV_DEC_STREAM_TYPE_VBI)
120                 ivtv_set_irq_mask(itv, IVTV_IRQ_DEC_VBI_RE_INSERT);
121
122         /* IVTV_DEC_STREAM_TYPE_MPG needs to release IVTV_DEC_STREAM_TYPE_VBI,
123            IVTV_ENC_STREAM_TYPE_MPG needs to release IVTV_ENC_STREAM_TYPE_VBI,
124            for all other streams we're done */
125         if (s->type == IVTV_DEC_STREAM_TYPE_MPG)
126                 s_vbi = &itv->streams[IVTV_DEC_STREAM_TYPE_VBI];
127         else if (s->type == IVTV_ENC_STREAM_TYPE_MPG)
128                 s_vbi = &itv->streams[IVTV_ENC_STREAM_TYPE_VBI];
129         else
130                 return;
131
132         /* clear internal use flag */
133         if (!test_and_clear_bit(IVTV_F_S_INTERNAL_USE, &s_vbi->s_flags)) {
134                 /* was already cleared */
135                 return;
136         }
137         if (s_vbi->id != -1) {
138                 /* VBI stream still claimed by a file descriptor */
139                 return;
140         }
141         /* disable reinsertion interrupt */
142         if (s_vbi->type == IVTV_DEC_STREAM_TYPE_VBI)
143                 ivtv_set_irq_mask(itv, IVTV_IRQ_DEC_VBI_RE_INSERT);
144         clear_bit(IVTV_F_S_CLAIMED, &s_vbi->s_flags);
145         ivtv_flush_queues(s_vbi);
146 }
147
148 static void ivtv_dualwatch(struct ivtv *itv)
149 {
150         struct v4l2_tuner vt;
151         u16 new_bitmap;
152         u16 new_stereo_mode;
153         const u16 stereo_mask = 0x0300;
154         const u16 dual = 0x0200;
155
156         new_stereo_mode = itv->params.audio_properties & stereo_mask;
157         memset(&vt, 0, sizeof(vt));
158         ivtv_call_i2c_clients(itv, VIDIOC_G_TUNER, &vt);
159         if (vt.audmode == V4L2_TUNER_MODE_LANG1_LANG2 && (vt.rxsubchans & V4L2_TUNER_SUB_LANG2))
160                 new_stereo_mode = dual;
161
162         if (new_stereo_mode == itv->dualwatch_stereo_mode)
163                 return;
164
165         new_bitmap = new_stereo_mode | (itv->params.audio_properties & ~stereo_mask);
166
167         IVTV_DEBUG_INFO("dualwatch: change stereo flag from 0x%x to 0x%x. new audio_bitmask=0x%ux\n",
168                            itv->dualwatch_stereo_mode, new_stereo_mode, new_bitmap);
169
170         if (ivtv_vapi(itv, CX2341X_ENC_SET_AUDIO_PROPERTIES, 1, new_bitmap) == 0) {
171                 itv->dualwatch_stereo_mode = new_stereo_mode;
172                 return;
173         }
174         IVTV_DEBUG_INFO("dualwatch: changing stereo flag failed\n");
175 }
176
177 static void ivtv_update_pgm_info(struct ivtv *itv)
178 {
179         u32 wr_idx = (read_enc(itv->pgm_info_offset) - itv->pgm_info_offset - 4) / 24;
180         int cnt;
181         int i = 0;
182
183         if (wr_idx >= itv->pgm_info_num) {
184                 IVTV_DEBUG_WARN("Invalid PGM index %d (>= %d)\n", wr_idx, itv->pgm_info_num);
185                 return;
186         }
187         cnt = (wr_idx + itv->pgm_info_num - itv->pgm_info_write_idx) % itv->pgm_info_num;
188         while (i < cnt) {
189                 int idx = (itv->pgm_info_write_idx + i) % itv->pgm_info_num;
190                 struct v4l2_enc_idx_entry *e = itv->pgm_info + idx;
191                 u32 addr = itv->pgm_info_offset + 4 + idx * 24;
192                 const int mapping[8] = { -1, V4L2_ENC_IDX_FRAME_I, V4L2_ENC_IDX_FRAME_P, -1,
193                         V4L2_ENC_IDX_FRAME_B, -1, -1, -1 };
194                                         // 1=I, 2=P, 4=B
195
196                 e->offset = read_enc(addr + 4) + ((u64)read_enc(addr + 8) << 32);
197                 if (e->offset > itv->mpg_data_received) {
198                         break;
199                 }
200                 e->offset += itv->vbi_data_inserted;
201                 e->length = read_enc(addr);
202                 e->pts = read_enc(addr + 16) + ((u64)(read_enc(addr + 20) & 1) << 32);
203                 e->flags = mapping[read_enc(addr + 12) & 7];
204                 i++;
205         }
206         itv->pgm_info_write_idx = (itv->pgm_info_write_idx + i) % itv->pgm_info_num;
207 }
208
209 static struct ivtv_buffer *ivtv_get_buffer(struct ivtv_stream *s, int non_block, int *err)
210 {
211         struct ivtv *itv = s->itv;
212         struct ivtv_stream *s_vbi = &itv->streams[IVTV_ENC_STREAM_TYPE_VBI];
213         struct ivtv_buffer *buf;
214         DEFINE_WAIT(wait);
215
216         *err = 0;
217         while (1) {
218                 if (s->type == IVTV_ENC_STREAM_TYPE_MPG) {
219                         /* Process pending program info updates and pending VBI data */
220                         ivtv_update_pgm_info(itv);
221
222                         if (time_after(jiffies,
223                                        itv->dualwatch_jiffies +
224                                        msecs_to_jiffies(1000))) {
225                                 itv->dualwatch_jiffies = jiffies;
226                                 ivtv_dualwatch(itv);
227                         }
228
229                         if (test_bit(IVTV_F_S_INTERNAL_USE, &s_vbi->s_flags) &&
230                             !test_bit(IVTV_F_S_APPL_IO, &s_vbi->s_flags)) {
231                                 while ((buf = ivtv_dequeue(s_vbi, &s_vbi->q_full))) {
232                                         /* byteswap and process VBI data */
233                                         ivtv_process_vbi_data(itv, buf, s_vbi->dma_pts, s_vbi->type);
234                                         ivtv_enqueue(s_vbi, buf, &s_vbi->q_free);
235                                 }
236                         }
237                         buf = &itv->vbi.sliced_mpeg_buf;
238                         if (buf->readpos != buf->bytesused) {
239                                 return buf;
240                         }
241                 }
242
243                 /* do we have leftover data? */
244                 buf = ivtv_dequeue(s, &s->q_io);
245                 if (buf)
246                         return buf;
247
248                 /* do we have new data? */
249                 buf = ivtv_dequeue(s, &s->q_full);
250                 if (buf) {
251                         if ((buf->b_flags & IVTV_F_B_NEED_BUF_SWAP) == 0)
252                                 return buf;
253                         buf->b_flags &= ~IVTV_F_B_NEED_BUF_SWAP;
254                         if (s->type == IVTV_ENC_STREAM_TYPE_MPG)
255                                 /* byteswap MPG data */
256                                 ivtv_buf_swap(buf);
257                         else if (s->type != IVTV_DEC_STREAM_TYPE_VBI) {
258                                 /* byteswap and process VBI data */
259                                 ivtv_process_vbi_data(itv, buf, s->dma_pts, s->type);
260                         }
261                         return buf;
262                 }
263
264                 /* return if end of stream */
265                 if (s->type != IVTV_DEC_STREAM_TYPE_VBI && !test_bit(IVTV_F_S_STREAMING, &s->s_flags)) {
266                         IVTV_DEBUG_INFO("EOS %s\n", s->name);
267                         return NULL;
268                 }
269
270                 /* return if file was opened with O_NONBLOCK */
271                 if (non_block) {
272                         *err = -EAGAIN;
273                         return NULL;
274                 }
275
276                 /* wait for more data to arrive */
277                 prepare_to_wait(&s->waitq, &wait, TASK_INTERRUPTIBLE);
278                 /* New buffers might have become available before we were added to the waitqueue */
279                 if (!s->q_full.buffers)
280                         schedule();
281                 finish_wait(&s->waitq, &wait);
282                 if (signal_pending(current)) {
283                         /* return if a signal was received */
284                         IVTV_DEBUG_INFO("User stopped %s\n", s->name);
285                         *err = -EINTR;
286                         return NULL;
287                 }
288         }
289 }
290
291 static void ivtv_setup_sliced_vbi_buf(struct ivtv *itv)
292 {
293         int idx = itv->vbi.inserted_frame % IVTV_VBI_FRAMES;
294
295         itv->vbi.sliced_mpeg_buf.buf = itv->vbi.sliced_mpeg_data[idx];
296         itv->vbi.sliced_mpeg_buf.bytesused = itv->vbi.sliced_mpeg_size[idx];
297         itv->vbi.sliced_mpeg_buf.readpos = 0;
298 }
299
300 static size_t ivtv_copy_buf_to_user(struct ivtv_stream *s, struct ivtv_buffer *buf,
301                 char __user *ubuf, size_t ucount)
302 {
303         struct ivtv *itv = s->itv;
304         size_t len = buf->bytesused - buf->readpos;
305
306         if (len > ucount) len = ucount;
307         if (itv->vbi.insert_mpeg && s->type == IVTV_ENC_STREAM_TYPE_MPG &&
308             !ivtv_raw_vbi(itv) && buf != &itv->vbi.sliced_mpeg_buf) {
309                 const char *start = buf->buf + buf->readpos;
310                 const char *p = start + 1;
311                 const u8 *q;
312                 u8 ch = itv->search_pack_header ? 0xba : 0xe0;
313                 int stuffing, i;
314
315                 while (start + len > p && (q = memchr(p, 0, start + len - p))) {
316                         p = q + 1;
317                         if ((char *)q + 15 >= buf->buf + buf->bytesused ||
318                             q[1] != 0 || q[2] != 1 || q[3] != ch) {
319                                 continue;
320                         }
321                         if (!itv->search_pack_header) {
322                                 if ((q[6] & 0xc0) != 0x80)
323                                         continue;
324                                 if (((q[7] & 0xc0) == 0x80 && (q[9] & 0xf0) == 0x20) ||
325                                     ((q[7] & 0xc0) == 0xc0 && (q[9] & 0xf0) == 0x30)) {
326                                         ch = 0xba;
327                                         itv->search_pack_header = 1;
328                                         p = q + 9;
329                                 }
330                                 continue;
331                         }
332                         stuffing = q[13] & 7;
333                         /* all stuffing bytes must be 0xff */
334                         for (i = 0; i < stuffing; i++)
335                                 if (q[14 + i] != 0xff)
336                                         break;
337                         if (i == stuffing && (q[4] & 0xc4) == 0x44 && (q[12] & 3) == 3 &&
338                                         q[14 + stuffing] == 0 && q[15 + stuffing] == 0 &&
339                                         q[16 + stuffing] == 1) {
340                                 itv->search_pack_header = 0;
341                                 len = (char *)q - start;
342                                 ivtv_setup_sliced_vbi_buf(itv);
343                                 break;
344                         }
345                 }
346         }
347         if (copy_to_user(ubuf, (u8 *)buf->buf + buf->readpos, len)) {
348                 IVTV_DEBUG_WARN("copy %zd bytes to user failed for %s\n", len, s->name);
349                 return -EFAULT;
350         }
351         /*IVTV_INFO("copied %lld %d %d %d %d %d vbi %d\n", itv->mpg_data_received, len, ucount,
352                         buf->readpos, buf->bytesused, buf->bytesused - buf->readpos - len,
353                         buf == &itv->vbi.sliced_mpeg_buf); */
354         buf->readpos += len;
355         if (s->type == IVTV_ENC_STREAM_TYPE_MPG && buf != &itv->vbi.sliced_mpeg_buf)
356                 itv->mpg_data_received += len;
357         return len;
358 }
359
360 static ssize_t ivtv_read(struct ivtv_stream *s, char __user *ubuf, size_t tot_count, int non_block)
361 {
362         struct ivtv *itv = s->itv;
363         size_t tot_written = 0;
364         int single_frame = 0;
365
366         if (atomic_read(&itv->capturing) == 0 && s->id == -1) {
367                 /* shouldn't happen */
368                 IVTV_DEBUG_WARN("Stream %s not initialized before read\n", s->name);
369                 return -EIO;
370         }
371
372         /* Each VBI buffer is one frame, the v4l2 API says that for VBI the frames should
373            arrive one-by-one, so make sure we never output more than one VBI frame at a time */
374         if (s->type == IVTV_DEC_STREAM_TYPE_VBI ||
375             (s->type == IVTV_ENC_STREAM_TYPE_VBI && !ivtv_raw_vbi(itv)))
376                 single_frame = 1;
377
378         for (;;) {
379                 struct ivtv_buffer *buf;
380                 int rc;
381
382                 buf = ivtv_get_buffer(s, non_block, &rc);
383                 /* if there is no data available... */
384                 if (buf == NULL) {
385                         /* if we got data, then return that regardless */
386                         if (tot_written)
387                                 break;
388                         /* EOS condition */
389                         if (rc == 0) {
390                                 clear_bit(IVTV_F_S_STREAMOFF, &s->s_flags);
391                                 clear_bit(IVTV_F_S_APPL_IO, &s->s_flags);
392                                 ivtv_release_stream(s);
393                         }
394                         /* set errno */
395                         return rc;
396                 }
397                 rc = ivtv_copy_buf_to_user(s, buf, ubuf + tot_written, tot_count - tot_written);
398                 if (buf != &itv->vbi.sliced_mpeg_buf) {
399                         ivtv_enqueue(s, buf, (buf->readpos == buf->bytesused) ? &s->q_free : &s->q_io);
400                 }
401                 else if (buf->readpos == buf->bytesused) {
402                         int idx = itv->vbi.inserted_frame % IVTV_VBI_FRAMES;
403                         itv->vbi.sliced_mpeg_size[idx] = 0;
404                         itv->vbi.inserted_frame++;
405                         itv->vbi_data_inserted += buf->bytesused;
406                 }
407                 if (rc < 0)
408                         return rc;
409                 tot_written += rc;
410
411                 if (tot_written == tot_count || single_frame)
412                         break;
413         }
414         return tot_written;
415 }
416
417 static ssize_t ivtv_read_pos(struct ivtv_stream *s, char __user *ubuf, size_t count,
418                         loff_t *pos, int non_block)
419 {
420         ssize_t rc = count ? ivtv_read(s, ubuf, count, non_block) : 0;
421         struct ivtv *itv = s->itv;
422
423         IVTV_DEBUG_HI_FILE("read %zd from %s, got %zd\n", count, s->name, rc);
424         if (rc > 0)
425                 pos += rc;
426         return rc;
427 }
428
429 int ivtv_start_capture(struct ivtv_open_id *id)
430 {
431         struct ivtv *itv = id->itv;
432         struct ivtv_stream *s = &itv->streams[id->type];
433         struct ivtv_stream *s_vbi;
434
435         if (s->type == IVTV_ENC_STREAM_TYPE_RAD ||
436             s->type == IVTV_DEC_STREAM_TYPE_MPG ||
437             s->type == IVTV_DEC_STREAM_TYPE_YUV ||
438             s->type == IVTV_DEC_STREAM_TYPE_VOUT) {
439                 /* you cannot read from these stream types. */
440                 return -EPERM;
441         }
442
443         /* Try to claim this stream. */
444         if (ivtv_claim_stream(id, s->type))
445                 return -EBUSY;
446
447         /* This stream does not need to start capturing */
448         if (s->type == IVTV_DEC_STREAM_TYPE_VBI) {
449                 set_bit(IVTV_F_S_APPL_IO, &s->s_flags);
450                 return 0;
451         }
452
453         /* If capture is already in progress, then we also have to
454            do nothing extra. */
455         if (test_bit(IVTV_F_S_STREAMOFF, &s->s_flags) || test_and_set_bit(IVTV_F_S_STREAMING, &s->s_flags)) {
456                 set_bit(IVTV_F_S_APPL_IO, &s->s_flags);
457                 return 0;
458         }
459
460         /* Start VBI capture if required */
461         s_vbi = &itv->streams[IVTV_ENC_STREAM_TYPE_VBI];
462         if (s->type == IVTV_ENC_STREAM_TYPE_MPG &&
463             test_bit(IVTV_F_S_INTERNAL_USE, &s_vbi->s_flags) &&
464             !test_and_set_bit(IVTV_F_S_STREAMING, &s_vbi->s_flags)) {
465                 /* Note: the IVTV_ENC_STREAM_TYPE_VBI is claimed
466                    automatically when the MPG stream is claimed.
467                    We only need to start the VBI capturing. */
468                 if (ivtv_start_v4l2_encode_stream(s_vbi)) {
469                         IVTV_DEBUG_WARN("VBI capture start failed\n");
470
471                         /* Failure, clean up and return an error */
472                         clear_bit(IVTV_F_S_STREAMING, &s_vbi->s_flags);
473                         clear_bit(IVTV_F_S_STREAMING, &s->s_flags);
474                         /* also releases the associated VBI stream */
475                         ivtv_release_stream(s);
476                         return -EIO;
477                 }
478                 IVTV_DEBUG_INFO("VBI insertion started\n");
479         }
480
481         /* Tell the card to start capturing */
482         if (!ivtv_start_v4l2_encode_stream(s)) {
483                 /* We're done */
484                 set_bit(IVTV_F_S_APPL_IO, &s->s_flags);
485                 /* Resume a possibly paused encoder */
486                 if (test_and_clear_bit(IVTV_F_I_ENC_PAUSED, &itv->i_flags))
487                         ivtv_vapi(itv, CX2341X_ENC_PAUSE_ENCODER, 1, 1);
488                 return 0;
489         }
490
491         /* failure, clean up */
492         IVTV_DEBUG_WARN("Failed to start capturing for stream %s\n", s->name);
493
494         /* Note: the IVTV_ENC_STREAM_TYPE_VBI is released
495            automatically when the MPG stream is released.
496            We only need to stop the VBI capturing. */
497         if (s->type == IVTV_ENC_STREAM_TYPE_MPG &&
498             test_bit(IVTV_F_S_STREAMING, &s_vbi->s_flags)) {
499                 ivtv_stop_v4l2_encode_stream(s_vbi, 0);
500                 clear_bit(IVTV_F_S_STREAMING, &s_vbi->s_flags);
501         }
502         clear_bit(IVTV_F_S_STREAMING, &s->s_flags);
503         ivtv_release_stream(s);
504         return -EIO;
505 }
506
507 ssize_t ivtv_v4l2_read(struct file * filp, char __user *buf, size_t count, loff_t * pos)
508 {
509         struct ivtv_open_id *id = filp->private_data;
510         struct ivtv *itv = id->itv;
511         struct ivtv_stream *s = &itv->streams[id->type];
512         int rc;
513
514         IVTV_DEBUG_HI_FILE("read %zd bytes from %s\n", count, s->name);
515
516         mutex_lock(&itv->serialize_lock);
517         rc = ivtv_start_capture(id);
518         mutex_unlock(&itv->serialize_lock);
519         if (rc)
520                 return rc;
521         return ivtv_read_pos(s, buf, count, pos, filp->f_flags & O_NONBLOCK);
522 }
523
524 int ivtv_start_decoding(struct ivtv_open_id *id, int speed)
525 {
526         struct ivtv *itv = id->itv;
527         struct ivtv_stream *s = &itv->streams[id->type];
528
529         if (atomic_read(&itv->decoding) == 0) {
530                 if (ivtv_claim_stream(id, s->type)) {
531                         /* someone else is using this stream already */
532                         IVTV_DEBUG_WARN("start decode, stream already claimed\n");
533                         return -EBUSY;
534                 }
535                 ivtv_start_v4l2_decode_stream(s, 0);
536         }
537         if (s->type == IVTV_DEC_STREAM_TYPE_MPG)
538                 return ivtv_set_speed(itv, speed);
539         return 0;
540 }
541
542 ssize_t ivtv_v4l2_write(struct file *filp, const char __user *user_buf, size_t count, loff_t *pos)
543 {
544         struct ivtv_open_id *id = filp->private_data;
545         struct ivtv *itv = id->itv;
546         struct ivtv_stream *s = &itv->streams[id->type];
547         struct yuv_playback_info *yi = &itv->yuv_info;
548         struct ivtv_buffer *buf;
549         struct ivtv_queue q;
550         int bytes_written = 0;
551         int mode;
552         int rc;
553         DEFINE_WAIT(wait);
554
555         IVTV_DEBUG_HI_FILE("write %zd bytes to %s\n", count, s->name);
556
557         if (s->type != IVTV_DEC_STREAM_TYPE_MPG &&
558             s->type != IVTV_DEC_STREAM_TYPE_YUV &&
559             s->type != IVTV_DEC_STREAM_TYPE_VOUT)
560                 /* not decoder streams */
561                 return -EPERM;
562
563         /* Try to claim this stream */
564         if (ivtv_claim_stream(id, s->type))
565                 return -EBUSY;
566
567         /* This stream does not need to start any decoding */
568         if (s->type == IVTV_DEC_STREAM_TYPE_VOUT) {
569                 int elems = count / sizeof(struct v4l2_sliced_vbi_data);
570
571                 set_bit(IVTV_F_S_APPL_IO, &s->s_flags);
572                 ivtv_write_vbi(itv, (const struct v4l2_sliced_vbi_data *)user_buf, elems);
573                 return elems * sizeof(struct v4l2_sliced_vbi_data);
574         }
575
576         mode = s->type == IVTV_DEC_STREAM_TYPE_MPG ? OUT_MPG : OUT_YUV;
577
578         if (ivtv_set_output_mode(itv, mode) != mode) {
579             ivtv_release_stream(s);
580             return -EBUSY;
581         }
582         ivtv_queue_init(&q);
583         set_bit(IVTV_F_S_APPL_IO, &s->s_flags);
584
585         /* Start decoder (returns 0 if already started) */
586         mutex_lock(&itv->serialize_lock);
587         rc = ivtv_start_decoding(id, itv->speed);
588         mutex_unlock(&itv->serialize_lock);
589         if (rc) {
590                 IVTV_DEBUG_WARN("Failed start decode stream %s\n", s->name);
591
592                 /* failure, clean up */
593                 clear_bit(IVTV_F_S_STREAMING, &s->s_flags);
594                 clear_bit(IVTV_F_S_APPL_IO, &s->s_flags);
595                 return rc;
596         }
597
598 retry:
599         /* If possible, just DMA the entire frame - Check the data transfer size
600         since we may get here before the stream has been fully set-up */
601         if (mode == OUT_YUV && s->q_full.length == 0 && itv->dma_data_req_size) {
602                 while (count >= itv->dma_data_req_size) {
603                         if (!ivtv_yuv_udma_stream_frame (itv, (void __user *)user_buf)) {
604                                 bytes_written += itv->dma_data_req_size;
605                                 user_buf += itv->dma_data_req_size;
606                                 count -= itv->dma_data_req_size;
607                         } else {
608                                 break;
609                         }
610                 }
611                 if (count == 0) {
612                         IVTV_DEBUG_HI_FILE("Wrote %d bytes to %s (%d)\n", bytes_written, s->name, s->q_full.bytesused);
613                         return bytes_written;
614                 }
615         }
616
617         for (;;) {
618                 /* Gather buffers */
619                 while (q.length - q.bytesused < count && (buf = ivtv_dequeue(s, &s->q_io)))
620                         ivtv_enqueue(s, buf, &q);
621                 while (q.length - q.bytesused < count && (buf = ivtv_dequeue(s, &s->q_free))) {
622                         ivtv_enqueue(s, buf, &q);
623                 }
624                 if (q.buffers)
625                         break;
626                 if (filp->f_flags & O_NONBLOCK)
627                         return -EAGAIN;
628                 prepare_to_wait(&s->waitq, &wait, TASK_INTERRUPTIBLE);
629                 /* New buffers might have become free before we were added to the waitqueue */
630                 if (!s->q_free.buffers)
631                         schedule();
632                 finish_wait(&s->waitq, &wait);
633                 if (signal_pending(current)) {
634                         IVTV_DEBUG_INFO("User stopped %s\n", s->name);
635                         return -EINTR;
636                 }
637         }
638
639         /* copy user data into buffers */
640         while ((buf = ivtv_dequeue(s, &q))) {
641                 /* yuv is a pain. Don't copy more data than needed for a single
642                    frame, otherwise we lose sync with the incoming stream */
643                 if (s->type == IVTV_DEC_STREAM_TYPE_YUV &&
644                     yi->stream_size + count > itv->dma_data_req_size)
645                         rc  = ivtv_buf_copy_from_user(s, buf, user_buf,
646                                 itv->dma_data_req_size - yi->stream_size);
647                 else
648                         rc = ivtv_buf_copy_from_user(s, buf, user_buf, count);
649
650                 /* Make sure we really got all the user data */
651                 if (rc < 0) {
652                         ivtv_queue_move(s, &q, NULL, &s->q_free, 0);
653                         return rc;
654                 }
655                 user_buf += rc;
656                 count -= rc;
657                 bytes_written += rc;
658
659                 if (s->type == IVTV_DEC_STREAM_TYPE_YUV) {
660                         yi->stream_size += rc;
661                         /* If we have a complete yuv frame, break loop now */
662                         if (yi->stream_size == itv->dma_data_req_size) {
663                                 ivtv_enqueue(s, buf, &s->q_full);
664                                 yi->stream_size = 0;
665                                 break;
666                         }
667                 }
668
669                 if (buf->bytesused != s->buf_size) {
670                         /* incomplete, leave in q_io for next time */
671                         ivtv_enqueue(s, buf, &s->q_io);
672                         break;
673                 }
674                 /* Byteswap MPEG buffer */
675                 if (s->type == IVTV_DEC_STREAM_TYPE_MPG)
676                         ivtv_buf_swap(buf);
677                 ivtv_enqueue(s, buf, &s->q_full);
678         }
679
680         if (test_bit(IVTV_F_S_NEEDS_DATA, &s->s_flags)) {
681                 if (s->q_full.length >= itv->dma_data_req_size) {
682                         int got_sig;
683
684                         if (mode == OUT_YUV)
685                                 ivtv_yuv_setup_stream_frame(itv);
686
687                         prepare_to_wait(&itv->dma_waitq, &wait, TASK_INTERRUPTIBLE);
688                         while (!(got_sig = signal_pending(current)) &&
689                                         test_bit(IVTV_F_S_DMA_PENDING, &s->s_flags)) {
690                                 schedule();
691                         }
692                         finish_wait(&itv->dma_waitq, &wait);
693                         if (got_sig) {
694                                 IVTV_DEBUG_INFO("User interrupted %s\n", s->name);
695                                 return -EINTR;
696                         }
697
698                         clear_bit(IVTV_F_S_NEEDS_DATA, &s->s_flags);
699                         ivtv_queue_move(s, &s->q_full, NULL, &s->q_predma, itv->dma_data_req_size);
700                         ivtv_dma_stream_dec_prepare(s, itv->dma_data_req_offset + IVTV_DECODER_OFFSET, 1);
701                 }
702         }
703         /* more user data is available, wait until buffers become free
704            to transfer the rest. */
705         if (count && !(filp->f_flags & O_NONBLOCK))
706                 goto retry;
707         IVTV_DEBUG_HI_FILE("Wrote %d bytes to %s (%d)\n", bytes_written, s->name, s->q_full.bytesused);
708         return bytes_written;
709 }
710
711 unsigned int ivtv_v4l2_dec_poll(struct file *filp, poll_table *wait)
712 {
713         struct ivtv_open_id *id = filp->private_data;
714         struct ivtv *itv = id->itv;
715         struct ivtv_stream *s = &itv->streams[id->type];
716         int res = 0;
717
718         /* add stream's waitq to the poll list */
719         IVTV_DEBUG_HI_FILE("Decoder poll\n");
720         poll_wait(filp, &s->waitq, wait);
721
722         set_bit(IVTV_F_I_EV_VSYNC_ENABLED, &itv->i_flags);
723         if (test_bit(IVTV_F_I_EV_VSYNC, &itv->i_flags) ||
724             test_bit(IVTV_F_I_EV_DEC_STOPPED, &itv->i_flags))
725                 res = POLLPRI;
726
727         /* Allow write if buffers are available for writing */
728         if (s->q_free.buffers)
729                 res |= POLLOUT | POLLWRNORM;
730         return res;
731 }
732
733 unsigned int ivtv_v4l2_enc_poll(struct file *filp, poll_table * wait)
734 {
735         struct ivtv_open_id *id = filp->private_data;
736         struct ivtv *itv = id->itv;
737         struct ivtv_stream *s = &itv->streams[id->type];
738         int eof = test_bit(IVTV_F_S_STREAMOFF, &s->s_flags);
739
740         /* Start a capture if there is none */
741         if (!eof && !test_bit(IVTV_F_S_STREAMING, &s->s_flags)) {
742                 int rc;
743
744                 mutex_lock(&itv->serialize_lock);
745                 rc = ivtv_start_capture(id);
746                 mutex_unlock(&itv->serialize_lock);
747                 if (rc) {
748                         IVTV_DEBUG_INFO("Could not start capture for %s (%d)\n",
749                                         s->name, rc);
750                         return POLLERR;
751                 }
752                 IVTV_DEBUG_FILE("Encoder poll started capture\n");
753         }
754
755         /* add stream's waitq to the poll list */
756         IVTV_DEBUG_HI_FILE("Encoder poll\n");
757         poll_wait(filp, &s->waitq, wait);
758
759         if (s->q_full.length || s->q_io.length)
760                 return POLLIN | POLLRDNORM;
761         if (eof)
762                 return POLLHUP;
763         return 0;
764 }
765
766 void ivtv_stop_capture(struct ivtv_open_id *id, int gop_end)
767 {
768         struct ivtv *itv = id->itv;
769         struct ivtv_stream *s = &itv->streams[id->type];
770
771         IVTV_DEBUG_FILE("close() of %s\n", s->name);
772
773         /* 'Unclaim' this stream */
774
775         /* Stop capturing */
776         if (test_bit(IVTV_F_S_STREAMING, &s->s_flags)) {
777                 struct ivtv_stream *s_vbi = &itv->streams[IVTV_ENC_STREAM_TYPE_VBI];
778
779                 IVTV_DEBUG_INFO("close stopping capture\n");
780                 /* Special case: a running VBI capture for VBI insertion
781                    in the mpeg stream. Need to stop that too. */
782                 if (id->type == IVTV_ENC_STREAM_TYPE_MPG &&
783                     test_bit(IVTV_F_S_STREAMING, &s_vbi->s_flags) &&
784                     !test_bit(IVTV_F_S_APPL_IO, &s_vbi->s_flags)) {
785                         IVTV_DEBUG_INFO("close stopping embedded VBI capture\n");
786                         ivtv_stop_v4l2_encode_stream(s_vbi, 0);
787                 }
788                 if ((id->type == IVTV_DEC_STREAM_TYPE_VBI ||
789                      id->type == IVTV_ENC_STREAM_TYPE_VBI) &&
790                     test_bit(IVTV_F_S_INTERNAL_USE, &s->s_flags)) {
791                         /* Also used internally, don't stop capturing */
792                         s->id = -1;
793                 }
794                 else {
795                         ivtv_stop_v4l2_encode_stream(s, gop_end);
796                 }
797         }
798         if (!gop_end) {
799                 clear_bit(IVTV_F_S_APPL_IO, &s->s_flags);
800                 clear_bit(IVTV_F_S_STREAMOFF, &s->s_flags);
801                 ivtv_release_stream(s);
802         }
803 }
804
805 static void ivtv_stop_decoding(struct ivtv_open_id *id, int flags, u64 pts)
806 {
807         struct ivtv *itv = id->itv;
808         struct ivtv_stream *s = &itv->streams[id->type];
809
810         IVTV_DEBUG_FILE("close() of %s\n", s->name);
811
812         /* Stop decoding */
813         if (test_bit(IVTV_F_S_STREAMING, &s->s_flags)) {
814                 IVTV_DEBUG_INFO("close stopping decode\n");
815
816                 ivtv_stop_v4l2_decode_stream(s, flags, pts);
817                 itv->output_mode = OUT_NONE;
818         }
819         clear_bit(IVTV_F_S_APPL_IO, &s->s_flags);
820         clear_bit(IVTV_F_S_STREAMOFF, &s->s_flags);
821         if (id->type == IVTV_DEC_STREAM_TYPE_YUV && test_bit(IVTV_F_I_DECODING_YUV, &itv->i_flags)) {
822                 /* Restore registers we've changed & clean up any mess we've made */
823                 ivtv_yuv_close(itv);
824         }
825         if (itv->output_mode == OUT_UDMA_YUV && id->yuv_frames)
826                 itv->output_mode = OUT_NONE;
827
828         itv->speed = 0;
829         clear_bit(IVTV_F_I_DEC_PAUSED, &itv->i_flags);
830         ivtv_release_stream(s);
831 }
832
833 int ivtv_v4l2_close(struct inode *inode, struct file *filp)
834 {
835         struct ivtv_open_id *id = filp->private_data;
836         struct ivtv *itv = id->itv;
837         struct ivtv_stream *s = &itv->streams[id->type];
838
839         IVTV_DEBUG_FILE("close %s\n", s->name);
840
841         v4l2_prio_close(&itv->prio, &id->prio);
842
843         /* Easy case first: this stream was never claimed by us */
844         if (s->id != id->open_id) {
845                 kfree(id);
846                 return 0;
847         }
848
849         /* 'Unclaim' this stream */
850
851         /* Stop radio */
852         mutex_lock(&itv->serialize_lock);
853         if (id->type == IVTV_ENC_STREAM_TYPE_RAD) {
854                 /* Closing radio device, return to TV mode */
855                 ivtv_mute(itv);
856                 /* Mark that the radio is no longer in use */
857                 clear_bit(IVTV_F_I_RADIO_USER, &itv->i_flags);
858                 /* Switch tuner to TV */
859                 ivtv_call_i2c_clients(itv, VIDIOC_S_STD, &itv->std);
860                 /* Select correct audio input (i.e. TV tuner or Line in) */
861                 ivtv_audio_set_io(itv);
862                 if (itv->hw_flags & IVTV_HW_SAA711X)
863                 {
864                         struct v4l2_crystal_freq crystal_freq;
865                         crystal_freq.freq = SAA7115_FREQ_32_11_MHZ;
866                         crystal_freq.flags = 0;
867                         ivtv_saa7115(itv, VIDIOC_INT_S_CRYSTAL_FREQ, &crystal_freq);
868                 }
869                 if (atomic_read(&itv->capturing) > 0) {
870                         /* Undo video mute */
871                         ivtv_vapi(itv, CX2341X_ENC_MUTE_VIDEO, 1,
872                                 itv->params.video_mute | (itv->params.video_mute_yuv << 8));
873                 }
874                 /* Done! Unmute and continue. */
875                 ivtv_unmute(itv);
876                 ivtv_release_stream(s);
877         } else if (s->type >= IVTV_DEC_STREAM_TYPE_MPG) {
878                 struct ivtv_stream *s_vout = &itv->streams[IVTV_DEC_STREAM_TYPE_VOUT];
879
880                 ivtv_stop_decoding(id, VIDEO_CMD_STOP_TO_BLACK | VIDEO_CMD_STOP_IMMEDIATELY, 0);
881
882                 /* If all output streams are closed, and if the user doesn't have
883                    IVTV_DEC_STREAM_TYPE_VOUT open, then disable CC on TV-out. */
884                 if (itv->output_mode == OUT_NONE && !test_bit(IVTV_F_S_APPL_IO, &s_vout->s_flags)) {
885                         /* disable CC on TV-out */
886                         ivtv_disable_cc(itv);
887                 }
888         } else {
889                 ivtv_stop_capture(id, 0);
890         }
891         kfree(id);
892         mutex_unlock(&itv->serialize_lock);
893         return 0;
894 }
895
896 static int ivtv_serialized_open(struct ivtv_stream *s, struct file *filp)
897 {
898         struct ivtv *itv = s->itv;
899         struct ivtv_open_id *item;
900
901         IVTV_DEBUG_FILE("open %s\n", s->name);
902
903         if (s->type == IVTV_DEC_STREAM_TYPE_MPG &&
904                 test_bit(IVTV_F_S_CLAIMED, &itv->streams[IVTV_DEC_STREAM_TYPE_YUV].s_flags))
905                 return -EBUSY;
906
907         if (s->type == IVTV_DEC_STREAM_TYPE_YUV &&
908                 test_bit(IVTV_F_S_CLAIMED, &itv->streams[IVTV_DEC_STREAM_TYPE_MPG].s_flags))
909                 return -EBUSY;
910
911         if (s->type == IVTV_DEC_STREAM_TYPE_YUV) {
912                 if (read_reg(0x82c) == 0) {
913                         IVTV_ERR("Tried to open YUV output device but need to send data to mpeg decoder before it can be used\n");
914                         /* return -ENODEV; */
915                 }
916                 ivtv_udma_alloc(itv);
917         }
918
919         /* Allocate memory */
920         item = kmalloc(sizeof(struct ivtv_open_id), GFP_KERNEL);
921         if (NULL == item) {
922                 IVTV_DEBUG_WARN("nomem on v4l2 open\n");
923                 return -ENOMEM;
924         }
925         item->itv = itv;
926         item->type = s->type;
927         v4l2_prio_open(&itv->prio, &item->prio);
928
929         item->open_id = itv->open_id++;
930         filp->private_data = item;
931
932         if (item->type == IVTV_ENC_STREAM_TYPE_RAD) {
933                 /* Try to claim this stream */
934                 if (ivtv_claim_stream(item, item->type)) {
935                         /* No, it's already in use */
936                         kfree(item);
937                         return -EBUSY;
938                 }
939
940                 if (!test_bit(IVTV_F_I_RADIO_USER, &itv->i_flags)) {
941                         if (atomic_read(&itv->capturing) > 0) {
942                                 /* switching to radio while capture is
943                                    in progress is not polite */
944                                 ivtv_release_stream(s);
945                                 kfree(item);
946                                 return -EBUSY;
947                         }
948                 }
949                 /* Mark that the radio is being used. */
950                 set_bit(IVTV_F_I_RADIO_USER, &itv->i_flags);
951                 /* We have the radio */
952                 ivtv_mute(itv);
953                 /* Switch tuner to radio */
954                 ivtv_call_i2c_clients(itv, AUDC_SET_RADIO, NULL);
955                 /* Select the correct audio input (i.e. radio tuner) */
956                 ivtv_audio_set_io(itv);
957                 if (itv->hw_flags & IVTV_HW_SAA711X)
958                 {
959                         struct v4l2_crystal_freq crystal_freq;
960                         crystal_freq.freq = SAA7115_FREQ_32_11_MHZ;
961                         crystal_freq.flags = SAA7115_FREQ_FL_APLL;
962                         ivtv_saa7115(itv, VIDIOC_INT_S_CRYSTAL_FREQ, &crystal_freq);
963                 }
964                 /* Done! Unmute and continue. */
965                 ivtv_unmute(itv);
966         }
967
968         /* YUV or MPG Decoding Mode? */
969         if (s->type == IVTV_DEC_STREAM_TYPE_MPG) {
970                 clear_bit(IVTV_F_I_DEC_YUV, &itv->i_flags);
971         } else if (s->type == IVTV_DEC_STREAM_TYPE_YUV) {
972                 set_bit(IVTV_F_I_DEC_YUV, &itv->i_flags);
973                 /* For yuv, we need to know the dma size before we start */
974                 itv->dma_data_req_size =
975                                 1080 * ((itv->yuv_info.v4l2_src_h + 31) & ~31);
976                 itv->yuv_info.stream_size = 0;
977         }
978         return 0;
979 }
980
981 int ivtv_v4l2_open(struct inode *inode, struct file *filp)
982 {
983         int res, x, y = 0;
984         struct ivtv *itv = NULL;
985         struct ivtv_stream *s = NULL;
986         int minor = iminor(inode);
987
988         /* Find which card this open was on */
989         spin_lock(&ivtv_cards_lock);
990         for (x = 0; itv == NULL && x < ivtv_cards_active; x++) {
991                 if (ivtv_cards[x] == NULL)
992                         continue;
993                 /* find out which stream this open was on */
994                 for (y = 0; y < IVTV_MAX_STREAMS; y++) {
995                         s = &ivtv_cards[x]->streams[y];
996                         if (s->v4l2dev && s->v4l2dev->minor == minor) {
997                                 itv = ivtv_cards[x];
998                                 break;
999                         }
1000                 }
1001         }
1002         spin_unlock(&ivtv_cards_lock);
1003
1004         if (itv == NULL) {
1005                 /* Couldn't find a device registered
1006                    on that minor, shouldn't happen! */
1007                 printk(KERN_WARNING "No ivtv device found on minor %d\n", minor);
1008                 return -ENXIO;
1009         }
1010
1011         mutex_lock(&itv->serialize_lock);
1012         if (ivtv_init_on_first_open(itv)) {
1013                 IVTV_ERR("Failed to initialize on minor %d\n", minor);
1014                 mutex_unlock(&itv->serialize_lock);
1015                 return -ENXIO;
1016         }
1017         res = ivtv_serialized_open(s, filp);
1018         mutex_unlock(&itv->serialize_lock);
1019         return res;
1020 }
1021
1022 void ivtv_mute(struct ivtv *itv)
1023 {
1024         if (atomic_read(&itv->capturing))
1025                 ivtv_vapi(itv, CX2341X_ENC_MUTE_AUDIO, 1, 1);
1026         IVTV_DEBUG_INFO("Mute\n");
1027 }
1028
1029 void ivtv_unmute(struct ivtv *itv)
1030 {
1031         if (atomic_read(&itv->capturing)) {
1032                 ivtv_msleep_timeout(100, 0);
1033                 ivtv_vapi(itv, CX2341X_ENC_MISC, 1, 12);
1034                 ivtv_vapi(itv, CX2341X_ENC_MUTE_AUDIO, 1, 0);
1035         }
1036         IVTV_DEBUG_INFO("Unmute\n");
1037 }