Invalidate imageCache if the pinned image from config exists

Signed-off-by: roman-kiselenko <roman.kiselenko.dev@gmail.com>

internal/storage/image.go

@@ -151,6 +151,10 @@ type ImageServer interface {
 	// CandidatesForPotentiallyShortImageName resolves an image name into a set of fully-qualified image names (domain/repo/image:tag|@digest).
 	// It will only return an empty slice if err != nil.
 	CandidatesForPotentiallyShortImageName(systemContext *types.SystemContext, imageName string) ([]RegistryImageReference, error)
+
+	// UpdateImageCache invalidate imageCache if the pinned image from config exists;
+	// this method doesn't change the Pinned attribute, only removes the image item.
+	UpdateImageCache(configPinnedImage []string) error
 }
 
 func parseImageNames(image *storage.Image) (someName *RegistryImageReference, tags []reference.NamedTagged, digests []reference.Canonical, err error) {
@@ -897,6 +901,26 @@ func (st nativeStorageTransport) ResolveReference(ref types.ImageReference) (typ
 	return istorage.ResolveReference(ref)
 }
 
+// UpdateImageCache invalidate imageCache if the pinned image from config exists;
+// this method doesn't change the Pinned attribute, only removes the image item.
+func (svc *imageService) UpdateImageCache(configPinnedImage []string) error {
+	images, err := svc.store.Images()
+	if err != nil {
+		return err
+	}
+	for i := range images {
+		img := &images[i]
+		for _, name := range img.Names {
+			if FilterPinnedImage(name, CompileRegexpsForPinnedImages(configPinnedImage)) {
+				svc.imageCacheLock.Lock()
+				delete(svc.imageCache, img.ID)
+				svc.imageCacheLock.Unlock()
+			}
+		}
+	}
+	return nil
+}
+
 // FilterPinnedImage checks if the given image needs to be pinned
 // and excluded from kubelet's image GC.
 func FilterPinnedImage(image string, pinnedImages []*regexp.Regexp) bool {

pkg/config/reload.go

@@ -20,7 +20,7 @@ func (c *Config) Reload() error {
 	// Reload the config
 	newConfig, err := DefaultConfig()
 	if err != nil {
-		return errors.New("unable to create default config")
+		return fmt.Errorf("unable to create default config: %w", err)
 	}
 
 	if _, err := os.Stat(c.singleConfigPath); !os.IsNotExist(err) {
@@ -51,7 +51,6 @@ func (c *Config) Reload() error {
 	if err := c.ReloadPauseImage(newConfig); err != nil {
 		return err
 	}
-	c.ReloadPinnedImages(newConfig)
 	if err := c.ReloadRegistries(); err != nil {
 		return err
 	}
@@ -76,6 +75,11 @@ func (c *Config) Reload() error {
 	return nil
 }
 
+// IsPinnedImagesUpdated return the state of pinned image updates
+func (c *Config) IsPinnedImagesUpdated() bool {
+	return c.ReloadPinnedImages(c)
+}
+
 // logConfig logs a config set operation as with info verbosity. Please always
 // use this function for setting configuration options to ensure consistent
 // log outputs
@@ -143,17 +147,24 @@ func (c *Config) ReloadPauseImage(newConfig *Config) error {
 }
 
 // ReloadPinnedImages updates the PinnedImages with the provided `newConfig`.
-func (c *Config) ReloadPinnedImages(newConfig *Config) {
+func (c *Config) ReloadPinnedImages(newConfig *Config) bool {
 	updatedPinnedImages := make([]string, len(newConfig.PinnedImages))
+	updatedPinnedImageList := false
+
 	for i, image := range newConfig.PinnedImages {
 		if i < len(c.PinnedImages) && image == c.PinnedImages[i] {
-			updatedPinnedImages[i] = c.PinnedImages[i]
-		} else {
-			updatedPinnedImages[i] = image
+			continue
 		}
+		updatedPinnedImages[i] = image
+		updatedPinnedImageList = true
 	}
-	logrus.Infof("Updated new pinned images: %+v", updatedPinnedImages)
-	c.PinnedImages = updatedPinnedImages
+
+	if updatedPinnedImageList {
+		logrus.Infof("Updated new pinned images: %+v", updatedPinnedImages)
+		c.PinnedImages = updatedPinnedImages
+	}
+
+	return updatedPinnedImageList
 }
 
 // ReloadRegistries reloads the registry configuration from the Configs

pkg/config/reload_test.go

@@ -427,15 +427,17 @@ var _ = t.Describe("Config", func() {
 			sut.PinnedImages = []string{"image1", "image4", "image3"}
 			newConfig := &config.Config{}
 			newConfig.PinnedImages = []string{"image5"}
-			sut.ReloadPinnedImages(newConfig)
+			ok := sut.ReloadPinnedImages(newConfig)
+			Expect(ok).To(BeTrue())
 			Expect(sut.PinnedImages).To(Equal([]string{"image5"}))
 		})
 
 		It("should not update PinnedImages if they are the same as newConfig's PinnedImages", func() {
 			sut.PinnedImages = []string{"image1", "image2", "image3"}
 			newConfig := &config.Config{}
 			newConfig.PinnedImages = []string{"image1", "image2", "image3"}
-			sut.ReloadPinnedImages(newConfig)
+			ok := sut.ReloadPinnedImages(newConfig)
+			Expect(ok).To(BeFalse())
 			Expect(sut.PinnedImages).To(Equal([]string{"image1", "image2", "image3"}))
 		})
 	})

server/server.go

@@ -577,6 +577,12 @@ func (s *Server) startReloadWatcher(ctx context.Context) {
 				logrus.Errorf("Unable to reload configuration: %v", err)
 				continue
 			}
+			if s.config.IsPinnedImagesUpdated() {
+				if err := s.StorageImageServer().UpdateImageCache(s.config.PinnedImages); err != nil {
+					logrus.Errorf("Unable to update pinned images: %v", err)
+					continue
+				}
+			}
 		}
 	}()
 

test/reload_config.bats

@@ -230,3 +230,17 @@ EOF
 	#then
 	wait_for_log '"updating runtime configuration"'
 }
+
+@test "reload config should update 'pinned_images'" {
+	# given
+	cat << EOF > "$CRIO_CONFIG_DIR/99-pinned-image.conf"
+[crio.image]
+pinned_images = [ "quay.io/crio/hello-wasm:latest" ]
+EOF
+
+	# when
+	reload_crio
+
+	#then
+	wait_for_log 'Updated new pinned images'
+}