source: tests/alloc.cfa @ dca5802

//
// Cforall Version 1.0.0 Copyright (C) 2016 University of Waterloo
//
// The contents of this file are covered under the licence agreement in the
// file "LICENCE" distributed with Cforall.
//
// alloc.cfa --
//
// Author           : Peter A. Buhr
// Created On       : Wed Feb  3 07:56:22 2016
// Last Modified By : Peter A. Buhr
// Last Modified On : Fri Nov 22 15:34:19 2019
// Update Count     : 404
//

#include <assert.h>
#include <malloc.h>                                                                             // malloc_usable_size
#include <stdint.h>                                                                             // uintptr_t
#include <stdlib.h>                                                                             // posix_memalign
#include <fstream.hfa>
#include <stdlib.hfa>                                                                   // access C malloc, realloc

int * foo( int * p, int c ) { return p; }
int * bar( int * p, int c ) { return p; }
int * baz( int * p, int c ) { return p; }

int main( void ) {
        size_t dim = 10;
        char fill = '\xde';
        int * p, * p1;

        // allocation, non-array types

        p = (int *)malloc( sizeof(*p) );                                        // C malloc, type unsafe
        *p = 0xdeadbeef;
        printf( "C   malloc %#x\n", *p );
        free( p );

        p = malloc();                                       // CFA malloc, type safe
        *p = 0xdeadbeef;
        printf( "CFA malloc %#x\n", *p );
        free( p );

        p = alloc();                                        // CFA alloc, type safe
        *p = 0xdeadbeef;
        printf( "CFA alloc %#x\n", *p );
        free( p );

        p = alloc_set( fill );                                                          // CFA alloc, fill
        printf( "CFA array alloc, fill %#hhx\n", fill );
        printf( "CFA alloc, fill %08x\n", *p );
        free( p );

        p = alloc_set( 3 );                                                                     // CFA alloc, fill
        printf( "CFA alloc, fill %d\n", *p );
        free( p );


        // allocation, array types
        printf( "\n" );

        p = (int *)calloc( dim, sizeof( *p ) );                         // C array calloc, type unsafe
        printf( "C   array calloc, fill 0\n" );
        for ( i; dim ) { printf( "%#x ", p[i] ); }
        printf( "\n" );
        free( p );

        p = calloc( dim );                                  // CFA array calloc, type safe
        printf( "CFA array calloc, fill 0\n" );
        for ( i; dim ) { printf( "%#x ", p[i] ); }
        printf( "\n" );
        free( p );

        p = alloc( dim );                                   // CFA array alloc, type safe
        for ( i; dim ) { p[i] = 0xdeadbeef; }
        printf( "CFA array alloc, no fill\n" );
        for ( i; dim ) { printf( "%#x ", p[i] ); }
        printf( "\n" );
        free( p );

        p = alloc_set( 2 * dim, fill );                                         // CFA array alloc, fill
        printf( "CFA array alloc, fill %#hhx\n", fill );
        for ( i; 2 * dim ) { printf( "%#x ", p[i] ); }
        printf( "\n" );
        free( p );

        p = alloc_set( 2 * dim, 0xdeadbeef );                           // CFA array alloc, fill
        printf( "CFA array alloc, fill %#hhx\n", 0xdeadbeef );
        for ( i; 2 * dim ) { printf( "%#x ", p[i] ); }
        printf( "\n" );
        // do not free

        p1 = alloc_set( 2 * dim, p );                                           // CFA array alloc, fill
        printf( "CFA array alloc, fill from array\n" );
        for ( i; 2 * dim ) { printf( "%#x %#x, ", p[i], p1[i] ); }
        free( p1 );
        printf( "\n" );


        // resize, non-array types
        printf( "\n" );

        p = (int *)realloc( p, dim * sizeof(*p) );                      // C realloc
        printf( "C realloc\n" );
        for ( i; dim ) { printf( "%#x ", p[i] ); }
        printf( "\n" );
        // do not free

        p = realloc( p, 2 * dim * sizeof(*p) );             // CFA realloc
        for ( i; dim ~ 2 * dim ) { p[i] = 0x1010101; }
        printf( "CFA realloc\n" );
        for ( i; 2 * dim ) { printf( "%#x ", p[i] ); }
        printf( "\n" );
        // do not free


        // resize, array types
        printf( "\n" );

        p = alloc( p, dim );                                // CFA resize array alloc
        for ( i; dim ) { p[i] = 0xdeadbeef; }
        printf( "CFA resize array alloc\n" );
        for ( i; dim ) { printf( "%#x ", p[i] ); }
        printf( "\n" );
        // do not free

        p = alloc( p, 2 * dim );                            // CFA resize array alloc
        for ( i; dim ~ 2 * dim ) { p[i] = 0x1010101; }          // fill upper part
        printf( "CFA resize array alloc\n" );
        for ( i; 2 * dim ) { printf( "%#x ", p[i] ); }
        printf( "\n" );
        // do not free

        p = alloc( p, dim );                                // CFA resize array alloc
        printf( "CFA resize array alloc\n" );
        for ( i; dim ) { printf( "%#x ", p[i] ); }
        printf( "\n" );
        // do not free

        p = alloc_set( p, 3 * dim, fill );                                      // CFA resize array alloc, fill
        printf( "CFA resize array alloc\n" );
        for ( i; 3 * dim ) { printf( "%#x ", p[i] ); }
        printf( "\n" );
        // do not free

        p = alloc_set( p, dim, fill );                                          // CFA resize array alloc, fill
        printf( "CFA resize array alloc\n" );
        for ( i; dim ) { printf( "%#x ", p[i] ); }
        printf( "\n" );
        // do not free

        p = alloc_set( p, 3 * dim, fill );                                      // CFA resize array alloc, fill
        printf( "CFA resize array alloc, fill\n" );
        for ( i; 3 * dim ) { printf( "%#x ", p[i] );; }
        printf( "\n" );
        free( p );


        struct Struct { int x; double y; };
        Struct st, st1, sta[dim], sta1[dim], * stp, * stp1;

        // alignment, non-array types
        printf( "\n" );
        enum { Alignment = 128 };

        stp = &(*(Struct*)memalign( Alignment, sizeof( *stp ) ) ){ 42, 42.5 }; // C memalign
        assert( (uintptr_t)stp % Alignment == 0 );
        printf( "C   memalign %d %g\n", stp->x, stp->y );
        free( stp );

        stp = &(*memalign( Alignment )){ 42, 42.5 };          // CFA memalign
        assert( (uintptr_t)stp % Alignment == 0 );
        printf( "CFA memalign %d %g\n", stp->x, stp->y );
        free( stp );

        posix_memalign( (void **)&stp, Alignment, sizeof( *stp ) ); // C posix_memalign
        *stp = (Struct){ 42, 42.5 };
        assert( (uintptr_t)stp % Alignment == 0 );
        printf( "CFA posix_memalign %d %g\n", stp->x, stp->y );
        free( stp );

        posix_memalign( &stp, Alignment );                  // CFA posix_memalign
        *stp = (Struct){ 42, 42.5 };
        assert( (uintptr_t)stp % Alignment == 0 );
        printf( "CFA posix_memalign %d %g\n", stp->x, stp->y );
        free( stp );

        stp = &(*alloc_align( Alignment)){ 42, 42.5 };          // CFA alloc_align
        assert( (uintptr_t)stp % Alignment == 0 );
        printf( "CFA alloc_align %d %g\n", stp->x, stp->y );
        free( stp );

        stp = &(*alloc_align( Alignment )){ 42, 42.5 };         // CFA alloc_align
        assert( (uintptr_t)stp % Alignment == 0 );
        printf( "CFA alloc_align %d %g\n", stp->x, stp->y );
        free( stp );

        stp = alloc_align_set( Alignment, fill );                       // CFA memalign, fill
        assert( (uintptr_t)stp % Alignment == 0 );
        printf( "CFA alloc_align fill %#x %a\n", stp->x, stp->y );
        free( stp );

        stp = alloc_align_set( Alignment, (Struct){ 42, 42.5 } ); // CFA memalign, fill
        assert( (uintptr_t)stp % Alignment == 0 );
        printf( "CFA alloc_align fill %d %g\n", stp->x, stp->y );
        // do not free

        stp = &(*alloc_align( stp, 4096 )){ 42, 42.5 };         // CFA realign
        assert( (uintptr_t)stp % 4096 == 0 );
        printf( "CFA alloc_align %d %g\n", stp->x, stp->y );
        free( stp );


        // alignment, array types
        printf( "\n" );

        stp = alloc_align( Alignment, dim );                // CFA array memalign
        assert( (uintptr_t)stp % Alignment == 0 );
        for ( i; dim ) { stp[i] = (Struct){ 42, 42.5 }; }
        printf( "CFA array alloc_align\n" );
        for ( i; dim ) { printf( "%d %g, ", stp[i].x, stp[i].y ); }
        printf( "\n" );
        free( stp );

        stp = alloc_align_set( Alignment, dim, fill );          // CFA array memalign, fill
        assert( (uintptr_t)stp % Alignment == 0 );
        printf( "CFA array alloc_align, fill\n" );
        for ( i; dim ) { printf( "%#x %a, ", stp[i].x, stp[i].y ); }
        printf( "\n" );
        free( stp );

        stp = alloc_align_set( Alignment, dim, (Struct){ 42, 42.5 } ); // CFA array memalign, fill
        assert( (uintptr_t)stp % Alignment == 0 );
        printf( "CFA array alloc_align, fill\n" );
        for ( i; dim ) { printf( "%d %g, ", stp[i].x, stp[i].y ); }
        printf( "\n" );
        // do not free

        stp1 = alloc_align_set( Alignment, dim, stp );          // CFA array memalign, fill
        assert( (uintptr_t)stp % Alignment == 0 );
        printf( "CFA array alloc_align, fill array\n" );
        for ( i; dim ) { printf( "%d %g, ", stp1[i].x, stp1[i].y ); }
        printf( "\n" );
        free( stp1 );

        stp = alloc_align( stp, 4096, dim );                            // CFA aligned realloc array
        assert( (uintptr_t)stp % 4096 == 0 );
        for ( i; dim ) { stp[i] = (Struct){ 42, 42.5 }; }
        printf( "CFA realloc array alloc_align\n" );
        for ( i; dim ) { printf( "%d %g, ", stp[i].x, stp[i].y ); }
        printf( "\n" );
        free( stp );


        // data, non-array types
        printf( "\n" );

        memset( &st, fill );                                // CFA memset, type safe
        printf( "CFA memset %#x %a\n", st.x, st.y );
        memcpy( &st1, &st );                                // CFA memcpy, type safe
        printf( "CFA memcpy %#x %a\n", st1.x, st1.y );


        // data, array types
        printf( "\n" );

        amemset( sta, fill, dim );                                                      // CFA array memset, type safe
        printf( "CFA array memset\n" );
        for ( i; dim ) { printf( "%#x %a, ", sta[i].x, sta[i].y ); }
        printf( "\n" );

        amemcpy( sta1, sta, dim );                                                      // CFA array memcpy, type safe
        printf( "CFA array memcpy\n" );
        for ( i; dim ) { printf( "%#x %a, ", sta1[i].x, sta1[i].y ); }
        printf( "\n" );


        // new, non-array types
        printf( "\n" );

        stp = new( 42, 42.5 );
        stp1 = new( 42, 42.5 );
        printf( "CFA new initialize\n%d %g %d %g\n", stp->x, stp->y, stp1->x, stp1->y );
        delete( stp, stp1 );

        // new, array types
        stp = anew( dim, 42, 42.5 );
        printf( "CFA array new initialize\n" );
        for ( i; dim ) { printf( "%d %g, ", stp[i].x, stp[i].y ); }
        printf( "\n" );
        stp1 = anew( dim, 42, 42.5 );
        for ( i; dim ) { printf( "%d %g, ", stp1[i].x, stp1[i].y ); }
        printf( "\n" );
        adelete( dim, stp, dim, stp1 );

        // extras
        printf( "\n" );

        float * fp = malloc() + 1;
        printf( "pointer arithmetic %d\n", fp == fp - 1 );
        free( fp - 1 );

        p = foo( bar( baz( malloc(), 0 ), 0 ), 0 );
        *p = 0xdeadbeef;
        printf( "CFA deep malloc %#x\n", *p );
        free( p );

#ifdef ERR1
        stp = malloc();
        printf( "\nSHOULD FAIL\n" );
        p = realloc( stp, dim * sizeof( *stp ) );
        p = alloc( stp, dim * sizeof( *stp ) );
        p = memset( stp, 10 );
        p = memcpy( &st1, &st );
#endif
} // main

// Local Variables: //
// tab-width: 4 //
// compile-command: "cfa alloc.cfa" //
// End: //