bd6d1895a9bdbabb50c379af6ce38007b47ac94a
[libguestfs.git] / src / inspect.c
1 /* libguestfs
2  * Copyright (C) 2010 Red Hat Inc.
3  *
4  * This library is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2 of the License, or (at your option) any later version.
8  *
9  * This library 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 GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with this library; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17  */
18
19 #include <config.h>
20
21 #include <stdio.h>
22 #include <stdlib.h>
23 #include <stdint.h>
24 #include <inttypes.h>
25 #include <unistd.h>
26 #include <string.h>
27 #include <sys/stat.h>
28
29 #ifdef HAVE_PCRE
30 #include <pcre.h>
31 #endif
32
33 #ifdef HAVE_HIVEX
34 #include <hivex.h>
35 #endif
36
37 #include "c-ctype.h"
38 #include "ignore-value.h"
39 #include "xstrtol.h"
40
41 #include "guestfs.h"
42 #include "guestfs-internal.h"
43 #include "guestfs-internal-actions.h"
44 #include "guestfs_protocol.h"
45
46 #if defined(HAVE_PCRE) && defined(HAVE_HIVEX)
47
48 /* Compile all the regular expressions once when the shared library is
49  * loaded.  PCRE is thread safe so we're supposedly OK here if
50  * multiple threads call into the libguestfs API functions below
51  * simultaneously.
52  */
53 static pcre *re_fedora;
54 static pcre *re_rhel_old;
55 static pcre *re_rhel;
56 static pcre *re_rhel_no_minor;
57 static pcre *re_major_minor;
58 static pcre *re_aug_seq;
59 static pcre *re_xdev;
60 static pcre *re_first_partition;
61 static pcre *re_freebsd;
62 static pcre *re_windows_version;
63
64 static void compile_regexps (void) __attribute__((constructor));
65 static void free_regexps (void) __attribute__((destructor));
66
67 static void
68 compile_regexps (void)
69 {
70   const char *err;
71   int offset;
72
73 #define COMPILE(re,pattern,options)                                     \
74   do {                                                                  \
75     re = pcre_compile ((pattern), (options), &err, &offset, NULL);      \
76     if (re == NULL) {                                                   \
77       ignore_value (write (2, err, strlen (err)));                      \
78       abort ();                                                         \
79     }                                                                   \
80   } while (0)
81
82   COMPILE (re_fedora, "Fedora release (\\d+)", 0);
83   COMPILE (re_rhel_old,
84            "(?:Red Hat Enterprise Linux|CentOS|Scientific Linux).*release (\\d+).*Update (\\d+)", 0);
85   COMPILE (re_rhel,
86            "(?:Red Hat Enterprise Linux|CentOS|Scientific Linux).*release (\\d+)\\.(\\d+)", 0);
87   COMPILE (re_rhel_no_minor,
88            "(?:Red Hat Enterprise Linux|CentOS|Scientific Linux).*release (\\d+)", 0);
89   COMPILE (re_major_minor, "(\\d+)\\.(\\d+)", 0);
90   COMPILE (re_aug_seq, "/\\d+$", 0);
91   COMPILE (re_xdev, "^/dev/(?:h|s|v|xv)d([a-z]\\d*)$", 0);
92   COMPILE (re_first_partition, "^/dev/(?:h|s|v)d.1$", 0);
93   COMPILE (re_freebsd, "^/dev/ad(\\d+)s(\\d+)([a-z])$", 0);
94   COMPILE (re_windows_version, "^(\\d+)\\.(\\d+)", 0);
95 }
96
97 static void
98 free_regexps (void)
99 {
100   pcre_free (re_fedora);
101   pcre_free (re_rhel_old);
102   pcre_free (re_rhel);
103   pcre_free (re_rhel_no_minor);
104   pcre_free (re_major_minor);
105   pcre_free (re_aug_seq);
106   pcre_free (re_xdev);
107   pcre_free (re_first_partition);
108   pcre_free (re_freebsd);
109   pcre_free (re_windows_version);
110 }
111
112 /* The main inspection code. */
113 static int check_for_filesystem_on (guestfs_h *g, const char *device);
114
115 char **
116 guestfs__inspect_os (guestfs_h *g)
117 {
118   /* Remove any information previously stored in the handle. */
119   guestfs___free_inspect_info (g);
120
121   if (guestfs_umount_all (g) == -1)
122     return NULL;
123
124   /* Iterate over all possible devices.  Try to mount each
125    * (read-only).  Examine ones which contain filesystems and add that
126    * information to the handle.
127    */
128   /* Look to see if any devices directly contain filesystems (RHBZ#590167). */
129   char **devices;
130   devices = guestfs_list_devices (g);
131   if (devices == NULL)
132     return NULL;
133
134   size_t i;
135   for (i = 0; devices[i] != NULL; ++i) {
136     if (check_for_filesystem_on (g, devices[i]) == -1) {
137       guestfs___free_string_list (devices);
138       guestfs___free_inspect_info (g);
139       return NULL;
140     }
141   }
142   guestfs___free_string_list (devices);
143
144   /* Look at all partitions. */
145   char **partitions;
146   partitions = guestfs_list_partitions (g);
147   if (partitions == NULL) {
148     guestfs___free_inspect_info (g);
149     return NULL;
150   }
151
152   for (i = 0; partitions[i] != NULL; ++i) {
153     if (check_for_filesystem_on (g, partitions[i]) == -1) {
154       guestfs___free_string_list (partitions);
155       guestfs___free_inspect_info (g);
156       return NULL;
157     }
158   }
159   guestfs___free_string_list (partitions);
160
161   /* Look at all LVs. */
162   if (guestfs___feature_available (g, "lvm2")) {
163     char **lvs;
164     lvs = guestfs_lvs (g);
165     if (lvs == NULL) {
166       guestfs___free_inspect_info (g);
167       return NULL;
168     }
169
170     for (i = 0; lvs[i] != NULL; ++i) {
171       if (check_for_filesystem_on (g, lvs[i]) == -1) {
172         guestfs___free_string_list (lvs);
173         guestfs___free_inspect_info (g);
174         return NULL;
175       }
176     }
177     guestfs___free_string_list (lvs);
178   }
179
180   /* At this point we have, in the handle, a list of all filesystems
181    * found and data about each one.  Now we assemble the list of
182    * filesystems which are root devices and return that to the user.
183    * Fall through to guestfs__inspect_get_roots to do that.
184    */
185   char **ret = guestfs__inspect_get_roots (g);
186   if (ret == NULL)
187     guestfs___free_inspect_info (g);
188   return ret;
189 }
190
191 /* Find out if 'device' contains a filesystem.  If it does, add
192  * another entry in g->fses.
193  */
194 static int check_filesystem (guestfs_h *g, const char *device);
195 static int check_linux_root (guestfs_h *g, struct inspect_fs *fs);
196 static int check_freebsd_root (guestfs_h *g, struct inspect_fs *fs);
197 static void check_architecture (guestfs_h *g, struct inspect_fs *fs);
198 static int check_fstab (guestfs_h *g, struct inspect_fs *fs);
199 static int check_windows_root (guestfs_h *g, struct inspect_fs *fs);
200 static int check_windows_arch (guestfs_h *g, struct inspect_fs *fs);
201 static int check_windows_registry (guestfs_h *g, struct inspect_fs *fs);
202 static char *resolve_windows_path_silently (guestfs_h *g, const char *);
203 static int extend_fses (guestfs_h *g);
204 static int parse_unsigned_int (guestfs_h *g, const char *str);
205 static int add_fstab_entry (guestfs_h *g, struct inspect_fs *fs,
206                             const char *spec, const char *mp);
207 static char *resolve_fstab_device (guestfs_h *g, const char *spec);
208 static void check_package_format (guestfs_h *g, struct inspect_fs *fs);
209 static void check_package_management (guestfs_h *g, struct inspect_fs *fs);
210
211 static int
212 check_for_filesystem_on (guestfs_h *g, const char *device)
213 {
214   /* Get vfs-type in order to check if it's a Linux(?) swap device.
215    * If there's an error we should ignore it, so to do that we have to
216    * temporarily replace the error handler with a null one.
217    */
218   guestfs_error_handler_cb old_error_cb = g->error_cb;
219   g->error_cb = NULL;
220   char *vfs_type = guestfs_vfs_type (g, device);
221   g->error_cb = old_error_cb;
222
223   int is_swap = vfs_type && STREQ (vfs_type, "swap");
224
225   if (g->verbose)
226     fprintf (stderr, "check_for_filesystem_on: %s (%s)\n",
227              device, vfs_type ? vfs_type : "failed to get vfs type");
228
229   if (is_swap) {
230     free (vfs_type);
231     if (extend_fses (g) == -1)
232       return -1;
233     g->fses[g->nr_fses-1].is_swap = 1;
234     return 0;
235   }
236
237   /* Try mounting the device.  As above, ignore errors. */
238   g->error_cb = NULL;
239   int r = guestfs_mount_ro (g, device, "/");
240   if (r == -1 && vfs_type && STREQ (vfs_type, "ufs")) /* Hack for the *BSDs. */
241     r = guestfs_mount_vfs (g, "ro,ufstype=ufs2", "ufs", device, "/");
242   free (vfs_type);
243   g->error_cb = old_error_cb;
244   if (r == -1)
245     return 0;
246
247   /* Do the rest of the checks. */
248   r = check_filesystem (g, device);
249
250   /* Unmount the filesystem. */
251   if (guestfs_umount_all (g) == -1)
252     return -1;
253
254   return r;
255 }
256
257 static int
258 check_filesystem (guestfs_h *g, const char *device)
259 {
260   if (extend_fses (g) == -1)
261     return -1;
262
263   struct inspect_fs *fs = &g->fses[g->nr_fses-1];
264
265   fs->device = safe_strdup (g, device);
266   fs->is_mountable = 1;
267
268   /* Optimize some of the tests by avoiding multiple tests of the same thing. */
269   int is_dir_etc = guestfs_is_dir (g, "/etc") > 0;
270   int is_dir_bin = guestfs_is_dir (g, "/bin") > 0;
271   int is_dir_share = guestfs_is_dir (g, "/share") > 0;
272
273   /* Grub /boot? */
274   if (guestfs_is_file (g, "/grub/menu.lst") > 0 ||
275       guestfs_is_file (g, "/grub/grub.conf") > 0)
276     fs->content = FS_CONTENT_LINUX_BOOT;
277   /* FreeBSD root? */
278   else if (is_dir_etc &&
279            is_dir_bin &&
280            guestfs_is_file (g, "/etc/freebsd-update.conf") > 0 &&
281            guestfs_is_file (g, "/etc/fstab") > 0) {
282     /* Ignore /dev/sda1 which is a shadow of the real root filesystem
283      * that is probably /dev/sda5 (see:
284      * http://www.freebsd.org/doc/handbook/disk-organization.html)
285      */
286     if (match (g, device, re_first_partition))
287       return 0;
288
289     fs->is_root = 1;
290     fs->content = FS_CONTENT_FREEBSD_ROOT;
291     if (check_freebsd_root (g, fs) == -1)
292       return -1;
293   }
294   /* Linux root? */
295   else if (is_dir_etc &&
296            is_dir_bin &&
297            guestfs_is_file (g, "/etc/fstab") > 0) {
298     fs->is_root = 1;
299     fs->content = FS_CONTENT_LINUX_ROOT;
300     if (check_linux_root (g, fs) == -1)
301       return -1;
302   }
303   /* Linux /usr/local? */
304   else if (is_dir_etc &&
305            is_dir_bin &&
306            is_dir_share &&
307            guestfs_exists (g, "/local") == 0 &&
308            guestfs_is_file (g, "/etc/fstab") == 0)
309     fs->content = FS_CONTENT_LINUX_USR_LOCAL;
310   /* Linux /usr? */
311   else if (is_dir_etc &&
312            is_dir_bin &&
313            is_dir_share &&
314            guestfs_exists (g, "/local") > 0 &&
315            guestfs_is_file (g, "/etc/fstab") == 0)
316     fs->content = FS_CONTENT_LINUX_USR;
317   /* Linux /var? */
318   else if (guestfs_is_dir (g, "/log") > 0 &&
319            guestfs_is_dir (g, "/run") > 0 &&
320            guestfs_is_dir (g, "/spool") > 0)
321     fs->content = FS_CONTENT_LINUX_VAR;
322   /* Windows root? */
323   else if (guestfs_is_file (g, "/AUTOEXEC.BAT") > 0 ||
324            guestfs_is_file (g, "/autoexec.bat") > 0 ||
325            guestfs_is_dir (g, "/Program Files") > 0 ||
326            guestfs_is_dir (g, "/WINDOWS") > 0 ||
327            guestfs_is_dir (g, "/Windows") > 0 ||
328            guestfs_is_dir (g, "/windows") > 0 ||
329            guestfs_is_dir (g, "/WIN32") > 0 ||
330            guestfs_is_dir (g, "/Win32") > 0 ||
331            guestfs_is_dir (g, "/WINNT") > 0 ||
332            guestfs_is_file (g, "/boot.ini") > 0 ||
333            guestfs_is_file (g, "/ntldr") > 0) {
334     fs->is_root = 1;
335     fs->content = FS_CONTENT_WINDOWS_ROOT;
336     if (check_windows_root (g, fs) == -1)
337       return -1;
338   }
339
340   return 0;
341 }
342
343 /* Set fs->product_name to the first line of the release file. */
344 static int
345 parse_release_file (guestfs_h *g, struct inspect_fs *fs,
346                     const char *release_filename)
347 {
348   char **product_name = guestfs_head_n (g, 1, release_filename);
349   if (product_name == NULL)
350     return -1;
351   if (product_name[0] == NULL) {
352     error (g, _("%s: file is empty"), release_filename);
353     guestfs___free_string_list (product_name);
354     return -1;
355   }
356
357   /* Note that this string becomes owned by the handle and will
358    * be freed by guestfs___free_inspect_info.
359    */
360   fs->product_name = product_name[0];
361   free (product_name);
362
363   return 0;
364 }
365
366 /* Parse generic MAJOR.MINOR from the fs->product_name string. */
367 static int
368 parse_major_minor (guestfs_h *g, struct inspect_fs *fs)
369 {
370   char *major, *minor;
371
372   if (match2 (g, fs->product_name, re_major_minor, &major, &minor)) {
373     fs->major_version = parse_unsigned_int (g, major);
374     free (major);
375     if (fs->major_version == -1) {
376       free (minor);
377       return -1;
378     }
379     fs->minor_version = parse_unsigned_int (g, minor);
380     free (minor);
381     if (fs->minor_version == -1)
382       return -1;
383   }
384   return 0;
385 }
386
387 /* Ubuntu has /etc/lsb-release containing:
388  *   DISTRIB_ID=Ubuntu                                # Distro
389  *   DISTRIB_RELEASE=10.04                            # Version
390  *   DISTRIB_CODENAME=lucid
391  *   DISTRIB_DESCRIPTION="Ubuntu 10.04.1 LTS"         # Product name
392  * In theory other distros could have this LSB file, but none do.
393  */
394 static int
395 parse_lsb_release (guestfs_h *g, struct inspect_fs *fs)
396 {
397   char **lines;
398   size_t i;
399   int r = 0;
400
401   lines = guestfs_head_n (g, 10, "/etc/lsb-release");
402   if (lines == NULL)
403     return -1;
404
405   for (i = 0; lines[i] != NULL; ++i) {
406     if (fs->distro == 0 &&
407         STREQ (lines[i], "DISTRIB_ID=Ubuntu")) {
408       fs->distro = OS_DISTRO_UBUNTU;
409       r = 1;
410     }
411     else if (STRPREFIX (lines[i], "DISTRIB_RELEASE=")) {
412       char *major, *minor;
413       if (match2 (g, &lines[i][16], re_major_minor, &major, &minor)) {
414         fs->major_version = parse_unsigned_int (g, major);
415         free (major);
416         if (fs->major_version == -1) {
417           free (minor);
418           guestfs___free_string_list (lines);
419           return -1;
420         }
421         fs->minor_version = parse_unsigned_int (g, minor);
422         free (minor);
423         if (fs->minor_version == -1) {
424           guestfs___free_string_list (lines);
425           return -1;
426         }
427       }
428     }
429     else if (fs->product_name == NULL &&
430              (STRPREFIX (lines[i], "DISTRIB_DESCRIPTION=\"") ||
431               STRPREFIX (lines[i], "DISTRIB_DESCRIPTION='"))) {
432       size_t len = strlen (lines[i]) - 21 - 1;
433       fs->product_name = safe_strndup (g, &lines[i][21], len);
434       r = 1;
435     }
436     else if (fs->product_name == NULL &&
437              STRPREFIX (lines[i], "DISTRIB_DESCRIPTION=")) {
438       size_t len = strlen (lines[i]) - 20;
439       fs->product_name = safe_strndup (g, &lines[i][20], len);
440       r = 1;
441     }
442   }
443
444   guestfs___free_string_list (lines);
445   return r;
446 }
447
448 /* The currently mounted device is known to be a Linux root.  Try to
449  * determine from this the distro, version, etc.  Also parse
450  * /etc/fstab to determine the arrangement of mountpoints and
451  * associated devices.
452  */
453 static int
454 check_linux_root (guestfs_h *g, struct inspect_fs *fs)
455 {
456   int r;
457
458   fs->type = OS_TYPE_LINUX;
459
460   if (guestfs_exists (g, "/etc/lsb-release") > 0) {
461     r = parse_lsb_release (g, fs);
462     if (r == -1)        /* error */
463       return -1;
464     if (r == 1)         /* ok - detected the release from this file */
465       goto skip_release_checks;
466   }
467
468   if (guestfs_exists (g, "/etc/redhat-release") > 0) {
469     fs->distro = OS_DISTRO_REDHAT_BASED; /* Something generic Red Hat-like. */
470
471     if (parse_release_file (g, fs, "/etc/redhat-release") == -1)
472       return -1;
473
474     char *major, *minor;
475     if ((major = match1 (g, fs->product_name, re_fedora)) != NULL) {
476       fs->distro = OS_DISTRO_FEDORA;
477       fs->major_version = parse_unsigned_int (g, major);
478       free (major);
479       if (fs->major_version == -1)
480         return -1;
481     }
482     else if (match2 (g, fs->product_name, re_rhel_old, &major, &minor) ||
483              match2 (g, fs->product_name, re_rhel, &major, &minor)) {
484       fs->distro = OS_DISTRO_RHEL;
485       fs->major_version = parse_unsigned_int (g, major);
486       free (major);
487       if (fs->major_version == -1) {
488         free (minor);
489         return -1;
490       }
491       fs->minor_version = parse_unsigned_int (g, minor);
492       free (minor);
493       if (fs->minor_version == -1)
494         return -1;
495     }
496     else if ((major = match1 (g, fs->product_name, re_rhel_no_minor)) != NULL) {
497       fs->distro = OS_DISTRO_RHEL;
498       fs->major_version = parse_unsigned_int (g, major);
499       free (major);
500       if (fs->major_version == -1)
501         return -1;
502       fs->minor_version = 0;
503     }
504   }
505   else if (guestfs_exists (g, "/etc/debian_version") > 0) {
506     fs->distro = OS_DISTRO_DEBIAN;
507
508     if (parse_release_file (g, fs, "/etc/debian_version") == -1)
509       return -1;
510
511     if (parse_major_minor (g, fs) == -1)
512       return -1;
513   }
514   else if (guestfs_exists (g, "/etc/pardus-release") > 0) {
515     fs->distro = OS_DISTRO_PARDUS;
516
517     if (parse_release_file (g, fs, "/etc/pardus-release") == -1)
518       return -1;
519
520     if (parse_major_minor (g, fs) == -1)
521       return -1;
522   }
523   else if (guestfs_exists (g, "/etc/arch-release") > 0) {
524     fs->distro = OS_DISTRO_ARCHLINUX;
525
526     /* /etc/arch-release file is empty and I can't see a way to
527      * determine the actual release or product string.
528      */
529   }
530   else if (guestfs_exists (g, "/etc/gentoo-release") > 0) {
531     fs->distro = OS_DISTRO_GENTOO;
532
533     if (parse_release_file (g, fs, "/etc/gentoo-release") == -1)
534       return -1;
535
536     if (parse_major_minor (g, fs) == -1)
537       return -1;
538   }
539   else if (guestfs_exists (g, "/etc/meego-release") > 0) {
540     fs->distro = OS_DISTRO_MEEGO;
541
542     if (parse_release_file (g, fs, "/etc/meego-release") == -1)
543       return -1;
544
545     if (parse_major_minor (g, fs) == -1)
546       return -1;
547   }
548
549  skip_release_checks:;
550
551   /* If distro test above was successful, work out the package format. */
552   check_package_format (g, fs);
553   check_package_management (g, fs);
554
555   /* Determine the architecture. */
556   check_architecture (g, fs);
557
558   /* We already know /etc/fstab exists because it's part of the test
559    * for Linux root above.  We must now parse this file to determine
560    * which filesystems are used by the operating system and how they
561    * are mounted.
562    */
563   if (check_fstab (g, fs) == -1)
564     return -1;
565
566   return 0;
567 }
568
569 /* The currently mounted device is known to be a FreeBSD root. */
570 static int
571 check_freebsd_root (guestfs_h *g, struct inspect_fs *fs)
572 {
573   int r;
574
575   fs->type = OS_TYPE_FREEBSD;
576
577   /* FreeBSD has no authoritative version file.  The version number is
578    * in /etc/motd, which the system administrator might edit, but
579    * we'll use that anyway.
580    */
581
582   if (guestfs_exists (g, "/etc/motd") > 0) {
583     if (parse_release_file (g, fs, "/etc/motd") == -1)
584       return -1;
585
586     if (parse_major_minor (g, fs) == -1)
587       return -1;
588   }
589
590   /* Determine the architecture. */
591   check_architecture (g, fs);
592
593   /* We already know /etc/fstab exists because it's part of the test above. */
594   if (check_fstab (g, fs) == -1)
595     return -1;
596
597   return 0;
598 }
599
600 static void
601 check_architecture (guestfs_h *g, struct inspect_fs *fs)
602 {
603   const char *binaries[] =
604     { "/bin/bash", "/bin/ls", "/bin/echo", "/bin/rm", "/bin/sh" };
605   size_t i;
606
607   for (i = 0; i < sizeof binaries / sizeof binaries[0]; ++i) {
608     if (guestfs_is_file (g, binaries[i]) > 0) {
609       /* Ignore errors from file_architecture call. */
610       guestfs_error_handler_cb old_error_cb = g->error_cb;
611       g->error_cb = NULL;
612       char *arch = guestfs_file_architecture (g, binaries[i]);
613       g->error_cb = old_error_cb;
614
615       if (arch) {
616         /* String will be owned by handle, freed by
617          * guestfs___free_inspect_info.
618          */
619         fs->arch = arch;
620         break;
621       }
622     }
623   }
624 }
625
626 static int check_fstab_aug_open (guestfs_h *g, struct inspect_fs *fs);
627
628 static int
629 check_fstab (guestfs_h *g, struct inspect_fs *fs)
630 {
631   int r;
632   int64_t size;
633
634   /* Security: Refuse to do this if /etc/fstab is huge. */
635   size = guestfs_filesize (g, "/etc/fstab");
636   if (size == -1 || size > 100000) {
637     error (g, _("size of /etc/fstab unreasonable (%" PRIi64 " bytes)"), size);
638     return -1;
639   }
640
641   /* XXX What if !feature_available (g, "augeas")? */
642   if (guestfs_aug_init (g, "/", 16|32) == -1)
643     return -1;
644
645   /* Tell Augeas to only load /etc/fstab (thanks Raphaël Pinson). */
646   guestfs_aug_rm (g, "/augeas/load//incl[. != \"/etc/fstab\"]");
647   guestfs_aug_load (g);
648
649   r = check_fstab_aug_open (g, fs);
650   guestfs_aug_close (g);
651   if (r == -1)
652     return -1;
653
654   return 0;
655 }
656
657 static int
658 check_fstab_aug_open (guestfs_h *g, struct inspect_fs *fs)
659 {
660   char **lines = guestfs_aug_ls (g, "/files/etc/fstab");
661   if (lines == NULL)
662     return -1;
663
664   if (lines[0] == NULL) {
665     error (g, _("could not parse /etc/fstab or empty file"));
666     guestfs___free_string_list (lines);
667     return -1;
668   }
669
670   size_t i;
671   char augpath[256];
672   for (i = 0; lines[i] != NULL; ++i) {
673     /* Ignore comments.  Only care about sequence lines which
674      * match m{/\d+$}.
675      */
676     if (match (g, lines[i], re_aug_seq)) {
677       snprintf (augpath, sizeof augpath, "%s/spec", lines[i]);
678       char *spec = guestfs_aug_get (g, augpath);
679       if (spec == NULL) {
680         guestfs___free_string_list (lines);
681         return -1;
682       }
683
684       snprintf (augpath, sizeof augpath, "%s/file", lines[i]);
685       char *mp = guestfs_aug_get (g, augpath);
686       if (mp == NULL) {
687         guestfs___free_string_list (lines);
688         free (spec);
689         return -1;
690       }
691
692       int r = add_fstab_entry (g, fs, spec, mp);
693       free (spec);
694       free (mp);
695
696       if (r == -1) {
697         guestfs___free_string_list (lines);
698         return -1;
699       }
700     }
701   }
702
703   guestfs___free_string_list (lines);
704   return 0;
705 }
706
707 /* Add a filesystem and possibly a mountpoint entry for
708  * the root filesystem 'fs'.
709  *
710  * 'spec' is the fstab spec field, which might be a device name or a
711  * pseudodevice or 'UUID=...' or 'LABEL=...'.
712  *
713  * 'mp' is the mount point, which could also be 'swap' or 'none'.
714  */
715 static int
716 add_fstab_entry (guestfs_h *g, struct inspect_fs *fs,
717                  const char *spec, const char *mp)
718 {
719   /* Ignore certain mountpoints. */
720   if (STRPREFIX (mp, "/dev/") ||
721       STREQ (mp, "/dev") ||
722       STRPREFIX (mp, "/media/") ||
723       STRPREFIX (mp, "/proc/") ||
724       STREQ (mp, "/proc") ||
725       STRPREFIX (mp, "/selinux/") ||
726       STREQ (mp, "/selinux") ||
727       STRPREFIX (mp, "/sys/") ||
728       STREQ (mp, "/sys"))
729     return 0;
730
731   /* Ignore /dev/fd (floppy disks) (RHBZ#642929) and CD-ROM drives. */
732   if ((STRPREFIX (spec, "/dev/fd") && c_isdigit (spec[7])) ||
733       STREQ (spec, "/dev/floppy") ||
734       STREQ (spec, "/dev/cdrom"))
735     return 0;
736
737   /* Resolve UUID= and LABEL= to the actual device. */
738   char *device = NULL;
739   if (STRPREFIX (spec, "UUID="))
740     device = guestfs_findfs_uuid (g, &spec[5]);
741   else if (STRPREFIX (spec, "LABEL="))
742     device = guestfs_findfs_label (g, &spec[6]);
743   /* Ignore "/.swap" (Pardus) and pseudo-devices like "tmpfs". */
744   else if (STRPREFIX (spec, "/dev/"))
745     /* Resolve guest block device names. */
746     device = resolve_fstab_device (g, spec);
747
748   /* If we haven't resolved the device successfully by this point,
749    * we don't care, just ignore it.
750    */
751   if (device == NULL)
752     return 0;
753
754   char *mountpoint = safe_strdup (g, mp);
755
756   /* Add this to the fstab entry in 'fs'.
757    * Note these are further filtered by guestfs_inspect_get_mountpoints
758    * and guestfs_inspect_get_filesystems.
759    */
760   size_t n = fs->nr_fstab + 1;
761   struct inspect_fstab_entry *p;
762
763   p = realloc (fs->fstab, n * sizeof (struct inspect_fstab_entry));
764   if (p == NULL) {
765     perrorf (g, "realloc");
766     free (device);
767     free (mountpoint);
768     return -1;
769   }
770
771   fs->fstab = p;
772   fs->nr_fstab = n;
773
774   /* These are owned by the handle and freed by guestfs___free_inspect_info. */
775   fs->fstab[n-1].device = device;
776   fs->fstab[n-1].mountpoint = mountpoint;
777
778   if (g->verbose)
779     fprintf (stderr, "fstab: device=%s mountpoint=%s\n", device, mountpoint);
780
781   return 0;
782 }
783
784 /* Resolve block device name to the libguestfs device name, eg.
785  * /dev/xvdb1 => /dev/vdb1; and /dev/mapper/VG-LV => /dev/VG/LV.  This
786  * assumes that disks were added in the same order as they appear to
787  * the real VM, which is a reasonable assumption to make.  Return
788  * anything we don't recognize unchanged.
789  */
790 static char *
791 resolve_fstab_device (guestfs_h *g, const char *spec)
792 {
793   char *a1;
794   char *device = NULL;
795   char *bsddisk, *bsdslice, *bsdpart;
796
797   if (STRPREFIX (spec, "/dev/mapper/")) {
798     /* LVM2 does some strange munging on /dev/mapper paths for VGs and
799      * LVs which contain '-' character:
800      *
801      * ><fs> lvcreate LV--test VG--test 32
802      * ><fs> debug ls /dev/mapper
803      * VG----test-LV----test
804      *
805      * This makes it impossible to reverse those paths directly, so
806      * we have implemented lvm_canonical_lv_name in the daemon.
807      */
808     device = guestfs_lvm_canonical_lv_name (g, spec);
809   }
810   else if ((a1 = match1 (g, spec, re_xdev)) != NULL) {
811     char **devices = guestfs_list_devices (g);
812     if (devices == NULL)
813       return NULL;
814
815     size_t count;
816     for (count = 0; devices[count] != NULL; count++)
817       ;
818
819     size_t i = a1[0] - 'a'; /* a1[0] is always [a-z] because of regex. */
820     if (i < count) {
821       size_t len = strlen (devices[i]) + strlen (a1) + 16;
822       device = safe_malloc (g, len);
823       snprintf (device, len, "%s%s", devices[i], &a1[1]);
824     }
825
826     free (a1);
827     guestfs___free_string_list (devices);
828   }
829   else if (match3 (g, spec, re_freebsd, &bsddisk, &bsdslice, &bsdpart)) {
830     /* FreeBSD disks are organized quite differently.  See:
831      * http://www.freebsd.org/doc/handbook/disk-organization.html
832      * FreeBSD "partitions" are exposed as quasi-extended partitions
833      * numbered from 5 in Linux.  I have no idea what happens when you
834      * have multiple "slices" (the FreeBSD term for MBR partitions).
835      */
836     int disk = parse_unsigned_int (g, bsddisk);
837     int slice = parse_unsigned_int (g, bsdslice);
838     int part = bsdpart[0] - 'a' /* counting from 0 */;
839     free (bsddisk);
840     free (bsdslice);
841     free (bsdpart);
842
843     if (disk == -1 || disk > 26 ||
844         slice <= 0 || slice > 1 /* > 4 .. see comment above */ ||
845         part < 0 || part >= 26)
846       goto out;
847
848     device = safe_asprintf (g, "/dev/sd%c%d", disk + 'a', part + 5);
849   }
850
851  out:
852   /* Didn't match device pattern, return original spec unchanged. */
853   if (device == NULL)
854     device = safe_strdup (g, spec);
855
856   return device;
857 }
858
859 /* XXX Handling of boot.ini in the Perl version was pretty broken.  It
860  * essentially didn't do anything for modern Windows guests.
861  * Therefore I've omitted all that code.
862  */
863 static int
864 check_windows_root (guestfs_h *g, struct inspect_fs *fs)
865 {
866   fs->type = OS_TYPE_WINDOWS;
867   fs->distro = OS_DISTRO_WINDOWS;
868
869   /* Try to find Windows systemroot using some common locations. */
870   const char *systemroots[] =
871     { "/windows", "/winnt", "/win32", "/win" };
872   size_t i;
873   char *systemroot = NULL;
874   for (i = 0;
875        systemroot == NULL && i < sizeof systemroots / sizeof systemroots[0];
876        ++i) {
877     systemroot = resolve_windows_path_silently (g, systemroots[i]);
878   }
879
880   if (!systemroot) {
881     error (g, _("cannot resolve Windows %%SYSTEMROOT%%"));
882     return -1;
883   }
884
885   if (g->verbose)
886     fprintf (stderr, "windows %%SYSTEMROOT%% = %s", systemroot);
887
888   /* Freed by guestfs___free_inspect_info. */
889   fs->windows_systemroot = systemroot;
890
891   if (check_windows_arch (g, fs) == -1)
892     return -1;
893
894   if (check_windows_registry (g, fs) == -1)
895     return -1;
896
897   check_package_format (g, fs);
898   check_package_management (g, fs);
899
900   return 0;
901 }
902
903 static int
904 check_windows_arch (guestfs_h *g, struct inspect_fs *fs)
905 {
906   size_t len = strlen (fs->windows_systemroot) + 32;
907   char cmd_exe[len];
908   snprintf (cmd_exe, len, "%s/system32/cmd.exe", fs->windows_systemroot);
909
910   char *cmd_exe_path = resolve_windows_path_silently (g, cmd_exe);
911   if (!cmd_exe_path)
912     return 0;
913
914   char *arch = guestfs_file_architecture (g, cmd_exe_path);
915   free (cmd_exe_path);
916
917   if (arch)
918     fs->arch = arch;        /* freed by guestfs___free_inspect_info */
919
920   return 0;
921 }
922
923 /* At the moment, pull just the ProductName and version numbers from
924  * the registry.  In future there is a case for making many more
925  * registry fields available to callers.
926  */
927 static int
928 check_windows_registry (guestfs_h *g, struct inspect_fs *fs)
929 {
930   TMP_TEMPLATE_ON_STACK (dir);
931 #define dir_len (strlen (dir))
932 #define software_hive_len (dir_len + 16)
933   char software_hive[software_hive_len];
934 #define cmd_len (dir_len + 16)
935   char cmd[cmd_len];
936
937   size_t len = strlen (fs->windows_systemroot) + 64;
938   char software[len];
939   snprintf (software, len, "%s/system32/config/software",
940             fs->windows_systemroot);
941
942   char *software_path = resolve_windows_path_silently (g, software);
943   if (!software_path)
944     /* If the software hive doesn't exist, just accept that we cannot
945      * find product_name etc.
946      */
947     return 0;
948
949   int ret = -1;
950   hive_h *h = NULL;
951   hive_value_h *values = NULL;
952
953   /* Security: Refuse to download registry if it is huge. */
954   int64_t size = guestfs_filesize (g, software_path);
955   if (size == -1 || size > 100000000) {
956     error (g, _("size of %s unreasonable (%" PRIi64 " bytes)"),
957            software_path, size);
958     goto out;
959   }
960
961   if (mkdtemp (dir) == NULL) {
962     perrorf (g, "mkdtemp");
963     goto out;
964   }
965
966   snprintf (software_hive, software_hive_len, "%s/software", dir);
967
968   if (guestfs_download (g, software_path, software_hive) == -1)
969     goto out;
970
971   h = hivex_open (software_hive, g->verbose ? HIVEX_OPEN_VERBOSE : 0);
972   if (h == NULL) {
973     perrorf (g, "hivex_open");
974     goto out;
975   }
976
977   hive_node_h node = hivex_root (h);
978   const char *hivepath[] =
979     { "Microsoft", "Windows NT", "CurrentVersion" };
980   size_t i;
981   for (i = 0;
982        node != 0 && i < sizeof hivepath / sizeof hivepath[0];
983        ++i) {
984     node = hivex_node_get_child (h, node, hivepath[i]);
985   }
986
987   if (node == 0) {
988     perrorf (g, "hivex: cannot locate HKLM\\SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion");
989     goto out;
990   }
991
992   values = hivex_node_values (h, node);
993
994   for (i = 0; values[i] != 0; ++i) {
995     char *key = hivex_value_key (h, values[i]);
996     if (key == NULL) {
997       perrorf (g, "hivex_value_key");
998       goto out;
999     }
1000
1001     if (STRCASEEQ (key, "ProductName")) {
1002       fs->product_name = hivex_value_string (h, values[i]);
1003       if (!fs->product_name) {
1004         perrorf (g, "hivex_value_string");
1005         free (key);
1006         goto out;
1007       }
1008     }
1009     else if (STRCASEEQ (key, "CurrentVersion")) {
1010       char *version = hivex_value_string (h, values[i]);
1011       if (!version) {
1012         perrorf (g, "hivex_value_string");
1013         free (key);
1014         goto out;
1015       }
1016       char *major, *minor;
1017       if (match2 (g, version, re_windows_version, &major, &minor)) {
1018         fs->major_version = parse_unsigned_int (g, major);
1019         free (major);
1020         if (fs->major_version == -1) {
1021           free (minor);
1022           free (key);
1023           free (version);
1024           goto out;
1025         }
1026         fs->minor_version = parse_unsigned_int (g, minor);
1027         free (minor);
1028         if (fs->minor_version == -1) {
1029           free (key);
1030           free (version);
1031           return -1;
1032         }
1033       }
1034
1035       free (version);
1036     }
1037
1038     free (key);
1039   }
1040
1041   ret = 0;
1042
1043  out:
1044   if (h) hivex_close (h);
1045   free (values);
1046   free (software_path);
1047
1048   /* Free up the temporary directory.  Note the directory name cannot
1049    * contain shell meta-characters because of the way it was
1050    * constructed above.
1051    */
1052   snprintf (cmd, cmd_len, "rm -rf %s", dir);
1053   ignore_value (system (cmd));
1054 #undef dir_len
1055 #undef software_hive_len
1056 #undef cmd_len
1057
1058   return ret;
1059 }
1060
1061 static char *
1062 resolve_windows_path_silently (guestfs_h *g, const char *path)
1063 {
1064   guestfs_error_handler_cb old_error_cb = g->error_cb;
1065   g->error_cb = NULL;
1066   char *ret = guestfs_case_sensitive_path (g, path);
1067   g->error_cb = old_error_cb;
1068   return ret;
1069 }
1070
1071 static int
1072 extend_fses (guestfs_h *g)
1073 {
1074   size_t n = g->nr_fses + 1;
1075   struct inspect_fs *p;
1076
1077   p = realloc (g->fses, n * sizeof (struct inspect_fs));
1078   if (p == NULL) {
1079     perrorf (g, "realloc");
1080     return -1;
1081   }
1082
1083   g->fses = p;
1084   g->nr_fses = n;
1085
1086   memset (&g->fses[n-1], 0, sizeof (struct inspect_fs));
1087
1088   return 0;
1089 }
1090
1091 /* Parse small, unsigned ints, as used in version numbers. */
1092 static int
1093 parse_unsigned_int (guestfs_h *g, const char *str)
1094 {
1095   long ret;
1096   int r = xstrtol (str, NULL, 10, &ret, "");
1097   if (r != LONGINT_OK) {
1098     error (g, _("could not parse integer in version number: %s"), str);
1099     return -1;
1100   }
1101   return ret;
1102 }
1103
1104 /* At the moment, package format and package management is just a
1105  * simple function of the distro and major_version fields, so these
1106  * can never return an error.  We might be cleverer in future.
1107  */
1108 static void
1109 check_package_format (guestfs_h *g, struct inspect_fs *fs)
1110 {
1111   switch (fs->distro) {
1112   case OS_DISTRO_FEDORA:
1113   case OS_DISTRO_MEEGO:
1114   case OS_DISTRO_REDHAT_BASED:
1115   case OS_DISTRO_RHEL:
1116     fs->package_format = OS_PACKAGE_FORMAT_RPM;
1117     break;
1118
1119   case OS_DISTRO_DEBIAN:
1120   case OS_DISTRO_UBUNTU:
1121     fs->package_format = OS_PACKAGE_FORMAT_DEB;
1122     break;
1123
1124   case OS_DISTRO_ARCHLINUX:
1125     fs->package_format = OS_PACKAGE_FORMAT_PACMAN;
1126     break;
1127   case OS_DISTRO_GENTOO:
1128     fs->package_format = OS_PACKAGE_FORMAT_EBUILD;
1129     break;
1130   case OS_DISTRO_PARDUS:
1131     fs->package_format = OS_PACKAGE_FORMAT_PISI;
1132     break;
1133
1134   case OS_DISTRO_WINDOWS:
1135   case OS_DISTRO_UNKNOWN:
1136   default:
1137     fs->package_format = OS_PACKAGE_FORMAT_UNKNOWN;
1138     break;
1139   }
1140 }
1141
1142 static void
1143 check_package_management (guestfs_h *g, struct inspect_fs *fs)
1144 {
1145   switch (fs->distro) {
1146   case OS_DISTRO_FEDORA:
1147   case OS_DISTRO_MEEGO:
1148     fs->package_management = OS_PACKAGE_MANAGEMENT_YUM;
1149     break;
1150
1151   case OS_DISTRO_REDHAT_BASED:
1152   case OS_DISTRO_RHEL:
1153     if (fs->major_version >= 5)
1154       fs->package_management = OS_PACKAGE_MANAGEMENT_YUM;
1155     else
1156       fs->package_management = OS_PACKAGE_MANAGEMENT_UP2DATE;
1157     break;
1158
1159   case OS_DISTRO_DEBIAN:
1160   case OS_DISTRO_UBUNTU:
1161     fs->package_management = OS_PACKAGE_MANAGEMENT_APT;
1162     break;
1163
1164   case OS_DISTRO_ARCHLINUX:
1165     fs->package_management = OS_PACKAGE_MANAGEMENT_PACMAN;
1166     break;
1167   case OS_DISTRO_GENTOO:
1168     fs->package_management = OS_PACKAGE_MANAGEMENT_PORTAGE;
1169     break;
1170   case OS_DISTRO_PARDUS:
1171     fs->package_management = OS_PACKAGE_MANAGEMENT_PISI;
1172     break;
1173
1174   case OS_DISTRO_WINDOWS:
1175   case OS_DISTRO_UNKNOWN:
1176   default:
1177     fs->package_management = OS_PACKAGE_MANAGEMENT_UNKNOWN;
1178     break;
1179   }
1180 }
1181
1182 static struct inspect_fs *
1183 search_for_root (guestfs_h *g, const char *root)
1184 {
1185   if (g->nr_fses == 0) {
1186     error (g, _("no inspection data: call guestfs_inspect_os first"));
1187     return NULL;
1188   }
1189
1190   size_t i;
1191   struct inspect_fs *fs;
1192   for (i = 0; i < g->nr_fses; ++i) {
1193     fs = &g->fses[i];
1194     if (fs->is_root && STREQ (root, fs->device))
1195       return fs;
1196   }
1197
1198   error (g, _("%s: root device not found: only call this function with a root device previously returned by guestfs_inspect_os"),
1199          root);
1200   return NULL;
1201 }
1202
1203 char **
1204 guestfs__inspect_get_roots (guestfs_h *g)
1205 {
1206   /* NB. Doesn't matter if g->nr_fses == 0.  We just return an empty
1207    * list in this case.
1208    */
1209
1210   size_t i;
1211   size_t count = 0;
1212   for (i = 0; i < g->nr_fses; ++i)
1213     if (g->fses[i].is_root)
1214       count++;
1215
1216   char **ret = calloc (count+1, sizeof (char *));
1217   if (ret == NULL) {
1218     perrorf (g, "calloc");
1219     return NULL;
1220   }
1221
1222   count = 0;
1223   for (i = 0; i < g->nr_fses; ++i) {
1224     if (g->fses[i].is_root) {
1225       ret[count] = safe_strdup (g, g->fses[i].device);
1226       count++;
1227     }
1228   }
1229   ret[count] = NULL;
1230
1231   return ret;
1232 }
1233
1234 char *
1235 guestfs__inspect_get_type (guestfs_h *g, const char *root)
1236 {
1237   struct inspect_fs *fs = search_for_root (g, root);
1238   if (!fs)
1239     return NULL;
1240
1241   char *ret;
1242   switch (fs->type) {
1243   case OS_TYPE_LINUX: ret = safe_strdup (g, "linux"); break;
1244   case OS_TYPE_WINDOWS: ret = safe_strdup (g, "windows"); break;
1245   case OS_TYPE_FREEBSD: ret = safe_strdup (g, "freebsd"); break;
1246   case OS_TYPE_UNKNOWN: default: ret = safe_strdup (g, "unknown"); break;
1247   }
1248
1249   return ret;
1250 }
1251
1252 char *
1253 guestfs__inspect_get_arch (guestfs_h *g, const char *root)
1254 {
1255   struct inspect_fs *fs = search_for_root (g, root);
1256   if (!fs)
1257     return NULL;
1258
1259   return safe_strdup (g, fs->arch ? : "unknown");
1260 }
1261
1262 char *
1263 guestfs__inspect_get_distro (guestfs_h *g, const char *root)
1264 {
1265   struct inspect_fs *fs = search_for_root (g, root);
1266   if (!fs)
1267     return NULL;
1268
1269   char *ret;
1270   switch (fs->distro) {
1271   case OS_DISTRO_ARCHLINUX: ret = safe_strdup (g, "archlinux"); break;
1272   case OS_DISTRO_DEBIAN: ret = safe_strdup (g, "debian"); break;
1273   case OS_DISTRO_FEDORA: ret = safe_strdup (g, "fedora"); break;
1274   case OS_DISTRO_GENTOO: ret = safe_strdup (g, "gentoo"); break;
1275   case OS_DISTRO_MEEGO: ret = safe_strdup (g, "meego"); break;
1276   case OS_DISTRO_PARDUS: ret = safe_strdup (g, "pardus"); break;
1277   case OS_DISTRO_REDHAT_BASED: ret = safe_strdup (g, "redhat-based"); break;
1278   case OS_DISTRO_RHEL: ret = safe_strdup (g, "rhel"); break;
1279   case OS_DISTRO_WINDOWS: ret = safe_strdup (g, "windows"); break;
1280   case OS_DISTRO_UBUNTU: ret = safe_strdup (g, "ubuntu"); break;
1281   case OS_DISTRO_UNKNOWN: default: ret = safe_strdup (g, "unknown"); break;
1282   }
1283
1284   return ret;
1285 }
1286
1287 int
1288 guestfs__inspect_get_major_version (guestfs_h *g, const char *root)
1289 {
1290   struct inspect_fs *fs = search_for_root (g, root);
1291   if (!fs)
1292     return -1;
1293
1294   return fs->major_version;
1295 }
1296
1297 int
1298 guestfs__inspect_get_minor_version (guestfs_h *g, const char *root)
1299 {
1300   struct inspect_fs *fs = search_for_root (g, root);
1301   if (!fs)
1302     return -1;
1303
1304   return fs->minor_version;
1305 }
1306
1307 char *
1308 guestfs__inspect_get_product_name (guestfs_h *g, const char *root)
1309 {
1310   struct inspect_fs *fs = search_for_root (g, root);
1311   if (!fs)
1312     return NULL;
1313
1314   return safe_strdup (g, fs->product_name ? : "unknown");
1315 }
1316
1317 char *
1318 guestfs__inspect_get_windows_systemroot (guestfs_h *g, const char *root)
1319 {
1320   struct inspect_fs *fs = search_for_root (g, root);
1321   if (!fs)
1322     return NULL;
1323
1324   if (!fs->windows_systemroot) {
1325     error (g, _("not a Windows guest, or systemroot could not be determined"));
1326     return NULL;
1327   }
1328
1329   return safe_strdup (g, fs->windows_systemroot);
1330 }
1331
1332 char **
1333 guestfs__inspect_get_mountpoints (guestfs_h *g, const char *root)
1334 {
1335   struct inspect_fs *fs = search_for_root (g, root);
1336   if (!fs)
1337     return NULL;
1338
1339   char **ret;
1340
1341   /* If no fstab information (Windows) return just the root. */
1342   if (fs->nr_fstab == 0) {
1343     ret = calloc (3, sizeof (char *));
1344     ret[0] = safe_strdup (g, "/");
1345     ret[1] = safe_strdup (g, root);
1346     ret[2] = NULL;
1347     return ret;
1348   }
1349
1350 #define CRITERION fs->fstab[i].mountpoint[0] == '/'
1351   size_t i, count = 0;
1352   for (i = 0; i < fs->nr_fstab; ++i)
1353     if (CRITERION)
1354       count++;
1355
1356   /* Hashtables have 2N+1 entries. */
1357   ret = calloc (2*count+1, sizeof (char *));
1358   if (ret == NULL) {
1359     perrorf (g, "calloc");
1360     return NULL;
1361   }
1362
1363   count = 0;
1364   for (i = 0; i < fs->nr_fstab; ++i)
1365     if (CRITERION) {
1366       ret[2*count] = safe_strdup (g, fs->fstab[i].mountpoint);
1367       ret[2*count+1] = safe_strdup (g, fs->fstab[i].device);
1368       count++;
1369     }
1370 #undef CRITERION
1371
1372   return ret;
1373 }
1374
1375 char **
1376 guestfs__inspect_get_filesystems (guestfs_h *g, const char *root)
1377 {
1378   struct inspect_fs *fs = search_for_root (g, root);
1379   if (!fs)
1380     return NULL;
1381
1382   char **ret;
1383
1384   /* If no fstab information (Windows) return just the root. */
1385   if (fs->nr_fstab == 0) {
1386     ret = calloc (2, sizeof (char *));
1387     ret[0] = safe_strdup (g, root);
1388     ret[1] = NULL;
1389     return ret;
1390   }
1391
1392   ret = calloc (fs->nr_fstab + 1, sizeof (char *));
1393   if (ret == NULL) {
1394     perrorf (g, "calloc");
1395     return NULL;
1396   }
1397
1398   size_t i;
1399   for (i = 0; i < fs->nr_fstab; ++i)
1400     ret[i] = safe_strdup (g, fs->fstab[i].device);
1401
1402   return ret;
1403 }
1404
1405 char *
1406 guestfs__inspect_get_package_format (guestfs_h *g, const char *root)
1407 {
1408   struct inspect_fs *fs = search_for_root (g, root);
1409   if (!fs)
1410     return NULL;
1411
1412   char *ret;
1413   switch (fs->package_format) {
1414   case OS_PACKAGE_FORMAT_RPM: ret = safe_strdup (g, "rpm"); break;
1415   case OS_PACKAGE_FORMAT_DEB: ret = safe_strdup (g, "deb"); break;
1416   case OS_PACKAGE_FORMAT_PACMAN: ret = safe_strdup (g, "pacman"); break;
1417   case OS_PACKAGE_FORMAT_EBUILD: ret = safe_strdup (g, "ebuild"); break;
1418   case OS_PACKAGE_FORMAT_PISI: ret = safe_strdup (g, "pisi"); break;
1419   case OS_PACKAGE_FORMAT_UNKNOWN:
1420   default:
1421     ret = safe_strdup (g, "unknown");
1422     break;
1423   }
1424
1425   return ret;
1426 }
1427
1428 char *
1429 guestfs__inspect_get_package_management (guestfs_h *g, const char *root)
1430 {
1431   struct inspect_fs *fs = search_for_root (g, root);
1432   if (!fs)
1433     return NULL;
1434
1435   char *ret;
1436   switch (fs->package_management) {
1437   case OS_PACKAGE_MANAGEMENT_YUM: ret = safe_strdup (g, "yum"); break;
1438   case OS_PACKAGE_MANAGEMENT_UP2DATE: ret = safe_strdup (g, "up2date"); break;
1439   case OS_PACKAGE_MANAGEMENT_APT: ret = safe_strdup (g, "apt"); break;
1440   case OS_PACKAGE_MANAGEMENT_PACMAN: ret = safe_strdup (g, "pacman"); break;
1441   case OS_PACKAGE_MANAGEMENT_PORTAGE: ret = safe_strdup (g, "portage"); break;
1442   case OS_PACKAGE_MANAGEMENT_PISI: ret = safe_strdup (g, "pisi"); break;
1443   case OS_PACKAGE_MANAGEMENT_UNKNOWN:
1444   default:
1445     ret = safe_strdup (g, "unknown");
1446     break;
1447   }
1448
1449   return ret;
1450 }
1451
1452 #else /* no PCRE or hivex at compile time */
1453
1454 /* XXX These functions should be in an optgroup. */
1455
1456 #define NOT_IMPL(r)                                                     \
1457   error (g, _("inspection API not available since this version of libguestfs was compiled without PCRE or hivex libraries")); \
1458   return r
1459
1460 char **
1461 guestfs__inspect_os (guestfs_h *g)
1462 {
1463   NOT_IMPL(NULL);
1464 }
1465
1466 char **
1467 guestfs__inspect_get_roots (guestfs_h *g)
1468 {
1469   NOT_IMPL(NULL);
1470 }
1471
1472 char *
1473 guestfs__inspect_get_type (guestfs_h *g, const char *root)
1474 {
1475   NOT_IMPL(NULL);
1476 }
1477
1478 char *
1479 guestfs__inspect_get_arch (guestfs_h *g, const char *root)
1480 {
1481   NOT_IMPL(NULL);
1482 }
1483
1484 char *
1485 guestfs__inspect_get_distro (guestfs_h *g, const char *root)
1486 {
1487   NOT_IMPL(NULL);
1488 }
1489
1490 int
1491 guestfs__inspect_get_major_version (guestfs_h *g, const char *root)
1492 {
1493   NOT_IMPL(-1);
1494 }
1495
1496 int
1497 guestfs__inspect_get_minor_version (guestfs_h *g, const char *root)
1498 {
1499   NOT_IMPL(-1);
1500 }
1501
1502 char *
1503 guestfs__inspect_get_product_name (guestfs_h *g, const char *root)
1504 {
1505   NOT_IMPL(NULL);
1506 }
1507
1508 char *
1509 guestfs__inspect_get_windows_systemroot (guestfs_h *g, const char *root)
1510 {
1511   NOT_IMPL(NULL);
1512 }
1513
1514 char **
1515 guestfs__inspect_get_mountpoints (guestfs_h *g, const char *root)
1516 {
1517   NOT_IMPL(NULL);
1518 }
1519
1520 char **
1521 guestfs__inspect_get_filesystems (guestfs_h *g, const char *root)
1522 {
1523   NOT_IMPL(NULL);
1524 }
1525
1526 char *
1527 guestfs__inspect_get_package_format (guestfs_h *g, const char *root)
1528 {
1529   NOT_IMPL(NULL);
1530 }
1531
1532 char *
1533 guestfs__inspect_get_package_management (guestfs_h *g, const char *root)
1534 {
1535   NOT_IMPL(NULL);
1536 }
1537
1538 #endif /* no PCRE or hivex at compile time */
1539
1540 void
1541 guestfs___free_inspect_info (guestfs_h *g)
1542 {
1543   size_t i;
1544   for (i = 0; i < g->nr_fses; ++i) {
1545     free (g->fses[i].device);
1546     free (g->fses[i].product_name);
1547     free (g->fses[i].arch);
1548     free (g->fses[i].windows_systemroot);
1549     size_t j;
1550     for (j = 0; j < g->fses[i].nr_fstab; ++j) {
1551       free (g->fses[i].fstab[j].device);
1552       free (g->fses[i].fstab[j].mountpoint);
1553     }
1554     free (g->fses[i].fstab);
1555   }
1556   free (g->fses);
1557   g->nr_fses = 0;
1558   g->fses = NULL;
1559 }
1560
1561 /* In the Perl code this is a public function. */
1562 int
1563 guestfs___feature_available (guestfs_h *g, const char *feature)
1564 {
1565   /* If there's an error we should ignore it, so to do that we have to
1566    * temporarily replace the error handler with a null one.
1567    */
1568   guestfs_error_handler_cb old_error_cb = g->error_cb;
1569   g->error_cb = NULL;
1570
1571   const char *groups[] = { feature, NULL };
1572   int r = guestfs_available (g, (char * const *) groups);
1573
1574   g->error_cb = old_error_cb;
1575
1576   return r == 0 ? 1 : 0;
1577 }
1578
1579 #ifdef HAVE_PCRE
1580
1581 /* Match a regular expression which contains no captures.  Returns
1582  * true if it matches or false if it doesn't.
1583  */
1584 int
1585 guestfs___match (guestfs_h *g, const char *str, const pcre *re)
1586 {
1587   size_t len = strlen (str);
1588   int vec[30], r;
1589
1590   r = pcre_exec (re, NULL, str, len, 0, 0, vec, sizeof vec / sizeof vec[0]);
1591   if (r == PCRE_ERROR_NOMATCH)
1592     return 0;
1593   if (r != 1) {
1594     /* Internal error -- should not happen. */
1595     fprintf (stderr, "libguestfs: %s: %s: internal error: pcre_exec returned unexpected error code %d when matching against the string \"%s\"\n",
1596              __FILE__, __func__, r, str);
1597     return 0;
1598   }
1599
1600   return 1;
1601 }
1602
1603 /* Match a regular expression which contains exactly one capture.  If
1604  * the string matches, return the capture, otherwise return NULL.  The
1605  * caller must free the result.
1606  */
1607 char *
1608 guestfs___match1 (guestfs_h *g, const char *str, const pcre *re)
1609 {
1610   size_t len = strlen (str);
1611   int vec[30], r;
1612
1613   r = pcre_exec (re, NULL, str, len, 0, 0, vec, sizeof vec / sizeof vec[0]);
1614   if (r == PCRE_ERROR_NOMATCH)
1615     return NULL;
1616   if (r != 2) {
1617     /* Internal error -- should not happen. */
1618     fprintf (stderr, "libguestfs: %s: %s: internal error: pcre_exec returned unexpected error code %d when matching against the string \"%s\"\n",
1619              __FILE__, __func__, r, str);
1620     return NULL;
1621   }
1622
1623   return safe_strndup (g, &str[vec[2]], vec[3]-vec[2]);
1624 }
1625
1626 /* Match a regular expression which contains exactly two captures. */
1627 int
1628 guestfs___match2 (guestfs_h *g, const char *str, const pcre *re,
1629                   char **ret1, char **ret2)
1630 {
1631   size_t len = strlen (str);
1632   int vec[30], r;
1633
1634   r = pcre_exec (re, NULL, str, len, 0, 0, vec, 30);
1635   if (r == PCRE_ERROR_NOMATCH)
1636     return 0;
1637   if (r != 3) {
1638     /* Internal error -- should not happen. */
1639     fprintf (stderr, "libguestfs: %s: %s: internal error: pcre_exec returned unexpected error code %d when matching against the string \"%s\"\n",
1640              __FILE__, __func__, r, str);
1641     return 0;
1642   }
1643
1644   *ret1 = safe_strndup (g, &str[vec[2]], vec[3]-vec[2]);
1645   *ret2 = safe_strndup (g, &str[vec[4]], vec[5]-vec[4]);
1646
1647   return 1;
1648 }
1649
1650 /* Match a regular expression which contains exactly three captures. */
1651 int
1652 guestfs___match3 (guestfs_h *g, const char *str, const pcre *re,
1653                   char **ret1, char **ret2, char **ret3)
1654 {
1655   size_t len = strlen (str);
1656   int vec[30], r;
1657
1658   r = pcre_exec (re, NULL, str, len, 0, 0, vec, 30);
1659   if (r == PCRE_ERROR_NOMATCH)
1660     return 0;
1661   if (r != 4) {
1662     /* Internal error -- should not happen. */
1663     fprintf (stderr, "libguestfs: %s: %s: internal error: pcre_exec returned unexpected error code %d when matching against the string \"%s\"\n",
1664              __FILE__, __func__, r, str);
1665     return 0;
1666   }
1667
1668   *ret1 = safe_strndup (g, &str[vec[2]], vec[3]-vec[2]);
1669   *ret2 = safe_strndup (g, &str[vec[4]], vec[5]-vec[4]);
1670   *ret3 = safe_strndup (g, &str[vec[6]], vec[7]-vec[6]);
1671
1672   return 1;
1673 }
1674
1675 #endif /* HAVE_PCRE */