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(pa_config_parser_state
*state
) {
89 if (!(n
= pa_proplist_from_string(state
->rvalue
)))
93 pa_proplist_update(r
->proplist
, PA_UPDATE_MERGE
, n
);
101 static int parse_categories(pa_config_parser_state
*state
) {
103 const char *split_state
= NULL
;
110 while ((c
= pa_split(state
->rvalue
, ";", &split_state
))) {
112 if (pa_streq(c
, "Game")) {
114 r
->role
= pa_xstrdup("game");
115 } else if (pa_streq(c
, "Telephony")) {
117 r
->role
= pa_xstrdup("phone");
126 static int check_type(pa_config_parser_state
*state
) {
129 return pa_streq(state
->rvalue
, "Application") ? 0 : -1;
132 static int catch_all(pa_config_parser_state
*state
) {
136 static void update_rule(struct rule
*r
) {
139 static pa_config_item table
[] = {
140 { "Name", pa_config_parse_string
, NULL
, "Desktop Entry" },
141 { "Icon", pa_config_parse_string
, NULL
, "Desktop Entry" },
142 { "Type", check_type
, NULL
, "Desktop Entry" },
143 { "X-PulseAudio-Properties", parse_properties
, NULL
, "Desktop Entry" },
144 { "Categories", parse_categories
, NULL
, "Desktop Entry" },
145 { NULL
, catch_all
, NULL
, NULL
},
146 { NULL
, NULL
, NULL
, NULL
},
148 pa_bool_t found
= FALSE
;
151 fn
= pa_sprintf_malloc(DESKTOPFILEDIR PA_PATH_SEP
"%s.desktop", r
->process_name
);
153 if (stat(fn
, &st
) == 0)
157 DIR *desktopfiles_dir
;
160 /* Let's try a more aggressive search, but only one level */
161 if ((desktopfiles_dir
= opendir(DESKTOPFILEDIR
))) {
162 while ((dir
= readdir(desktopfiles_dir
))) {
163 if (dir
->d_type
!= DT_DIR
164 || pa_streq(dir
->d_name
, ".")
165 || pa_streq(dir
->d_name
, ".."))
169 fn
= pa_sprintf_malloc(DESKTOPFILEDIR PA_PATH_SEP
"%s" PA_PATH_SEP
"%s.desktop", dir
->d_name
, r
->process_name
);
171 if (stat(fn
, &st
) == 0) {
176 closedir(desktopfiles_dir
);
187 if (st
.st_mtime
== r
->mtime
) {
188 /* Theoretically the filename could have changed, but if so
189 having the same mtime is very unlikely so not worth tracking it in r */
193 pa_log_debug("Found %s (which has been updated since we last checked).", fn
);
195 pa_log_debug("Found %s.", fn
);
198 r
->mtime
= st
.st_mtime
;
199 pa_xfree(r
->application_name
);
200 pa_xfree(r
->icon_name
);
202 r
->application_name
= r
->icon_name
= r
->role
= NULL
;
204 pa_proplist_clear(r
->proplist
);
206 table
[0].data
= &r
->application_name
;
207 table
[1].data
= &r
->icon_name
;
209 if (pa_config_parse(fn
, NULL
, table
, NULL
, r
) < 0)
210 pa_log_warn("Failed to parse .desktop file %s.", fn
);
215 static void apply_rule(struct rule
*r
, pa_proplist
*p
) {
223 pa_proplist_update(p
, PA_UPDATE_MERGE
, r
->proplist
);
226 if (!pa_proplist_contains(p
, PA_PROP_APPLICATION_ICON_NAME
))
227 pa_proplist_sets(p
, PA_PROP_APPLICATION_ICON_NAME
, r
->icon_name
);
229 if (r
->application_name
) {
232 t
= pa_proplist_gets(p
, PA_PROP_APPLICATION_NAME
);
234 if (!t
|| pa_streq(t
, r
->process_name
))
235 pa_proplist_sets(p
, PA_PROP_APPLICATION_NAME
, r
->application_name
);
239 if (!pa_proplist_contains(p
, PA_PROP_MEDIA_ROLE
))
240 pa_proplist_sets(p
, PA_PROP_MEDIA_ROLE
, r
->role
);
243 static void make_room(pa_hashmap
*cache
) {
246 while (pa_hashmap_size(cache
) >= MAX_CACHE_SIZE
) {
249 pa_assert_se(r
= pa_hashmap_steal_first(cache
));
254 static pa_hook_result_t
process(struct userdata
*u
, pa_proplist
*p
) {
262 if (!(pn
= pa_proplist_gets(p
, PA_PROP_APPLICATION_PROCESS_BINARY
)))
265 if (*pn
== '.' || strchr(pn
, '/'))
270 pa_log_debug("Looking for .desktop file for %s", pn
);
272 if ((r
= pa_hashmap_get(u
->cache
, pn
))) {
273 if (now
-r
->timestamp
> STAT_INTERVAL
) {
280 r
= pa_xnew0(struct rule
, 1);
281 r
->process_name
= pa_xstrdup(pn
);
283 pa_hashmap_put(u
->cache
, r
->process_name
, r
);
291 static pa_hook_result_t
client_new_cb(pa_core
*core
, pa_client_new_data
*data
, struct userdata
*u
) {
292 pa_core_assert_ref(core
);
296 return process(u
, data
->proplist
);
299 static pa_hook_result_t
client_proplist_changed_cb(pa_core
*core
, pa_client
*client
, struct userdata
*u
) {
300 pa_core_assert_ref(core
);
304 return process(u
, client
->proplist
);
307 int pa__init(pa_module
*m
) {
308 pa_modargs
*ma
= NULL
;
313 if (!(ma
= pa_modargs_new(m
->argument
, valid_modargs
))) {
314 pa_log("Failed to parse module arguments");
318 m
->userdata
= u
= pa_xnew(struct userdata
, 1);
320 u
->cache
= pa_hashmap_new(pa_idxset_string_hash_func
, pa_idxset_string_compare_func
);
321 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
);
322 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
);
337 void pa__done(pa_module
*m
) {
342 if (!(u
= m
->userdata
))
345 if (u
->client_new_slot
)
346 pa_hook_slot_free(u
->client_new_slot
);
347 if (u
->client_proplist_changed_slot
)
348 pa_hook_slot_free(u
->client_proplist_changed_slot
);
353 while ((r
= pa_hashmap_steal_first(u
->cache
)))
356 pa_hashmap_free(u
->cache
, NULL
, NULL
);