]> git.sesse.net Git - vlc/blob - src/audio_output/dec.c
Fix malloc error handling
[vlc] / src / audio_output / dec.c
1 /*****************************************************************************
2  * dec.c : audio output API towards decoders
3  *****************************************************************************
4  * Copyright (C) 2002-2007 the VideoLAN team
5  * $Id$
6  *
7  * Authors: Christophe Massiot <massiot@via.ecp.fr>
8  *
9  * This program is free software; you can redistribute it and/or modify
10  * it under the terms of the GNU General Public License as published by
11  * the Free Software Foundation; either version 2 of the License, or
12  * (at your option) any later version.
13  *
14  * This program is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17  * GNU General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program; if not, write to the Free Software
21  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
22  *****************************************************************************/
23
24 /*****************************************************************************
25  * Preamble
26  *****************************************************************************/
27 #include <stdlib.h>                            /* calloc(), malloc(), free() */
28 #include <string.h>
29
30 #include <vlc/vlc.h>
31
32 #ifdef HAVE_ALLOCA_H
33 #   include <alloca.h>
34 #endif
35
36 #include <vlc_aout.h>
37 #include <vlc_input.h>
38
39 #include "aout_internal.h"
40
41 /** FIXME: Ugly but needed to access the counters */
42 #include "input/input_internal.h"
43
44 /*****************************************************************************
45  * aout_DecNew : create a decoder
46  *****************************************************************************/
47 static aout_input_t * DecNew( vlc_object_t * p_this, aout_instance_t * p_aout,
48                               audio_sample_format_t *p_format,
49                               audio_replay_gain_t *p_replay_gain )
50 {
51     aout_input_t * p_input;
52     input_thread_t * p_input_thread;
53     vlc_value_t val;
54
55     /* Sanitize audio format */
56     if( p_format->i_channels > 32 )
57     {
58         msg_Err( p_aout, "too many audio channels (%u)",
59                  p_format->i_channels );
60         goto error;
61     }
62
63     if( p_format->i_rate > 192000 )
64     {
65         msg_Err( p_aout, "excessive audio sample frequency (%u)",
66                  p_format->i_rate );
67         goto error;
68     }
69
70     /* We can only be called by the decoder, so no need to lock
71      * p_input->lock. */
72     vlc_mutex_lock( &p_aout->mixer_lock );
73
74     if ( p_aout->i_nb_inputs >= AOUT_MAX_INPUTS )
75     {
76         msg_Err( p_aout, "too many inputs already (%d)", p_aout->i_nb_inputs );
77         goto error;
78     }
79
80     p_input = malloc(sizeof(aout_input_t));
81     if ( p_input == NULL )
82     {
83         msg_Err( p_aout, "out of memory" );
84         goto error;
85     }
86     memset( p_input, 0, sizeof(aout_input_t) );
87
88     vlc_mutex_init( p_aout, &p_input->lock );
89
90     p_input->b_changed = 0;
91     p_input->b_error = 1;
92     aout_FormatPrepare( p_format );
93     memcpy( &p_input->input, p_format,
94             sizeof(audio_sample_format_t) );
95     if( p_replay_gain )
96         p_input->replay_gain = *p_replay_gain;
97
98     p_aout->pp_inputs[p_aout->i_nb_inputs] = p_input;
99     p_aout->i_nb_inputs++;
100
101     if ( p_aout->mixer.b_error )
102     {
103         int i;
104
105         var_Destroy( p_aout, "audio-device" );
106         var_Destroy( p_aout, "audio-channels" );
107
108         /* Recreate the output using the new format. */
109         if ( aout_OutputNew( p_aout, p_format ) < 0 )
110         {
111             for ( i = 0; i < p_aout->i_nb_inputs - 1; i++ )
112             {
113                 vlc_mutex_lock( &p_aout->pp_inputs[i]->lock );
114                 aout_InputDelete( p_aout, p_aout->pp_inputs[i] );
115                 vlc_mutex_unlock( &p_aout->pp_inputs[i]->lock );
116             }
117             vlc_mutex_unlock( &p_aout->mixer_lock );
118             return p_input;
119         }
120
121         /* Create other input streams. */
122         for ( i = 0; i < p_aout->i_nb_inputs - 1; i++ )
123         {
124             vlc_mutex_lock( &p_aout->pp_inputs[i]->lock );
125             aout_InputDelete( p_aout, p_aout->pp_inputs[i] );
126             aout_InputNew( p_aout, p_aout->pp_inputs[i] );
127             vlc_mutex_unlock( &p_aout->pp_inputs[i]->lock );
128         }
129     }
130     else
131     {
132         aout_MixerDelete( p_aout );
133     }
134
135     if ( aout_MixerNew( p_aout ) == -1 )
136     {
137         aout_OutputDelete( p_aout );
138         goto error;
139     }
140
141     aout_InputNew( p_aout, p_input );
142
143     vlc_mutex_unlock( &p_aout->mixer_lock );
144     var_Create( p_this, "audio-desync", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
145     var_Get( p_this, "audio-desync", &val );
146     p_input->i_desync = val.i_int * 1000;
147
148     p_input_thread = (input_thread_t *)vlc_object_find( p_this,
149                                            VLC_OBJECT_INPUT, FIND_PARENT );
150     if( p_input_thread )
151     {
152         p_input->i_pts_delay = p_input_thread->i_pts_delay;
153         p_input->i_pts_delay += p_input->i_desync;
154         p_input->p_input_thread = p_input_thread;
155         vlc_object_release( p_input_thread );
156     }
157     else
158     {
159         p_input->i_pts_delay = DEFAULT_PTS_DELAY;
160         p_input->i_pts_delay += p_input->i_desync;
161         p_input->p_input_thread = NULL;
162     }
163
164     return p_input;
165
166 error:
167     vlc_mutex_unlock( &p_aout->mixer_lock );
168     return NULL;
169 }
170
171 aout_input_t * __aout_DecNew( vlc_object_t * p_this,
172                               aout_instance_t ** pp_aout,
173                               audio_sample_format_t * p_format,
174                               audio_replay_gain_t *p_replay_gain )
175 {
176     if ( *pp_aout == NULL )
177     {
178         /* Create an audio output if there is none. */
179         *pp_aout = vlc_object_find( p_this, VLC_OBJECT_AOUT, FIND_ANYWHERE );
180
181         if( *pp_aout == NULL )
182         {
183             msg_Dbg( p_this, "no aout present, spawning one" );
184
185             *pp_aout = aout_New( p_this );
186             /* Everything failed, I'm a loser, I just wanna die */
187             if( *pp_aout == NULL )
188             {
189                 return NULL;
190             }
191             vlc_object_attach( *pp_aout, p_this->p_libvlc );
192         }
193         else
194         {
195             vlc_object_release( *pp_aout );
196         }
197     }
198
199     return DecNew( p_this, *pp_aout, p_format, p_replay_gain );
200 }
201
202 /*****************************************************************************
203  * aout_DecDelete : delete a decoder
204  *****************************************************************************/
205 int aout_DecDelete( aout_instance_t * p_aout, aout_input_t * p_input )
206 {
207     int i_input;
208
209     /* This function can only be called by the decoder itself, so no need
210      * to lock p_input->lock. */
211     vlc_mutex_lock( &p_aout->mixer_lock );
212
213     for ( i_input = 0; i_input < p_aout->i_nb_inputs; i_input++ )
214     {
215         if ( p_aout->pp_inputs[i_input] == p_input )
216         {
217             break;
218         }
219     }
220
221     if ( i_input == p_aout->i_nb_inputs )
222     {
223         msg_Err( p_aout, "cannot find an input to delete" );
224         return -1;
225     }
226
227     /* Remove the input from the list. */
228     memmove( &p_aout->pp_inputs[i_input], &p_aout->pp_inputs[i_input + 1],
229              (AOUT_MAX_INPUTS - i_input - 1) * sizeof(aout_input_t *) );
230     p_aout->i_nb_inputs--;
231
232     aout_InputDelete( p_aout, p_input );
233
234     vlc_mutex_destroy( &p_input->lock );
235     free( p_input );
236
237     if ( !p_aout->i_nb_inputs )
238     {
239         aout_OutputDelete( p_aout );
240         aout_MixerDelete( p_aout );
241         if ( var_Type( p_aout, "audio-device" ) != 0 )
242         {
243             var_Destroy( p_aout, "audio-device" );
244         }
245         if ( var_Type( p_aout, "audio-channels" ) != 0 )
246         {
247             var_Destroy( p_aout, "audio-channels" );
248         }
249     }
250
251     vlc_mutex_unlock( &p_aout->mixer_lock );
252
253     return 0;
254 }
255
256
257 /*
258  * Buffer management
259  */
260
261 /*****************************************************************************
262  * aout_DecNewBuffer : ask for a new empty buffer
263  *****************************************************************************/
264 aout_buffer_t * aout_DecNewBuffer( aout_instance_t * p_aout,
265                                    aout_input_t * p_input,
266                                    size_t i_nb_samples )
267 {
268     aout_buffer_t * p_buffer;
269     mtime_t duration;
270
271     vlc_mutex_lock( &p_input->lock );
272
273     if ( p_input->b_error )
274     {
275         vlc_mutex_unlock( &p_input->lock );
276         return NULL;
277     }
278
279     duration = (1000000 * (mtime_t)i_nb_samples) / p_input->input.i_rate;
280
281     /* This necessarily allocates in the heap. */
282     aout_BufferAlloc( &p_input->input_alloc, duration, NULL, p_buffer );
283     if( p_buffer != NULL )
284         p_buffer->i_nb_bytes = i_nb_samples * p_input->input.i_bytes_per_frame
285                                   / p_input->input.i_frame_length;
286
287     /* Suppose the decoder doesn't have more than one buffered buffer */
288     p_input->b_changed = 0;
289
290     vlc_mutex_unlock( &p_input->lock );
291
292     if( p_buffer == NULL )
293         return NULL;
294
295     p_buffer->i_nb_samples = i_nb_samples;
296     p_buffer->start_date = p_buffer->end_date = 0;
297     return p_buffer;
298 }
299
300 /*****************************************************************************
301  * aout_DecDeleteBuffer : destroy an undecoded buffer
302  *****************************************************************************/
303 void aout_DecDeleteBuffer( aout_instance_t * p_aout, aout_input_t * p_input,
304                            aout_buffer_t * p_buffer )
305 {
306     (void)p_aout; (void)p_input;
307     aout_BufferFree( p_buffer );
308 }
309
310 /*****************************************************************************
311  * aout_DecPlay : filter & mix the decoded buffer
312  *****************************************************************************/
313 int aout_DecPlay( aout_instance_t * p_aout, aout_input_t * p_input,
314                   aout_buffer_t * p_buffer, int i_input_rate )
315 {
316     if ( p_buffer->start_date == 0 )
317     {
318         msg_Warn( p_aout, "non-dated buffer received" );
319         aout_BufferFree( p_buffer );
320         return -1;
321     }
322
323     if( i_input_rate > INPUT_RATE_DEFAULT * AOUT_MAX_INPUT_RATE ||
324         i_input_rate < INPUT_RATE_DEFAULT / AOUT_MAX_INPUT_RATE )
325     {
326         aout_BufferFree( p_buffer );
327         return 0;
328     }
329
330     /* Apply the desynchronisation requested by the user */
331     p_buffer->start_date += p_input->i_desync;
332     p_buffer->end_date += p_input->i_desync;
333
334     if ( p_buffer->start_date > mdate() + p_input->i_pts_delay +
335          AOUT_MAX_ADVANCE_TIME )
336     {
337         msg_Warn( p_aout, "received buffer in the future ("I64Fd")",
338                   p_buffer->start_date - mdate());
339         if( p_input->p_input_thread )
340         {
341             vlc_mutex_lock( &p_input->p_input_thread->p->counters.counters_lock);
342             stats_UpdateInteger( p_aout,
343                            p_input->p_input_thread->p->counters.p_lost_abuffers,
344                            1, NULL );
345             vlc_mutex_unlock( &p_input->p_input_thread->p->counters.counters_lock);
346         }
347         aout_BufferFree( p_buffer );
348         return -1;
349     }
350
351     p_buffer->end_date = p_buffer->start_date
352                             + (mtime_t)p_buffer->i_nb_samples * 1000000
353                                 / p_input->input.i_rate;
354
355     vlc_mutex_lock( &p_input->lock );
356
357     if ( p_input->b_error )
358     {
359         vlc_mutex_unlock( &p_input->lock );
360         aout_BufferFree( p_buffer );
361         return -1;
362     }
363
364     if ( p_input->b_changed )
365     {
366         /* Maybe the allocation size has changed. Re-allocate a buffer. */
367         aout_buffer_t * p_new_buffer;
368         mtime_t duration = (1000000 * (mtime_t)p_buffer->i_nb_samples)
369                             / p_input->input.i_rate;
370
371         aout_BufferAlloc( &p_input->input_alloc, duration, NULL, p_new_buffer );
372         p_aout->p_libvlc->pf_memcpy( p_new_buffer->p_buffer, p_buffer->p_buffer,
373                                   p_buffer->i_nb_bytes );
374         p_new_buffer->i_nb_samples = p_buffer->i_nb_samples;
375         p_new_buffer->i_nb_bytes = p_buffer->i_nb_bytes;
376         p_new_buffer->start_date = p_buffer->start_date;
377         p_new_buffer->end_date = p_buffer->end_date;
378         aout_BufferFree( p_buffer );
379         p_buffer = p_new_buffer;
380         p_input->b_changed = 0;
381     }
382
383     /* If the buffer is too early, wait a while. */
384     mwait( p_buffer->start_date - AOUT_MAX_PREPARE_TIME );
385
386     if ( aout_InputPlay( p_aout, p_input, p_buffer, i_input_rate ) == -1 )
387     {
388         vlc_mutex_unlock( &p_input->lock );
389         return -1;
390     }
391
392     vlc_mutex_unlock( &p_input->lock );
393
394     /* Run the mixer if it is able to run. */
395     vlc_mutex_lock( &p_aout->mixer_lock );
396     aout_MixerRun( p_aout );
397     if( p_input->p_input_thread )
398     {
399         vlc_mutex_lock( &p_input->p_input_thread->p->counters.counters_lock);
400         stats_UpdateInteger( p_aout,
401                              p_input->p_input_thread->p->counters.p_played_abuffers,
402                              1, NULL );
403         vlc_mutex_unlock( &p_input->p_input_thread->p->counters.counters_lock);
404     }
405     vlc_mutex_unlock( &p_aout->mixer_lock );
406
407     return 0;
408 }