[U-Boot] [PATCHv4] [RFC] DM: early_malloc for DM added.

[Marek Vasut <marex@denx.de>]

Dear Tomas Hlavacek,

> early_malloc for DM with support for more heaps and lightweight
> first heap on stack.
> 
> Adaptation layer for seamless calling of early_malloc or dlmalloc from
> DM based on init stage added (dmmalloc() and related functions).
> 
> Signed-off-by: Tomas Hlavacek <tmshlvck@gmail.com>
> ---

It looks mostly OK, few comments

I'd say, pull out the modification of global data into separate patch and put it 
before this patch. That'd make review of the core code much easier.

[...]

> +
> +#include <common.h> /* for ROUND_UP */
> +#include <asm/u-boot.h>
> +#include <asm/global_data.h> /* for gd_t and gd */
> +#include <asm/types.h> /* for phys_addr_t and size_addt_t */
> +
> +#include <dmmalloc.h>
> +#include <malloc.h>
> +
> +DECLARE_GLOBAL_DATA_PTR;
> +
> +#ifdef CONFIG_SYS_EARLY_MALLOC
> +static struct early_heap_header *def_early_brk(size_t size)
> +{
> +	struct early_heap_header *h =
> +		(struct early_heap_header *)CONFIG_SYS_EARLY_HEAP_ADDR;
> +
> +	h->free_space_pointer = (void *)(roundup(
> +				(phys_addr_t)CONFIG_SYS_EARLY_HEAP_ADDR +
> +				sizeof(struct early_heap_header),
> +				sizeof(phys_addr_t)));
> +	h->free_bytes = size - roundup(sizeof(struct early_heap_header),
> +				sizeof(phys_addr_t));
> +	h->next_early_heap = NULL;
> +
> +	return h;
> +}
> +
> +struct early_heap_header *early_brk(size_t size)
> +	__attribute__((weak, alias("def_early_brk")));

what about using (it needs <linux/compiler.h>):

__weak struct early_heap_header *early_brk(size_t size)
{
...
body
...
}

> +void *early_malloc(size_t size)
> +{
> +	phys_addr_t addr;
> +	struct early_heap_header *h;
> +
> +	/* Align size. */
> +	size = roundup(size, sizeof(phys_addr_t));
> +
> +	/* Choose early_heap with enough space. */
> +	h = gd->early_heap_first;
> +	while ((h->free_bytes < size) && (h->next_early_heap != NULL))
> +		h = h->next_early_heap;
> +
> +	if (h->free_bytes < size) {
> +		debug("Early heap overflow. Heap %p, free %d, required %d.",
> +			h, h->free_bytes, size);
> +		return NULL;
> +	}
> +
> +	/* Choose block beginning address and mark next free space. */
> +	addr = (phys_addr_t)h->free_space_pointer;
> +
> +	h->free_space_pointer += size;
> +	h->free_bytes -= size;
> +
> +	return (void *)addr;
> +}
> +
> +static int is_early_malloc_active(void)
> +{
> +	if (gd->flags & GD_FLG_RELOC)
> +		return 0;
> +
> +	return 1;
> +}
> +
> +#endif /* CONFIG_SYS_EARLY_MALLOC */
> +
> +void *dmmalloc(size_t size)
> +{
> +#ifdef CONFIG_SYS_EARLY_MALLOC
> +	if (is_early_malloc_active())
> +		return early_malloc(size);
> +#endif /* CONFIG_SYS_EARLY_MALLOC */

Or you can implement empty prototypes for these functions in case CONFIG_SYS ... 
isn't defined to punt this preprocessor bloat.

> +	return malloc(size);
> +}

[...]

> diff --git a/include/configs/zipitz2.h b/include/configs/zipitz2.h
> index 26204af..5cd0dcb 100644
> --- a/include/configs/zipitz2.h
> +++ b/include/configs/zipitz2.h
> @@ -176,8 +176,13 @@ unsigned char zipitz2_spi_read(void);
> 
>  #define	CONFIG_SYS_LOAD_ADDR		CONFIG_SYS_DRAM_BASE
> 
> +#define CONFIG_SYS_EARLY_HEAP_ADDR     (GENERATED_GBL_DATA_SIZE + \
> +	PHYS_SDRAM_1)
> +#define CONFIG_SYS_EARLY_HEAP_SIZE     256
> +

1) Pull this file into separate patch and order it afterwards this patch.
2) You're putting your early thingie into SDRAM, which works on PXA (sadly) ... 
looking through the PXA init code, it needs cleanup, damn

... there's no real hint so far, just a rant.

>  #define CONFIG_SYS_SDRAM_BASE		PHYS_SDRAM_1
> -#define	CONFIG_SYS_INIT_SP_ADDR		(GENERATED_GBL_DATA_SIZE + 
PHYS_SDRAM_1 +
> 2048) +#define CONFIG_SYS_INIT_SP_ADDR		(GENERATED_GBL_DATA_SIZE 
+ \
> +	CONFIG_SYS_EARLY_HEAP_SIZE + PHYS_SDRAM_1 + 2048)
> 
>  /*
>   * NOR FLASH
> @@ -260,4 +265,9 @@ unsigned char zipitz2_spi_read(void);
>  #define CONFIG_SYS_MCIO0_VAL	0x0001430f
>  #define CONFIG_SYS_MCIO1_VAL	0x0001430f

[...]