mirror of https://github.com/Qortal/Brooklyn
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
129 lines
2.7 KiB
129 lines
2.7 KiB
// SPDX-License-Identifier: GPL-2.0-only |
|
/* |
|
* Parser for TRX format partitions |
|
* |
|
* Copyright (C) 2012 - 2017 Rafał Miłecki <[email protected]> |
|
*/ |
|
|
|
#include <linux/module.h> |
|
#include <linux/slab.h> |
|
#include <linux/mtd/mtd.h> |
|
#include <linux/mtd/partitions.h> |
|
|
|
#define TRX_PARSER_MAX_PARTS 4 |
|
|
|
/* Magics */ |
|
#define TRX_MAGIC 0x30524448 |
|
#define UBI_EC_MAGIC 0x23494255 /* UBI# */ |
|
|
|
struct trx_header { |
|
uint32_t magic; |
|
uint32_t length; |
|
uint32_t crc32; |
|
uint16_t flags; |
|
uint16_t version; |
|
uint32_t offset[3]; |
|
} __packed; |
|
|
|
static const char *parser_trx_data_part_name(struct mtd_info *master, |
|
size_t offset) |
|
{ |
|
uint32_t buf; |
|
size_t bytes_read; |
|
int err; |
|
|
|
err = mtd_read(master, offset, sizeof(buf), &bytes_read, |
|
(uint8_t *)&buf); |
|
if (err && !mtd_is_bitflip(err)) { |
|
pr_err("mtd_read error while parsing (offset: 0x%zX): %d\n", |
|
offset, err); |
|
goto out_default; |
|
} |
|
|
|
if (buf == UBI_EC_MAGIC) |
|
return "ubi"; |
|
|
|
out_default: |
|
return "rootfs"; |
|
} |
|
|
|
static int parser_trx_parse(struct mtd_info *mtd, |
|
const struct mtd_partition **pparts, |
|
struct mtd_part_parser_data *data) |
|
{ |
|
struct mtd_partition *parts; |
|
struct mtd_partition *part; |
|
struct trx_header trx; |
|
size_t bytes_read; |
|
uint8_t curr_part = 0, i = 0; |
|
int err; |
|
|
|
parts = kcalloc(TRX_PARSER_MAX_PARTS, sizeof(struct mtd_partition), |
|
GFP_KERNEL); |
|
if (!parts) |
|
return -ENOMEM; |
|
|
|
err = mtd_read(mtd, 0, sizeof(trx), &bytes_read, (uint8_t *)&trx); |
|
if (err) { |
|
pr_err("MTD reading error: %d\n", err); |
|
kfree(parts); |
|
return err; |
|
} |
|
|
|
if (trx.magic != TRX_MAGIC) { |
|
kfree(parts); |
|
return -ENOENT; |
|
} |
|
|
|
/* We have LZMA loader if there is address in offset[2] */ |
|
if (trx.offset[2]) { |
|
part = &parts[curr_part++]; |
|
part->name = "loader"; |
|
part->offset = trx.offset[i]; |
|
i++; |
|
} |
|
|
|
if (trx.offset[i]) { |
|
part = &parts[curr_part++]; |
|
part->name = "linux"; |
|
part->offset = trx.offset[i]; |
|
i++; |
|
} |
|
|
|
if (trx.offset[i]) { |
|
part = &parts[curr_part++]; |
|
part->name = parser_trx_data_part_name(mtd, trx.offset[i]); |
|
part->offset = trx.offset[i]; |
|
i++; |
|
} |
|
|
|
/* |
|
* Assume that every partition ends at the beginning of the one it is |
|
* followed by. |
|
*/ |
|
for (i = 0; i < curr_part; i++) { |
|
u64 next_part_offset = (i < curr_part - 1) ? |
|
parts[i + 1].offset : mtd->size; |
|
|
|
parts[i].size = next_part_offset - parts[i].offset; |
|
} |
|
|
|
*pparts = parts; |
|
return i; |
|
}; |
|
|
|
static const struct of_device_id mtd_parser_trx_of_match_table[] = { |
|
{ .compatible = "brcm,trx" }, |
|
{}, |
|
}; |
|
MODULE_DEVICE_TABLE(of, mtd_parser_trx_of_match_table); |
|
|
|
static struct mtd_part_parser mtd_parser_trx = { |
|
.parse_fn = parser_trx_parse, |
|
.name = "trx", |
|
.of_match_table = mtd_parser_trx_of_match_table, |
|
}; |
|
module_mtd_part_parser(mtd_parser_trx); |
|
|
|
MODULE_LICENSE("GPL v2"); |
|
MODULE_DESCRIPTION("Parser for TRX format partitions");
|
|
|