96 lines
2.3 KiB
Factor
96 lines
2.3 KiB
Factor
! Copyright (C) 2004, 2005 Mackenzie Straight
|
|
! Copyright (C) 2007, 2009 Slava Pestov
|
|
! Copyright (C) 2007, 2008 Doug Coleman
|
|
! See http://factorcode.org/license.txt for BSD license.
|
|
USING: alien assocs continuations alien.destructors kernel
|
|
namespaces accessors sets summary destructors destructors.private ;
|
|
IN: libc
|
|
|
|
: errno ( -- int )
|
|
"int" "factor" "err_no" { } alien-invoke ;
|
|
|
|
: clear-errno ( -- )
|
|
"void" "factor" "clear_err_no" { } alien-invoke ;
|
|
|
|
<PRIVATE
|
|
|
|
: (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 ;
|
|
|
|
! 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" { "char*" } alien-invoke ;
|
|
|
|
DESTRUCTOR: free
|