lib/aes: add aesShiftRows

1 files changed, 14 insertions, 0 deletions

diff --git a/lib/aes.go b/lib/aes.go
index 9440b52..a388db8 100644
--- a/lib/aes.go
+++ b/lib/aes.go
@@ -98,6 +98,20 @@ func aesInvSubBytes(state [][]byte) [][]byte {
 	return state
 }
 
+func aesShiftRows(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] = state[r][(c+r)%nb]
+
+		}
+	}
+	return n_state
+}
+
 func aesInvShiftRows(state [][]byte) [][]byte {
 	n_state := make([][]byte, 4) // New state.