diff options
| author | Andrey V. Elsukov <ae@FreeBSD.org> | 2013-08-08 11:24:25 +0000 |
|---|---|---|
| committer | Andrey V. Elsukov <ae@FreeBSD.org> | 2013-08-08 11:24:25 +0000 |
| commit | 4371b649aad3cee78651ff9d6a8e286669b8bb89 (patch) | |
| tree | 336a7f4f1529b232ed07e58d765d76e4470699e3 /sys/boot | |
| parent | c08f11b07c1e4a1cc30cade438adbdd4fefe5da2 (diff) | |
Notes
Diffstat (limited to 'sys/boot')
| -rw-r--r-- | sys/boot/common/part.c | 15 |
1 files changed, 9 insertions, 6 deletions
diff --git a/sys/boot/common/part.c b/sys/boot/common/part.c index 2b3a16fbd851..d37eee52ca77 100644 --- a/sys/boot/common/part.c +++ b/sys/boot/common/part.c @@ -181,7 +181,7 @@ gpt_checkhdr(struct gpt_hdr *hdr, uint64_t lba_self, uint64_t lba_last, } hdr->hdr_entries = le32toh(hdr->hdr_entries); hdr->hdr_entsz = le32toh(hdr->hdr_entsz); - if (hdr->hdr_entries < 128 || + if (hdr->hdr_entries == 0 || hdr->hdr_entsz < sizeof(struct gpt_ent) || sectorsize % hdr->hdr_entsz != 0) { DEBUG("invalid entry size or number of entries"); @@ -203,11 +203,14 @@ gpt_checktbl(const struct gpt_hdr *hdr, u_char *tbl, size_t size, int i, cnt; cnt = size / hdr->hdr_entsz; - /* Check CRC only when buffer size is enough for table. */ - if (hdr->hdr_entries <= cnt && - crc32(tbl, size) != hdr->hdr_crc_table) { - DEBUG("GPT table's CRC doesn't match"); - return (-1); + if (hdr->hdr_entries <= cnt) { + cnt = hdr->hdr_entries; + /* Check CRC only when buffer size is enough for table. */ + if (hdr->hdr_crc_table != + crc32(tbl, hdr->hdr_entries * hdr->hdr_entsz)) { + DEBUG("GPT table's CRC doesn't match"); + return (-1); + } } ent = (struct gpt_ent *)tbl; for (i = 0; i < cnt; i++, ent++) { |
