2 * Copyright (C) 2006-2009 Andre Noll <maan@systemlinux.org>
4 * Licensed under the GPL v2. For licencing details see COPYING.
7 /** \file grab_client.c Functions for grabbing the audio stream. */
10 #include <sys/types.h>
18 #include "buffer_tree.h"
20 #include "grab_client.h"
27 * How to handle blocking writes for the grab client fds.
30 /** Ignore the data and do not write. */
32 /** Write anyway (default). */
34 /** Close fd if write would block. */
38 /** Flags specified as arguments to the grab command. */
40 /** Stop grabbing if audio file changes. */
44 /** Describes one active grab client. */
46 /* The value of the -p option. */
48 /** The file descriptor to send the grabbed stream to. */
50 /** See \ref grab_mode. */
52 /** Flags given at the command line. */
53 enum grab_flags flags;
54 /** The point of the grab client's node in the buffer tree. */
55 struct btr_node *btrn;
56 /* The task of this grab client. */
58 /** Belongs to either the active or the inactive list. */
59 struct list_head node;
62 /* Grab clients that are attached to a btr node. */
63 INITIALIZED_LIST_HEAD(active_grab_client_list);
64 /* Grab clients that are not currently attached any btr node. */
65 INITIALIZED_LIST_HEAD(inactive_grab_client_list);
67 static int gc_write(struct grab_client *gc, char *buf, size_t len)
69 int ret = write_ok(gc->fd);
73 if (ret == 0) { /* fd not ready */
74 if (gc->mode == GM_PEDANTIC)
76 if (gc->mode == GM_SLOPPY)
79 ret = write_nonblock(gc->fd, buf, len, 0);
85 if (gc->mode == GM_PEDANTIC)
87 if (gc->mode == GM_SLOPPY)
95 static void gc_pre_select(struct sched *s, struct task *t)
97 struct grab_client *gc = container_of(t, struct grab_client, task);
98 int ret = btr_node_status(gc->btrn, 0, BTR_NT_LEAF);
103 s->timeout.tv_sec = 0;
104 s->timeout.tv_usec = 0;
107 para_fd_set(gc->fd, &s->wfds, &s->max_fileno);
111 * We need this forward declaration as post_select() needs
112 * activate_grab_client and vice versa.
114 static void gc_post_select(struct sched *s, struct task *t);
117 * Move a grab client to the active list and start it.
119 * \param gc The grab client to activate.
121 static void gc_activate(struct grab_client *gc)
123 struct btr_node *root = audiod_get_btr_root(), *parent;
127 parent = btr_search_node(gc->parent, root);
130 PARA_INFO_LOG("activating fd %d\n", gc->fd);
131 list_move(&gc->node, &active_grab_client_list);
132 gc->btrn = btr_new_node("grab", parent, NULL, NULL);
133 if (!gc->task.pre_select) {
134 gc->task.pre_select = gc_pre_select;
135 gc->task.post_select = gc_post_select;
136 sprintf(gc->task.status, "grab");
137 register_task(&gc->task);
142 * Activate inactive grab clients if possible.
144 * This is called from audiod.c when the current audio file changes. It loops
145 * over all inactive grab clients and checks each grab client's configuration
146 * to determine if the client in question wishes to grab the new stream. If
147 * yes, this grab client is moved from the inactive to the active grab client list.
149 * This function also garbage collects all grab clients whose tasks have been
152 void activate_grab_clients(void)
154 struct grab_client *gc, *tmp;
156 list_for_each_entry_safe(gc, tmp, &inactive_grab_client_list, node) {
157 if (gc->task.error == -E_TASK_UNREGISTERED) {
166 static int gc_close(struct grab_client *gc, int err)
168 btr_remove_node(gc->btrn);
169 btr_free_node(gc->btrn);
171 PARA_INFO_LOG("closing gc: %s\n", para_strerror(-err));
172 list_move(&gc->node, &inactive_grab_client_list);
173 if (err == -E_GC_WRITE || (gc->flags & GF_ONE_SHOT)) {
175 * We must not free the gc structure here as it contains ->task
176 * which is still used because this function is called from
187 static void gc_post_select(__a_unused struct sched *s, struct task *t)
189 struct grab_client *gc = container_of(t, struct grab_client, task);
190 struct btr_node *btrn = gc->btrn;
196 ret = btr_node_status(btrn, 0, BTR_NT_LEAF);
201 sz = btr_next_buffer(btrn, &buf);
203 ret = gc_write(gc, buf, sz);
207 btr_consume(btrn, ret);
210 t->error = gc_close(gc, ret)? ret : 0;
213 static int gc_check_args(int argc, char **argv, struct grab_client *gc)
217 for (i = 1; i < argc; i++) {
218 const char *arg = argv[i];
221 if (!strcmp(arg, "--")) {
225 if (!strncmp(arg, "-m", 2)) {
230 gc->mode = GM_SLOPPY;
233 gc->mode = GM_AGGRESSIVE;
236 gc->mode = GM_PEDANTIC;
242 if (!strcmp(arg, "-o")) {
243 gc->flags |= GF_ONE_SHOT;
246 if (!strncmp(arg, "-p=", 3)) {
247 gc->parent = para_strdup(arg + 3);
258 * Check the command line options and allocate a grab_client structure.
260 * \param fd The file descriptor of the client.
261 * \param argc Argument count.
262 * \param argv Argument vector.
264 * If the command line options given by \a argc and \a argv are valid.
265 * allocate a struct grab_client and initialize it with this valid
268 * If the new grab client can be added to an existing buffer tree, activate it.
269 * Otherwise, add it to the inactive list for later activation.
273 int grab_client_new(int fd, int argc, char **argv)
276 struct grab_client *gc = para_calloc(sizeof(struct grab_client));
278 ret = gc_check_args(argc, argv, gc);
282 para_list_add(&gc->node, &inactive_grab_client_list);