520cc8
commit bd13cb19f5e15e9e9a92a536e755fd93a97a67f6
520cc8
Author: Florian Weimer <fweimer@redhat.com>
520cc8
Date:   Fri Aug 19 11:16:32 2022 +0200
520cc8
520cc8
    scripts/glibcelf.py: Add hashing support
520cc8
    
520cc8
    ELF and GNU hashes can now be computed using the elf_hash and
520cc8
    gnu_hash functions.
520cc8
    
520cc8
    Reviewed-by: Carlos O'Donell <carlos@redhat.com>
520cc8
    Tested-by: Carlos O'Donell <carlos@redhat.com>
520cc8
520cc8
diff --git a/elf/tst-glibcelf.py b/elf/tst-glibcelf.py
520cc8
index bf15a3bad4479e08..e5026e2289df206b 100644
520cc8
--- a/elf/tst-glibcelf.py
520cc8
+++ b/elf/tst-glibcelf.py
520cc8
@@ -240,6 +240,24 @@ def check_constant_values(cc):
520cc8
             error('{}: glibcelf has {!r}, <elf.h> has {!r}'.format(
520cc8
                 name, glibcelf_value, elf_h_value))
520cc8
 
520cc8
+def check_hashes():
520cc8
+    for name, expected_elf, expected_gnu in (
520cc8
+            ('', 0, 0x1505),
520cc8
+            ('PPPPPPPPPPPP', 0, 0x9f105c45),
520cc8
+            ('GLIBC_2.0', 0xd696910, 0xf66c3dd5),
520cc8
+            ('GLIBC_2.34', 0x69691b4, 0xc3f3f90c),
520cc8
+            ('GLIBC_PRIVATE', 0x963cf85, 0x692a260)):
520cc8
+        for convert in (lambda x: x, lambda x: x.encode('UTF-8')):
520cc8
+            name = convert(name)
520cc8
+            actual_elf = glibcelf.elf_hash(name)
520cc8
+            if actual_elf != expected_elf:
520cc8
+                error('elf_hash({!r}): {:x} != 0x{:x}'.format(
520cc8
+                    name, actual_elf, expected_elf))
520cc8
+            actual_gnu = glibcelf.gnu_hash(name)
520cc8
+            if actual_gnu != expected_gnu:
520cc8
+                error('gnu_hash({!r}): {:x} != 0x{:x}'.format(
520cc8
+                    name, actual_gnu, expected_gnu))
520cc8
+
520cc8
 def main():
520cc8
     """The main entry point."""
520cc8
     parser = argparse.ArgumentParser(
520cc8
@@ -251,6 +269,7 @@ def main():
520cc8
     check_duplicates()
520cc8
     check_constant_prefixes()
520cc8
     check_constant_values(cc=args.cc)
520cc8
+    check_hashes()
520cc8
 
520cc8
     if errors_encountered > 0:
520cc8
         print("note: errors encountered:", errors_encountered)
520cc8
diff --git a/scripts/glibcelf.py b/scripts/glibcelf.py
520cc8
index de0509130ed9ad47..5c8f46f590722384 100644
520cc8
--- a/scripts/glibcelf.py
520cc8
+++ b/scripts/glibcelf.py
520cc8
@@ -1158,5 +1158,24 @@ class Image:
520cc8
         self._stringtab[sh_link] = strtab
520cc8
         return strtab
520cc8
 
520cc8
+def elf_hash(s):
520cc8
+    """Computes the ELF hash of the string."""
520cc8
+    acc = 0
520cc8
+    for ch in s:
520cc8
+        if type(ch) is not int:
520cc8
+            ch = ord(ch)
520cc8
+        acc = ((acc << 4) + ch) & 0xffffffff
520cc8
+        top = acc & 0xf0000000
520cc8
+        acc = (acc ^ (top >> 24)) & ~top
520cc8
+    return acc
520cc8
+
520cc8
+def gnu_hash(s):
520cc8
+    """Computes the GNU hash of the string."""
520cc8
+    h = 5381
520cc8
+    for ch in s:
520cc8
+        if type(ch) is not int:
520cc8
+            ch = ord(ch)
520cc8
+        h = (h * 33 + ch) & 0xffffffff
520cc8
+    return h
520cc8
 
520cc8
 __all__ = [name for name in dir() if name[0].isupper()]