/* Linux/i386 CET initializers function.

   Copyright (C) 2018 Free Software Foundation, Inc.

   The GNU C Library is free software; you can redistribute it and/or

   modify it under the terms of the GNU Lesser General Public

   License as published by the Free Software Foundation; either

   version 2.1 of the License, or (at your option) any later version.

   The GNU C Library is distributed in the hope that it will be useful,

   but WITHOUT ANY WARRANTY; without even the implied warranty of

   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

   Lesser General Public License for more details.

   You should have received a copy of the GNU Lesser General Public

   License along with the GNU C Library; if not, see

   <http://www.gnu.org/licenses/>.  */

#define LINKAGE static inline

#define _dl_cet_check cet_check

#include <sysdeps/x86/dl-cet.c>

#undef _dl_cet_check

#ifdef SHARED

void

_dl_cet_check (struct link_map *main_map, const char *program)

{

  cet_check (main_map, program);

  if ((GL(dl_x86_feature_1)[0] & GNU_PROPERTY_X86_FEATURE_1_SHSTK))

    {

      /* Replace _dl_runtime_resolve and _dl_runtime_profile with

         _dl_runtime_resolve_shstk and _dl_runtime_profile_shstk,

	 respectively if SHSTK is enabled.  */

      extern void _dl_runtime_resolve (Elf32_Word) attribute_hidden;

      extern void _dl_runtime_resolve_shstk (Elf32_Word) attribute_hidden;

      extern void _dl_runtime_profile (Elf32_Word) attribute_hidden;

      extern void _dl_runtime_profile_shstk (Elf32_Word) attribute_hidden;

      unsigned int i;

      struct link_map *l;

      Elf32_Addr *got;

      if (main_map->l_info[DT_JMPREL])

	{

	  got = (Elf32_Addr *) D_PTR (main_map, l_info[DT_PLTGOT]);

	  if (got[2] == (Elf32_Addr) &_dl_runtime_resolve)

	    got[2] = (Elf32_Addr) &_dl_runtime_resolve_shstk;

	  else if (got[2] == (Elf32_Addr) &_dl_runtime_profile)

	    got[2] = (Elf32_Addr) &_dl_runtime_profile_shstk;

	}

      i = main_map->l_searchlist.r_nlist;

      while (i-- > 0)

	{

	  l = main_map->l_initfini[i];

	  if (l->l_info[DT_JMPREL])

	    {

	      got = (Elf32_Addr *) D_PTR (l, l_info[DT_PLTGOT]);

	      if (got[2] == (Elf32_Addr) &_dl_runtime_resolve)

		got[2] = (Elf32_Addr) &_dl_runtime_resolve_shstk;

	      else if (got[2] == (Elf32_Addr) &_dl_runtime_profile)

		got[2] = (Elf32_Addr) &_dl_runtime_profile_shstk;

	    }

	}

    }

}

#endif