From fd4a6bc9794c3138b322753c8ce1b628c7b3e86b Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@iotlink.com>
Date: 星期四, 21 十一月 2019 18:27:33 +0800
Subject: [PATCH] fix: add prefix on create key file

---
 licence.go |  154 ++++++++++++++++++++++++++++++++++++---------------
 1 files changed, 108 insertions(+), 46 deletions(-)

diff --git a/licence.go b/licence.go
index 6469996..31d5799 100644
--- a/licence.go
+++ b/licence.go
@@ -1,85 +1,147 @@
 package licence
 
 import (
+	"encoding/base64"
 	"encoding/json"
-	"strings"
+	"io/ioutil"
+	"os"
 	"time"
-
-	"github.com/shirou/gopsutil/cpu"
-	// "github.com/shirou/gopsutil/disk"
-	"github.com/shirou/gopsutil/host"
 )
 
-type Licence struct {
+type RegisterCode struct {
 	MachineCode string
-	Expires     int64
+	Company     string
+	Email       string
+	Phone       string
+	Version     string
+}
+
+type Licence struct {
+	RegCode   RegisterCode
+	Expires   int64
+	Timestamp int64
 }
 
 const (
-	ValidationErrorMalformed        uint32 = iota + 1 // Licence is malformed
-	ValidationErrorUnverifiableHost                   // Licence could not be verified because of signing problems
-	ValidationErrorExpired                            // Signature validation failed
+	ValidationErrorDecrypt          uint32 = iota + 1 // Licence is malformed
+	ValidationErrorUnverifiableHost                   // Host unauthorized
+	ValidationErrorExpired                            // Signature expired
+	ValidationErrorMalformed
+
+	Version = "1.0.1"
+	aesKey  = "www.aiotlink.com"
 )
 
-func GetMachineCode() string {
-	var machineCode string
+func GetRegisterCode(company, email, phone string) string {
+	mCode := GetMachineCode()
 
-	// CPU
-	if cpu, err := cpu.Info(); err == nil {
-		for _, c := range cpu {
-			strings.Join([]string{machineCode, c.String()}, "-")
-		}
-	}
+	regCode := RegisterCode{mCode, company, email, phone, Version}
 
-	// // Disk
-	// if diskInfo, err := disk.Partitions(false); err == nil {
-	// 	for _, d := range diskInfo {
-	// 		diskSerialNumber := disk.GetDiskSerialNumber(d.Device)
-	// 		strings.Join([]string{machineCode, diskSerialNumber}, "-")
-	// 	}
-	// }
+	json, _ := json.Marshal(regCode)
 
-	// Host
-	if host, err := host.Info(); err == nil {
-		strings.Join([]string{machineCode, host.HostID}, "-")
-	}
-
-	return GetMd5String(machineCode, true, false)
+	return AESEncodeStr(json, aesKey)
 }
 
-func GenerateLicence(machineCode, timeOut, key string) string {
+func DecryptRegisterCode(regCode string) []byte {
+	return AESDecodeStr(regCode, aesKey)
+}
+
+func GenerateLicence(regCode, timeOut string, privateKey []byte) (string, error) {
 	timeLayout := "2006-01-02 15:04:05"  //杞寲鎵�闇�妯℃澘
 	loc, _ := time.LoadLocation("Local") //鑾峰彇鏃跺尯
 	tmp, _ := time.ParseInLocation(timeLayout, timeOut, loc)
-	timestamp := tmp.Unix()
+	exp := tmp.Unix()
+	now := time.Now().Unix()
 
-	licence := Licence{machineCode, timestamp}
+	regCodeText := AESDecodeStr(regCode, aesKey)
+	var registerCode RegisterCode
+	if err := json.Unmarshal(regCodeText, &registerCode); err != nil {
+		return "", err
+	}
 
-	json, _ := json.Marshal(licence)
-	return AESEncodeStr(json, key)
+	licence := Licence{registerCode, exp, now}
+	json, err := json.Marshal(licence)
+	if err != nil {
+		return "", err
+	}
+
+	RSA := &RSASecurity{}
+	if err := RSA.SetPrivateKey(privateKey); err != nil {
+		return "", err
+	}
+
+	licenceHex, err := RSA.PriKeyENCTYPT(json)
+	if err != nil {
+		return "", err
+	}
+
+	licenceText := base64.StdEncoding.EncodeToString(licenceHex)
+
+	return licenceText, nil
 }
 
-func VerifyLicence(licenceCode, key string) uint32 {
-	decodeData := AESDecodeStr(licenceCode, key)
-	if decodeData == nil {
-		return ValidationErrorMalformed
+func DecryptLicence(licencePath, publicKeyPath string) ([]byte, error) {
+	var publicKey, licenceCode []byte
+	var fdLic, fdPub *os.File
+	var err error
+
+	// 璇诲彇Licence File
+	fdLic, err = os.Open(licencePath)
+	if err != nil {
+		return nil, err
+	}
+	defer fdLic.Close()
+
+	licenceCode, err = ioutil.ReadAll(fdLic)
+	if err != nil {
+		return nil, err
+	}
+
+	// 璇诲彇鍏挜
+	fdPub, err = os.Open(publicKeyPath)
+	if err != nil {
+		return nil, err
+	}
+	defer fdPub.Close()
+
+	publicKey, err = ioutil.ReadAll(fdPub)
+	if err != nil {
+		return nil, err
+	}
+
+	RSA := &RSASecurity{}
+	if err := RSA.SetPublicKey(publicKey); err != nil {
+		return nil, err
+	}
+
+	licenceHex, _ := base64.StdEncoding.DecodeString(string(licenceCode))
+
+	return RSA.PubKeyDECRYPT(licenceHex)
+}
+
+func VerifyLicence(licencePath, publicKeyPath string) uint32 {
+	licenceText, err := DecryptLicence(licencePath, publicKeyPath)
+	if err != nil {
+		return ValidationErrorDecrypt
 	}
 
 	var licence Licence
 
-	if err := json.Unmarshal(decodeData, &licence); err != nil {
+	if err := json.Unmarshal(licenceText, &licence); err != nil {
 		return ValidationErrorMalformed
 	}
 
-	code := GetMachineCode()
-	if licence.MachineCode != code {
-		return ValidationErrorUnverifiableHost
-	}
-
+	// 鍒ゆ柇杩囨湡
 	now := time.Now().Unix()
 	if now > licence.Expires {
 		return ValidationErrorExpired
 	}
 
+	// 鍒ゆ柇鏈哄櫒鐮�
+	mCode := GetMachineCode()
+	if licence.RegCode.MachineCode != mCode {
+		return ValidationErrorUnverifiableHost
+	}
+
 	return 0
 }

--
Gitblit v1.8.0