"

 COPYRIGHT (c) 1993 by Claus Gittinger

              All Rights Reserved

 This software is furnished under a license and may be used

 only in accordance with the terms of that license and with the

 inclusion of the above copyright notice.   This software may not

 be provided or otherwise made available to, or used by, any

 other person.  No title to or ownership of the software is

 hereby transferred.

"

"{ Package: 'stx:libbasic' }"

UninterpretedBytes subclass:#ExternalBytes

	instanceVariableNames:'address* size'

	classVariableNames:'AllocatedInstances Lobby DebugMalloc TraceMalloc AccessLock'

	poolDictionaries:''

	category:'System-Support'

!

!ExternalBytes primitiveFunctions!

%{

struct mallocList {

        char *chunk;

        unsigned size;

        struct mallocList *next;

};

static struct mallocList *mallocList = (struct mallocList *)0;

static mallocCount = 0;

char *

__stx_malloc(nBytes) {

        char *ptr;

        char *malloc();

        struct mallocList *e, *this;

        int found;

        ptr = malloc(nBytes);

        if (@global(TraceMalloc) == true) {

#ifdef alpha64

            printf("EXTBYTES [info]: allocated %d bytes at: %016lx\n", nBytes, ptr);

#else

            printf("EXTBYTES [info]: allocated %d bytes at: %08x\n", nBytes, ptr);

#endif

        }

        if (@global(DebugMalloc) == true) {

            if (ptr) {

                found = 0;

                for (this=mallocList; this; this=this->next) {

                    if (this->chunk == ptr) {

#ifdef alpha64

                        printf("EXTBYTES [warning]: **** %016lx already allocated (freed somewhere else ?)\n", ptr);

#else

                        printf("EXTBYTES [warning]: **** %08x already allocated (freed somewhere else ?)\n", ptr);

#endif

                        found++;

                    }

                }

                if (! found) {

                    e = (struct mallocList *) malloc(sizeof(struct mallocList));

                    e->next = mallocList;

                    e->chunk = ptr;

                    e->size = nBytes;

                    mallocList = e;

                    mallocCount++;

                }

            }

        }

        return ptr;

}

char *

__stx_calloc(n, size) {

        char *ptr;

        ptr = __stx_malloc(n * size);

        if (ptr != (char *)0) {

            bzero(ptr, (n * size));

        }

        return ptr;

}

char *

__stx_realloc(ptr, nBytes)

    char *ptr;

{

        char *newPtr;

        char *realloc(), *malloc();

        struct mallocList *this, *prev, *next, *e;

        int foundOld, foundNew;

        newPtr = realloc(ptr, nBytes);

        if (@global(TraceMalloc) == true) {

#ifdef alpha64

            printf("EXTBYTES [info]: reallocated %d bytes for %016lx at: %016lx\n", nBytes, ptr, newPtr);

#else

            printf("EXTBYTES [info]: reallocated %d bytes for %08x at: %08x\n", nBytes, ptr, newPtr);

#endif

        }

        if (@global(DebugMalloc) == true) {

            if (newPtr) {

                foundNew = 0;

                for (this=mallocList; this; this=this->next) {

                    if (this->chunk == newPtr) {

#ifdef alpha64

                        printf("EXTBYTES [warning]: **** %016lx already allocated (freed somewhere else ?)\n", newPtr);

#else

                        printf("EXTBYTES [warning]: **** %08x already allocated (freed somewhere else ?)\n", newPtr);

#endif

                        foundNew = 1;

                    }

                }

            }

            if (ptr) {

                foundOld = 0;

                for (this=mallocList, prev=0; this; this=next) {

                    next = this->next;

                    if (this->chunk == ptr) {

                        if (prev) {

                            prev->next = next;

                        } else {

                            mallocList = next;

                        }

                        free(this);

                        foundOld++;

                    } else {

                        prev = this;

                    }

                }

                if (! foundOld) {

#ifdef alpha64

                    printf("EXTBYTES [warning]: **** realloc alien %lx (allocated somewhere else ?)\n", ptr);

#else

                    printf("EXTBYTES [warning]: **** realloc alien %x (allocated somewhere else ?)\n", ptr);

#endif

                }

            }

            if (! foundNew) {

                e = (struct mallocList *) malloc(sizeof(struct mallocList));

                e->next = mallocList;

                e->chunk = newPtr;

                e->size = nBytes;

                mallocList = e;

                if (! ptr) {

                    mallocCount++;

                }

            }

        }

        return newPtr;

}

__stx_free(ptr)

    char *ptr;

{

        struct mallocList *this, *prev, *next;

        int found;

        if (@global(TraceMalloc) == true) {

            printf("EXTBYTES: free bytes at: %08x\n", ptr);

        }

        if (@global(DebugMalloc) == true) {

            if (ptr) {

                found = 0;

                for (this=mallocList, prev=0; this; this=next) {

                    next = this->next;

                    if (this->chunk == ptr) {

                        if (prev) {

                            prev->next = next;

                        } else {

                            mallocList = next;

                        }

                        free(this);

                        found++;

                        mallocCount--;

                    } else {

                        prev = this;

                    }

                }

                if (! found) {

#ifdef alpha64

                    printf("EXTBYTES [warning]: **** free: alien %lx (allocated somewhere else ?))\n", ptr);

#else

                    printf("EXTBYTES [warning]: **** free: alien %x (allocated somewhere else ?))\n", ptr);

#endif

                }

            }

        }

        return free(ptr);

}

__stx_mallocStatistics() {

	struct mallocList *this;

	int amount = 0;

	int n = 0;

	for (this=mallocList; this; this=this->next) {

	    n++;

	    amount += this->size;

	}

	printf("EXTBYTES [info]: allocated %d blocks with %d bytes overall\n", n, amount);

}

%}

! !

!ExternalBytes class methodsFor:'documentation'!

copyright

"

 COPYRIGHT (c) 1993 by Claus Gittinger

              All Rights Reserved

 This software is furnished under a license and may be used

 only in accordance with the terms of that license and with the

 inclusion of the above copyright notice.   This software may not

 be provided or otherwise made available to, or used by, any

 other person.  No title to or ownership of the software is

 hereby transferred.

"

!

documentation

"

    This class provides access to any memory in the system. Its main purpose

    is to provide a baseclass for objects referencing structured external data. 

    Normally, instances are created by primitive code which wants to pass C-data

    to Smalltalk AND grants smalltalk access to individual bytes afterwards.

    Primitives which do not want to grant this access should return instances of

    ExternalAddress. See more info there. Also, have a look at ExternalFunction

    which is another similar class, but specialized to represent callable C-functions.

    Since the memory address of an instance stays fixed (once allocated),

    it can also be used to share data with external C-parts 

    (which are not prepared for objects to change their address).

    Use with great care - access is not always checked for out-of-bounds 

    or valid addresses.

    Since the data is allocated outside the garbage collected smalltalk space,

    its address stays fix. Thus, it can be passed to external C-functions without

    any danger. However, you have to take care for freeing the memory yourself.

    To help in avoiding memory bugs, instances created with #new: are 

    registered in a local classvar and deregistered when the underlying memory

    is explicitely freed. Since a life reference (from that classvar) exists,

    the garbage collector will never find these to be reclaimable, and the

    underlying memory stays allocated (at a fix address) forever.

    To release the memory, either #free it or #unprotect it.

    The first will immediately release the memory, while the second will delay

    freeing until the next garbage collect occurs.

    If you need memory which is automatically freed, create

    the instance via #unprotectedNew: right away; the underlying malloced-memory

    will be released as soon as no smalltalk reference to the ExtBytes object

    exists any more (however, you have to know for sure, that no C-references

    exist to this memory).

    To release all memory call #releaseAllMemory which simply sets the 

    AllocatedInstances class variable to nil (thus releasing those refs).

    Example (automatic freeing as soon as ref to buffer is gone):

        |buffer|

        buffer := ExternalBytes unprotectedNew:100.

        ...

    Example (manual freeing - never freed, if ref to buffer is gone):

        |buffer|

        buffer := ExternalBytes new:100.

        ...

        buffer free

    Example (delayed automatic freeing as soon as ref to buffer is gone):

        |buffer|

        buffer := ExternalBytes new:100.

        ...

        buffer unregister

    This class only supports unstructured external data 

    - see the companion class ExternalStructure for more.

    Notice: support for external data is still being developed -

            a parser for C structure syntax and typedefs is on the way,

            making shared data with C programs much easier in the future.

    Also notice, that this class may not be available or behave different

    in other smalltalk systems, making code using it very unportable.

    It is provided for C interfacing only.

    Finally note, that ST/X's memory system is much faster than malloc/free

    in the normal case - especially for short term temporary objects,

    automatically reclaimed object memory is about 5-10 times faster than

    malloc/free.

    Things may be different for huge byte-valued objects, which are to be

    reclaimed by the oldspace colletor. 

    Anyway, for portability, we strongly warn from using this as a substitute 

    for byteArrays; it is meant for shared data with external C-functions ONLY.

    Debugging: 

        since all manual memory systems are subject of obscure errors,

        you may want to turn malloc-tracing on; this traces all allocations/frees

        done here. To do this, evaluate: 'ExternalBytes mallocTrace:true'.

        In addition, you may turn on full debugging (with 'ExternalBytes mallocDebug:true');

        if turned on, all malloc/realloc requests are remembered and later free / realloc

        requests validated against this list (i.e. to detect freeing unallocated chunks).

        To benefit from this in C-code, we recommend you use __stx_malloc() / __stx_free()

        instead of malloc() / free(). To do so, redefine them in a header file (or cc comand line)

        and recompile your external c-libraries with this.

        I used this here to find memory leaks in the Xt libraries (there are still some in

        the HTML widget ...). If mallocDebug is on, #dumpMallocChunks will print out what is

        leftOver. This may help to find trouble spots in your C-code.

    [author:]

        Claus Gittinger

    [see also:]

        ExternalAddress ExternalFunction

        ByteArray

        ( how to write primitive code :html: programming/primitive.html )

"

!

examples 

"

    These examples below are not directly executable;

    some require primitive code to be compiled, or are simply given

    as sceletton.

    Notice, that in C, indexing is 0-based, while in Smalltalk, indices start

    at 1.

    allocating memory in ST, passing it to C:

      in smalltalk:

        ...

        bytes := ExternalBytes new:100.

        ...

      in C (assuming that the bytes-object has been passed):

        ...

        char *bytePtr;

        bytePtr = (char *)(__externalBytesAddress(bytes));

        if (bytePtr) {

            ... do something with bytes at bytePtr ...

        }

      freeing (in ST):

        ...

        bytes free.

        ...

    allocating memory in C, passing it to ST:

      in C:

        ...

        char *bytePtr;

        bytePtr = (char *)(malloc(100));

        ...

        RETURN (__MKEXTERNALBYTES(bytePtr));

      in Smalltalk (extracting bytes, integers or strings):

        byteVal := bytes byteAt:1.

        ...

        intVal := bytes doubleWordAt:1 MSB:true.

        ...

        string := bytes stringAt:20.    

        ...

"

! !

!ExternalBytes class methodsFor:'initialization'!

initialize

    Lobby isNil ifTrue:[

        Lobby := Registry new.

        AccessLock := Semaphore forMutualExclusion.

    ]

! !

!ExternalBytes class methodsFor:'instance creation'!

address:aNumber

    "return a new ExternalBytes object to access bytes starting at aNumber.

     The memory at aNumber has been allocated elsewhere. The size is not known,

     therefore byte accesses will NOT be checked for valid index.

     Use this, if you get a pointer from some external source (such as a

     C-callBack function) and you have to extract bytes from that.

     DANGER ALERT: this method allows very bad things to be done to the

                   system - use with GREAT care (better: do not use it)"

    ^ self basicNew setAddress:aNumber size:nil

!

address:aNumber size:size

    "return a new ExternalBytes object to access bytes starting at aNumber.

     The memory at aNumber has been allocated elsewhere. The size is known,

     which allows byte accesses to be checked for valid index.

     Use this, if you get a pointer to a structure from some external source 

     (such as a C-callBack function) and you have to extract things from that.

     DANGER ALERT: this method allows very bad things to be done to the

                   system - use with GREAT care (better: do not use it)"

    ^ self basicNew setAddress:aNumber size:size 

!

new:numberOfBytes

    "allocate some memory usable for data;

     the memory is not controlled by the garbage collector.

     Return a corresponding ExternalBytes object or nil (if malloc fails).

     Use this, if you have to pass a block of bytes to some 

     external destination (such as a C function) which does not copy the

     data, but instead keeps a reference to it. For example, many functions

     which expect strings simply keep a ref to the passed string - for those,

     an ST/X string-pointer is not the right thing to pass, since ST/X objects

     may change their address.

     DANGER ALERT: the memory is NOT automatically freed until it is either

                   MANUALLY freed (see #free) or the returned externalBytes object

                   is unprotected or the classes releaseAllMemory method is called."

    |newInst|

    newInst := self unprotectedNew:numberOfBytes.

    newInst protectFromGC.

    ^ newInst

    "

     |bytes|

     bytes := ExternalBytes new:100.

     bytes wordAt:1 put:1.

     bytes doubleWordAt:3 put:16r12345678.

     bytes inspect

    "

!

unprotectedNew:numberOfBytes

    "allocate some memory usable for data; 

     the memory is under the control of the garbage collector.

     Return a corresponding ExternalBytes object or nil (if malloc fails).

     DANGER ALERT: the memory block as allocated will be automatically freed

                   as soon as the reference to the returned externalBytes object

                   is gone (by the next garbage collect).

                   If the memory has been passed to a C-function which 

                   remembers this pointer, bad things may happen ...."

    |newInst|

    newInst := self newForData:numberOfBytes.

    Lobby register:newInst.

    ^ newInst

! !

!ExternalBytes class methodsFor:'instance release'!

releaseAllMemory

    AllocatedInstances := nil

    "

     ... the next GC will get 'em

    "

! !

!ExternalBytes class methodsFor:'malloc debug'!

dumpMallocChunks

%{  /* NOCONTEXT */

    struct mallocList *entry;

    for (entry = mallocList; entry; entry=entry->next) {

#ifdef alpha64

        printf("  %016lx (%d)\n", entry->chunk, entry->size);

#else

        printf("  %08x (%d)\n", entry->chunk, entry->size);

#endif

    }

%}

    "

     self dumpMallocChunks

    "

!

freeAllMallocChunks

    "free all stx_malloc'd memory. Be careful, this does no validation at all;