You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
93 lines
2.6 KiB
93 lines
2.6 KiB
// Copyright 2020 The Android Open Source Project
|
|
//
|
|
// 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 "android/base/misc/FileUtils.h"
|
|
|
|
#include "android/base/files/ScopedFd.h"
|
|
#include "android/utils/eintr_wrapper.h"
|
|
#include "android/utils/file_io.h"
|
|
#include "android/utils/tempfile.h"
|
|
|
|
#include <fcntl.h>
|
|
#include <gtest/gtest.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/types.h>
|
|
#include <limits>
|
|
#include <string>
|
|
|
|
using android::base::ScopedFd;
|
|
|
|
namespace android {
|
|
|
|
TEST(FileUtils, stringToFile) {
|
|
const char test_pattern[] = "test pattern";
|
|
TempFile* tf = tempfile_create();
|
|
|
|
ScopedFd fd(HANDLE_EINTR(open(tempfile_path(tf), O_RDWR, 0600)));
|
|
EXPECT_NE(-1, fd.get());
|
|
|
|
EXPECT_FALSE(writeStringToFile(-1, test_pattern));
|
|
EXPECT_TRUE(writeStringToFile(fd.get(), test_pattern));
|
|
|
|
std::string dummy_buffer("dummy");
|
|
EXPECT_FALSE(readFileIntoString(-1, &dummy_buffer));
|
|
EXPECT_STREQ("dummy", dummy_buffer.c_str());
|
|
|
|
std::string read_buffer;
|
|
EXPECT_TRUE(readFileIntoString(fd.get(), &read_buffer));
|
|
EXPECT_STREQ(test_pattern, read_buffer.c_str());
|
|
|
|
fd.close();
|
|
|
|
auto readFailedRes = readFileIntoString("!@#%R#$%W$*@#$*");
|
|
EXPECT_FALSE(readFailedRes);
|
|
|
|
auto readSucceededRes = readFileIntoString(tempfile_path(tf));
|
|
EXPECT_TRUE(readSucceededRes);
|
|
EXPECT_STREQ(test_pattern, readSucceededRes->c_str());
|
|
|
|
tempfile_close(tf);
|
|
}
|
|
|
|
// Tests readFileIntoString
|
|
TEST(FileUtils, fileToString) {
|
|
const int bytesToTest = 100;
|
|
|
|
std::vector<uint8_t> testPattern;
|
|
|
|
for (int i = 0; i < bytesToTest; i++) {
|
|
testPattern.push_back(i % 3 == 0 ? 0x0 : 0x1a);
|
|
}
|
|
|
|
TempFile* tf = tempfile_create();
|
|
|
|
ScopedFd fd(HANDLE_EINTR(open(tempfile_path(tf), O_RDWR, 0600)));
|
|
EXPECT_NE(-1, fd.get());
|
|
|
|
HANDLE_EINTR(write(fd.get(), testPattern.data(), bytesToTest));
|
|
|
|
fd.close();
|
|
|
|
const auto testOutput = readFileIntoString(tempfile_path(tf));
|
|
|
|
EXPECT_TRUE(testOutput);
|
|
EXPECT_EQ(bytesToTest, testOutput->size());
|
|
|
|
for (int i = 0; i < bytesToTest; i++) {
|
|
EXPECT_EQ((char)testPattern[i], (char)testOutput->at(i));
|
|
}
|
|
}
|
|
|
|
} // namespace android
|