opal: add code patcher framework

This commit adds a framework to abstract runtime code patching. Components in the new framework can provide functions for either patching a named function or a function pointer. The later functionality is not being used but may provide a way to allow memory hooks when dlopen functionality is disabled. This commit adds two different flavors of code patching. The first is provided by the overwrite component. This component overwrites the first several instructions of the target function with code to jump to the provided hook function. The hook is expected to provide the full functionality of the hooked function. The linux patcher component is based on the memory hooks in ucx. It only works on linux and operates by overwriting function pointers in the symbol table. In this case the hook is free to call the original function using the function pointer returned by dlsym. Both components restore the original functions when the patcher framework closes. Signed-off-by: Nathan Hjelm <[email protected]>
open-mpi · Mar 29, 2016 · 5783f52 · 5783f52
1 parent beb4f86
commit 5783f52
Show file tree

Hide file tree

Showing 25 changed files with 1,390 additions and 226 deletions.
diff --git a/contrib/platform/lanl/cray_xe6/debug-common b/contrib/platform/lanl/cray_xe6/debug-common
@@ -32,8 +32,5 @@ enable_mca_direct=pml-ob1
 # enable development headers
 with_devel_headers=yes
 
-# enable ptmalloc (enables lazy deregistration)
-with_memory_manager=linux
-
 # disable valgrind
 with_valgrind=no
diff --git a/contrib/platform/lanl/cray_xe6/optimized-common b/contrib/platform/lanl/cray_xe6/optimized-common
@@ -38,8 +38,5 @@ enable_mca_direct=pml-ob1
 # enable development headers
 with_devel_headers=yes
 
-# enable ptmalloc (enables lazy deregistration)
-with_memory_manager=linux
-
 # disable valgrind
 with_valgrind=no
diff --git a/opal/common_sym_whitelist.txt b/opal/common_sym_whitelist.txt
@@ -4,3 +4,4 @@ opal_show_help_yyleng
 opal_show_help_yytext
 opal_util_keyval_yyleng
 opal_util_keyval_yytext
+__curbrk
diff --git a/opal/mca/memory/patcher/configure.m4 b/opal/mca/memory/patcher/configure.m4
@@ -21,7 +21,7 @@
 #
 # $HEADER$
 #
-AC_DEFUN([MCA_opal_memory_patcher_PRIORITY], [40])
+AC_DEFUN([MCA_opal_memory_patcher_PRIORITY], [41])
 
 AC_DEFUN([MCA_opal_memory_patcher_COMPILE_MODE], [
     AC_MSG_CHECKING([for MCA component $2:$3 compile mode])

diff --git a/opal/mca/memory/patcher/memory_patcher.h b/opal/mca/memory/patcher/memory_patcher.h
@@ -16,7 +16,7 @@
 #include "opal_config.h"
 
 #include "opal/mca/memory/memory.h"
-#include "opal/util/opal_patcher.h"
+#include "opal/mca/patcher/patcher.h"
 
 typedef struct opal_memory_patcher_component_t {
   opal_memory_base_component_2_0_0_t super;

diff --git a/opal/mca/memory/patcher/memory_patcher_component.c b/opal/mca/memory/patcher/memory_patcher_component.c
@@ -28,6 +28,7 @@
 #include "opal/util/output.h"
 #include "opal/util/show_help.h"
 #include "opal/mca/memory/base/empty.h"
+#include "opal/mca/memory/base/base.h"
 #include "opal/memoryhooks/memory.h"
 
 #include <stdlib.h>
@@ -95,78 +96,94 @@ void *__mmap (void *start, size_t length, int prot, int flags, int fd, off_t off
 #define memory_patcher_syscall syscall
 #endif
 
+static void *(*original_mmap)(void *, size_t, int, int, int, off_t);
+
 static void *intercept_mmap(void *start, size_t length, int prot, int flags, int fd, off_t offset)
 {
     OPAL_PATCHER_BEGIN;
     void *result = 0;
 
     if (prot == PROT_NONE) {
-        opal_mem_hooks_release_hook (start, length, 0);
+        opal_mem_hooks_release_hook (start, length, true);
     }
 
+    if (!original_mmap) {
 #if OPAL_MEMORY_PATCHER_HAVE___MMAP
-    /* the darwin syscall returns an int not a long so call the underlying __mmap function */
-    result = __mmap (start, length, prot, flags, fd, offset);
+        /* the darwin syscall returns an int not a long so call the underlying __mmap function */
+        result = __mmap (start, length, prot, flags, fd, offset);
 #else
-    result = (void*)(intptr_t) memory_patcher_syscall(SYS_mmap, start, length, prot, flags, fd, offset);
+        result = (void*)(intptr_t) memory_patcher_syscall(SYS_mmap, start, length, prot, flags, fd, offset);
 #endif
 
-// I thought we had some issue in the past with the above line for IA32,
-// like maybe syscall() wouldn't handle that many arguments. But just now
-// I used gcc -m32 and it worked on a recent system. But there's a possibility
-// that older ia32 systems may need some other code to make the above syscall.
+        // I thought we had some issue in the past with the above line for IA32,
+        // like maybe syscall() wouldn't handle that many arguments. But just now
+        // I used gcc -m32 and it worked on a recent system. But there's a possibility
+        // that older ia32 systems may need some other code to make the above syscall.
+    } else {
+        result = original_mmap (start, length, prot, flags, fd, offset);
+    }
 
     OPAL_PATCHER_END;
     return result;
 }
 
+static int (*original_munmap) (void *, size_t);
+
 static int intercept_munmap(void *start, size_t length)
 {
     OPAL_PATCHER_BEGIN;
     int result = 0;
 
-    opal_mem_hooks_release_hook (start, length, 0);
+    opal_mem_hooks_release_hook (start, length, false);
 
-    result=memory_patcher_syscall(SYS_munmap, start, length);
+    if (!original_munmap) {
+        result = memory_patcher_syscall(SYS_munmap, start, length);
+    } else {
+        result = original_munmap (start, length);
+    }
 
     OPAL_PATCHER_END;
     return result;
 }
 
 #if defined (SYS_mremap)
 
+static void *(*original_mremap) (void *, size_t, size_t, int, ...);
+
 static void *intercept_mremap (void *start, size_t oldlen, size_t newlen, int flags, ...)
 {
     OPAL_PATCHER_BEGIN;
     void *result = 0;
 #ifdef MREMAP_FIXED
     va_list ap;
-    void *new_address;
 #endif
+    void *new_address = NULL;
 
-    opal_mem_hooks_release_hook (start, oldlen, 0);
+    opal_mem_hooks_release_hook (start, oldlen, false);
 
 #ifdef MREMAP_FIXED
     if (flags & MREMAP_FIXED) {
         va_start(ap, flags);
         new_address = va_arg(ap, void*);
-        result=(void *)(intptr_t) memory_patcher_syscall(
-            SYS_mremap, start, oldlen, newlen, flags, new_address);
+
         va_end(ap);
-    } else {
-        result=(void*)memory_patcher_syscall(
-            SYS_mremap, start, oldlen, newlen, flags);
     }
-#else
-    result=(void*)(intptr_t) memory_patcher_syscall(SYS_mremap, start, oldlen, newlen, flags);
 #endif
 
+    if (!original_mremap) {
+        result = (void *)(intptr_t) memory_patcher_syscall (SYS_mremap, start, oldlen, newlen, flags, new_address);
+    } else {
+        result = original_mremap (start, oldlen, newlen, flags, new_address);
+    }
+
     OPAL_PATCHER_END;
     return result;
 }
 
 #endif
 
+static int (*original_madvise) (void *, size_t, int);
+
 static int intercept_madvise (void *start, size_t length, int advice)
 {
     OPAL_PATCHER_BEGIN;
@@ -178,9 +195,14 @@ static int intercept_madvise (void *start, size_t length, int advice)
 #endif
         advice == POSIX_MADV_DONTNEED)
     {
-        opal_mem_hooks_release_hook (start, length, 0);
+        opal_mem_hooks_release_hook (start, length, false);
+    }
+
+    if (!original_madvise) {
+        result = memory_patcher_syscall(SYS_madvise, start, length, advice);
+    } else {
+        result = original_madvise (start, length, advice);
     }
-    result = memory_patcher_syscall(SYS_madvise, start, length, advice);
 
     OPAL_PATCHER_END;
     return result;
@@ -192,6 +214,8 @@ static int intercept_madvise (void *start, size_t length, int advice)
 void *__curbrk; /* in libc */
 #endif
 
+static int (*original_brk) (void *);
+
 static int intercept_brk (void *addr)
 {
     OPAL_PATCHER_BEGIN;
@@ -204,23 +228,32 @@ static int intercept_brk (void *addr)
     old_addr = sbrk (0);
 #endif
 
-    /* get the current_addr */
-    new_addr = (void *) (intptr_t) memory_patcher_syscall(SYS_brk, addr);
+    if (!original_brk) {
+        /* get the current_addr */
+        new_addr = (void *) (intptr_t) memory_patcher_syscall(SYS_brk, addr);
 
 #if OPAL_MEMORY_PATCHER_HAVE___CURBRK
-    /*
-     * Note: if we were using glibc brk/sbrk, their __curbrk would get
-     * updated, but since we're going straight to the syscall, we have
-     * to update __curbrk or else glibc won't see it.
-     */
-    __curbrk = new_addr;
+        /*
+         * Note: if we were using glibc brk/sbrk, their __curbrk would get
+         * updated, but since we're going straight to the syscall, we have
+         * to update __curbrk or else glibc won't see it.
+         */
+        __curbrk = new_addr;
 #endif
+    } else {
+        result = original_brk (addr);
+#if OPAL_MEMORY_PATCHER_HAVE___CURBRK
+        new_addr = __curbrk;
+#else
+        new_addr = sbrk (0);
+#endif
+    }
 
     if (new_addr < addr) {
         errno = ENOMEM;
         result = -1;
     } else if (new_addr < old_addr) {
-        opal_mem_hooks_release_hook (new_addr, (intptr_t) old_addr - (intptr_t) new_addr, 0);
+        opal_mem_hooks_release_hook (new_addr, (intptr_t) old_addr - (intptr_t) new_addr, true);
     }
     OPAL_PATCHER_END;
     return result;
@@ -241,7 +274,7 @@ static int patcher_register (void)
 
 static int patcher_query (int *priority)
 {
-    if (opal_patch_supported ()) {
+    if (opal_patcher->patch_symbol) {
         *priority = mca_memory_patcher_priority;
     } else {
         *priority = -1;
@@ -263,30 +296,30 @@ static int patcher_open (void)
     /* set memory hooks support level */
     opal_mem_hooks_set_support (OPAL_MEMORY_FREE_SUPPORT | OPAL_MEMORY_MUNMAP_SUPPORT);
 
-    rc = opal_patch_symbol ("mmap", (uintptr_t) intercept_mmap);
+    rc = opal_patcher->patch_symbol ("mmap", (uintptr_t) intercept_mmap, (uintptr_t *) &original_mmap);
     if (OPAL_SUCCESS != rc) {
         return rc;
     }
 
-    rc = opal_patch_symbol ("munmap", (uintptr_t)intercept_munmap);
+    rc = opal_patcher->patch_symbol ("munmap", (uintptr_t)intercept_munmap, (uintptr_t *) &original_munmap);
     if (OPAL_SUCCESS != rc) {
         return rc;
     }
 
 #if defined (SYS_mremap)
-    rc = opal_patch_symbol ("mremap",(uintptr_t)intercept_mremap);
+    rc = opal_patcher->patch_symbol ("mremap",(uintptr_t)intercept_mremap, (uintptr_t *) &original_mremap);
     if (OPAL_SUCCESS != rc) {
         return rc;
     }
 #endif
 
-    rc = opal_patch_symbol ("madvise", (uintptr_t)intercept_madvise);
+    rc = opal_patcher->patch_symbol ("madvise", (uintptr_t)intercept_madvise, (uintptr_t *) &original_madvise);
     if (OPAL_SUCCESS != rc) {
         return rc;
     }
 
 #if defined (SYS_brk)
-    rc = opal_patch_symbol ("brk", (uintptr_t)intercept_brk);
+    rc = opal_patcher->patch_symbol ("brk", (uintptr_t)intercept_brk, (uintptr_t *) &original_brk);
 #endif
 
     return rc;

diff --git a/opal/mca/patcher/Makefile.am b/opal/mca/patcher/Makefile.am
@@ -0,0 +1,39 @@
+#
+# Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
+#                         University Research and Technology
+#                         Corporation.  All rights reserved.
+# Copyright (c) 2004-2005 The University of Tennessee and The University
+#                         of Tennessee Research Foundation.  All rights
+#                         reserved.
+# Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
+#                         University of Stuttgart.  All rights reserved.
+# Copyright (c) 2004-2005 The Regents of the University of California.
+#                         All rights reserved.
+# Copyright (c) 2010 Cisco Systems, Inc.  All rights reserved.
+# Copyright (c) 2016      Los Alamos National Security, LLC. All rights
+#                         reserved.
+# $COPYRIGHT$
+#
+# Additional copyrights may follow
+#
+# $HEADER$
+#
+
+# main library setup
+noinst_LTLIBRARIES = libmca_patcher.la
+libmca_patcher_la_SOURCES =
+
+# local files
+headers = patcher.h
+libmca_patcher_la_SOURCES += $(headers)
+
+# Conditionally install the header files
+if WANT_INSTALL_HEADERS
+opaldir = $(opalincludedir)/$(subdir)
+nobase_opal_HEADERS = $(headers)
+endif
+
+include base/Makefile.am
+
+distclean-local:
+	rm -f base/static-components.h
diff --git a/opal/mca/patcher/base/Makefile.am b/opal/mca/patcher/base/Makefile.am
@@ -0,0 +1,24 @@
+#
+# Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
+#                         University Research and Technology
+#                         Corporation.  All rights reserved.
+# Copyright (c) 2004-2005 The University of Tennessee and The University
+#                         of Tennessee Research Foundation.  All rights
+#                         reserved.
+# Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
+#                         University of Stuttgart.  All rights reserved.
+# Copyright (c) 2004-2005 The Regents of the University of California.
+#                         All rights reserved.
+# Copyright (c) 2009      Cisco Systems, Inc.  All rights reserved.
+# Copyright (c) 2016      Los Alamos National Security, LLC. All rights
+#                         reserved.
+# $COPYRIGHT$
+#
+# Additional copyrights may follow
+#
+# $HEADER$
+#
+
+headers += base/base.h
+
+libmca_patcher_la_SOURCES += base/patcher_base_frame.c
diff --git a/opal/mca/patcher/base/base.h b/opal/mca/patcher/base/base.h
@@ -0,0 +1,40 @@
+/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil -*- */
+/*
+ * Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
+ *                         University Research and Technology
+ *                         Corporation.  All rights reserved.
+ * Copyright (c) 2004-2006 The University of Tennessee and The University
+ *                         of Tennessee Research Foundation.  All rights
+ *                         reserved.
+ * Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
+ *                         University of Stuttgart.  All rights reserved.
+ * Copyright (c) 2004-2005 The Regents of the University of California.
+ *                         All rights reserved.
+ * Copyright (c) 2016      Los Alamos National Security, LLC. All rights
+ *                         reserved.
+ * $COPYRIGHT$
+ *
+ * Additional copyrights may follow
+ *
+ * $HEADER$
+ *
+ */
+
+#ifndef OPAL_PATCHER_BASE_H
+#define OPAL_PATCHER_BASE_H
+
+#include "opal_config.h"
+#include "opal/mca/base/mca_base_framework.h"
+#include "opal/mca/patcher/patcher.h"
+
+
+BEGIN_C_DECLS
+
+/**
+ * Framework struct declaration for this framework
+ */
+OPAL_DECLSPEC extern mca_base_framework_t opal_patcher_base_framework;
+OPAL_DECLSPEC int opal_patcher_base_select (void);
+
+END_C_DECLS
+#endif /* OPAL_BASE_PATCHER_H */