mirror of https://github.com/grpc/grpc.git
Mark D. Roth
1 year ago
parent
f22c954ef5
commit
a5133740f0
71 changed files with 465 additions and 1241 deletions
@ -1,78 +0,0 @@ |
|||||||
//
|
|
||||||
//
|
|
||||||
// Copyright 2015 gRPC authors.
|
|
||||||
//
|
|
||||||
// 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 <grpc/support/port_platform.h> |
|
||||||
|
|
||||||
#include "src/core/lib/iomgr/load_file.h" |
|
||||||
|
|
||||||
#include <errno.h> |
|
||||||
#include <string.h> |
|
||||||
|
|
||||||
#include <grpc/support/alloc.h> |
|
||||||
#include <grpc/support/log.h> |
|
||||||
#include <grpc/support/string_util.h> |
|
||||||
|
|
||||||
#include "src/core/lib/gpr/string.h" |
|
||||||
#include "src/core/lib/gprpp/crash.h" |
|
||||||
#include "src/core/lib/iomgr/block_annotate.h" |
|
||||||
|
|
||||||
grpc_error_handle grpc_load_file(const char* filename, int add_null_terminator, |
|
||||||
grpc_slice* output) { |
|
||||||
unsigned char* contents = nullptr; |
|
||||||
size_t contents_size = 0; |
|
||||||
grpc_slice result = grpc_empty_slice(); |
|
||||||
FILE* file; |
|
||||||
size_t bytes_read = 0; |
|
||||||
grpc_error_handle error; |
|
||||||
|
|
||||||
GRPC_SCHEDULING_START_BLOCKING_REGION; |
|
||||||
file = fopen(filename, "rb"); |
|
||||||
if (file == nullptr) { |
|
||||||
error = GRPC_OS_ERROR(errno, "fopen"); |
|
||||||
goto end; |
|
||||||
} |
|
||||||
fseek(file, 0, SEEK_END); |
|
||||||
// Converting to size_t on the assumption that it will not fail
|
|
||||||
contents_size = static_cast<size_t>(ftell(file)); |
|
||||||
fseek(file, 0, SEEK_SET); |
|
||||||
contents = static_cast<unsigned char*>( |
|
||||||
gpr_malloc(contents_size + (add_null_terminator ? 1 : 0))); |
|
||||||
bytes_read = fread(contents, 1, contents_size, file); |
|
||||||
if (bytes_read < contents_size) { |
|
||||||
gpr_free(contents); |
|
||||||
error = GRPC_OS_ERROR(errno, "fread"); |
|
||||||
GPR_ASSERT(ferror(file)); |
|
||||||
goto end; |
|
||||||
} |
|
||||||
if (add_null_terminator) { |
|
||||||
contents[contents_size++] = 0; |
|
||||||
} |
|
||||||
result = grpc_slice_new(contents, contents_size, gpr_free); |
|
||||||
|
|
||||||
end: |
|
||||||
*output = result; |
|
||||||
if (file != nullptr) fclose(file); |
|
||||||
if (!error.ok()) { |
|
||||||
grpc_error_handle error_out = grpc_error_set_str( |
|
||||||
GRPC_ERROR_CREATE_REFERENCING("Failed to load file", &error, 1), |
|
||||||
grpc_core::StatusStrProperty::kFilename, filename); |
|
||||||
error = error_out; |
|
||||||
} |
|
||||||
GRPC_SCHEDULING_END_BLOCKING_REGION_NO_EXEC_CTX; |
|
||||||
return error; |
|
||||||
} |
|
||||||
@ -1,35 +0,0 @@ |
|||||||
//
|
|
||||||
//
|
|
||||||
// Copyright 2015 gRPC authors.
|
|
||||||
//
|
|
||||||
// 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.
|
|
||||||
//
|
|
||||||
//
|
|
||||||
|
|
||||||
#ifndef GRPC_SRC_CORE_LIB_IOMGR_LOAD_FILE_H |
|
||||||
#define GRPC_SRC_CORE_LIB_IOMGR_LOAD_FILE_H |
|
||||||
|
|
||||||
#include <grpc/support/port_platform.h> |
|
||||||
|
|
||||||
#include <stdio.h> |
|
||||||
|
|
||||||
#include <grpc/slice.h> |
|
||||||
|
|
||||||
#include "src/core/lib/iomgr/error.h" |
|
||||||
|
|
||||||
// Loads the content of a file into a slice. add_null_terminator will add
|
|
||||||
// a NULL terminator if non-zero.
|
|
||||||
grpc_error_handle grpc_load_file(const char* filename, int add_null_terminator, |
|
||||||
grpc_slice* output); |
|
||||||
|
|
||||||
#endif // GRPC_SRC_CORE_LIB_IOMGR_LOAD_FILE_H
|
|
||||||
@ -1,161 +0,0 @@ |
|||||||
//
|
|
||||||
//
|
|
||||||
// Copyright 2015 gRPC authors.
|
|
||||||
//
|
|
||||||
// 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/core/lib/iomgr/load_file.h" |
|
||||||
|
|
||||||
#include <stdio.h> |
|
||||||
#include <string.h> |
|
||||||
|
|
||||||
#include <gtest/gtest.h> |
|
||||||
|
|
||||||
#include <grpc/grpc.h> |
|
||||||
#include <grpc/slice.h> |
|
||||||
#include <grpc/support/alloc.h> |
|
||||||
#include <grpc/support/log.h> |
|
||||||
|
|
||||||
#include "src/core/lib/gpr/string.h" |
|
||||||
#include "src/core/lib/gpr/tmpfile.h" |
|
||||||
#include "src/core/lib/gprpp/crash.h" |
|
||||||
#include "test/core/util/test_config.h" |
|
||||||
|
|
||||||
#define LOG_TEST_NAME(x) gpr_log(GPR_INFO, "%s", x) |
|
||||||
|
|
||||||
static const char prefix[] = "file_test"; |
|
||||||
|
|
||||||
TEST(LoadFileTest, TestLoadEmptyFile) { |
|
||||||
FILE* tmp = nullptr; |
|
||||||
grpc_slice slice; |
|
||||||
grpc_slice slice_with_null_term; |
|
||||||
grpc_error_handle error; |
|
||||||
char* tmp_name; |
|
||||||
|
|
||||||
LOG_TEST_NAME("test_load_empty_file"); |
|
||||||
|
|
||||||
tmp = gpr_tmpfile(prefix, &tmp_name); |
|
||||||
ASSERT_NE(tmp_name, nullptr); |
|
||||||
ASSERT_NE(tmp, nullptr); |
|
||||||
fclose(tmp); |
|
||||||
|
|
||||||
error = grpc_load_file(tmp_name, 0, &slice); |
|
||||||
ASSERT_TRUE(error.ok()); |
|
||||||
ASSERT_EQ(GRPC_SLICE_LENGTH(slice), 0); |
|
||||||
|
|
||||||
error = grpc_load_file(tmp_name, 1, &slice_with_null_term); |
|
||||||
ASSERT_TRUE(error.ok()); |
|
||||||
ASSERT_EQ(GRPC_SLICE_LENGTH(slice_with_null_term), 1); |
|
||||||
ASSERT_EQ(GRPC_SLICE_START_PTR(slice_with_null_term)[0], 0); |
|
||||||
|
|
||||||
remove(tmp_name); |
|
||||||
gpr_free(tmp_name); |
|
||||||
grpc_slice_unref(slice); |
|
||||||
grpc_slice_unref(slice_with_null_term); |
|
||||||
} |
|
||||||
|
|
||||||
TEST(LoadFileTest, TestLoadFailure) { |
|
||||||
FILE* tmp = nullptr; |
|
||||||
grpc_slice slice; |
|
||||||
grpc_error_handle error; |
|
||||||
char* tmp_name; |
|
||||||
|
|
||||||
LOG_TEST_NAME("test_load_failure"); |
|
||||||
|
|
||||||
tmp = gpr_tmpfile(prefix, &tmp_name); |
|
||||||
ASSERT_NE(tmp_name, nullptr); |
|
||||||
ASSERT_NE(tmp, nullptr); |
|
||||||
fclose(tmp); |
|
||||||
remove(tmp_name); |
|
||||||
|
|
||||||
error = grpc_load_file(tmp_name, 0, &slice); |
|
||||||
ASSERT_FALSE(error.ok()); |
|
||||||
ASSERT_EQ(GRPC_SLICE_LENGTH(slice), 0); |
|
||||||
gpr_free(tmp_name); |
|
||||||
grpc_slice_unref(slice); |
|
||||||
} |
|
||||||
|
|
||||||
TEST(LoadFileTest, TestLoadSmallFile) { |
|
||||||
FILE* tmp = nullptr; |
|
||||||
grpc_slice slice; |
|
||||||
grpc_slice slice_with_null_term; |
|
||||||
grpc_error_handle error; |
|
||||||
char* tmp_name; |
|
||||||
const char* blah = "blah"; |
|
||||||
|
|
||||||
LOG_TEST_NAME("test_load_small_file"); |
|
||||||
|
|
||||||
tmp = gpr_tmpfile(prefix, &tmp_name); |
|
||||||
ASSERT_NE(tmp_name, nullptr); |
|
||||||
ASSERT_NE(tmp, nullptr); |
|
||||||
ASSERT_EQ(fwrite(blah, 1, strlen(blah), tmp), strlen(blah)); |
|
||||||
fclose(tmp); |
|
||||||
|
|
||||||
error = grpc_load_file(tmp_name, 0, &slice); |
|
||||||
ASSERT_TRUE(error.ok()); |
|
||||||
ASSERT_EQ(GRPC_SLICE_LENGTH(slice), strlen(blah)); |
|
||||||
ASSERT_FALSE(memcmp(GRPC_SLICE_START_PTR(slice), blah, strlen(blah))); |
|
||||||
|
|
||||||
error = grpc_load_file(tmp_name, 1, &slice_with_null_term); |
|
||||||
ASSERT_TRUE(error.ok()); |
|
||||||
ASSERT_EQ(GRPC_SLICE_LENGTH(slice_with_null_term), (strlen(blah) + 1)); |
|
||||||
ASSERT_STREQ((const char*)GRPC_SLICE_START_PTR(slice_with_null_term), blah); |
|
||||||
|
|
||||||
remove(tmp_name); |
|
||||||
gpr_free(tmp_name); |
|
||||||
grpc_slice_unref(slice); |
|
||||||
grpc_slice_unref(slice_with_null_term); |
|
||||||
} |
|
||||||
|
|
||||||
TEST(LoadFileTest, TestLoadBigFile) { |
|
||||||
FILE* tmp = nullptr; |
|
||||||
grpc_slice slice; |
|
||||||
grpc_error_handle error; |
|
||||||
char* tmp_name; |
|
||||||
static const size_t buffer_size = 124631; |
|
||||||
unsigned char* buffer = static_cast<unsigned char*>(gpr_malloc(buffer_size)); |
|
||||||
unsigned char* current; |
|
||||||
size_t i; |
|
||||||
|
|
||||||
LOG_TEST_NAME("test_load_big_file"); |
|
||||||
|
|
||||||
memset(buffer, 42, buffer_size); |
|
||||||
|
|
||||||
tmp = gpr_tmpfile(prefix, &tmp_name); |
|
||||||
ASSERT_NE(tmp, nullptr); |
|
||||||
ASSERT_NE(tmp_name, nullptr); |
|
||||||
ASSERT_EQ(fwrite(buffer, 1, buffer_size, tmp), buffer_size); |
|
||||||
fclose(tmp); |
|
||||||
|
|
||||||
error = grpc_load_file(tmp_name, 0, &slice); |
|
||||||
ASSERT_TRUE(error.ok()); |
|
||||||
ASSERT_EQ(GRPC_SLICE_LENGTH(slice), buffer_size); |
|
||||||
current = GRPC_SLICE_START_PTR(slice); |
|
||||||
for (i = 0; i < buffer_size; i++) { |
|
||||||
ASSERT_EQ(current[i], 42); |
|
||||||
} |
|
||||||
|
|
||||||
remove(tmp_name); |
|
||||||
gpr_free(tmp_name); |
|
||||||
grpc_slice_unref(slice); |
|
||||||
gpr_free(buffer); |
|
||||||
} |
|
||||||
|
|
||||||
int main(int argc, char** argv) { |
|
||||||
grpc::testing::TestEnvironment env(&argc, argv); |
|
||||||
::testing::InitGoogleTest(&argc, argv); |
|
||||||
grpc::testing::TestGrpcScope grpc_scope; |
|
||||||
return RUN_ALL_TESTS(); |
|
||||||
} |
|
||||||
Loading…
Reference in new issue