--- /dev/null
+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 ;