First commit

Author: jdieguez89

.gitignore

@@ -13,3 +13,4 @@
 
 # Dependency directories (remove the comment below to include it)
 # vendor/
+.idea

atlas_aes.go

@@ -0,0 +1,69 @@
+package AtlasInsideAES
+
+import (
+	"bytes"
+	"crypto/aes"
+	"crypto/cipher"
+	"crypto/sha1"
+	b64 "encoding/base64"
+	"golang.org/x/crypto/pbkdf2"
+)
+
+const IterationCount = 65536
+const SaltLength = 16
+const KeyLength = 16
+
+func setKey(key []byte) (cipher.Block, []byte, error) {
+	h := sha1.New()
+	h.Write(key)
+	salt := h.Sum(nil)
+	keyEnc := pbkdf2.Key(key, salt, IterationCount, KeyLength, sha1.New)
+	block, err := aes.NewCipher(keyEnc)
+	if err != nil {
+		return nil, nil, err
+	}
+	return block, salt[:SaltLength], nil
+}
+
+func AESEncrypt(src string, key []byte) (string, error) {
+	if len(src) == 0 {
+		return "", &InvalidEncryptedDataError{"Invalid crypto"}
+	}
+	blkEncrypt, ivEncrypt, err := setKey(key)
+	if err != nil {
+		return "", &InvalidAESKeyError{"Invalid crypto"}
+	}
+	ecb := cipher.NewCBCEncrypter(blkEncrypt, ivEncrypt)
+	content := []byte(src)
+	content = PKCS5Padding(content, blkEncrypt.BlockSize())
+	crypted := make([]byte, len(content))
+	ecb.CryptBlocks(crypted, content)
+	base64 := b64.StdEncoding.EncodeToString(crypted)
+	return base64, nil
+}
+
+func AESDecrypt(crypt string, key []byte) (string, error) {
+	encryptedData, _ := b64.StdEncoding.DecodeString(crypt)
+	if len(crypt) == 0 {
+		return "", &InvalidPassphraseError{"Invalid crypto"}
+	}
+	blk, iv, err := setKey(key)
+	if err != nil {
+		return "", &InvalidAESKeyError{"Invalid crypto"}
+	}
+	ecb := cipher.NewCBCDecrypter(blk, iv)
+	decrypted := make([]byte, len(encryptedData))
+	ecb.CryptBlocks(decrypted, encryptedData)
+	return string(PKCS5Trimming(decrypted)), nil
+}
+
+func PKCS5Padding(ciphertext []byte, blockSize int) []byte {
+	padding := blockSize - len(ciphertext)%blockSize
+	padtext := bytes.Repeat([]byte{byte(padding)}, padding)
+	return append(ciphertext, padtext...)
+}
+
+func PKCS5Trimming(encrypt []byte) []byte {
+	padding := encrypt[len(encrypt)-1]
+	return encrypt[:len(encrypt)-int(padding)]
+}

errors.go

@@ -0,0 +1,63 @@
+package AtlasInsideAES
+
+type InvalidPassphraseError struct {
+	Msg string
+}
+
+func (err *InvalidPassphraseError) Error() string {
+	return "Invalid encryption or decryption passphrase: " + err.Msg
+}
+
+type InvalidEncryptedDataError struct {
+	Msg string
+}
+
+func (err *InvalidEncryptedDataError) Error() string {
+	return "Invalid encrypted data: " + err.Msg
+}
+
+type CouldNotObtainRandomSaltError struct {
+	Msg string
+}
+
+func (err *CouldNotObtainRandomSaltError) Error() string {
+	return "Could not obtain random salt: " + err.Msg
+}
+
+type CouldNotObtainRandomIVError struct {
+	Msg string
+}
+
+func (err *CouldNotObtainRandomIVError) Error() string {
+	return "Could not obtain random IV: " + err.Msg
+}
+
+type InvalidAESKeyError struct {
+	Msg string
+}
+
+func (err *InvalidAESKeyError) Error() string {
+	return "Invalid AES key: " + err.Msg
+}
+
+type InvalidIVError struct {
+	Msg string
+}
+
+func (err *InvalidIVError) Error() string {
+	return "Invalid IV: " + err.Msg
+}
+
+type InvalidSaltError struct {
+	Msg string
+}
+
+func (err *InvalidSaltError) Error() string {
+	return "Invalid Salt: " + err.Msg
+}
+
+type IncorrectPassphraseError struct{}
+
+func (err *IncorrectPassphraseError) Error() string {
+	return "Invalid decryption passphrase"
+}

go.mod

@@ -0,0 +1,5 @@
+module github.com/AtlasInsideCorp/AtlasInsideAES
+
+go 1.17
+
+require golang.org/x/crypto v0.0.0-20220331220935-ae2d96664a29

go.sum

@@ -0,0 +1,9 @@
+golang.org/x/crypto v0.0.0-20220331220935-ae2d96664a29 h1:tkVvjkPTB7pnW3jnid7kNyAMPVWllTNOf/qKDze4p9o=
+golang.org/x/crypto v0.0.0-20220331220935-ae2d96664a29/go.mod h1:IxCIyHEi3zRg3s0A5j5BB6A9Jmi73HwBIUl50j+osU4=
+golang.org/x/net v0.0.0-20211112202133-69e39bad7dc2/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=
+golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
+golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
+golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=