1 /* 2 * CDDL HEADER START 3 * 4 * The contents of this file are subject to the terms of the 5 * Common Development and Distribution License (the "License"). 6 * You may not use this file except in compliance with the License. 7 * 8 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE 9 * or http://www.opensolaris.org/os/licensing. 10 * See the License for the specific language governing permissions 11 * and limitations under the License. 12 * 13 * When distributing Covered Code, include this CDDL HEADER in each 14 * file and include the License file at usr/src/OPENSOLARIS.LICENSE. 15 * If applicable, add the following below this CDDL HEADER, with the 16 * fields enclosed by brackets "[]" replaced with your own identifying 17 * information: Portions Copyright [yyyy] [name of copyright owner] 18 * 19 * CDDL HEADER END 20 */ 21 /* 22 * Copyright 2009 Sun Microsystems, Inc. All rights reserved. 23 * Use is subject to license terms. 24 */ 25 26 #include <stdio.h> 27 #include <stdlib.h> 28 #include <sys/types.h> 29 #include <sys/stat.h> 30 #include <dirent.h> 31 #include <signal.h> 32 #include <unistd.h> 33 #include <fcntl.h> 34 #include <strings.h> 35 #include <libgen.h> 36 #include <libintl.h> 37 #include <errno.h> 38 #include <sys/syscall.h> 39 40 #include <dbus/dbus.h> 41 #include <dbus/dbus-glib.h> 42 #include <dbus/dbus-glib-lowlevel.h> 43 #include <libhal.h> 44 45 #include <rmm_common.h> 46 47 char *progname; 48 49 static boolean_t d_opt, l_opt, o_opt, u_opt, eject_opt, 50 closetray_opt, query_opt; 51 52 static void usage(); 53 static void nomem(); 54 55 static void 56 usage() 57 { 58 if (!u_opt) { 59 (void) fprintf(stderr, 60 "%s: [-dlu] [-o options] [nickname | device] " 61 "[mount_point]\n", progname); 62 } else { 63 (void) fprintf(stderr, 64 "%s: [-dl] [nickname | device]\n", progname); 65 } 66 } 67 68 static int 69 rmmount(int argc, char **argv) 70 { 71 int c; 72 action_t action; 73 LibHalContext *hal_ctx; 74 DBusError error; 75 rmm_error_t rmm_error; 76 LibHalDrive *d; 77 GSList *volumes; 78 const char *default_name; 79 char **opts = NULL; 80 int num_opts = 0; 81 char *mountpoint = NULL; 82 char **p; 83 int print_mask; 84 int ret = 0; 85 86 progname = basename(argv[0]); 87 88 if (strcmp(progname, "rmumount") == 0) { 89 u_opt = B_TRUE; 90 } 91 92 if (getenv("RMMOUNT_DEBUG") != NULL) { 93 rmm_debug = 1; 94 } 95 96 while ((c = getopt(argc, argv, "?dlo:u")) != -1) { 97 switch (c) { 98 case 'd': 99 d_opt = B_TRUE; 100 break; 101 case 'l': 102 l_opt = B_TRUE; 103 break; 104 case 'o': 105 o_opt = B_TRUE; 106 if ((opts = g_strsplit(optarg, ",", 10)) == NULL) { 107 nomem(); 108 } 109 for (num_opts = 0, p = &opts[0]; *p != NULL; p++) { 110 num_opts++; 111 } 112 break; 113 case 'u': 114 u_opt = B_TRUE; 115 break; 116 case '?': 117 usage(); 118 return (0); 119 default: 120 usage(); 121 return (1); 122 } 123 } 124 125 if (u_opt) { 126 action = UNMOUNT; 127 } else if (closetray_opt) { 128 action = CLOSETRAY; 129 } else if (eject_opt) { 130 action = EJECT; 131 } else { 132 action = INSERT; 133 } 134 135 if ((hal_ctx = rmm_hal_init(0, 0, 0, 0, &error, &rmm_error)) == NULL) { 136 (void) fprintf(stderr, gettext("warning: %s\n"), 137 rmm_strerror(&error, rmm_error)); 138 rmm_dbus_error_free(&error); 139 if ((rmm_error == RMM_EDBUS_CONNECT) || 140 (rmm_error == RMM_EHAL_CONNECT)) { 141 return (99); 142 } else { 143 return (1); 144 } 145 } 146 147 if (d_opt) { 148 /* -d: print default name and exit */ 149 if ((d = rmm_hal_volume_find_default(hal_ctx, &error, 150 &default_name, &volumes)) == NULL) { 151 default_name = "nothing inserted"; 152 } else { 153 rmm_volumes_free(volumes); 154 libhal_drive_free(d); 155 } 156 (void) printf(gettext("Default device is: %s\n"), default_name); 157 } else if (l_opt) { 158 /* -l: list volumes and exit */ 159 print_mask = RMM_PRINT_MOUNTABLE; 160 if (eject_opt) { 161 print_mask |= RMM_PRINT_EJECTABLE; 162 } 163 rmm_print_volume_nicknames(hal_ctx, &error, print_mask); 164 } else if (optind == argc) { 165 /* no name provided, use default */ 166 if ((d = rmm_hal_volume_find_default(hal_ctx, &error, 167 &default_name, &volumes)) == NULL) { 168 (void) fprintf(stderr, 169 gettext("No default media available\n")); 170 ret = 1; 171 } else { 172 rmm_volumes_free(volumes); 173 libhal_drive_free(d); 174 175 if (query_opt) { 176 ret = rmm_rescan(hal_ctx, default_name, 177 B_TRUE) ? 0 : 1; 178 } else { 179 ret = rmm_action(hal_ctx, default_name, action, 180 0, 0, 0, 0) ? 0 : 1; 181 } 182 } 183 } else { 184 if (argc - optind > 1) { 185 mountpoint = argv[optind + 1]; 186 } 187 if (query_opt) { 188 ret = rmm_rescan(hal_ctx, argv[optind], 189 B_TRUE) ? 0 : 1; 190 } else { 191 ret = rmm_action(hal_ctx, argv[optind], action, 192 0, opts, num_opts, mountpoint) ? 0 : 1; 193 } 194 } 195 196 rmm_dbus_error_free(&error); 197 rmm_hal_fini(hal_ctx); 198 199 return (ret); 200 } 201 202 static int 203 rmumount(int argc, char **argv) 204 { 205 return (rmmount(argc, argv)); 206 } 207 208 static int 209 eject(int argc, char **argv) 210 { 211 if (getenv("EJECT_CLOSETRAY") != NULL) { 212 closetray_opt = B_TRUE; 213 } else if (getenv("EJECT_QUERY") != NULL) { 214 query_opt = B_TRUE; 215 } else { 216 eject_opt = B_TRUE; 217 } 218 return (rmmount(argc, argv)); 219 } 220 221 static void 222 nomem(void) 223 { 224 (void) fprintf(stderr, gettext("%s: Out of memory\n"), progname); 225 exit(1); 226 } 227 228 229 /* 230 * get the name by which this program was called 231 */ 232 static char * 233 get_progname(char *path) 234 { 235 char *s; 236 char *p; 237 238 if ((s = strdup(path)) == NULL) { 239 perror(path); 240 exit(1); 241 } 242 243 p = strrchr(s, '/'); 244 if (p != NULL) { 245 strcpy(s, p + 1); 246 } 247 248 return (s); 249 } 250 251 int 252 main(int argc, char **argv) 253 { 254 int ret = 1; 255 256 vold_init(argc, argv); 257 258 progname = get_progname(argv[0]); 259 260 if (strcmp(progname, "rmmount") == 0) { 261 if ((getenv("VOLUME_ACTION") != NULL) && 262 (getenv("VOLUME_PATH") != NULL)) { 263 ret = vold_rmmount(argc, argv); 264 } else { 265 ret = rmmount(argc, argv); 266 } 267 } else if (strcmp(progname, "rmumount") == 0) { 268 ret = rmumount(argc, argv); 269 } else if (strcmp(progname, "eject") == 0) { 270 ret = eject(argc, argv); 271 } else { 272 (void) fprintf(stderr, "rmmount: invalid program name\n"); 273 ret = 1; 274 } 275 276 return (ret); 277 } 278