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 = smp_processor_id();
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
	spin_unlock_irqrestore(&per_cpu(l1_data_sram_lock, cpu), flags);
Bryan Wu's avatar
Bryan Wu committed
414 415 416 417 418

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

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

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

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

436 437
	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
438 439

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

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

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

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

460 461
	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
462 463

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

	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
479
	unsigned long flags;
Bryan Wu's avatar
Bryan Wu committed
480
	int ret;
481
	unsigned int cpu;
Bryan Wu's avatar
Bryan Wu committed
482

483
	cpu = smp_processor_id();
Bryan Wu's avatar
Bryan Wu committed
484
	/* add mutex operation */
485
	spin_lock_irqsave(&per_cpu(l1_data_sram_lock, cpu), flags);
Bryan Wu's avatar
Bryan Wu committed
486

487 488
	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
489 490

	/* add mutex operation */
491
	spin_unlock_irqrestore(&per_cpu(l1_data_sram_lock, cpu), flags);
Bryan Wu's avatar
Bryan Wu committed
492 493 494 495 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

	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)
{
534
#if L1_CODE_LENGTH != 0
535
	unsigned long flags;
Bryan Wu's avatar
Bryan Wu committed
536
	void *addr;
537
	unsigned int cpu;
Bryan Wu's avatar
Bryan Wu committed
538

539
	cpu = smp_processor_id();
Bryan Wu's avatar
Bryan Wu committed
540
	/* add mutex operation */
541
	spin_lock_irqsave(&per_cpu(l1_inst_sram_lock, cpu), flags);
Bryan Wu's avatar
Bryan Wu committed
542

543 544
	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
545 546

	/* add mutex operation */
547
	spin_unlock_irqrestore(&per_cpu(l1_inst_sram_lock, cpu), flags);
Bryan Wu's avatar
Bryan Wu committed
548 549 550 551 552 553 554 555 556 557 558 559 560 561

	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
562
	unsigned long flags;
Bryan Wu's avatar
Bryan Wu committed
563
	int ret;
564
	unsigned int cpu;
Bryan Wu's avatar
Bryan Wu committed
565

566
	cpu = smp_processor_id();
Bryan Wu's avatar
Bryan Wu committed
567
	/* add mutex operation */
568
	spin_lock_irqsave(&per_cpu(l1_inst_sram_lock, cpu), flags);
Bryan Wu's avatar
Bryan Wu committed
569

570 571
	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
572 573

	/* add mutex operation */
574
	spin_unlock_irqrestore(&per_cpu(l1_inst_sram_lock, cpu), flags);
Bryan Wu's avatar
Bryan Wu committed
575 576 577 578 579 580 581 582 583 584 585

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

/* L1 Scratchpad memory allocate function */
void *l1sram_alloc(size_t size)
{
586
	unsigned long flags;
Bryan Wu's avatar
Bryan Wu committed
587
	void *addr;
588
	unsigned int cpu;
Bryan Wu's avatar
Bryan Wu committed
589

590
	cpu = smp_processor_id();
Bryan Wu's avatar
Bryan Wu committed
591
	/* add mutex operation */
592
	spin_lock_irqsave(&per_cpu(l1sram_lock, cpu), flags);
Bryan Wu's avatar
Bryan Wu committed
593

594 595
	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
596 597

	/* add mutex operation */
598
	spin_unlock_irqrestore(&per_cpu(l1sram_lock, cpu), flags);
Bryan Wu's avatar
Bryan Wu committed
599 600 601 602 603 604 605

	return addr;
}

/* L1 Scratchpad memory allocate function */
void *l1sram_alloc_max(size_t *psize)
{
606
	unsigned long flags;
Bryan Wu's avatar
Bryan Wu committed
607
	void *addr;
608
	unsigned int cpu;
Bryan Wu's avatar
Bryan Wu committed
609

610
	cpu = smp_processor_id();
Bryan Wu's avatar
Bryan Wu committed
611
	/* add mutex operation */
612
	spin_lock_irqsave(&per_cpu(l1sram_lock, cpu), flags);
Bryan Wu's avatar
Bryan Wu committed
613

614 615
	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
616 617

	/* add mutex operation */
618
	spin_unlock_irqrestore(&per_cpu(l1sram_lock, cpu), flags);
Bryan Wu's avatar
Bryan Wu committed
619 620 621 622 623 624 625

	return addr;
}

/* L1 Scratchpad memory free function */
int l1sram_free(const void *addr)
{
626
	unsigned long flags;
Bryan Wu's avatar
Bryan Wu committed
627
	int ret;
628
	unsigned int cpu;
Bryan Wu's avatar
Bryan Wu committed
629

630
	cpu = smp_processor_id();
Bryan Wu's avatar
Bryan Wu committed
631
	/* add mutex operation */
632
	spin_lock_irqsave(&per_cpu(l1sram_lock, cpu), flags);
Bryan Wu's avatar
Bryan Wu committed
633

634 635
	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
636 637

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

	return ret;
}

643 644
void *l2_sram_alloc(size_t size)
{
645
#if L2_LENGTH != 0
646
	unsigned long flags;
647 648 649 650 651 652 653 654 655 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
	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)
{
681
#if L2_LENGTH != 0
682
	unsigned long flags;
683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700
	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
701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719
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);

720 721 722 723
/* 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
724 725 726 727 728 729
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;

730
	lsl = kzalloc(sizeof(struct sram_list_struct), GFP_KERNEL);
Bryan Wu's avatar
Bryan Wu committed
731 732 733 734 735 736 737 738 739 740 741 742
	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);

743 744 745
	if (addr == NULL && (flags & L2_SRAM))
		addr = l2_sram_alloc(size);

Bryan Wu's avatar
Bryan Wu committed
746 747 748 749 750 751 752 753 754 755 756
	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);
757 758 759 760 761

#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.
 */
762 763 764
/* Need to keep line of output the same.  Currently, that is 44 bytes
 * (including newline).
 */
765
static int _sram_proc_read(char *buf, int *len, int count, const char *desc,
766 767
		struct sram_piece *pfree_head,
		struct sram_piece *pused_head)
768
{
769 770 771 772
	struct sram_piece *pslot;

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

774
	*len += sprintf(&buf[*len], "--- SRAM %-14s Size   PID State     \n", desc);
775 776 777 778 779

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

	while (pslot != NULL) {
780
		*len += sprintf(&buf[*len], "%p-%p %10i %5i %-10s\n",
781 782 783 784 785 786 787 788 789
			pslot->paddr, pslot->paddr + pslot->size,
			pslot->size, pslot->pid, "ALLOCATED");

		pslot = pslot->next;
	}

	pslot = pfree_head->next;

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

		pslot = pslot->next;
795
	}
796 797

	return 0;
798
}
799
static int sram_proc_read(char *buf, char **start, off_t offset, int count,
800 801 802
		int *eof, void *data)
{
	int len = 0;
803
	unsigned int cpu;
804

805 806 807 808
	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;
809
#if L1_DATA_A_LENGTH != 0
810 811 812 813
		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;
814 815
#endif
#if L1_DATA_B_LENGTH != 0
816 817 818 819
		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;
820 821
#endif
#if L1_CODE_LENGTH != 0
822 823 824 825
		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;
826
#endif
827
	}
828
#if L2_LENGTH != 0
829 830
	if (_sram_proc_read(buf, &len, count, "L2", &free_l2_sram_head,
		&used_l2_sram_head))
831 832
		goto not_done;
#endif
833 834
	*eof = 1;
 not_done:
835 836 837
	return len;
}

838
static int __init sram_proc_init(void)
839 840 841 842 843 844 845
{
	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;
	}
846
	ptr->read_proc = sram_proc_read;
847 848
	return 0;
}
849
late_initcall(sram_proc_init);
850
#endif