feat: 设置序列号

cmd/agent/core/agent.go

@@ -139,17 +139,21 @@ func genVersion() string {
 
 func initSvcCfg() error {
 	// 异常情况处理
-	if _, err := os.Stat(global.SelfOperatedAndroid); os.IsExist(err) {
+	if info, err := os.Stat(global.SelfOperatedAndroid); err == nil && info != nil && !info.IsDir() {
 		l.Infof("the host is a self operated Android device")
 		for {
 			time.Sleep(24 * time.Hour)
 		}
 	}
-	// TODO: 序列号设置
+	// 序列号设置
 	if option.Opt.SerialNumber != "" {
 		global.DeviceSerialNumber = option.Opt.SerialNumber
 	} else {
-
+		sn, err := global.GetDeviceID()
+		if err != nil {
+			return err
+		}
+		global.DeviceSerialNumber = sn
 	}
 	global.SubscribePrefixInfo = global.DeviceSerialNumber + "/publish/"
 

cmd/agent/global/service.go

 package global
 
+import (
+	"crypto/hmac"
+	"crypto/sha256"
+	"crypto/sha512"
+	"encoding/hex"
+	"fmt"
+	"os"
+	"path"
+	"path/filepath"
+	"strings"
+
+	"agent/cmd/agent/config"
+	"agent/cmd/agent/option"
+	"linkfog.com/public/lib/l"
+
+	"github.com/gofrs/uuid"
+)
+
 const (
 	DeviceIDFile         = ".deviceID"
 	DeviceRegisterFIle   = ".reg2lw.lock"
@@ -21,5 +39,49 @@ const (
 	Dcs9942URL             = "https://newdcs.bkdomain.cn:9992"
 	DcsUploadFlowDeviceAPI = "/api/v4/sds/uploadFlowDevice"
 
-	HmacKey = "4N][zmn;bPW!a>F~iU+["
+	HmacKey  = "4N][zmn;bPW!a>F~iU+["
+	HmacKey2 = "semigroupal"
 )
+
+func GetDeviceID() (string, error) {
+	sn := ""
+	if _, err := os.Stat(path.Join(config.WorkDir, DeviceIDFile)); os.IsNotExist(err) {
+		l.Warnf("the device has been registered and needs to be re registered")
+		sn, err = produceDeviceID()
+		if err != nil {
+			return "", err
+		}
+	} else {
+		content, err := os.ReadFile(path.Join(config.WorkDir, DeviceIDFile))
+		if err != nil {
+			return "", err
+		}
+		sn = strings.TrimSpace(string(content))
+	}
+
+	return sn, nil
+}
+
+func produceDeviceID() (string, error) {
+	if option.Opt.Vendor == "" {
+		return "", fmt.Errorf("first registration requires setting up a vendor")
+	}
+	//create sn
+	uuidV4, _ := uuid.NewV4()
+	uuid32 := uuidV4.String()
+	shaSum256 := fmt.Sprintf("%x", sha256.Sum256([]byte(uuid32)))
+	shaSum256 = strings.ToUpper(shaSum256[:12])
+	mac := hmac.New(sha512.New, []byte(HmacKey2))
+	mac.Write([]byte("32_" + option.Opt.Vendor + "_" + shaSum256))
+	last4 := hex.EncodeToString(mac.Sum(nil))[:4]
+	tmp := "32_" + option.Opt.Vendor + "_" + shaSum256 + last4 + "-DISKSN"
+	snByte := []byte(strings.ToUpper(tmp))
+
+	//generate sn file
+	err := os.WriteFile(filepath.Join(config.WorkDir, DeviceIDFile), snByte, 0666)
+	if err != nil {
+		return "", fmt.Errorf("failed to write device id file, err:%v", err)
+	}
+
+	return string(snByte), nil
+}

cmd/agent/option/option.go

@@ -38,6 +38,7 @@ type options struct {
 	MonitorDefaultBackupFileName string
 
 	SerialNumber string
+	Vendor       string
 
 	// 调试信息
 	MQTTBrokerURL  string
@@ -66,7 +67,8 @@ const (
 	ArgMonitorDefaultMaxRetryTimes  = "monitor-default-max-retry-times"
 	ArgMonitorDefaultBackupFileName = "monitor-default-backup-file-name"
 
-	ArgSerialNumber = "SN"
+	ArgSerialNumber = "sn"
+	ArgVendor       = "vendor"
 )
 
 func flagSet() {
@@ -96,6 +98,7 @@ func flagSet() {
 	flag.StringVar(&Opt.MonitorDefaultBackupFileName, ArgMonitorDefaultBackupFileName, "agent.bak", "agent file backup")
 
 	flag.StringVar(&Opt.SerialNumber, ArgSerialNumber, "", "serial number")
+	flag.StringVar(&Opt.Vendor, ArgVendor, "", "vendor")
 }
 
 func Parse() {

go.mod

@@ -9,6 +9,7 @@ replace linkfog.com/pluginx => web.lueluesay.top/git/lil/pluginx v0.0.0-20241017
 require (
 	github.com/beevik/ntp v1.4.3
 	github.com/eclipse/paho.mqtt.golang v1.5.0
+	github.com/gofrs/uuid v4.4.0+incompatible
 	github.com/judwhite/go-svc v1.2.1
 	github.com/olekukonko/tablewriter v0.0.5
 	github.com/prometheus/procfs v0.15.1

go.sum

@@ -6,6 +6,8 @@ github.com/eclipse/paho.mqtt.golang v1.5.0 h1:EH+bUVJNgttidWFkLLVKaQPGmkTUfQQqjO
 github.com/eclipse/paho.mqtt.golang v1.5.0/go.mod h1:du/2qNQVqJf/Sqs4MEL77kR8QTqANF7XU7Fk0aOTAgk=
 github.com/go-ole/go-ole v1.2.6 h1:/Fpf6oFPoeFik9ty7siob0G6Ke8QvQEuVcuChpwXzpY=
 github.com/go-ole/go-ole v1.2.6/go.mod h1:pprOEPIfldk/42T2oK7lQ4v4JSDwmV0As9GaiUsvbm0=
+github.com/gofrs/uuid v4.4.0+incompatible h1:3qXRTX8/NbyulANqlc0lchS1gqAVxRgsuW1YrTJupqA=
+github.com/gofrs/uuid v4.4.0+incompatible/go.mod h1:b2aQJv3Z4Fp6yNu3cdSllBxTCLRxnplIgP/c0N/04lM=
 github.com/golang/protobuf v1.5.4 h1:i7eJL8qZTpSEXOPTxNKhASYpMn+8e5Q6AdndVa1dWek=
 github.com/golang/protobuf v1.5.4/go.mod h1:lnTiLA8Wa4RWRcIUkrtSVa5nRhsEGBg48fD6rSs7xps=
 github.com/google/go-cmp v0.6.0 h1:ofyhxvXcZhMsU5ulbFiLKl/XBFqE1GSq7atu8tAmTRI=