]> gitweb.factorcode.org Git - factor.git/blobdiff - basis/checksums/sha/sha.factor
factor: trim using lists
[factor.git] / basis / checksums / sha / sha.factor
index 82d2c2b4919606e33739a5e274842e295868b15f..cff9445067268d3adb3ab42aa0bf7447904f5872 100644 (file)
@@ -1,42 +1,56 @@
 ! Copyright (C) 2008 Doug Coleman.
 ! See http://factorcode.org/license.txt for BSD license.
-USING: accessors checksums checksums.common checksums.stream
-combinators combinators.smart fry generalizations grouping
-io.binary kernel literals locals make math math.bitwise
-math.ranges multiline namespaces sbufs sequences
-sequences.generalizations sequences.private splitting strings ;
+USING: accessors alien.c-types checksums checksums.common
+combinators combinators.smart endian grouping kernel
+kernel.private literals math math.bitwise ranges
+sequences sequences.generalizations sequences.private
+specialized-arrays ;
+SPECIALIZED-ARRAY: uint
+SPECIALIZED-ARRAY: ulong
 IN: checksums.sha
 
+MIXIN: sha
+INSTANCE: sha block-checksum
+
 SINGLETON: sha1
-INSTANCE: sha1 stream-checksum
+INSTANCE: sha1 sha
 
 SINGLETON: sha-224
 SINGLETON: sha-256
 
-INSTANCE: sha-224 stream-checksum
-INSTANCE: sha-256 stream-checksum
+INSTANCE: sha-224 sha
+INSTANCE: sha-256 sha
+
+<PRIVATE
 
-TUPLE: sha1-state < checksum-state K H W word-size ;
+TUPLE: sha1-state < block-checksum-state
+{ K uint-array }
+{ H uint-array }
+{ W uint-array }
+{ word-size fixnum } ;
 
 CONSTANT: initial-H-sha1
-    { 
-        HEX: 67452301
-        HEX: efcdab89
-        HEX: 98badcfe
-        HEX: 10325476
-        HEX: c3d2e1f0
+    uint-array{
+        0x67452301
+        0xefcdab89
+        0x98badcfe
+        0x10325476
+        0xc3d2e1f0
     }
 
 CONSTANT: K-sha1
     $[
-        20 HEX: 5a827999 <repetition>
-        20 HEX: 6ed9eba1 <repetition>
-        20 HEX: 8f1bbcdc <repetition>
-        20 HEX: ca62c1d6 <repetition> 
-        4 { } nappend-as
+        20 0x5a827999 <repetition>
+        20 0x6ed9eba1 <repetition>
+        20 0x8f1bbcdc <repetition>
+        20 0xca62c1d6 <repetition>
+        4 uint-array{ } nappend-as
     ]
 
-TUPLE: sha2-state < checksum-state K H word-size ;
+TUPLE: sha2-state < block-checksum-state
+{ K uint-array }
+{ H uint-array }
+{ word-size fixnum } ;
 
 TUPLE: sha2-short < sha2-state ;
 
@@ -51,8 +65,6 @@ M: sha2-state clone
     [ clone ] change-H
     [ clone ] change-K ;
 
-<PRIVATE
-
 CONSTANT: a 0
 CONSTANT: b 1
 CONSTANT: c 2
@@ -63,84 +75,83 @@ CONSTANT: g 6
 CONSTANT: h 7
 
 CONSTANT: initial-H-224
-    {
-        HEX: c1059ed8 HEX: 367cd507 HEX: 3070dd17 HEX: f70e5939
-        HEX: ffc00b31 HEX: 68581511 HEX: 64f98fa7 HEX: befa4fa4
+    uint-array{
+        0xc1059ed8 0x367cd507 0x3070dd17 0xf70e5939
+        0xffc00b31 0x68581511 0x64f98fa7 0xbefa4fa4
     }
 
 CONSTANT: initial-H-256
-    {
-        HEX: 6a09e667 HEX: bb67ae85 HEX: 3c6ef372 HEX: a54ff53a
-        HEX: 510e527f HEX: 9b05688c HEX: 1f83d9ab HEX: 5be0cd19
+    uint-array{
+        0x6a09e667 0xbb67ae85 0x3c6ef372 0xa54ff53a
+        0x510e527f 0x9b05688c 0x1f83d9ab 0x5be0cd19
     }
 
 CONSTANT: initial-H-384
-    {
-        HEX: cbbb9d5dc1059ed8
-        HEX: 629a292a367cd507
-        HEX: 9159015a3070dd17
-        HEX: 152fecd8f70e5939
-        HEX: 67332667ffc00b31
-        HEX: 8eb44a8768581511
-        HEX: db0c2e0d64f98fa7
-        HEX: 47b5481dbefa4fa4
+    ulong-array{
+        0xcbbb9d5dc1059ed8
+        0x629a292a367cd507
+        0x9159015a3070dd17
+        0x152fecd8f70e5939
+        0x67332667ffc00b31
+        0x8eb44a8768581511
+        0xdb0c2e0d64f98fa7
+        0x47b5481dbefa4fa4
     }
 
 CONSTANT: initial-H-512
-    {
-        HEX: 6a09e667f3bcc908
-        HEX: bb67ae8584caa73b
-        HEX: 3c6ef372fe94f82b
-        HEX: a54ff53a5f1d36f1
-        HEX: 510e527fade682d1
-        HEX: 9b05688c2b3e6c1f
-        HEX: 1f83d9abfb41bd6b
-        HEX: 5be0cd19137e2179
+    ulong-array{
+        0x6a09e667f3bcc908
+        0xbb67ae8584caa73b
+        0x3c6ef372fe94f82b
+        0xa54ff53a5f1d36f1
+        0x510e527fade682d1
+        0x9b05688c2b3e6c1f
+        0x1f83d9abfb41bd6b
+        0x5be0cd19137e2179
     }
 
 CONSTANT: K-256
-    {
-        HEX: 428a2f98 HEX: 71374491 HEX: b5c0fbcf HEX: e9b5dba5
-        HEX: 3956c25b HEX: 59f111f1 HEX: 923f82a4 HEX: ab1c5ed5
-        HEX: d807aa98 HEX: 12835b01 HEX: 243185be HEX: 550c7dc3
-        HEX: 72be5d74 HEX: 80deb1fe HEX: 9bdc06a7 HEX: c19bf174
-        HEX: e49b69c1 HEX: efbe4786 HEX: 0fc19dc6 HEX: 240ca1cc
-        HEX: 2de92c6f HEX: 4a7484aa HEX: 5cb0a9dc HEX: 76f988da
-        HEX: 983e5152 HEX: a831c66d HEX: b00327c8 HEX: bf597fc7
-        HEX: c6e00bf3 HEX: d5a79147 HEX: 06ca6351 HEX: 14292967
-        HEX: 27b70a85 HEX: 2e1b2138 HEX: 4d2c6dfc HEX: 53380d13
-        HEX: 650a7354 HEX: 766a0abb HEX: 81c2c92e HEX: 92722c85
-        HEX: a2bfe8a1 HEX: a81a664b HEX: c24b8b70 HEX: c76c51a3
-        HEX: d192e819 HEX: d6990624 HEX: f40e3585 HEX: 106aa070
-        HEX: 19a4c116 HEX: 1e376c08 HEX: 2748774c HEX: 34b0bcb5
-        HEX: 391c0cb3 HEX: 4ed8aa4a HEX: 5b9cca4f HEX: 682e6ff3
-        HEX: 748f82ee HEX: 78a5636f HEX: 84c87814 HEX: 8cc70208
-        HEX: 90befffa HEX: a4506ceb HEX: bef9a3f7 HEX: c67178f2
+    uint-array{
+        0x428a2f98 0x71374491 0xb5c0fbcf 0xe9b5dba5
+        0x3956c25b 0x59f111f1 0x923f82a4 0xab1c5ed5
+        0xd807aa98 0x12835b01 0x243185be 0x550c7dc3
+        0x72be5d74 0x80deb1fe 0x9bdc06a7 0xc19bf174
+        0xe49b69c1 0xefbe4786 0x0fc19dc6 0x240ca1cc
+        0x2de92c6f 0x4a7484aa 0x5cb0a9dc 0x76f988da
+        0x983e5152 0xa831c66d 0xb00327c8 0xbf597fc7
+        0xc6e00bf3 0xd5a79147 0x06ca6351 0x14292967
+        0x27b70a85 0x2e1b2138 0x4d2c6dfc 0x53380d13
+        0x650a7354 0x766a0abb 0x81c2c92e 0x92722c85
+        0xa2bfe8a1 0xa81a664b 0xc24b8b70 0xc76c51a3
+        0xd192e819 0xd6990624 0xf40e3585 0x106aa070
+        0x19a4c116 0x1e376c08 0x2748774c 0x34b0bcb5
+        0x391c0cb3 0x4ed8aa4a 0x5b9cca4f 0x682e6ff3
+        0x748f82ee 0x78a5636f 0x84c87814 0x8cc70208
+        0x90befffa 0xa4506ceb 0xbef9a3f7 0xc67178f2
     }
 
 CONSTANT: K-384
-    {
-
-        HEX: 428a2f98d728ae22 HEX: 7137449123ef65cd HEX: b5c0fbcfec4d3b2f HEX: e9b5dba58189dbbc 
-        HEX: 3956c25bf348b538 HEX: 59f111f1b605d019 HEX: 923f82a4af194f9b HEX: ab1c5ed5da6d8118 
-        HEX: d807aa98a3030242 HEX: 12835b0145706fbe HEX: 243185be4ee4b28c HEX: 550c7dc3d5ffb4e2
-        HEX: 72be5d74f27b896f HEX: 80deb1fe3b1696b1 HEX: 9bdc06a725c71235 HEX: c19bf174cf692694 
-        HEX: e49b69c19ef14ad2 HEX: efbe4786384f25e3 HEX: 0fc19dc68b8cd5b5 HEX: 240ca1cc77ac9c65 
-        HEX: 2de92c6f592b0275 HEX: 4a7484aa6ea6e483 HEX: 5cb0a9dcbd41fbd4 HEX: 76f988da831153b5 
-        HEX: 983e5152ee66dfab HEX: a831c66d2db43210 HEX: b00327c898fb213f HEX: bf597fc7beef0ee4 
-        HEX: c6e00bf33da88fc2 HEX: d5a79147930aa725 HEX: 06ca6351e003826f HEX: 142929670a0e6e70 
-        HEX: 27b70a8546d22ffc HEX: 2e1b21385c26c926 HEX: 4d2c6dfc5ac42aed HEX: 53380d139d95b3df 
-        HEX: 650a73548baf63de HEX: 766a0abb3c77b2a8 HEX: 81c2c92e47edaee6 HEX: 92722c851482353b 
-        HEX: a2bfe8a14cf10364 HEX: a81a664bbc423001 HEX: c24b8b70d0f89791 HEX: c76c51a30654be30 
-        HEX: d192e819d6ef5218 HEX: d69906245565a910 HEX: f40e35855771202a HEX: 106aa07032bbd1b8 
-        HEX: 19a4c116b8d2d0c8 HEX: 1e376c085141ab53 HEX: 2748774cdf8eeb99 HEX: 34b0bcb5e19b48a8 
-        HEX: 391c0cb3c5c95a63 HEX: 4ed8aa4ae3418acb HEX: 5b9cca4f7763e373 HEX: 682e6ff3d6b2b8a3 
-        HEX: 748f82ee5defb2fc HEX: 78a5636f43172f60 HEX: 84c87814a1f0ab72 HEX: 8cc702081a6439ec 
-        HEX: 90befffa23631e28 HEX: a4506cebde82bde9 HEX: bef9a3f7b2c67915 HEX: c67178f2e372532b 
-        HEX: ca273eceea26619c HEX: d186b8c721c0c207 HEX: eada7dd6cde0eb1e HEX: f57d4f7fee6ed178 
-        HEX: 06f067aa72176fba HEX: 0a637dc5a2c898a6 HEX: 113f9804bef90dae HEX: 1b710b35131c471b 
-        HEX: 28db77f523047d84 HEX: 32caab7b40c72493 HEX: 3c9ebe0a15c9bebc HEX: 431d67c49c100d4c 
-        HEX: 4cc5d4becb3e42b6 HEX: 597f299cfc657e2a HEX: 5fcb6fab3ad6faec HEX: 6c44198c4a475817
+    ulong-array{
+        0x428a2f98d728ae22 0x7137449123ef65cd 0xb5c0fbcfec4d3b2f 0xe9b5dba58189dbbc
+        0x3956c25bf348b538 0x59f111f1b605d019 0x923f82a4af194f9b 0xab1c5ed5da6d8118
+        0xd807aa98a3030242 0x12835b0145706fbe 0x243185be4ee4b28c 0x550c7dc3d5ffb4e2
+        0x72be5d74f27b896f 0x80deb1fe3b1696b1 0x9bdc06a725c71235 0xc19bf174cf692694
+        0xe49b69c19ef14ad2 0xefbe4786384f25e3 0x0fc19dc68b8cd5b5 0x240ca1cc77ac9c65
+        0x2de92c6f592b0275 0x4a7484aa6ea6e483 0x5cb0a9dcbd41fbd4 0x76f988da831153b5
+        0x983e5152ee66dfab 0xa831c66d2db43210 0xb00327c898fb213f 0xbf597fc7beef0ee4
+        0xc6e00bf33da88fc2 0xd5a79147930aa725 0x06ca6351e003826f 0x142929670a0e6e70
+        0x27b70a8546d22ffc 0x2e1b21385c26c926 0x4d2c6dfc5ac42aed 0x53380d139d95b3df
+        0x650a73548baf63de 0x766a0abb3c77b2a8 0x81c2c92e47edaee6 0x92722c851482353b
+        0xa2bfe8a14cf10364 0xa81a664bbc423001 0xc24b8b70d0f89791 0xc76c51a30654be30
+        0xd192e819d6ef5218 0xd69906245565a910 0xf40e35855771202a 0x106aa07032bbd1b8
+        0x19a4c116b8d2d0c8 0x1e376c085141ab53 0x2748774cdf8eeb99 0x34b0bcb5e19b48a8
+        0x391c0cb3c5c95a63 0x4ed8aa4ae3418acb 0x5b9cca4f7763e373 0x682e6ff3d6b2b8a3
+        0x748f82ee5defb2fc 0x78a5636f43172f60 0x84c87814a1f0ab72 0x8cc702081a6439ec
+        0x90befffa23631e28 0xa4506cebde82bde9 0xbef9a3f7b2c67915 0xc67178f2e372532b
+        0xca273eceea26619c 0xd186b8c721c0c207 0xeada7dd6cde0eb1e 0xf57d4f7fee6ed178
+        0x06f067aa72176fba 0x0a637dc5a2c898a6 0x113f9804bef90dae 0x1b710b35131c471b
+        0x28db77f523047d84 0x32caab7b40c72493 0x3c9ebe0a15c9bebc 0x431d67c49c100d4c
+        0x4cc5d4becb3e42b6 0x597f299cfc657e2a 0x5fcb6fab3ad6faec 0x6c44198c4a475817
     }
 
 ALIAS: K-512 K-384
@@ -229,6 +240,7 @@ M: sha-256 initialize-checksum-state drop <sha-256-state> ;
     ] [ bitxor ] reduce-outputs ; inline
 
 : prepare-M-256 ( n seq -- )
+    { uint-array } declare
     {
         [ [ 16 - ] dip nth-unsafe ]
         [ [ 15 - ] dip nth-unsafe s0-256 ]
@@ -238,6 +250,7 @@ M: sha-256 initialize-checksum-state drop <sha-256-state> ;
     } 2cleave set-nth-unsafe ; inline
 
 : prepare-M-512 ( n seq -- )
+    { ulong-array } declare
     {
         [ [ 16 - ] dip nth-unsafe ]
         [ [ 15 - ] dip nth-unsafe s0-512 ]
@@ -264,9 +277,9 @@ GENERIC: pad-initial-bytes ( string sha2 -- padded-string )
     e H nth-unsafe S1-256 w+
     h H nth-unsafe w+ ; inline
 
-: T2-256 ( H -- T2 )
-    [ a swap nth-unsafe S0-256 ]
-    [ a swap slice3 maj w+ ] bi ; inline
+:: T2-256 ( H -- T2 )
+    a H nth-unsafe S0-256
+    a H slice3 maj w+ ; inline
 
 :: T1-512 ( n M H sha2 -- T1 )
     n M nth-unsafe
@@ -275,26 +288,24 @@ GENERIC: pad-initial-bytes ( string sha2 -- padded-string )
     e H nth-unsafe S1-512 w+
     h H nth-unsafe w+ ; inline
 
-: T2-512 ( H -- T2 )
-    [ a swap nth-unsafe S0-512 ]
-    [ a swap slice3 maj w+ ] bi ; inline
+:: T2-512 ( H -- T2 )
+    a H nth-unsafe S0-512
+    a H slice3 maj w+ ; inline
 
-: update-H ( T1 T2 H -- )
-    h g pick exchange-unsafe
-    g f pick exchange-unsafe
-    f e pick exchange-unsafe
-    pick d pick nth-unsafe w+ e pick set-nth-unsafe
-    d c pick exchange-unsafe
-    c b pick exchange-unsafe
-    b a pick exchange-unsafe
-    [ w+ a ] dip set-nth-unsafe ; inline
+:: update-H ( T1 T2 H -- )
+    h g H exchange-unsafe
+    g f H exchange-unsafe
+    f e H exchange-unsafe
+    T1 d H nth-unsafe w+ e H set-nth-unsafe
+    d c H exchange-unsafe
+    c b H exchange-unsafe
+    b a H exchange-unsafe
+    T1 T2 w+ a H set-nth-unsafe ; inline
 
 : prepare-message-schedule ( seq sha2 -- w-seq )
-    [ word-size>> <sliced-groups> [ be> ] map ]
-    [
-        block-size>> [ 0 pad-tail 16 ] keep [a,b) over
-        '[ _ prepare-M-256 ] each
-    ] bi ; inline
+    [ word-size>> <groups> ] [ block-size>> <uint-array> ] bi
+    [ '[ [ be> ] dip _ set-nth-unsafe ] each-index ]
+    [ 16 over length [a..b) over '[ _ prepare-M-256 ] each ] bi ; inline
 
 :: process-chunk ( M block-size cloned-H sha2 -- )
     block-size [
@@ -308,23 +319,21 @@ M: sha2-short checksum-block
     [ prepare-message-schedule ]
     [ [ block-size>> ] [ H>> clone ] [ ] tri process-chunk ] bi ;
 
-: seq>byte-array ( seq n -- string )
-    '[ _ >be ] map B{ } concat-as ;
+: sequence>byte-array ( seq n -- bytes )
+    '[ _ >be ] { } map-as B{ } concat-as ; inline
 
 : sha1>checksum ( sha2 -- bytes )
-    H>> 4 seq>byte-array ;
+    H>> 4 sequence>byte-array ; inline
 
 : sha-224>checksum ( sha2 -- bytes )
-    H>> 7 head 4 seq>byte-array ;
+    H>> 7 head 4 sequence>byte-array ; inline
 
 : sha-256>checksum ( sha2 -- bytes )
-    H>> 4 seq>byte-array ;
+    H>> 4 sequence>byte-array ; inline
 
 : pad-last-short-block ( state -- )
     [ bytes>> t ] [ bytes-read>> pad-last-block ] [ ] tri
-    [ checksum-block ] curry each ;
-
-PRIVATE>
+    [ checksum-block ] curry each ; inline
 
 M: sha-224-state get-checksum
     clone
@@ -334,27 +343,20 @@ M: sha-256-state get-checksum
     clone
     [ pad-last-short-block ] [ sha-256>checksum ] bi ;
 
-M: sha-224 checksum-stream ( stream checksum -- byte-array )
-    drop
-    [ <sha-224-state> ] dip add-checksum-stream get-checksum ;
-
-M: sha-256 checksum-stream ( stream checksum -- byte-array )
-    drop
-    [ <sha-256-state> ] dip add-checksum-stream get-checksum ;
-
 : sha1-W ( t seq -- )
+    { uint-array } declare
     {
         [ [ 3 - ] dip nth-unsafe ]
         [ [ 8 - ] dip nth-unsafe bitxor ]
         [ [ 14 - ] dip nth-unsafe bitxor ]
         [ [ 16 - ] dip nth-unsafe bitxor 1 bitroll-32 ]
         [ ]
-    } 2cleave set-nth-unsafe ;
+    } 2cleave set-nth-unsafe ; inline
 
 : prepare-sha1-message-schedule ( seq -- w-seq )
-    4 <sliced-groups> [ be> ] map
-    80 0 pad-tail 16 80 [a,b) over
-    '[ _ sha1-W ] each ; inline
+    4 <groups> 80 <uint-array>
+    [ '[ [ be> ] dip _ set-nth-unsafe ] each-index ]
+    [ 16 80 [a..b) over '[ _ sha1-W ] each ] bi ; inline
 
 : sha1-f ( B C D n -- f_nbcd )
     20 /i
@@ -363,7 +365,7 @@ M: sha-256 checksum-stream ( stream checksum -- byte-array )
         { 1 [ bitxor bitxor ] }
         { 2 [ 2dup bitand [ pick bitand [ bitand ] dip ] dip bitor bitor ] }
         { 3 [ bitxor bitxor ] }
-    } case ;
+    } case ; inline
 
 :: inner-loop ( n H W K -- temp )
     a H nth-unsafe :> A
@@ -374,16 +376,16 @@ M: sha-256 checksum-stream ( stream checksum -- byte-array )
     [
         A 5 bitroll-32
 
-        B C D n sha1-f 
+        B C D n sha1-f
 
         E
 
         n K nth-unsafe
 
         n W nth-unsafe
-    ] sum-outputs 32 bits ;
+    ] sum-outputs 32 bits ; inline
 
-:: process-sha1-chunk ( bytes H W K state -- )
+:: process-sha1-chunk ( H W K state -- )
     80 [
         H W K inner-loop
         d H nth-unsafe e H set-nth-unsafe
@@ -397,13 +399,10 @@ M: sha-256 checksum-stream ( stream checksum -- byte-array )
 M:: sha1-state checksum-block ( bytes state -- )
     bytes prepare-sha1-message-schedule state W<<
 
-    bytes
     state [ H>> clone ] [ W>> ] [ K>> ] tri state process-sha1-chunk ;
 
 M: sha1-state get-checksum
     clone
     [ pad-last-short-block ] [ sha-256>checksum ] bi ;
 
-M: sha1 checksum-stream ( stream checksum -- byte-array )
-    drop
-    [ <sha1-state> ] dip add-checksum-stream get-checksum ;
+PRIVATE>