summaryrefslogtreecommitdiffstats
path: root/lib
diff options
context:
space:
mode:
Diffstat (limited to 'lib')
-rw-r--r--lib/aes.go13
1 files changed, 13 insertions, 0 deletions
diff --git a/lib/aes.go b/lib/aes.go
index 389dc44..7d98a60 100644
--- a/lib/aes.go
+++ b/lib/aes.go
@@ -3,6 +3,19 @@
package lib
+func InvShiftRows(state [][]byte) [][]byte {
+ n_state := make([][]byte, 4) // New state.
+
+ nb := 4
+ for r := 0; r < 4; r++ {
+ n_state[r] = make([]byte, nb)
+ for c := 0; c < nb; c++ {
+ n_state[r][(c+r)%nb] = state[r][c]
+ }
+ }
+ return n_state
+}
+
func AddRoundKey(state, ks [][]byte) [][]byte {
if len(ks) != 4 {
return state