checksums.common checksums.stream combinators combinators.smart ;
IN: checksums.md5
-TUPLE: md5-state bytes-read state old-state ;
+SINGLETON: md5
+INSTANCE: md5 stream-checksum
+
+TUPLE: md5-state < checksum-state state old-state ;
: <md5-state> ( -- md5-state )
- md5-state new
- 0 >>bytes-read
+ 64 md5-state new-checksum-state
{ HEX: 67452301 HEX: efcdab89 HEX: 98badcfe HEX: 10325476 }
[ clone >>state ] [ >>old-state ] bi ;
[ b c d a 9 S44 64 ]
} [ I ] with-md5-round ; inline
-: (process-md5-block) ( block state -- )
+M: md5-state checksum-block ( block state -- )
[
[ 4 <groups> [ le> ] map ] [ state>> ] bi* {
[ (process-md5-block-F) ]
nip update-md5-state
] 2bi ;
-:: process-md5-block ( block state -- )
- block length
- [ state [ + ] change-bytes-read drop ] [ 64 = ] bi [
- block state (process-md5-block)
- ] [
- block f state bytes-read>> pad-last-block
- [ state (process-md5-block) ] each
- ] if ;
-
-: get-md5 ( md5-state -- bytes )
+: md5-state>checksum ( md5-state -- bytes )
state>> [ 4 >le ] map B{ } concat-as ;
-:: stream>md5 ( state stream -- )
- 64 stream stream-read
- [ state process-md5-block ] [ length 64 = ] bi
- [ state stream stream>md5 ] when ;
+M: md5-state get-checksum ( md5-state -- bytes )
+ clone [ bytes>> f ] [ bytes-read>> pad-last-block ] [ ] tri
+ [ [ checksum-block ] curry each ] [ md5-state>checksum ] bi ;
PRIVATE>
-
-SINGLETON: md5
-
-INSTANCE: md5 stream-checksum
-
-M: md5 checksum-stream
- drop [ <md5-state> ] dip [ stream>md5 ] [ drop get-md5 ] 2bi ;
-
-GENERIC: initialize-checksum ( checksum -- state )
-GENERIC# add-bytes 1 ( state bytes -- state )
-GENERIC# add-stream 1 ( state stream -- state )
-GENERIC: finish-checksum ( state -- bytes )
-
-M: md5 initialize-checksum drop <md5-state> ;
-
-M: md5-state finish-checksum get-md5 ;
-
-M: md5-state add-bytes over [ binary <byte-reader> stream>md5 ] dip ;
-
-M: md5-state add-stream over [ stream>md5 ] dip ;
! Copyright (c) 2008 Slava Pestov
! See http://factorcode.org/license.txt for BSD license.
-USING: sequences math.parser io io.backend io.files
-kernel ;
+USING: accessors io io.backend io.files kernel math math.parser
+sequences vectors io.encodings.binary ;
IN: checksums
MIXIN: checksum
+TUPLE: checksum-state bytes-read block-size bytes ;
+
+: new-checksum-state ( block-size class -- checksum-state )
+ new
+ swap >>block-size
+ 0 >>bytes-read
+ V{ } clone >>bytes ; inline
+
+GENERIC: checksum-block ( bytes checksum -- )
+
+GENERIC: get-checksum ( checksum -- value )
+
+: add-checksum-bytes ( checksum-state data -- checksum-state )
+ over bytes>> [ push-all ] keep
+ [ dup length pick block-size>> >= ]
+ [
+ 64 cut-slice [
+ over [ checksum-block ]
+ [ [ 64 + ] change-bytes-read drop ] bi
+ ] dip
+ ] while >vector >>bytes ;
+
+: add-checksum-stream ( checksum-state stream -- checksum-state )
+ [
+ [ '[ [ _ ] dip add-checksum-bytes drop ] each-block ] keep
+ ] with-input-stream ;
+
+: add-checksum-file ( checksum-state path -- checksum-state )
+ binary <file-reader> add-checksum-stream ;
+
GENERIC: checksum-bytes ( bytes checksum -- value )
GENERIC: checksum-stream ( stream checksum -- value )