blob: 87f2f022ef9f996d4dcf2dc78611a701a5387bfa [file] [log] [blame] [edit]
// Copyright 2014 The Bazel 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 "src/main/cpp/util/file.h"
#include <limits.h> // PATH_MAX
#include <algorithm>
#include <cstdlib>
#include <vector>
#include "src/main/cpp/util/errors.h"
#include "src/main/cpp/util/exit_code.h"
#include "src/main/cpp/util/path.h"
#include "src/main/cpp/util/strings.h"
namespace blaze_util {
using std::string;
using std::vector;
bool ReadFrom(file_handle_type handle, string *content, int max_size) {
static const size_t kReadSize = 4096; // read 4K chunks
content->clear();
char buf[kReadSize];
// OPT: This loop generates one spurious read on regular files.
int error;
while (int r = ReadFromHandle(
handle, buf,
max_size > 0 ? std::min(static_cast<size_t>(max_size), kReadSize)
: kReadSize,
&error)) {
if (r < 0) {
if (error == ReadFileResult::INTERRUPTED ||
error == ReadFileResult::AGAIN) {
continue;
}
return false;
}
content->append(buf, r);
if (max_size > 0) {
if (max_size > r) {
max_size -= r;
} else {
break;
}
}
}
return true;
}
bool ReadFrom(file_handle_type handle, void *data, size_t size) {
static const size_t kReadSize = 4096; // read 4K chunks
size_t offset = 0;
int error;
while (int r = ReadFromHandle(handle, reinterpret_cast<char *>(data) + offset,
std::min(kReadSize, size), &error)) {
if (r < 0) {
if (error == ReadFileResult::INTERRUPTED ||
error == ReadFileResult::AGAIN) {
continue;
}
return false;
}
offset += r;
if (size > static_cast<size_t>(r)) {
size -= r;
} else {
break;
}
}
return true;
}
bool WriteFile(const std::string &content, const std::string &filename,
unsigned int perm) {
return WriteFile(content.c_str(), content.size(), filename, perm);
}
bool WriteFile(const std::string &content, const Path &path,
unsigned int perm) {
return WriteFile(content.c_str(), content.size(), path, perm);
}
class DirectoryTreeWalker : public DirectoryEntryConsumer {
public:
DirectoryTreeWalker(vector<string> *files,
_ForEachDirectoryEntry walk_entries)
: _files(files), _walk_entries(walk_entries) {}
void Consume(const string &path, bool follow_directory) override {
if (follow_directory) {
Walk(path);
} else {
_files->push_back(path);
}
}
void Walk(const string &path) { _walk_entries(path, this); }
private:
vector<string> *_files;
_ForEachDirectoryEntry _walk_entries;
};
void GetAllFilesUnder(const string &path, vector<string> *result) {
_GetAllFilesUnder(path, result, &ForEachDirectoryEntry);
}
void _GetAllFilesUnder(const string &path,
vector<string> *result,
_ForEachDirectoryEntry walk_entries) {
DirectoryTreeWalker(result, walk_entries).Walk(path);
}
} // namespace blaze_util