feat(lvol): add snapshot checksum APIs

app/cmd/basic/bdev_lvol.go

@@ -33,6 +33,8 @@
 			BdevLvolGetXattrCmd(),
 			BdevLvolGetFragmapCmd(),
 			BdevLvolRenameCmd(),
+			BdevLvolRegisterSnapshotChecksumCmd(),
+			BdevLvolGetSnapshotChecksumCmd(),
 		},
 	}
 }
@@ -654,3 +656,96 @@
 
 	return util.PrintObject(renamed)
 }
+
+func BdevLvolRegisterSnapshotChecksumCmd() cli.Command {
+	return cli.Command{
+		Name: "register-snapshot-checksum",
+		Flags: []cli.Flag{
+			cli.StringFlag{
+				Name:  "alias",
+				Usage: "The alias of a snapshot is <LVSTORE NAME>/<SNAPSHOT NAME>. Specify this or uuid",
+			},
+			cli.StringFlag{
+				Name:  "uuid",
+				Usage: "Specify this or alias",
+			},
+		},
+		Usage: "compute and store checksum of snapshot's data: \"register-snapshot-checksum --alias <LVSTORE NAME>/<LVOL NAME>\"," +
+			" or \"register-snapshot-checksum --uuid <LVOL UUID>\"",
+		Action: func(c *cli.Context) {
+			if err := bdevLvolRegisterSnapshotChecksum(c); err != nil {
+				logrus.WithError(err).Fatalf("Failed to run register snapshot checksum command")
+			}
+		},
+	}
+}
+
+func bdevLvolRegisterSnapshotChecksum(c *cli.Context) error {
+	spdkCli, err := client.NewClient(context.Background())
+	if err != nil {
+		return err
+	}
+
+	name := c.String("alias")
+	if name == "" {
+		name = c.String("uuid")
+	}
+	if name == "" {
+		return fmt.Errorf("either alias or uuid must be provided")
+	}
+
+	registered, err := spdkCli.BdevLvolRegisterSnapshotChecksum(name)
+	if err != nil {
+		return fmt.Errorf("failed to register checksum for snapshot %q: %v", name, err)
+	}
+
+	return util.PrintObject(registered)
+}
+
+func BdevLvolGetSnapshotChecksumCmd() cli.Command {
+	return cli.Command{
+		Name: "get-snapshot-checksum",
+		Flags: []cli.Flag{
+			cli.StringFlag{
+				Name:  "alias",
+				Usage: "The alias of a snapshot is <LVSTORE NAME>/<SNAPSHOT NAME>. Specify this or uuid",
+			},
+			cli.StringFlag{
+				Name:  "uuid",
+				Usage: "Specify this or alias",
+			},
+		},
+		Usage: "get checksum of snapshot's data: \"get-snapshot-checksum --alias <LVSTORE NAME>/<LVOL NAME>\"," +
+			" or \"get-snapshot-checksum --uuid <LVOL UUID>\"",
+		Action: func(c *cli.Context) {
+			if err := bdevLvolGetSnapshotChecksum(c); err != nil {
+				logrus.WithError(err).Fatalf("Failed to run get snapshot checksum command")
+			}
+		},
+	}
+}
+
+func bdevLvolGetSnapshotChecksum(c *cli.Context) error {
+	spdkCli, err := client.NewClient(context.Background())
+	if err != nil {
+		return err
+	}
+
+	name := c.String("alias")
+	if name == "" {
+		name = c.String("uuid")
+	}
+	if name == "" {
+		return fmt.Errorf("either alias or uuid must be provided")
+	}
+
+	checksum, err := spdkCli.BdevLvolGetSnapshotChecksum(name)
+	if err != nil {
+		return fmt.Errorf("failed to get checksum for snapshot %q: %v", name, err)
+	}
+	if checksum == nil {
+		return fmt.Errorf("no checksum found for snapshot %q", name)
+	}
+
+	return util.PrintObject(*checksum)
+}

pkg/spdk/client/basic.go

@@ -502,6 +502,44 @@
 	return &result, nil
 }
 
+// BdevLvolRegisterSnapshotChecksum compute and store checksum of snapshot's data. Overwrite old checksum if already registered.
+//
+//	"name": Required. UUID or alias of the snapshot. The alias of a snapshot is <LVSTORE NAME>/<SNAPSHOT NAME>.
+func (c *Client) BdevLvolRegisterSnapshotChecksum(name string) (registered bool, err error) {
+	req := spdktypes.BdevLvolRegisterSnapshotChecksumRequest{
+		Name: name,
+	}
+
+	cmdOutput, err := c.jsonCli.SendCommandWithLongTimeout("bdev_lvol_register_snapshot_checksum", req)
+	if err != nil {
+		return false, err
+	}
+
+	return registered, json.Unmarshal(cmdOutput, &registered)
+}
+
+// BdevLvolGetSnapshotChecksum gets snapshot's stored checksum. The checksum must has been previously registered.
+//
+//	"name": Required. UUID or alias of the snapshot. The alias of a snapshot is <LVSTORE NAME>/<SNAPSHOT NAME>.
+func (c *Client) BdevLvolGetSnapshotChecksum(name string) (checksum *uint64, err error) {
+	req := spdktypes.BdevLvolGetSnapshotChecksumRequest{
+		Name: name,
+	}
+
+	cmdOutput, err := c.jsonCli.SendCommandWithLongTimeout("bdev_lvol_get_snapshot_checksum", req)
+	if err != nil {
+		return nil, err
+	}
+
+	var snapshotChecksum spdktypes.BdevLvolSnapshotChecksum
+	err = json.Unmarshal(cmdOutput, &snapshotChecksum)
+	if err != nil {
+		return nil, err
+	}
+
+	return &snapshotChecksum.Checksum, nil
+}
+
 // BdevLvolRename renames a logical volume.
 //
 //	"oldName": Required. UUID or alias of the existing logical volume.

pkg/spdk/types/lvol.go

@@ -150,6 +150,18 @@ type BdevLvolRenameRequest struct {
 	NewName string `json:"new_name"`
 }
 
+type BdevLvolRegisterSnapshotChecksumRequest struct {
+	Name string `json:"name"`
+}
+
+type BdevLvolGetSnapshotChecksumRequest struct {
+	Name string `json:"name"`
+}
+
+type BdevLvolSnapshotChecksum struct {
+	Checksum uint64 `json:"checksum"`
+}
+
 func GetLvolAlias(lvsName, lvolName string) string {
 	return fmt.Sprintf("%s/%s", lvsName, lvolName)
 }