Create & Init Project...

app/interface/main/report-click/service/crypto/aes/BUILD

@@ -0,0 +1,44 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_test",
+    "go_library",
+)
+
+go_test(
+    name = "go_default_test",
+    srcs = ["aes_test.go"],
+    embed = [":go_default_library"],
+    rundir = ".",
+    tags = ["automanaged"],
+    deps = [
+        "//app/interface/main/report-click/service/crypto/padding:go_default_library",
+        "//vendor/github.com/smartystreets/goconvey/convey:go_default_library",
+    ],
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = ["aes.go"],
+    importpath = "go-common/app/interface/main/report-click/service/crypto/aes",
+    tags = ["automanaged"],
+    deps = [
+        "//app/interface/main/report-click/service/crypto/cipher:go_default_library",
+        "//app/interface/main/report-click/service/crypto/padding:go_default_library",
+    ],
+)
+
+filegroup(
+    name = "package-srcs",
+    srcs = glob(["**"]),
+    tags = ["automanaged"],
+    visibility = ["//visibility:private"],
+)
+
+filegroup(
+    name = "all-srcs",
+    srcs = [":package-srcs"],
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+)

app/interface/main/report-click/service/crypto/aes/aes.go

@@ -0,0 +1,103 @@
+package aes
+
+import (
+	iaes "crypto/aes"
+	icipher "crypto/cipher"
+	"errors"
+
+	"go-common/app/interface/main/report-click/service/crypto/cipher"
+	"go-common/app/interface/main/report-click/service/crypto/padding"
+)
+
+var (
+	ErrAesBlockSize = errors.New("plaintext is not a multiple of the block size")
+	ErrAesSrcSize   = errors.New("ciphertext too short")
+	ErrAesIVSize    = errors.New("iv size is not a block size")
+)
+
+// ECBEncrypt aes ecb encrypt.
+func ECBEncrypt(src, key []byte, p padding.Padding) ([]byte, error) {
+	if p == nil {
+		if len(src) < iaes.BlockSize || len(src)%iaes.BlockSize != 0 {
+			return nil, ErrAesBlockSize
+		}
+	} else {
+		src = p.Padding(src, iaes.BlockSize)
+	}
+	b, err := iaes.NewCipher(key)
+	if err != nil {
+		return nil, err
+	}
+	mode := cipher.NewECBEncrypter(b)
+	encryptText := make([]byte, len(src))
+	mode.CryptBlocks(encryptText, src)
+	return encryptText, nil
+}
+
+// ECBDecrypt aes ecb decrypt.
+func ECBDecrypt(src, key []byte, p padding.Padding) ([]byte, error) {
+	if len(src) < iaes.BlockSize || len(src)%iaes.BlockSize != 0 {
+		return nil, ErrAesSrcSize
+	}
+	b, err := iaes.NewCipher(key)
+	if err != nil {
+		return nil, err
+	}
+	mode := cipher.NewECBDecrypter(b)
+	decryptText := make([]byte, len(src))
+	mode.CryptBlocks(decryptText, src)
+	if p == nil {
+		return decryptText, nil
+	} else {
+		return p.Unpadding(decryptText, iaes.BlockSize)
+	}
+}
+
+// CBCEncrypt aes cbc encrypt.
+func CBCEncrypt(src, key, iv []byte, p padding.Padding) ([]byte, error) {
+	// check iv
+	if len(iv) != iaes.BlockSize {
+		return nil, ErrAesIVSize
+	}
+	if p == nil {
+		// if no padding check src
+		if len(src) < iaes.BlockSize || len(src)%iaes.BlockSize != 0 {
+			return nil, ErrAesSrcSize
+		}
+	} else {
+		// padding
+		src = p.Padding(src, iaes.BlockSize)
+	}
+	block, err := iaes.NewCipher(key)
+	if err != nil {
+		return nil, err
+	}
+	mode := icipher.NewCBCEncrypter(block, iv)
+	encryptText := make([]byte, len(src))
+	mode.CryptBlocks(encryptText, src)
+	return encryptText, nil
+}
+
+// CBCDecrypt aes cbc decrypt.
+func CBCDecrypt(src, key, iv []byte, p padding.Padding) ([]byte, error) {
+	// check src
+	if len(src) < iaes.BlockSize || len(src)%iaes.BlockSize != 0 {
+		return nil, ErrAesSrcSize
+	}
+	// check iv
+	if len(iv) != iaes.BlockSize {
+		return nil, ErrAesIVSize
+	}
+	block, err := iaes.NewCipher(key)
+	if err != nil {
+		return nil, err
+	}
+	mode := icipher.NewCBCDecrypter(block, iv)
+	decryptText := make([]byte, len(src))
+	mode.CryptBlocks(decryptText, src)
+	if p == nil {
+		return decryptText, nil
+	} else {
+		return p.Unpadding(decryptText, iaes.BlockSize)
+	}
+}

app/interface/main/report-click/service/crypto/aes/aes_test.go

@@ -0,0 +1,24 @@
+package aes
+
+import (
+	"testing"
+
+	"go-common/app/interface/main/report-click/service/crypto/padding"
+
+	. "github.com/smartystreets/goconvey/convey"
+)
+
+func TestAES(t *testing.T) {
+	var (
+		aesKey = ""
+		aesIv  = ""
+		bs     = []byte("this is  test massage ")
+	)
+	Convey("aes test ", t, func() {
+		bs, _ = CBCDecrypt(bs, []byte(aesKey), []byte(aesIv), padding.PKCS5)
+		ECBDecrypt(bs, []byte(aesKey), padding.PKCS5)
+		bs = []byte("this is  test massage ")
+		bs, _ = CBCEncrypt(bs, []byte(aesKey), []byte(aesIv), padding.PKCS5)
+		CBCDecrypt(bs, []byte(aesKey), []byte(aesIv), padding.PKCS5)
+	})
+}