rpm&deb

lib/global/global.go

@@ -52,6 +52,7 @@ var (
 	ErrCMNotFoundRollbackNeVersion = errors.New("not found the rollback NE version")
 
 	ErrCMNotFoundTargetBackupFile = errors.New("not found the target NE backup")
+	ErrCMUnknownSoftwareFormat    = errors.New("unknown software package format") // 未知软件包格式
 
 	// TRACE module error message
 	ErrTraceFailedDistributeToNEs = errors.New("failed to distribute trace task to target NEs")

lib/global/kits.go

@@ -585,3 +585,31 @@ func ZipDirectoryFile(srcDir, dstZip string) error {
 
 	return err
 }
+
+// 判断软件包是rpm或者deb, 1:rpm, 2:deb, 0:unknown format
+func JudgeRpmOrDebPackage(filePath string) (int, error) {
+	var fileType int = 0
+	file, err := os.Open(filePath)
+	if err != nil {
+		return fileType, err
+	}
+	defer file.Close()
+
+	// Read the first 6 bytes of the file
+	header := make([]byte, 6)
+	_, err = file.Read(header)
+	if err != nil {
+		return fileType, err
+	}
+
+	// Check the magic numbers to determine the package format
+	if string(header) == "!<arch>" {
+		fileType = 1
+	} else if string(header) == "!<arch\n" || string(header) == "!<arch\r" {
+		fileType = 2
+	} else {
+		fileType = 3
+	}
+
+	return fileType, nil
+}