Plug a leak
[platform/upstream/glib.git] / gio / gunixvolumemonitor.c
1 /* -*- mode: C; c-file-style: "gnu"; indent-tabs-mode: nil; -*- */
2
3 /* GIO - GLib Input, Output and Streaming Library
4  * 
5  * Copyright (C) 2006-2007 Red Hat, Inc.
6  *
7  * This library is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2 of the License, or (at your option) any later version.
11  *
12  * This library is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General
18  * Public License along with this library; if not, write to the
19  * Free Software Foundation, Inc., 59 Temple Place, Suite 330,
20  * Boston, MA 02111-1307, USA.
21  *
22  * Author: Alexander Larsson <alexl@redhat.com>
23  *         David Zeuthen <davidz@redhat.com>
24  */
25
26 #include "config.h"
27
28 #include <string.h>
29
30 #include <glib.h>
31 #include "gunixvolumemonitor.h"
32 #include "gunixmounts.h"
33 #include "gunixmount.h"
34 #include "gunixvolume.h"
35 #include "gmount.h"
36 #include "gmountprivate.h"
37 #include "giomodule.h"
38 #include "glibintl.h"
39
40 #include "gioalias.h"
41
42 struct _GUnixVolumeMonitor {
43   GNativeVolumeMonitor parent;
44
45   GUnixMountMonitor *mount_monitor;
46
47   GList *last_mountpoints;
48   GList *last_mounts;
49
50   GList *volumes;
51   GList *mounts;
52 };
53
54 static void mountpoints_changed      (GUnixMountMonitor  *mount_monitor,
55                                       gpointer            user_data);
56 static void mounts_changed           (GUnixMountMonitor  *mount_monitor,
57                                       gpointer            user_data);
58 static void update_volumes           (GUnixVolumeMonitor *monitor);
59 static void update_mounts            (GUnixVolumeMonitor *monitor);
60
61 #define g_unix_volume_monitor_get_type _g_unix_volume_monitor_get_type
62 G_DEFINE_TYPE_WITH_CODE (GUnixVolumeMonitor, g_unix_volume_monitor, G_TYPE_NATIVE_VOLUME_MONITOR,
63                          g_io_extension_point_implement (G_NATIVE_VOLUME_MONITOR_EXTENSION_POINT_NAME,
64                                                          g_define_type_id,
65                                                          "unix",
66                                                          0));
67
68 static void
69 g_unix_volume_monitor_finalize (GObject *object)
70 {
71   GUnixVolumeMonitor *monitor;
72   
73   monitor = G_UNIX_VOLUME_MONITOR (object);
74
75   g_signal_handlers_disconnect_by_func (monitor->mount_monitor, mountpoints_changed, monitor);
76   g_signal_handlers_disconnect_by_func (monitor->mount_monitor, mounts_changed, monitor);
77                                         
78   g_object_unref (monitor->mount_monitor);
79
80   g_list_foreach (monitor->last_mountpoints, (GFunc)g_unix_mount_point_free, NULL);
81   g_list_free (monitor->last_mountpoints);
82   g_list_foreach (monitor->last_mounts, (GFunc)g_unix_mount_free, NULL);
83   g_list_free (monitor->last_mounts);
84
85   g_list_foreach (monitor->volumes, (GFunc)g_object_unref, NULL);
86   g_list_free (monitor->volumes);
87   g_list_foreach (monitor->mounts, (GFunc)g_object_unref, NULL);
88   g_list_free (monitor->mounts);
89
90   G_OBJECT_CLASS (g_unix_volume_monitor_parent_class)->finalize (object);
91 }
92
93 static GList *
94 get_mounts (GVolumeMonitor *volume_monitor)
95 {
96   GUnixVolumeMonitor *monitor;
97   GList *l;
98   
99   monitor = G_UNIX_VOLUME_MONITOR (volume_monitor);
100
101   l = g_list_copy (monitor->mounts);
102   g_list_foreach (l, (GFunc)g_object_ref, NULL);
103
104   return l;
105 }
106
107 static GList *
108 get_volumes (GVolumeMonitor *volume_monitor)
109 {
110   GUnixVolumeMonitor *monitor;
111   GList *l;
112   
113   monitor = G_UNIX_VOLUME_MONITOR (volume_monitor);
114
115   l = g_list_copy (monitor->volumes);
116   g_list_foreach (l, (GFunc)g_object_ref, NULL);
117
118   return l;
119 }
120
121 static GList *
122 get_connected_drives (GVolumeMonitor *volume_monitor)
123 {
124   return NULL;
125 }
126
127 static GVolume *
128 get_volume_for_uuid (GVolumeMonitor *volume_monitor, const char *uuid)
129 {
130   return NULL;
131 }
132
133 static GMount *
134 get_mount_for_uuid (GVolumeMonitor *volume_monitor, const char *uuid)
135 {
136   return NULL;
137 }
138
139 static gboolean
140 is_supported (void)
141 {
142   return TRUE;
143 }
144
145 static GMount *
146 get_mount_for_mount_path (const char *mount_path,
147                           GCancellable *cancellable)
148 {
149   GUnixMountEntry *mount_entry;
150   GUnixMount *mount;
151
152   mount_entry = g_unix_mount_at (mount_path, NULL);
153   
154   /* TODO: Set mountable volume? */
155   mount = _g_unix_mount_new (NULL, mount_entry, NULL);
156
157   g_unix_mount_free (mount_entry);
158
159   return G_MOUNT (mount);
160 }
161
162 static void
163 g_unix_volume_monitor_class_init (GUnixVolumeMonitorClass *klass)
164 {
165   GObjectClass *gobject_class = G_OBJECT_CLASS (klass);
166   GVolumeMonitorClass *monitor_class = G_VOLUME_MONITOR_CLASS (klass);
167   GNativeVolumeMonitorClass *native_class = G_NATIVE_VOLUME_MONITOR_CLASS (klass);
168   
169   gobject_class->finalize = g_unix_volume_monitor_finalize;
170
171   monitor_class->get_mounts = get_mounts;
172   monitor_class->get_volumes = get_volumes;
173   monitor_class->get_connected_drives = get_connected_drives;
174   monitor_class->get_volume_for_uuid = get_volume_for_uuid;
175   monitor_class->get_mount_for_uuid = get_mount_for_uuid;
176   monitor_class->is_supported = is_supported;
177
178   native_class->get_mount_for_mount_path = get_mount_for_mount_path;
179 }
180
181 static void
182 mountpoints_changed (GUnixMountMonitor *mount_monitor,
183                      gpointer           user_data)
184 {
185   GUnixVolumeMonitor *unix_monitor = user_data;
186
187   /* Update both to make sure volumes are created before mounts */
188   update_volumes (unix_monitor);
189   update_mounts (unix_monitor);
190 }
191
192 static void
193 mounts_changed (GUnixMountMonitor *mount_monitor,
194                 gpointer           user_data)
195 {
196   GUnixVolumeMonitor *unix_monitor = user_data;
197
198   /* Update both to make sure volumes are created before mounts */
199   update_volumes (unix_monitor);
200   update_mounts (unix_monitor);
201 }
202
203 static void
204 g_unix_volume_monitor_init (GUnixVolumeMonitor *unix_monitor)
205 {
206
207   unix_monitor->mount_monitor = g_unix_mount_monitor_new ();
208
209   g_signal_connect (unix_monitor->mount_monitor,
210                     "mounts_changed", G_CALLBACK (mounts_changed),
211                     unix_monitor);
212   
213   g_signal_connect (unix_monitor->mount_monitor,
214                     "mountpoints_changed", G_CALLBACK (mountpoints_changed),
215                     unix_monitor);
216                     
217   update_volumes (unix_monitor);
218   update_mounts (unix_monitor);
219 }
220
221 /**
222  * g_unix_volume_monitor_new:
223  * 
224  * Returns:  a new #GVolumeMonitor.
225  **/
226 GVolumeMonitor *
227 _g_unix_volume_monitor_new (void)
228 {
229   GUnixVolumeMonitor *monitor;
230
231   monitor = g_object_new (G_TYPE_UNIX_VOLUME_MONITOR, NULL);
232   
233   return G_VOLUME_MONITOR (monitor);
234 }
235
236 static void
237 diff_sorted_lists (GList         *list1, 
238                    GList         *list2, 
239                    GCompareFunc   compare,
240                    GList        **added, 
241                    GList        **removed)
242 {
243   int order;
244   
245   *added = *removed = NULL;
246   
247   while (list1 != NULL &&
248          list2 != NULL)
249     {
250       order = (*compare) (list1->data, list2->data);
251       if (order < 0)
252         {
253           *removed = g_list_prepend (*removed, list1->data);
254           list1 = list1->next;
255         }
256       else if (order > 0)
257         {
258           *added = g_list_prepend (*added, list2->data);
259           list2 = list2->next;
260         }
261       else
262         { /* same item */
263           list1 = list1->next;
264           list2 = list2->next;
265         }
266     }
267
268   while (list1 != NULL)
269     {
270       *removed = g_list_prepend (*removed, list1->data);
271       list1 = list1->next;
272     }
273   while (list2 != NULL)
274     {
275       *added = g_list_prepend (*added, list2->data);
276       list2 = list2->next;
277     }
278 }
279
280 /**
281  * _g_unix_volume_monitor_lookup_volume_for_mount_path: 
282  * @monitor:
283  * @mount_path:
284  * 
285  * Returns:  #GUnixVolume for the given @mount_path.
286  **/
287 GUnixVolume *
288 _g_unix_volume_monitor_lookup_volume_for_mount_path (GUnixVolumeMonitor *monitor,
289                                                      const char         *mount_path)
290 {
291   GList *l;
292
293   for (l = monitor->volumes; l != NULL; l = l->next)
294     {
295       GUnixVolume *volume = l->data;
296
297       if (_g_unix_volume_has_mount_path (volume, mount_path))
298         return volume;
299     }
300   
301   return NULL;
302 }
303
304 static GUnixMount *
305 find_mount_by_mountpath (GUnixVolumeMonitor *monitor,
306                          const char *mount_path)
307 {
308   GList *l;
309
310   for (l = monitor->mounts; l != NULL; l = l->next)
311     {
312       GUnixMount *mount = l->data;
313
314       if (_g_unix_mount_has_mount_path (mount, mount_path))
315         return mount;
316     }
317   
318   return NULL;
319 }
320
321 static void
322 update_volumes (GUnixVolumeMonitor *monitor)
323 {
324   GList *new_mountpoints;
325   GList *removed, *added;
326   GList *l;
327   GUnixVolume *volume;
328   
329   new_mountpoints = g_unix_mount_points_get (NULL);
330   
331   new_mountpoints = g_list_sort (new_mountpoints, (GCompareFunc) g_unix_mount_point_compare);
332   
333   diff_sorted_lists (monitor->last_mountpoints,
334                      new_mountpoints, (GCompareFunc) g_unix_mount_point_compare,
335                      &added, &removed);
336   
337   for (l = removed; l != NULL; l = l->next)
338     {
339       GUnixMountPoint *mountpoint = l->data;
340       
341       volume = _g_unix_volume_monitor_lookup_volume_for_mount_path (monitor,
342                                                                     g_unix_mount_point_get_mount_path (mountpoint));
343       if (volume)
344         {
345           _g_unix_volume_disconnected (volume);
346           monitor->volumes = g_list_remove (monitor->volumes, volume);
347           g_signal_emit_by_name (monitor, "volume_removed", volume);
348           g_signal_emit_by_name (volume, "removed");
349           g_object_unref (volume);
350         }
351     }
352   
353   for (l = added; l != NULL; l = l->next)
354     {
355       GUnixMountPoint *mountpoint = l->data;
356       
357       volume = _g_unix_volume_new (G_VOLUME_MONITOR (monitor), mountpoint);
358       if (volume)
359         {
360           monitor->volumes = g_list_prepend (monitor->volumes, volume);
361           g_signal_emit_by_name (monitor, "volume_added", volume);
362         }
363     }
364   
365   g_list_free (added);
366   g_list_free (removed);
367   g_list_foreach (monitor->last_mountpoints,
368                   (GFunc)g_unix_mount_point_free, NULL);
369   g_list_free (monitor->last_mountpoints);
370   monitor->last_mountpoints = new_mountpoints;
371 }
372
373 static void
374 update_mounts (GUnixVolumeMonitor *monitor)
375 {
376   GList *new_mounts;
377   GList *removed, *added;
378   GList *l;
379   GUnixMount *mount;
380   GUnixVolume *volume;
381   const char *mount_path;
382   
383   new_mounts = g_unix_mounts_get (NULL);
384   
385   new_mounts = g_list_sort (new_mounts, (GCompareFunc) g_unix_mount_compare);
386   
387   diff_sorted_lists (monitor->last_mounts,
388                      new_mounts, (GCompareFunc) g_unix_mount_compare,
389                      &added, &removed);
390   
391   for (l = removed; l != NULL; l = l->next)
392     {
393       GUnixMountEntry *mount_entry = l->data;
394       
395       mount = find_mount_by_mountpath (monitor, g_unix_mount_get_mount_path (mount_entry));
396       if (mount)
397         {
398           _g_unix_mount_unmounted (mount);
399           monitor->mounts = g_list_remove (monitor->mounts, mount);
400           g_signal_emit_by_name (monitor, "mount_removed", mount);
401           g_signal_emit_by_name (mount, "unmounted");
402           g_object_unref (mount);
403         }
404     }
405   
406   for (l = added; l != NULL; l = l->next)
407     {
408       GUnixMountEntry *mount_entry = l->data;
409
410       mount_path = g_unix_mount_get_mount_path (mount_entry);
411       
412       volume = _g_unix_volume_monitor_lookup_volume_for_mount_path (monitor, mount_path);
413       mount = _g_unix_mount_new (G_VOLUME_MONITOR (monitor), mount_entry, volume);
414       if (mount)
415         {
416           monitor->mounts = g_list_prepend (monitor->mounts, mount);
417           g_signal_emit_by_name (monitor, "mount_added", mount);
418         }
419     }
420   
421   g_list_free (added);
422   g_list_free (removed);
423   g_list_foreach (monitor->last_mounts,
424                   (GFunc)g_unix_mount_free, NULL);
425   g_list_free (monitor->last_mounts);
426   monitor->last_mounts = new_mounts;
427 }