2 * Copyright (C) 2006 Andre Noll <maan@systemlinux.org>
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation; either version 2 of the License, or
7 * (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA.
20 * \file grab_client.c functions for grabbing the stream at any position
23 * \sa filter_chain filter_chain_info filter
27 #include "close_on_fork.h"
28 #include "grab_client.cmdline.h"
31 #include "grab_client.h"
37 /** grab clients that are not yet attached to a filter node */
38 struct list_head inactive_grab_client_list;
40 static int max_num_filters(void)
43 for (i = 0; audio_formats[i]; i++) {
44 PARA_INFO_LOG("%s filter chain length: %d\n", audio_formats[i],
46 ret = PARA_MAX(ret, num_filters(i));
48 PARA_INFO_LOG("maximal filter chain length: %d\n", ret);
52 static int gc_write(char *buf, size_t len, struct filter_callback *fcb)
54 struct grab_client *gc = fcb->data;
55 struct timeval tv = {0, 100};
58 // PARA_INFO_LOG("writing %d bytes to fd %d\n", len, gc->fd);
61 FD_SET(gc->fd, &wfds);
62 ret = para_select(gc->fd + 1, NULL, &wfds, &tv);
64 if (gc->mode == GRAB_PEDANTIC)
65 return -E_PEDANTIC_GRAB;
66 if (gc->mode == GRAB_SLOPPY)
70 ret = write(gc->fd, buf, len);
73 gc->error = E_GC_WRITE;
76 if (gc->mode == GRAB_PEDANTIC)
77 return -E_PEDANTIC_GRAB;
78 if (gc->mode == GRAB_AGGRESSIVE) {
80 memmove(buf, buf + ret, len);
88 static int check_gc_args(struct grab_client *gc)
91 struct grab_client_args_info *conf = gc->conf;
92 char **mv = grab_client_cmdline_parser_mode_values;
94 PARA_INFO_LOG("filter_num: %d\n", gc->conf->filter_num_arg);
95 for (i = 0; mv[i]; i++)
96 if (!strcmp(conf->mode_arg, mv[i]))
99 if (conf->audio_format_given) {
100 gc->audio_format_num = get_audio_format_num(conf->audio_format_arg);
101 if (gc->audio_format_num < 0)
102 return gc->audio_format_num;
104 if (conf->slot_arg > MAX_STREAM_SLOTS)
105 return -E_BAD_GC_SLOT;
106 if (conf->filter_num_arg <= 0)
107 return -E_BAD_GC_FILTER_NUM;
108 if (conf->audio_format_given) {
109 if (num_filters(gc->audio_format_num) < conf->filter_num_arg)
110 return -E_BAD_GC_FILTER_NUM;
112 if (conf->filter_num_arg > max_num_filters())
113 return -E_BAD_GC_FILTER_NUM;
118 static void add_inactive_gc(struct grab_client *gc)
120 PARA_INFO_LOG("adding grab client %p (fd %d) to inactive list\n",
122 list_add(&gc->node, &inactive_grab_client_list);
125 static void gc_free(struct grab_client *gc)
129 for (i = 0; i < gc->argc; i++)
137 static void gc_close(struct filter_callback *fcb)
139 struct grab_client *gc = fcb->data;
141 if (gc->conf->one_shot_given || gc->error) {
142 PARA_INFO_LOG("closing fd %d (grab client %p)\n", gc->fd, gc);
143 del_close_on_fork_list(gc->fd);
153 * move a grab client from the inactive list to a filter node
155 * \param gc the grab client to activate
156 * \param fn the filter node \a gc gets attached to
158 * \sa filter_node::callbacks, inactive_grab_client_list
160 void activate_grab_client(struct grab_client *gc, struct filter_node *fn)
162 PARA_INFO_LOG("activating %p (fd %d, filter node: %p)\n", gc, gc->fd, fn);
164 list_add(&gc->fcb.node, &fn->callbacks);
168 * activate inactive grab clients if possible
170 * \param slot audiod's slot for the new audio file
171 * \param audio_format_num the number of the audio format of the new audio file
172 * \param filter_list the list of activated filters for that new audio file
174 * This is called from audiod.c when the current audio file changes. It loops
175 * over all inactive grab clients and checks each grab client's configuration
176 * to determine if the client in question wishes to grab the new stream. If
177 * yes, this grab client is moved from the inactive grab client list to an
178 * appropriate filter_node.
180 * \sa filter_chain_info::filters, inactive_grab_client_list,
181 * activate_grab_client
183 void activate_inactive_grab_clients(int slot, int audio_format_num,
184 struct list_head *filter_list)
186 struct grab_client *gc, *tmp;
188 struct filter_node *fn;
190 list_for_each_entry_safe(gc, tmp, &inactive_grab_client_list, node) {
191 // PARA_INFO_LOG("checking inactive grab client %p\n", gc);
192 if (gc->conf->slot_arg >= 0 && gc->conf->slot_arg != slot)
194 if (gc->audio_format_num >= 0 && gc->audio_format_num !=
197 if (gc->conf->filter_num_arg >= 0 &&
198 num_filters(gc->audio_format_num)
199 < gc->conf->filter_num_arg)
202 list_for_each_entry(fn, filter_list, node) {
203 if (gc->conf->filter_num_arg <= 0
204 || i == gc->conf->filter_num_arg)
208 activate_grab_client(gc, fn);
213 * check the command line options and allocate a grab_client structure
215 * \param fd the file descriptor of the client
216 * \param argc the number of command line options
217 * \param argv pointers to the command line options
218 * \param err non-zero if an error occured
220 * If the command line options given by \a argc and \a argv are valid.
221 * allocate a struct grab_client and initialize it with this valid
222 * configuration. Moreover, add the new grab client to the inactive list.
224 * \return On success, this function returns a pointer to the newly created
225 * struct. On errors, it returns NULL and sets \a err appropriately.
227 * \sa grab_client, inactive_grab_client_list, activate_grab_client,
228 * filter_node::callbacks
231 * argc, argv get freed when com_grab() returns, so we have to make a
234 struct grab_client *grab_client_new(int fd, int argc, char **argv, int *err)
237 struct grab_client *gc = para_calloc(sizeof(struct grab_client));
239 gc->conf = para_calloc(sizeof(struct grab_client_args_info));
241 gc->argv = para_calloc((argc + 2) * sizeof(char *));
243 for (i = 0; argv[i]; i++) {
244 gc->argv[i] = para_strdup(argv[i]);
245 PARA_INFO_LOG("argc: %d, argv[%d]: %s\n", argc, i, gc->argv[i]);
247 ret = grab_client_cmdline_parser(gc->argc, gc->argv , gc->conf);
251 *err = -E_GC_HELP_GIVEN;
252 if (gc->conf->help_given)
254 *err = -E_GC_VERSION_GIVEN;
255 if (gc->conf->version_given)
257 *err = check_gc_args(gc);
260 if (gc->conf->input_grab_given) {
261 gc->fcb.input_cb = gc_write;
262 gc->fcb.output_cb = NULL;
264 gc->fcb.output_cb = gc_write;
265 gc->fcb.input_cb = NULL;
268 gc->fcb.close = gc_close;
273 for (i = 0; i < argc; i++)
281 /** initialize the grabbing subsystem.
283 * This has to be called once during startup before any other function from
284 * grab_client.c may be used. It initializes \a inactive_grab_client_list.
288 PARA_INFO_LOG("%s", "grab init\n");
289 INIT_LIST_HEAD(&inactive_grab_client_list);