module: refactor out section header rewriting: FIX modversions

We can't do the find_sec after removing the SHF_ALLOC flags; it won't
find the sections.

Signed-off-by: Rusty Russell <rusty@rustcorp.com.au>

diff --git a/kernel/module.c b/kernel/module.c
index 91f3ebe230e3b3672c3ecef94de53fda6e240139..ab07f67de996a01472ad605b24dadfb08d1d54a3 100644 (file)
--- a/kernel/module.c
+++ b/kernel/module.c
@@ -2225,11 +2225,13 @@ static int rewrite_section_headers(struct load_info *info)
                if (strstarts(info->secstrings+shdr->sh_name, ".exit"))
                        shdr->sh_flags &= ~(unsigned long)SHF_ALLOC;
 #endif
-               /* Don't keep modinfo and version sections. */
-               if (!strcmp(info->secstrings+shdr->sh_name, "__versions")
-                   || !strcmp(info->secstrings+shdr->sh_name, ".modinfo"))
-                       shdr->sh_flags &= ~(unsigned long)SHF_ALLOC;
        }
+
+       /* Track but don't keep modinfo and version sections. */
+       info->index.vers = find_sec(info->hdr, info->sechdrs, info->secstrings, "__versions");
+       info->index.info = find_sec(info->hdr, info->sechdrs, info->secstrings, ".modinfo");
+       info->sechdrs[info->index.info].sh_flags &= ~(unsigned long)SHF_ALLOC;
+       info->sechdrs[info->index.vers].sh_flags &= ~(unsigned long)SHF_ALLOC;
        return 0;
 }
 
@@ -2282,8 +2284,6 @@ static struct module *setup_load_info(struct load_info *info)
                return ERR_PTR(-ENOEXEC);
        }
 
-       info->index.vers = find_sec(info->hdr, info->sechdrs, info->secstrings, "__versions");
-       info->index.info = find_sec(info->hdr, info->sechdrs, info->secstrings, ".modinfo");
        info->index.pcpu = find_pcpusec(info->hdr, info->sechdrs, info->secstrings);
 
        /* Check module struct version now, before we try to use module. */