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