Merge pull request #14305 from NixOS/alignment-utils

libutil: Add alignUp helper function, use in archive.cc

src/libutil/archive.cc

@@ -6,6 +6,7 @@
 #include <strings.h> // for strcasecmp
 
 #include "nix/util/archive.hh"
+#include "nix/util/alignment.hh"
 #include "nix/util/config-global.hh"
 #include "nix/util/posix-source-accessor.hh"
 #include "nix/util/source-path.hh"
@@ -133,7 +134,7 @@ static void parseContents(CreateRegularFileSink & sink, Source & source)
     sink.preallocateContents(size);
 
     if (sink.skipContents) {
-        source.skip(size + (size % 8 ? 8 - (size % 8) : 0));
+        source.skip(alignUp(size, 8));
         return;
     }
 

src/libutil/include/nix/util/alignment.hh

@@ -0,0 +1,23 @@
+#pragma once
+///@file
+
+#include <cassert>
+#include <type_traits>
+#include <cstdint>
+#include <bit>
+
+namespace nix {
+
+/// Aligns val upwards to be a multiple of alignment.
+///
+/// @pre alignment must be a power of 2.
+template<typename T>
+    requires std::is_unsigned_v<T>
+constexpr T alignUp(T val, unsigned alignment)
+{
+    assert(std::has_single_bit(alignment) && "alignment must be a power of 2");
+    T mask = ~(T{alignment} - 1u);
+    return (val + alignment - 1) & mask;
+}
+
+} // namespace nix

src/libutil/include/nix/util/meson.build

@@ -4,6 +4,7 @@ include_dirs = [ include_directories('../..') ]
 
 headers = files(
   'abstract-setting-to-json.hh',
+  'alignment.hh',
   'ansicolor.hh',
   'archive.hh',
   'args.hh',