From 851bf31df898b1569571244ce2094b59a9efcb7f Mon Sep 17 00:00:00 2001 From: Paul Elder Date: Thu, 6 Jun 2019 15:01:20 -0400 Subject: libcamera: ipa_module: elfLoadSymbol find symbol regardless of size Make elfLoadSymbol more generic by making the symbol size an output rather than an input. Also move the memcpy out of elfLoadSymbol. If the size of struct IPAModuleInfo changes between versions, we still want to be able to load it and perhaps do conversions for backwards compatibility. In this case the size should not be a restriction when searching for the symbol. Signed-off-by: Paul Elder Reviewed-by: Laurent Pinchart --- src/libcamera/ipa_module.cpp | 68 ++++++++++++++++++++++++++++---------------- 1 file changed, 43 insertions(+), 25 deletions(-) (limited to 'src/libcamera/ipa_module.cpp') diff --git a/src/libcamera/ipa_module.cpp b/src/libcamera/ipa_module.cpp index d201b4f8..d2e3c366 100644 --- a/src/libcamera/ipa_module.cpp +++ b/src/libcamera/ipa_module.cpp @@ -15,6 +15,7 @@ #include #include #include +#include #include #include "log.h" @@ -81,18 +82,27 @@ int elfVerifyIdent(void *map, size_t soSize) return 0; } +/** + * \brief Retrieve address and size of a symbol from an mmap'ed ELF file + * \param[in] map Address of mmap'ed ELF file + * \param[in] soSize Size of mmap'ed ELF file (in bytes) + * \param[in] symbol Symbol name + * + * \return zero or error code, address or nullptr, size of symbol or zero, + * respectively + */ template -int elfLoadSymbol(void *dst, size_t size, void *map, size_t soSize, - const char *symbol) +std::tuple +elfLoadSymbol(void *map, size_t soSize, const char *symbol) { ElfHeader *eHdr = elfPointer(map, 0, soSize); if (!eHdr) - return -ENOEXEC; + return std::make_tuple(nullptr, 0); off_t offset = eHdr->e_shoff + eHdr->e_shentsize * eHdr->e_shstrndx; SecHeader *sHdr = elfPointer(map, offset, soSize); if (!sHdr) - return -ENOEXEC; + return std::make_tuple(nullptr, 0); off_t shnameoff = sHdr->sh_offset; /* Locate .dynsym section header. */ @@ -101,12 +111,12 @@ int elfLoadSymbol(void *dst, size_t size, void *map, size_t soSize, offset = eHdr->e_shoff + eHdr->e_shentsize * i; sHdr = elfPointer(map, offset, soSize); if (!sHdr) - return -ENOEXEC; + return std::make_tuple(nullptr, 0); offset = shnameoff + sHdr->sh_name; char *name = elfPointer(map, offset, soSize); if (!name) - return -ENOEXEC; + return std::make_tuple(nullptr, 0); if (sHdr->sh_type == SHT_DYNSYM && !strcmp(name, ".dynsym")) { dynsym = sHdr; @@ -116,13 +126,13 @@ int elfLoadSymbol(void *dst, size_t size, void *map, size_t soSize, if (dynsym == nullptr) { LOG(IPAModule, Error) << "ELF has no .dynsym section"; - return -ENOEXEC; + return std::make_tuple(nullptr, 0); } offset = eHdr->e_shoff + eHdr->e_shentsize * dynsym->sh_link; sHdr = elfPointer(map, offset, soSize); if (!sHdr) - return -ENOEXEC; + return std::make_tuple(nullptr, 0); off_t dynsym_nameoff = sHdr->sh_offset; /* Locate symbol in the .dynsym section. */ @@ -132,16 +142,16 @@ int elfLoadSymbol(void *dst, size_t size, void *map, size_t soSize, offset = dynsym->sh_offset + dynsym->sh_entsize * i; SymHeader *sym = elfPointer(map, offset, soSize); if (!sym) - return -ENOEXEC; + return std::make_tuple(nullptr, 0); offset = dynsym_nameoff + sym->st_name; char *name = elfPointer(map, offset, soSize, strlen(symbol) + 1); if (!name) - return -ENOEXEC; + return std::make_tuple(nullptr, 0); if (!strcmp(name, symbol) && - sym->st_info & STB_GLOBAL && sym->st_size == size) { + sym->st_info & STB_GLOBAL) { targetSymbol = sym; break; } @@ -149,24 +159,22 @@ int elfLoadSymbol(void *dst, size_t size, void *map, size_t soSize, if (targetSymbol == nullptr) { LOG(IPAModule, Error) << "Symbol " << symbol << " not found"; - return -ENOEXEC; + return std::make_tuple(nullptr, 0); } /* Locate and return data of symbol. */ if (targetSymbol->st_shndx >= eHdr->e_shnum) - return -ENOEXEC; + return std::make_tuple(nullptr, 0); offset = eHdr->e_shoff + targetSymbol->st_shndx * eHdr->e_shentsize; sHdr = elfPointer(map, offset, soSize); if (!sHdr) - return -ENOEXEC; + return std::make_tuple(nullptr, 0); offset = sHdr->sh_offset + (targetSymbol->st_value - sHdr->sh_addr); - char *data = elfPointer(map, offset, soSize, size); + char *data = elfPointer(map, offset, soSize, targetSymbol->st_size); if (!data) - return -ENOEXEC; - - memcpy(dst, data, size); + return std::make_tuple(nullptr, 0); - return 0; + return std::make_tuple(data, targetSymbol->st_size); } } /* namespace */ @@ -257,8 +265,10 @@ int IPAModule::loadIPAModuleInfo() return ret; } - size_t soSize; + void *data; + size_t dataSize; void *map; + size_t soSize; struct stat st; int ret = fstat(fd, &st); if (ret < 0) @@ -275,11 +285,19 @@ int IPAModule::loadIPAModuleInfo() goto unmap; if (sizeof(unsigned long) == 4) - ret = elfLoadSymbol - (&info_, sizeof(info_), map, soSize, "ipaModuleInfo"); + std::tie(data, dataSize) = + elfLoadSymbol + (map, soSize, "ipaModuleInfo"); else - ret = elfLoadSymbol - (&info_, sizeof(info_), map, soSize, "ipaModuleInfo"); + std::tie(data, dataSize) = + elfLoadSymbol + (map, soSize, "ipaModuleInfo"); + + if (data && dataSize == sizeof(info_)) + memcpy(&info_, data, dataSize); + + if (!data) + goto unmap; if (info_.moduleAPIVersion != IPA_MODULE_API_VERSION) { LOG(IPAModule, Error) << "IPA module API version mismatch"; @@ -289,7 +307,7 @@ int IPAModule::loadIPAModuleInfo() unmap: munmap(map, soSize); close: - if (ret) + if (ret || !data) LOG(IPAModule, Error) << "Error loading IPA module info for " << libPath_; -- cgit v1.2.1