75d70b7306
Before this change, a separate static method named cleanup was used to cleanup the file. Instead, now the close method cleans up the full file object using the platform's close function. Reviewed By: lntue Differential Revision: https://reviews.llvm.org/D153377
100 lines
3.1 KiB
C++
100 lines
3.1 KiB
C++
//===-- Implementation of fopencookie -------------------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "src/stdio/fopencookie.h"
|
|
#include "src/__support/CPP/new.h"
|
|
#include "src/__support/File/file.h"
|
|
|
|
#include "src/errno/libc_errno.h"
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
namespace __llvm_libc {
|
|
|
|
namespace {
|
|
|
|
class CookieFile : public __llvm_libc::File {
|
|
void *cookie;
|
|
cookie_io_functions_t ops;
|
|
|
|
static FileIOResult cookie_write(File *f, const void *data, size_t size);
|
|
static FileIOResult cookie_read(File *f, void *data, size_t size);
|
|
static ErrorOr<long> cookie_seek(File *f, long offset, int whence);
|
|
static int cookie_close(File *f);
|
|
|
|
public:
|
|
CookieFile(void *c, cookie_io_functions_t cops, uint8_t *buffer,
|
|
size_t bufsize, File::ModeFlags mode)
|
|
: File(&cookie_write, &cookie_read, &CookieFile::cookie_seek,
|
|
&cookie_close, buffer, bufsize, 0 /* default buffering mode */,
|
|
true /* File owns buffer */, mode),
|
|
cookie(c), ops(cops) {}
|
|
};
|
|
|
|
FileIOResult CookieFile::cookie_write(File *f, const void *data, size_t size) {
|
|
auto cookie_file = reinterpret_cast<CookieFile *>(f);
|
|
if (cookie_file->ops.write == nullptr)
|
|
return 0;
|
|
return cookie_file->ops.write(cookie_file->cookie,
|
|
reinterpret_cast<const char *>(data), size);
|
|
}
|
|
|
|
FileIOResult CookieFile::cookie_read(File *f, void *data, size_t size) {
|
|
auto cookie_file = reinterpret_cast<CookieFile *>(f);
|
|
if (cookie_file->ops.read == nullptr)
|
|
return 0;
|
|
return cookie_file->ops.read(cookie_file->cookie,
|
|
reinterpret_cast<char *>(data), size);
|
|
}
|
|
|
|
ErrorOr<long> CookieFile::cookie_seek(File *f, long offset, int whence) {
|
|
auto cookie_file = reinterpret_cast<CookieFile *>(f);
|
|
if (cookie_file->ops.seek == nullptr) {
|
|
return Error(EINVAL);
|
|
}
|
|
off64_t offset64 = offset;
|
|
int result = cookie_file->ops.seek(cookie_file->cookie, &offset64, whence);
|
|
if (result == 0)
|
|
return offset64;
|
|
else
|
|
return -1;
|
|
}
|
|
|
|
int CookieFile::cookie_close(File *f) {
|
|
auto cookie_file = reinterpret_cast<CookieFile *>(f);
|
|
if (cookie_file->ops.close == nullptr)
|
|
return 0;
|
|
int retval = cookie_file->ops.close(cookie_file->cookie);
|
|
if (retval != 0)
|
|
return retval;
|
|
delete cookie_file;
|
|
return 0;
|
|
}
|
|
|
|
} // anonymous namespace
|
|
|
|
LLVM_LIBC_FUNCTION(::FILE *, fopencookie,
|
|
(void *cookie, const char *mode,
|
|
cookie_io_functions_t ops)) {
|
|
uint8_t *buffer;
|
|
{
|
|
AllocChecker ac;
|
|
buffer = new (ac) uint8_t[File::DEFAULT_BUFFER_SIZE];
|
|
if (!ac)
|
|
return nullptr;
|
|
}
|
|
AllocChecker ac;
|
|
auto *file = new (ac) CookieFile(
|
|
cookie, ops, buffer, File::DEFAULT_BUFFER_SIZE, File::mode_flags(mode));
|
|
if (!ac)
|
|
return nullptr;
|
|
return reinterpret_cast<::FILE *>(file);
|
|
}
|
|
|
|
} // namespace __llvm_libc
|