1 /*************************************************************************** 2 * CVSID: $Id: hal-storage-mount.c,v 1.7 2006/06/21 00:44:03 david Exp $ 3 * 4 * hal-storage-cleanup-mountpoint.c : Cleanup mount point when hald detects 5 * that an unmount not done through Unmount() 6 * 7 * Copyright (C) 2006 David Zeuthen, <david@fubar.dk> 8 * 9 * This program is free software; you can redistribute it and/or modify 10 * it under the terms of the GNU General Public License as published by 11 * the Free Software Foundation; either version 2 of the License, or 12 * (at your option) any later version. 13 * 14 * This program is distributed in the hope that it will be useful, 15 * but WITHOUT ANY WARRANTY; without even the implied warranty of 16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 17 * GNU General Public License for more details. 18 * 19 * You should have received a copy of the GNU General Public License 20 * along with this program; if not, write to the Free Software 21 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA 22 * 23 **************************************************************************/ 24 25 #ifdef HAVE_CONFIG_H 26 # include <config.h> 27 #endif 28 29 #include <unistd.h> 30 #include <stdio.h> 31 #include <stdlib.h> 32 #include <string.h> 33 #include <glib.h> 34 #include <glib/gstdio.h> 35 36 #include "hal-storage-shared.h" 37 38 /*#define DEBUG*/ 39 #define DEBUG 40 41 static void 42 usage (void) 43 { 44 fprintf (stderr, "This program should only be started by hald.\n"); 45 exit (1); 46 } 47 48 static void 49 do_cleanup (const char *mount_point) 50 { 51 int i, j; 52 FILE *hal_mtab_orig; 53 int hal_mtab_orig_len; 54 int num_read; 55 char *hal_mtab_buf; 56 char **lines; 57 FILE *hal_mtab_new; 58 gboolean found; 59 60 hal_mtab_orig = fopen ("/media/.hal-mtab", "r"); 61 if (hal_mtab_orig == NULL) { 62 unknown_error ("Cannot open /media/.hal-mtab"); 63 } 64 if (fseek (hal_mtab_orig, 0L, SEEK_END) != 0) { 65 unknown_error ("Cannot seek to end of /media/.hal-mtab"); 66 } 67 hal_mtab_orig_len = ftell (hal_mtab_orig); 68 if (hal_mtab_orig_len < 0) { 69 unknown_error ("Cannot determine size of /media/.hal-mtab"); 70 } 71 rewind (hal_mtab_orig); 72 hal_mtab_buf = g_new0 (char, hal_mtab_orig_len + 1); 73 num_read = fread (hal_mtab_buf, 1, hal_mtab_orig_len, hal_mtab_orig); 74 if (num_read != hal_mtab_orig_len) { 75 unknown_error ("Cannot read from /media/.hal-mtab"); 76 } 77 fclose (hal_mtab_orig); 78 79 #ifdef DEBUG 80 printf ("hal_mtab = '%s'\n", hal_mtab_buf); 81 #endif 82 83 lines = g_strsplit (hal_mtab_buf, "\n", 0); 84 g_free (hal_mtab_buf); 85 86 /* find the entry we're going to unmount */ 87 found = FALSE; 88 for (i = 0; lines[i] != NULL && !found; i++) { 89 char **line_elements; 90 91 #ifdef DEBUG 92 printf (" line = '%s'\n", lines[i]); 93 #endif 94 95 if ((lines[i])[0] == '#') 96 continue; 97 98 line_elements = g_strsplit (lines[i], "\t", 6); 99 if (g_strv_length (line_elements) == 6) { 100 101 #ifdef DEBUG 102 printf (" devfile = '%s'\n", line_elements[0]); 103 printf (" uid = '%s'\n", line_elements[1]); 104 printf (" session id = '%s'\n", line_elements[2]); 105 printf (" fs = '%s'\n", line_elements[3]); 106 printf (" options = '%s'\n", line_elements[4]); 107 printf (" mount_point = '%s'\n", line_elements[5]); 108 #endif 109 110 if (strcmp (line_elements[5], mount_point) == 0) { 111 char *line_to_free; 112 113 found = TRUE; 114 115 line_to_free = lines[i]; 116 for (j = i; lines[j] != NULL; j++) { 117 lines[j] = lines[j+1]; 118 } 119 lines[j] = NULL; 120 g_free (line_to_free); 121 } 122 } 123 124 g_strfreev (line_elements); 125 } 126 127 if (!found) { 128 unknown_error ("mount point is not /media/.hal-mtab"); 129 } 130 131 #ifdef DEBUG 132 printf ("Found entry for mount point '%s' in /media/.hal-mtab", mount_point); 133 #endif 134 135 /* create new .hal-mtab~ file without the entry we're going to unmount */ 136 hal_mtab_new = fopen ("/media/.hal-mtab~", "w"); 137 if (hal_mtab_new == NULL) { 138 unknown_error ("Cannot create /media/.hal-mtab~"); 139 } 140 for (i = 0; lines[i] != NULL; i++) { 141 if (i > 0) { 142 char anewl[2] = "\n\0"; 143 if (fwrite (anewl, 1, 1, hal_mtab_new) != 1) { 144 unknown_error ("Cannot write to /media/.hal-mtab~"); 145 } 146 } 147 148 if (fwrite (lines[i], 1, strlen (lines[i]), hal_mtab_new) != strlen (lines[i])) { 149 unknown_error ("Cannot write to /media/.hal-mtab~"); 150 } 151 152 } 153 fclose (hal_mtab_new); 154 155 g_strfreev (lines); 156 157 /* remove directory */ 158 if (g_rmdir (mount_point) != 0) { 159 unlink ("/media/.hal-mtab~"); 160 unknown_error ("Cannot remove directory"); 161 } 162 163 /* set new .hal-mtab file */ 164 if (rename ("/media/.hal-mtab~", "/media/.hal-mtab") != 0) { 165 unlink ("/media/.hal-mtab~"); 166 unknown_error ("Cannot rename /media/.hal-mtab~ to /media/.hal-mtab"); 167 } 168 169 } 170 171 int 172 main (int argc, char *argv[]) 173 { 174 char *mount_point; 175 176 if (!lock_hal_mtab ()) { 177 unknown_error ("Cannot obtain lock on /media/.hal-mtab"); 178 } 179 180 mount_point = getenv ("HALD_CLEANUP"); 181 if (mount_point == NULL) 182 usage (); 183 184 #ifdef DEBUG 185 printf ("in hal-storage-cleanup-mountpoint for mount point '%s'\n", mount_point); 186 #endif 187 do_cleanup (mount_point); 188 189 190 unlock_hal_mtab (); 191 return 0; 192 } 193