fastmap.c revision fe24c6e5f51ec5ea42d432f67618a084047a23ed
1dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger/* 2dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * Copyright (c) 2012 Linutronix GmbH 3dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * Author: Richard Weinberger <richard@nod.at> 4dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * 5dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * This program is free software; you can redistribute it and/or modify 6dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * it under the terms of the GNU General Public License as published by 7dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * the Free Software Foundation; version 2. 8dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * 9dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * This program is distributed in the hope that it will be useful, 10dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * but WITHOUT ANY WARRANTY; without even the implied warranty of 11dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See 12dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * the GNU General Public License for more details. 13dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * 14dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger */ 15dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 16dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger#include <linux/crc32.h> 17dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger#include "ubi.h" 18dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 19dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger/** 20dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * ubi_calc_fm_size - calculates the fastmap size in bytes for an UBI device. 21dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @ubi: UBI device description object 22dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger */ 23dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinbergersize_t ubi_calc_fm_size(struct ubi_device *ubi) 24dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger{ 25dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger size_t size; 26dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 27dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger size = sizeof(struct ubi_fm_hdr) + \ 28dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger sizeof(struct ubi_fm_scan_pool) + \ 29dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger sizeof(struct ubi_fm_scan_pool) + \ 30dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger (ubi->peb_count * sizeof(struct ubi_fm_ec)) + \ 31dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger (sizeof(struct ubi_fm_eba) + \ 32dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger (ubi->peb_count * sizeof(__be32))) + \ 33dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger sizeof(struct ubi_fm_volhdr) * UBI_MAX_VOLUMES; 34dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return roundup(size, ubi->leb_size); 35dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger} 36dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 37dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 38dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger/** 39dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * new_fm_vhdr - allocate a new volume header for fastmap usage. 40dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @ubi: UBI device description object 41dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @vol_id: the VID of the new header 42dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * 43dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * Returns a new struct ubi_vid_hdr on success. 44dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * NULL indicates out of memory. 45dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger */ 46dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinbergerstatic struct ubi_vid_hdr *new_fm_vhdr(struct ubi_device *ubi, int vol_id) 47dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger{ 48dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_vid_hdr *new; 49dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 50dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger new = ubi_zalloc_vid_hdr(ubi, GFP_KERNEL); 51dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (!new) 52dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto out; 53dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 54dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger new->vol_type = UBI_VID_DYNAMIC; 55dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger new->vol_id = cpu_to_be32(vol_id); 56dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 57dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger /* UBI implementations without fastmap support have to delete the 58dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * fastmap. 59dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger */ 60dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger new->compat = UBI_COMPAT_DELETE; 61dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 62dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinbergerout: 63dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return new; 64dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger} 65dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 66dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger/** 67dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * add_aeb - create and add a attach erase block to a given list. 68dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @ai: UBI attach info object 69dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @list: the target list 70dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @pnum: PEB number of the new attach erase block 71dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @ec: erease counter of the new LEB 72dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @scrub: scrub this PEB after attaching 73dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * 74dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * Returns 0 on success, < 0 indicates an internal error. 75dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger */ 76dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinbergerstatic int add_aeb(struct ubi_attach_info *ai, struct list_head *list, 77dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger int pnum, int ec, int scrub) 78dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger{ 79dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_ainf_peb *aeb; 80dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 81dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger aeb = kmem_cache_alloc(ai->aeb_slab_cache, GFP_KERNEL); 82dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (!aeb) 83dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return -ENOMEM; 84dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 85dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger aeb->pnum = pnum; 86dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger aeb->ec = ec; 87dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger aeb->lnum = -1; 88dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger aeb->scrub = scrub; 89dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger aeb->copy_flag = aeb->sqnum = 0; 90dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 91dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ai->ec_sum += aeb->ec; 92dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ai->ec_count++; 93dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 94dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ai->max_ec < aeb->ec) 95dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ai->max_ec = aeb->ec; 96dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 97dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ai->min_ec > aeb->ec) 98dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ai->min_ec = aeb->ec; 99dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 100dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger list_add_tail(&aeb->u.list, list); 101dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 102dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return 0; 103dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger} 104dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 105dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger/** 106dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * add_vol - create and add a new volume to ubi_attach_info. 107dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @ai: ubi_attach_info object 108dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @vol_id: VID of the new volume 109dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @used_ebs: number of used EBS 110dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @data_pad: data padding value of the new volume 111dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @vol_type: volume type 112dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @last_eb_bytes: number of bytes in the last LEB 113dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * 114dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * Returns the new struct ubi_ainf_volume on success. 115dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * NULL indicates an error. 116dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger */ 117dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinbergerstatic struct ubi_ainf_volume *add_vol(struct ubi_attach_info *ai, int vol_id, 118dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger int used_ebs, int data_pad, u8 vol_type, 119dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger int last_eb_bytes) 120dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger{ 121dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_ainf_volume *av; 122dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct rb_node **p = &ai->volumes.rb_node, *parent = NULL; 123dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 124dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger while (*p) { 125dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger parent = *p; 126dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger av = rb_entry(parent, struct ubi_ainf_volume, rb); 127dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 128dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (vol_id > av->vol_id) 129dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger p = &(*p)->rb_left; 130dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger else if (vol_id > av->vol_id) 131dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger p = &(*p)->rb_right; 132dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 133dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 134dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger av = kmalloc(sizeof(struct ubi_ainf_volume), GFP_KERNEL); 135dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (!av) 136dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto out; 137dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 138dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger av->highest_lnum = av->leb_count = 0; 139dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger av->vol_id = vol_id; 140dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger av->used_ebs = used_ebs; 141dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger av->data_pad = data_pad; 142dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger av->last_data_size = last_eb_bytes; 143dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger av->compat = 0; 144dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger av->vol_type = vol_type; 145dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger av->root = RB_ROOT; 146dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 147dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger dbg_bld("found volume (ID %i)", vol_id); 148dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 149dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger rb_link_node(&av->rb, parent, p); 150dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger rb_insert_color(&av->rb, &ai->volumes); 151dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 152dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinbergerout: 153dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return av; 154dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger} 155dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 156dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger/** 157dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * assign_aeb_to_av - assigns a SEB to a given ainf_volume and removes it 158dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * from it's original list. 159dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @ai: ubi_attach_info object 160dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @aeb: the to be assigned SEB 161dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @av: target scan volume 162dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger */ 163dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinbergerstatic void assign_aeb_to_av(struct ubi_attach_info *ai, 164dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_ainf_peb *aeb, 165dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_ainf_volume *av) 166dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger{ 167dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_ainf_peb *tmp_aeb; 168dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct rb_node **p = &ai->volumes.rb_node, *parent = NULL; 169dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 170dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger p = &av->root.rb_node; 171dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger while (*p) { 172dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger parent = *p; 173dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 174dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger tmp_aeb = rb_entry(parent, struct ubi_ainf_peb, u.rb); 175dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (aeb->lnum != tmp_aeb->lnum) { 176dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (aeb->lnum < tmp_aeb->lnum) 177dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger p = &(*p)->rb_left; 178dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger else 179dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger p = &(*p)->rb_right; 180dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 181dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger continue; 182dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } else 183dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger break; 184dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 185dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 186dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger list_del(&aeb->u.list); 187dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger av->leb_count++; 188dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 189dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger rb_link_node(&aeb->u.rb, parent, p); 190dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger rb_insert_color(&aeb->u.rb, &av->root); 191dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger} 192dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 193dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger/** 194dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * update_vol - inserts or updates a LEB which was found a pool. 195dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @ubi: the UBI device object 196dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @ai: attach info object 197dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @av: the volume this LEB belongs to 198dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @new_vh: the volume header derived from new_aeb 199dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @new_aeb: the AEB to be examined 200dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * 201dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * Returns 0 on success, < 0 indicates an internal error. 202dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger */ 203dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinbergerstatic int update_vol(struct ubi_device *ubi, struct ubi_attach_info *ai, 204dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_ainf_volume *av, struct ubi_vid_hdr *new_vh, 205dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_ainf_peb *new_aeb) 206dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger{ 207dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct rb_node **p = &av->root.rb_node, *parent = NULL; 208dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_ainf_peb *aeb, *victim; 209dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger int cmp_res; 210dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 211dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger while (*p) { 212dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger parent = *p; 213dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger aeb = rb_entry(parent, struct ubi_ainf_peb, u.rb); 214dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 215dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (be32_to_cpu(new_vh->lnum) != aeb->lnum) { 216dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (be32_to_cpu(new_vh->lnum) < aeb->lnum) 217dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger p = &(*p)->rb_left; 218dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger else 219dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger p = &(*p)->rb_right; 220dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 221dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger continue; 222dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 223dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 224dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger /* This case can happen if the fastmap gets written 225dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * because of a volume change (creation, deletion, ..). 226dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * Then a PEB can be within the persistent EBA and the pool. 227dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger */ 228dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (aeb->pnum == new_aeb->pnum) { 229dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_assert(aeb->lnum == new_aeb->lnum); 230dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger kmem_cache_free(ai->aeb_slab_cache, new_aeb); 231dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 232dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return 0; 233dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 234dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 235dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger cmp_res = ubi_compare_lebs(ubi, aeb, new_aeb->pnum, new_vh); 236dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (cmp_res < 0) 237dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return cmp_res; 238dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 239dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger /* new_aeb is newer */ 240dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (cmp_res & 1) { 241dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger victim = kmem_cache_alloc(ai->aeb_slab_cache, 242dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger GFP_KERNEL); 243dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (!victim) 244dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return -ENOMEM; 245dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 246dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger victim->ec = aeb->ec; 247dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger victim->pnum = aeb->pnum; 248dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger list_add_tail(&victim->u.list, &ai->erase); 249dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 250dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (av->highest_lnum == be32_to_cpu(new_vh->lnum)) 251dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger av->last_data_size = \ 252dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger be32_to_cpu(new_vh->data_size); 253dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 254dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger dbg_bld("vol %i: AEB %i's PEB %i is the newer", 255dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger av->vol_id, aeb->lnum, new_aeb->pnum); 256dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 257dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger aeb->ec = new_aeb->ec; 258dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger aeb->pnum = new_aeb->pnum; 259dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger aeb->copy_flag = new_vh->copy_flag; 260dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger aeb->scrub = new_aeb->scrub; 261dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger kmem_cache_free(ai->aeb_slab_cache, new_aeb); 262dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 263dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger /* new_aeb is older */ 264dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } else { 265dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger dbg_bld("vol %i: AEB %i's PEB %i is old, dropping it", 266dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger av->vol_id, aeb->lnum, new_aeb->pnum); 267dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger list_add_tail(&new_aeb->u.list, &ai->erase); 268dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 269dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 270dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return 0; 271dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 272dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger /* This LEB is new, let's add it to the volume */ 273dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 274dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (av->highest_lnum <= be32_to_cpu(new_vh->lnum)) { 275dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger av->highest_lnum = be32_to_cpu(new_vh->lnum); 276dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger av->last_data_size = be32_to_cpu(new_vh->data_size); 277dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 278dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 279dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (av->vol_type == UBI_STATIC_VOLUME) 280dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger av->used_ebs = be32_to_cpu(new_vh->used_ebs); 281dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 282dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger av->leb_count++; 283dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 284dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger rb_link_node(&new_aeb->u.rb, parent, p); 285dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger rb_insert_color(&new_aeb->u.rb, &av->root); 286dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 287dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return 0; 288dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger} 289dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 290dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger/** 291dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * process_pool_aeb - we found a non-empty PEB in a pool. 292dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @ubi: UBI device object 293dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @ai: attach info object 294dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @new_vh: the volume header derived from new_aeb 295dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @new_aeb: the AEB to be examined 296dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * 297dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * Returns 0 on success, < 0 indicates an internal error. 298dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger */ 299dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinbergerstatic int process_pool_aeb(struct ubi_device *ubi, struct ubi_attach_info *ai, 300dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_vid_hdr *new_vh, 301dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_ainf_peb *new_aeb) 302dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger{ 303dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_ainf_volume *av, *tmp_av = NULL; 304dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct rb_node **p = &ai->volumes.rb_node, *parent = NULL; 305dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger int found = 0; 306dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 307dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (be32_to_cpu(new_vh->vol_id) == UBI_FM_SB_VOLUME_ID || 308dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger be32_to_cpu(new_vh->vol_id) == UBI_FM_DATA_VOLUME_ID) { 309dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger kmem_cache_free(ai->aeb_slab_cache, new_aeb); 310dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 311dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return 0; 312dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 313dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 314dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger /* Find the volume this SEB belongs to */ 315dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger while (*p) { 316dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger parent = *p; 317dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger tmp_av = rb_entry(parent, struct ubi_ainf_volume, rb); 318dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 319dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (be32_to_cpu(new_vh->vol_id) > tmp_av->vol_id) 320dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger p = &(*p)->rb_left; 321dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger else if (be32_to_cpu(new_vh->vol_id) < tmp_av->vol_id) 322dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger p = &(*p)->rb_right; 323dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger else { 324dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger found = 1; 325dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger break; 326dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 327dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 328dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 329dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (found) 330dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger av = tmp_av; 331dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger else { 332dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("orphaned volume in fastmap pool!"); 333dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return UBI_BAD_FASTMAP; 334dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 335dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 336dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_assert(be32_to_cpu(new_vh->vol_id) == av->vol_id); 337dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 338dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return update_vol(ubi, ai, av, new_vh, new_aeb); 339dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger} 340dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 341dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger/** 342dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * unmap_peb - unmap a PEB. 343dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * If fastmap detects a free PEB in the pool it has to check whether 344dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * this PEB has been unmapped after writing the fastmap. 345dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * 346dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @ai: UBI attach info object 347dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @pnum: The PEB to be unmapped 348dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger */ 349dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinbergerstatic void unmap_peb(struct ubi_attach_info *ai, int pnum) 350dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger{ 351dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_ainf_volume *av; 352dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct rb_node *node, *node2; 353dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_ainf_peb *aeb; 354dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 355dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger for (node = rb_first(&ai->volumes); node; node = rb_next(node)) { 356dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger av = rb_entry(node, struct ubi_ainf_volume, rb); 357dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 358dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger for (node2 = rb_first(&av->root); node2; 359dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger node2 = rb_next(node2)) { 360dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger aeb = rb_entry(node2, struct ubi_ainf_peb, u.rb); 361dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (aeb->pnum == pnum) { 362dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger rb_erase(&aeb->u.rb, &av->root); 363dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger kmem_cache_free(ai->aeb_slab_cache, aeb); 364dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return; 365dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 366dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 367dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 368dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger} 369dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 370dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger/** 371dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * scan_pool - scans a pool for changed (no longer empty PEBs). 372dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @ubi: UBI device object 373dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @ai: attach info object 374dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @pebs: an array of all PEB numbers in the to be scanned pool 375dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @pool_size: size of the pool (number of entries in @pebs) 376dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @max_sqnum: pointer to the maximal sequence number 377dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @eba_orphans: list of PEBs which need to be scanned 378dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @free: list of PEBs which are most likely free (and go into @ai->free) 379dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * 380dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * Returns 0 on success, if the pool is unusable UBI_BAD_FASTMAP is returned. 381dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * < 0 indicates an internal error. 382dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger */ 383dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinbergerstatic int scan_pool(struct ubi_device *ubi, struct ubi_attach_info *ai, 384dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger int *pebs, int pool_size, unsigned long long *max_sqnum, 385dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct list_head *eba_orphans, struct list_head *free) 386dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger{ 387dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_vid_hdr *vh; 388dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_ec_hdr *ech; 389dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_ainf_peb *new_aeb, *tmp_aeb; 390dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger int i, pnum, err, found_orphan, ret = 0; 391dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 392dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ech = kzalloc(ubi->ec_hdr_alsize, GFP_KERNEL); 393dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (!ech) 394dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return -ENOMEM; 395dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 396dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger vh = ubi_zalloc_vid_hdr(ubi, GFP_KERNEL); 397dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (!vh) { 398dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger kfree(ech); 399dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return -ENOMEM; 400dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 401dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 402dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger dbg_bld("scanning fastmap pool: size = %i", pool_size); 403dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 404dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger /* 405dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * Now scan all PEBs in the pool to find changes which have been made 406dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * after the creation of the fastmap 407dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger */ 408dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger for (i = 0; i < pool_size; i++) { 409dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger int scrub = 0; 410c22301ad4fa0f4cf71e9c877d072e6f07a0bf682Richard Genoud int image_seq; 411dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 412dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger pnum = be32_to_cpu(pebs[i]); 413dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 414dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ubi_io_is_bad(ubi, pnum)) { 415dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("bad PEB in fastmap pool!"); 416dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = UBI_BAD_FASTMAP; 417dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto out; 418dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 419dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 420dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger err = ubi_io_read_ec_hdr(ubi, pnum, ech, 0); 421dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (err && err != UBI_IO_BITFLIPS) { 422dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("unable to read EC header! PEB:%i err:%i", 423dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger pnum, err); 424dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = err > 0 ? UBI_BAD_FASTMAP : err; 425dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto out; 426dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } else if (ret == UBI_IO_BITFLIPS) 427dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger scrub = 1; 428dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 429c22301ad4fa0f4cf71e9c877d072e6f07a0bf682Richard Genoud /* 430c22301ad4fa0f4cf71e9c877d072e6f07a0bf682Richard Genoud * Older UBI implementations have image_seq set to zero, so 431c22301ad4fa0f4cf71e9c877d072e6f07a0bf682Richard Genoud * we shouldn't fail if image_seq == 0. 432c22301ad4fa0f4cf71e9c877d072e6f07a0bf682Richard Genoud */ 433c22301ad4fa0f4cf71e9c877d072e6f07a0bf682Richard Genoud image_seq = be32_to_cpu(ech->image_seq); 434c22301ad4fa0f4cf71e9c877d072e6f07a0bf682Richard Genoud 435c22301ad4fa0f4cf71e9c877d072e6f07a0bf682Richard Genoud if (image_seq && (image_seq != ubi->image_seq)) { 436dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("bad image seq: 0x%x, expected: 0x%x", 437dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger be32_to_cpu(ech->image_seq), ubi->image_seq); 438f240dca8f2fdf11d52a7bacd867e274c3b21a7d3Richard Weinberger ret = UBI_BAD_FASTMAP; 439dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto out; 440dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 441dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 442dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger err = ubi_io_read_vid_hdr(ubi, pnum, vh, 0); 443dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (err == UBI_IO_FF || err == UBI_IO_FF_BITFLIPS) { 444dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger unsigned long long ec = be64_to_cpu(ech->ec); 445dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger unmap_peb(ai, pnum); 446dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger dbg_bld("Adding PEB to free: %i", pnum); 447dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (err == UBI_IO_FF_BITFLIPS) 448dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger add_aeb(ai, free, pnum, ec, 1); 449dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger else 450dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger add_aeb(ai, free, pnum, ec, 0); 451dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger continue; 452dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } else if (err == 0 || err == UBI_IO_BITFLIPS) { 453dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger dbg_bld("Found non empty PEB:%i in pool", pnum); 454dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 455dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (err == UBI_IO_BITFLIPS) 456dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger scrub = 1; 457dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 458dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger found_orphan = 0; 459dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger list_for_each_entry(tmp_aeb, eba_orphans, u.list) { 460dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (tmp_aeb->pnum == pnum) { 461dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger found_orphan = 1; 462dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger break; 463dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 464dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 465dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (found_orphan) { 466dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger kmem_cache_free(ai->aeb_slab_cache, tmp_aeb); 467dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger list_del(&tmp_aeb->u.list); 468dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 469dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 470dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger new_aeb = kmem_cache_alloc(ai->aeb_slab_cache, 471dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger GFP_KERNEL); 472dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (!new_aeb) { 473dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = -ENOMEM; 474dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto out; 475dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 476dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 477dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger new_aeb->ec = be64_to_cpu(ech->ec); 478dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger new_aeb->pnum = pnum; 479dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger new_aeb->lnum = be32_to_cpu(vh->lnum); 480dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger new_aeb->sqnum = be64_to_cpu(vh->sqnum); 481dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger new_aeb->copy_flag = vh->copy_flag; 482dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger new_aeb->scrub = scrub; 483dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 484dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (*max_sqnum < new_aeb->sqnum) 485dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger *max_sqnum = new_aeb->sqnum; 486dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 487dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger err = process_pool_aeb(ubi, ai, vh, new_aeb); 488dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (err) { 489dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = err > 0 ? UBI_BAD_FASTMAP : err; 490dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto out; 491dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 492dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } else { 493dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger /* We are paranoid and fall back to scanning mode */ 494dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("fastmap pool PEBs contains damaged PEBs!"); 495dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = err > 0 ? UBI_BAD_FASTMAP : err; 496dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto out; 497dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 498dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 499dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 500dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 501dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinbergerout: 502dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_free_vid_hdr(ubi, vh); 503dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger kfree(ech); 504dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return ret; 505dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger} 506dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 507dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger/** 508dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * count_fastmap_pebs - Counts the PEBs found by fastmap. 509dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @ai: The UBI attach info object 510dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger */ 511dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinbergerstatic int count_fastmap_pebs(struct ubi_attach_info *ai) 512dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger{ 513dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_ainf_peb *aeb; 514dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_ainf_volume *av; 515dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct rb_node *rb1, *rb2; 516dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger int n = 0; 517dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 518dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger list_for_each_entry(aeb, &ai->erase, u.list) 519dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger n++; 520dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 521dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger list_for_each_entry(aeb, &ai->free, u.list) 522dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger n++; 523dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 524dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_rb_for_each_entry(rb1, av, &ai->volumes, rb) 525dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_rb_for_each_entry(rb2, aeb, &av->root, u.rb) 526dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger n++; 527dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 528dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return n; 529dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger} 530dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 531dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger/** 532dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * ubi_attach_fastmap - creates ubi_attach_info from a fastmap. 533dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @ubi: UBI device object 534dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @ai: UBI attach info object 535dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @fm: the fastmap to be attached 536dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * 537dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * Returns 0 on success, UBI_BAD_FASTMAP if the found fastmap was unusable. 538dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * < 0 indicates an internal error. 539dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger */ 540dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinbergerstatic int ubi_attach_fastmap(struct ubi_device *ubi, 541dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_attach_info *ai, 542dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_fastmap_layout *fm) 543dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger{ 544dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct list_head used, eba_orphans, free; 545dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_ainf_volume *av; 546dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_ainf_peb *aeb, *tmp_aeb, *_tmp_aeb; 547dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_ec_hdr *ech; 548dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_fm_sb *fmsb; 549dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_fm_hdr *fmhdr; 550dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_fm_scan_pool *fmpl1, *fmpl2; 551dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_fm_ec *fmec; 552dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_fm_volhdr *fmvhdr; 553dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_fm_eba *fm_eba; 554dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger int ret, i, j, pool_size, wl_pool_size; 555dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger size_t fm_pos = 0, fm_size = ubi->fm_size; 556dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger unsigned long long max_sqnum = 0; 557dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger void *fm_raw = ubi->fm_buf; 558dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 559dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger INIT_LIST_HEAD(&used); 560dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger INIT_LIST_HEAD(&free); 561dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger INIT_LIST_HEAD(&eba_orphans); 562dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger INIT_LIST_HEAD(&ai->corr); 563dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger INIT_LIST_HEAD(&ai->free); 564dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger INIT_LIST_HEAD(&ai->erase); 565dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger INIT_LIST_HEAD(&ai->alien); 566dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ai->volumes = RB_ROOT; 567dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ai->min_ec = UBI_MAX_ERASECOUNTER; 568dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 569dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ai->aeb_slab_cache = kmem_cache_create("ubi_ainf_peb_slab", 570dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger sizeof(struct ubi_ainf_peb), 571dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 0, 0, NULL); 572dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (!ai->aeb_slab_cache) { 573dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = -ENOMEM; 574dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto fail; 575dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 576dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 577dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmsb = (struct ubi_fm_sb *)(fm_raw); 578dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ai->max_sqnum = fmsb->sqnum; 579dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm_pos += sizeof(struct ubi_fm_sb); 580dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (fm_pos >= fm_size) 581dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto fail_bad; 582dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 583dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmhdr = (struct ubi_fm_hdr *)(fm_raw + fm_pos); 584dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm_pos += sizeof(*fmhdr); 585dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (fm_pos >= fm_size) 586dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto fail_bad; 587dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 588dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (be32_to_cpu(fmhdr->magic) != UBI_FM_HDR_MAGIC) { 589dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("bad fastmap header magic: 0x%x, expected: 0x%x", 590dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger be32_to_cpu(fmhdr->magic), UBI_FM_HDR_MAGIC); 591dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto fail_bad; 592dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 593dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 594dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmpl1 = (struct ubi_fm_scan_pool *)(fm_raw + fm_pos); 595dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm_pos += sizeof(*fmpl1); 596dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (fm_pos >= fm_size) 597dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto fail_bad; 598dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (be32_to_cpu(fmpl1->magic) != UBI_FM_POOL_MAGIC) { 599dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("bad fastmap pool magic: 0x%x, expected: 0x%x", 600dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger be32_to_cpu(fmpl1->magic), UBI_FM_POOL_MAGIC); 601dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto fail_bad; 602dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 603dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 604dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmpl2 = (struct ubi_fm_scan_pool *)(fm_raw + fm_pos); 605dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm_pos += sizeof(*fmpl2); 606dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (fm_pos >= fm_size) 607dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto fail_bad; 608dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (be32_to_cpu(fmpl2->magic) != UBI_FM_POOL_MAGIC) { 609dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("bad fastmap pool magic: 0x%x, expected: 0x%x", 610dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger be32_to_cpu(fmpl2->magic), UBI_FM_POOL_MAGIC); 611dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto fail_bad; 612dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 613dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 614dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger pool_size = be16_to_cpu(fmpl1->size); 615dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger wl_pool_size = be16_to_cpu(fmpl2->size); 616dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm->max_pool_size = be16_to_cpu(fmpl1->max_size); 617dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm->max_wl_pool_size = be16_to_cpu(fmpl2->max_size); 618dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 619dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (pool_size > UBI_FM_MAX_POOL_SIZE || pool_size < 0) { 620dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("bad pool size: %i", pool_size); 621dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto fail_bad; 622dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 623dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 624dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (wl_pool_size > UBI_FM_MAX_POOL_SIZE || wl_pool_size < 0) { 625dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("bad WL pool size: %i", wl_pool_size); 626dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto fail_bad; 627dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 628dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 629dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 630dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (fm->max_pool_size > UBI_FM_MAX_POOL_SIZE || 631dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm->max_pool_size < 0) { 632dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("bad maximal pool size: %i", fm->max_pool_size); 633dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto fail_bad; 634dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 635dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 636dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (fm->max_wl_pool_size > UBI_FM_MAX_POOL_SIZE || 637dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm->max_wl_pool_size < 0) { 638dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("bad maximal WL pool size: %i", fm->max_wl_pool_size); 639dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto fail_bad; 640dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 641dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 642dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger /* read EC values from free list */ 643dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger for (i = 0; i < be32_to_cpu(fmhdr->free_peb_count); i++) { 644dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmec = (struct ubi_fm_ec *)(fm_raw + fm_pos); 645dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm_pos += sizeof(*fmec); 646dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (fm_pos >= fm_size) 647dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto fail_bad; 648dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 649dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger add_aeb(ai, &ai->free, be32_to_cpu(fmec->pnum), 650dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger be32_to_cpu(fmec->ec), 0); 651dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 652dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 653dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger /* read EC values from used list */ 654dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger for (i = 0; i < be32_to_cpu(fmhdr->used_peb_count); i++) { 655dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmec = (struct ubi_fm_ec *)(fm_raw + fm_pos); 656dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm_pos += sizeof(*fmec); 657dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (fm_pos >= fm_size) 658dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto fail_bad; 659dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 660dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger add_aeb(ai, &used, be32_to_cpu(fmec->pnum), 661dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger be32_to_cpu(fmec->ec), 0); 662dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 663dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 664dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger /* read EC values from scrub list */ 665dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger for (i = 0; i < be32_to_cpu(fmhdr->scrub_peb_count); i++) { 666dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmec = (struct ubi_fm_ec *)(fm_raw + fm_pos); 667dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm_pos += sizeof(*fmec); 668dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (fm_pos >= fm_size) 669dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto fail_bad; 670dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 671dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger add_aeb(ai, &used, be32_to_cpu(fmec->pnum), 672dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger be32_to_cpu(fmec->ec), 1); 673dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 674dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 675dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger /* read EC values from erase list */ 676dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger for (i = 0; i < be32_to_cpu(fmhdr->erase_peb_count); i++) { 677dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmec = (struct ubi_fm_ec *)(fm_raw + fm_pos); 678dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm_pos += sizeof(*fmec); 679dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (fm_pos >= fm_size) 680dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto fail_bad; 681dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 682dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger add_aeb(ai, &ai->erase, be32_to_cpu(fmec->pnum), 683dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger be32_to_cpu(fmec->ec), 1); 684dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 685dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 686dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ai->mean_ec = div_u64(ai->ec_sum, ai->ec_count); 687dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ai->bad_peb_count = be32_to_cpu(fmhdr->bad_peb_count); 688dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 689dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger /* Iterate over all volumes and read their EBA table */ 690dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger for (i = 0; i < be32_to_cpu(fmhdr->vol_count); i++) { 691dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmvhdr = (struct ubi_fm_volhdr *)(fm_raw + fm_pos); 692dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm_pos += sizeof(*fmvhdr); 693dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (fm_pos >= fm_size) 694dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto fail_bad; 695dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 696dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (be32_to_cpu(fmvhdr->magic) != UBI_FM_VHDR_MAGIC) { 697dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("bad fastmap vol header magic: 0x%x, " \ 698dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger "expected: 0x%x", 699dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger be32_to_cpu(fmvhdr->magic), UBI_FM_VHDR_MAGIC); 700dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto fail_bad; 701dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 702dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 703dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger av = add_vol(ai, be32_to_cpu(fmvhdr->vol_id), 704dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger be32_to_cpu(fmvhdr->used_ebs), 705dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger be32_to_cpu(fmvhdr->data_pad), 706dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmvhdr->vol_type, 707dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger be32_to_cpu(fmvhdr->last_eb_bytes)); 708dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 709dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (!av) 710dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto fail_bad; 711dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 712dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ai->vols_found++; 713dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ai->highest_vol_id < be32_to_cpu(fmvhdr->vol_id)) 714dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ai->highest_vol_id = be32_to_cpu(fmvhdr->vol_id); 715dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 716dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm_eba = (struct ubi_fm_eba *)(fm_raw + fm_pos); 717dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm_pos += sizeof(*fm_eba); 718dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm_pos += (sizeof(__be32) * be32_to_cpu(fm_eba->reserved_pebs)); 719dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (fm_pos >= fm_size) 720dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto fail_bad; 721dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 722dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (be32_to_cpu(fm_eba->magic) != UBI_FM_EBA_MAGIC) { 723dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("bad fastmap EBA header magic: 0x%x, " \ 724dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger "expected: 0x%x", 725dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger be32_to_cpu(fm_eba->magic), UBI_FM_EBA_MAGIC); 726dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto fail_bad; 727dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 728dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 729dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger for (j = 0; j < be32_to_cpu(fm_eba->reserved_pebs); j++) { 730dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger int pnum = be32_to_cpu(fm_eba->pnum[j]); 731dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 732dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if ((int)be32_to_cpu(fm_eba->pnum[j]) < 0) 733dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger continue; 734dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 735dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger aeb = NULL; 736dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger list_for_each_entry(tmp_aeb, &used, u.list) { 737584d4623357db8f79029db76ec2b4953418018f0Brian Pomerantz if (tmp_aeb->pnum == pnum) { 738dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger aeb = tmp_aeb; 739584d4623357db8f79029db76ec2b4953418018f0Brian Pomerantz break; 740584d4623357db8f79029db76ec2b4953418018f0Brian Pomerantz } 741dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 742dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 743dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger /* This can happen if a PEB is already in an EBA known 744dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * by this fastmap but the PEB itself is not in the used 745dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * list. 746dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * In this case the PEB can be within the fastmap pool 747dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * or while writing the fastmap it was in the protection 748dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * queue. 749dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger */ 750dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (!aeb) { 751dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger aeb = kmem_cache_alloc(ai->aeb_slab_cache, 752dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger GFP_KERNEL); 753dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (!aeb) { 754dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = -ENOMEM; 755dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 756dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto fail; 757dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 758dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 759dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger aeb->lnum = j; 760dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger aeb->pnum = be32_to_cpu(fm_eba->pnum[j]); 761dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger aeb->ec = -1; 762dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger aeb->scrub = aeb->copy_flag = aeb->sqnum = 0; 763dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger list_add_tail(&aeb->u.list, &eba_orphans); 764dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger continue; 765dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 766dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 767dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger aeb->lnum = j; 768dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 769dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (av->highest_lnum <= aeb->lnum) 770dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger av->highest_lnum = aeb->lnum; 771dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 772dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger assign_aeb_to_av(ai, aeb, av); 773dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 774dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger dbg_bld("inserting PEB:%i (LEB %i) to vol %i", 775dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger aeb->pnum, aeb->lnum, av->vol_id); 776dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 777dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 778dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ech = kzalloc(ubi->ec_hdr_alsize, GFP_KERNEL); 779dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (!ech) { 780dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = -ENOMEM; 781dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto fail; 782dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 783dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 784dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger list_for_each_entry_safe(tmp_aeb, _tmp_aeb, &eba_orphans, 785dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger u.list) { 786dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger int err; 787dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 788dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ubi_io_is_bad(ubi, tmp_aeb->pnum)) { 789dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("bad PEB in fastmap EBA orphan list"); 790dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = UBI_BAD_FASTMAP; 791dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger kfree(ech); 792dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto fail; 793dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 794dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 795dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger err = ubi_io_read_ec_hdr(ubi, tmp_aeb->pnum, ech, 0); 796dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (err && err != UBI_IO_BITFLIPS) { 797dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("unable to read EC header! PEB:%i " \ 798dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger "err:%i", tmp_aeb->pnum, err); 799dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = err > 0 ? UBI_BAD_FASTMAP : err; 800dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger kfree(ech); 801dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 802dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto fail; 803dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } else if (err == UBI_IO_BITFLIPS) 804dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger tmp_aeb->scrub = 1; 805dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 806dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger tmp_aeb->ec = be64_to_cpu(ech->ec); 807dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger assign_aeb_to_av(ai, tmp_aeb, av); 808dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 809dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 810dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger kfree(ech); 811dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 812dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 813dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = scan_pool(ubi, ai, fmpl1->pebs, pool_size, &max_sqnum, 814dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger &eba_orphans, &free); 815dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ret) 816dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto fail; 817dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 818dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = scan_pool(ubi, ai, fmpl2->pebs, wl_pool_size, &max_sqnum, 819dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger &eba_orphans, &free); 820dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ret) 821dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto fail; 822dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 823dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (max_sqnum > ai->max_sqnum) 824dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ai->max_sqnum = max_sqnum; 825dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 8266a059abdbae7b6b35ddc69f6336f7afc1ce79bacWei Yongjun list_for_each_entry_safe(tmp_aeb, _tmp_aeb, &free, u.list) 8276a059abdbae7b6b35ddc69f6336f7afc1ce79bacWei Yongjun list_move_tail(&tmp_aeb->u.list, &ai->free); 828dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 829dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger /* 830dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * If fastmap is leaking PEBs (must not happen), raise a 831dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * fat warning and fall back to scanning mode. 832dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * We do this here because in ubi_wl_init() it's too late 833dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * and we cannot fall back to scanning. 834dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger */ 835dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (WARN_ON(count_fastmap_pebs(ai) != ubi->peb_count - 836dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ai->bad_peb_count - fm->used_blocks)) 837dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto fail_bad; 838dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 839dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return 0; 840dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 841dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinbergerfail_bad: 842dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = UBI_BAD_FASTMAP; 843dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinbergerfail: 844fe24c6e5f51ec5ea42d432f67618a084047a23edRichard Weinberger list_for_each_entry_safe(tmp_aeb, _tmp_aeb, &used, u.list) { 845fe24c6e5f51ec5ea42d432f67618a084047a23edRichard Weinberger kmem_cache_free(ai->aeb_slab_cache, tmp_aeb); 846fe24c6e5f51ec5ea42d432f67618a084047a23edRichard Weinberger list_del(&tmp_aeb->u.list); 847fe24c6e5f51ec5ea42d432f67618a084047a23edRichard Weinberger } 848fe24c6e5f51ec5ea42d432f67618a084047a23edRichard Weinberger list_for_each_entry_safe(tmp_aeb, _tmp_aeb, &eba_orphans, u.list) { 849fe24c6e5f51ec5ea42d432f67618a084047a23edRichard Weinberger kmem_cache_free(ai->aeb_slab_cache, tmp_aeb); 850fe24c6e5f51ec5ea42d432f67618a084047a23edRichard Weinberger list_del(&tmp_aeb->u.list); 851fe24c6e5f51ec5ea42d432f67618a084047a23edRichard Weinberger } 852fe24c6e5f51ec5ea42d432f67618a084047a23edRichard Weinberger list_for_each_entry_safe(tmp_aeb, _tmp_aeb, &free, u.list) { 853fe24c6e5f51ec5ea42d432f67618a084047a23edRichard Weinberger kmem_cache_free(ai->aeb_slab_cache, tmp_aeb); 854fe24c6e5f51ec5ea42d432f67618a084047a23edRichard Weinberger list_del(&tmp_aeb->u.list); 855fe24c6e5f51ec5ea42d432f67618a084047a23edRichard Weinberger } 856fe24c6e5f51ec5ea42d432f67618a084047a23edRichard Weinberger 857dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return ret; 858dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger} 859dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 860dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger/** 861dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * ubi_scan_fastmap - scan the fastmap. 862dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @ubi: UBI device object 863dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @ai: UBI attach info to be filled 864dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @fm_anchor: The fastmap starts at this PEB 865dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * 866dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * Returns 0 on success, UBI_NO_FASTMAP if no fastmap was found, 867dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * UBI_BAD_FASTMAP if one was found but is not usable. 868dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * < 0 indicates an internal error. 869dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger */ 870dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinbergerint ubi_scan_fastmap(struct ubi_device *ubi, struct ubi_attach_info *ai, 871dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger int fm_anchor) 872dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger{ 873dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_fm_sb *fmsb, *fmsb2; 874dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_vid_hdr *vh; 875dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_ec_hdr *ech; 876dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_fastmap_layout *fm; 877dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger int i, used_blocks, pnum, ret = 0; 878dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger size_t fm_size; 879dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger __be32 crc, tmp_crc; 880dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger unsigned long long sqnum = 0; 881dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 882dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger mutex_lock(&ubi->fm_mutex); 883dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger memset(ubi->fm_buf, 0, ubi->fm_size); 884dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 885dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmsb = kmalloc(sizeof(*fmsb), GFP_KERNEL); 886dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (!fmsb) { 887dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = -ENOMEM; 888dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto out; 889dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 890dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 891dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm = kzalloc(sizeof(*fm), GFP_KERNEL); 892dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (!fm) { 893dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = -ENOMEM; 894dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger kfree(fmsb); 895dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto out; 896dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 897dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 898dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = ubi_io_read(ubi, fmsb, fm_anchor, ubi->leb_start, sizeof(*fmsb)); 899dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ret && ret != UBI_IO_BITFLIPS) 900dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto free_fm_sb; 901dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger else if (ret == UBI_IO_BITFLIPS) 902dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm->to_be_tortured[0] = 1; 903dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 904dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (be32_to_cpu(fmsb->magic) != UBI_FM_SB_MAGIC) { 905dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("bad super block magic: 0x%x, expected: 0x%x", 906dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger be32_to_cpu(fmsb->magic), UBI_FM_SB_MAGIC); 907dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = UBI_BAD_FASTMAP; 908dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto free_fm_sb; 909dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 910dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 911dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (fmsb->version != UBI_FM_FMT_VERSION) { 912dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("bad fastmap version: %i, expected: %i", 913dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmsb->version, UBI_FM_FMT_VERSION); 914dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = UBI_BAD_FASTMAP; 915dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto free_fm_sb; 916dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 917dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 918dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger used_blocks = be32_to_cpu(fmsb->used_blocks); 919dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (used_blocks > UBI_FM_MAX_BLOCKS || used_blocks < 1) { 920dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("number of fastmap blocks is invalid: %i", used_blocks); 921dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = UBI_BAD_FASTMAP; 922dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto free_fm_sb; 923dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 924dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 925dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm_size = ubi->leb_size * used_blocks; 926dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (fm_size != ubi->fm_size) { 927dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("bad fastmap size: %zi, expected: %zi", fm_size, 928dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi->fm_size); 929dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = UBI_BAD_FASTMAP; 930dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto free_fm_sb; 931dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 932dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 933dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ech = kzalloc(ubi->ec_hdr_alsize, GFP_KERNEL); 934dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (!ech) { 935dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = -ENOMEM; 936dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto free_fm_sb; 937dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 938dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 939dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger vh = ubi_zalloc_vid_hdr(ubi, GFP_KERNEL); 940dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (!vh) { 941dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = -ENOMEM; 942dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto free_hdr; 943dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 944dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 945dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger for (i = 0; i < used_blocks; i++) { 946c22301ad4fa0f4cf71e9c877d072e6f07a0bf682Richard Genoud int image_seq; 947c22301ad4fa0f4cf71e9c877d072e6f07a0bf682Richard Genoud 948dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger pnum = be32_to_cpu(fmsb->block_loc[i]); 949dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 950dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ubi_io_is_bad(ubi, pnum)) { 951dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = UBI_BAD_FASTMAP; 952dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto free_hdr; 953dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 954dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 955dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = ubi_io_read_ec_hdr(ubi, pnum, ech, 0); 956dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ret && ret != UBI_IO_BITFLIPS) { 957dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("unable to read fastmap block# %i EC (PEB: %i)", 958dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger i, pnum); 959dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ret > 0) 960dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = UBI_BAD_FASTMAP; 961dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto free_hdr; 962dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } else if (ret == UBI_IO_BITFLIPS) 963dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm->to_be_tortured[i] = 1; 964dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 965c22301ad4fa0f4cf71e9c877d072e6f07a0bf682Richard Genoud image_seq = be32_to_cpu(ech->image_seq); 966dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (!ubi->image_seq) 967c22301ad4fa0f4cf71e9c877d072e6f07a0bf682Richard Genoud ubi->image_seq = image_seq; 968dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 969c22301ad4fa0f4cf71e9c877d072e6f07a0bf682Richard Genoud /* 970c22301ad4fa0f4cf71e9c877d072e6f07a0bf682Richard Genoud * Older UBI implementations have image_seq set to zero, so 971c22301ad4fa0f4cf71e9c877d072e6f07a0bf682Richard Genoud * we shouldn't fail if image_seq == 0. 972c22301ad4fa0f4cf71e9c877d072e6f07a0bf682Richard Genoud */ 973c22301ad4fa0f4cf71e9c877d072e6f07a0bf682Richard Genoud if (image_seq && (image_seq != ubi->image_seq)) { 974c22301ad4fa0f4cf71e9c877d072e6f07a0bf682Richard Genoud ubi_err("wrong image seq:%d instead of %d", 975c22301ad4fa0f4cf71e9c877d072e6f07a0bf682Richard Genoud be32_to_cpu(ech->image_seq), ubi->image_seq); 976dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = UBI_BAD_FASTMAP; 977dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto free_hdr; 978dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 979dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 980dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = ubi_io_read_vid_hdr(ubi, pnum, vh, 0); 981dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ret && ret != UBI_IO_BITFLIPS) { 982dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("unable to read fastmap block# %i (PEB: %i)", 983dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger i, pnum); 984dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto free_hdr; 985dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 986dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 987dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (i == 0) { 988dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (be32_to_cpu(vh->vol_id) != UBI_FM_SB_VOLUME_ID) { 989dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("bad fastmap anchor vol_id: 0x%x," \ 990dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger " expected: 0x%x", 991dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger be32_to_cpu(vh->vol_id), 992dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger UBI_FM_SB_VOLUME_ID); 993dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = UBI_BAD_FASTMAP; 994dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto free_hdr; 995dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 996dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } else { 997dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (be32_to_cpu(vh->vol_id) != UBI_FM_DATA_VOLUME_ID) { 998dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("bad fastmap data vol_id: 0x%x," \ 999dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger " expected: 0x%x", 1000dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger be32_to_cpu(vh->vol_id), 1001dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger UBI_FM_DATA_VOLUME_ID); 1002dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = UBI_BAD_FASTMAP; 1003dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto free_hdr; 1004dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1005dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1006dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1007dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (sqnum < be64_to_cpu(vh->sqnum)) 1008dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger sqnum = be64_to_cpu(vh->sqnum); 1009dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1010dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = ubi_io_read(ubi, ubi->fm_buf + (ubi->leb_size * i), pnum, 1011dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi->leb_start, ubi->leb_size); 1012dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ret && ret != UBI_IO_BITFLIPS) { 1013dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("unable to read fastmap block# %i (PEB: %i, " \ 1014dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger "err: %i)", i, pnum, ret); 1015dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto free_hdr; 1016dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1017dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1018dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1019dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger kfree(fmsb); 1020dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmsb = NULL; 1021dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1022dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmsb2 = (struct ubi_fm_sb *)(ubi->fm_buf); 1023dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger tmp_crc = be32_to_cpu(fmsb2->data_crc); 1024dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmsb2->data_crc = 0; 1025dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger crc = crc32(UBI_CRC32_INIT, ubi->fm_buf, fm_size); 1026dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (crc != tmp_crc) { 1027dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("fastmap data CRC is invalid"); 1028dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("CRC should be: 0x%x, calc: 0x%x", tmp_crc, crc); 1029dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = UBI_BAD_FASTMAP; 1030dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto free_hdr; 1031dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1032dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1033dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmsb2->sqnum = sqnum; 1034dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1035dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm->used_blocks = used_blocks; 1036dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1037dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = ubi_attach_fastmap(ubi, ai, fm); 1038dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ret) { 1039dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ret > 0) 1040dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = UBI_BAD_FASTMAP; 1041dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto free_hdr; 1042dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1043dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1044dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger for (i = 0; i < used_blocks; i++) { 1045dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_wl_entry *e; 1046dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1047dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger e = kmem_cache_alloc(ubi_wl_entry_slab, GFP_KERNEL); 1048dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (!e) { 1049dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger while (i--) 1050dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger kfree(fm->e[i]); 1051dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1052dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = -ENOMEM; 1053dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto free_hdr; 1054dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1055dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1056dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger e->pnum = be32_to_cpu(fmsb2->block_loc[i]); 1057dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger e->ec = be32_to_cpu(fmsb2->block_ec[i]); 1058dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm->e[i] = e; 1059dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1060dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1061dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi->fm = fm; 1062dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi->fm_pool.max_size = ubi->fm->max_pool_size; 1063dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi->fm_wl_pool.max_size = ubi->fm->max_wl_pool_size; 1064dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_msg("attached by fastmap"); 1065dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_msg("fastmap pool size: %d", ubi->fm_pool.max_size); 1066dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_msg("fastmap WL pool size: %d", ubi->fm_wl_pool.max_size); 1067dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi->fm_disabled = 0; 1068dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1069dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_free_vid_hdr(ubi, vh); 1070dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger kfree(ech); 1071dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinbergerout: 1072dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger mutex_unlock(&ubi->fm_mutex); 1073dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ret == UBI_BAD_FASTMAP) 1074dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("Attach by fastmap failed, doing a full scan!"); 1075dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return ret; 1076dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1077dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinbergerfree_hdr: 1078dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_free_vid_hdr(ubi, vh); 1079dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger kfree(ech); 1080dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinbergerfree_fm_sb: 1081dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger kfree(fmsb); 1082dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger kfree(fm); 1083dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto out; 1084dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger} 1085dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1086dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger/** 1087dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * ubi_write_fastmap - writes a fastmap. 1088dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @ubi: UBI device object 1089dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @new_fm: the to be written fastmap 1090dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * 1091dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * Returns 0 on success, < 0 indicates an internal error. 1092dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger */ 1093dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinbergerstatic int ubi_write_fastmap(struct ubi_device *ubi, 1094dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_fastmap_layout *new_fm) 1095dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger{ 1096dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger size_t fm_pos = 0; 1097dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger void *fm_raw; 1098dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_fm_sb *fmsb; 1099dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_fm_hdr *fmh; 1100dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_fm_scan_pool *fmpl1, *fmpl2; 1101dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_fm_ec *fec; 1102dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_fm_volhdr *fvh; 1103dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_fm_eba *feba; 1104dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct rb_node *node; 1105dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_wl_entry *wl_e; 1106dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_volume *vol; 1107dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_vid_hdr *avhdr, *dvhdr; 1108dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_work *ubi_wrk; 1109dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger int ret, i, j, free_peb_count, used_peb_count, vol_count; 1110dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger int scrub_peb_count, erase_peb_count; 1111dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1112dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm_raw = ubi->fm_buf; 1113dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger memset(ubi->fm_buf, 0, ubi->fm_size); 1114dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1115dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger avhdr = new_fm_vhdr(ubi, UBI_FM_SB_VOLUME_ID); 1116dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (!avhdr) { 1117dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = -ENOMEM; 1118dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto out; 1119dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1120dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1121dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger dvhdr = new_fm_vhdr(ubi, UBI_FM_DATA_VOLUME_ID); 1122dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (!dvhdr) { 1123dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = -ENOMEM; 1124dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto out_kfree; 1125dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1126dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1127dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger spin_lock(&ubi->volumes_lock); 1128dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger spin_lock(&ubi->wl_lock); 1129dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1130dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmsb = (struct ubi_fm_sb *)fm_raw; 1131dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm_pos += sizeof(*fmsb); 1132dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_assert(fm_pos <= ubi->fm_size); 1133dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1134dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmh = (struct ubi_fm_hdr *)(fm_raw + fm_pos); 1135dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm_pos += sizeof(*fmh); 1136dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_assert(fm_pos <= ubi->fm_size); 1137dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1138dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmsb->magic = cpu_to_be32(UBI_FM_SB_MAGIC); 1139dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmsb->version = UBI_FM_FMT_VERSION; 1140dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmsb->used_blocks = cpu_to_be32(new_fm->used_blocks); 1141dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger /* the max sqnum will be filled in while *reading* the fastmap */ 1142dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmsb->sqnum = 0; 1143dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1144dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmh->magic = cpu_to_be32(UBI_FM_HDR_MAGIC); 1145dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger free_peb_count = 0; 1146dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger used_peb_count = 0; 1147dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger scrub_peb_count = 0; 1148dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger erase_peb_count = 0; 1149dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger vol_count = 0; 1150dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1151dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmpl1 = (struct ubi_fm_scan_pool *)(fm_raw + fm_pos); 1152dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm_pos += sizeof(*fmpl1); 1153dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmpl1->magic = cpu_to_be32(UBI_FM_POOL_MAGIC); 1154dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmpl1->size = cpu_to_be16(ubi->fm_pool.size); 1155dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmpl1->max_size = cpu_to_be16(ubi->fm_pool.max_size); 1156dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1157dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger for (i = 0; i < ubi->fm_pool.size; i++) 1158dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmpl1->pebs[i] = cpu_to_be32(ubi->fm_pool.pebs[i]); 1159dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1160dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmpl2 = (struct ubi_fm_scan_pool *)(fm_raw + fm_pos); 1161dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm_pos += sizeof(*fmpl2); 1162dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmpl2->magic = cpu_to_be32(UBI_FM_POOL_MAGIC); 1163dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmpl2->size = cpu_to_be16(ubi->fm_wl_pool.size); 1164dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmpl2->max_size = cpu_to_be16(ubi->fm_wl_pool.max_size); 1165dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1166dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger for (i = 0; i < ubi->fm_wl_pool.size; i++) 1167dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmpl2->pebs[i] = cpu_to_be32(ubi->fm_wl_pool.pebs[i]); 1168dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1169dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger for (node = rb_first(&ubi->free); node; node = rb_next(node)) { 1170dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger wl_e = rb_entry(node, struct ubi_wl_entry, u.rb); 1171dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fec = (struct ubi_fm_ec *)(fm_raw + fm_pos); 1172dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1173dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fec->pnum = cpu_to_be32(wl_e->pnum); 1174dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fec->ec = cpu_to_be32(wl_e->ec); 1175dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1176dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger free_peb_count++; 1177dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm_pos += sizeof(*fec); 1178dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_assert(fm_pos <= ubi->fm_size); 1179dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1180dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmh->free_peb_count = cpu_to_be32(free_peb_count); 1181dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1182dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger for (node = rb_first(&ubi->used); node; node = rb_next(node)) { 1183dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger wl_e = rb_entry(node, struct ubi_wl_entry, u.rb); 1184dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fec = (struct ubi_fm_ec *)(fm_raw + fm_pos); 1185dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1186dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fec->pnum = cpu_to_be32(wl_e->pnum); 1187dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fec->ec = cpu_to_be32(wl_e->ec); 1188dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1189dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger used_peb_count++; 1190dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm_pos += sizeof(*fec); 1191dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_assert(fm_pos <= ubi->fm_size); 1192dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1193dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmh->used_peb_count = cpu_to_be32(used_peb_count); 1194dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1195dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger for (node = rb_first(&ubi->scrub); node; node = rb_next(node)) { 1196dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger wl_e = rb_entry(node, struct ubi_wl_entry, u.rb); 1197dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fec = (struct ubi_fm_ec *)(fm_raw + fm_pos); 1198dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1199dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fec->pnum = cpu_to_be32(wl_e->pnum); 1200dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fec->ec = cpu_to_be32(wl_e->ec); 1201dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1202dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger scrub_peb_count++; 1203dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm_pos += sizeof(*fec); 1204dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_assert(fm_pos <= ubi->fm_size); 1205dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1206dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmh->scrub_peb_count = cpu_to_be32(scrub_peb_count); 1207dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1208dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1209dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger list_for_each_entry(ubi_wrk, &ubi->works, list) { 1210dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ubi_is_erase_work(ubi_wrk)) { 1211dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger wl_e = ubi_wrk->e; 1212dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_assert(wl_e); 1213dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1214dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fec = (struct ubi_fm_ec *)(fm_raw + fm_pos); 1215dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1216dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fec->pnum = cpu_to_be32(wl_e->pnum); 1217dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fec->ec = cpu_to_be32(wl_e->ec); 1218dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1219dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger erase_peb_count++; 1220dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm_pos += sizeof(*fec); 1221dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_assert(fm_pos <= ubi->fm_size); 1222dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1223dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1224dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmh->erase_peb_count = cpu_to_be32(erase_peb_count); 1225dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1226dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger for (i = 0; i < UBI_MAX_VOLUMES + UBI_INT_VOL_COUNT; i++) { 1227dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger vol = ubi->volumes[i]; 1228dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1229dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (!vol) 1230dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger continue; 1231dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1232dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger vol_count++; 1233dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1234dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fvh = (struct ubi_fm_volhdr *)(fm_raw + fm_pos); 1235dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm_pos += sizeof(*fvh); 1236dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_assert(fm_pos <= ubi->fm_size); 1237dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1238dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fvh->magic = cpu_to_be32(UBI_FM_VHDR_MAGIC); 1239dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fvh->vol_id = cpu_to_be32(vol->vol_id); 1240dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fvh->vol_type = vol->vol_type; 1241dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fvh->used_ebs = cpu_to_be32(vol->used_ebs); 1242dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fvh->data_pad = cpu_to_be32(vol->data_pad); 1243dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fvh->last_eb_bytes = cpu_to_be32(vol->last_eb_bytes); 1244dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1245dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_assert(vol->vol_type == UBI_DYNAMIC_VOLUME || 1246dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger vol->vol_type == UBI_STATIC_VOLUME); 1247dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1248dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger feba = (struct ubi_fm_eba *)(fm_raw + fm_pos); 1249dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fm_pos += sizeof(*feba) + (sizeof(__be32) * vol->reserved_pebs); 1250dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_assert(fm_pos <= ubi->fm_size); 1251dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1252dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger for (j = 0; j < vol->reserved_pebs; j++) 1253dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger feba->pnum[j] = cpu_to_be32(vol->eba_tbl[j]); 1254dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1255dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger feba->reserved_pebs = cpu_to_be32(j); 1256dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger feba->magic = cpu_to_be32(UBI_FM_EBA_MAGIC); 1257dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1258dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmh->vol_count = cpu_to_be32(vol_count); 1259dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmh->bad_peb_count = cpu_to_be32(ubi->bad_peb_count); 1260dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1261dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger avhdr->sqnum = cpu_to_be64(ubi_next_sqnum(ubi)); 1262dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger avhdr->lnum = 0; 1263dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1264dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger spin_unlock(&ubi->wl_lock); 1265dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger spin_unlock(&ubi->volumes_lock); 1266dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1267dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger dbg_bld("writing fastmap SB to PEB %i", new_fm->e[0]->pnum); 1268dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = ubi_io_write_vid_hdr(ubi, new_fm->e[0]->pnum, avhdr); 1269dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ret) { 1270dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("unable to write vid_hdr to fastmap SB!"); 1271dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto out_kfree; 1272dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1273dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1274dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger for (i = 0; i < new_fm->used_blocks; i++) { 1275dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmsb->block_loc[i] = cpu_to_be32(new_fm->e[i]->pnum); 1276dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmsb->block_ec[i] = cpu_to_be32(new_fm->e[i]->ec); 1277dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1278dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1279dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmsb->data_crc = 0; 1280dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger fmsb->data_crc = cpu_to_be32(crc32(UBI_CRC32_INIT, fm_raw, 1281dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi->fm_size)); 1282dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1283dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger for (i = 1; i < new_fm->used_blocks; i++) { 1284dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger dvhdr->sqnum = cpu_to_be64(ubi_next_sqnum(ubi)); 1285dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger dvhdr->lnum = cpu_to_be32(i); 1286dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger dbg_bld("writing fastmap data to PEB %i sqnum %llu", 1287dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger new_fm->e[i]->pnum, be64_to_cpu(dvhdr->sqnum)); 1288dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = ubi_io_write_vid_hdr(ubi, new_fm->e[i]->pnum, dvhdr); 1289dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ret) { 1290dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("unable to write vid_hdr to PEB %i!", 1291dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger new_fm->e[i]->pnum); 1292dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto out_kfree; 1293dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1294dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1295dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1296dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger for (i = 0; i < new_fm->used_blocks; i++) { 1297dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = ubi_io_write(ubi, fm_raw + (i * ubi->leb_size), 1298dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger new_fm->e[i]->pnum, ubi->leb_start, ubi->leb_size); 1299dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ret) { 1300dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("unable to write fastmap to PEB %i!", 1301dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger new_fm->e[i]->pnum); 1302dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto out_kfree; 1303dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1304dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1305dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1306dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_assert(new_fm); 1307dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi->fm = new_fm; 1308dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1309dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger dbg_bld("fastmap written!"); 1310dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1311dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinbergerout_kfree: 1312dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_free_vid_hdr(ubi, avhdr); 1313dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_free_vid_hdr(ubi, dvhdr); 1314dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinbergerout: 1315dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return ret; 1316dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger} 1317dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1318dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger/** 1319dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * erase_block - Manually erase a PEB. 1320dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @ubi: UBI device object 1321dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @pnum: PEB to be erased 1322dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * 1323dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * Returns the new EC value on success, < 0 indicates an internal error. 1324dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger */ 1325dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinbergerstatic int erase_block(struct ubi_device *ubi, int pnum) 1326dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger{ 1327dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger int ret; 1328dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_ec_hdr *ec_hdr; 1329dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger long long ec; 1330dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1331dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ec_hdr = kzalloc(ubi->ec_hdr_alsize, GFP_KERNEL); 1332dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (!ec_hdr) 1333dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return -ENOMEM; 1334dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1335dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = ubi_io_read_ec_hdr(ubi, pnum, ec_hdr, 0); 1336dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ret < 0) 1337dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto out; 1338dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger else if (ret && ret != UBI_IO_BITFLIPS) { 1339dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = -EINVAL; 1340dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto out; 1341dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1342dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1343dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = ubi_io_sync_erase(ubi, pnum, 0); 1344dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ret < 0) 1345dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto out; 1346dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1347dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ec = be64_to_cpu(ec_hdr->ec); 1348dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ec += ret; 1349dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ec > UBI_MAX_ERASECOUNTER) { 1350dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = -EINVAL; 1351dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto out; 1352dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1353dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1354dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ec_hdr->ec = cpu_to_be64(ec); 1355dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = ubi_io_write_ec_hdr(ubi, pnum, ec_hdr); 1356dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ret < 0) 1357dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto out; 1358dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1359dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = ec; 1360dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinbergerout: 1361dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger kfree(ec_hdr); 1362dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return ret; 1363dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger} 1364dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1365dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger/** 1366dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * invalidate_fastmap - destroys a fastmap. 1367dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @ubi: UBI device object 1368dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @fm: the fastmap to be destroyed 1369dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * 1370dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * Returns 0 on success, < 0 indicates an internal error. 1371dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger */ 1372dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinbergerstatic int invalidate_fastmap(struct ubi_device *ubi, 1373dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_fastmap_layout *fm) 1374dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger{ 13758930fa500f55215f0211db9b78255e727ac3f282Richard Weinberger int ret; 1376dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_vid_hdr *vh; 1377dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1378dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = erase_block(ubi, fm->e[0]->pnum); 1379dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ret < 0) 1380dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return ret; 1381dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1382dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger vh = new_fm_vhdr(ubi, UBI_FM_SB_VOLUME_ID); 1383dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (!vh) 1384dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return -ENOMEM; 1385dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1386dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger /* deleting the current fastmap SB is not enough, an old SB may exist, 1387dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * so create a (corrupted) SB such that fastmap will find it and fall 1388dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * back to scanning mode in any case */ 1389dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger vh->sqnum = cpu_to_be64(ubi_next_sqnum(ubi)); 1390dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = ubi_io_write_vid_hdr(ubi, fm->e[0]->pnum, vh); 1391dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1392dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return ret; 1393dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger} 1394dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1395dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger/** 1396dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * ubi_update_fastmap - will be called by UBI if a volume changes or 1397dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * a fastmap pool becomes full. 1398dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * @ubi: UBI device object 1399dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * 1400dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger * Returns 0 on success, < 0 indicates an internal error. 1401dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger */ 1402dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinbergerint ubi_update_fastmap(struct ubi_device *ubi) 1403dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger{ 1404dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger int ret, i; 1405dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_fastmap_layout *new_fm, *old_fm; 1406dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger struct ubi_wl_entry *tmp_e; 1407dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1408dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger mutex_lock(&ubi->fm_mutex); 1409dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1410dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_refill_pools(ubi); 1411dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1412dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ubi->ro_mode || ubi->fm_disabled) { 1413dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger mutex_unlock(&ubi->fm_mutex); 1414dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return 0; 1415dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1416dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1417dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = ubi_ensure_anchor_pebs(ubi); 1418dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ret) { 1419dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger mutex_unlock(&ubi->fm_mutex); 1420dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return ret; 1421dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1422dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1423dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger new_fm = kzalloc(sizeof(*new_fm), GFP_KERNEL); 1424dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (!new_fm) { 1425dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger mutex_unlock(&ubi->fm_mutex); 1426dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return -ENOMEM; 1427dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1428dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1429dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger new_fm->used_blocks = ubi->fm_size / ubi->leb_size; 1430dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1431dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger for (i = 0; i < new_fm->used_blocks; i++) { 1432dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger new_fm->e[i] = kmem_cache_alloc(ubi_wl_entry_slab, GFP_KERNEL); 1433dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (!new_fm->e[i]) { 1434dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger while (i--) 1435dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger kfree(new_fm->e[i]); 1436dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1437dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger kfree(new_fm); 1438dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger mutex_unlock(&ubi->fm_mutex); 1439dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return -ENOMEM; 1440dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1441dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1442dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1443dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger old_fm = ubi->fm; 1444dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi->fm = NULL; 1445dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1446dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (new_fm->used_blocks > UBI_FM_MAX_BLOCKS) { 1447dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("fastmap too large"); 1448dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = -ENOSPC; 1449dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto err; 1450dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1451dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1452dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger for (i = 1; i < new_fm->used_blocks; i++) { 1453dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger spin_lock(&ubi->wl_lock); 1454dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger tmp_e = ubi_wl_get_fm_peb(ubi, 0); 1455dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger spin_unlock(&ubi->wl_lock); 1456dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1457dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (!tmp_e && !old_fm) { 1458dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger int j; 1459dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("could not get any free erase block"); 1460dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1461dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger for (j = 1; j < i; j++) 1462dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_wl_put_fm_peb(ubi, new_fm->e[j], j, 0); 1463dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1464dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = -ENOSPC; 1465dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto err; 1466dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } else if (!tmp_e && old_fm) { 1467dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = erase_block(ubi, old_fm->e[i]->pnum); 1468dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ret < 0) { 1469dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger int j; 1470dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1471dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger for (j = 1; j < i; j++) 1472dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_wl_put_fm_peb(ubi, new_fm->e[j], 1473dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger j, 0); 1474dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1475dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("could not erase old fastmap PEB"); 1476dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto err; 1477dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1478dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1479dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger new_fm->e[i]->pnum = old_fm->e[i]->pnum; 1480dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger new_fm->e[i]->ec = old_fm->e[i]->ec; 1481dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } else { 1482dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger new_fm->e[i]->pnum = tmp_e->pnum; 1483dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger new_fm->e[i]->ec = tmp_e->ec; 1484dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1485dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (old_fm) 1486dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_wl_put_fm_peb(ubi, old_fm->e[i], i, 1487dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger old_fm->to_be_tortured[i]); 1488dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1489dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1490dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1491dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger spin_lock(&ubi->wl_lock); 1492dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger tmp_e = ubi_wl_get_fm_peb(ubi, 1); 1493dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger spin_unlock(&ubi->wl_lock); 1494dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1495dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (old_fm) { 1496dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger /* no fresh anchor PEB was found, reuse the old one */ 1497dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (!tmp_e) { 1498dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = erase_block(ubi, old_fm->e[0]->pnum); 1499dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ret < 0) { 1500dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger int i; 1501dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("could not erase old anchor PEB"); 1502dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1503dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger for (i = 1; i < new_fm->used_blocks; i++) 1504dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_wl_put_fm_peb(ubi, new_fm->e[i], 1505dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger i, 0); 1506dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto err; 1507dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1508dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1509dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger new_fm->e[0]->pnum = old_fm->e[0]->pnum; 1510dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger new_fm->e[0]->ec = ret; 1511dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } else { 1512dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger /* we've got a new anchor PEB, return the old one */ 1513dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_wl_put_fm_peb(ubi, old_fm->e[0], 0, 1514dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger old_fm->to_be_tortured[0]); 1515dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1516dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger new_fm->e[0]->pnum = tmp_e->pnum; 1517dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger new_fm->e[0]->ec = tmp_e->ec; 1518dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1519dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } else { 1520dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (!tmp_e) { 1521dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger int i; 1522dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("could not find any anchor PEB"); 1523dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1524dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger for (i = 1; i < new_fm->used_blocks; i++) 1525dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_wl_put_fm_peb(ubi, new_fm->e[i], i, 0); 1526dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1527dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = -ENOSPC; 1528dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto err; 1529dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1530dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1531dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger new_fm->e[0]->pnum = tmp_e->pnum; 1532dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger new_fm->e[0]->ec = tmp_e->ec; 1533dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1534dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1535dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger down_write(&ubi->work_sem); 1536dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger down_write(&ubi->fm_sem); 1537dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = ubi_write_fastmap(ubi, new_fm); 1538dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger up_write(&ubi->fm_sem); 1539dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger up_write(&ubi->work_sem); 1540dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1541dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ret) 1542dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto err; 1543dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1544dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinbergerout_unlock: 1545dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger mutex_unlock(&ubi->fm_mutex); 1546dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger kfree(old_fm); 1547dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger return ret; 1548dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1549dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinbergererr: 1550dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger kfree(new_fm); 1551dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1552dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_warn("Unable to write new fastmap, err=%i", ret); 1553dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger 1554dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = 0; 1555dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (old_fm) { 1556dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = invalidate_fastmap(ubi, old_fm); 1557dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger if (ret < 0) 1558dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ubi_err("Unable to invalidiate current fastmap!"); 1559dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger else if (ret) 1560dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger ret = 0; 1561dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger } 1562dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger goto out_unlock; 1563dbb7d2a88d2a7bd3624e090bee42cdee048d9290Richard Weinberger} 1564