Refactor RC4 implementation

This commit is contained in:
shenjack-5600u 2024-05-02 12:29:52 +08:00
parent fe46122537
commit 6bc2637263
Signed by: shenjack
GPG Key ID: FDF9864E11C7E79F

View File

@ -88,7 +88,8 @@ impl RC4 {
self.i = (self.i + 1) & 255;
self.j = (self.j + self.main_val[self.i as usize] as u32) & 255;
self.main_val.swap(self.i as usize, self.j as usize);
*byte ^= self.main_val[(self.main_val[self.i as usize] as u32 + self.main_val[self.j as usize] as u32) as usize & 255];
*byte ^=
self.main_val[(self.main_val[self.i as usize] as u32 + self.main_val[self.j as usize] as u32) as usize & 255];
}
}
@ -113,7 +114,8 @@ impl RC4 {
self.i = (self.i + 1) & 255;
self.j = (self.j + self.main_val[self.i as usize] as u32) & 255;
self.main_val.swap(self.i as usize, self.j as usize);
*byte ^= self.main_val[(self.main_val[self.i as usize] as u32 + self.main_val[self.j as usize] as u32) as usize & 255];
*byte ^=
self.main_val[(self.main_val[self.i as usize] as u32 + self.main_val[self.j as usize] as u32) as usize & 255];
self.j = (self.j + *byte as u32) & 255;
}
}
@ -141,7 +143,8 @@ impl RC4 {
self.j = (self.j + self.main_val[self.i as usize] as u32) & 255;
self.main_val.swap(self.i as usize, self.j as usize);
let byte_v = *byte;
*byte ^= self.main_val[(self.main_val[self.i as usize] as u32 + self.main_val[self.j as usize] as u32) as usize & 255];
*byte ^=
self.main_val[(self.main_val[self.i as usize] as u32 + self.main_val[self.j as usize] as u32) as usize & 255];
self.j = (self.j + byte_v as u32) & 255;
}
}
@ -155,7 +158,7 @@ impl RC4 {
/// S[i] = S[j];
/// S[j] = t;
/// return S[(S[i] + S[j]) & 255];
///}
/// }
/// ```
pub fn next_u8(&mut self) -> u8 {
self.i = (self.i + 1) & 255;
@ -179,7 +182,6 @@ impl RC4 {
/// } while (round !== 0)
/// return n
/// }
///
/// ```
pub fn next_i32(&mut self, max: i32) -> i32 {
if max == 0 {