49 lines
1.5 KiB
Factor
49 lines
1.5 KiB
Factor
|
USING: alien fry generalizations io io.ports kernel locals math
|
||
|
|
sequences sequences.private tools.annotations ;
|
||
|
|
IN: tools.annotations.assertions
|
||
|
|
|
||
|
|
ERROR: invalid-nth-unsafe n seq word ;
|
||
|
|
|
||
|
|
: check-nth-unsafe ( n seq word -- n seq )
|
||
|
|
2over length >= [ invalid-nth-unsafe ] [ drop ] if ; inline
|
||
|
|
|
||
|
|
: (assert-nth-unsafe) ( word -- )
|
||
|
|
dup [ swap '[ _ check-nth-unsafe @ ] ] curry annotate ;
|
||
|
|
|
||
|
|
: assert-nth-unsafe ( -- )
|
||
|
|
\ nth-unsafe (assert-nth-unsafe)
|
||
|
|
\ set-nth-unsafe (assert-nth-unsafe) ;
|
||
|
|
|
||
|
|
: reset-nth-unsafe ( -- )
|
||
|
|
\ nth-unsafe reset
|
||
|
|
\ set-nth-unsafe reset ;
|
||
|
|
|
||
|
|
ERROR: invalid-stream-read-unsafe len buf port word ;
|
||
|
|
ERROR: invalid-stream-read-unsafe-return out-len in-len buf port word ;
|
||
|
|
|
||
|
|
:: check-stream-read-unsafe-before ( n buf stream word -- n buf stream )
|
||
|
|
buf alien? [ n buf port ] [
|
||
|
|
n buf byte-length >
|
||
|
|
[ n buf stream word invalid-stream-read-unsafe ]
|
||
|
|
[ n buf stream ] if
|
||
|
|
] if ; inline
|
||
|
|
|
||
|
|
:: check-stream-read-unsafe-after ( count n buf stream word -- count )
|
||
|
|
count n >
|
||
|
|
[ count n buf stream word invalid-stream-read-unsafe-return ]
|
||
|
|
[ count ] if ;
|
||
|
|
|
||
|
|
: (assert-stream-read-unsafe) ( word -- )
|
||
|
|
dup [ swap '[ _
|
||
|
|
[ check-stream-read-unsafe-before @ ]
|
||
|
|
[ check-stream-read-unsafe-after ] 4 nbi
|
||
|
|
] ] curry annotate ;
|
||
|
|
|
||
|
|
: assert-stream-read-unsafe ( -- )
|
||
|
|
\ stream-read-unsafe (assert-stream-read-unsafe)
|
||
|
|
\ stream-read-partial-unsafe (assert-stream-read-unsafe) ;
|
||
|
|
|
||
|
|
: reset-stream-read-unsafe ( -- )
|
||
|
|
\ stream-read-unsafe reset
|
||
|
|
\ stream-read-partial-unsafe reset ;
|