| /* cpio.c - cpio and tar filesystem. */ |
| /* |
| * GRUB -- GRand Unified Bootloader |
| * Copyright (C) 2007,2008,2009 Free Software Foundation, Inc. |
| * |
| * This program is free software: you can redistribute it and/or modify |
| * it under the terms of the GNU General Public License as published by |
| * the Free Software Foundation, either version 3 of the License, or |
| * (at your option) any later version. |
| * |
| * This program is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| * GNU General Public License for more details. |
| * |
| * You should have received a copy of the GNU General Public License |
| * along with this program. If not, see <http://www.gnu.org/licenses/>. |
| */ |
| |
| #include <grub/file.h> |
| #include <grub/mm.h> |
| #include <grub/misc.h> |
| #include <grub/disk.h> |
| #include <grub/dl.h> |
| #include <grub/i18n.h> |
| |
| GRUB_MOD_LICENSE ("GPLv3+"); |
| |
| #define ATTR_TYPE 0160000 |
| #define ATTR_FILE 0100000 |
| #define ATTR_DIR 0040000 |
| #define ATTR_LNK 0120000 |
| |
| #ifdef MODE_ODC |
| #define ALIGN_CPIO(x) x |
| #define MAGIC "070707" |
| struct head |
| { |
| char magic[6]; |
| char dev[6]; |
| char ino[6]; |
| char mode[6]; |
| char uid[6]; |
| char gid[6]; |
| char nlink[6]; |
| char rdev[6]; |
| char mtime[11]; |
| char namesize[6]; |
| char filesize[11]; |
| } __attribute__ ((packed)); |
| #elif defined (MODE_NEWC) |
| #define ALIGN_CPIO(x) (ALIGN_UP ((x), 4)) |
| #define MAGIC "070701" |
| #define MAGIC2 "070702" |
| struct head |
| { |
| char magic[6]; |
| char ino[8]; |
| char mode[8]; |
| char uid[8]; |
| char gid[8]; |
| char nlink[8]; |
| char mtime[8]; |
| char filesize[8]; |
| char devmajor[8]; |
| char devminor[8]; |
| char rdevmajor[8]; |
| char rdevminor[8]; |
| char namesize[8]; |
| char check[8]; |
| } __attribute__ ((packed)); |
| #elif !defined (MODE_USTAR) |
| /* cpio support */ |
| #define ALIGN_CPIO(x) (ALIGN_UP ((x), 2)) |
| #ifdef MODE_BIGENDIAN |
| #define MAGIC "\x71\xc7" |
| #else |
| #define MAGIC "\xc7\x71" |
| #endif |
| struct head |
| { |
| grub_uint16_t magic[1]; |
| grub_uint16_t dev; |
| grub_uint16_t ino; |
| grub_uint16_t mode[1]; |
| grub_uint16_t uid; |
| grub_uint16_t gid; |
| grub_uint16_t nlink; |
| grub_uint16_t rdev; |
| grub_uint16_t mtime[2]; |
| grub_uint16_t namesize[1]; |
| grub_uint16_t filesize[2]; |
| } __attribute__ ((packed)); |
| #else |
| /* tar support */ |
| #define MAGIC "ustar" |
| struct head |
| { |
| char name[100]; |
| char mode[8]; |
| char uid[8]; |
| char gid[8]; |
| char size[12]; |
| char mtime[12]; |
| char chksum[8]; |
| char typeflag; |
| char linkname[100]; |
| char magic[6]; |
| char version[2]; |
| char uname[32]; |
| char gname[32]; |
| char devmajor[8]; |
| char devminor[8]; |
| char prefix[155]; |
| } __attribute__ ((packed)); |
| #endif |
| |
| struct grub_cpio_data |
| { |
| grub_disk_t disk; |
| grub_off_t hofs; |
| grub_off_t dofs; |
| grub_off_t size; |
| #ifdef MODE_USTAR |
| char *linkname; |
| grub_size_t linkname_alloc; |
| #endif |
| }; |
| |
| static grub_dl_t my_mod; |
| |
| static inline void |
| canonicalize (char *name) |
| { |
| char *iptr, *optr; |
| for (iptr = name, optr = name; *iptr; ) |
| { |
| while (*iptr == '/') |
| iptr++; |
| if (iptr[0] == '.' && (iptr[1] == '/' || iptr[1] == 0)) |
| { |
| iptr += 2; |
| continue; |
| } |
| if (iptr[0] == '.' && iptr[1] == '.' && (iptr[2] == '/' || iptr[2] == 0)) |
| { |
| iptr += 3; |
| if (optr == name) |
| continue; |
| for (optr -= 2; optr >= name && *optr != '/'; optr--); |
| optr++; |
| continue; |
| } |
| while (*iptr && *iptr != '/') |
| *optr++ = *iptr++; |
| if (*iptr) |
| *optr++ = *iptr++; |
| } |
| *optr = 0; |
| } |
| |
| #if defined (MODE_ODC) || defined (MODE_USTAR) |
| static inline unsigned long long |
| read_number (const char *str, grub_size_t size) |
| { |
| unsigned long long ret = 0; |
| while (size-- && *str >= '0' && *str <= '7') |
| ret = (ret << 3) | (*str++ & 0xf); |
| return ret; |
| } |
| #elif defined (MODE_NEWC) |
| static inline unsigned long long |
| read_number (const char *str, grub_size_t size) |
| { |
| unsigned long long ret = 0; |
| while (size-- && grub_isxdigit (*str)) |
| { |
| char dig = *str++; |
| if (dig >= '0' && dig <= '9') |
| dig &= 0xf; |
| else if (dig >= 'a' && dig <= 'f') |
| dig -= 'a' - 10; |
| else |
| dig -= 'A' - 10; |
| ret = (ret << 4) | (dig); |
| } |
| return ret; |
| } |
| #else |
| static inline unsigned long long |
| read_number (const grub_uint16_t *arr, grub_size_t size) |
| { |
| long long ret = 0; |
| #ifdef MODE_BIGENDIAN |
| while (size--) |
| ret = (ret << 16) | grub_be_to_cpu16 (*arr++); |
| #else |
| while (size--) |
| ret = (ret << 16) | grub_le_to_cpu16 (*arr++); |
| #endif |
| return ret; |
| } |
| #endif |
| |
| static grub_err_t |
| grub_cpio_find_file (struct grub_cpio_data *data, char **name, |
| grub_int32_t *mtime, grub_disk_addr_t *ofs, |
| grub_uint32_t *mode) |
| { |
| #ifndef MODE_USTAR |
| struct head hd; |
| grub_size_t namesize; |
| grub_uint32_t modeval; |
| |
| if (grub_disk_read (data->disk, 0, data->hofs, sizeof (hd), &hd)) |
| return grub_errno; |
| |
| if (grub_memcmp (hd.magic, MAGIC, sizeof (hd.magic)) != 0 |
| #ifdef MAGIC2 |
| && grub_memcmp (hd.magic, MAGIC2, sizeof (hd.magic)) != 0 |
| #endif |
| ) |
| return grub_error (GRUB_ERR_BAD_FS, "invalid cpio archive"); |
| data->size = read_number (hd.filesize, ARRAY_SIZE (hd.filesize)); |
| if (mtime) |
| *mtime = read_number (hd.mtime, ARRAY_SIZE (hd.mtime)); |
| modeval = read_number (hd.mode, ARRAY_SIZE (hd.mode)); |
| namesize = read_number (hd.namesize, ARRAY_SIZE (hd.namesize)); |
| |
| if (mode) |
| *mode = modeval; |
| |
| *name = grub_malloc (namesize + 1); |
| if (*name == NULL) |
| return grub_errno; |
| |
| if (grub_disk_read (data->disk, 0, data->hofs + sizeof (hd), |
| namesize, *name)) |
| { |
| grub_free (*name); |
| return grub_errno; |
| } |
| (*name)[namesize] = 0; |
| |
| if (data->size == 0 && modeval == 0 && namesize == 11 |
| && grub_memcmp(*name, "TRAILER!!!", 11) == 0) |
| { |
| *ofs = 0; |
| grub_free (*name); |
| return GRUB_ERR_NONE; |
| } |
| |
| canonicalize (*name); |
| |
| data->dofs = data->hofs + ALIGN_CPIO (sizeof (hd) + namesize); |
| *ofs = data->dofs + ALIGN_CPIO (data->size); |
| #else |
| struct head hd; |
| int reread = 0, have_longname = 0, have_longlink = 0; |
| |
| for (reread = 0; reread < 3; reread++) |
| { |
| if (grub_disk_read (data->disk, 0, data->hofs, sizeof (hd), &hd)) |
| return grub_errno; |
| |
| if (!hd.name[0] && !hd.prefix[0]) |
| { |
| *ofs = 0; |
| return GRUB_ERR_NONE; |
| } |
| |
| if (grub_memcmp (hd.magic, MAGIC, sizeof (MAGIC) - 1)) |
| return grub_error (GRUB_ERR_BAD_FS, "invalid tar archive"); |
| |
| if (hd.typeflag == 'L') |
| { |
| grub_err_t err; |
| grub_size_t namesize = read_number (hd.size, sizeof (hd.size)); |
| *name = grub_malloc (namesize + 1); |
| if (*name == NULL) |
| return grub_errno; |
| err = grub_disk_read (data->disk, 0, |
| data->hofs + GRUB_DISK_SECTOR_SIZE, namesize, |
| *name); |
| (*name)[namesize] = 0; |
| if (err) |
| return err; |
| data->hofs += GRUB_DISK_SECTOR_SIZE |
| + ((namesize + GRUB_DISK_SECTOR_SIZE - 1) & |
| ~(GRUB_DISK_SECTOR_SIZE - 1)); |
| have_longname = 1; |
| continue; |
| } |
| |
| if (hd.typeflag == 'K') |
| { |
| grub_err_t err; |
| grub_size_t linksize = read_number (hd.size, sizeof (hd.size)); |
| if (data->linkname_alloc < linksize + 1) |
| { |
| char *n; |
| n = grub_malloc (2 * (linksize + 1)); |
| if (!n) |
| return grub_errno; |
| grub_free (data->linkname); |
| data->linkname = n; |
| data->linkname_alloc = 2 * (linksize + 1); |
| } |
| |
| err = grub_disk_read (data->disk, 0, |
| data->hofs + GRUB_DISK_SECTOR_SIZE, linksize, |
| data->linkname); |
| if (err) |
| return err; |
| data->linkname[linksize] = 0; |
| data->hofs += GRUB_DISK_SECTOR_SIZE |
| + ((linksize + GRUB_DISK_SECTOR_SIZE - 1) & |
| ~(GRUB_DISK_SECTOR_SIZE - 1)); |
| have_longlink = 1; |
| continue; |
| } |
| |
| if (!have_longname) |
| { |
| grub_size_t extra_size = 0; |
| |
| while (extra_size < sizeof (hd.prefix) |
| && hd.prefix[extra_size]) |
| extra_size++; |
| *name = grub_malloc (sizeof (hd.name) + extra_size + 2); |
| if (*name == NULL) |
| return grub_errno; |
| if (hd.prefix[0]) |
| { |
| grub_memcpy (*name, hd.prefix, extra_size); |
| (*name)[extra_size++] = '/'; |
| } |
| grub_memcpy (*name + extra_size, hd.name, sizeof (hd.name)); |
| (*name)[extra_size + sizeof (hd.name)] = 0; |
| } |
| |
| data->size = read_number (hd.size, sizeof (hd.size)); |
| data->dofs = data->hofs + GRUB_DISK_SECTOR_SIZE; |
| *ofs = data->dofs + ((data->size + GRUB_DISK_SECTOR_SIZE - 1) & |
| ~(GRUB_DISK_SECTOR_SIZE - 1)); |
| if (mtime) |
| *mtime = read_number (hd.mtime, sizeof (hd.mtime)); |
| if (mode) |
| { |
| *mode = read_number (hd.mode, sizeof (hd.mode)); |
| switch (hd.typeflag) |
| { |
| /* Hardlink. */ |
| case '1': |
| /* Symlink. */ |
| case '2': |
| *mode |= ATTR_LNK; |
| break; |
| case '0': |
| *mode |= ATTR_FILE; |
| break; |
| case '5': |
| *mode |= ATTR_DIR; |
| break; |
| } |
| } |
| if (!have_longlink) |
| { |
| if (data->linkname_alloc < 101) |
| { |
| char *n; |
| n = grub_malloc (101); |
| if (!n) |
| return grub_errno; |
| grub_free (data->linkname); |
| data->linkname = n; |
| data->linkname_alloc = 101; |
| } |
| grub_memcpy (data->linkname, hd.linkname, sizeof (hd.linkname)); |
| data->linkname[100] = 0; |
| } |
| |
| canonicalize (*name); |
| return GRUB_ERR_NONE; |
| } |
| #endif |
| return GRUB_ERR_NONE; |
| } |
| |
| static struct grub_cpio_data * |
| grub_cpio_mount (grub_disk_t disk) |
| { |
| struct head hd; |
| struct grub_cpio_data *data; |
| |
| if (grub_disk_read (disk, 0, 0, sizeof (hd), &hd)) |
| goto fail; |
| |
| if (grub_memcmp (hd.magic, MAGIC, sizeof (MAGIC) - 1) |
| #ifdef MAGIC2 |
| && grub_memcmp (hd.magic, MAGIC2, sizeof (MAGIC2) - 1) |
| #endif |
| ) |
| goto fail; |
| |
| data = (struct grub_cpio_data *) grub_zalloc (sizeof (*data)); |
| if (!data) |
| goto fail; |
| |
| data->disk = disk; |
| |
| return data; |
| |
| fail: |
| grub_error (GRUB_ERR_BAD_FS, "not a " |
| #ifdef MODE_USTAR |
| "tar" |
| #else |
| "cpio" |
| #endif |
| " filesystem"); |
| return 0; |
| } |
| |
| static grub_err_t |
| handle_symlink (struct grub_cpio_data *data, |
| const char *fn, char **name, |
| grub_uint32_t mode, int *restart) |
| { |
| grub_size_t flen; |
| char *target; |
| #ifndef MODE_USTAR |
| grub_err_t err; |
| #endif |
| char *ptr; |
| char *lastslash; |
| grub_size_t prefixlen; |
| char *rest; |
| grub_size_t size; |
| |
| *restart = 0; |
| |
| if ((mode & ATTR_TYPE) != ATTR_LNK) |
| return GRUB_ERR_NONE; |
| flen = grub_strlen (fn); |
| if (grub_memcmp (*name, fn, flen) != 0 |
| || ((*name)[flen] != 0 && (*name)[flen] != '/')) |
| return GRUB_ERR_NONE; |
| rest = *name + flen; |
| lastslash = rest; |
| if (*rest) |
| rest++; |
| while (lastslash >= *name && *lastslash != '/') |
| lastslash--; |
| if (lastslash >= *name) |
| prefixlen = lastslash - *name; |
| else |
| prefixlen = 0; |
| |
| if (prefixlen) |
| prefixlen++; |
| |
| #ifdef MODE_USTAR |
| size = grub_strlen (data->linkname); |
| #else |
| size = data->size; |
| #endif |
| if (size == 0) |
| return GRUB_ERR_NONE; |
| target = grub_malloc (size + grub_strlen (*name) + 2); |
| if (!target) |
| return grub_errno; |
| |
| #ifdef MODE_USTAR |
| grub_memcpy (target + prefixlen, data->linkname, size); |
| #else |
| err = grub_disk_read (data->disk, 0, data->dofs, data->size, |
| target + prefixlen); |
| if (err) |
| return err; |
| #endif |
| if (target[prefixlen] == '/') |
| { |
| grub_memmove (target, target + prefixlen, size); |
| ptr = target + size; |
| ptr = grub_stpcpy (ptr, rest); |
| *ptr = 0; |
| grub_dprintf ("cpio", "symlink redirected %s to %s\n", |
| *name, target); |
| grub_free (*name); |
| |
| canonicalize (target); |
| *name = target; |
| *restart = 1; |
| return GRUB_ERR_NONE; |
| } |
| if (prefixlen) |
| { |
| grub_memcpy (target, *name, prefixlen); |
| target[prefixlen-1] = '/'; |
| } |
| ptr = target + prefixlen + size; |
| ptr = grub_stpcpy (ptr, rest); |
| *ptr = 0; |
| grub_dprintf ("cpio", "symlink redirected %s to %s\n", |
| *name, target); |
| grub_free (*name); |
| canonicalize (target); |
| *name = target; |
| *restart = 1; |
| return GRUB_ERR_NONE; |
| } |
| |
| static grub_err_t |
| grub_cpio_dir (grub_device_t device, const char *path_in, |
| int (*hook) (const char *filename, |
| const struct grub_dirhook_info *info)) |
| { |
| struct grub_cpio_data *data; |
| grub_disk_addr_t ofs; |
| char *prev, *name, *path, *ptr; |
| grub_size_t len; |
| int symlinknest = 0; |
| |
| path = grub_strdup (path_in + 1); |
| if (!path) |
| return grub_errno; |
| canonicalize (path); |
| for (ptr = path + grub_strlen (path) - 1; ptr >= path && *ptr == '/'; ptr--) |
| *ptr = 0; |
| |
| grub_dl_ref (my_mod); |
| |
| prev = 0; |
| |
| data = grub_cpio_mount (device->disk); |
| if (!data) |
| { |
| grub_free (path); |
| return grub_errno; |
| } |
| |
| len = grub_strlen (path); |
| data->hofs = 0; |
| while (1) |
| { |
| grub_int32_t mtime; |
| grub_uint32_t mode; |
| grub_err_t err; |
| |
| if (grub_cpio_find_file (data, &name, &mtime, &ofs, &mode)) |
| goto fail; |
| |
| if (!ofs) |
| break; |
| |
| if (grub_memcmp (path, name, len) == 0 |
| && (name[len] == 0 || name[len] == '/' || len == 0)) |
| { |
| char *p, *n; |
| |
| n = name + len; |
| while (*n == '/') |
| n++; |
| |
| p = grub_strchr (n, '/'); |
| if (p) |
| *p = 0; |
| |
| if (((!prev) || (grub_strcmp (prev, name) != 0)) && *n != 0) |
| { |
| struct grub_dirhook_info info; |
| grub_memset (&info, 0, sizeof (info)); |
| info.dir = (p != NULL) || ((mode & ATTR_TYPE) == ATTR_DIR); |
| info.mtime = mtime; |
| info.mtimeset = 1; |
| |
| if (hook (n, &info)) |
| { |
| grub_free (name); |
| goto fail; |
| } |
| grub_free (prev); |
| prev = name; |
| } |
| else |
| { |
| int restart = 0; |
| err = handle_symlink (data, name, &path, mode, &restart); |
| grub_free (name); |
| if (err) |
| goto fail; |
| if (restart) |
| { |
| len = grub_strlen (path); |
| if (++symlinknest == 8) |
| { |
| grub_error (GRUB_ERR_SYMLINK_LOOP, |
| N_("too deep nesting of symlinks")); |
| goto fail; |
| } |
| ofs = 0; |
| } |
| } |
| } |
| else |
| grub_free (name); |
| data->hofs = ofs; |
| } |
| |
| fail: |
| |
| grub_free (path); |
| grub_free (prev); |
| #ifdef MODE_USTAR |
| grub_free (data->linkname); |
| #endif |
| grub_free (data); |
| |
| grub_dl_unref (my_mod); |
| |
| return grub_errno; |
| } |
| |
| static grub_err_t |
| grub_cpio_open (grub_file_t file, const char *name_in) |
| { |
| struct grub_cpio_data *data; |
| grub_disk_addr_t ofs; |
| char *fn; |
| char *name = grub_strdup (name_in + 1); |
| int symlinknest = 0; |
| |
| if (!name) |
| return grub_errno; |
| |
| canonicalize (name); |
| |
| grub_dl_ref (my_mod); |
| |
| data = grub_cpio_mount (file->device->disk); |
| if (!data) |
| { |
| grub_free (name); |
| return grub_errno; |
| } |
| |
| data->hofs = 0; |
| while (1) |
| { |
| grub_uint32_t mode; |
| int restart; |
| |
| if (grub_cpio_find_file (data, &fn, NULL, &ofs, &mode)) |
| goto fail; |
| |
| if (!ofs) |
| { |
| grub_error (GRUB_ERR_FILE_NOT_FOUND, N_("file `%s' not found"), name_in); |
| break; |
| } |
| |
| if (handle_symlink (data, fn, &name, mode, &restart)) |
| { |
| grub_free (fn); |
| goto fail; |
| } |
| |
| if (restart) |
| { |
| ofs = 0; |
| if (++symlinknest == 8) |
| { |
| grub_error (GRUB_ERR_SYMLINK_LOOP, |
| N_("too deep nesting of symlinks")); |
| goto fail; |
| } |
| goto no_match; |
| } |
| |
| if (grub_strcmp (name, fn) != 0) |
| goto no_match; |
| |
| file->data = data; |
| file->size = data->size; |
| grub_free (fn); |
| grub_free (name); |
| |
| return GRUB_ERR_NONE; |
| |
| no_match: |
| |
| grub_free (fn); |
| data->hofs = ofs; |
| } |
| |
| fail: |
| #ifdef MODE_USTAR |
| grub_free (data->linkname); |
| #endif |
| grub_free (data); |
| grub_free (name); |
| |
| grub_dl_unref (my_mod); |
| |
| return grub_errno; |
| } |
| |
| static grub_ssize_t |
| grub_cpio_read (grub_file_t file, char *buf, grub_size_t len) |
| { |
| struct grub_cpio_data *data; |
| |
| data = file->data; |
| return (grub_disk_read (data->disk, 0, data->dofs + file->offset, |
| len, buf)) ? -1 : (grub_ssize_t) len; |
| } |
| |
| static grub_err_t |
| grub_cpio_close (grub_file_t file) |
| { |
| struct grub_cpio_data *data; |
| |
| data = file->data; |
| #ifdef MODE_USTAR |
| grub_free (data->linkname); |
| #endif |
| grub_free (data); |
| |
| grub_dl_unref (my_mod); |
| |
| return grub_errno; |
| } |
| |
| static struct grub_fs grub_cpio_fs = { |
| #ifdef MODE_USTAR |
| .name = "tarfs", |
| #elif defined (MODE_ODC) |
| .name = "odc", |
| #elif defined (MODE_NEWC) |
| .name = "newc", |
| #else |
| .name = "cpiofs", |
| #endif |
| .dir = grub_cpio_dir, |
| .open = grub_cpio_open, |
| .read = grub_cpio_read, |
| .close = grub_cpio_close, |
| #ifdef GRUB_UTIL |
| .reserved_first_sector = 0, |
| .blocklist_install = 0, |
| #endif |
| }; |
| |
| #ifdef MODE_USTAR |
| GRUB_MOD_INIT (tar) |
| #elif defined (MODE_ODC) |
| GRUB_MOD_INIT (odc) |
| #elif defined (MODE_NEWC) |
| GRUB_MOD_INIT (newc) |
| #elif defined (MODE_BIGENDIAN) |
| GRUB_MOD_INIT (cpio_be) |
| #else |
| GRUB_MOD_INIT (cpio) |
| #endif |
| { |
| grub_fs_register (&grub_cpio_fs); |
| my_mod = mod; |
| } |
| |
| #ifdef MODE_USTAR |
| GRUB_MOD_FINI (tar) |
| #elif defined (MODE_ODC) |
| GRUB_MOD_FINI (odc) |
| #elif defined (MODE_NEWC) |
| GRUB_MOD_FINI (newc) |
| #elif defined (MODE_BIGENDIAN) |
| GRUB_MOD_FINI (cpio_be) |
| #else |
| GRUB_MOD_FINI (cpio) |
| #endif |
| { |
| grub_fs_unregister (&grub_cpio_fs); |
| } |