8992cd49f1e97ad58cb5ee98a04e067ebe536e35
[vlc.git] / modules / access / rtp / input.c
1 /**
2  * @file input.c
3  * @brief RTP packet input
4  */
5 /*****************************************************************************
6  * Copyright © 2008 Rémi Denis-Courmont
7  *
8  * This library is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU General Public License
10  * as published by the Free Software Foundation; either version 2.0
11  * of the License, or (at your option) any later version.
12  *
13  * This library is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16  * GNU General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with this library; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
21  ****************************************************************************/
22
23 #ifdef HAVE_CONFIG_H
24 # include <config.h>
25 #endif
26
27 #include <vlc_common.h>
28 #include <vlc_demux.h>
29 #include <vlc_block.h>
30 #include <vlc_network.h>
31
32 #include <unistd.h>
33 #ifdef HAVE_POLL
34 # include <poll.h>
35 #endif
36
37 #include "rtp.h"
38 #include <srtp.h>
39
40 /**
41  * Gets a datagram from the network.
42  * @param fd datagram file descriptor
43  * @return a block or NULL on fatal error (socket dead)
44  */
45 static block_t *rtp_dgram_recv (int fd)
46 {
47     block_t *block = block_Alloc (0xffff);
48     ssize_t len;
49
50     do
51     {
52         struct pollfd ufd = { .fd = fd, .events = POLLIN, };
53
54         block_cleanup_push (block);
55         poll (&ufd, 1, -1);
56         len = read (fd, block->p_buffer, block->i_buffer);
57         vlc_cleanup_pop ();
58
59         if ((len <= 0) && (ufd.revents & POLLHUP))
60         {   /* POLLHUP -> permanent (DCCP) socket error */
61             block_Release (block);
62             return NULL;
63         }
64     }
65     while (len == -1);
66
67     return block_Realloc (block, 0, len);
68 }
69
70
71 /**
72  * Gets a framed RTP packet.
73  * @param fd stream file descriptor
74  * @return a block or NULL in case of fatal error
75  */
76 static block_t *rtp_stream_recv (int fd)
77 {
78     ssize_t len = 0;
79     struct pollfd ufd = { .fd = fd, .events = POLLIN, };
80     uint8_t hdr[2]; /* frame header */
81
82     /* Receives the RTP frame header */
83     do
84     {
85         ssize_t val;
86
87         poll (&ufd, 1, -1);
88         val = read (fd, hdr + len, 2 - len);
89         if (val <= 0)
90             return NULL;
91         len += val;
92     }
93     while (len < 2);
94
95     block_t *block = block_Alloc (GetWBE (hdr));
96
97     /* Receives the RTP packet */
98     for (ssize_t i = 0; i < len;)
99     {
100         ssize_t val;
101
102         block_cleanup_push (block);
103         poll (&ufd, 1, -1);
104         val = read (fd, block->p_buffer + i, block->i_buffer - i);
105         vlc_cleanup_pop ();
106
107         if (val <= 0)
108         {
109             block_Release (block);
110             return NULL;
111         }
112         i += val;
113     }
114
115     return block;
116 }
117
118
119 static block_t *rtp_recv (demux_t *demux)
120 {
121     demux_sys_t *p_sys = demux->p_sys;
122
123     for (block_t *block;; block_Release (block))
124     {
125         block = p_sys->framed_rtp
126                 ? rtp_stream_recv (p_sys->fd)
127                 : rtp_dgram_recv (p_sys->fd);
128         if (block == NULL)
129         {
130             msg_Err (demux, "RTP flow stopped");
131             break; /* fatal error */
132         }
133
134         if (block->i_buffer < 2)
135             continue;
136
137         /* FIXME */
138         const uint8_t ptype = rtp_ptype (block);
139         if (ptype >= 72 && ptype <= 76)
140             continue; /* Muxed RTCP, ignore for now */
141
142         if (p_sys->srtp)
143         {
144             size_t len = block->i_buffer;
145             int canc, err;
146
147             canc = vlc_savecancel ();
148             err = srtp_recv (p_sys->srtp, block->p_buffer, &len);
149             vlc_restorecancel (canc);
150             if (err)
151             {
152                 msg_Dbg (demux, "SRTP authentication/decryption failed");
153                 continue;
154             }
155             block->i_buffer = len;
156         }
157         return block; /* success! */
158     }
159     return NULL;
160 }
161
162
163 void *rtp_thread (void *data)
164 {
165     demux_t *demux = data;
166     demux_sys_t *p_sys = demux->p_sys;
167
168     for (;;)
169     {
170         block_t *block = rtp_recv (demux);
171         if (block == NULL)
172             break; /* fatal error: abort */
173
174         vlc_mutex_lock (&p_sys->lock);
175
176         /* Autodetect payload type, _before_ rtp_queue() */
177         if (p_sys->autodetect)
178         {
179             if (rtp_autodetect (demux, p_sys->session, block))
180             {
181                 block_Release (block);
182                 continue;
183             }
184             p_sys->autodetect = false;
185         }
186
187         rtp_queue (demux, p_sys->session, block);
188         vlc_cond_signal (&p_sys->wait);
189         vlc_mutex_unlock (&p_sys->lock);
190     }
191     /* TODO: return 0 from Demux */
192     return NULL;
193 }
194
195
196 void rtp_process (demux_t *demux)
197 {
198     demux_sys_t *p_sys = demux->p_sys;
199     mtime_t deadline;
200
201     vlc_mutex_lock (&p_sys->lock);
202     if (rtp_dequeue (demux, p_sys->session, &deadline))
203         /* Pace the demux thread */
204         vlc_cond_timedwait (&p_sys->wait, &p_sys->lock, deadline);
205     else
206         vlc_cond_wait (&p_sys->wait, &p_sys->lock);
207     vlc_mutex_unlock (&p_sys->lock);
208 }