Version 1.10.0.
[libguestfs.git] / fish / inspect.c
1 /* libguestfs - guestfish and guestmount shared option parsing
2  * Copyright (C) 2010 Red Hat Inc.
3  *
4  * This program is free software; you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation; either version 2 of the License, or
7  * (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software
16  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
17  */
18
19 #include <config.h>
20
21 #include <stdio.h>
22 #include <stdlib.h>
23 #include <string.h>
24
25 #include "c-ctype.h"
26
27 #include "guestfs.h"
28
29 #include "options.h"
30
31 /* Global that saves the root device between inspect_mount and
32  * print_inspect_prompt.
33  */
34 static char *root = NULL;
35
36 static void
37 free_strings (char **argv)
38 {
39   int argc;
40
41   for (argc = 0; argv[argc] != NULL; ++argc)
42     free (argv[argc]);
43   free (argv);
44 }
45
46 static int
47 count_strings (char *const *argv)
48 {
49   int c;
50
51   for (c = 0; argv[c]; ++c)
52     ;
53   return c;
54 }
55
56 static int
57 compare_keys_len (const void *p1, const void *p2)
58 {
59   const char *key1 = * (char * const *) p1;
60   const char *key2 = * (char * const *) p2;
61   return strlen (key1) - strlen (key2);
62 }
63
64 static int
65 compare_keys (const void *p1, const void *p2)
66 {
67   const char *key1 = * (char * const *) p1;
68   const char *key2 = * (char * const *) p2;
69   return strcasecmp (key1, key2);
70 }
71
72 /* This function implements the -i option. */
73 void
74 inspect_mount (void)
75 {
76   if (live) {
77     fprintf (stderr, _("%s: don't use --live and -i options together\n"),
78              program_name);
79     exit (EXIT_FAILURE);
80   }
81
82   inspect_do_decrypt ();
83
84   char **roots = guestfs_inspect_os (g);
85   if (roots == NULL)
86     exit (EXIT_FAILURE);
87
88   if (roots[0] == NULL) {
89     fprintf (stderr, _("%s: no operating system was found on this disk\n"),
90              program_name);
91     exit (EXIT_FAILURE);
92   }
93
94   if (roots[1] != NULL) {
95     fprintf (stderr, _("%s: multi-boot operating systems are not supported by the -i option\n"),
96              program_name);
97     exit (EXIT_FAILURE);
98   }
99
100   root = roots[0];
101   free (roots);
102
103   inspect_mount_root (root);
104 }
105
106 void
107 inspect_mount_root (const char *root)
108 {
109   char **mountpoints = guestfs_inspect_get_mountpoints (g, root);
110   if (mountpoints == NULL)
111     exit (EXIT_FAILURE);
112
113   /* Sort by key length, shortest key first, so that we end up
114    * mounting the filesystems in the correct order.
115    */
116   qsort (mountpoints, count_strings (mountpoints) / 2, 2 * sizeof (char *),
117          compare_keys_len);
118
119   size_t i;
120   size_t mount_errors = 0;
121   for (i = 0; mountpoints[i] != NULL; i += 2) {
122     int r;
123     if (!read_only)
124       r = guestfs_mount_options (g, "", mountpoints[i+1], mountpoints[i]);
125     else
126       r = guestfs_mount_ro (g, mountpoints[i+1], mountpoints[i]);
127     if (r == -1)
128       mount_errors++;
129   }
130
131   free_strings (mountpoints);
132
133   if (mount_errors)
134     fprintf (stderr, _("%s: some filesystems could not be mounted (ignored)\n"),
135              program_name);
136 }
137
138 /* This function is called only if the above function was called,
139  * and only after we've printed the prompt in interactive mode.
140  */
141 void
142 print_inspect_prompt (void)
143 {
144   char *name = guestfs_inspect_get_product_name (g, root);
145   if (STRNEQ (name, "unknown"))
146     printf (_("Operating system: %s\n"), name);
147   free (name);
148
149   char **mountpoints = guestfs_inspect_get_mountpoints (g, root);
150   if (mountpoints == NULL)
151     return;
152
153   /* Sort by key. */
154   qsort (mountpoints, count_strings (mountpoints) / 2, 2 * sizeof (char *),
155          compare_keys);
156
157   size_t i;
158   for (i = 0; mountpoints[i] != NULL; i += 2)
159     printf (_("%s mounted on %s\n"), mountpoints[i+1], mountpoints[i]);
160
161   free_strings (mountpoints);
162 }
163
164 /* Make a LUKS map name from the partition name,
165  * eg "/dev/vda2" => "luksvda2"
166  */
167 static void
168 make_mapname (const char *device, char *mapname, size_t len)
169 {
170   size_t i = 0;
171
172   if (len < 5)
173     abort ();
174   strcpy (mapname, "luks");
175   mapname += 4;
176   len -= 4;
177
178   if (STRPREFIX (device, "/dev/"))
179     i = 5;
180
181   for (; device[i] != '\0' && len >= 1; ++i) {
182     if (c_isalnum (device[i])) {
183       *mapname++ = device[i];
184       len--;
185     }
186   }
187
188   *mapname = '\0';
189 }
190
191 /* Simple implementation of decryption: look for any crypto_LUKS
192  * partitions and decrypt them, then rescan for VGs.  This only works
193  * for Fedora whole-disk encryption.  WIP to make this work for other
194  * encryption schemes.
195  */
196 void
197 inspect_do_decrypt (void)
198 {
199   char **partitions = guestfs_list_partitions (g);
200   if (partitions == NULL)
201     exit (EXIT_FAILURE);
202
203   int need_rescan = 0;
204   size_t i;
205   for (i = 0; partitions[i] != NULL; ++i) {
206     char *type = guestfs_vfs_type (g, partitions[i]);
207     if (type && STREQ (type, "crypto_LUKS")) {
208       char mapname[32];
209       make_mapname (partitions[i], mapname, sizeof mapname);
210
211       char *key = read_key (partitions[i]);
212       /* XXX Should we call guestfs_luks_open_ro if readonly flag
213        * is set?  This might break 'mount_ro'.
214        */
215       if (guestfs_luks_open (g, partitions[i], key, mapname) == -1)
216         exit (EXIT_FAILURE);
217
218       free (key);
219
220       need_rescan = 1;
221     }
222     free (type);
223   }
224
225   free_strings (partitions);
226
227   if (need_rescan) {
228     if (guestfs_vgscan (g) == -1)
229       exit (EXIT_FAILURE);
230     if (guestfs_vg_activate_all (g, 1) == -1)
231       exit (EXIT_FAILURE);
232   }
233 }