sf: sst: setup read func
[oweals/u-boot.git] / drivers / mtd / mtdcore.c
1 /*
2  * Core registration and callback routines for MTD
3  * drivers and users.
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License version 2 as
7  * published by the Free Software Foundation.
8  */
9
10 #include <linux/mtd/mtd.h>
11 #include <linux/mtd/compat.h>
12 #include <ubi_uboot.h>
13
14 struct mtd_info *mtd_table[MAX_MTD_DEVICES];
15
16 int add_mtd_device(struct mtd_info *mtd)
17 {
18         int i;
19
20         BUG_ON(mtd->writesize == 0);
21
22         for (i = 0; i < MAX_MTD_DEVICES; i++)
23                 if (!mtd_table[i]) {
24                         mtd_table[i] = mtd;
25                         mtd->index = i;
26                         mtd->usecount = 0;
27
28                         /* No need to get a refcount on the module containing
29                            the notifier, since we hold the mtd_table_mutex */
30
31                         /* We _know_ we aren't being removed, because
32                            our caller is still holding us here. So none
33                            of this try_ nonsense, and no bitching about it
34                            either. :) */
35                         return 0;
36                 }
37
38         return 1;
39 }
40
41 /**
42  *      del_mtd_device - unregister an MTD device
43  *      @mtd: pointer to MTD device info structure
44  *
45  *      Remove a device from the list of MTD devices present in the system,
46  *      and notify each currently active MTD 'user' of its departure.
47  *      Returns zero on success or 1 on failure, which currently will happen
48  *      if the requested device does not appear to be present in the list.
49  */
50 int del_mtd_device(struct mtd_info *mtd)
51 {
52         int ret;
53
54         if (mtd_table[mtd->index] != mtd) {
55                 ret = -ENODEV;
56         } else if (mtd->usecount) {
57                 printk(KERN_NOTICE "Removing MTD device #%d (%s)"
58                                 " with use count %d\n",
59                                 mtd->index, mtd->name, mtd->usecount);
60                 ret = -EBUSY;
61         } else {
62                 /* No need to get a refcount on the module containing
63                  * the notifier, since we hold the mtd_table_mutex */
64                 mtd_table[mtd->index] = NULL;
65
66                 ret = 0;
67         }
68
69         return ret;
70 }
71
72 /**
73  *      get_mtd_device - obtain a validated handle for an MTD device
74  *      @mtd: last known address of the required MTD device
75  *      @num: internal device number of the required MTD device
76  *
77  *      Given a number and NULL address, return the num'th entry in the device
78  *      table, if any.  Given an address and num == -1, search the device table
79  *      for a device with that address and return if it's still present. Given
80  *      both, return the num'th driver only if its address matches. Return
81  *      error code if not.
82  */
83 struct mtd_info *get_mtd_device(struct mtd_info *mtd, int num)
84 {
85         struct mtd_info *ret = NULL;
86         int i, err = -ENODEV;
87
88         if (num == -1) {
89                 for (i = 0; i < MAX_MTD_DEVICES; i++)
90                         if (mtd_table[i] == mtd)
91                                 ret = mtd_table[i];
92         } else if (num < MAX_MTD_DEVICES) {
93                 ret = mtd_table[num];
94                 if (mtd && mtd != ret)
95                         ret = NULL;
96         }
97
98         if (!ret)
99                 goto out_unlock;
100
101         ret->usecount++;
102         return ret;
103
104 out_unlock:
105         return ERR_PTR(err);
106 }
107
108 /**
109  *      get_mtd_device_nm - obtain a validated handle for an MTD device by
110  *      device name
111  *      @name: MTD device name to open
112  *
113  *      This function returns MTD device description structure in case of
114  *      success and an error code in case of failure.
115  */
116 struct mtd_info *get_mtd_device_nm(const char *name)
117 {
118         int i, err = -ENODEV;
119         struct mtd_info *mtd = NULL;
120
121         for (i = 0; i < MAX_MTD_DEVICES; i++) {
122                 if (mtd_table[i] && !strcmp(name, mtd_table[i]->name)) {
123                         mtd = mtd_table[i];
124                         break;
125                 }
126         }
127
128         if (!mtd)
129                 goto out_unlock;
130
131         mtd->usecount++;
132         return mtd;
133
134 out_unlock:
135         return ERR_PTR(err);
136 }
137
138 void put_mtd_device(struct mtd_info *mtd)
139 {
140         int c;
141
142         c = --mtd->usecount;
143         BUG_ON(c < 0);
144 }
145
146 #if defined(CONFIG_CMD_MTDPARTS_SPREAD)
147 /**
148  * mtd_get_len_incl_bad
149  *
150  * Check if length including bad blocks fits into device.
151  *
152  * @param mtd an MTD device
153  * @param offset offset in flash
154  * @param length image length
155  * @return image length including bad blocks in *len_incl_bad and whether or not
156  *         the length returned was truncated in *truncated
157  */
158 void mtd_get_len_incl_bad(struct mtd_info *mtd, uint64_t offset,
159                           const uint64_t length, uint64_t *len_incl_bad,
160                           int *truncated)
161 {
162         *truncated = 0;
163         *len_incl_bad = 0;
164
165         if (!mtd->block_isbad) {
166                 *len_incl_bad = length;
167                 return;
168         }
169
170         uint64_t len_excl_bad = 0;
171         uint64_t block_len;
172
173         while (len_excl_bad < length) {
174                 if (offset >= mtd->size) {
175                         *truncated = 1;
176                         return;
177                 }
178
179                 block_len = mtd->erasesize - (offset & (mtd->erasesize - 1));
180
181                 if (!mtd->block_isbad(mtd, offset & ~(mtd->erasesize - 1)))
182                         len_excl_bad += block_len;
183
184                 *len_incl_bad += block_len;
185                 offset       += block_len;
186         }
187 }
188 #endif /* defined(CONFIG_CMD_MTDPARTS_SPREAD) */