inspect: Refuse to parse /etc/fstab if it is huge.
[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   if (mkdtemp (dir) == NULL) {
954     perrorf (g, "mkdtemp");
955     goto out;
956   }
957
958   snprintf (software_hive, software_hive_len, "%s/software", dir);
959
960   if (guestfs_download (g, software_path, software_hive) == -1)
961     goto out;
962
963   h = hivex_open (software_hive, g->verbose ? HIVEX_OPEN_VERBOSE : 0);
964   if (h == NULL) {
965     perrorf (g, "hivex_open");
966     goto out;
967   }
968
969   hive_node_h node = hivex_root (h);
970   const char *hivepath[] =
971     { "Microsoft", "Windows NT", "CurrentVersion" };
972   size_t i;
973   for (i = 0;
974        node != 0 && i < sizeof hivepath / sizeof hivepath[0];
975        ++i) {
976     node = hivex_node_get_child (h, node, hivepath[i]);
977   }
978
979   if (node == 0) {
980     perrorf (g, "hivex: cannot locate HKLM\\SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion");
981     goto out;
982   }
983
984   values = hivex_node_values (h, node);
985
986   for (i = 0; values[i] != 0; ++i) {
987     char *key = hivex_value_key (h, values[i]);
988     if (key == NULL) {
989       perrorf (g, "hivex_value_key");
990       goto out;
991     }
992
993     if (STRCASEEQ (key, "ProductName")) {
994       fs->product_name = hivex_value_string (h, values[i]);
995       if (!fs->product_name) {
996         perrorf (g, "hivex_value_string");
997         free (key);
998         goto out;
999       }
1000     }
1001     else if (STRCASEEQ (key, "CurrentVersion")) {
1002       char *version = hivex_value_string (h, values[i]);
1003       if (!version) {
1004         perrorf (g, "hivex_value_string");
1005         free (key);
1006         goto out;
1007       }
1008       char *major, *minor;
1009       if (match2 (g, version, re_windows_version, &major, &minor)) {
1010         fs->major_version = parse_unsigned_int (g, major);
1011         free (major);
1012         if (fs->major_version == -1) {
1013           free (minor);
1014           free (key);
1015           free (version);
1016           goto out;
1017         }
1018         fs->minor_version = parse_unsigned_int (g, minor);
1019         free (minor);
1020         if (fs->minor_version == -1) {
1021           free (key);
1022           free (version);
1023           return -1;
1024         }
1025       }
1026
1027       free (version);
1028     }
1029
1030     free (key);
1031   }
1032
1033   ret = 0;
1034
1035  out:
1036   if (h) hivex_close (h);
1037   free (values);
1038   free (software_path);
1039
1040   /* Free up the temporary directory.  Note the directory name cannot
1041    * contain shell meta-characters because of the way it was
1042    * constructed above.
1043    */
1044   snprintf (cmd, cmd_len, "rm -rf %s", dir);
1045   ignore_value (system (cmd));
1046 #undef dir_len
1047 #undef software_hive_len
1048 #undef cmd_len
1049
1050   return ret;
1051 }
1052
1053 static char *
1054 resolve_windows_path_silently (guestfs_h *g, const char *path)
1055 {
1056   guestfs_error_handler_cb old_error_cb = g->error_cb;
1057   g->error_cb = NULL;
1058   char *ret = guestfs_case_sensitive_path (g, path);
1059   g->error_cb = old_error_cb;
1060   return ret;
1061 }
1062
1063 static int
1064 extend_fses (guestfs_h *g)
1065 {
1066   size_t n = g->nr_fses + 1;
1067   struct inspect_fs *p;
1068
1069   p = realloc (g->fses, n * sizeof (struct inspect_fs));
1070   if (p == NULL) {
1071     perrorf (g, "realloc");
1072     return -1;
1073   }
1074
1075   g->fses = p;
1076   g->nr_fses = n;
1077
1078   memset (&g->fses[n-1], 0, sizeof (struct inspect_fs));
1079
1080   return 0;
1081 }
1082
1083 /* Parse small, unsigned ints, as used in version numbers. */
1084 static int
1085 parse_unsigned_int (guestfs_h *g, const char *str)
1086 {
1087   long ret;
1088   int r = xstrtol (str, NULL, 10, &ret, "");
1089   if (r != LONGINT_OK) {
1090     error (g, _("could not parse integer in version number: %s"), str);
1091     return -1;
1092   }
1093   return ret;
1094 }
1095
1096 /* At the moment, package format and package management is just a
1097  * simple function of the distro and major_version fields, so these
1098  * can never return an error.  We might be cleverer in future.
1099  */
1100 static void
1101 check_package_format (guestfs_h *g, struct inspect_fs *fs)
1102 {
1103   switch (fs->distro) {
1104   case OS_DISTRO_FEDORA:
1105   case OS_DISTRO_MEEGO:
1106   case OS_DISTRO_REDHAT_BASED:
1107   case OS_DISTRO_RHEL:
1108     fs->package_format = OS_PACKAGE_FORMAT_RPM;
1109     break;
1110
1111   case OS_DISTRO_DEBIAN:
1112   case OS_DISTRO_UBUNTU:
1113     fs->package_format = OS_PACKAGE_FORMAT_DEB;
1114     break;
1115
1116   case OS_DISTRO_ARCHLINUX:
1117     fs->package_format = OS_PACKAGE_FORMAT_PACMAN;
1118     break;
1119   case OS_DISTRO_GENTOO:
1120     fs->package_format = OS_PACKAGE_FORMAT_EBUILD;
1121     break;
1122   case OS_DISTRO_PARDUS:
1123     fs->package_format = OS_PACKAGE_FORMAT_PISI;
1124     break;
1125
1126   case OS_DISTRO_WINDOWS:
1127   case OS_DISTRO_UNKNOWN:
1128   default:
1129     fs->package_format = OS_PACKAGE_FORMAT_UNKNOWN;
1130     break;
1131   }
1132 }
1133
1134 static void
1135 check_package_management (guestfs_h *g, struct inspect_fs *fs)
1136 {
1137   switch (fs->distro) {
1138   case OS_DISTRO_FEDORA:
1139   case OS_DISTRO_MEEGO:
1140     fs->package_management = OS_PACKAGE_MANAGEMENT_YUM;
1141     break;
1142
1143   case OS_DISTRO_REDHAT_BASED:
1144   case OS_DISTRO_RHEL:
1145     if (fs->major_version >= 5)
1146       fs->package_management = OS_PACKAGE_MANAGEMENT_YUM;
1147     else
1148       fs->package_management = OS_PACKAGE_MANAGEMENT_UP2DATE;
1149     break;
1150
1151   case OS_DISTRO_DEBIAN:
1152   case OS_DISTRO_UBUNTU:
1153     fs->package_management = OS_PACKAGE_MANAGEMENT_APT;
1154     break;
1155
1156   case OS_DISTRO_ARCHLINUX:
1157     fs->package_management = OS_PACKAGE_MANAGEMENT_PACMAN;
1158     break;
1159   case OS_DISTRO_GENTOO:
1160     fs->package_management = OS_PACKAGE_MANAGEMENT_PORTAGE;
1161     break;
1162   case OS_DISTRO_PARDUS:
1163     fs->package_management = OS_PACKAGE_MANAGEMENT_PISI;
1164     break;
1165
1166   case OS_DISTRO_WINDOWS:
1167   case OS_DISTRO_UNKNOWN:
1168   default:
1169     fs->package_management = OS_PACKAGE_MANAGEMENT_UNKNOWN;
1170     break;
1171   }
1172 }
1173
1174 static struct inspect_fs *
1175 search_for_root (guestfs_h *g, const char *root)
1176 {
1177   if (g->nr_fses == 0) {
1178     error (g, _("no inspection data: call guestfs_inspect_os first"));
1179     return NULL;
1180   }
1181
1182   size_t i;
1183   struct inspect_fs *fs;
1184   for (i = 0; i < g->nr_fses; ++i) {
1185     fs = &g->fses[i];
1186     if (fs->is_root && STREQ (root, fs->device))
1187       return fs;
1188   }
1189
1190   error (g, _("%s: root device not found: only call this function with a root device previously returned by guestfs_inspect_os"),
1191          root);
1192   return NULL;
1193 }
1194
1195 char **
1196 guestfs__inspect_get_roots (guestfs_h *g)
1197 {
1198   /* NB. Doesn't matter if g->nr_fses == 0.  We just return an empty
1199    * list in this case.
1200    */
1201
1202   size_t i;
1203   size_t count = 0;
1204   for (i = 0; i < g->nr_fses; ++i)
1205     if (g->fses[i].is_root)
1206       count++;
1207
1208   char **ret = calloc (count+1, sizeof (char *));
1209   if (ret == NULL) {
1210     perrorf (g, "calloc");
1211     return NULL;
1212   }
1213
1214   count = 0;
1215   for (i = 0; i < g->nr_fses; ++i) {
1216     if (g->fses[i].is_root) {
1217       ret[count] = safe_strdup (g, g->fses[i].device);
1218       count++;
1219     }
1220   }
1221   ret[count] = NULL;
1222
1223   return ret;
1224 }
1225
1226 char *
1227 guestfs__inspect_get_type (guestfs_h *g, const char *root)
1228 {
1229   struct inspect_fs *fs = search_for_root (g, root);
1230   if (!fs)
1231     return NULL;
1232
1233   char *ret;
1234   switch (fs->type) {
1235   case OS_TYPE_LINUX: ret = safe_strdup (g, "linux"); break;
1236   case OS_TYPE_WINDOWS: ret = safe_strdup (g, "windows"); break;
1237   case OS_TYPE_FREEBSD: ret = safe_strdup (g, "freebsd"); break;
1238   case OS_TYPE_UNKNOWN: default: ret = safe_strdup (g, "unknown"); break;
1239   }
1240
1241   return ret;
1242 }
1243
1244 char *
1245 guestfs__inspect_get_arch (guestfs_h *g, const char *root)
1246 {
1247   struct inspect_fs *fs = search_for_root (g, root);
1248   if (!fs)
1249     return NULL;
1250
1251   return safe_strdup (g, fs->arch ? : "unknown");
1252 }
1253
1254 char *
1255 guestfs__inspect_get_distro (guestfs_h *g, const char *root)
1256 {
1257   struct inspect_fs *fs = search_for_root (g, root);
1258   if (!fs)
1259     return NULL;
1260
1261   char *ret;
1262   switch (fs->distro) {
1263   case OS_DISTRO_ARCHLINUX: ret = safe_strdup (g, "archlinux"); break;
1264   case OS_DISTRO_DEBIAN: ret = safe_strdup (g, "debian"); break;
1265   case OS_DISTRO_FEDORA: ret = safe_strdup (g, "fedora"); break;
1266   case OS_DISTRO_GENTOO: ret = safe_strdup (g, "gentoo"); break;
1267   case OS_DISTRO_MEEGO: ret = safe_strdup (g, "meego"); break;
1268   case OS_DISTRO_PARDUS: ret = safe_strdup (g, "pardus"); break;
1269   case OS_DISTRO_REDHAT_BASED: ret = safe_strdup (g, "redhat-based"); break;
1270   case OS_DISTRO_RHEL: ret = safe_strdup (g, "rhel"); break;
1271   case OS_DISTRO_WINDOWS: ret = safe_strdup (g, "windows"); break;
1272   case OS_DISTRO_UBUNTU: ret = safe_strdup (g, "ubuntu"); break;
1273   case OS_DISTRO_UNKNOWN: default: ret = safe_strdup (g, "unknown"); break;
1274   }
1275
1276   return ret;
1277 }
1278
1279 int
1280 guestfs__inspect_get_major_version (guestfs_h *g, const char *root)
1281 {
1282   struct inspect_fs *fs = search_for_root (g, root);
1283   if (!fs)
1284     return -1;
1285
1286   return fs->major_version;
1287 }
1288
1289 int
1290 guestfs__inspect_get_minor_version (guestfs_h *g, const char *root)
1291 {
1292   struct inspect_fs *fs = search_for_root (g, root);
1293   if (!fs)
1294     return -1;
1295
1296   return fs->minor_version;
1297 }
1298
1299 char *
1300 guestfs__inspect_get_product_name (guestfs_h *g, const char *root)
1301 {
1302   struct inspect_fs *fs = search_for_root (g, root);
1303   if (!fs)
1304     return NULL;
1305
1306   return safe_strdup (g, fs->product_name ? : "unknown");
1307 }
1308
1309 char *
1310 guestfs__inspect_get_windows_systemroot (guestfs_h *g, const char *root)
1311 {
1312   struct inspect_fs *fs = search_for_root (g, root);
1313   if (!fs)
1314     return NULL;
1315
1316   if (!fs->windows_systemroot) {
1317     error (g, _("not a Windows guest, or systemroot could not be determined"));
1318     return NULL;
1319   }
1320
1321   return safe_strdup (g, fs->windows_systemroot);
1322 }
1323
1324 char **
1325 guestfs__inspect_get_mountpoints (guestfs_h *g, const char *root)
1326 {
1327   struct inspect_fs *fs = search_for_root (g, root);
1328   if (!fs)
1329     return NULL;
1330
1331   char **ret;
1332
1333   /* If no fstab information (Windows) return just the root. */
1334   if (fs->nr_fstab == 0) {
1335     ret = calloc (3, sizeof (char *));
1336     ret[0] = safe_strdup (g, "/");
1337     ret[1] = safe_strdup (g, root);
1338     ret[2] = NULL;
1339     return ret;
1340   }
1341
1342 #define CRITERION fs->fstab[i].mountpoint[0] == '/'
1343   size_t i, count = 0;
1344   for (i = 0; i < fs->nr_fstab; ++i)
1345     if (CRITERION)
1346       count++;
1347
1348   /* Hashtables have 2N+1 entries. */
1349   ret = calloc (2*count+1, sizeof (char *));
1350   if (ret == NULL) {
1351     perrorf (g, "calloc");
1352     return NULL;
1353   }
1354
1355   count = 0;
1356   for (i = 0; i < fs->nr_fstab; ++i)
1357     if (CRITERION) {
1358       ret[2*count] = safe_strdup (g, fs->fstab[i].mountpoint);
1359       ret[2*count+1] = safe_strdup (g, fs->fstab[i].device);
1360       count++;
1361     }
1362 #undef CRITERION
1363
1364   return ret;
1365 }
1366
1367 char **
1368 guestfs__inspect_get_filesystems (guestfs_h *g, const char *root)
1369 {
1370   struct inspect_fs *fs = search_for_root (g, root);
1371   if (!fs)
1372     return NULL;
1373
1374   char **ret;
1375
1376   /* If no fstab information (Windows) return just the root. */
1377   if (fs->nr_fstab == 0) {
1378     ret = calloc (2, sizeof (char *));
1379     ret[0] = safe_strdup (g, root);
1380     ret[1] = NULL;
1381     return ret;
1382   }
1383
1384   ret = calloc (fs->nr_fstab + 1, sizeof (char *));
1385   if (ret == NULL) {
1386     perrorf (g, "calloc");
1387     return NULL;
1388   }
1389
1390   size_t i;
1391   for (i = 0; i < fs->nr_fstab; ++i)
1392     ret[i] = safe_strdup (g, fs->fstab[i].device);
1393
1394   return ret;
1395 }
1396
1397 char *
1398 guestfs__inspect_get_package_format (guestfs_h *g, const char *root)
1399 {
1400   struct inspect_fs *fs = search_for_root (g, root);
1401   if (!fs)
1402     return NULL;
1403
1404   char *ret;
1405   switch (fs->package_format) {
1406   case OS_PACKAGE_FORMAT_RPM: ret = safe_strdup (g, "rpm"); break;
1407   case OS_PACKAGE_FORMAT_DEB: ret = safe_strdup (g, "deb"); break;
1408   case OS_PACKAGE_FORMAT_PACMAN: ret = safe_strdup (g, "pacman"); break;
1409   case OS_PACKAGE_FORMAT_EBUILD: ret = safe_strdup (g, "ebuild"); break;
1410   case OS_PACKAGE_FORMAT_PISI: ret = safe_strdup (g, "pisi"); break;
1411   case OS_PACKAGE_FORMAT_UNKNOWN:
1412   default:
1413     ret = safe_strdup (g, "unknown");
1414     break;
1415   }
1416
1417   return ret;
1418 }
1419
1420 char *
1421 guestfs__inspect_get_package_management (guestfs_h *g, const char *root)
1422 {
1423   struct inspect_fs *fs = search_for_root (g, root);
1424   if (!fs)
1425     return NULL;
1426
1427   char *ret;
1428   switch (fs->package_management) {
1429   case OS_PACKAGE_MANAGEMENT_YUM: ret = safe_strdup (g, "yum"); break;
1430   case OS_PACKAGE_MANAGEMENT_UP2DATE: ret = safe_strdup (g, "up2date"); break;
1431   case OS_PACKAGE_MANAGEMENT_APT: ret = safe_strdup (g, "apt"); break;
1432   case OS_PACKAGE_MANAGEMENT_PACMAN: ret = safe_strdup (g, "pacman"); break;
1433   case OS_PACKAGE_MANAGEMENT_PORTAGE: ret = safe_strdup (g, "portage"); break;
1434   case OS_PACKAGE_MANAGEMENT_PISI: ret = safe_strdup (g, "pisi"); break;
1435   case OS_PACKAGE_MANAGEMENT_UNKNOWN:
1436   default:
1437     ret = safe_strdup (g, "unknown");
1438     break;
1439   }
1440
1441   return ret;
1442 }
1443
1444 #else /* no PCRE or hivex at compile time */
1445
1446 /* XXX These functions should be in an optgroup. */
1447
1448 #define NOT_IMPL(r)                                                     \
1449   error (g, _("inspection API not available since this version of libguestfs was compiled without PCRE or hivex libraries")); \
1450   return r
1451
1452 char **
1453 guestfs__inspect_os (guestfs_h *g)
1454 {
1455   NOT_IMPL(NULL);
1456 }
1457
1458 char **
1459 guestfs__inspect_get_roots (guestfs_h *g)
1460 {
1461   NOT_IMPL(NULL);
1462 }
1463
1464 char *
1465 guestfs__inspect_get_type (guestfs_h *g, const char *root)
1466 {
1467   NOT_IMPL(NULL);
1468 }
1469
1470 char *
1471 guestfs__inspect_get_arch (guestfs_h *g, const char *root)
1472 {
1473   NOT_IMPL(NULL);
1474 }
1475
1476 char *
1477 guestfs__inspect_get_distro (guestfs_h *g, const char *root)
1478 {
1479   NOT_IMPL(NULL);
1480 }
1481
1482 int
1483 guestfs__inspect_get_major_version (guestfs_h *g, const char *root)
1484 {
1485   NOT_IMPL(-1);
1486 }
1487
1488 int
1489 guestfs__inspect_get_minor_version (guestfs_h *g, const char *root)
1490 {
1491   NOT_IMPL(-1);
1492 }
1493
1494 char *
1495 guestfs__inspect_get_product_name (guestfs_h *g, const char *root)
1496 {
1497   NOT_IMPL(NULL);
1498 }
1499
1500 char *
1501 guestfs__inspect_get_windows_systemroot (guestfs_h *g, const char *root)
1502 {
1503   NOT_IMPL(NULL);
1504 }
1505
1506 char **
1507 guestfs__inspect_get_mountpoints (guestfs_h *g, const char *root)
1508 {
1509   NOT_IMPL(NULL);
1510 }
1511
1512 char **
1513 guestfs__inspect_get_filesystems (guestfs_h *g, const char *root)
1514 {
1515   NOT_IMPL(NULL);
1516 }
1517
1518 char *
1519 guestfs__inspect_get_package_format (guestfs_h *g, const char *root)
1520 {
1521   NOT_IMPL(NULL);
1522 }
1523
1524 char *
1525 guestfs__inspect_get_package_management (guestfs_h *g, const char *root)
1526 {
1527   NOT_IMPL(NULL);
1528 }
1529
1530 #endif /* no PCRE or hivex at compile time */
1531
1532 void
1533 guestfs___free_inspect_info (guestfs_h *g)
1534 {
1535   size_t i;
1536   for (i = 0; i < g->nr_fses; ++i) {
1537     free (g->fses[i].device);
1538     free (g->fses[i].product_name);
1539     free (g->fses[i].arch);
1540     free (g->fses[i].windows_systemroot);
1541     size_t j;
1542     for (j = 0; j < g->fses[i].nr_fstab; ++j) {
1543       free (g->fses[i].fstab[j].device);
1544       free (g->fses[i].fstab[j].mountpoint);
1545     }
1546     free (g->fses[i].fstab);
1547   }
1548   free (g->fses);
1549   g->nr_fses = 0;
1550   g->fses = NULL;
1551 }
1552
1553 /* In the Perl code this is a public function. */
1554 int
1555 guestfs___feature_available (guestfs_h *g, const char *feature)
1556 {
1557   /* If there's an error we should ignore it, so to do that we have to
1558    * temporarily replace the error handler with a null one.
1559    */
1560   guestfs_error_handler_cb old_error_cb = g->error_cb;
1561   g->error_cb = NULL;
1562
1563   const char *groups[] = { feature, NULL };
1564   int r = guestfs_available (g, (char * const *) groups);
1565
1566   g->error_cb = old_error_cb;
1567
1568   return r == 0 ? 1 : 0;
1569 }
1570
1571 #ifdef HAVE_PCRE
1572
1573 /* Match a regular expression which contains no captures.  Returns
1574  * true if it matches or false if it doesn't.
1575  */
1576 int
1577 guestfs___match (guestfs_h *g, const char *str, const pcre *re)
1578 {
1579   size_t len = strlen (str);
1580   int vec[30], r;
1581
1582   r = pcre_exec (re, NULL, str, len, 0, 0, vec, sizeof vec / sizeof vec[0]);
1583   if (r == PCRE_ERROR_NOMATCH)
1584     return 0;
1585   if (r != 1) {
1586     /* Internal error -- should not happen. */
1587     fprintf (stderr, "libguestfs: %s: %s: internal error: pcre_exec returned unexpected error code %d when matching against the string \"%s\"\n",
1588              __FILE__, __func__, r, str);
1589     return 0;
1590   }
1591
1592   return 1;
1593 }
1594
1595 /* Match a regular expression which contains exactly one capture.  If
1596  * the string matches, return the capture, otherwise return NULL.  The
1597  * caller must free the result.
1598  */
1599 char *
1600 guestfs___match1 (guestfs_h *g, const char *str, const pcre *re)
1601 {
1602   size_t len = strlen (str);
1603   int vec[30], r;
1604
1605   r = pcre_exec (re, NULL, str, len, 0, 0, vec, sizeof vec / sizeof vec[0]);
1606   if (r == PCRE_ERROR_NOMATCH)
1607     return NULL;
1608   if (r != 2) {
1609     /* Internal error -- should not happen. */
1610     fprintf (stderr, "libguestfs: %s: %s: internal error: pcre_exec returned unexpected error code %d when matching against the string \"%s\"\n",
1611              __FILE__, __func__, r, str);
1612     return NULL;
1613   }
1614
1615   return safe_strndup (g, &str[vec[2]], vec[3]-vec[2]);
1616 }
1617
1618 /* Match a regular expression which contains exactly two captures. */
1619 int
1620 guestfs___match2 (guestfs_h *g, const char *str, const pcre *re,
1621                   char **ret1, char **ret2)
1622 {
1623   size_t len = strlen (str);
1624   int vec[30], r;
1625
1626   r = pcre_exec (re, NULL, str, len, 0, 0, vec, 30);
1627   if (r == PCRE_ERROR_NOMATCH)
1628     return 0;
1629   if (r != 3) {
1630     /* Internal error -- should not happen. */
1631     fprintf (stderr, "libguestfs: %s: %s: internal error: pcre_exec returned unexpected error code %d when matching against the string \"%s\"\n",
1632              __FILE__, __func__, r, str);
1633     return 0;
1634   }
1635
1636   *ret1 = safe_strndup (g, &str[vec[2]], vec[3]-vec[2]);
1637   *ret2 = safe_strndup (g, &str[vec[4]], vec[5]-vec[4]);
1638
1639   return 1;
1640 }
1641
1642 /* Match a regular expression which contains exactly three captures. */
1643 int
1644 guestfs___match3 (guestfs_h *g, const char *str, const pcre *re,
1645                   char **ret1, char **ret2, char **ret3)
1646 {
1647   size_t len = strlen (str);
1648   int vec[30], r;
1649
1650   r = pcre_exec (re, NULL, str, len, 0, 0, vec, 30);
1651   if (r == PCRE_ERROR_NOMATCH)
1652     return 0;
1653   if (r != 4) {
1654     /* Internal error -- should not happen. */
1655     fprintf (stderr, "libguestfs: %s: %s: internal error: pcre_exec returned unexpected error code %d when matching against the string \"%s\"\n",
1656              __FILE__, __func__, r, str);
1657     return 0;
1658   }
1659
1660   *ret1 = safe_strndup (g, &str[vec[2]], vec[3]-vec[2]);
1661   *ret2 = safe_strndup (g, &str[vec[4]], vec[5]-vec[4]);
1662   *ret3 = safe_strndup (g, &str[vec[6]], vec[7]-vec[6]);
1663
1664   return 1;
1665 }
1666
1667 #endif /* HAVE_PCRE */