103 lines
		
	
	
		
			2.4 KiB
		
	
	
	
		
			Factor
		
	
	
			
		
		
	
	
			103 lines
		
	
	
		
			2.4 KiB
		
	
	
	
		
			Factor
		
	
	
! Copyright (C) 2004, 2005 Mackenzie Straight
 | 
						|
! Copyright (C) 2007, 2010 Slava Pestov
 | 
						|
! Copyright (C) 2007, 2008 Doug Coleman
 | 
						|
! See http://factorcode.org/license.txt for BSD license.
 | 
						|
USING: alien alien.c-types assocs continuations alien.destructors kernel
 | 
						|
namespaces accessors sets summary destructors destructors.private ;
 | 
						|
IN: libc
 | 
						|
 | 
						|
: errno ( -- int )
 | 
						|
    int "factor" "err_no" { } alien-invoke ;
 | 
						|
 | 
						|
: set-errno ( int -- )
 | 
						|
    void "factor" "set_err_no" { int } alien-invoke ;
 | 
						|
 | 
						|
: clear-errno ( -- )
 | 
						|
    0 set-errno ;
 | 
						|
 | 
						|
: preserve-errno ( quot -- )
 | 
						|
    errno [ call ] dip set-errno ; inline
 | 
						|
 | 
						|
: (malloc) ( size -- alien )
 | 
						|
    void* "libc" "malloc" { ulong } alien-invoke ;
 | 
						|
 | 
						|
: (calloc) ( count size -- alien )
 | 
						|
    void* "libc" "calloc" { ulong ulong } alien-invoke ;
 | 
						|
 | 
						|
: (free) ( alien -- )
 | 
						|
    void "libc" "free" { void* } alien-invoke ;
 | 
						|
 | 
						|
: (realloc) ( alien size -- newalien )
 | 
						|
    void* "libc" "realloc" { void* ulong } alien-invoke ;
 | 
						|
 | 
						|
<PRIVATE
 | 
						|
 | 
						|
! We stick malloc-ptr instances in the global disposables set
 | 
						|
TUPLE: malloc-ptr value continuation ;
 | 
						|
 | 
						|
M: malloc-ptr hashcode* value>> hashcode* ;
 | 
						|
 | 
						|
M: malloc-ptr equal?
 | 
						|
    over malloc-ptr? [ [ value>> ] bi@ = ] [ 2drop f ] if ;
 | 
						|
 | 
						|
: <malloc-ptr> ( value -- malloc-ptr )
 | 
						|
    malloc-ptr new swap >>value ;
 | 
						|
 | 
						|
PRIVATE>
 | 
						|
 | 
						|
ERROR: bad-ptr ;
 | 
						|
 | 
						|
M: bad-ptr summary
 | 
						|
    drop "Memory allocation failed" ;
 | 
						|
 | 
						|
: check-ptr ( c-ptr -- c-ptr )
 | 
						|
    [ bad-ptr ] unless* ;
 | 
						|
 | 
						|
ERROR: realloc-error ptr size ;
 | 
						|
 | 
						|
M: realloc-error summary
 | 
						|
    drop "Memory reallocation failed" ;
 | 
						|
 | 
						|
<PRIVATE
 | 
						|
 | 
						|
: add-malloc ( alien -- alien )
 | 
						|
    dup <malloc-ptr> register-disposable ;
 | 
						|
 | 
						|
: delete-malloc ( alien -- )
 | 
						|
    [ <malloc-ptr> unregister-disposable ] when* ;
 | 
						|
 | 
						|
: malloc-exists? ( alien -- ? )
 | 
						|
    <malloc-ptr> disposables get key? ;
 | 
						|
 | 
						|
PRIVATE>
 | 
						|
 | 
						|
: malloc ( size -- alien )
 | 
						|
    (malloc) check-ptr add-malloc ;
 | 
						|
 | 
						|
: calloc ( count size -- alien )
 | 
						|
    (calloc) check-ptr add-malloc ;
 | 
						|
 | 
						|
: realloc ( alien size -- newalien )
 | 
						|
    [ >c-ptr ] dip
 | 
						|
    over malloc-exists? [ realloc-error ] unless
 | 
						|
    [ drop ] [ (realloc) check-ptr ] 2bi
 | 
						|
    [ delete-malloc ] [ add-malloc ] bi* ;
 | 
						|
 | 
						|
: free ( alien -- )
 | 
						|
    >c-ptr [ delete-malloc ] [ (free) ] bi ;
 | 
						|
 | 
						|
: memcpy ( dst src size -- )
 | 
						|
    void "libc" "memcpy" { void* void* ulong } alien-invoke ;
 | 
						|
 | 
						|
: memcmp ( a b size -- cmp )
 | 
						|
    int "libc" "memcmp" { void* void* ulong } alien-invoke ;
 | 
						|
 | 
						|
: memory= ( a b size -- ? )
 | 
						|
    memcmp 0 = ;
 | 
						|
 | 
						|
: strlen ( alien -- len )
 | 
						|
    size_t "libc" "strlen" { c-string } alien-invoke ;
 | 
						|
 | 
						|
DESTRUCTOR: free
 | 
						|
DESTRUCTOR: (free)
 |