From a41613ba3ab004719155db48e38897843dd7ece7 Mon Sep 17 00:00:00 2001
From: James Hewitt <james.hewitt@uk.ibm.com>
Date: Fri, 18 Aug 2023 17:23:49 +0100
Subject: [PATCH] Don't make a new buffer for catalog listing

We are given a slice to fill, write catalog entries directly to the
slice until it is full.

Signed-off-by: James Hewitt <james.hewitt@uk.ibm.com>
---
 registry/storage/catalog.go | 32 +++++++++++++++++---------------
 1 file changed, 17 insertions(+), 15 deletions(-)

diff --git a/registry/storage/catalog.go b/registry/storage/catalog.go
index 55500694..7c8551fe 100644
--- a/registry/storage/catalog.go
+++ b/registry/storage/catalog.go
@@ -14,12 +14,12 @@ import (
 // Returns a list, or partial list, of repositories in the registry.
 // Because it's a quite expensive operation, it should only be used when building up
 // an initial set of repositories.
-func (reg *registry) Repositories(ctx context.Context, repos []string, last string) (n int, err error) {
-	var finishedWalk bool
-	var foundRepos []string
+func (reg *registry) Repositories(ctx context.Context, repos []string, last string) (int, error) {
+	filledBuffer := false
+	foundRepos := 0
 
 	if len(repos) == 0 {
-		return 0, errors.New("no space in slice")
+		return 0, errors.New("Attempted to list 0 repositories")
 	}
 
 	root, err := pathFor(repositoriesRootPathSpec{})
@@ -29,32 +29,34 @@ func (reg *registry) Repositories(ctx context.Context, repos []string, last stri
 
 	err = reg.blobStore.driver.Walk(ctx, root, func(fileInfo driver.FileInfo) error {
 		err := handleRepository(fileInfo, root, last, func(repoPath string) error {
-			foundRepos = append(foundRepos, repoPath)
+			repos[foundRepos] = repoPath
+			foundRepos += 1
 			return nil
 		})
 		if err != nil {
 			return err
 		}
 
-		// if we've filled our array, no need to walk any further
-		if len(foundRepos) == len(repos) {
-			finishedWalk = true
+		// if we've filled our slice, no need to walk any further
+		if foundRepos == len(repos) {
+			filledBuffer = true
 			return driver.ErrSkipDir
 		}
 
 		return nil
 	})
 
-	n = copy(repos, foundRepos)
-
 	if err != nil {
-		return n, err
-	} else if !finishedWalk {
-		// We didn't fill buffer. No more records are available.
-		return n, io.EOF
+		return foundRepos, err
 	}
 
-	return n, err
+	if filledBuffer {
+		// There are potentially more repositories to list
+		return foundRepos, nil
+	}
+
+	// We didn't fill the buffer, so that's the end of the list of repos
+	return foundRepos, io.EOF
 }
 
 // Enumerate applies ingester to each repository