From f7e5ed3c8dffcdfc2c55c63f2e3cbcb390bbf013 Mon Sep 17 00:00:00 2001
From: Julian Andres Klode <jak@debian.org>
Date: Sat, 13 Apr 2024 20:27:48 +0200
Subject: Show space estimate for /boot, if separate; or estimate initrd for
 /usr

Calculate an estimate of 110% of the biggest initrd + system.map
as the additional space a kernel needs in /boot.

If /boot is a different file system than /usr, print the size of
the kernels + the additional space they will need separately;
otherwise include it in our /usr figure.
---
 apt-pkg/depcache.cc | 56 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 56 insertions(+)

(limited to 'apt-pkg/depcache.cc')

diff --git a/apt-pkg/depcache.cc b/apt-pkg/depcache.cc
index e3e8d627c..76a5c09ba 100644
--- a/apt-pkg/depcache.cc
+++ b/apt-pkg/depcache.cc
@@ -42,6 +42,7 @@
 #include <unordered_map>
 #include <utility>
 #include <vector>
+#include <dirent.h>
 
 #include <sys/stat.h>
 
@@ -2613,3 +2614,58 @@ bool pkgDepCache::PhasingApplied(pkgCache::PkgIterator Pkg) const
    return true;
 }
 									/*}}}*/
+
+// DepCache::BootSize						/*{{{*/
+double pkgDepCache::BootSize(bool initrdOnly)
+{
+   double BootSize = 0;
+   int BootCount = 0;
+   auto VirtualKernelPkg = FindPkg("$kernel", "any");
+   if (VirtualKernelPkg.end())
+      return 0;
+
+   for (pkgCache::PrvIterator Prv = VirtualKernelPkg.ProvidesList(); Prv.end() == false; ++Prv)
+   {
+      auto Pkg = Prv.OwnerPkg();
+      if ((*this)[Pkg].NewInstall())
+	 BootSize += (*this)[Pkg].InstallVer->InstalledSize, BootCount++;
+   }
+   if (BootCount == 0)
+      return 0;
+   if (initrdOnly)
+      BootSize = 0;
+
+   DIR *boot = opendir(_config->FindDir("Dir::Boot").c_str());
+   struct dirent *ent;
+   if (boot)
+   {
+      double initrdSize = 0;
+      double mapSize = 0;
+      while ((ent = readdir(boot)))
+      {
+	 enum
+	 {
+	    INITRD,
+	    MAP
+	 } type;
+	 if (APT::String::Startswith(ent->d_name, "initrd.img-"))
+	    type = INITRD;
+	 else if (APT::String::Startswith(ent->d_name, "System.map-"))
+	    type = MAP;
+	 else
+	    continue;
+
+	 auto path = _config->FindDir("Dir::Boot") + ent->d_name;
+
+	 if (struct stat st; stat(path.c_str(), &st) == 0)
+	 {
+	    double &targetSize = type == INITRD ? initrdSize : mapSize;
+	    targetSize = std::max(targetSize, double(st.st_size));
+	 }
+      }
+      closedir(boot);
+      return initrdSize ? BootSize + BootCount * (initrdSize + mapSize) * 1.1 : 0;
+   }
+   return 0;
+}
+									/*}}}*/
-- 
cgit v1.2.3-70-g09d2