Root/target/linux/ar7/files/drivers/mtd/titanpart.c

1#include <linux/kernel.h>
2#include <linux/slab.h>
3
4#include <linux/mtd/mtd.h>
5#include <linux/mtd/partitions.h>
6#include <linux/bootmem.h>
7#include <linux/magic.h>
8#include <asm/mach-ar7/prom.h>
9
10#define IMAGE_A_SIZE 0X3c0000
11#define WRTP_PARTS 14
12#define NSP_IMG_MAGIC_NUMBER le32_to_cpu(0x4D544443)
13#define NSP_IMG_SECTION_TYPE_KERNEL (0x01)
14#define NSP_IMG_SECTION_TYPE_FILESYSTEM_ROOT (0x02)
15#define NSP_IMG_SECTION_TYPE_FILESYSTEM (0x03)
16#define MAX_NUM_PARTITIONS 14
17
18static int part_count=0;
19static struct mtd_partition titan_parts[WRTP_PARTS];
20
21
22struct nsp_img_hdr_head
23{
24    unsigned int magic; /* Magic number to identify this image header */
25    unsigned int boot_offset; /* Offset from start of header to kernel code. */
26    unsigned int flags; /* Image flags. */
27    unsigned int hdr_version; /* Version of this header. */
28    unsigned int hdr_size; /* The complete size of all portions of the header */
29    unsigned int prod_id; /* This product id */
30    unsigned int rel_id; /* Which release this is */
31    unsigned int version; /* name-MMM.nnn.ooo-rxx => 0xMMnnooxx. See comment
32                       below */
33    unsigned int image_size; /* Image size (including header) */
34    unsigned int info_offset; /* Offset from start of header to info block */
35    unsigned int sect_info_offset; /* Offset from start of header to section desc */
36    unsigned int chksum_offset; /* Offset from start of header to chksum block */
37    unsigned int pad1;
38};
39
40struct nsp_img_hdr_section_info
41{
42    unsigned int num_sects; /* Number of section (and section desc blocks) in this image */
43    unsigned int sect_size; /* Size of a SINGLE section_desc block */
44    unsigned int sections_offset; /* Offset to from start of header to the start of the section blocks */
45};
46
47/* There will be one of more of the following stuctures in the image header. Each
48   section will have one of these blocks. */
49struct nsp_img_hdr_sections
50{
51    unsigned int offset; /* Offset of section from start of NSP_IMG_HDR_HEAD */
52    unsigned int total_size; /* Size of section (including pad size.) */
53    unsigned int raw_size; /* Size of section only */
54    unsigned int flags; /* Section flags */
55    unsigned int chksum; /* Section checksum */
56    unsigned int type; /* Section type. What kind of info does this section describe */
57    char name[16]; /* Reference name for this section. */
58};
59
60
61
62
63
64static int titan_parse_env_address(char *env_name, unsigned int *flash_base,
65                unsigned int *flash_end)
66{
67    char image_name[30];
68    char *env_ptr;
69    char *base_ptr;
70    char *end_ptr;
71    char * string_ptr;
72    /* Get the image variable */
73    env_ptr = prom_getenv(env_name);
74    if(!env_ptr){
75        printk("titan: invalid env name, %s.\n", env_name);
76        return -1; /* Error, no image variable */
77    }
78    strncpy(image_name, env_ptr, 30);
79    image_name[29]=0;
80    string_ptr = image_name;
81    /* Extract the start and stop addresses of the partition */
82    base_ptr = strsep(&string_ptr, ",");
83    end_ptr = strsep(&string_ptr, ",");
84    if ((base_ptr == NULL) || (end_ptr == NULL)) {
85        printk("titan: Couldn't tokenize %s start,end.\n", image_name);
86        return -1;
87    }
88
89    *flash_base = (unsigned int) simple_strtol(base_ptr, NULL, 0);
90    *flash_end = (unsigned int) simple_strtol(end_ptr, NULL, 0);
91    if((!*flash_base) || (!*flash_end)) {
92        printk("titan: Unable to convert :%s: :%s: into start,end values.\n",
93                env_name, image_name);
94        return -1;
95    }
96    *flash_base &= 0x0fffffff;
97    *flash_end &= 0x0fffffff;
98    return 0;
99}
100
101
102
103static int titan_get_single_image(char *bootcfg_name, unsigned int *flash_base,
104                unsigned int *flash_end)
105{
106    char *env_ptr;
107    char *base_ptr;
108    char *end_ptr;
109    char image_name[30];
110    char * string_ptr;
111
112    if(!bootcfg_name || !flash_base || !flash_end)
113        return -1;
114
115    env_ptr = prom_getenv(bootcfg_name);
116    if(!env_ptr){
117        printk("titan: %s variable not found.\n", bootcfg_name);
118        return -1; /* Error, no bootcfg variable */
119    }
120
121    string_ptr = image_name;
122    /* Save off the image name */
123    strncpy(image_name, env_ptr, 30);
124    image_name[29]=0;
125
126    end_ptr=strsep(&string_ptr, "\"");
127    base_ptr=strsep(&string_ptr, "\""); /* Loose the last " */
128    if(!end_ptr || !base_ptr){
129        printk("titan: invalid bootcfg format, %s.\n", image_name);
130        return -1; /* Error, invalid bootcfg variable */
131    }
132
133    /* Now, parse the addresses */
134    return titan_parse_env_address(base_ptr, flash_base, flash_end);
135}
136
137
138
139static void titan_add_partition(char * env_name, unsigned int flash_base, unsigned int flash_end)
140{
141        titan_parts[part_count].name = env_name;
142        titan_parts[part_count].offset = flash_base;
143        titan_parts[part_count].size = flash_end-flash_base;
144        titan_parts[part_count].mask_flags = (strcmp(env_name, "bootloader")==0||
145            strcmp(env_name, "boot_env")==0 ||
146            strcmp(env_name, "full_image")==0 )?MTD_WRITEABLE:0;
147        part_count++;
148
149}
150int create_titan_partitions(struct mtd_info *master,
151    struct mtd_partition **pparts,
152    unsigned long origin)
153{
154    struct nsp_img_hdr_head hdr;
155    struct nsp_img_hdr_section_info sect_info;
156    struct nsp_img_hdr_sections section;
157    unsigned int flash_base, flash_end;
158    unsigned int start, end;
159    char *name;
160    int i;
161    int total_sects=0;
162    size_t len;
163
164    /* Get the bootcfg env variable first */
165    if(titan_get_single_image("BOOTCFG", &flash_base, &flash_end)) {
166        /* Error, fallback */
167        return -1;
168    }
169
170    /* Get access to the header, and do some validation checks */
171    //hdr=(struct nsp_img_hdr_head*)flash_base;
172    master->read(master, flash_base, sizeof(struct nsp_img_hdr_head), &len, (uint8_t *)&hdr);
173    if(hdr.magic != NSP_IMG_MAGIC_NUMBER)
174        return -1; /* Not a single image */
175
176    master->read(master, flash_base + hdr.sect_info_offset, sizeof(struct nsp_img_hdr_section_info), &len, (uint8_t *)&sect_info);
177
178    /* Look for the root fs, and add it first. This way we KNOW where the rootfs is */
179    for(i=0; i< sect_info.num_sects && i<MAX_NUM_PARTITIONS; i++){
180        master->read(master, flash_base + sect_info.sections_offset + (i * sect_info.sect_size) , sizeof(struct nsp_img_hdr_sections), &len, (uint8_t *)&section);
181        /* Add only the root partition */
182        if(section.type != NSP_IMG_SECTION_TYPE_FILESYSTEM_ROOT){
183            continue;
184        }
185        start=flash_base + section.offset;
186        end=start + section.total_size;
187        titan_add_partition("root", start, end);
188        total_sects++;
189        
190    }
191
192    for(i=0; i< sect_info.num_sects && i<MAX_NUM_PARTITIONS; i++){
193
194        master->read(master, flash_base + sect_info.sections_offset + (i * sect_info.sect_size) , sizeof(struct nsp_img_hdr_sections), &len, (uint8_t *)&section);
195
196        name=section.name;
197        if(section.type == NSP_IMG_SECTION_TYPE_FILESYSTEM_ROOT)
198        {
199            name = "rootfs";
200            start=flash_base + section.offset;
201            end=flash_end;
202            titan_add_partition(name, start, end);
203            total_sects++;
204        }
205        else if(section.type == NSP_IMG_SECTION_TYPE_KERNEL)
206        {
207            name = "kernel";
208            start=flash_base + section.offset;
209            end=start + section.total_size;
210            titan_add_partition(name, start, end);
211            total_sects++;
212        }
213
214    }
215
216    /* Next, lets add the single image */
217    titan_add_partition("primary_image", flash_base, flash_end);
218    total_sects++;
219
220
221    titan_add_partition("full_image", 0, master->size);
222    total_sects++;
223
224    if (!titan_parse_env_address("BOOTLOADER", &start, &end)){
225        titan_add_partition("bootloader", start, end);
226        total_sects++;
227    }
228    if (!titan_parse_env_address("boot_env", &start, &end)){
229        titan_add_partition("boot_env", start, end);
230        total_sects++;
231    }
232    *pparts = titan_parts;
233    return total_sects;
234}
235

Archive Download this file

Branches:
backfire
history
master
release_2010-11-17
release_2010-12-14
release_2011-02-23
release_2011-05-28
release_2011-08-27
release_2011-11-13
release_2012-03-18
release_2012-04-09
release_2012-10-11
uclibc_nptl



interactive