]> bbs.cooldavid.org Git - net-next-2.6.git/blame - drivers/staging/dt3155/allocator.c
Staging: Eliminate a NULL pointer dereference
[net-next-2.6.git] / drivers / staging / dt3155 / allocator.c
CommitLineData
aa337ef1
SS
1/*
2 * allocator.c -- allocate after high_memory, if available
3 *
4 * NOTE: this is different from my previous allocator, the one that
5 * assembles pages, which revealed itself both slow and unreliable.
6 *
7 * Copyright (C) 1998 rubini@linux.it (Alessandro Rubini)
8 *
9 * This program is free software; you can redistribute it and/or modify
10 * it under the terms of the GNU General Public License as published by
11 * the Free Software Foundation; either version 2 of the License, or
12 * (at your option) any later version.
13 *
14 * This program is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 * GNU General Public License for more details.
18 *
19 * You should have received a copy of the GNU General Public License
20 * along with this program; if not, write to the Free Software
21 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
22 *
23
24-- Changes --
25
26 Date Programmer Description of changes made
27 -------------------------------------------------------------------
28 02-Aug-2002 NJC allocator now steps in 1MB increments, rather
1769fd86 29 than doubling its size each time.
dcff74ce 30 Also, allocator_init(u32 *) now returns
aa337ef1
SS
31 (in the first arg) the size of the free
32 space. This is no longer consistent with
33 using the allocator as a module, and some changes
34 may be necessary for that purpose. This was
35 designed to work with the DT3155 driver, in
36 stand alone mode only!!!
37 26-Oct-2009 SS Port to 2.6.30 kernel.
38 */
39
40
41#ifndef __KERNEL__
42# define __KERNEL__
43#endif
44#ifndef MODULE
45# define MODULE
46#endif
47
aa337ef1
SS
48
49#include <linux/sched.h>
50#include <linux/kernel.h>
51#include <linux/fs.h>
52#include <linux/proc_fs.h>
53#include <linux/errno.h>
54#include <linux/types.h>
55#include <linux/mm.h> /* PAGE_ALIGN() */
2141ec62 56#include <linux/io.h>
5a0e3ad6 57#include <linux/slab.h>
aa337ef1
SS
58
59#include <asm/page.h>
60
7d4984d8
JP
61#include "allocator.h"
62
aa337ef1
SS
63/*#define ALL_DEBUG*/
64#define ALL_MSG "allocator: "
65
66#undef PDEBUG /* undef it, just in case */
67#ifdef ALL_DEBUG
68# define __static
69# define DUMP_LIST() dump_list()
70# ifdef __KERNEL__
71 /* This one if debugging is on, and kernel space */
1769fd86 72# define PDEBUG(fmt, args...) printk(KERN_DEBUG ALL_MSG fmt, ## args)
aa337ef1
SS
73# else
74 /* This one for user space */
75# define PDEBUG(fmt, args...) fprintf(stderr, fmt, ## args)
76# endif
77#else
78# define PDEBUG(fmt, args...) /* not debugging: nothing */
79# define DUMP_LIST()
80# define __static static
81#endif
82
83#undef PDEBUGG
84#define PDEBUGG(fmt, args...)
85/*#define PDEBUGG(fmt, args...) printk( KERN_DEBUG ALL_MSG fmt, ## args)*/
86
87
7d4984d8
JP
88static int allocator_himem = 1; /* 0 = probe, pos. = megs, neg. = disable */
89static int allocator_step = 1; /* This is the step size in MB */
90static int allocator_probe = 1; /* This is a flag -- 1=probe, 0=don't probe */
aa337ef1 91
1769fd86
GKH
92static unsigned long allocator_buffer; /* physical address */
93static unsigned long allocator_buffer_size; /* kilobytes */
aa337ef1
SS
94
95/*
96 * The allocator keeps a list of DMA areas, so multiple devices
97 * can coexist. The list is kept sorted by address
98 */
99
100struct allocator_struct {
1769fd86
GKH
101 unsigned long address;
102 unsigned long size;
103 struct allocator_struct *next;
aa337ef1
SS
104};
105
7d4984d8 106static struct allocator_struct *allocator_list;
aa337ef1
SS
107
108#ifdef ALL_DEBUG
109static int dump_list(void)
110{
1769fd86 111 struct allocator_struct *ptr;
aa337ef1 112
1769fd86
GKH
113 PDEBUG("Current list:\n");
114 for (ptr = allocator_list; ptr; ptr = ptr->next)
115 PDEBUG("0x%08lx (size %likB)\n", ptr->address, ptr->size>>10);
116 return 0;
aa337ef1
SS
117}
118#endif
119
120/* ========================================================================
121 * This function is the actual allocator.
122 *
123 * If space is available in high memory (as detected at load time), that
124 * one is returned. The return value is a physical address (i.e., it can
125 * be used straight ahead for DMA, but needs remapping for program use).
126 */
127
7d4984d8 128unsigned long allocator_allocate_dma(unsigned long kilobytes, gfp_t flags)
aa337ef1 129{
1769fd86
GKH
130 struct allocator_struct *ptr = allocator_list, *newptr;
131 unsigned long bytes = kilobytes << 10;
132
133 /* check if high memory is available */
134 if (!allocator_buffer)
135 return 0;
136
137 /* Round it to a multiple of the pagesize */
138 bytes = PAGE_ALIGN(bytes);
139 PDEBUG("request for %li bytes\n", bytes);
140
141 while (ptr && ptr->next) {
142 if (ptr->next->address - (ptr->address + ptr->size) >= bytes)
143 break; /* enough space */
144 ptr = ptr->next;
145 }
146 if (!ptr->next) {
147 DUMP_LIST();
148 PDEBUG("alloc failed\n");
149 return 0; /* end of list */
150 }
7d4984d8 151 newptr = kmalloc(sizeof(struct allocator_struct), flags);
1769fd86
GKH
152 if (!newptr)
153 return 0;
154
155 /* ok, now stick it after ptr */
156 newptr->address = ptr->address + ptr->size;
157 newptr->size = bytes;
158 newptr->next = ptr->next;
159 ptr->next = newptr;
160
161 DUMP_LIST();
162 PDEBUG("returning 0x%08lx\n", newptr->address);
163 return newptr->address;
aa337ef1
SS
164}
165
1769fd86 166int allocator_free_dma(unsigned long address)
aa337ef1 167{
1769fd86 168 struct allocator_struct *ptr = allocator_list, *prev;
aa337ef1 169
1769fd86
GKH
170 while (ptr && ptr->next) {
171 if (ptr->next->address == address)
172 break;
173 ptr = ptr->next;
174 }
175 /* the one being freed is ptr->next */
176 prev = ptr; ptr = ptr->next;
177
178 if (!ptr) {
c60e55f3 179 pr_err(ALL_MSG "free_dma but add. not allocated\n");
1769fd86 180 return -EINVAL;
aa337ef1 181 }
1769fd86
GKH
182 PDEBUGG("freeing: %08lx (%li) next %08lx\n", ptr->address, ptr->size,
183 ptr->next->address);
184 prev->next = ptr->next;
185 kfree(ptr);
186
187 /* dump_list(); */
188 return 0;
aa337ef1
SS
189}
190
191/* ========================================================================
192 * Init and cleanup
193 *
194 * On cleanup everything is released. If the list is not empty, that a
195 * problem of our clients
196 */
3a8954e8 197int allocator_init(u32 *allocator_max)
aa337ef1 198{
1769fd86
GKH
199 /* check how much free memory is there */
200 void *remapped;
201 unsigned long max;
202 unsigned long trial_size = allocator_himem<<20;
203 unsigned long last_trial = 0;
204 unsigned long step = allocator_step<<20;
205 unsigned long i = 0;
206 struct allocator_struct *head, *tail;
207 char test_string[] = "0123456789abcde"; /* 16 bytes */
208
209 PDEBUGG("himem = %i\n", allocator_himem);
210 if (allocator_himem < 0) /* don't even try */
211 return -EINVAL;
212
213 if (!trial_size)
214 trial_size = 1<<20; /* not specified: try one meg */
215
216 while (1) {
217 remapped = ioremap(__pa(high_memory), trial_size);
218 if (!remapped) {
219 PDEBUGG("%li megs failed!\n", trial_size>>20);
220 break;
221 }
222 PDEBUGG("Trying %li megs (at %p, %p)\n", trial_size>>20,
223 (void *)__pa(high_memory), remapped);
224 for (i = last_trial; i < trial_size; i += 16) {
225 strcpy((char *)(remapped)+i, test_string);
226 if (strcmp((char *)(remapped)+i, test_string))
227 break;
228 }
229 iounmap((void *)remapped);
230 schedule();
231 last_trial = trial_size;
232 if (i == trial_size)
233 trial_size += step; /* increment, if all went well */
234 else {
235 PDEBUGG("%li megs copy test failed!\n", trial_size>>20);
236 break;
237 }
238 if (!allocator_probe)
239 break;
240 }
241 PDEBUG("%li megs (%li k, %li b)\n", i>>20, i>>10, i);
242 allocator_buffer_size = i>>10; /* kilobytes */
243 allocator_buffer = __pa(high_memory);
244 if (!allocator_buffer_size) {
245 printk(KERN_WARNING ALL_MSG "no free high memory to use\n");
246 return -ENOMEM;
247 }
248
249 /*
250 * to simplify things, always have two cells in the list:
251 * the first and the last. This avoids some conditionals and
252 * extra code when allocating and deallocating: we only play
253 * in the middle of the list
254 */
255 head = kmalloc(sizeof(struct allocator_struct), GFP_KERNEL);
256 if (!head)
257 return -ENOMEM;
258 tail = kmalloc(sizeof(struct allocator_struct), GFP_KERNEL);
259 if (!tail) {
260 kfree(head);
261 return -ENOMEM;
262 }
263
264 max = allocator_buffer_size<<10;
265
266 head->size = tail->size = 0;
267 head->address = allocator_buffer;
268 tail->address = allocator_buffer + max;
269 head->next = tail;
270 tail->next = NULL;
271 allocator_list = head;
272
273 /* Back to the user code, in KB */
274 *allocator_max = allocator_buffer_size;
275
276 return 0; /* ok, ready */
aa337ef1
SS
277}
278
279void allocator_cleanup(void)
280{
1769fd86 281 struct allocator_struct *ptr, *next;
aa337ef1 282
1769fd86
GKH
283 for (ptr = allocator_list; ptr; ptr = next) {
284 next = ptr->next;
285 PDEBUG("freeing list: 0x%08lx\n", ptr->address);
286 kfree(ptr);
287 }
aa337ef1 288
1769fd86
GKH
289 allocator_buffer = 0;
290 allocator_buffer_size = 0;
291 allocator_list = NULL;
aa337ef1
SS
292}
293
294