sram-alloc.c 19.9 KB
Newer Older
Bryan Wu's avatar
Bryan Wu committed
1
/*
2
 * SRAM allocator for Blackfin on-chip memory
Bryan Wu's avatar
Bryan Wu committed
3
 *
4
 * Copyright 2004-2009 Analog Devices Inc.
Bryan Wu's avatar
Bryan Wu committed
5
 *
6
 * Licensed under the GPL-2 or later.
Bryan Wu's avatar
Bryan Wu committed
7 8 9 10 11 12 13 14 15 16 17 18 19 20
 */

#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/types.h>
#include <linux/miscdevice.h>
#include <linux/ioport.h>
#include <linux/fcntl.h>
#include <linux/init.h>
#include <linux/poll.h>
#include <linux/proc_fs.h>
#include <linux/spinlock.h>
#include <linux/rtc.h>
#include <asm/blackfin.h>
21
#include <asm/mem_map.h>
Bryan Wu's avatar
Bryan Wu committed
22 23 24
#include "blackfin_sram.h"

/* the data structure for L1 scratchpad and DATA SRAM */
25
struct sram_piece {
Bryan Wu's avatar
Bryan Wu committed
26 27
	void *paddr;
	int size;
28
	pid_t pid;
29
	struct sram_piece *next;
Bryan Wu's avatar
Bryan Wu committed
30 31
};

32
static DEFINE_PER_CPU_SHARED_ALIGNED(spinlock_t, l1sram_lock);
33 34
static DEFINE_PER_CPU(struct sram_piece, free_l1_ssram_head);
static DEFINE_PER_CPU(struct sram_piece, used_l1_ssram_head);
Bryan Wu's avatar
Bryan Wu committed
35 36

#if L1_DATA_A_LENGTH != 0
37 38
static DEFINE_PER_CPU(struct sram_piece, free_l1_data_A_sram_head);
static DEFINE_PER_CPU(struct sram_piece, used_l1_data_A_sram_head);
Bryan Wu's avatar
Bryan Wu committed
39 40 41
#endif

#if L1_DATA_B_LENGTH != 0
42 43
static DEFINE_PER_CPU(struct sram_piece, free_l1_data_B_sram_head);
static DEFINE_PER_CPU(struct sram_piece, used_l1_data_B_sram_head);
Bryan Wu's avatar
Bryan Wu committed
44 45
#endif

46 47 48 49
#if L1_DATA_A_LENGTH || L1_DATA_B_LENGTH
static DEFINE_PER_CPU_SHARED_ALIGNED(spinlock_t, l1_data_sram_lock);
#endif

Bryan Wu's avatar
Bryan Wu committed
50
#if L1_CODE_LENGTH != 0
51
static DEFINE_PER_CPU_SHARED_ALIGNED(spinlock_t, l1_inst_sram_lock);
52 53
static DEFINE_PER_CPU(struct sram_piece, free_l1_inst_sram_head);
static DEFINE_PER_CPU(struct sram_piece, used_l1_inst_sram_head);
Bryan Wu's avatar
Bryan Wu committed
54 55
#endif

56
#if L2_LENGTH != 0
57
static spinlock_t l2_sram_lock ____cacheline_aligned_in_smp;
58 59 60
static struct sram_piece free_l2_sram_head, used_l2_sram_head;
#endif

61 62
static struct kmem_cache *sram_piece_cache;

Bryan Wu's avatar
Bryan Wu committed
63
/* L1 Scratchpad SRAM initialization function */
64
static void __init l1sram_init(void)
Bryan Wu's avatar
Bryan Wu committed
65
{
66
	unsigned int cpu;
67 68 69 70 71 72 73 74
	unsigned long reserve;

#ifdef CONFIG_SMP
	reserve = 0;
#else
	reserve = sizeof(struct l1_scratch_task_info);
#endif

75 76 77 78 79 80 81 82
	for (cpu = 0; cpu < num_possible_cpus(); ++cpu) {
		per_cpu(free_l1_ssram_head, cpu).next =
			kmem_cache_alloc(sram_piece_cache, GFP_KERNEL);
		if (!per_cpu(free_l1_ssram_head, cpu).next) {
			printk(KERN_INFO "Fail to initialize Scratchpad data SRAM.\n");
			return;
		}

83 84
		per_cpu(free_l1_ssram_head, cpu).next->paddr = (void *)get_l1_scratch_start_cpu(cpu) + reserve;
		per_cpu(free_l1_ssram_head, cpu).next->size = L1_SCRATCH_LENGTH - reserve;
85 86 87 88 89 90 91 92 93
		per_cpu(free_l1_ssram_head, cpu).next->pid = 0;
		per_cpu(free_l1_ssram_head, cpu).next->next = NULL;

		per_cpu(used_l1_ssram_head, cpu).next = NULL;

		/* mutex initialize */
		spin_lock_init(&per_cpu(l1sram_lock, cpu));
		printk(KERN_INFO "Blackfin Scratchpad data SRAM: %d KB\n",
			L1_SCRATCH_LENGTH >> 10);
94
	}
Bryan Wu's avatar
Bryan Wu committed
95 96
}

97
static void __init l1_data_sram_init(void)
Bryan Wu's avatar
Bryan Wu committed
98
{
99
#if L1_DATA_A_LENGTH != 0 || L1_DATA_B_LENGTH != 0
100
	unsigned int cpu;
101
#endif
Bryan Wu's avatar
Bryan Wu committed
102
#if L1_DATA_A_LENGTH != 0
103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122
	for (cpu = 0; cpu < num_possible_cpus(); ++cpu) {
		per_cpu(free_l1_data_A_sram_head, cpu).next =
			kmem_cache_alloc(sram_piece_cache, GFP_KERNEL);
		if (!per_cpu(free_l1_data_A_sram_head, cpu).next) {
			printk(KERN_INFO "Fail to initialize L1 Data A SRAM.\n");
			return;
		}

		per_cpu(free_l1_data_A_sram_head, cpu).next->paddr =
			(void *)get_l1_data_a_start_cpu(cpu) + (_ebss_l1 - _sdata_l1);
		per_cpu(free_l1_data_A_sram_head, cpu).next->size =
			L1_DATA_A_LENGTH - (_ebss_l1 - _sdata_l1);
		per_cpu(free_l1_data_A_sram_head, cpu).next->pid = 0;
		per_cpu(free_l1_data_A_sram_head, cpu).next->next = NULL;

		per_cpu(used_l1_data_A_sram_head, cpu).next = NULL;

		printk(KERN_INFO "Blackfin L1 Data A SRAM: %d KB (%d KB free)\n",
			L1_DATA_A_LENGTH >> 10,
			per_cpu(free_l1_data_A_sram_head, cpu).next->size >> 10);
123
	}
Bryan Wu's avatar
Bryan Wu committed
124 125
#endif
#if L1_DATA_B_LENGTH != 0
126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146
	for (cpu = 0; cpu < num_possible_cpus(); ++cpu) {
		per_cpu(free_l1_data_B_sram_head, cpu).next =
			kmem_cache_alloc(sram_piece_cache, GFP_KERNEL);
		if (!per_cpu(free_l1_data_B_sram_head, cpu).next) {
			printk(KERN_INFO "Fail to initialize L1 Data B SRAM.\n");
			return;
		}

		per_cpu(free_l1_data_B_sram_head, cpu).next->paddr =
			(void *)get_l1_data_b_start_cpu(cpu) + (_ebss_b_l1 - _sdata_b_l1);
		per_cpu(free_l1_data_B_sram_head, cpu).next->size =
			L1_DATA_B_LENGTH - (_ebss_b_l1 - _sdata_b_l1);
		per_cpu(free_l1_data_B_sram_head, cpu).next->pid = 0;
		per_cpu(free_l1_data_B_sram_head, cpu).next->next = NULL;

		per_cpu(used_l1_data_B_sram_head, cpu).next = NULL;

		printk(KERN_INFO "Blackfin L1 Data B SRAM: %d KB (%d KB free)\n",
			L1_DATA_B_LENGTH >> 10,
			per_cpu(free_l1_data_B_sram_head, cpu).next->size >> 10);
		/* mutex initialize */
147
	}
Bryan Wu's avatar
Bryan Wu committed
148 149
#endif

150 151 152 153
#if L1_DATA_A_LENGTH != 0 || L1_DATA_B_LENGTH != 0
	for (cpu = 0; cpu < num_possible_cpus(); ++cpu)
		spin_lock_init(&per_cpu(l1_data_sram_lock, cpu));
#endif
Bryan Wu's avatar
Bryan Wu committed
154 155
}

156
static void __init l1_inst_sram_init(void)
Bryan Wu's avatar
Bryan Wu committed
157 158
{
#if L1_CODE_LENGTH != 0
159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182
	unsigned int cpu;
	for (cpu = 0; cpu < num_possible_cpus(); ++cpu) {
		per_cpu(free_l1_inst_sram_head, cpu).next =
			kmem_cache_alloc(sram_piece_cache, GFP_KERNEL);
		if (!per_cpu(free_l1_inst_sram_head, cpu).next) {
			printk(KERN_INFO "Failed to initialize L1 Instruction SRAM\n");
			return;
		}

		per_cpu(free_l1_inst_sram_head, cpu).next->paddr =
			(void *)get_l1_code_start_cpu(cpu) + (_etext_l1 - _stext_l1);
		per_cpu(free_l1_inst_sram_head, cpu).next->size =
			L1_CODE_LENGTH - (_etext_l1 - _stext_l1);
		per_cpu(free_l1_inst_sram_head, cpu).next->pid = 0;
		per_cpu(free_l1_inst_sram_head, cpu).next->next = NULL;

		per_cpu(used_l1_inst_sram_head, cpu).next = NULL;

		printk(KERN_INFO "Blackfin L1 Instruction SRAM: %d KB (%d KB free)\n",
			L1_CODE_LENGTH >> 10,
			per_cpu(free_l1_inst_sram_head, cpu).next->size >> 10);

		/* mutex initialize */
		spin_lock_init(&per_cpu(l1_inst_sram_lock, cpu));
183
	}
Bryan Wu's avatar
Bryan Wu committed
184 185 186
#endif
}

187 188
static void __init l2_sram_init(void)
{
189
#if L2_LENGTH != 0
190 191 192
	free_l2_sram_head.next =
		kmem_cache_alloc(sram_piece_cache, GFP_KERNEL);
	if (!free_l2_sram_head.next) {
193
		printk(KERN_INFO "Fail to initialize L2 SRAM.\n");
194 195 196
		return;
	}

197 198 199 200
	free_l2_sram_head.next->paddr =
		(void *)L2_START + (_ebss_l2 - _stext_l2);
	free_l2_sram_head.next->size =
		L2_LENGTH - (_ebss_l2 - _stext_l2);
201 202 203 204 205 206 207 208 209 210 211
	free_l2_sram_head.next->pid = 0;
	free_l2_sram_head.next->next = NULL;

	used_l2_sram_head.next = NULL;

	printk(KERN_INFO "Blackfin L2 SRAM: %d KB (%d KB free)\n",
		L2_LENGTH >> 10,
		free_l2_sram_head.next->size >> 10);

	/* mutex initialize */
	spin_lock_init(&l2_sram_lock);
212
#endif
213
}
214

215
static int __init bfin_sram_init(void)
216 217 218 219 220 221 222 223
{
	sram_piece_cache = kmem_cache_create("sram_piece_cache",
				sizeof(struct sram_piece),
				0, SLAB_PANIC, NULL);

	l1sram_init();
	l1_data_sram_init();
	l1_inst_sram_init();
224
	l2_sram_init();
225 226

	return 0;
227
}
228
pure_initcall(bfin_sram_init);
229

230 231
/* SRAM allocate function */
static void *_sram_alloc(size_t size, struct sram_piece *pfree_head,
232
		struct sram_piece *pused_head)
Bryan Wu's avatar
Bryan Wu committed
233
{
234
	struct sram_piece *pslot, *plast, *pavail;
Bryan Wu's avatar
Bryan Wu committed
235

236
	if (size <= 0 || !pfree_head || !pused_head)
Bryan Wu's avatar
Bryan Wu committed
237 238 239 240 241
		return NULL;

	/* Align the size */
	size = (size + 3) & ~3;

242 243 244 245 246 247 248
	pslot = pfree_head->next;
	plast = pfree_head;

	/* search an available piece slot */
	while (pslot != NULL && size > pslot->size) {
		plast = pslot;
		pslot = pslot->next;
Bryan Wu's avatar
Bryan Wu committed
249
	}
250 251

	if (!pslot)
Bryan Wu's avatar
Bryan Wu committed
252 253
		return NULL;

254 255 256 257 258 259 260 261 262 263 264 265 266
	if (pslot->size == size) {
		plast->next = pslot->next;
		pavail = pslot;
	} else {
		pavail = kmem_cache_alloc(sram_piece_cache, GFP_KERNEL);

		if (!pavail)
			return NULL;

		pavail->paddr = pslot->paddr;
		pavail->size = size;
		pslot->paddr += size;
		pslot->size -= size;
Bryan Wu's avatar
Bryan Wu committed
267 268
	}

269 270 271 272 273 274 275 276 277 278 279 280 281 282 283
	pavail->pid = current->pid;

	pslot = pused_head->next;
	plast = pused_head;

	/* insert new piece into used piece list !!! */
	while (pslot != NULL && pavail->paddr < pslot->paddr) {
		plast = pslot;
		pslot = pslot->next;
	}

	pavail->next = pslot;
	plast->next = pavail;

	return pavail->paddr;
Bryan Wu's avatar
Bryan Wu committed
284 285 286
}

/* Allocate the largest available block.  */
287
static void *_sram_alloc_max(struct sram_piece *pfree_head,
288
				struct sram_piece *pused_head,
Bryan Wu's avatar
Bryan Wu committed
289 290
				unsigned long *psize)
{
291 292 293 294 295 296
	struct sram_piece *pslot, *pmax;

	if (!pfree_head || !pused_head)
		return NULL;

	pmax = pslot = pfree_head->next;
Bryan Wu's avatar
Bryan Wu committed
297

298 299 300 301 302
	/* search an available piece slot */
	while (pslot != NULL) {
		if (pslot->size > pmax->size)
			pmax = pslot;
		pslot = pslot->next;
Bryan Wu's avatar
Bryan Wu committed
303
	}
304 305

	if (!pmax)
Bryan Wu's avatar
Bryan Wu committed
306 307
		return NULL;

308 309
	*psize = pmax->size;

310
	return _sram_alloc(*psize, pfree_head, pused_head);
Bryan Wu's avatar
Bryan Wu committed
311 312
}

313 314
/* SRAM free function */
static int _sram_free(const void *addr,
315 316
			struct sram_piece *pfree_head,
			struct sram_piece *pused_head)
Bryan Wu's avatar
Bryan Wu committed
317
{
318 319 320 321
	struct sram_piece *pslot, *plast, *pavail;

	if (!pfree_head || !pused_head)
		return -1;
Bryan Wu's avatar
Bryan Wu committed
322 323

	/* search the relevant memory slot */
324 325 326 327 328 329 330
	pslot = pused_head->next;
	plast = pused_head;

	/* search an available piece slot */
	while (pslot != NULL && pslot->paddr != addr) {
		plast = pslot;
		pslot = pslot->next;
Bryan Wu's avatar
Bryan Wu committed
331
	}
332 333

	if (!pslot)
Bryan Wu's avatar
Bryan Wu committed
334 335
		return -1;

336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352
	plast->next = pslot->next;
	pavail = pslot;
	pavail->pid = 0;

	/* insert free pieces back to the free list */
	pslot = pfree_head->next;
	plast = pfree_head;

	while (pslot != NULL && addr > pslot->paddr) {
		plast = pslot;
		pslot = pslot->next;
	}

	if (plast != pfree_head && plast->paddr + plast->size == pavail->paddr) {
		plast->size += pavail->size;
		kmem_cache_free(sram_piece_cache, pavail);
	} else {
353
		pavail->next = plast->next;
354 355
		plast->next = pavail;
		plast = pavail;
Bryan Wu's avatar
Bryan Wu committed
356 357
	}

358 359 360 361
	if (pslot && plast->paddr + plast->size == pslot->paddr) {
		plast->size += pslot->size;
		plast->next = pslot->next;
		kmem_cache_free(sram_piece_cache, pslot);
Bryan Wu's avatar
Bryan Wu committed
362 363 364 365 366 367 368
	}

	return 0;
}

int sram_free(const void *addr)
{
369

Bryan Wu's avatar
Bryan Wu committed
370
#if L1_CODE_LENGTH != 0
371 372
	if (addr >= (void *)get_l1_code_start()
		 && addr < (void *)(get_l1_code_start() + L1_CODE_LENGTH))
Bryan Wu's avatar
Bryan Wu committed
373
		return l1_inst_sram_free(addr);
374
	else
Bryan Wu's avatar
Bryan Wu committed
375 376
#endif
#if L1_DATA_A_LENGTH != 0
377 378
	if (addr >= (void *)get_l1_data_a_start()
		 && addr < (void *)(get_l1_data_a_start() + L1_DATA_A_LENGTH))
Bryan Wu's avatar
Bryan Wu committed
379
		return l1_data_A_sram_free(addr);
380
	else
Bryan Wu's avatar
Bryan Wu committed
381 382
#endif
#if L1_DATA_B_LENGTH != 0
383 384
	if (addr >= (void *)get_l1_data_b_start()
		 && addr < (void *)(get_l1_data_b_start() + L1_DATA_B_LENGTH))
Bryan Wu's avatar
Bryan Wu committed
385
		return l1_data_B_sram_free(addr);
386
	else
387
#endif
388
#if L2_LENGTH != 0
389
	if (addr >= (void *)L2_START
390 391
		 && addr < (void *)(L2_START + L2_LENGTH))
		return l2_sram_free(addr);
Bryan Wu's avatar
Bryan Wu committed
392
	else
393
#endif
Bryan Wu's avatar
Bryan Wu committed
394 395 396 397 398 399
		return -1;
}
EXPORT_SYMBOL(sram_free);

void *l1_data_A_sram_alloc(size_t size)
{
400
#if L1_DATA_A_LENGTH != 0
401
	unsigned long flags;
402
	void *addr;
403
	unsigned int cpu;
Bryan Wu's avatar
Bryan Wu committed
404

405
	cpu = get_cpu();
Bryan Wu's avatar
Bryan Wu committed
406
	/* add mutex operation */
407
	spin_lock_irqsave(&per_cpu(l1_data_sram_lock, cpu), flags);
Bryan Wu's avatar
Bryan Wu committed
408

409 410
	addr = _sram_alloc(size, &per_cpu(free_l1_data_A_sram_head, cpu),
			&per_cpu(used_l1_data_A_sram_head, cpu));
Bryan Wu's avatar
Bryan Wu committed
411 412

	/* add mutex operation */
413 414
	spin_unlock_irqrestore(&per_cpu(l1_data_sram_lock, cpu), flags);
	put_cpu();
Bryan Wu's avatar
Bryan Wu committed
415 416 417 418 419

	pr_debug("Allocated address in l1_data_A_sram_alloc is 0x%lx+0x%lx\n",
		 (long unsigned int)addr, size);

	return addr;
420 421 422
#else
	return NULL;
#endif
Bryan Wu's avatar
Bryan Wu committed
423 424 425 426 427
}
EXPORT_SYMBOL(l1_data_A_sram_alloc);

int l1_data_A_sram_free(const void *addr)
{
428
#if L1_DATA_A_LENGTH != 0
429
	unsigned long flags;
Bryan Wu's avatar
Bryan Wu committed
430
	int ret;
431
	unsigned int cpu;
Bryan Wu's avatar
Bryan Wu committed
432

433
	cpu = get_cpu();
Bryan Wu's avatar
Bryan Wu committed
434
	/* add mutex operation */
435
	spin_lock_irqsave(&per_cpu(l1_data_sram_lock, cpu), flags);
Bryan Wu's avatar
Bryan Wu committed
436

437 438
	ret = _sram_free(addr, &per_cpu(free_l1_data_A_sram_head, cpu),
			&per_cpu(used_l1_data_A_sram_head, cpu));
Bryan Wu's avatar
Bryan Wu committed
439 440

	/* add mutex operation */
441 442
	spin_unlock_irqrestore(&per_cpu(l1_data_sram_lock, cpu), flags);
	put_cpu();
Bryan Wu's avatar
Bryan Wu committed
443 444

	return ret;
445 446 447
#else
	return -1;
#endif
Bryan Wu's avatar
Bryan Wu committed
448 449 450 451 452 453
}
EXPORT_SYMBOL(l1_data_A_sram_free);

void *l1_data_B_sram_alloc(size_t size)
{
#if L1_DATA_B_LENGTH != 0
454
	unsigned long flags;
Bryan Wu's avatar
Bryan Wu committed
455
	void *addr;
456
	unsigned int cpu;
Bryan Wu's avatar
Bryan Wu committed
457

458
	cpu = get_cpu();
Bryan Wu's avatar
Bryan Wu committed
459
	/* add mutex operation */
460
	spin_lock_irqsave(&per_cpu(l1_data_sram_lock, cpu), flags);
Bryan Wu's avatar
Bryan Wu committed
461

462 463
	addr = _sram_alloc(size, &per_cpu(free_l1_data_B_sram_head, cpu),
			&per_cpu(used_l1_data_B_sram_head, cpu));
Bryan Wu's avatar
Bryan Wu committed
464 465

	/* add mutex operation */
466 467
	spin_unlock_irqrestore(&per_cpu(l1_data_sram_lock, cpu), flags);
	put_cpu();
Bryan Wu's avatar
Bryan Wu committed
468 469 470 471 472 473 474 475 476 477 478 479 480 481

	pr_debug("Allocated address in l1_data_B_sram_alloc is 0x%lx+0x%lx\n",
		 (long unsigned int)addr, size);

	return addr;
#else
	return NULL;
#endif
}
EXPORT_SYMBOL(l1_data_B_sram_alloc);

int l1_data_B_sram_free(const void *addr)
{
#if L1_DATA_B_LENGTH != 0
482
	unsigned long flags;
Bryan Wu's avatar
Bryan Wu committed
483
	int ret;
484
	unsigned int cpu;
Bryan Wu's avatar
Bryan Wu committed
485

486
	cpu = get_cpu();
Bryan Wu's avatar
Bryan Wu committed
487
	/* add mutex operation */
488
	spin_lock_irqsave(&per_cpu(l1_data_sram_lock, cpu), flags);
Bryan Wu's avatar
Bryan Wu committed
489

490 491
	ret = _sram_free(addr, &per_cpu(free_l1_data_B_sram_head, cpu),
			&per_cpu(used_l1_data_B_sram_head, cpu));
Bryan Wu's avatar
Bryan Wu committed
492 493

	/* add mutex operation */
494 495
	spin_unlock_irqrestore(&per_cpu(l1_data_sram_lock, cpu), flags);
	put_cpu();
Bryan Wu's avatar
Bryan Wu committed
496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537

	return ret;
#else
	return -1;
#endif
}
EXPORT_SYMBOL(l1_data_B_sram_free);

void *l1_data_sram_alloc(size_t size)
{
	void *addr = l1_data_A_sram_alloc(size);

	if (!addr)
		addr = l1_data_B_sram_alloc(size);

	return addr;
}
EXPORT_SYMBOL(l1_data_sram_alloc);

void *l1_data_sram_zalloc(size_t size)
{
	void *addr = l1_data_sram_alloc(size);

	if (addr)
		memset(addr, 0x00, size);

	return addr;
}
EXPORT_SYMBOL(l1_data_sram_zalloc);

int l1_data_sram_free(const void *addr)
{
	int ret;
	ret = l1_data_A_sram_free(addr);
	if (ret == -1)
		ret = l1_data_B_sram_free(addr);
	return ret;
}
EXPORT_SYMBOL(l1_data_sram_free);

void *l1_inst_sram_alloc(size_t size)
{
538
#if L1_CODE_LENGTH != 0
539
	unsigned long flags;
Bryan Wu's avatar
Bryan Wu committed
540
	void *addr;
541
	unsigned int cpu;
Bryan Wu's avatar
Bryan Wu committed
542

543
	cpu = get_cpu();
Bryan Wu's avatar
Bryan Wu committed
544
	/* add mutex operation */
545
	spin_lock_irqsave(&per_cpu(l1_inst_sram_lock, cpu), flags);
Bryan Wu's avatar
Bryan Wu committed
546

547 548
	addr = _sram_alloc(size, &per_cpu(free_l1_inst_sram_head, cpu),
			&per_cpu(used_l1_inst_sram_head, cpu));
Bryan Wu's avatar
Bryan Wu committed
549 550

	/* add mutex operation */
551 552
	spin_unlock_irqrestore(&per_cpu(l1_inst_sram_lock, cpu), flags);
	put_cpu();
Bryan Wu's avatar
Bryan Wu committed
553 554 555 556 557 558 559 560 561 562 563 564 565 566

	pr_debug("Allocated address in l1_inst_sram_alloc is 0x%lx+0x%lx\n",
		 (long unsigned int)addr, size);

	return addr;
#else
	return NULL;
#endif
}
EXPORT_SYMBOL(l1_inst_sram_alloc);

int l1_inst_sram_free(const void *addr)
{
#if L1_CODE_LENGTH != 0
567
	unsigned long flags;
Bryan Wu's avatar
Bryan Wu committed
568
	int ret;
569
	unsigned int cpu;
Bryan Wu's avatar
Bryan Wu committed
570

571
	cpu = get_cpu();
Bryan Wu's avatar
Bryan Wu committed
572
	/* add mutex operation */
573
	spin_lock_irqsave(&per_cpu(l1_inst_sram_lock, cpu), flags);
Bryan Wu's avatar
Bryan Wu committed
574

575 576
	ret = _sram_free(addr, &per_cpu(free_l1_inst_sram_head, cpu),
			&per_cpu(used_l1_inst_sram_head, cpu));
Bryan Wu's avatar
Bryan Wu committed
577 578

	/* add mutex operation */
579 580
	spin_unlock_irqrestore(&per_cpu(l1_inst_sram_lock, cpu), flags);
	put_cpu();
Bryan Wu's avatar
Bryan Wu committed
581 582 583 584 585 586 587 588 589 590 591

	return ret;
#else
	return -1;
#endif
}
EXPORT_SYMBOL(l1_inst_sram_free);

/* L1 Scratchpad memory allocate function */
void *l1sram_alloc(size_t size)
{
592
	unsigned long flags;
Bryan Wu's avatar
Bryan Wu committed
593
	void *addr;
594
	unsigned int cpu;
Bryan Wu's avatar
Bryan Wu committed
595

596
	cpu = get_cpu();
Bryan Wu's avatar
Bryan Wu committed
597
	/* add mutex operation */
598
	spin_lock_irqsave(&per_cpu(l1sram_lock, cpu), flags);
Bryan Wu's avatar
Bryan Wu committed
599

600 601
	addr = _sram_alloc(size, &per_cpu(free_l1_ssram_head, cpu),
			&per_cpu(used_l1_ssram_head, cpu));
Bryan Wu's avatar
Bryan Wu committed
602 603

	/* add mutex operation */
604 605
	spin_unlock_irqrestore(&per_cpu(l1sram_lock, cpu), flags);
	put_cpu();
Bryan Wu's avatar
Bryan Wu committed
606 607 608 609 610 611 612

	return addr;
}

/* L1 Scratchpad memory allocate function */
void *l1sram_alloc_max(size_t *psize)
{
613
	unsigned long flags;
Bryan Wu's avatar
Bryan Wu committed
614
	void *addr;
615
	unsigned int cpu;
Bryan Wu's avatar
Bryan Wu committed
616

617
	cpu = get_cpu();
Bryan Wu's avatar
Bryan Wu committed
618
	/* add mutex operation */
619
	spin_lock_irqsave(&per_cpu(l1sram_lock, cpu), flags);
Bryan Wu's avatar
Bryan Wu committed
620

621 622
	addr = _sram_alloc_max(&per_cpu(free_l1_ssram_head, cpu),
			&per_cpu(used_l1_ssram_head, cpu), psize);
Bryan Wu's avatar
Bryan Wu committed
623 624

	/* add mutex operation */
625 626
	spin_unlock_irqrestore(&per_cpu(l1sram_lock, cpu), flags);
	put_cpu();
Bryan Wu's avatar
Bryan Wu committed
627 628 629 630 631 632 633

	return addr;
}

/* L1 Scratchpad memory free function */
int l1sram_free(const void *addr)
{
634
	unsigned long flags;
Bryan Wu's avatar
Bryan Wu committed
635
	int ret;
636
	unsigned int cpu;
Bryan Wu's avatar
Bryan Wu committed
637

638
	cpu = get_cpu();
Bryan Wu's avatar
Bryan Wu committed
639
	/* add mutex operation */
640
	spin_lock_irqsave(&per_cpu(l1sram_lock, cpu), flags);
Bryan Wu's avatar
Bryan Wu committed
641

642 643
	ret = _sram_free(addr, &per_cpu(free_l1_ssram_head, cpu),
			&per_cpu(used_l1_ssram_head, cpu));
Bryan Wu's avatar
Bryan Wu committed
644 645

	/* add mutex operation */
646 647
	spin_unlock_irqrestore(&per_cpu(l1sram_lock, cpu), flags);
	put_cpu();
Bryan Wu's avatar
Bryan Wu committed
648 649 650 651

	return ret;
}

652 653
void *l2_sram_alloc(size_t size)
{
654
#if L2_LENGTH != 0
655
	unsigned long flags;
656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689
	void *addr;

	/* add mutex operation */
	spin_lock_irqsave(&l2_sram_lock, flags);

	addr = _sram_alloc(size, &free_l2_sram_head,
			&used_l2_sram_head);

	/* add mutex operation */
	spin_unlock_irqrestore(&l2_sram_lock, flags);

	pr_debug("Allocated address in l2_sram_alloc is 0x%lx+0x%lx\n",
		 (long unsigned int)addr, size);

	return addr;
#else
	return NULL;
#endif
}
EXPORT_SYMBOL(l2_sram_alloc);

void *l2_sram_zalloc(size_t size)
{
	void *addr = l2_sram_alloc(size);

	if (addr)
		memset(addr, 0x00, size);

	return addr;
}
EXPORT_SYMBOL(l2_sram_zalloc);

int l2_sram_free(const void *addr)
{
690
#if L2_LENGTH != 0
691
	unsigned long flags;
692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709
	int ret;

	/* add mutex operation */
	spin_lock_irqsave(&l2_sram_lock, flags);

	ret = _sram_free(addr, &free_l2_sram_head,
			&used_l2_sram_head);

	/* add mutex operation */
	spin_unlock_irqrestore(&l2_sram_lock, flags);

	return ret;
#else
	return -1;
#endif
}
EXPORT_SYMBOL(l2_sram_free);

Bryan Wu's avatar
Bryan Wu committed
710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728
int sram_free_with_lsl(const void *addr)
{
	struct sram_list_struct *lsl, **tmp;
	struct mm_struct *mm = current->mm;

	for (tmp = &mm->context.sram_list; *tmp; tmp = &(*tmp)->next)
		if ((*tmp)->addr == addr)
			goto found;
	return -1;
found:
	lsl = *tmp;
	sram_free(addr);
	*tmp = lsl->next;
	kfree(lsl);

	return 0;
}
EXPORT_SYMBOL(sram_free_with_lsl);

729 730 731 732
/* Allocate memory and keep in L1 SRAM List (lsl) so that the resources are
 * tracked.  These are designed for userspace so that when a process exits,
 * we can safely reap their resources.
 */
Bryan Wu's avatar
Bryan Wu committed
733 734 735 736 737 738
void *sram_alloc_with_lsl(size_t size, unsigned long flags)
{
	void *addr = NULL;
	struct sram_list_struct *lsl = NULL;
	struct mm_struct *mm = current->mm;

739
	lsl = kzalloc(sizeof(struct sram_list_struct), GFP_KERNEL);
Bryan Wu's avatar
Bryan Wu committed
740 741 742 743 744 745 746 747 748 749 750 751
	if (!lsl)
		return NULL;

	if (flags & L1_INST_SRAM)
		addr = l1_inst_sram_alloc(size);

	if (addr == NULL && (flags & L1_DATA_A_SRAM))
		addr = l1_data_A_sram_alloc(size);

	if (addr == NULL && (flags & L1_DATA_B_SRAM))
		addr = l1_data_B_sram_alloc(size);

752 753 754
	if (addr == NULL && (flags & L2_SRAM))
		addr = l2_sram_alloc(size);

Bryan Wu's avatar
Bryan Wu committed
755 756 757 758 759 760 761 762 763 764 765
	if (addr == NULL) {
		kfree(lsl);
		return NULL;
	}
	lsl->addr = addr;
	lsl->length = size;
	lsl->next = mm->context.sram_list;
	mm->context.sram_list = lsl;
	return addr;
}
EXPORT_SYMBOL(sram_alloc_with_lsl);
766 767 768 769 770

#ifdef CONFIG_PROC_FS
/* Once we get a real allocator, we'll throw all of this away.
 * Until then, we need some sort of visibility into the L1 alloc.
 */
771 772 773
/* Need to keep line of output the same.  Currently, that is 44 bytes
 * (including newline).
 */
774
static int _sram_proc_read(char *buf, int *len, int count, const char *desc,
775 776
		struct sram_piece *pfree_head,
		struct sram_piece *pused_head)
777
{
778 779 780 781
	struct sram_piece *pslot;

	if (!pfree_head || !pused_head)
		return -1;
782

783
	*len += sprintf(&buf[*len], "--- SRAM %-14s Size   PID State     \n", desc);
784 785 786 787 788

	/* search the relevant memory slot */
	pslot = pused_head->next;

	while (pslot != NULL) {
789
		*len += sprintf(&buf[*len], "%p-%p %10i %5i %-10s\n",
790 791 792 793 794 795 796 797 798
			pslot->paddr, pslot->paddr + pslot->size,
			pslot->size, pslot->pid, "ALLOCATED");

		pslot = pslot->next;
	}

	pslot = pfree_head->next;

	while (pslot != NULL) {
799
		*len += sprintf(&buf[*len], "%p-%p %10i %5i %-10s\n",
800 801 802 803
			pslot->paddr, pslot->paddr + pslot->size,
			pslot->size, pslot->pid, "FREE");

		pslot = pslot->next;
804
	}
805 806

	return 0;
807
}
808
static int sram_proc_read(char *buf, char **start, off_t offset, int count,
809 810 811
		int *eof, void *data)
{
	int len = 0;
812
	unsigned int cpu;
813

814 815 816 817
	for (cpu = 0; cpu < num_possible_cpus(); ++cpu) {
		if (_sram_proc_read(buf, &len, count, "Scratchpad",
			&per_cpu(free_l1_ssram_head, cpu), &per_cpu(used_l1_ssram_head, cpu)))
			goto not_done;
818
#if L1_DATA_A_LENGTH != 0
819 820 821 822
		if (_sram_proc_read(buf, &len, count, "L1 Data A",
			&per_cpu(free_l1_data_A_sram_head, cpu),
			&per_cpu(used_l1_data_A_sram_head, cpu)))
			goto not_done;
823 824
#endif
#if L1_DATA_B_LENGTH != 0
825 826 827 828
		if (_sram_proc_read(buf, &len, count, "L1 Data B",
			&per_cpu(free_l1_data_B_sram_head, cpu),
			&per_cpu(used_l1_data_B_sram_head, cpu)))
			goto not_done;
829 830
#endif
#if L1_CODE_LENGTH != 0
831 832 833 834
		if (_sram_proc_read(buf, &len, count, "L1 Instruction",
			&per_cpu(free_l1_inst_sram_head, cpu),
			&per_cpu(used_l1_inst_sram_head, cpu)))
			goto not_done;
835
#endif
836
	}
837
#if L2_LENGTH != 0
838 839
	if (_sram_proc_read(buf, &len, count, "L2", &free_l2_sram_head,
		&used_l2_sram_head))
840 841
		goto not_done;
#endif
842 843
	*eof = 1;
 not_done:
844 845 846
	return len;
}

847
static int __init sram_proc_init(void)
848 849 850 851 852 853 854
{
	struct proc_dir_entry *ptr;
	ptr = create_proc_entry("sram", S_IFREG | S_IRUGO, NULL);
	if (!ptr) {
		printk(KERN_WARNING "unable to create /proc/sram\n");
		return -1;
	}
855
	ptr->read_proc = sram_proc_read;
856 857
	return 0;
}
858
late_initcall(sram_proc_init);
859
#endif