sbv-2.10: SMT Based Verification: Symbolic Haskell theorem prover using SMT solving.

Safe HaskellNone




An implementation of RC4 (AKA Rivest Cipher 4 or Alleged RC4/ARC4), using SBV. For information on RC4, see:

We make no effort to optimize the code, and instead focus on a clear implementation. In fact, the RC4 algorithm relies on in-place update of its state heavily for efficiency, and is therefore unsuitable for a purely functional implementation.



type S = STree Word8 Word8Source

RC4 State contains 256 8-bit values. We use the symbolically accessible full-binary type STree to represent the state, since RC4 needs access to the array via a symbolic index and it's important to minimize access time.

initS :: SSource

Construct the fully balanced initial tree, where the leaves are simply the numbers 0 through 255.

type Key = [SWord8]Source

The key is a stream of Word8 values.

type RC4 = (S, SWord8, SWord8)Source

Represents the current state of the RC4 stream: it is the S array along with the i and j index values used by the PRGA.


swap :: SWord8 -> SWord8 -> S -> SSource

Swaps two elements in the RC4 array.

prga :: RC4 -> (SWord8, RC4)Source

Implements the PRGA used in RC4. We return the new state and the next key value generated.

Key schedule

initRC4 :: Key -> SSource

Constructs the state to be used by the PRGA using the given key.

keySchedule :: Key -> [SWord8]Source

The key-schedule. Note that this function returns an infinite list.

keyScheduleString :: String -> [SWord8]Source

Generate a key-schedule from a given key-string.

Encryption and Decryption

encrypt :: String -> String -> [SWord8]Source

RC4 encryption. We generate key-words and xor it with the input. The following test-vectors are from Wikipedia

>>> concatMap hex $ encrypt "Key" "Plaintext"
>>> concatMap hex $ encrypt "Wiki" "pedia"
>>> concatMap hex $ encrypt "Secret" "Attack at dawn"

decrypt :: String -> [SWord8] -> StringSource

RC4 decryption. Essentially the same as decryption. For the above test vectors we have:

>>> decrypt "Key" [0xbb, 0xf3, 0x16, 0xe8, 0xd9, 0x40, 0xaf, 0x0a, 0xd3]
>>> decrypt "Wiki" [0x10, 0x21, 0xbf, 0x04, 0x20]
>>> decrypt "Secret" [0x45, 0xa0, 0x1f, 0x64, 0x5f, 0xc3, 0x5b, 0x38, 0x35, 0x52, 0x54, 0x4b, 0x9b, 0xf5]
"Attack at dawn"


rc4IsCorrect :: IO ThmResultSource

Prove that round-trip encryption/decryption leaves the plain-text unchanged. The theorem is stated parametrically over key and plain-text sizes. The expression performs the proof for a 40-bit key (5 bytes) and 40-bit plaintext (again 5 bytes).

Note that this theorem is trivial to prove, since it is essentially establishing xor'in the same value twice leaves a word unchanged (i.e., x xor y xor y = x). However, the proof takes quite a while to complete, as it gives rise to a fairly large symbolic trace.