2 This file is part of PulseAudio.
4 Copyright 2009 Lennart Poettering
6 PulseAudio is free software; you can redistribute it and/or modify
7 it under the terms of the GNU Lesser General Public License as published
8 by the Free Software Foundation; either version 2.1 of the License,
9 or (at your option) any later version.
11 PulseAudio is distributed in the hope that it will be useful, but
12 WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 General Public License for more details.
16 You should have received a copy of the GNU Lesser General Public License
17 along with PulseAudio; if not, write to the Free Software
18 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
30 #include <pulse/xmalloc.h>
32 #include <pulsecore/module.h>
33 #include <pulsecore/core-util.h>
34 #include <pulsecore/modargs.h>
35 #include <pulsecore/log.h>
36 #include <pulsecore/client.h>
37 #include <pulsecore/conf-parser.h>
39 #include "module-augment-properties-symdef.h"
41 PA_MODULE_AUTHOR("Lennart Poettering");
42 PA_MODULE_DESCRIPTION("Augment the property sets of streams with additional static information");
43 PA_MODULE_VERSION(PACKAGE_VERSION
);
44 PA_MODULE_LOAD_ONCE(TRUE
);
46 #define STAT_INTERVAL 30
47 #define MAX_CACHE_SIZE 50
49 static const char* const valid_modargs
[] = {
58 char *application_name
;
61 pa_proplist
*proplist
;
66 pa_hook_slot
*client_new_slot
, *client_proplist_changed_slot
;
69 static void rule_free(struct rule
*r
) {
72 pa_xfree(r
->process_name
);
73 pa_xfree(r
->application_name
);
74 pa_xfree(r
->icon_name
);
77 pa_proplist_free(r
->proplist
);
81 static int parse_properties(
90 struct rule
*r
= userdata
;
93 if (!(n
= pa_proplist_from_string(rvalue
)))
97 pa_proplist_update(r
->proplist
, PA_UPDATE_MERGE
, n
);
105 static int parse_categories(
106 const char *filename
,
114 struct rule
*r
= userdata
;
115 const char *state
= NULL
;
118 while ((c
= pa_split(rvalue
, ";", &state
))) {
120 if (pa_streq(c
, "Game")) {
122 r
->role
= pa_xstrdup("game");
123 } else if (pa_streq(c
, "Telephony")) {
125 r
->role
= pa_xstrdup("phone");
134 static int check_type(
135 const char *filename
,
143 return pa_streq(rvalue
, "Application") ? 0 : -1;
146 static int catch_all(
147 const char *filename
,
158 static void update_rule(struct rule
*r
) {
161 static pa_config_item table
[] = {
162 { "Name", pa_config_parse_string
, NULL
, "Desktop Entry" },
163 { "Icon", pa_config_parse_string
, NULL
, "Desktop Entry" },
164 { "Type", check_type
, NULL
, "Desktop Entry" },
165 { "X-PulseAudio-Properties", parse_properties
, NULL
, "Desktop Entry" },
166 { "Categories", parse_categories
, NULL
, "Desktop Entry" },
167 { NULL
, catch_all
, NULL
, NULL
},
168 { NULL
, NULL
, NULL
, NULL
},
170 pa_bool_t found
= FALSE
;
173 fn
= pa_sprintf_malloc(DESKTOPFILEDIR PA_PATH_SEP
"%s.desktop", r
->process_name
);
175 if (stat(fn
, &st
) == 0)
178 DIR *desktopfiles_dir
;
182 /* Let's try a more aggressive search, but only one level */
183 if ((desktopfiles_dir
= opendir(DESKTOPFILEDIR
))) {
184 while ((dir
= readdir(desktopfiles_dir
))) {
185 if (dir
->d_type
!= DT_DIR
186 || strcmp(dir
->d_name
, ".") == 0
187 || strcmp(dir
->d_name
, "..") == 0)
191 fn
= pa_sprintf_malloc(DESKTOPFILEDIR PA_PATH_SEP
"%s" PA_PATH_SEP
"%s.desktop", dir
->d_name
, r
->process_name
);
193 if (stat(fn
, &st
) == 0) {
198 closedir(desktopfiles_dir
);
209 if (st
.st_mtime
== r
->mtime
) {
210 /* Theoretically the filename could have changed, but if so
211 having the same mtime is very unlikely so not worth tracking it in r */
215 pa_log_debug("Found %s (which has been updated since we last checked).", fn
);
217 pa_log_debug("Found %s.", fn
);
220 r
->mtime
= st
.st_mtime
;
221 pa_xfree(r
->application_name
);
222 pa_xfree(r
->icon_name
);
224 r
->application_name
= r
->icon_name
= r
->role
= NULL
;
226 pa_proplist_clear(r
->proplist
);
228 table
[0].data
= &r
->application_name
;
229 table
[1].data
= &r
->icon_name
;
231 if (pa_config_parse(fn
, NULL
, table
, r
) < 0)
232 pa_log_warn("Failed to parse .desktop file %s.", fn
);
237 static void apply_rule(struct rule
*r
, pa_proplist
*p
) {
245 pa_proplist_update(p
, PA_UPDATE_MERGE
, r
->proplist
);
248 if (!pa_proplist_contains(p
, PA_PROP_APPLICATION_ICON_NAME
))
249 pa_proplist_sets(p
, PA_PROP_APPLICATION_ICON_NAME
, r
->icon_name
);
251 if (r
->application_name
) {
254 t
= pa_proplist_gets(p
, PA_PROP_APPLICATION_NAME
);
256 if (!t
|| pa_streq(t
, r
->process_name
))
257 pa_proplist_sets(p
, PA_PROP_APPLICATION_NAME
, r
->application_name
);
261 if (!pa_proplist_contains(p
, PA_PROP_MEDIA_ROLE
))
262 pa_proplist_sets(p
, PA_PROP_MEDIA_ROLE
, r
->role
);
265 static void make_room(pa_hashmap
*cache
) {
268 while (pa_hashmap_size(cache
) >= MAX_CACHE_SIZE
) {
271 pa_assert_se(r
= pa_hashmap_steal_first(cache
));
276 static pa_hook_result_t
process(struct userdata
*u
, pa_proplist
*p
) {
284 if (!(pn
= pa_proplist_gets(p
, PA_PROP_APPLICATION_PROCESS_BINARY
)))
287 if (*pn
== '.' || strchr(pn
, '/'))
292 pa_log_debug("Looking for .desktop file for %s", pn
);
294 if ((r
= pa_hashmap_get(u
->cache
, pn
))) {
295 if (now
-r
->timestamp
> STAT_INTERVAL
) {
302 r
= pa_xnew0(struct rule
, 1);
303 r
->process_name
= pa_xstrdup(pn
);
305 pa_hashmap_put(u
->cache
, r
->process_name
, r
);
313 static pa_hook_result_t
client_new_cb(pa_core
*core
, pa_client_new_data
*data
, struct userdata
*u
) {
314 pa_core_assert_ref(core
);
318 return process(u
, data
->proplist
);
321 static pa_hook_result_t
client_proplist_changed_cb(pa_core
*core
, pa_client
*client
, struct userdata
*u
) {
322 pa_core_assert_ref(core
);
326 return process(u
, client
->proplist
);
329 int pa__init(pa_module
*m
) {
330 pa_modargs
*ma
= NULL
;
335 if (!(ma
= pa_modargs_new(m
->argument
, valid_modargs
))) {
336 pa_log("Failed to parse module arguments");
340 m
->userdata
= u
= pa_xnew(struct userdata
, 1);
342 u
->cache
= pa_hashmap_new(pa_idxset_string_hash_func
, pa_idxset_string_compare_func
);
343 u
->client_new_slot
= pa_hook_connect(&m
->core
->hooks
[PA_CORE_HOOK_CLIENT_NEW
], PA_HOOK_EARLY
, (pa_hook_cb_t
) client_new_cb
, u
);
344 u
->client_proplist_changed_slot
= pa_hook_connect(&m
->core
->hooks
[PA_CORE_HOOK_CLIENT_PROPLIST_CHANGED
], PA_HOOK_EARLY
, (pa_hook_cb_t
) client_proplist_changed_cb
, u
);
359 void pa__done(pa_module
*m
) {
364 if (!(u
= m
->userdata
))
367 if (u
->client_new_slot
)
368 pa_hook_slot_free(u
->client_new_slot
);
369 if (u
->client_proplist_changed_slot
)
370 pa_hook_slot_free(u
->client_proplist_changed_slot
);
375 while ((r
= pa_hashmap_steal_first(u
->cache
)))
378 pa_hashmap_free(u
->cache
, NULL
, NULL
);