xref: /system/chre/platform/shared/include/chre/platform/shared/nanoapp_loader.h

/*
 * Copyright (C) 2020 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef CHRE_PLATFORM_SHARED_NANOAPP_LOADER_H_
#define CHRE_PLATFORM_SHARED_NANOAPP_LOADER_H_

#include <cinttypes>
#include <cstdlib>

#include "chre/platform/shared/loader_util.h"

#include "chre/util/dynamic_vector.h"
#include "chre/util/optional.h"

namespace chre {

/**
 * @struct:
 *   AtExitCallback
 *
 * @description:
 *   Store callback information for both atexit and __cxa_atexit.
 *
 * @fields:
 *   func0 ::
 *     Callback function for atexit (no arg).
 *
 *   func1 ::
 *     Callback function for __cxa_atexit (one arg).
 *
 *   arg ::
 *     Optional arg for __cxa_atexit only.
 */
struct AtExitCallback {
  union {
    void (*func0)(void);
    void (*func1)(void *);
  };
  Optional<void *> arg;

  AtExitCallback(void (*func)(void)) {
    func0 = func;
  }

  AtExitCallback(void (*func)(void *), void *a) {
    func1 = func;
    arg = a;
  }
};

/**
 * Provides dynamic loading support for nanoapps on FreeRTOS-based platforms.
 * At a high level, this class is responsible for mapping the provided binary
 * into CHRE's address space, relocating and resolving symbols, and initializing
 * and freeing static data.
 */
class NanoappLoader {
 public:
  NanoappLoader() = delete;

  explicit NanoappLoader(void *elfInput, bool mapIntoTcm) {
    mBinary = static_cast<uint8_t *>(elfInput);
    mIsTcmBinary = mapIntoTcm;
  }

  /**
   * Factory method to create a NanoappLoader Instance after loading
   * the buffer containing the ELF binary.
   *
   * @param elfInput Buffer containing the elf file
   * @param mapIntoTcm Indicates whether the elfBinary should be mapped into
   *     tightly coupled memory.
   * @return Class instance on successful load and verification,
   *     nullptr otherwise.
   */
  static void *create(void *elfInput, bool mapIntoTcm);

  /**
   * Closes and destroys the NanoappLoader instance.
   *
   * @param loader A non-null pointer to the loader that must be destroyed.
   */
  static void destroy(NanoappLoader *loader);

  /**
   * Attempts to locate the exported symbol specified by the given function
   * name.
   *
   * @param name A null-terminated char array that is the name of the function
   *     to be found.
   * @return The address of the function. nullptr if not found.
   */
  static void *findExportedSymbol(const char *name);

  /**
   * Method for pointer lookup by symbol name. Only function pointers
   * are currently supported.
   *
   * @return function pointer on successful lookup, nullptr otherwise
   */
  void *findSymbolByName(const char *name);

  /**
   * Registers a function provided through atexit during static initialization
   * that should be called prior to unloading a nanoapp.
   *
   * @param callback Callback info that should be invoked prior to unloading a
   *     nanoapp.
   */
  void registerAtexitFunction(struct AtExitCallback &cb);

  /**
   * Rounds the given address down to the closest alignment boundary.
   *
   * The alignment follows ELF's p_align semantics:
   *
   * [p_align] holds the value to which the segments are aligned in memory and
   * in the file. Loadable process segments must have congruent values for
   * p_vaddr and p_offset, modulo the page size. Values of zero and one mean no
   * alignment is required. Otherwise, p_align should be a positive, integral
   * power of two, and p_vaddr should equal p_offset, modulo p_align.
   *
   * @param virtualAddr The address to be rounded.
   * @param alignment Alignment to which the address is rounded to.
   * @return An address that is a multiple of the platform's alignment and is
   *     less than or equal to virtualAddr.
   */
  static uintptr_t roundDownToAlign(uintptr_t virtualAddr, size_t alignment);

 private:
  /**
   * Opens the ELF binary. This maps the binary into memory, resolves symbols,
   * and invokes any static initializers.
   *
   * @return true if all required opening steps were completed.
   */
  bool open();

  /**
   * Closes the loader, freeing any state associated with the loaded ELF binary
   * and unmapping it from memory. Prior to unmapping from memory, any static
   * termination functions will be invoked.
   */
  void close();

  using DynamicHeader = ElfW(Dyn);
  using ElfAddr = ElfW(Addr);
  using ElfHeader = ElfW(Ehdr);
  using ElfRel = ElfW(Rel);  // Relocation table entry,
                             // in section of type SHT_REL
  using ElfRela = ElfW(Rela);
  using ElfSym = ElfW(Sym);
  using ElfWord = ElfW(Word);
  using ProgramHeader = ElfW(Phdr);
  using SectionHeader = ElfW(Shdr);

  //! Name of various segments in the ELF that need to be looked up
  static constexpr const char *kSymTableName = ".symtab";
  static constexpr const char *kStrTableName = ".strtab";
  static constexpr const char *kInitArrayName = ".init_array";
  static constexpr const char *kFiniArrayName = ".fini_array";

  //! Pointer to the table of all the section names.
  char *mSectionNamesPtr = nullptr;
  //! Pointer to the table of symbol names of defined symbols.
  char *mStringTablePtr = nullptr;
  //! Pointer to the table of symbol information for defined symbols.
  uint8_t *mSymbolTablePtr = nullptr;
  //! Pointer to the array of section header entries.
  SectionHeader *mSectionHeadersPtr = nullptr;
  //! Number of SectionHeaders pointed to by mSectionHeadersPtr.
  size_t mNumSectionHeaders = 0;
  //! Size of the data pointed to by mSymbolTablePtr.
  size_t mSymbolTableSize = 0;

  //! The ELF that is being mapped into the system. This pointer will be invalid
  //! after open returns.
  uint8_t *mBinary = nullptr;
  //! The starting location of the memory that has been mapped into the system.
  uint8_t *mMapping = nullptr;
  //! The span of memory that has been mapped into the system.
  size_t mMemorySpan = 0;
  //! The difference between where the first load segment was mapped into
  //! virtual memory and what the virtual load offset was of that segment.
  ElfAddr mLoadBias = 0;
  //! Dynamic vector containing functions that should be invoked prior to
  //! unloading this nanoapp. Note that functions are stored in the order they
  //! were added and should be called in reverse.
  DynamicVector<struct AtExitCallback> mAtexitFunctions;
  //! Whether this loader instance is managing a TCM nanoapp binary.
  bool mIsTcmBinary = false;

  /**
   * Invokes all functions registered via atexit during static initialization.
   */
  void callAtexitFunctions();

  /**
   * Invokes all initialization functions in .init_array segment.
   *
   * @return true if static initialization succeeded.
   */
  bool callInitArray();

  /**
   * Invokes all termination functions in the .fini_array segment.
   */
  void callTerminatorArray();

  /**
   * Allocates memory for all load segments that need to be mapped into virtual
   * memory and copies the load segments into the newly allocated memory.
   *
   * @return true if the memory for mapping was allocated and the load segments
   *     were formatted correctly.
   */
  bool createMappings();

  /**
   * Copies various sections and headers from the ELF while verifying that they
   * match the ELF format specification.
   *
   * @return true if all data was copied and verified.
   */
  bool copyAndVerifyHeaders();

  /**
   * Resolves all relocated symbols located in the DT_REL table.
   *
   * @return true if all relocated symbols were resolved.
   */
  bool fixRelocations();

  /**
   * Resolves entries in the Global Offset Table (GOT) to facility the ELF's
   * compiled using position independent code (PIC).
   *
   * @return true if all symbols were resolved.
   */
  bool resolveGot();

  /**
   * Verifies the ELF header has correct values based on the ELF spec.
   *
   * @return true if the header passed verification.
   */
  bool verifyElfHeader();

  /**
   * Verifies basic information about program headers.
   *
   * @return true if the headers passed verification.
   */
  bool verifyProgramHeaders();

  /**
   * Verifies basic information about section headers.
   *
   * @return true if the headers passed verification.
   */
  bool verifySectionHeaders();

  /**
   * Retrieves the symbol at the given position in the symbol table.
   *
   * @param posInSymbolTable The position in the symbol table where information
   *     about the symbol can be found.
   * @return The symbol or nullptr if not found.
   */
  ElfSym *getDynamicSymbol(size_t posInSymbolTable);

  /**
   * Retrieves the symbol name.
   *
   * @param symbol A pointer to the symbol.
   * @return The symbol's name or nullptr if not found.
   */
  const char *getDataName(const ElfSym *symbol);

  /**
   * Retrieves the target address of the symbol.
   *
   * @param symbol A pointer to the symbol.
   * @return The target address or nullptr if the symbol is not defined.
   */
  void *getSymbolTarget(const ElfSym *symbol);

  /**
   * Retrieves the name of the section header located at the given offset in
   * the section name table.
   *
   * @param headerOffset The offset in the section names table where the
   * header is located.
   * @return The section's name or the empty string if the offset is 0.
   */
  const char *getSectionHeaderName(size_t headerOffset);

  /**
   * Frees any data that was allocated as part of loading the ELF into memory.
   */
  void freeAllocatedData();

  /**
   * Ensures the BSS section is properly mapped into memory. If there is a
   * difference between the size of the BSS section in the ELF binary and the
   * size it needs to be in memory, the rest of the section is zeroed out.
   *
   * @param header The ProgramHeader of the BSS section that is being mapped in.
   */
  void mapBss(const ProgramHeader *header);

  /**
   * Resolves the address of an undefined symbol located at the given position
   * in the symbol table. This symbol must be defined and exposed by the given
   * platform in order for it to be resolved successfully.
   *
   * @param posInSymbolTable The position of the undefined symbol in the symbol
   *     table.
   * @return The address of the resolved symbol. nullptr if not found.
   */
  void *resolveData(size_t posInSymbolTable);

  /**
   * @return The address for the dynamic segment. nullptr if not found.
   */
  DynamicHeader *getDynamicHeader();

  /**
   * @return The address of the first read-only segment. nullptr if not found.
   */
  ProgramHeader *getFirstRoSegHeader();

  /**
   * Retrieves the section header with the given name.
   *
   * @param headerName The name of the section header to find.
   * @return The address of the section. nullptr if not found.
   */
  SectionHeader *getSectionHeader(const char *headerName);

  /**
   * @return The ELF header for the binary being loaded. nullptr if it doesn't
   *    exist or no binary is being loaded.
   */
  ElfHeader *getElfHeader();

  /**
   * @return The array of program headers for the binary being loaded. nullptr
   *    if it doesn't exist or no binary is being loaded.
   */
  ProgramHeader *getProgramHeaderArray();

  /**
   * @return The size of the array of program headers for the binary being
   *    loaded. 0 if it doesn't exist or no binary is being loaded.
   */
  size_t getProgramHeaderArraySize();

  /**
   * @return An array of characters containing the symbol names for dynamic
   *    symbols inside the binary being loaded. nullptr if it doesn't exist or
   *    no binary is being loaded.
   */
  char *getDynamicStringTable();

  /**
   * @return An array of dynamic symbol information for the binary being loaded.
   *     nullptr if it doesn't exist or no binary is being loaded.
   */
  uint8_t *getDynamicSymbolTable();

  /**
   * @return The size of the array of dynamic symbol information for the binary
   *     being loaded. 0 if it doesn't exist or no binary is being loaded.
   */
  size_t getDynamicSymbolTableSize();

  /**
   * Returns the first entry in the dynamic header that has a tag that matches
   * the given field.
   *
   * @param dyn The dynamic header for the binary.
   * @param field The field to be searched for.
   * @return The value found at the entry. 0 if the entry isn't found.
   */
  static ElfWord getDynEntry(DynamicHeader *dyn, int field);

  /**
   * Handle reolcation for entries in the specified table.
   *
   * @param dyn The dynamic header for the binary.
   * @param tableTag The dynamic tag (DT_x) of the relocation table.
   * @return True if success or unsupported, false if failure.
   */
  bool relocateTable(DynamicHeader *dyn, int tableTag);
};

}  // namespace chre

#endif  // CHRE_PLATFORM_SHARED_NANOAPP_LOADER_H_