-/* Match a regular expression which contains no captures. Returns
- * true if it matches or false if it doesn't.
- */
-static int
-match (guestfs_h *g, const char *str, const pcre *re)
-{
- size_t len = strlen (str);
- int vec[30], r;
-
- r = pcre_exec (re, NULL, str, len, 0, 0, vec, sizeof vec / sizeof vec[0]);
- if (r == PCRE_ERROR_NOMATCH)
- return 0;
- if (r != 1) {
- /* Internal error -- should not happen. */
- fprintf (stderr, "libguestfs: %s: %s: internal error: pcre_exec returned unexpected error code %d when matching against the string \"%s\"\n",
- __FILE__, __func__, r, str);
- return 0;
- }
-
- return 1;
-}
-
-/* Match a regular expression which contains exactly one capture. If
- * the string matches, return the capture, otherwise return NULL. The
- * caller must free the result.
- */
-static char *
-match1 (guestfs_h *g, const char *str, const pcre *re)
-{
- size_t len = strlen (str);
- int vec[30], r;
-
- r = pcre_exec (re, NULL, str, len, 0, 0, vec, sizeof vec / sizeof vec[0]);
- if (r == PCRE_ERROR_NOMATCH)
- return NULL;
- if (r != 2) {
- /* Internal error -- should not happen. */
- fprintf (stderr, "libguestfs: %s: %s: internal error: pcre_exec returned unexpected error code %d when matching against the string \"%s\"\n",
- __FILE__, __func__, r, str);
- return NULL;
- }
-
- return safe_strndup (g, &str[vec[2]], vec[3]-vec[2]);
-}
-
-/* Match a regular expression which contains exactly two captures. */
-static int
-match2 (guestfs_h *g, const char *str, const pcre *re, char **ret1, char **ret2)
-{
- size_t len = strlen (str);
- int vec[30], r;
-
- r = pcre_exec (re, NULL, str, len, 0, 0, vec, 30);
- if (r == PCRE_ERROR_NOMATCH)
- return 0;
- if (r != 3) {
- /* Internal error -- should not happen. */
- fprintf (stderr, "libguestfs: %s: %s: internal error: pcre_exec returned unexpected error code %d when matching against the string \"%s\"\n",
- __FILE__, __func__, r, str);
- return 0;
- }
-
- *ret1 = safe_strndup (g, &str[vec[2]], vec[3]-vec[2]);
- *ret2 = safe_strndup (g, &str[vec[4]], vec[5]-vec[4]);
-
- return 1;
-}
-
-/* Convert output from 'file' command on ELF files to the canonical
- * architecture string. Caller must free the result.
- */
-static char *
-canonical_elf_arch (guestfs_h *g, const char *elf_arch)
-{
- const char *r;
-
- if (strstr (elf_arch, "Intel 80386"))
- r = "i386";
- else if (strstr (elf_arch, "Intel 80486"))
- r = "i486";
- else if (strstr (elf_arch, "x86-64"))
- r = "x86_64";
- else if (strstr (elf_arch, "AMD x86-64"))
- r = "x86_64";
- else if (strstr (elf_arch, "SPARC32"))
- r = "sparc";
- else if (strstr (elf_arch, "SPARC V9"))
- r = "sparc64";
- else if (strstr (elf_arch, "IA-64"))
- r = "ia64";
- else if (match (g, elf_arch, re_elf_ppc64))
- r = "ppc64";
- else if (strstr (elf_arch, "PowerPC"))
- r = "ppc";
- else
- r = elf_arch;
-
- char *ret = safe_strdup (g, r);
- return ret;
-}
-
-static int
-is_regular_file (const char *filename)
-{
- struct stat statbuf;
-
- return lstat (filename, &statbuf) == 0 && S_ISREG (statbuf.st_mode);
-}
-
-/* Download and uncompress the cpio file to find binaries within.
- * Notes:
- * (1) Two lists must be identical.
- * (2) Implicit limit of 31 bytes for length of each element (see code
- * below).
- */
-#define INITRD_BINARIES1 "bin/ls bin/rm bin/modprobe sbin/modprobe bin/sh bin/bash bin/dash bin/nash"
-#define INITRD_BINARIES2 {"bin/ls", "bin/rm", "bin/modprobe", "sbin/modprobe", "bin/sh", "bin/bash", "bin/dash", "bin/nash"}
-
-static char *
-cpio_arch (guestfs_h *g, const char *file, const char *path)
-{
- TMP_TEMPLATE_ON_STACK (dir);
-#define dir_len (strlen (dir))
-#define initrd_len (dir_len + 16)
- char initrd[initrd_len];
-#define cmd_len (dir_len + 256)
- char cmd[cmd_len];
-#define bin_len (dir_len + 32)
- char bin[bin_len];
-
- char *ret = NULL;
-
- const char *method;
- if (strstr (file, "gzip"))
- method = "zcat";
- else if (strstr (file, "bzip2"))
- method = "bzcat";
- else
- method = "cat";
-
- if (mkdtemp (dir) == NULL) {
- perrorf (g, "mkdtemp");
- goto out;
- }
-
- snprintf (initrd, initrd_len, "%s/initrd", dir);
- if (guestfs_download (g, path, initrd) == -1)
- goto out;
-
- snprintf (cmd, cmd_len,
- "cd %s && %s initrd | cpio --quiet -id " INITRD_BINARIES1,
- dir, method);
- int r = system (cmd);
- if (r == -1 || WEXITSTATUS (r) != 0) {
- perrorf (g, "cpio command failed");
- goto out;
- }
-
- const char *bins[] = INITRD_BINARIES2;
- size_t i;
- for (i = 0; i < sizeof bins / sizeof bins[0]; ++i) {
- snprintf (bin, bin_len, "%s/%s", dir, bins[i]);
-
- if (is_regular_file (bin)) {
- int flags = g->verbose ? MAGIC_DEBUG : 0;
- flags |= MAGIC_ERROR | MAGIC_RAW;
-
- magic_t m = magic_open (flags);
- if (m == NULL) {
- perrorf (g, "magic_open");
- goto out;
- }
-
- if (magic_load (m, NULL) == -1) {
- perrorf (g, "magic_load: default magic database file");
- magic_close (m);
- goto out;
- }
-
- const char *line = magic_file (m, bin);
- if (line == NULL) {
- perrorf (g, "magic_file: %s", bin);
- magic_close (m);
- goto out;
- }
-
- char *elf_arch;
- if ((elf_arch = match1 (g, line, re_file_elf)) != NULL) {
- ret = canonical_elf_arch (g, elf_arch);
- free (elf_arch);
- magic_close (m);
- goto out;
- }
- magic_close (m);
- }
- }
- error (g, "file_architecture: could not determine architecture of cpio archive");
-
- out:
- /* Free up the temporary directory. Note the directory name cannot
- * contain shell meta-characters because of the way it was
- * constructed above.
- */
- snprintf (cmd, cmd_len, "rm -rf %s", dir);
- ignore_value (system (cmd));
-
- return ret;
-#undef dir_len
-#undef initrd_len
-#undef cmd_len
-#undef bin_len
-}
-
-char *
-guestfs__file_architecture (guestfs_h *g, const char *path)
-{
- char *file = NULL;
- char *elf_arch = NULL;
- char *ret = NULL;
-
- /* Get the output of the "file" command. Note that because this
- * runs in the daemon, LANG=C so it's in English.
- */
- file = guestfs_file (g, path);
- if (file == NULL)
- return NULL;
-
- if ((elf_arch = match1 (g, file, re_file_elf)) != NULL)
- ret = canonical_elf_arch (g, elf_arch);
- else if (strstr (file, "PE32 executable"))
- ret = safe_strdup (g, "i386");
- else if (strstr (file, "PE32+ executable"))
- ret = safe_strdup (g, "x86_64");
- else if (strstr (file, "cpio archive"))
- ret = cpio_arch (g, file, path);
- else
- error (g, "file_architecture: unknown architecture: %s", path);
-
- free (file);
- free (elf_arch);
- return ret; /* caller frees */
-}
-