v/vlib/crypto/rc4/rc4.v

80 lines
1.9 KiB
V
Raw Normal View History

2020-02-03 05:00:36 +01:00
// Copyright (c) 2019-2020 Alexander Medvednikov. All rights reserved.
2019-07-26 16:48:49 +02:00
// Use of this source code is governed by an MIT license
// that can be found in the LICENSE file.
// Package rc4 implements RC4 encryption, as defined in Bruce Schneier's
// Applied Cryptography.
//
// RC4 is cryptographically broken and should not be used for secure
// applications.
// Based off: https://github.com/golang/go/blob/master/src/crypto/rc4
// Last commit: https://github.com/golang/go/commit/b35dacaac57b039205d9b07ea24098e2c3fcb12e
2019-07-26 16:48:49 +02:00
module rc4
import crypto.internal.subtle
// A Cipher is an instance of RC4 using a particular key.
struct Cipher {
mut:
s []u32
i byte
j byte
2019-07-26 16:48:49 +02:00
}
// new_cipher creates and returns a new Cipher. The key argument should be the
// RC4 key, at least 1 byte and at most 256 bytes.
pub fn new_cipher(key []byte) ?Cipher {
if key.len < 1 || key.len > 256 {
return error('crypto.rc4: invalid key size ' + key.len.str())
}
mut c := Cipher{
s: []u32{len: (256)}
2019-07-26 16:48:49 +02:00
}
for i in 0 .. 256 {
2019-07-26 16:48:49 +02:00
c.s[i] = u32(i)
}
mut j := byte(0)
for i in 0 .. 256 {
j += byte(c.s[i]) + key[i % key.len]
2019-07-26 16:48:49 +02:00
tmp := c.s[i]
c.s[i] = c.s[j]
c.s[j] = tmp
}
return c
}
2019-08-07 13:37:07 +02:00
// reset zeros the key data and makes the Cipher unusable.good to com
2019-07-26 16:48:49 +02:00
//
// Deprecated: Reset can't guarantee that the key will be entirely removed from
// the process's memory.
2020-05-17 13:51:18 +02:00
pub fn (mut c Cipher) reset() {
2019-07-26 16:48:49 +02:00
for i in c.s {
c.s[i] = 0
2019-07-26 16:48:49 +02:00
}
c.i = 0
c.j = 0
2019-07-26 16:48:49 +02:00
}
// xor_key_stream sets dst to the result of XORing src with the key stream.
// Dst and src must overlap entirely or not at all.
pub fn (mut c Cipher) xor_key_stream(mut dst []byte, mut src []byte) {
2019-07-26 16:48:49 +02:00
if src.len == 0 {
return
}
if subtle.inexact_overlap(dst, src) {
2019-07-26 16:48:49 +02:00
panic('crypto.rc4: invalid buffer overlap')
}
mut i := c.i
mut j := c.j
for k, v in src {
i += byte(1)
2019-07-26 16:48:49 +02:00
x := c.s[i]
j += byte(x)
2019-07-26 16:48:49 +02:00
y := c.s[j]
c.s[i] = y
c.s[j] = x
dst[k] = v ^ byte(c.s[byte(x + y)])
2019-07-26 16:48:49 +02:00
}
c.i = i
c.j = j
}