+impl pack opcodeRaw

WebAssembly · Jan 27, 2025 · 5fcaded · 5fcaded
1 parent b3ad677
commit 5fcaded
Show file tree

Hide file tree

Showing 3 changed files with 35 additions and 1 deletion.
diff --git a/include/wabt/binary-writer.h b/include/wabt/binary-writer.h
@@ -46,6 +46,11 @@ struct WriteBinaryOptions {
 
 Result WriteBinaryModule(Stream*, const Module*, const WriteBinaryOptions&);
 
+class OpcodeRawExpr;
+Result PackOpcodeRawExpr(OpcodeRawExpr& expr,
+                         Module& module,
+                         const WriteBinaryOptions& options);
+
 void WriteType(Stream* stream, Type type, const char* desc = nullptr);
 
 void WriteStr(Stream* stream,

diff --git a/src/binary-reader-ir.cc b/src/binary-reader-ir.cc
@@ -1911,4 +1911,4 @@ Result ExtractOpcodeRawExpr(OpcodeRawExpr &expr, Module &module,
   return result;
 }
 
-} // namespace wabt
+}  // namespace wabt
diff --git a/src/binary-writer.cc b/src/binary-writer.cc
@@ -384,6 +384,9 @@ class BinaryWriter {
   WABT_DISALLOW_COPY_AND_ASSIGN(BinaryWriter);
 
  public:
+  friend Result wabt::PackOpcodeRawExpr(OpcodeRawExpr& expr,
+                                        Module& module,
+                                        const WriteBinaryOptions& options);
   BinaryWriter(Stream*,
                const WriteBinaryOptions& options,
                const Module* module);
@@ -1865,4 +1868,30 @@ Result WriteBinaryModule(Stream* stream,
   return binary_writer.WriteModule();
 }
 
+Result PackOpcodeRawExpr(OpcodeRawExpr& expr,
+                         Module& module,
+                         const WriteBinaryOptions& options) {
+  if (!expr.is_extracted) {
+    return Result::Ok;
+  }
+
+  MemoryStream stream;
+  BinaryWriter binary_writer(&stream, options, &module);
+
+  binary_writer.WriteExprList(nullptr, expr.extracted_exprs);
+  WriteOpcode(&stream, Opcode::End);
+  auto& ob = stream.output_buffer();
+
+  if (ob.data.size()) {
+    expr.opcode_buffer = std::move(ob.data);
+    stream.ReleaseOutputBuffer();
+    expr.is_extracted = false;
+    expr.extracted_exprs.clear();
+    return Result::Ok;
+  } else {
+    stream.ReleaseOutputBuffer();
+    return Result::Error;
+  }
+}
+
 }  // namespace wabt