Merge pull request #2254 from will4j/feature/os-iso-checksum

feature/support checksum of downloaded repository iso file

cmd/kk/apis/kubekey/v1alpha2/manifest_types.go

@@ -24,6 +24,7 @@ import (
 type Iso struct {
 	LocalPath string `yaml:"localPath" json:"localPath"`
 	Url       string `yaml:"url" json:"url"`
+	Checksum  string `yaml:"checksum" json:"checksum"`
 }
 
 type Repository struct {

cmd/kk/pkg/artifact/tasks.go

@@ -30,6 +30,7 @@ import (
 	"github.com/kubesphere/kubekey/v3/cmd/kk/pkg/core/connector"
 	"github.com/kubesphere/kubekey/v3/cmd/kk/pkg/core/logger"
 	coreutil "github.com/kubesphere/kubekey/v3/cmd/kk/pkg/core/util"
+	"github.com/kubesphere/kubekey/v3/cmd/kk/pkg/files"
 )
 
 type DownloadISOFile struct {
@@ -44,6 +45,17 @@ func (d *DownloadISOFile) Execute(runtime connector.Runtime) error {
 
 		fileName := fmt.Sprintf("%s-%s-%s.iso", sys.Id, sys.Version, sys.Arch)
 		filePath := filepath.Join(runtime.GetWorkDir(), fileName)
+
+		checksumEqual, err := files.SHA256CheckEqual(filePath, sys.Repository.Iso.Checksum)
+		if err != nil {
+			return err
+		}
+		if checksumEqual {
+			logger.Log.Infof("Skip download exists iso file %s", fileName)
+			d.Manifest.Spec.OperatingSystems[i].Repository.Iso.LocalPath = filePath
+			continue
+		}
+
 		getCmd := d.Manifest.Arg.DownloadCommand(filePath, sys.Repository.Iso.Url)
 
 		cmd := exec.Command("/bin/sh", "-c", getCmd)

cmd/kk/pkg/files/file.go

@@ -328,6 +328,23 @@ func (b *KubeBinary) SHA256Check() error {
 	return nil
 }
 
+func SHA256CheckEqual(filePath string, checksum string) (bool, error) {
+	if strings.TrimSpace(checksum) == "" {
+		return false, nil
+	}
+
+	if !util.IsExist(filePath) {
+		return false, nil
+	}
+
+	output, err := sha256sum(filePath)
+	if err != nil {
+		return false, errors.Wrap(err, fmt.Sprintf("Failed to check SHA256 of %s", filePath))
+	}
+
+	return output == checksum, nil
+}
+
 func sha256sum(path string) (string, error) {
 	file, err := os.Open(path)
 	if err != nil {