3 files changed, 44 insertions, 11 deletions

diff --git a/pkg/api/server/server.go b/pkg/api/server/server.go
index 9576fd437..ce2d152e0 100644
--- a/pkg/api/server/server.go
+++ b/pkg/api/server/server.go
@@ -8,6 +8,7 @@ import (
 	"os"
 	"os/signal"
 	"runtime"
+	goRuntime "runtime"
 	"strings"
 	"sync"
 	"syscall"
@@ -30,6 +31,7 @@ type APIServer struct {
 	net.Listener                    // mux for routing HTTP API calls to libpod routines
 	context.CancelFunc              // Stop APIServer
 	idleTracker        *IdleTracker // Track connections to support idle shutdown
+	pprof              *http.Server // Sidecar http server for providing performance data
 }
 
 // Number of seconds to wait for next request, if exceeded shutdown server
@@ -145,6 +147,20 @@ func (s *APIServer) Serve() error {
 		_ = s.Shutdown()
 	}()
 
+	if logrus.IsLevelEnabled(logrus.DebugLevel) {
+		go func() {
+			pprofMux := mux.NewRouter()
+			pprofMux.PathPrefix("/debug/pprof").Handler(http.DefaultServeMux)
+			goRuntime.SetMutexProfileFraction(1)
+			goRuntime.SetBlockProfileRate(1)
+			s.pprof = &http.Server{Addr: "localhost:8888", Handler: pprofMux}
+			err := s.pprof.ListenAndServe()
+			if err != nil && err != http.ErrServerClosed {
+				logrus.Warn("Profiler Service failed: " + err.Error())
+			}
+		}()
+	}
+
 	go func() {
 		err := s.Server.Serve(s.Listener)
 		if err != nil && err != http.ErrServerClosed {
@@ -166,26 +182,29 @@ func (s *APIServer) Serve() error {
 
 // Shutdown is a clean shutdown waiting on existing clients
 func (s *APIServer) Shutdown() error {
-	if logrus.IsLevelEnabled(logrus.DebugLevel) {
-		_, file, line, _ := runtime.Caller(1)
-		logrus.Debugf("APIServer.Shutdown by %s:%d, %d/%d connection(s)",
-			file, line, s.idleTracker.ActiveConnections(), s.idleTracker.TotalConnections())
-	}
-
-	// Duration == 0 flags no auto-shutdown of the server
-	if s.idleTracker.Duration == 0 {
-		logrus.Debug("APIServer.Shutdown ignored as Duration == 0")
+	if s.idleTracker.Duration == UnlimitedServiceDuration {
+		logrus.Debug("APIServer.Shutdown ignored as Duration is UnlimitedService.")
 		return nil
 	}
 
-	// Gracefully shutdown server, duration of wait same as idle window
+	// Gracefully shutdown server(s), duration of wait same as idle window
 	// TODO: Should we really wait the idle window for shutdown?
 	ctx, cancel := context.WithTimeout(context.Background(), s.idleTracker.Duration)
 	defer cancel()
+
+	if logrus.IsLevelEnabled(logrus.DebugLevel) {
+		_, file, line, _ := runtime.Caller(1)
+		logrus.Debugf("APIServer.Shutdown by %s:%d, %d/%d connection(s)",
+			file, line, s.idleTracker.ActiveConnections(), s.idleTracker.TotalConnections())
+		if err := s.pprof.Shutdown(ctx); err != nil {
+			logrus.Warn("Failed to cleanly shutdown pprof Server: " + err.Error())
+		}
+	}
+
 	go func() {
 		err := s.Server.Shutdown(ctx)
 		if err != nil && err != context.Canceled && err != http.ErrServerClosed {
-			logrus.Errorf("Failed to cleanly shutdown APIServer: %s", err.Error())
+			logrus.Error("Failed to cleanly shutdown APIServer: " + err.Error())
 		}
 	}()
 	<-ctx.Done()
diff --git a/pkg/domain/infra/abi/images.go b/pkg/domain/infra/abi/images.go
index 32f7d75e5..d0b7b42b5 100644
--- a/pkg/domain/infra/abi/images.go
+++ b/pkg/domain/infra/abi/images.go
@@ -303,6 +303,10 @@ func (ir *ImageEngine) Untag(ctx context.Context, nameOrId string, tags []string
 	if err != nil {
 		return err
 	}
+	// If only one arg is provided, all names are to be untagged
+	if len(tags) == 0 {
+		tags = newImage.Names()
+	}
 	for _, tag := range tags {
 		if err := newImage.UntagImage(tag); err != nil {
 			return err
diff --git a/pkg/domain/infra/tunnel/images.go b/pkg/domain/infra/tunnel/images.go
index 822842936..27ed9f1ec 100644
--- a/pkg/domain/infra/tunnel/images.go
+++ b/pkg/domain/infra/tunnel/images.go
@@ -7,6 +7,7 @@ import (
 
 	"github.com/containers/common/pkg/config"
 	"github.com/containers/image/v5/docker/reference"
+	"github.com/containers/libpod/pkg/bindings"
 	images "github.com/containers/libpod/pkg/bindings/images"
 	"github.com/containers/libpod/pkg/domain/entities"
 	"github.com/containers/libpod/pkg/domain/utils"
@@ -109,6 +110,15 @@ func (ir *ImageEngine) Tag(ctx context.Context, nameOrId string, tags []string,
 }
 
 func (ir *ImageEngine) Untag(ctx context.Context, nameOrId string, tags []string, options entities.ImageUntagOptions) error {
+	// Remove all tags if none are provided
+	if len(tags) == 0 {
+		newImage, err := images.GetImage(ir.ClientCxt, nameOrId, &bindings.PFalse)
+		if err != nil {
+			return err
+		}
+		tags = newImage.NamesHistory
+	}
+
 	for _, newTag := range tags {
 		var (
 			tag, repo string