diff --git a/internal/rbd/rbd_util.go b/internal/rbd/rbd_util.go index 64d786d734da..795b4511b546 100644 --- a/internal/rbd/rbd_util.go +++ b/internal/rbd/rbd_util.go @@ -697,47 +697,59 @@ func (ri *rbdImage) trashRemoveImage(ctx context.Context) error { return nil } +func getImageSpec(spec librbd.ImageSpec) string { + if spec.Trash { + return spec.ImageID + } + + if spec.PoolNamespace != "" { + return fmt.Sprintf("%s/%s/%s", spec.PoolName, spec.PoolNamespace, spec.ImageName) + } + + return fmt.Sprintf("%s/%s", spec.PoolName, spec.ImageName) +} + +// getCloneDepth walks the parents of the image and returns the number of +// images in the chain. +// +// This function re-uses the ioctx of the image to open all images in the +// chain. There is no need to open new ioctx's for every image. func (ri *rbdImage) getCloneDepth(ctx context.Context) (uint, error) { - var depth uint - vol := rbdVolume{} + var ( + depth uint + parent librbd.ImageSpec + ) - vol.Pool = ri.Pool - vol.Monitors = ri.Monitors - vol.RbdImageName = ri.RbdImageName - vol.RadosNamespace = ri.RadosNamespace - vol.conn = ri.conn.Copy() + image, err := ri.open() + if err != nil { + return 0, fmt.Errorf("failed to open image %s: %w", ri, err) + } + defer image.Close() for { - if vol.RbdImageName == "" { - return depth, nil - } - err := vol.openIoctx() - if err != nil { - return depth, err + // get the librbd.ImageSpec of the parent + info, err := image.GetParent() + if errors.Is(err, librbd.ErrNotFound) { + // image does not have more parents + break + } else if err != nil { + return 0, fmt.Errorf("failed to get parent of image %s: %w", ri, err) } - err = vol.getImageInfo() - // FIXME: create and destroy the vol inside the loop. - // see https://github.com/ceph/ceph-csi/pull/1838#discussion_r598530807 - vol.ioctx.Destroy() - vol.ioctx = nil - if err != nil { - // if the parent image is moved to trash the name will be present - // in rbd image info but the image will be in trash, in that case - // return the found depth - if errors.Is(err, ErrImageNotFound) { - return depth, nil - } - log.ErrorLog(ctx, "failed to check depth on image %s: %s", &vol, err) + // if there is a parent, count it to the depth + depth++ - return depth, err - } - if vol.ParentName != "" { - depth++ + // open the parent image, so that the for-loop can continue + // with checking for the parent of the parent + parent = info.Image + image, err = librbd.OpenImageById(ri.ioctx, parent.ImageID, librbd.NoSnapshot) + if err != nil { + return 0, fmt.Errorf("failed to open parent image ID %q, at depth %d: %w", parent.ImageID, depth, err) } - vol.RbdImageName = vol.ParentName - vol.Pool = vol.ParentPool + defer image.Close() } + + return depth, nil } type trashSnapInfo struct {