/src/WasmEdge/lib/loader/serialize/serial_module.cpp

Source (jump to first uncovered line)
// SPDX-License-Identifier: Apache-2.0
// SPDX-FileCopyrightText: 2019-2024 Second State INC

#include "loader/serialize.h"

#include <functional>

namespace WasmEdge {
namespace Loader {

// Serialize module. See "include/loader/serialize.h".
Expect<std::vector<uint8_t>>
Serializer::serializeModule(const AST::Module &Mod) const noexcept {
  std::vector<uint8_t> OutVec;
  OutVec.reserve(Mod.getMagic().size() + Mod.getVersion().size());
  // Serialize Magic and Version.
  OutVec.insert(OutVec.end(), Mod.getMagic().begin(), Mod.getMagic().end());
  OutVec.insert(OutVec.end(), Mod.getVersion().begin(), Mod.getVersion().end());

  // Sort sections according to start offset.
  using SecVariant =
      std::variant<const AST::CustomSection *, const AST::TypeSection *,
                   const AST::ImportSection *, const AST::FunctionSection *,
                   const AST::TableSection *, const AST::MemorySection *,
                   const AST::GlobalSection *, const AST::ExportSection *,
                   const AST::StartSection *, const AST::ElementSection *,
                   const AST::CodeSection *, const AST::DataSection *,
                   const AST::DataCountSection *, const AST::TagSection *>;
  std::vector<SecVariant> Sections;
  Sections.reserve(Mod.getCustomSections().size() + 13);
  for (auto &CustomSec : Mod.getCustomSections()) {
    Sections.push_back(&CustomSec);
  }
  Sections.push_back(&Mod.getTypeSection());
  Sections.push_back(&Mod.getImportSection());
  Sections.push_back(&Mod.getFunctionSection());
  Sections.push_back(&Mod.getTableSection());
  Sections.push_back(&Mod.getMemorySection());
  Sections.push_back(&Mod.getGlobalSection());
  Sections.push_back(&Mod.getExportSection());
  Sections.push_back(&Mod.getStartSection());
  Sections.push_back(&Mod.getElementSection());
  Sections.push_back(&Mod.getCodeSection());
  Sections.push_back(&Mod.getDataSection());
  Sections.push_back(&Mod.getDataCountSection());
  Sections.push_back(&Mod.getTagSection());
  std::sort(Sections.begin(), Sections.end(), [&](auto &A, auto &B) {
    auto Getter = [](auto &Sec) { return Sec->getStartOffset(); };
    return std::visit(Getter, A) < std::visit(Getter, B);
  });

  // Serialize sections.
  for (auto &Sec : Sections) {
    auto SerVisit = [&OutVec, this](auto &A) -> Expect<void> {
      return serializeSection(*A, OutVec);
    };
    EXPECTED_TRY(std::visit(SerVisit, Sec).map_error([](auto E) {
      spdlog::error(ErrInfo::InfoAST(ASTNodeAttr::Module));
      return E;
    }));
  }
  return OutVec;
}

} // namespace Loader
} // namespace WasmEdge

Coverage Report

Created: 2025-07-01 06:18

Line	Count	Source (jump to first uncovered line)
1		// SPDX-License-Identifier: Apache-2.0
2		// SPDX-FileCopyrightText: 2019-2024 Second State INC
3
4		#include "loader/serialize.h"
5
6		#include <functional>
7
8		namespace WasmEdge {
9		namespace Loader {
10
11		// Serialize module. See "include/loader/serialize.h".
12		Expect<std::vector<uint8_t>>
13	0	Serializer::serializeModule(const AST::Module &Mod) const noexcept {
14	0	std::vector<uint8_t> OutVec;
15	0	OutVec.reserve(Mod.getMagic().size() + Mod.getVersion().size());
16		// Serialize Magic and Version.
17	0	OutVec.insert(OutVec.end(), Mod.getMagic().begin(), Mod.getMagic().end());
18	0	OutVec.insert(OutVec.end(), Mod.getVersion().begin(), Mod.getVersion().end());
19
20		// Sort sections according to start offset.
21	0	using SecVariant =
22	0	std::variant<const AST::CustomSection , const AST::TypeSection ,
23	0	const AST::ImportSection , const AST::FunctionSection ,
24	0	const AST::TableSection , const AST::MemorySection ,
25	0	const AST::GlobalSection , const AST::ExportSection ,
26	0	const AST::StartSection , const AST::ElementSection ,
27	0	const AST::CodeSection , const AST::DataSection ,
28	0	const AST::DataCountSection , const AST::TagSection >;
29	0	std::vector<SecVariant> Sections;
30	0	Sections.reserve(Mod.getCustomSections().size() + 13);
31	0	for (auto &CustomSec : Mod.getCustomSections()) {
32	0	Sections.push_back(&CustomSec);
33	0	}
34	0	Sections.push_back(&Mod.getTypeSection());
35	0	Sections.push_back(&Mod.getImportSection());
36	0	Sections.push_back(&Mod.getFunctionSection());
37	0	Sections.push_back(&Mod.getTableSection());
38	0	Sections.push_back(&Mod.getMemorySection());
39	0	Sections.push_back(&Mod.getGlobalSection());