118c2aff7Sartem /***************************************************************************
218c2aff7Sartem * CVSID: $Id$
318c2aff7Sartem *
418c2aff7Sartem * hal-storage-unmount.c : Unmount wrapper
518c2aff7Sartem *
618c2aff7Sartem * Copyright (C) 2006 David Zeuthen, <david@fubar.dk>
718c2aff7Sartem *
818c2aff7Sartem * This program is free software; you can redistribute it and/or modify
918c2aff7Sartem * it under the terms of the GNU General Public License as published by
1018c2aff7Sartem * the Free Software Foundation; either version 2 of the License, or
1118c2aff7Sartem * (at your option) any later version.
1218c2aff7Sartem *
1318c2aff7Sartem * This program is distributed in the hope that it will be useful,
1418c2aff7Sartem * but WITHOUT ANY WARRANTY; without even the implied warranty of
1518c2aff7Sartem * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
1618c2aff7Sartem * GNU General Public License for more details.
1718c2aff7Sartem *
1818c2aff7Sartem * You should have received a copy of the GNU General Public License
1918c2aff7Sartem * along with this program; if not, write to the Free Software
2018c2aff7Sartem * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
2118c2aff7Sartem *
2218c2aff7Sartem **************************************************************************/
2318c2aff7Sartem
2418c2aff7Sartem
2518c2aff7Sartem #ifdef HAVE_CONFIG_H
2618c2aff7Sartem # include <config.h>
2718c2aff7Sartem #endif
2818c2aff7Sartem
2918c2aff7Sartem #include <stdio.h>
3018c2aff7Sartem #include <stdlib.h>
3118c2aff7Sartem #include <string.h>
3218c2aff7Sartem #include <glib.h>
3318c2aff7Sartem #include <glib/gstdio.h>
3418c2aff7Sartem #ifdef __FreeBSD__
3518c2aff7Sartem #include <fstab.h>
3618c2aff7Sartem #include <sys/param.h>
3718c2aff7Sartem #include <sys/ucred.h>
3818c2aff7Sartem #include <sys/mount.h>
3918c2aff7Sartem #include <limits.h>
4018c2aff7Sartem #include <pwd.h>
4118c2aff7Sartem #elif sun
4218c2aff7Sartem #include <fcntl.h>
4318c2aff7Sartem #include <sys/mnttab.h>
4418c2aff7Sartem #include <sys/vfstab.h>
4518c2aff7Sartem #else
4618c2aff7Sartem #include <mntent.h>
4718c2aff7Sartem #endif
4818c2aff7Sartem #include <sys/types.h>
4918c2aff7Sartem #include <unistd.h>
5018c2aff7Sartem
5118c2aff7Sartem #include <libhal.h>
5218c2aff7Sartem #include <libhal-storage.h>
5318c2aff7Sartem #ifdef HAVE_POLKIT
5418c2aff7Sartem #include <libpolkit.h>
5518c2aff7Sartem #endif
5618c2aff7Sartem
5718c2aff7Sartem #include "hal-storage-shared.h"
5818c2aff7Sartem
5918c2aff7Sartem
6018c2aff7Sartem static void
usage(void)6118c2aff7Sartem usage (void)
6218c2aff7Sartem {
6318c2aff7Sartem fprintf (stderr, "This program should only be started by hald.\n");
6418c2aff7Sartem exit (1);
6518c2aff7Sartem }
6618c2aff7Sartem
6718c2aff7Sartem static void
invalid_unmount_option(const char * option,const char * uid)6818c2aff7Sartem invalid_unmount_option (const char *option, const char *uid)
6918c2aff7Sartem {
7018c2aff7Sartem fprintf (stderr, "org.freedesktop.Hal.Device.Volume.InvalidUnmountOption\n");
7118c2aff7Sartem fprintf (stderr, "The option '%s' is not allowed for uid=%s\n", option, uid);
7218c2aff7Sartem exit (1);
7318c2aff7Sartem }
7418c2aff7Sartem
7518c2aff7Sartem int
main(int argc,char * argv[])7618c2aff7Sartem main (int argc, char *argv[])
7718c2aff7Sartem {
7818c2aff7Sartem char *udi;
7918c2aff7Sartem char *device;
8018c2aff7Sartem LibHalVolume *volume;
8118c2aff7Sartem DBusError error;
8218c2aff7Sartem LibHalContext *hal_ctx = NULL;
8318c2aff7Sartem DBusConnection *system_bus = NULL;
8418c2aff7Sartem #ifdef HAVE_POLKIT
8518c2aff7Sartem LibPolKitContext *pol_ctx = NULL;
8618c2aff7Sartem #endif
8718c2aff7Sartem char *invoked_by_uid;
8818c2aff7Sartem char *invoked_by_syscon_name;
8918c2aff7Sartem int i;
9018c2aff7Sartem char unmount_options[1024];
9118c2aff7Sartem char **given_options;
9218c2aff7Sartem gboolean use_lazy;
9318c2aff7Sartem gboolean use_force;
9418c2aff7Sartem const char *end;
9518c2aff7Sartem
9618c2aff7Sartem if (!lock_hal_mtab ()) {
9718c2aff7Sartem unknown_error ("Cannot obtain lock on /media/.hal-mtab");
9818c2aff7Sartem }
9918c2aff7Sartem
10018c2aff7Sartem device = getenv ("HAL_PROP_BLOCK_DEVICE");
10118c2aff7Sartem if (device == NULL)
10218c2aff7Sartem usage ();
10318c2aff7Sartem
10418c2aff7Sartem udi = getenv ("HAL_PROP_INFO_UDI");
10518c2aff7Sartem if (udi == NULL)
10618c2aff7Sartem usage ();
10718c2aff7Sartem
10818c2aff7Sartem invoked_by_uid = getenv ("HAL_METHOD_INVOKED_BY_UID");
10918c2aff7Sartem
11018c2aff7Sartem invoked_by_syscon_name = getenv ("HAL_METHOD_INVOKED_BY_SYSTEMBUS_CONNECTION_NAME");
11118c2aff7Sartem
11218c2aff7Sartem dbus_error_init (&error);
11318c2aff7Sartem if ((hal_ctx = libhal_ctx_init_direct (&error)) == NULL) {
11418c2aff7Sartem printf ("Cannot connect to hald\n");
11518c2aff7Sartem LIBHAL_FREE_DBUS_ERROR (&error);
11618c2aff7Sartem usage ();
11718c2aff7Sartem }
11818c2aff7Sartem
11918c2aff7Sartem dbus_error_init (&error);
12018c2aff7Sartem system_bus = dbus_bus_get (DBUS_BUS_SYSTEM, &error);
12118c2aff7Sartem if (system_bus == NULL) {
12218c2aff7Sartem printf ("Cannot connect to the system bus\n");
12318c2aff7Sartem LIBHAL_FREE_DBUS_ERROR (&error);
12418c2aff7Sartem usage ();
12518c2aff7Sartem }
12618c2aff7Sartem #ifdef HAVE_POLKIT
12718c2aff7Sartem pol_ctx = libpolkit_new_context (system_bus);
12818c2aff7Sartem if (pol_ctx == NULL) {
12918c2aff7Sartem printf ("Cannot get libpolkit context\n");
13018c2aff7Sartem unknown_error ("Cannot get libpolkit context");
13118c2aff7Sartem }
13218c2aff7Sartem #endif
13318c2aff7Sartem
13418c2aff7Sartem /* read from stdin */
13518c2aff7Sartem if (strlen (fgets (unmount_options, sizeof (unmount_options), stdin)) > 0)
13618c2aff7Sartem unmount_options [strlen (unmount_options) - 1] = '\0';
13718c2aff7Sartem /* validate that input from stdin is UTF-8 */
13818c2aff7Sartem if (!g_utf8_validate (unmount_options, -1, &end))
13918c2aff7Sartem unknown_error ("Error validating unmount_options as UTF-8");
14018c2aff7Sartem #ifdef DEBUG
14118c2aff7Sartem printf ("unmount_options = '%s'\n", unmount_options);
14218c2aff7Sartem #endif
14318c2aff7Sartem
14418c2aff7Sartem /* delete any trailing whitespace options from splitting the string */
14518c2aff7Sartem given_options = g_strsplit (unmount_options, "\t", 0);
14618c2aff7Sartem for (i = g_strv_length (given_options) - 1; i >= 0; --i) {
14718c2aff7Sartem if (strlen (given_options[i]) > 0)
14818c2aff7Sartem break;
14918c2aff7Sartem given_options[i] = NULL;
15018c2aff7Sartem }
151*55fea89dSDan Cross
15218c2aff7Sartem use_lazy = FALSE;
15318c2aff7Sartem use_force = FALSE;
15418c2aff7Sartem
15518c2aff7Sartem /* check unmount options */
15618c2aff7Sartem for (i = 0; given_options[i] != NULL; i++) {
15718c2aff7Sartem char *given = given_options[i];
15818c2aff7Sartem
15918c2aff7Sartem if (strcmp (given, "lazy") == 0) {
16018c2aff7Sartem use_lazy = TRUE;
16118c2aff7Sartem } else if (strcmp (given, "force") == 0) {
16218c2aff7Sartem use_force = TRUE;
16318c2aff7Sartem } else {
16418c2aff7Sartem invalid_unmount_option (given, invoked_by_uid);
16518c2aff7Sartem }
16618c2aff7Sartem }
16718c2aff7Sartem g_strfreev (given_options);
16818c2aff7Sartem
16918c2aff7Sartem
17018c2aff7Sartem volume = libhal_volume_from_udi (hal_ctx, udi);
17118c2aff7Sartem if (volume == NULL) {
17218c2aff7Sartem LibHalDrive *drive;
17318c2aff7Sartem
17418c2aff7Sartem drive = libhal_drive_from_udi (hal_ctx, udi);
17518c2aff7Sartem if (drive == NULL) {
17618c2aff7Sartem usage ();
17718c2aff7Sartem } else {
178*55fea89dSDan Cross handle_unmount (hal_ctx,
17918c2aff7Sartem #ifdef HAVE_POLKIT
180*55fea89dSDan Cross pol_ctx,
18118c2aff7Sartem #endif
182*55fea89dSDan Cross udi, NULL, drive, device, invoked_by_uid,
18318c2aff7Sartem invoked_by_syscon_name, use_lazy, use_force,
18418c2aff7Sartem system_bus);
18518c2aff7Sartem }
18618c2aff7Sartem
18718c2aff7Sartem } else {
18818c2aff7Sartem const char *drive_udi;
18918c2aff7Sartem LibHalDrive *drive;
19018c2aff7Sartem
19118c2aff7Sartem drive_udi = libhal_volume_get_storage_device_udi (volume);
192*55fea89dSDan Cross
19318c2aff7Sartem if (drive_udi == NULL)
19418c2aff7Sartem unknown_error ("Cannot get drive_udi from volume");
19518c2aff7Sartem drive = libhal_drive_from_udi (hal_ctx, drive_udi);
19618c2aff7Sartem if (drive == NULL)
19718c2aff7Sartem unknown_error ("Cannot get drive from hal");
19818c2aff7Sartem
199*55fea89dSDan Cross handle_unmount (hal_ctx,
20018c2aff7Sartem #ifdef HAVE_POLKIT
201*55fea89dSDan Cross pol_ctx,
20218c2aff7Sartem #endif
203*55fea89dSDan Cross udi, volume, drive, device, invoked_by_uid,
20418c2aff7Sartem invoked_by_syscon_name, use_lazy, use_force,
20518c2aff7Sartem system_bus);
20618c2aff7Sartem
20718c2aff7Sartem }
20818c2aff7Sartem
20918c2aff7Sartem unlock_hal_mtab ();
21018c2aff7Sartem
21118c2aff7Sartem return 0;
21218c2aff7Sartem }
213