summaryrefslogtreecommitdiffstats
path: root/lib/aes.go
diff options
context:
space:
mode:
authorrsiddharth <s@ricketyspace.net>2021-03-03 20:14:16 -0500
committerrsiddharth <s@ricketyspace.net>2021-03-03 20:14:16 -0500
commit51cce8132d7bd21f51613a1a78ea6652ffb222f4 (patch)
tree7c7683bb027aa34a1f7e1224808374afe592ac9e /lib/aes.go
parente9f7ddf6c40844043d517931e157f5c53f8d36b4 (diff)
lib: add AESGenCTRFunc
Diffstat (limited to 'lib/aes.go')
-rw-r--r--lib/aes.go23
1 files changed, 23 insertions, 0 deletions
diff --git a/lib/aes.go b/lib/aes.go
index 82ee8ed..f3fd6d9 100644
--- a/lib/aes.go
+++ b/lib/aes.go
@@ -3,6 +3,29 @@
package lib
+// Generates counter function for AES CTR mode.
+func AESGenCTRFunc(nonce uint64) func() []byte {
+ ctr := uint64(0) // Counter
+ ff := uint64(0xFF)
+ cf := func() []byte {
+ cb := make([]byte, 16) // counter block
+ var i, j uint
+ // Put nonce in the first 8 bytes in cb in little endian format
+ for i = 0; i < 8; i++ {
+ n := nonce & (ff << (i * 8)) // Reset all except the i^th byte of the nonce
+ cb[i] = byte(n >> (i * 8)) // Retrieve i^th byte of the nonce
+ }
+ // Put counter in the next 8 bytes in cb in little endian format
+ for i, j = 8, 0; i < 16; i, j = i+1, j+1 {
+ n := ctr & (ff << (j * 8)) // Reset all except the j^th byte of the counter
+ cb[i] = byte(n >> (j * 8)) // Retrieve j^th byte of the counter
+ }
+ ctr += 1 // Increment counter by 1
+ return cb
+ }
+ return cf
+}
+
func AESEncryptCTR(plain, key []byte, ctrFunc func() []byte) ([]byte, error) {
if len(key) != 16 {
return []byte{}, CPError{"key length != 16"}