1 files changed, 111 insertions, 0 deletions
diff --git a/src/tools/wasm-strip.cc b/src/tools/wasm-strip.cc
new file mode 100644
index 00000000..c12d105a
--- /dev/null
+++ b/src/tools/wasm-strip.cc
@@ -0,0 +1,111 @@
+/*
+ * Copyright 2018 WebAssembly Community Group participants
+ *
+ * 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.
+ */
+
+#include "src/binary.h"
+#include "src/binary-reader.h"
+#include "src/binary-reader-nop.h"
+#include "src/error-handler.h"
+#include "src/leb128.h"
+#include "src/option-parser.h"
+#include "src/stream.h"
+
+using namespace wabt;
+
+static std::string s_filename;
+
+static const char s_description[] =
+R"(  Remove sections of a WebAssembly binary file.
+
+examples:
+  # Remove all custom sections from test.wasm
+  $ wasm-strip test.wasm
+)";
+
+static void ParseOptions(int argc, char** argv) {
+  OptionParser parser("wasm-strip", s_description);
+
+  parser.AddHelpOption();
+  parser.AddArgument("filename", OptionParser::ArgumentCount::One,
+                     [](const char* argument) {
+                       s_filename = argument;
+                       ConvertBackslashToSlash(&s_filename);
+                     });
+  parser.Parse(argc, argv);
+}
+
+class BinaryReaderStrip : public BinaryReaderNop {
+ public:
+  explicit BinaryReaderStrip(ErrorHandler* error_handler)
+      : error_handler_(error_handler) {
+    stream_.WriteU32(WABT_BINARY_MAGIC, "WASM_BINARY_MAGIC");
+    stream_.WriteU32(WABT_BINARY_VERSION, "WASM_BINARY_VERSION");
+  }
+
+  bool OnError(ErrorLevel error_level, const char* message) override {
+    return error_handler_->OnError(error_level, state->offset, message);
+  }
+
+  Result BeginSection(BinarySection section_type, Offset size) override {
+    if (section_type == BinarySection::Custom) {
+      return Result::Ok;
+    }
+    stream_.WriteU8Enum(section_type, "section code");
+    WriteU32Leb128(&stream_, size, "section size");
+    stream_.WriteData(state->data + state->offset, size, "section data");
+    return Result::Ok;
+  }
+
+  Result WriteToFile(string_view filename) {
+    return stream_.WriteToFile(filename);
+  }
+
+ private:
+  MemoryStream stream_;
+  ErrorHandler* error_handler_;
+};
+
+int ProgramMain(int argc, char** argv) {
+  Result result;
+
+  InitStdio();
+  ParseOptions(argc, argv);
+
+  std::vector<uint8_t> file_data;
+  result = ReadFile(s_filename.c_str(), &file_data);
+  if (Succeeded(result)) {
+    Features features;
+    ErrorHandlerFile error_handler(Location::Type::Binary);
+    const bool kReadDebugNames = false;
+    const bool kStopOnFirstError = true;
+    const bool kFailOnCustomSectionError = false;
+    ReadBinaryOptions options(features, nullptr, kReadDebugNames,
+                              kStopOnFirstError, kFailOnCustomSectionError);
+
+    BinaryReaderStrip reader(&error_handler);
+    result = ReadBinary(file_data.data(), file_data.size(), &reader, options);
+
+    if (Succeeded(result)) {
+      result = reader.WriteToFile(s_filename);
+    }
+  }
+  return result != Result::Ok;
+}
+
+int main(int argc, char** argv) {
+  WABT_TRY
+  return ProgramMain(argc, argv);
+  WABT_CATCH_BAD_ALLOC_AND_EXIT
+}