Implement custom section reading/writing (#2284)

1 files changed, 13 insertions, 0 deletions

diff --git a/src/wat-writer.cc b/src/wat-writer.cc
index d696894e..88ff60ba 100644
--- a/src/wat-writer.cc
+++ b/src/wat-writer.cc
@@ -164,6 +164,7 @@ class WatWriter : ModuleContext {
   void WriteTypeEntry(const TypeEntry& type);
   void WriteField(const Field& field);
   void WriteStartFunction(const Var& start);
+  void WriteCustom(const Custom& custom);
 
   class ExprVisitorDelegate;
 
@@ -1613,6 +1614,13 @@ void WatWriter::WriteStartFunction(const Var& start) {
   WriteCloseNewline();
 }
 
+void WatWriter::WriteCustom(const Custom& custom) {
+  WriteOpenSpace("@custom");
+  WriteQuotedString(custom.name, NextChar::Space);
+  WriteQuotedData(custom.data.data(), custom.data.size());
+  WriteCloseNewline();
+}
+
 Result WatWriter::WriteModule() {
   BuildInlineExportMap();
   BuildInlineImportMap();
@@ -1659,6 +1667,11 @@ Result WatWriter::WriteModule() {
         break;
     }
   }
+  if (options_.features.annotations_enabled()) {
+    for (const Custom& custom : module.customs) {
+      WriteCustom(custom);
+    }
+  }
   WriteCloseNewline();
   /* force the newline to be written */
   WriteNextChar();