diff options
Diffstat (limited to 'src/support')
-rw-r--r-- | src/support/file.cpp | 51 | ||||
-rw-r--r-- | src/support/file.h | 51 |
2 files changed, 102 insertions, 0 deletions
diff --git a/src/support/file.cpp b/src/support/file.cpp new file mode 100644 index 000000000..470d07737 --- /dev/null +++ b/src/support/file.cpp @@ -0,0 +1,51 @@ +/* + * Copyright 2015 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 "support/file.h" + +#include <cstdlib> + +std::string wasm::read_file(const std::string &filename, bool debug) { + if (debug) std::cerr << "Loading '" << filename << "'..." << std::endl; + std::ifstream infile(filename); + if (!infile.is_open()) { + std::cerr << "Failed opening '" << filename << "'" << std::endl; + exit(EXIT_FAILURE); + } + infile.seekg(0, std::ios::end); + size_t insize = infile.tellg(); + std::string input(insize + 1, '\0'); + infile.seekg(0); + infile.read(&input[0], insize); + return input; +} + +wasm::Output::Output(const std::string &filename, bool debug) + : outfile(), out([this, filename, debug]() { + std::streambuf *buffer; + if (filename.size()) { + if (debug) std::cerr << "Opening '" << filename << std::endl; + outfile.open(filename, std::ofstream::out | std::ofstream::trunc); + if (!outfile.is_open()) { + std::cerr << "Failed opening '" << filename << "'" << std::endl; + exit(EXIT_FAILURE); + } + buffer = outfile.rdbuf(); + } else { + buffer = std::cout.rdbuf(); + } + return buffer; + }()) {} diff --git a/src/support/file.h b/src/support/file.h new file mode 100644 index 000000000..e9c35f5f3 --- /dev/null +++ b/src/support/file.h @@ -0,0 +1,51 @@ +/* + * Copyright 2015 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. + */ + +// +// FIle helpers. +// + +#ifndef wasm_support_file_h +#define wasm_support_file_h + +#include <fstream> +#include <iostream> +#include <string> +#include <utility> + +namespace wasm { +std::string read_file(const std::string &filename, bool debug); + +class Output { + public: + // An empty filename will open stdout instead. + Output(const std::string &filename, bool debug); + ~Output() = default; + template <typename T> + std::ostream &operator<<(const T &v) { + return out << v; + } + + private: + Output() = delete; + Output(const Output &) = delete; + Output &operator=(const Output &) = delete; + std::ofstream outfile; + std::ostream out; +}; +} + +#endif // wasm_support_file_h |