create-diff-object: Add new entries to special sections array

Handle .livepatch.hooks* and .altinstr_replacement sections as the
special sections with assigned group_size resolution function.
By default each .livepatch.hooks* sections' entry is 8 bytes long (a
pointer). The .altinstr_replacement section has undefined group_size.

Allow to specify different .livepatch.hooks* section entry size using
shell environment variable HOOK_STRUCT_SIZE.

Signed-off-by: Pawel Wieczorkiewicz <wipawel@amazon.de>
Reviewed-by: Andra-Irina Paraschiv <andraprs@amazon.com>
Reviewed-by: Bjoern Doebel <doebel@amazon.de>
Reviewed-by: Norbert Manthey <nmanthey@amazon.de>

diff --git a/create-diff-object.c b/create-diff-object.c
index c6183c3a65e4f46ce70d7d509b7d1840ddd241a6..8365af082ec80791f1223f4fbfc8f19ef6dbba2d 100644 (file)
--- a/create-diff-object.c
+++ b/create-diff-object.c
@@ -995,6 +995,24 @@ static int altinstructions_group_size(struct kpatch_elf *kelf, int offset)
        return size;
 }
 
+static int livepatch_hooks_group_size(struct kpatch_elf *kelf, int offset)
+{
+       static int size = 0;
+       char *str;
+       if (!size) {
+               str = getenv("HOOK_STRUCT_SIZE");
+               size = str ? atoi(str) : 8;
+       }
+
+       log_debug("livepatch_hooks_size=%d\n", size);
+       return size;
+}
+
+static int undefined_group_size(struct kpatch_elf *kelf, int offset)
+{
+       return 0;
+}
+
 /*
  * The rela groups in the .fixup section vary in size.  The beginning of each
  * .fixup rela group is referenced by the .ex_table section. To find the size
@@ -1072,6 +1090,18 @@ static struct special_section special_sections[] = {
                .name           = ".altinstructions",
                .group_size     = altinstructions_group_size,
        },
+       {
+               .name           = ".altinstr_replacement",
+               .group_size     = undefined_group_size,
+       },
+       {
+               .name           = ".livepatch.hooks.load",
+               .group_size     = livepatch_hooks_group_size,
+       },
+       {
+               .name           = ".livepatch.hooks.unload",
+               .group_size     = livepatch_hooks_group_size,
+       },
        {},
 };