ws2_32: Fix test output string to match the test.
[wine] / dlls / mountmgr.sys / hal.c
1 /*
2  * HAL devices support
3  *
4  * Copyright 2006 Alexandre Julliard
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19  */
20
21 #include "config.h"
22 #include "wine/port.h"
23
24 #include <assert.h>
25 #include <errno.h>
26 #include <stdarg.h>
27 #include <stdio.h>
28 #include <sys/time.h>
29
30 #include "windef.h"
31 #include "winbase.h"
32 #include "winreg.h"
33 #include "winuser.h"
34 #include "excpt.h"
35
36 #include "wine/library.h"
37 #include "wine/exception.h"
38 #include "wine/debug.h"
39 #include "mountmgr.h"
40
41 WINE_DEFAULT_DEBUG_CHANNEL(mountmgr);
42
43 #ifdef SONAME_LIBHAL
44
45 #include <dbus/dbus.h>
46 #include <hal/libhal.h>
47
48 #define DBUS_FUNCS \
49     DO_FUNC(dbus_bus_get); \
50     DO_FUNC(dbus_connection_close); \
51     DO_FUNC(dbus_connection_read_write_dispatch); \
52     DO_FUNC(dbus_error_init); \
53     DO_FUNC(dbus_error_free); \
54     DO_FUNC(dbus_error_is_set)
55
56 #define HAL_FUNCS \
57     DO_FUNC(libhal_ctx_free); \
58     DO_FUNC(libhal_ctx_init); \
59     DO_FUNC(libhal_ctx_new); \
60     DO_FUNC(libhal_ctx_set_dbus_connection); \
61     DO_FUNC(libhal_ctx_set_device_added); \
62     DO_FUNC(libhal_ctx_set_device_property_modified); \
63     DO_FUNC(libhal_ctx_set_device_removed); \
64     DO_FUNC(libhal_ctx_shutdown); \
65     DO_FUNC(libhal_device_get_property_bool); \
66     DO_FUNC(libhal_device_get_property_string); \
67     DO_FUNC(libhal_device_add_property_watch); \
68     DO_FUNC(libhal_device_remove_property_watch); \
69     DO_FUNC(libhal_free_string); \
70     DO_FUNC(libhal_free_string_array); \
71     DO_FUNC(libhal_get_all_devices)
72
73 #define DO_FUNC(f) static typeof(f) * p_##f
74 DBUS_FUNCS;
75 HAL_FUNCS;
76 #undef DO_FUNC
77
78 static BOOL load_functions(void)
79 {
80     void *hal_handle;
81     char error[128];
82
83     /* Load libhal with RTLD_GLOBAL so that the dbus symbols are available.
84      * We can't load libdbus directly since libhal may have been built against a
85      * different version but with the same soname. Binary compatibility is for wimps. */
86
87     if (!(hal_handle = wine_dlopen(SONAME_LIBHAL, RTLD_NOW|RTLD_GLOBAL, error, sizeof(error))))
88         goto failed;
89
90 #define DO_FUNC(f) if (!(p_##f = wine_dlsym( RTLD_DEFAULT, #f, error, sizeof(error) ))) goto failed
91     DBUS_FUNCS;
92 #undef DO_FUNC
93
94 #define DO_FUNC(f) if (!(p_##f = wine_dlsym( hal_handle, #f, error, sizeof(error) ))) goto failed
95     HAL_FUNCS;
96 #undef DO_FUNC
97
98     return TRUE;
99
100 failed:
101     WARN( "failed to load HAL support: %s\n", error );
102     return FALSE;
103 }
104
105 static LONG WINAPI assert_fault(EXCEPTION_POINTERS *eptr)
106 {
107     if (eptr->ExceptionRecord->ExceptionCode == EXCEPTION_WINE_ASSERTION)
108         return EXCEPTION_EXECUTE_HANDLER;
109     return EXCEPTION_CONTINUE_SEARCH;
110 }
111
112 /* HAL callback for new device */
113 static void new_device( LibHalContext *ctx, const char *udi )
114 {
115     DBusError error;
116     char *parent = NULL;
117     char *mount_point = NULL;
118     char *device = NULL;
119     char *type = NULL;
120
121     p_dbus_error_init( &error );
122
123     if (!(device = p_libhal_device_get_property_string( ctx, udi, "block.device", &error )))
124         goto done;
125
126     if (!(mount_point = p_libhal_device_get_property_string( ctx, udi, "volume.mount_point", &error )))
127         goto done;
128
129     if (!(parent = p_libhal_device_get_property_string( ctx, udi, "info.parent", &error )))
130         goto done;
131
132     if (!p_libhal_device_get_property_bool( ctx, parent, "storage.removable", &error ))
133         goto done;
134
135     if (!(type = p_libhal_device_get_property_string( ctx, parent, "storage.drive_type", &error )))
136         p_dbus_error_free( &error );  /* ignore error */
137
138     add_dos_device( udi, device, mount_point, type );
139
140     /* add property watch for mount point */
141     p_libhal_device_add_property_watch( ctx, udi, &error );
142
143 done:
144     if (type) p_libhal_free_string( type );
145     if (parent) p_libhal_free_string( parent );
146     if (device) p_libhal_free_string( device );
147     if (mount_point) p_libhal_free_string( mount_point );
148     p_dbus_error_free( &error );
149 }
150
151 /* HAL callback for removed device */
152 static void removed_device( LibHalContext *ctx, const char *udi )
153 {
154     DBusError error;
155
156     TRACE( "removed %s\n", wine_dbgstr_a(udi) );
157
158     if (remove_dos_device( udi ))
159     {
160         p_dbus_error_init( &error );
161         p_libhal_device_remove_property_watch( ctx, udi, &error );
162         p_dbus_error_free( &error );
163     }
164 }
165
166 /* HAL callback for property changes */
167 static void property_modified (LibHalContext *ctx, const char *udi,
168                                const char *key, dbus_bool_t is_removed, dbus_bool_t is_added)
169 {
170     TRACE( "udi %s key %s %s\n", wine_dbgstr_a(udi), wine_dbgstr_a(key),
171                 is_added ? "added" : is_removed ? "removed" : "modified" );
172
173     if (!strcmp( key, "volume.mount_point" )) new_device( ctx, udi );
174 }
175
176
177 static DWORD WINAPI hal_thread( void *arg )
178 {
179     DBusError error;
180     DBusConnection *dbc;
181     LibHalContext *ctx;
182     int i, num;
183     char **list;
184
185     if (!(ctx = p_libhal_ctx_new())) return 1;
186
187     p_dbus_error_init( &error );
188     if (!(dbc = p_dbus_bus_get( DBUS_BUS_SYSTEM, &error )))
189     {
190         WARN( "failed to get system dbus connection: %s\n", error.message );
191         p_dbus_error_free( &error );
192         return 1;
193     }
194
195     p_libhal_ctx_set_dbus_connection( ctx, dbc );
196     p_libhal_ctx_set_device_added( ctx, new_device );
197     p_libhal_ctx_set_device_removed( ctx, removed_device );
198     p_libhal_ctx_set_device_property_modified( ctx, property_modified );
199
200     if (!p_libhal_ctx_init( ctx, &error ))
201     {
202         WARN( "HAL context init failed: %s\n", error.message );
203         p_dbus_error_free( &error );
204         return 1;
205     }
206
207     /* retrieve all existing devices */
208     if (!(list = p_libhal_get_all_devices( ctx, &num, &error ))) p_dbus_error_free( &error );
209     else
210     {
211         for (i = 0; i < num; i++) new_device( ctx, list[i] );
212         p_libhal_free_string_array( list );
213     }
214
215     __TRY
216     {
217         while (p_dbus_connection_read_write_dispatch( dbc, -1 )) /* nothing */ ;
218     }
219     __EXCEPT( assert_fault )
220     {
221         WARN( "dbus assertion failure, disabling HAL support\n" );
222         return 1;
223     }
224     __ENDTRY;
225
226     p_libhal_ctx_shutdown( ctx, &error );
227     p_dbus_error_free( &error );  /* just in case */
228     p_dbus_connection_close( dbc );
229     p_libhal_ctx_free( ctx );
230     return 0;
231 }
232
233 void initialize_hal(void)
234 {
235     HANDLE handle;
236
237     if (!load_functions()) return;
238     if (!(handle = CreateThread( NULL, 0, hal_thread, NULL, 0, NULL ))) return;
239     CloseHandle( handle );
240 }
241
242 #else  /* SONAME_LIBHAL */
243
244 void initialize_hal(void)
245 {
246     TRACE( "Skipping, HAL support not compiled in\n" );
247 }
248
249 #endif  /* SONAME_LIBHAL */