libshaderc_util/src/io.cc

// Copyright 2015 The Shaderc Authors. All rights reserved.
//
// 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 "libshaderc_util/io.h"

#include "libshaderc_util/universal_unistd.h"

#include <errno.h>
#include <cstring>
#include <fstream>
#include <iostream>

namespace {

// Outputs a descriptive message for errno_value to cerr.
// This may be truncated to 1023 bytes on certain platforms.
void OutputFileErrorMessage(int errno_value) {
#ifdef _MSC_VER
  // If the error message is more than 1023 bytes it will be truncated.
  char buffer[1024];
  strerror_s(buffer, errno_value);
  std::cerr << ": " << buffer << std::endl;
#else
  std::cerr << ": " << strerror(errno_value) << std::endl;
#endif
}

}  // anonymous namespace

namespace shaderc_util {

bool ReadFile(const std::string& input_file_name,
              std::vector<char>* input_data) {
  std::istream* stream = &std::cin;
  std::ifstream input_file;
  if (input_file_name != "-") {
    input_file.open(input_file_name, std::ios_base::binary);
    stream = &input_file;
    if (input_file.fail()) {
      std::cerr << "glslc: error: cannot open input file: '" << input_file_name
                << "'";
      if (access(input_file_name.c_str(), R_OK) != 0) {
        OutputFileErrorMessage(errno);
        return false;
      }
      std::cerr << std::endl;
      return false;
    }
  }
  *input_data = std::vector<char>((std::istreambuf_iterator<char>(*stream)),
                                  std::istreambuf_iterator<char>());
  return true;
}

bool WriteFile(const std::string& output_file_name, string_piece output_data) {
  std::ostream* stream = &std::cout;
  std::ofstream output_file;
  if (output_file_name != "-") {
    output_file.open(output_file_name, std::ios_base::binary);
    stream = &output_file;
    if (output_file.fail()) {
      std::cerr << "glslc: error: cannot open output file: '"
                << output_file_name << "'";
      if (access(output_file_name.c_str(), W_OK) != 0) {
        OutputFileErrorMessage(errno);
        return false;
      }
      std::cerr << std::endl;
      return false;
    }
  }
  if (output_data.size() > 0) {
    stream->write(&output_data[0], output_data.size());
    if (!stream->good()) {
      if (stream == &std::cout) {
        std::cerr << "glslc: error: error writing to standard output"
                  << std::endl;
      } else {
        std::cerr << "glslc: error: error writing to output file: '"
                  << output_file_name << "'" << std::endl;
      }
      return false;
    }
  }
  stream->flush();
  return true;
}

}  // namespace shaderc_util