treewide: fix replace nbd@openwrt.org with nbd@nbd.name
[openwrt.org/openwrt.git] / target / linux / generic / files / drivers / mtd / mtdsplit / mtdsplit_tplink.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
/*
 *  Copyright (C) 2013 Gabor Juhos <juhosg@openwrt.org>
 *  Copyright (C) 2014 Felix Fietkau <nbd@nbd.name>
 *
 *  This program is free software; you can redistribute it and/or modify it
 *  under the terms of the GNU General Public License version 2 as published
 *  by the Free Software Foundation.
 *
 */
 
#include <linux/module.h>
#include <linux/init.h>
#include <linux/kernel.h>
#include <linux/slab.h>
#include <linux/mtd/mtd.h>
#include <linux/mtd/partitions.h>
#include <linux/byteorder/generic.h>
 
#include "mtdsplit.h"
 
#define TPLINK_NR_PARTS         2
#define TPLINK_MIN_ROOTFS_OFFS  0x80000 /* 512KiB */
 
#define MD5SUM_LEN  16
 
struct fw_v1 {
        char            vendor_name[24];
        char            fw_version[36];
        uint32_t        hw_id;          /* hardware id */
        uint32_t        hw_rev;         /* hardware revision */
        uint32_t        unk1;
        uint8_t         md5sum1[MD5SUM_LEN];
        uint32_t        unk2;
        uint8_t         md5sum2[MD5SUM_LEN];
        uint32_t        unk3;
        uint32_t        kernel_la;      /* kernel load address */
        uint32_t        kernel_ep;      /* kernel entry point */
        uint32_t        fw_length;      /* total length of the firmware */
        uint32_t        kernel_ofs;     /* kernel data offset */
        uint32_t        kernel_len;     /* kernel data length */
        uint32_t        rootfs_ofs;     /* rootfs data offset */
        uint32_t        rootfs_len;     /* rootfs data length */
        uint32_t        boot_ofs;       /* bootloader data offset */
        uint32_t        boot_len;       /* bootloader data length */
        uint8_t         pad[360];
} __attribute__ ((packed));
 
struct fw_v2 {
        char            fw_version[48]; /* 0x04: fw version string */
        uint32_t        hw_id;          /* 0x34: hardware id */
        uint32_t        hw_rev;         /* 0x38: FIXME: hardware revision? */
        uint32_t        unk1;           /* 0x3c: 0x00000000 */
        uint8_t         md5sum1[MD5SUM_LEN]; /* 0x40 */
        uint32_t        unk2;           /* 0x50: 0x00000000 */
        uint8_t         md5sum2[MD5SUM_LEN]; /* 0x54 */
        uint32_t        unk3;           /* 0x64: 0xffffffff */
 
        uint32_t        kernel_la;      /* 0x68: kernel load address */
        uint32_t        kernel_ep;      /* 0x6c: kernel entry point */
        uint32_t        fw_length;      /* 0x70: total length of the image */
        uint32_t        kernel_ofs;     /* 0x74: kernel data offset */
        uint32_t        kernel_len;     /* 0x78: kernel data length */
        uint32_t        rootfs_ofs;     /* 0x7c: rootfs data offset */
        uint32_t        rootfs_len;     /* 0x80: rootfs data length */
        uint32_t        boot_ofs;       /* 0x84: FIXME: seems to be unused */
        uint32_t        boot_len;       /* 0x88: FIXME: seems to be unused */
        uint16_t        unk4;           /* 0x8c: 0x55aa */
        uint8_t         sver_hi;        /* 0x8e */
        uint8_t         sver_lo;        /* 0x8f */
        uint8_t         unk5;           /* 0x90: magic: 0xa5 */
        uint8_t         ver_hi;         /* 0x91 */
        uint8_t         ver_mid;        /* 0x92 */
        uint8_t         ver_lo;         /* 0x93 */
        uint8_t         pad[364];
} __attribute__ ((packed));
 
struct tplink_fw_header {
        uint32_t version;
        union {
                struct fw_v1 v1;
                struct fw_v2 v2;
        };
};
 
static int mtdsplit_parse_tplink(struct mtd_info *master,
                                struct mtd_partition **pparts,
                                struct mtd_part_parser_data *data)
{
        struct tplink_fw_header hdr;
        size_t hdr_len, retlen, kernel_size;
        size_t rootfs_offset;
        struct mtd_partition *parts;
        int err;
 
        hdr_len = sizeof(hdr);
        err = mtd_read(master, 0, hdr_len, &retlen, (void *) &hdr);
        if (err)
                return err;
 
        if (retlen != hdr_len)
                return -EIO;
 
        switch (le32_to_cpu(hdr.version)) {
        case 1:
                if (be32_to_cpu(hdr.v1.kernel_ofs) != sizeof(hdr))
                        return -EINVAL;
 
                kernel_size = sizeof(hdr) + be32_to_cpu(hdr.v1.kernel_len);
                break;
        case 2:
        case 3:
                if (be32_to_cpu(hdr.v2.kernel_ofs) != sizeof(hdr))
                        return -EINVAL;
 
                kernel_size = sizeof(hdr) + be32_to_cpu(hdr.v2.kernel_len);
                break;
        default:
                return -EINVAL;
        }
 
        if (kernel_size > master->size)
                return -EINVAL;
 
        /* Find the rootfs after the kernel. */
        err = mtd_check_rootfs_magic(master, kernel_size, NULL);
        if (!err) {
                rootfs_offset = kernel_size;
        } else {
                /*
                 * The size in the header might cover the rootfs as well.
                 * Start the search from an arbitrary offset.
                 */
                err = mtd_find_rootfs_from(master, TPLINK_MIN_ROOTFS_OFFS,
                                           master->size, &rootfs_offset, NULL);
                if (err)
                        return err;
        }
 
        parts = kzalloc(TPLINK_NR_PARTS * sizeof(*parts), GFP_KERNEL);
        if (!parts)
                return -ENOMEM;
 
        parts[0].name = KERNEL_PART_NAME;
        parts[0].offset = 0;
        parts[0].size = rootfs_offset;
 
        parts[1].name = ROOTFS_PART_NAME;
        parts[1].offset = rootfs_offset;
        parts[1].size = master->size - rootfs_offset;
 
        *pparts = parts;
        return TPLINK_NR_PARTS;
}
 
static struct mtd_part_parser mtdsplit_tplink_parser = {
        .owner = THIS_MODULE,
        .name = "tplink-fw",
        .parse_fn = mtdsplit_parse_tplink,
        .type = MTD_PARSER_TYPE_FIRMWARE,
};
 
static int __init mtdsplit_tplink_init(void)
{
        register_mtd_parser(&mtdsplit_tplink_parser);
 
        return 0;
}
 
subsys_initcall(mtdsplit_tplink_init);
 
comments