1 /*****************************************************************************
2 * select.c: Select individual es to enable or disable from stream
3 *****************************************************************************
4 * Copyright (C) 2006-2011 the VideoLAN team
7 * Authors: Jean-Paul Saman <jpsaman@videolan.org>
8 * Based upon autodel.c written by: Christophe Massiot <massiot@via.ecp.fr>
10 * This program is free software; you can redistribute it and/or modify
11 * it under the terms of the GNU General Public License as published by
12 * the Free Software Foundation; either version 2 of the License, or
13 * (at your option) any later version.
15 * This program is distributed in the hope that it will be useful,
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18 * GNU General Public License for more details.
20 * You should have received a copy of the GNU General Public License
21 * along with this program; if not, write to the Free Software
22 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA.
23 *****************************************************************************/
25 /*****************************************************************************
27 *****************************************************************************/
33 #include <vlc_common.h>
34 #include <vlc_plugin.h>
36 #include <vlc_block.h>
37 #include <vlc_network.h>
39 /*****************************************************************************
41 *****************************************************************************/
42 static int Open ( vlc_object_t * );
43 static void Close ( vlc_object_t * );
45 #define PORT_TEXT N_("Command UDP port")
46 #define PORT_LONGTEXT N_( \
47 "UDP port to listen to for commands (show | enable <pid> | disable <pid>)." )
49 #define DISABLE_TEXT N_("Disable ES id")
50 #define DISABLE_LONGTEXT N_( \
51 "Disable ES id at startup." )
53 #define ENABLE_TEXT N_("Enable ES id")
54 #define ENABLE_LONGTEXT N_( \
55 "Only enable ES id at startup." )
57 #define SOUT_CFG_PREFIX "sout-select-"
60 set_shortname(N_("Select"))
61 set_description(N_("Select individual es to enable or disable from stream"))
62 set_capability("sout stream", 50 )
63 add_integer(SOUT_CFG_PREFIX "port", 5001, PORT_TEXT, PORT_LONGTEXT, true)
64 add_integer(SOUT_CFG_PREFIX "disable", -1, DISABLE_TEXT, DISABLE_LONGTEXT, false)
65 add_integer(SOUT_CFG_PREFIX "enable", -1, ENABLE_TEXT, ENABLE_LONGTEXT, false)
66 add_shortcut("select")
67 set_callbacks(Open, Close)
70 /*****************************************************************************
72 *****************************************************************************/
73 static sout_stream_id_t *Add (sout_stream_t *, es_format_t *);
74 static int Del (sout_stream_t *, sout_stream_id_t *);
75 static int Send (sout_stream_t *, sout_stream_id_t *, block_t *);
77 static void* Command(void *);
79 struct sout_stream_id_t
87 struct sout_stream_sys_t
89 sout_stream_id_t **pp_es;
100 static const char *const ppsz_sout_options[] = {
101 "enable", "disable", "port", NULL
104 /*****************************************************************************
106 *****************************************************************************/
107 static int Open(vlc_object_t *p_this)
109 sout_stream_t *p_stream = (sout_stream_t*)p_this;
110 sout_stream_sys_t *p_sys;
112 p_sys = malloc(sizeof(sout_stream_sys_t));
116 if (!p_stream->p_next)
118 msg_Err( p_stream, "cannot create chain" );
123 config_ChainParse(p_stream, SOUT_CFG_PREFIX, ppsz_sout_options,
126 int port = var_GetInteger(p_stream, SOUT_CFG_PREFIX "port");
127 p_sys->i_fd = net_ListenUDP1(VLC_OBJECT(p_stream), NULL, port);
133 p_sys->i_id_disable = var_GetInteger(p_stream, SOUT_CFG_PREFIX "disable");
134 p_sys->i_id_enable = var_GetInteger(p_stream, SOUT_CFG_PREFIX "enable");
139 p_stream->pf_add = Add;
140 p_stream->pf_del = Del;
141 p_stream->pf_send = Send;
143 p_stream->p_sys = p_sys;
145 vlc_mutex_init(&p_sys->es_lock);
147 if(vlc_clone(&p_sys->thread, Command, p_stream, VLC_THREAD_PRIORITY_LOW))
149 vlc_mutex_destroy(&p_sys->es_lock);
154 /* update p_sout->i_out_pace_nocontrol */
155 p_stream->p_sout->i_out_pace_nocontrol++;
160 /*****************************************************************************
162 *****************************************************************************/
163 static void Close (vlc_object_t * p_this)
165 sout_stream_t *p_stream = (sout_stream_t*)p_this;
166 sout_stream_sys_t *p_sys = (sout_stream_sys_t *)p_stream->p_sys;
168 /* Stop the thread */
169 vlc_cancel(p_sys->thread);
170 vlc_join(p_sys->thread, NULL);
172 /* Free the ressources */
173 net_Close( p_sys->i_fd );
174 vlc_mutex_destroy(&p_sys->es_lock);
176 p_stream->p_sout->i_out_pace_nocontrol--;
181 /****************************************************************************
183 ****************************************************************************/
184 static void* Command(void *p_this)
186 sout_stream_t *p_stream = (sout_stream_t *)p_this;
187 sout_stream_sys_t *p_sys = p_stream->p_sys;
189 while (vlc_object_alive(p_stream))
193 int i_len = recv(p_sys->i_fd, psz_buffer, sizeof(psz_buffer)-1, 0);
197 psz_buffer[i_len] = '\0';
198 msg_Info( p_stream, "command: %s", psz_buffer );
200 if (strncmp(psz_buffer, "show", 4) == 0)
202 vlc_mutex_lock(&p_sys->es_lock);
203 mutex_cleanup_push(&p_sys->es_lock);
204 for (int i = 0; i < p_sys->i_es_num; i++)
206 msg_Info(p_stream, "%.4s : %d",
207 (char *)&p_sys->pp_es[i]->fmt.i_codec,
208 p_sys->pp_es[i]->fmt.i_id);
214 bool b_apply = false;
215 bool b_select = false;
218 if (strncmp(psz_buffer, "enable", 6) == 0)
220 i_pid = atol(psz_buffer+7);
224 else if (strncmp(psz_buffer, "disable", 7) == 0)
226 i_pid = atol(psz_buffer+8);
232 vlc_mutex_lock(&p_sys->es_lock);
233 mutex_cleanup_push(&p_sys->es_lock);
234 for (int i = 0; i < p_sys->i_es_num; i++)
236 msg_Info(p_stream, "elementary stream pid %d",
237 p_sys->pp_es[i]->fmt.i_id);
238 if (p_sys->pp_es[i]->fmt.i_id == i_pid)
240 p_sys->pp_es[i]->b_enabled = b_select;
241 msg_Info(p_stream, "%s: %d", b_select ? "enable" : "disable", i_pid);
252 static sout_stream_id_t *Add(sout_stream_t *p_stream, es_format_t *p_fmt)
254 sout_stream_sys_t *p_sys = (sout_stream_sys_t *)p_stream->p_sys;
255 sout_stream_id_t *p_es = malloc(sizeof(sout_stream_id_t));
261 p_es->b_error = false;
262 if (p_sys->i_id_disable >= -1)
263 p_es->b_enabled = (p_es->fmt.i_id == p_sys->i_id_disable) ? false : true;
264 else if (p_sys->i_id_enable >= -1)
265 p_es->b_enabled = (p_es->fmt.i_id == p_sys->i_id_enable) ? true: false;
267 vlc_mutex_lock(&p_sys->es_lock);
268 TAB_APPEND(p_sys->i_es_num, p_sys->pp_es, p_es);
269 vlc_mutex_unlock(&p_sys->es_lock);
274 static int Del(sout_stream_t *p_stream, sout_stream_id_t *p_es)
276 sout_stream_sys_t *p_sys = (sout_stream_sys_t *)p_stream->p_sys;
277 sout_stream_id_t *id = p_es->id;
279 vlc_mutex_lock(&p_sys->es_lock);
280 TAB_REMOVE(p_sys->i_es_num, p_sys->pp_es, p_es);
281 vlc_mutex_unlock(&p_sys->es_lock);
286 return p_stream->p_next->pf_del(p_stream->p_next, id);
291 static int Send(sout_stream_t *p_stream, sout_stream_id_t *p_es,
294 if (p_es->id == NULL && !p_es->b_error)
296 p_es->id = p_stream->p_next->pf_add(p_stream->p_next, &p_es->fmt);
297 if (p_es->id == NULL)
299 p_es->b_error = true;
300 msg_Err(p_stream, "couldn't create chain for id %d",
305 if (!p_es->b_error && p_es->b_enabled)
306 p_stream->p_next->pf_send(p_stream->p_next, p_es->id, p_buffer);
308 block_ChainRelease(p_buffer);