Merge "Revert "Add a way to turn off unique_fd's operator int.""

This commit is contained in:
Ian Kasprzak 2019-04-25 17:16:55 +00:00 committed by Gerrit Code Review
commit e30fe5fe89
33 changed files with 313 additions and 329 deletions

View file

@ -26,7 +26,6 @@ cc_defaults {
"-Wvla", "-Wvla",
"-DADB_HOST=1", // overridden by adbd_defaults "-DADB_HOST=1", // overridden by adbd_defaults
"-DALLOW_ADBD_ROOT=0", // overridden by adbd_defaults "-DALLOW_ADBD_ROOT=0", // overridden by adbd_defaults
"-DANDROID_BASE_UNIQUE_FD_DISABLE_IMPLICIT_CONVERSION=1",
], ],
cpp_std: "experimental", cpp_std: "experimental",

View file

@ -34,7 +34,7 @@
#include "adb_utils.h" #include "adb_utils.h"
#include "sysdeps.h" #include "sysdeps.h"
bool SendProtocolString(borrowed_fd fd, std::string_view s) { bool SendProtocolString(int fd, std::string_view s) {
unsigned int length = s.size(); unsigned int length = s.size();
if (length > MAX_PAYLOAD - 4) { if (length > MAX_PAYLOAD - 4) {
errno = EMSGSIZE; errno = EMSGSIZE;
@ -47,7 +47,7 @@ bool SendProtocolString(borrowed_fd fd, std::string_view s) {
return WriteFdExactly(fd, str); return WriteFdExactly(fd, str);
} }
bool ReadProtocolString(borrowed_fd fd, std::string* s, std::string* error) { bool ReadProtocolString(int fd, std::string* s, std::string* error) {
char buf[5]; char buf[5];
if (!ReadFdExactly(fd, buf, 4)) { if (!ReadFdExactly(fd, buf, 4)) {
*error = perror_str("protocol fault (couldn't read status length)"); *error = perror_str("protocol fault (couldn't read status length)");
@ -65,57 +65,57 @@ bool ReadProtocolString(borrowed_fd fd, std::string* s, std::string* error) {
return true; return true;
} }
bool SendOkay(borrowed_fd fd) { bool SendOkay(int fd) {
return WriteFdExactly(fd, "OKAY", 4); return WriteFdExactly(fd, "OKAY", 4);
} }
bool SendFail(borrowed_fd fd, std::string_view reason) { bool SendFail(int fd, std::string_view reason) {
return WriteFdExactly(fd, "FAIL", 4) && SendProtocolString(fd, reason); return WriteFdExactly(fd, "FAIL", 4) && SendProtocolString(fd, reason);
} }
bool ReadFdExactly(borrowed_fd fd, void* buf, size_t len) { bool ReadFdExactly(int fd, void* buf, size_t len) {
char* p = reinterpret_cast<char*>(buf); char* p = reinterpret_cast<char*>(buf);
size_t len0 = len; size_t len0 = len;
D("readx: fd=%d wanted=%zu", fd.get(), len); D("readx: fd=%d wanted=%zu", fd, len);
while (len > 0) { while (len > 0) {
int r = adb_read(fd, p, len); int r = adb_read(fd, p, len);
if (r > 0) { if (r > 0) {
len -= r; len -= r;
p += r; p += r;
} else if (r == -1) { } else if (r == -1) {
D("readx: fd=%d error %d: %s", fd.get(), errno, strerror(errno)); D("readx: fd=%d error %d: %s", fd, errno, strerror(errno));
return false; return false;
} else { } else {
D("readx: fd=%d disconnected", fd.get()); D("readx: fd=%d disconnected", fd);
errno = 0; errno = 0;
return false; return false;
} }
} }
VLOG(RWX) << "readx: fd=" << fd.get() << " wanted=" << len0 << " got=" << (len0 - len) << " " VLOG(RWX) << "readx: fd=" << fd << " wanted=" << len0 << " got=" << (len0 - len)
<< dump_hex(reinterpret_cast<const unsigned char*>(buf), len0); << " " << dump_hex(reinterpret_cast<const unsigned char*>(buf), len0);
return true; return true;
} }
bool WriteFdExactly(borrowed_fd fd, const void* buf, size_t len) { bool WriteFdExactly(int fd, const void* buf, size_t len) {
const char* p = reinterpret_cast<const char*>(buf); const char* p = reinterpret_cast<const char*>(buf);
int r; int r;
VLOG(RWX) << "writex: fd=" << fd.get() << " len=" << len << " " VLOG(RWX) << "writex: fd=" << fd << " len=" << len
<< dump_hex(reinterpret_cast<const unsigned char*>(buf), len); << " " << dump_hex(reinterpret_cast<const unsigned char*>(buf), len);
while (len > 0) { while (len > 0) {
r = adb_write(fd, p, len); r = adb_write(fd, p, len);
if (r == -1) { if (r == -1) {
D("writex: fd=%d error %d: %s", fd.get(), errno, strerror(errno)); D("writex: fd=%d error %d: %s", fd, errno, strerror(errno));
if (errno == EAGAIN) { if (errno == EAGAIN) {
std::this_thread::yield(); std::this_thread::yield();
continue; continue;
} else if (errno == EPIPE) { } else if (errno == EPIPE) {
D("writex: fd=%d disconnected", fd.get()); D("writex: fd=%d disconnected", fd);
errno = 0; errno = 0;
return false; return false;
} else { } else {
@ -129,15 +129,15 @@ bool WriteFdExactly(borrowed_fd fd, const void* buf, size_t len) {
return true; return true;
} }
bool WriteFdExactly(borrowed_fd fd, const char* str) { bool WriteFdExactly(int fd, const char* str) {
return WriteFdExactly(fd, str, strlen(str)); return WriteFdExactly(fd, str, strlen(str));
} }
bool WriteFdExactly(borrowed_fd fd, const std::string& str) { bool WriteFdExactly(int fd, const std::string& str) {
return WriteFdExactly(fd, str.c_str(), str.size()); return WriteFdExactly(fd, str.c_str(), str.size());
} }
bool WriteFdFmt(borrowed_fd fd, const char* fmt, ...) { bool WriteFdFmt(int fd, const char* fmt, ...) {
std::string str; std::string str;
va_list ap; va_list ap;
@ -148,7 +148,7 @@ bool WriteFdFmt(borrowed_fd fd, const char* fmt, ...) {
return WriteFdExactly(fd, str); return WriteFdExactly(fd, str);
} }
bool ReadOrderlyShutdown(borrowed_fd fd) { bool ReadOrderlyShutdown(int fd) {
char buf[16]; char buf[16];
// Only call this function if you're sure that the peer does // Only call this function if you're sure that the peer does
@ -178,7 +178,7 @@ bool ReadOrderlyShutdown(borrowed_fd fd) {
// data. We don't repeatedly call adb_read() until we get zero because // data. We don't repeatedly call adb_read() until we get zero because
// we don't know how long that would take, but we do know that the // we don't know how long that would take, but we do know that the
// caller wants to close the socket soon. // caller wants to close the socket soon.
VLOG(RWX) << "ReadOrderlyShutdown(" << fd.get() << ") unexpectedly read " VLOG(RWX) << "ReadOrderlyShutdown(" << fd << ") unexpectedly read "
<< dump_hex(buf, result); << dump_hex(buf, result);
// Shutdown the socket to prevent the caller from reading or writing to // Shutdown the socket to prevent the caller from reading or writing to
// it which doesn't make sense if we just read and discarded some data. // it which doesn't make sense if we just read and discarded some data.

View file

@ -25,16 +25,16 @@
#include "adb_unique_fd.h" #include "adb_unique_fd.h"
// Sends the protocol "OKAY" message. // Sends the protocol "OKAY" message.
bool SendOkay(borrowed_fd fd); bool SendOkay(int fd);
// Sends the protocol "FAIL" message, with the given failure reason. // Sends the protocol "FAIL" message, with the given failure reason.
bool SendFail(borrowed_fd fd, std::string_view reason); bool SendFail(int fd, std::string_view reason);
// Writes a protocol-format string; a four hex digit length followed by the string data. // Writes a protocol-format string; a four hex digit length followed by the string data.
bool SendProtocolString(borrowed_fd fd, std::string_view s); bool SendProtocolString(int fd, std::string_view s);
// Reads a protocol-format string; a four hex digit length followed by the string data. // Reads a protocol-format string; a four hex digit length followed by the string data.
bool ReadProtocolString(borrowed_fd fd, std::string* s, std::string* error); bool ReadProtocolString(int fd, std::string* s, std::string* error);
// Reads exactly len bytes from fd into buf. // Reads exactly len bytes from fd into buf.
// //
@ -42,7 +42,7 @@ bool ReadProtocolString(borrowed_fd fd, std::string* s, std::string* error);
// were read. If EOF was found, errno will be set to 0. // were read. If EOF was found, errno will be set to 0.
// //
// If this function fails, the contents of buf are undefined. // If this function fails, the contents of buf are undefined.
bool ReadFdExactly(borrowed_fd fd, void* buf, size_t len); bool ReadFdExactly(int fd, void* buf, size_t len);
// Given a client socket, wait for orderly/graceful shutdown. Call this: // Given a client socket, wait for orderly/graceful shutdown. Call this:
// //
@ -60,19 +60,19 @@ bool ReadFdExactly(borrowed_fd fd, void* buf, size_t len);
// connect()s from the client to fail with WSAEADDRINUSE on Windows. // connect()s from the client to fail with WSAEADDRINUSE on Windows.
// Returns true if it is sure that orderly/graceful shutdown has occurred with // Returns true if it is sure that orderly/graceful shutdown has occurred with
// no additional data read from the server. // no additional data read from the server.
bool ReadOrderlyShutdown(borrowed_fd fd); bool ReadOrderlyShutdown(int fd);
// Writes exactly len bytes from buf to fd. // Writes exactly len bytes from buf to fd.
// //
// Returns false if there is an error or if the fd was closed before the write // Returns false if there is an error or if the fd was closed before the write
// completed. If the other end of the fd (such as in a socket, pipe, or fifo), // completed. If the other end of the fd (such as in a socket, pipe, or fifo),
// is closed, errno will be set to 0. // is closed, errno will be set to 0.
bool WriteFdExactly(borrowed_fd fd, const void* buf, size_t len); bool WriteFdExactly(int fd, const void* buf, size_t len);
// Same as above, but for strings. // Same as above, but for strings.
bool WriteFdExactly(borrowed_fd fd, const char* s); bool WriteFdExactly(int fd, const char* s);
bool WriteFdExactly(borrowed_fd fd, const std::string& s); bool WriteFdExactly(int fd, const std::string& s);
// Same as above, but formats the string to send. // Same as above, but formats the string to send.
bool WriteFdFmt(borrowed_fd fd, const char* fmt, ...) __attribute__((__format__(__printf__, 2, 3))); bool WriteFdFmt(int fd, const char* fmt, ...) __attribute__((__format__(__printf__, 2, 3)));
#endif /* ADB_IO_H */ #endif /* ADB_IO_H */

View file

@ -32,8 +32,6 @@ using unique_fd = android::base::unique_fd_impl<AdbCloser>;
using unique_fd = android::base::unique_fd; using unique_fd = android::base::unique_fd;
#endif #endif
using android::base::borrowed_fd;
template <typename T> template <typename T>
int adb_close(const android::base::unique_fd_impl<T>&) int adb_close(const android::base::unique_fd_impl<T>&)
__attribute__((__unavailable__("adb_close called on unique_fd"))); __attribute__((__unavailable__("adb_close called on unique_fd")));

View file

@ -234,15 +234,15 @@ std::string perror_str(const char* msg) {
#if !defined(_WIN32) #if !defined(_WIN32)
// Windows version provided in sysdeps_win32.cpp // Windows version provided in sysdeps_win32.cpp
bool set_file_block_mode(borrowed_fd fd, bool block) { bool set_file_block_mode(int fd, bool block) {
int flags = fcntl(fd.get(), F_GETFL, 0); int flags = fcntl(fd, F_GETFL, 0);
if (flags == -1) { if (flags == -1) {
PLOG(ERROR) << "failed to fcntl(F_GETFL) for fd " << fd.get(); PLOG(ERROR) << "failed to fcntl(F_GETFL) for fd " << fd;
return false; return false;
} }
flags = block ? (flags & ~O_NONBLOCK) : (flags | O_NONBLOCK); flags = block ? (flags & ~O_NONBLOCK) : (flags | O_NONBLOCK);
if (fcntl(fd.get(), F_SETFL, flags) != 0) { if (fcntl(fd, F_SETFL, flags) != 0) {
PLOG(ERROR) << "failed to fcntl(F_SETFL) for fd " << fd.get() << ", flags " << flags; PLOG(ERROR) << "failed to fcntl(F_SETFL) for fd " << fd << ", flags " << flags;
return false; return false;
} }
return true; return true;

View file

@ -26,7 +26,6 @@
#include <android-base/macros.h> #include <android-base/macros.h>
#include "adb.h" #include "adb.h"
#include "adb_unique_fd.h"
void close_stdin(); void close_stdin();
@ -52,7 +51,7 @@ std::string perror_str(const char* msg);
[[noreturn]] void error_exit(const char* fmt, ...) __attribute__((__format__(__printf__, 1, 2))); [[noreturn]] void error_exit(const char* fmt, ...) __attribute__((__format__(__printf__, 1, 2)));
[[noreturn]] void perror_exit(const char* fmt, ...) __attribute__((__format__(__printf__, 1, 2))); [[noreturn]] void perror_exit(const char* fmt, ...) __attribute__((__format__(__printf__, 1, 2)));
bool set_file_block_mode(borrowed_fd fd, bool block); bool set_file_block_mode(int fd, bool block);
// Given forward/reverse targets, returns true if they look sane. If an error is found, fills // Given forward/reverse targets, returns true if they look sane. If an error is found, fills
// |error| and returns false. // |error| and returns false.

View file

@ -149,13 +149,13 @@ TEST(adb_utils, mkdirs) {
TEST(adb_utils, set_file_block_mode) { TEST(adb_utils, set_file_block_mode) {
unique_fd fd(adb_open("/dev/null", O_RDWR | O_APPEND)); unique_fd fd(adb_open("/dev/null", O_RDWR | O_APPEND));
ASSERT_GE(fd, 0); ASSERT_GE(fd, 0);
int flags = fcntl(fd.get(), F_GETFL, 0); int flags = fcntl(fd, F_GETFL, 0);
ASSERT_EQ(O_RDWR | O_APPEND, (flags & (O_RDWR | O_APPEND))); ASSERT_EQ(O_RDWR | O_APPEND, (flags & (O_RDWR | O_APPEND)));
ASSERT_TRUE(set_file_block_mode(fd, false)); ASSERT_TRUE(set_file_block_mode(fd, false));
int new_flags = fcntl(fd.get(), F_GETFL, 0); int new_flags = fcntl(fd, F_GETFL, 0);
ASSERT_EQ(flags | O_NONBLOCK, new_flags); ASSERT_EQ(flags | O_NONBLOCK, new_flags);
ASSERT_TRUE(set_file_block_mode(fd, true)); ASSERT_TRUE(set_file_block_mode(fd, true));
new_flags = fcntl(fd.get(), F_GETFL, 0); new_flags = fcntl(fd, F_GETFL, 0);
ASSERT_EQ(flags, new_flags); ASSERT_EQ(flags, new_flags);
} }
#endif #endif

View file

@ -128,7 +128,7 @@ static std::optional<TransportId> switch_socket_transport(int fd, std::string* e
return result; return result;
} }
bool adb_status(borrowed_fd fd, std::string* error) { bool adb_status(int fd, std::string* error) {
char buf[5]; char buf[5];
if (!ReadFdExactly(fd, buf, 4)) { if (!ReadFdExactly(fd, buf, 4)) {
*error = perror_str("protocol fault (couldn't read status)"); *error = perror_str("protocol fault (couldn't read status)");

View file

@ -16,14 +16,13 @@
#pragma once #pragma once
#include <optional>
#include <string>
#include "adb.h" #include "adb.h"
#include "adb_unique_fd.h"
#include "sysdeps.h" #include "sysdeps.h"
#include "transport.h" #include "transport.h"
#include <optional>
#include <string>
// Explicitly check the adb server version. // Explicitly check the adb server version.
// All of the commands below do this implicitly. // All of the commands below do this implicitly.
// Only the first invocation of this function will check the server version. // Only the first invocation of this function will check the server version.
@ -65,7 +64,7 @@ int adb_send_emulator_command(int argc, const char* _Nonnull* _Nonnull argv,
// Reads a standard adb status response (OKAY|FAIL) and returns true in the // Reads a standard adb status response (OKAY|FAIL) and returns true in the
// event of OKAY, false in the event of FAIL or protocol error. // event of OKAY, false in the event of FAIL or protocol error.
bool adb_status(borrowed_fd fd, std::string* _Nonnull error); bool adb_status(int fd, std::string* _Nonnull error);
// Create a host command corresponding to selected transport type/serial. // Create a host command corresponding to selected transport type/serial.
std::string format_host_command(const char* _Nonnull command); std::string format_host_command(const char* _Nonnull command);

View file

@ -262,7 +262,7 @@ static void stdin_raw_restore() {
// stdout/stderr are routed independently and the remote exit code will be // stdout/stderr are routed independently and the remote exit code will be
// returned. // returned.
// if |callback| is non-null, stdout/stderr output will be handled by it. // if |callback| is non-null, stdout/stderr output will be handled by it.
int read_and_dump(borrowed_fd fd, bool use_shell_protocol = false, int read_and_dump(int fd, bool use_shell_protocol = false,
StandardStreamsCallbackInterface* callback = &DEFAULT_STANDARD_STREAMS_CALLBACK) { StandardStreamsCallbackInterface* callback = &DEFAULT_STANDARD_STREAMS_CALLBACK) {
int exit_code = 0; int exit_code = 0;
if (fd < 0) return exit_code; if (fd < 0) return exit_code;
@ -305,9 +305,9 @@ int read_and_dump(borrowed_fd fd, bool use_shell_protocol = false,
} }
length = protocol->data_length(); length = protocol->data_length();
} else { } else {
D("read_and_dump(): pre adb_read(fd=%d)", fd.get()); D("read_and_dump(): pre adb_read(fd=%d)", fd);
length = adb_read(fd, raw_buffer, sizeof(raw_buffer)); length = adb_read(fd, raw_buffer, sizeof(raw_buffer));
D("read_and_dump(): post adb_read(fd=%d): length=%d", fd.get(), length); D("read_and_dump(): post adb_read(fd=%d): length=%d", fd, length);
if (length <= 0) { if (length <= 0) {
break; break;
} }
@ -887,7 +887,7 @@ static int adb_sideload_install(const char* filename, bool rescue_mode) {
return -1; return -1;
} }
fprintf(stderr, "adb: trying pre-KitKat sideload method...\n"); fprintf(stderr, "adb: trying pre-KitKat sideload method...\n");
return adb_sideload_legacy(filename, package_fd.get(), static_cast<int>(sb.st_size)); return adb_sideload_legacy(filename, package_fd, static_cast<int>(sb.st_size));
} }
int opt = SIDELOAD_HOST_BLOCK_SIZE; int opt = SIDELOAD_HOST_BLOCK_SIZE;
@ -1180,14 +1180,14 @@ static int logcat(int argc, const char** argv) {
return send_shell_command(cmd); return send_shell_command(cmd);
} }
static void write_zeros(int bytes, borrowed_fd fd) { static void write_zeros(int bytes, int fd) {
int old_stdin_mode = -1; int old_stdin_mode = -1;
int old_stdout_mode = -1; int old_stdout_mode = -1;
std::vector<char> buf(bytes); std::vector<char> buf(bytes);
D("write_zeros(%d) -> %d", bytes, fd.get()); D("write_zeros(%d) -> %d", bytes, fd);
stdinout_raw_prologue(-1, fd.get(), old_stdin_mode, old_stdout_mode); stdinout_raw_prologue(-1, fd, old_stdin_mode, old_stdout_mode);
if (fd == STDOUT_FILENO) { if (fd == STDOUT_FILENO) {
fwrite(buf.data(), 1, bytes, stdout); fwrite(buf.data(), 1, bytes, stdout);
@ -1196,7 +1196,7 @@ static void write_zeros(int bytes, borrowed_fd fd) {
adb_write(fd, buf.data(), bytes); adb_write(fd, buf.data(), bytes);
} }
stdinout_raw_prologue(-1, fd.get(), old_stdin_mode, old_stdout_mode); stdinout_raw_prologue(-1, fd, old_stdin_mode, old_stdout_mode);
D("write_zeros() finished"); D("write_zeros() finished");
} }

View file

@ -28,11 +28,11 @@ namespace {
class AdbFdTextOutput : public android::TextOutput { class AdbFdTextOutput : public android::TextOutput {
public: public:
explicit AdbFdTextOutput(borrowed_fd fd) : fd_(fd) {} explicit AdbFdTextOutput(int fd) : mFD(fd) {}
private: private:
android::status_t print(const char* txt, size_t len) override { android::status_t print(const char* txt, size_t len) override {
return WriteFdExactly(fd_, txt, len) ? android::OK : -errno; return WriteFdExactly(mFD, txt, len) ? android::OK : -errno;
} }
void moveIndent(int delta) override { /*not implemented*/ void moveIndent(int delta) override { /*not implemented*/
} }
@ -43,7 +43,7 @@ class AdbFdTextOutput : public android::TextOutput {
} }
private: private:
borrowed_fd fd_; int mFD;
}; };
std::vector<std::string_view> parseCmdArgs(std::string_view args) { std::vector<std::string_view> parseCmdArgs(std::string_view args) {
@ -67,11 +67,10 @@ std::vector<std::string_view> parseCmdArgs(std::string_view args) {
} // namespace } // namespace
static int execCmd(std::string_view args, borrowed_fd in, borrowed_fd out, borrowed_fd err) { static int execCmd(std::string_view args, int in, int out, int err) {
AdbFdTextOutput oin(out); AdbFdTextOutput oin(out);
AdbFdTextOutput oerr(err); AdbFdTextOutput oerr(err);
return cmdMain(parseCmdArgs(args), oin, oerr, in.get(), out.get(), err.get(), return cmdMain(parseCmdArgs(args), oin, oerr, in, out, err, RunMode::kLibrary);
RunMode::kLibrary);
} }
int main(int argc, char* const argv[]) { int main(int argc, char* const argv[]) {

View file

@ -235,8 +235,8 @@ static bool handle_send_file(int s, const char* path, uint32_t* timestamp, uid_t
unique_fd fd(adb_open_mode(path, O_WRONLY | O_CREAT | O_EXCL | O_CLOEXEC, mode)); unique_fd fd(adb_open_mode(path, O_WRONLY | O_CREAT | O_EXCL | O_CLOEXEC, mode));
if (posix_fadvise(fd.get(), 0, 0, if (posix_fadvise(fd, 0, 0, POSIX_FADV_SEQUENTIAL | POSIX_FADV_NOREUSE | POSIX_FADV_WILLNEED) <
POSIX_FADV_SEQUENTIAL | POSIX_FADV_NOREUSE | POSIX_FADV_WILLNEED) < 0) { 0) {
D("[ Failed to fadvise: %d ]", errno); D("[ Failed to fadvise: %d ]", errno);
} }
@ -464,9 +464,8 @@ static bool do_recv(int s, const char* path, std::vector<char>& buffer) {
return false; return false;
} }
int rc = posix_fadvise(fd.get(), 0, 0, POSIX_FADV_SEQUENTIAL | POSIX_FADV_NOREUSE); if (posix_fadvise(fd.get(), 0, 0, POSIX_FADV_SEQUENTIAL | POSIX_FADV_NOREUSE) < 0) {
if (rc != 0) { D("[ Failed to fadvise: %d ]", errno);
D("[ Failed to fadvise: %d ]", rc);
} }
syncmsg msg; syncmsg msg;

View file

@ -325,12 +325,12 @@ static int jdwp_control_init(JdwpControl* control, const char* sockname, int soc
addrlen = pathlen + sizeof(addr.sun_family); addrlen = pathlen + sizeof(addr.sun_family);
if (bind(s.get(), reinterpret_cast<sockaddr*>(&addr), addrlen) < 0) { if (bind(s, reinterpret_cast<sockaddr*>(&addr), addrlen) < 0) {
D("could not bind vm debug control socket: %d: %s", errno, strerror(errno)); D("could not bind vm debug control socket: %d: %s", errno, strerror(errno));
return -1; return -1;
} }
if (listen(s.get(), 4) < 0) { if (listen(s, 4) < 0) {
D("listen failed in jdwp control socket: %d: %s", errno, strerror(errno)); D("listen failed in jdwp control socket: %d: %s", errno, strerror(errno));
return -1; return -1;
} }

View file

@ -58,7 +58,7 @@ void reboot_service(unique_fd fd, const std::string& arg) {
sockaddr_un addr = {.sun_family = AF_UNIX}; sockaddr_un addr = {.sun_family = AF_UNIX};
strncpy(addr.sun_path, "/dev/socket/recovery", sizeof(addr.sun_path) - 1); strncpy(addr.sun_path, "/dev/socket/recovery", sizeof(addr.sun_path) - 1);
if (connect(sock.get(), reinterpret_cast<sockaddr*>(&addr), sizeof(addr)) == -1) { if (connect(sock, reinterpret_cast<sockaddr*>(&addr), sizeof(addr)) == -1) {
WriteFdFmt(fd, "reboot (%s) connect\n", strerror(errno)); WriteFdFmt(fd, "reboot (%s) connect\n", strerror(errno));
PLOG(ERROR) << "Couldn't connect to recovery socket"; PLOG(ERROR) << "Couldn't connect to recovery socket";
return; return;

View file

@ -58,7 +58,7 @@ static bool make_block_device_writable(const std::string& dev) {
} }
int OFF = 0; int OFF = 0;
bool result = (ioctl(fd.get(), BLKROSET, &OFF) != -1); bool result = (ioctl(fd, BLKROSET, &OFF) != -1);
return result; return result;
} }
@ -194,7 +194,7 @@ void set_verity_enabled_state_service(unique_fd fd, bool enable) {
} }
if (!android::base::GetBoolProperty("ro.secure", false)) { if (!android::base::GetBoolProperty("ro.secure", false)) {
overlayfs_setup(fd.get(), enable); overlayfs_setup(fd, enable);
WriteFdExactly(fd.get(), "verity not enabled - ENG build\n"); WriteFdExactly(fd.get(), "verity not enabled - ENG build\n");
return; return;
} }
@ -239,7 +239,7 @@ void set_verity_enabled_state_service(unique_fd fd, bool enable) {
} }
} }
} }
if (!any_changed) any_changed = overlayfs_setup(fd.get(), enable); if (!any_changed) any_changed = overlayfs_setup(fd, enable);
if (any_changed) { if (any_changed) {
WriteFdExactly(fd.get(), "Now reboot your device for settings to take effect\n"); WriteFdExactly(fd.get(), "Now reboot your device for settings to take effect\n");

View file

@ -114,7 +114,7 @@
namespace { namespace {
// Reads from |fd| until close or failure. // Reads from |fd| until close or failure.
std::string ReadAll(borrowed_fd fd) { std::string ReadAll(int fd) {
char buffer[512]; char buffer[512];
std::string received; std::string received;
@ -317,10 +317,9 @@ bool Subprocess::ForkAndExec(std::string* error) {
child_stdinout_sfd.reset(OpenPtyChildFd(pts_name, &child_error_sfd)); child_stdinout_sfd.reset(OpenPtyChildFd(pts_name, &child_error_sfd));
} }
dup2(child_stdinout_sfd.get(), STDIN_FILENO); dup2(child_stdinout_sfd, STDIN_FILENO);
dup2(child_stdinout_sfd.get(), STDOUT_FILENO); dup2(child_stdinout_sfd, STDOUT_FILENO);
dup2(child_stderr_sfd != -1 ? child_stderr_sfd.get() : child_stdinout_sfd.get(), dup2(child_stderr_sfd != -1 ? child_stderr_sfd : child_stdinout_sfd, STDERR_FILENO);
STDERR_FILENO);
// exec doesn't trigger destructors, close the FDs manually. // exec doesn't trigger destructors, close the FDs manually.
stdinout_sfd_.reset(-1); stdinout_sfd_.reset(-1);
@ -416,7 +415,7 @@ bool Subprocess::ExecInProcess(Command command, std::string* _Nonnull error) {
} }
} else { } else {
// Raw protocol doesn't support multiple output streams, so combine stdout and stderr. // Raw protocol doesn't support multiple output streams, so combine stdout and stderr.
child_stderr_sfd.reset(dup(child_stdinout_sfd.get())); child_stderr_sfd.reset(dup(child_stdinout_sfd));
} }
D("execinprocess: stdin/stdout FD = %d, stderr FD = %d", stdinout_sfd_.get(), D("execinprocess: stdin/stdout FD = %d, stderr FD = %d", stdinout_sfd_.get(),
@ -538,7 +537,7 @@ void Subprocess::PassDataStreams() {
FD_ZERO(&master_write_set); FD_ZERO(&master_write_set);
for (unique_fd* sfd : {&protocol_sfd_, &stdinout_sfd_, &stderr_sfd_}) { for (unique_fd* sfd : {&protocol_sfd_, &stdinout_sfd_, &stderr_sfd_}) {
if (*sfd != -1) { if (*sfd != -1) {
FD_SET(sfd->get(), &master_read_set); FD_SET(*sfd, &master_read_set);
} }
} }
@ -548,8 +547,8 @@ void Subprocess::PassDataStreams() {
unique_fd* dead_sfd = SelectLoop(&master_read_set, &master_write_set); unique_fd* dead_sfd = SelectLoop(&master_read_set, &master_write_set);
if (dead_sfd) { if (dead_sfd) {
D("closing FD %d", dead_sfd->get()); D("closing FD %d", dead_sfd->get());
FD_CLR(dead_sfd->get(), &master_read_set); FD_CLR(*dead_sfd, &master_read_set);
FD_CLR(dead_sfd->get(), &master_write_set); FD_CLR(*dead_sfd, &master_write_set);
if (dead_sfd == &protocol_sfd_) { if (dead_sfd == &protocol_sfd_) {
// Using SIGHUP is a decent general way to indicate that the // Using SIGHUP is a decent general way to indicate that the
// controlling process is going away. If specific signals are // controlling process is going away. If specific signals are
@ -574,7 +573,7 @@ void Subprocess::PassDataStreams() {
namespace { namespace {
inline bool ValidAndInSet(const unique_fd& sfd, fd_set* set) { inline bool ValidAndInSet(const unique_fd& sfd, fd_set* set) {
return sfd != -1 && FD_ISSET(sfd.get(), set); return sfd != -1 && FD_ISSET(sfd, set);
} }
} // namespace } // namespace
@ -582,8 +581,7 @@ inline bool ValidAndInSet(const unique_fd& sfd, fd_set* set) {
unique_fd* Subprocess::SelectLoop(fd_set* master_read_set_ptr, unique_fd* Subprocess::SelectLoop(fd_set* master_read_set_ptr,
fd_set* master_write_set_ptr) { fd_set* master_write_set_ptr) {
fd_set read_set, write_set; fd_set read_set, write_set;
int select_n = int select_n = std::max(std::max(protocol_sfd_, stdinout_sfd_), stderr_sfd_) + 1;
std::max(std::max(protocol_sfd_.get(), stdinout_sfd_.get()), stderr_sfd_.get()) + 1;
unique_fd* dead_sfd = nullptr; unique_fd* dead_sfd = nullptr;
// Keep calling select() and passing data until an FD closes/errors. // Keep calling select() and passing data until an FD closes/errors.
@ -616,8 +614,8 @@ unique_fd* Subprocess::SelectLoop(fd_set* master_read_set_ptr,
dead_sfd = PassInput(); dead_sfd = PassInput();
// If we didn't finish writing, block on stdin write. // If we didn't finish writing, block on stdin write.
if (input_bytes_left_) { if (input_bytes_left_) {
FD_CLR(protocol_sfd_.get(), master_read_set_ptr); FD_CLR(protocol_sfd_, master_read_set_ptr);
FD_SET(stdinout_sfd_.get(), master_write_set_ptr); FD_SET(stdinout_sfd_, master_write_set_ptr);
} }
} }
@ -626,8 +624,8 @@ unique_fd* Subprocess::SelectLoop(fd_set* master_read_set_ptr,
dead_sfd = PassInput(); dead_sfd = PassInput();
// If we finished writing, go back to blocking on protocol read. // If we finished writing, go back to blocking on protocol read.
if (!input_bytes_left_) { if (!input_bytes_left_) {
FD_SET(protocol_sfd_.get(), master_read_set_ptr); FD_SET(protocol_sfd_, master_read_set_ptr);
FD_CLR(stdinout_sfd_.get(), master_write_set_ptr); FD_CLR(stdinout_sfd_, master_write_set_ptr);
} }
} }
} // while (!dead_sfd) } // while (!dead_sfd)
@ -641,7 +639,7 @@ unique_fd* Subprocess::PassInput() {
if (!input_->Read()) { if (!input_->Read()) {
// Read() uses ReadFdExactly() which sets errno to 0 on EOF. // Read() uses ReadFdExactly() which sets errno to 0 on EOF.
if (errno != 0) { if (errno != 0) {
PLOG(ERROR) << "error reading protocol FD " << protocol_sfd_.get(); PLOG(ERROR) << "error reading protocol FD " << protocol_sfd_;
} }
return &protocol_sfd_; return &protocol_sfd_;
} }
@ -657,7 +655,7 @@ unique_fd* Subprocess::PassInput() {
ws.ws_col = cols; ws.ws_col = cols;
ws.ws_xpixel = x_pixels; ws.ws_xpixel = x_pixels;
ws.ws_ypixel = y_pixels; ws.ws_ypixel = y_pixels;
ioctl(stdinout_sfd_.get(), TIOCSWINSZ, &ws); ioctl(stdinout_sfd_, TIOCSWINSZ, &ws);
} }
break; break;
case ShellProtocol::kIdStdin: case ShellProtocol::kIdStdin:
@ -668,7 +666,8 @@ unique_fd* Subprocess::PassInput() {
if (adb_shutdown(stdinout_sfd_, SHUT_WR) == 0) { if (adb_shutdown(stdinout_sfd_, SHUT_WR) == 0) {
return nullptr; return nullptr;
} }
PLOG(ERROR) << "failed to shutdown writes to FD " << stdinout_sfd_.get(); PLOG(ERROR) << "failed to shutdown writes to FD "
<< stdinout_sfd_;
return &stdinout_sfd_; return &stdinout_sfd_;
} else { } else {
// PTYs can't close just input, so rather than close the // PTYs can't close just input, so rather than close the
@ -689,7 +688,7 @@ unique_fd* Subprocess::PassInput() {
int bytes = adb_write(stdinout_sfd_, input_->data() + index, input_bytes_left_); int bytes = adb_write(stdinout_sfd_, input_->data() + index, input_bytes_left_);
if (bytes == 0 || (bytes < 0 && errno != EAGAIN)) { if (bytes == 0 || (bytes < 0 && errno != EAGAIN)) {
if (bytes < 0) { if (bytes < 0) {
PLOG(ERROR) << "error reading stdin FD " << stdinout_sfd_.get(); PLOG(ERROR) << "error reading stdin FD " << stdinout_sfd_;
} }
// stdin is done, mark this packet as finished and we'll just start // stdin is done, mark this packet as finished and we'll just start
// dumping any further data received from the protocol FD. // dumping any further data received from the protocol FD.
@ -709,14 +708,14 @@ unique_fd* Subprocess::PassOutput(unique_fd* sfd, ShellProtocol::Id id) {
// read() returns EIO if a PTY closes; don't report this as an error, // read() returns EIO if a PTY closes; don't report this as an error,
// it just means the subprocess completed. // it just means the subprocess completed.
if (bytes < 0 && !(type_ == SubprocessType::kPty && errno == EIO)) { if (bytes < 0 && !(type_ == SubprocessType::kPty && errno == EIO)) {
PLOG(ERROR) << "error reading output FD " << sfd->get(); PLOG(ERROR) << "error reading output FD " << *sfd;
} }
return sfd; return sfd;
} }
if (bytes > 0 && !output_->Write(id, bytes)) { if (bytes > 0 && !output_->Write(id, bytes)) {
if (errno != 0) { if (errno != 0) {
PLOG(ERROR) << "error reading protocol FD " << protocol_sfd_.get(); PLOG(ERROR) << "error reading protocol FD " << protocol_sfd_;
} }
return &protocol_sfd_; return &protocol_sfd_;
} }

View file

@ -48,7 +48,7 @@ unique_fd StartSubprocess(std::string name, const char* terminal_type, Subproces
// Sets up in/out and error streams to emulate shell-like behavior. // Sets up in/out and error streams to emulate shell-like behavior.
// //
// Returns an open FD connected to the thread or -1 on failure. // Returns an open FD connected to the thread or -1 on failure.
using Command = int(std::string_view args, borrowed_fd in, borrowed_fd out, borrowed_fd err); using Command = int(std::string_view args, int in, int out, int err);
unique_fd StartCommandInProcess(std::string name, Command command, SubprocessProtocol protocol); unique_fd StartCommandInProcess(std::string name, Command command, SubprocessProtocol protocol);
// Create a pipe containing the error. // Create a pipe containing the error.

View file

@ -77,7 +77,7 @@ void ShellServiceTest::StartTestCommandInProcess(std::string name, Command comma
namespace { namespace {
// Reads raw data from |fd| until it closes or errors. // Reads raw data from |fd| until it closes or errors.
std::string ReadRaw(borrowed_fd fd) { std::string ReadRaw(int fd) {
char buffer[1024]; char buffer[1024];
char *cur_ptr = buffer, *end_ptr = buffer + sizeof(buffer); char *cur_ptr = buffer, *end_ptr = buffer + sizeof(buffer);
@ -93,12 +93,12 @@ std::string ReadRaw(borrowed_fd fd) {
// Reads shell protocol data from |fd| until it closes or errors. Fills // Reads shell protocol data from |fd| until it closes or errors. Fills
// |stdout| and |stderr| with their respective data, and returns the exit code // |stdout| and |stderr| with their respective data, and returns the exit code
// read from the protocol or -1 if an exit code packet was not received. // read from the protocol or -1 if an exit code packet was not received.
int ReadShellProtocol(borrowed_fd fd, std::string* stdout, std::string* stderr) { int ReadShellProtocol(int fd, std::string* stdout, std::string* stderr) {
int exit_code = -1; int exit_code = -1;
stdout->clear(); stdout->clear();
stderr->clear(); stderr->clear();
auto protocol = std::make_unique<ShellProtocol>(fd.get()); auto protocol = std::make_unique<ShellProtocol>(fd);
while (protocol->Read()) { while (protocol->Read()) {
switch (protocol->id()) { switch (protocol->id()) {
case ShellProtocol::kIdStdout: case ShellProtocol::kIdStdout:

View file

@ -21,7 +21,6 @@
#include <android-base/macros.h> #include <android-base/macros.h>
#include "adb.h" #include "adb.h"
#include "adb_unique_fd.h"
// Class to send and receive shell protocol packets. // Class to send and receive shell protocol packets.
// //
@ -61,7 +60,7 @@ class ShellProtocol {
// should be dynamically allocated on the heap instead. // should be dynamically allocated on the heap instead.
// //
// |fd| is an open file descriptor to be used to send or receive packets. // |fd| is an open file descriptor to be used to send or receive packets.
explicit ShellProtocol(borrowed_fd fd); explicit ShellProtocol(int fd);
virtual ~ShellProtocol(); virtual ~ShellProtocol();
// Returns a pointer to the data buffer. // Returns a pointer to the data buffer.
@ -104,7 +103,7 @@ class ShellProtocol {
kHeaderSize = sizeof(Id) + sizeof(length_t) kHeaderSize = sizeof(Id) + sizeof(length_t)
}; };
borrowed_fd fd_; int fd_;
char buffer_[kBufferSize]; char buffer_[kBufferSize];
size_t data_length_ = 0, bytes_left_ = 0; size_t data_length_ = 0, bytes_left_ = 0;

View file

@ -22,7 +22,7 @@
#include "adb_io.h" #include "adb_io.h"
ShellProtocol::ShellProtocol(borrowed_fd fd) : fd_(fd) { ShellProtocol::ShellProtocol(int fd) : fd_(fd) {
buffer_[0] = kIdInvalid; buffer_[0] = kIdInvalid;
} }

View file

@ -314,14 +314,14 @@ int socket_spec_listen(std::string_view spec, std::string* error, int* resolved_
addr.svm_port = port == 0 ? VMADDR_PORT_ANY : port; addr.svm_port = port == 0 ? VMADDR_PORT_ANY : port;
addr.svm_cid = VMADDR_CID_ANY; addr.svm_cid = VMADDR_CID_ANY;
socklen_t addr_len = sizeof(addr); socklen_t addr_len = sizeof(addr);
if (bind(serverfd.get(), reinterpret_cast<struct sockaddr*>(&addr), addr_len)) { if (bind(serverfd, reinterpret_cast<struct sockaddr*>(&addr), addr_len)) {
return -1; return -1;
} }
if (listen(serverfd.get(), 4)) { if (listen(serverfd, 4)) {
return -1; return -1;
} }
if (serverfd >= 0 && resolved_port) { if (serverfd >= 0 && resolved_port) {
if (getsockname(serverfd.get(), reinterpret_cast<sockaddr*>(&addr), &addr_len) == 0) { if (getsockname(serverfd, reinterpret_cast<sockaddr*>(&addr), &addr_len) == 0) {
*resolved_port = addr.svm_port; *resolved_port = addr.svm_port;
} else { } else {
return -1; return -1;

View file

@ -36,7 +36,6 @@
#include <android-base/unique_fd.h> #include <android-base/unique_fd.h>
#include <android-base/utf8.h> #include <android-base/utf8.h>
#include "adb_unique_fd.h"
#include "sysdeps/errno.h" #include "sysdeps/errno.h"
#include "sysdeps/network.h" #include "sysdeps/network.h"
#include "sysdeps/stat.h" #include "sysdeps/stat.h"
@ -77,40 +76,42 @@ static __inline__ bool adb_is_separator(char c) {
extern int adb_thread_setname(const std::string& name); extern int adb_thread_setname(const std::string& name);
static __inline__ void close_on_exec(borrowed_fd fd) { static __inline__ void close_on_exec(int fd)
{
/* nothing really */ /* nothing really */
} }
extern int adb_unlink(const char* path); extern int adb_unlink(const char* path);
#undef unlink #undef unlink
#define unlink ___xxx_unlink #define unlink ___xxx_unlink
extern int adb_mkdir(const std::string& path, int mode); extern int adb_mkdir(const std::string& path, int mode);
#undef mkdir #undef mkdir
#define mkdir ___xxx_mkdir #define mkdir ___xxx_mkdir
// See the comments for the !defined(_WIN32) versions of adb_*(). // See the comments for the !defined(_WIN32) versions of adb_*().
extern int adb_open(const char* path, int options); extern int adb_open(const char* path, int options);
extern int adb_creat(const char* path, int mode); extern int adb_creat(const char* path, int mode);
extern int adb_read(borrowed_fd fd, void* buf, int len); extern int adb_read(int fd, void* buf, int len);
extern int adb_write(borrowed_fd fd, const void* buf, int len); extern int adb_write(int fd, const void* buf, int len);
extern int64_t adb_lseek(borrowed_fd fd, int64_t pos, int where); extern int64_t adb_lseek(int fd, int64_t pos, int where);
extern int adb_shutdown(borrowed_fd fd, int direction = SHUT_RDWR); extern int adb_shutdown(int fd, int direction = SHUT_RDWR);
extern int adb_close(int fd); extern int adb_close(int fd);
extern int adb_register_socket(SOCKET s); extern int adb_register_socket(SOCKET s);
// See the comments for the !defined(_WIN32) version of unix_close(). // See the comments for the !defined(_WIN32) version of unix_close().
static __inline__ int unix_close(int fd) { static __inline__ int unix_close(int fd)
{
return close(fd); return close(fd);
} }
#undef close #undef close
#define close ____xxx_close #define close ____xxx_close
// Like unix_read(), but may return EINTR. // Like unix_read(), but may return EINTR.
extern int unix_read_interruptible(borrowed_fd fd, void* buf, size_t len); extern int unix_read_interruptible(int fd, void* buf, size_t len);
// See the comments for the !defined(_WIN32) version of unix_read(). // See the comments for the !defined(_WIN32) version of unix_read().
static __inline__ int unix_read(borrowed_fd fd, void* buf, size_t len) { static __inline__ int unix_read(int fd, void* buf, size_t len) {
return TEMP_FAILURE_RETRY(unix_read_interruptible(fd, buf, len)); return TEMP_FAILURE_RETRY(unix_read_interruptible(fd, buf, len));
} }
@ -118,21 +119,23 @@ static __inline__ int unix_read(borrowed_fd fd, void* buf, size_t len) {
#define read ___xxx_read #define read ___xxx_read
// See the comments for the !defined(_WIN32) version of unix_write(). // See the comments for the !defined(_WIN32) version of unix_write().
static __inline__ int unix_write(borrowed_fd fd, const void* buf, size_t len) { static __inline__ int unix_write(int fd, const void* buf, size_t len)
return write(fd.get(), buf, len); {
return write(fd, buf, len);
} }
#undef write #undef write
#define write ___xxx_write #define write ___xxx_write
// See the comments for the !defined(_WIN32) version of unix_lseek(). // See the comments for the !defined(_WIN32) version of unix_lseek().
static __inline__ int unix_lseek(borrowed_fd fd, int pos, int where) { static __inline__ int unix_lseek(int fd, int pos, int where) {
return lseek(fd.get(), pos, where); return lseek(fd, pos, where);
} }
#undef lseek #undef lseek
#define lseek ___xxx_lseek #define lseek ___xxx_lseek
// See the comments for the !defined(_WIN32) version of adb_open_mode(). // See the comments for the !defined(_WIN32) version of adb_open_mode().
static __inline__ int adb_open_mode(const char* path, int options, int mode) { static __inline__ int adb_open_mode(const char* path, int options, int mode)
{
return adb_open(path, options); return adb_open(path, options);
} }
@ -149,7 +152,7 @@ extern int unix_open(std::string_view path, int options, ...);
// with |fd| must have GENERIC_READ access (which console FDs have by default). // with |fd| must have GENERIC_READ access (which console FDs have by default).
// Returns 1 if |fd| is a console FD, 0 otherwise. The value of errno after // Returns 1 if |fd| is a console FD, 0 otherwise. The value of errno after
// calling this function is unreliable and should not be used. // calling this function is unreliable and should not be used.
int unix_isatty(borrowed_fd fd); int unix_isatty(int fd);
#define isatty ___xxx_isatty #define isatty ___xxx_isatty
int network_inaddr_any_server(int port, int type, std::string* error); int network_inaddr_any_server(int port, int type, std::string* error);
@ -165,21 +168,20 @@ inline int network_local_server(const char* name, int namespace_id, int type, st
int network_connect(const std::string& host, int port, int type, int timeout, int network_connect(const std::string& host, int port, int type, int timeout,
std::string* error); std::string* error);
extern int adb_socket_accept(borrowed_fd serverfd, struct sockaddr* addr, socklen_t* addrlen); extern int adb_socket_accept(int serverfd, struct sockaddr* addr, socklen_t *addrlen);
#undef accept #undef accept
#define accept ___xxx_accept #define accept ___xxx_accept
// Returns the local port number of a bound socket, or -1 on failure. // Returns the local port number of a bound socket, or -1 on failure.
int adb_socket_get_local_port(borrowed_fd fd); int adb_socket_get_local_port(int fd);
extern int adb_setsockopt(borrowed_fd fd, int level, int optname, const void* optval, extern int adb_setsockopt(int fd, int level, int optname, const void* optval, socklen_t optlen);
socklen_t optlen);
#undef setsockopt #undef setsockopt
#define setsockopt ___xxx_setsockopt #define setsockopt ___xxx_setsockopt
extern int adb_socketpair(int sv[2]); extern int adb_socketpair( int sv[2] );
struct adb_pollfd { struct adb_pollfd {
int fd; int fd;
@ -212,7 +214,8 @@ extern int adb_fputs(const char* buf, FILE* stream);
extern int adb_fputc(int ch, FILE* stream); extern int adb_fputc(int ch, FILE* stream);
extern int adb_putchar(int ch); extern int adb_putchar(int ch);
extern int adb_puts(const char* buf); extern int adb_puts(const char* buf);
extern size_t adb_fwrite(const void* ptr, size_t size, size_t nmemb, FILE* stream); extern size_t adb_fwrite(const void* ptr, size_t size, size_t nmemb,
FILE* stream);
extern FILE* adb_fopen(const char* f, const char* m); extern FILE* adb_fopen(const char* f, const char* m);
@ -341,8 +344,9 @@ static __inline__ bool adb_is_separator(char c) {
return c == '/'; return c == '/';
} }
static __inline__ void close_on_exec(borrowed_fd fd) { static __inline__ void close_on_exec(int fd)
fcntl(fd.get(), F_SETFD, FD_CLOEXEC); {
fcntl( fd, F_SETFD, FD_CLOEXEC );
} }
// Open a file and return a file descriptor that may be used with unix_read(), // Open a file and return a file descriptor that may be used with unix_read(),
@ -370,10 +374,12 @@ static __inline__ int unix_open(std::string_view path, int options, ...) {
// Similar to the two-argument adb_open(), but takes a mode parameter for file // Similar to the two-argument adb_open(), but takes a mode parameter for file
// creation. See adb_open() for more info. // creation. See adb_open() for more info.
static __inline__ int adb_open_mode(const char* pathname, int options, int mode) { static __inline__ int adb_open_mode( const char* pathname, int options, int mode )
return TEMP_FAILURE_RETRY(open(pathname, options, mode)); {
return TEMP_FAILURE_RETRY( open( pathname, options, mode ) );
} }
// Open a file and return a file descriptor that may be used with adb_read(), // Open a file and return a file descriptor that may be used with adb_read(),
// adb_write(), adb_close(), but not unix_read(), unix_write(), unix_close(). // adb_write(), adb_close(), but not unix_read(), unix_write(), unix_close().
// //
@ -381,21 +387,23 @@ static __inline__ int adb_open_mode(const char* pathname, int options, int mode)
// sysdeps_win32.cpp) uses Windows native file I/O and bypasses the C Runtime // sysdeps_win32.cpp) uses Windows native file I/O and bypasses the C Runtime
// and its CR/LF translation. The returned file descriptor should be used with // and its CR/LF translation. The returned file descriptor should be used with
// adb_read(), adb_write(), adb_close(), etc. // adb_read(), adb_write(), adb_close(), etc.
static __inline__ int adb_open(const char* pathname, int options) { static __inline__ int adb_open( const char* pathname, int options )
int fd = TEMP_FAILURE_RETRY(open(pathname, options)); {
if (fd < 0) return -1; int fd = TEMP_FAILURE_RETRY( open( pathname, options ) );
close_on_exec(fd); if (fd < 0)
return -1;
close_on_exec( fd );
return fd; return fd;
} }
#undef open #undef open
#define open ___xxx_open #define open ___xxx_open
static __inline__ int adb_shutdown(borrowed_fd fd, int direction = SHUT_RDWR) { static __inline__ int adb_shutdown(int fd, int direction = SHUT_RDWR) {
return shutdown(fd.get(), direction); return shutdown(fd, direction);
} }
#undef shutdown #undef shutdown
#define shutdown ____xxx_shutdown #define shutdown ____xxx_shutdown
// Closes a file descriptor that came from adb_open() or adb_open_mode(), but // Closes a file descriptor that came from adb_open() or adb_open_mode(), but
// not designed to take a file descriptor from unix_open(). See the comments // not designed to take a file descriptor from unix_open(). See the comments
@ -403,76 +411,81 @@ static __inline__ int adb_shutdown(borrowed_fd fd, int direction = SHUT_RDWR) {
__inline__ int adb_close(int fd) { __inline__ int adb_close(int fd) {
return close(fd); return close(fd);
} }
#undef close #undef close
#define close ____xxx_close #define close ____xxx_close
// On Windows, ADB has an indirection layer for file descriptors. If we get a // On Windows, ADB has an indirection layer for file descriptors. If we get a
// Win32 SOCKET object from an external library, we have to map it in to that // Win32 SOCKET object from an external library, we have to map it in to that
// indirection layer, which this does. // indirection layer, which this does.
__inline__ int adb_register_socket(int s) { __inline__ int adb_register_socket(int s) {
return s; return s;
} }
static __inline__ int adb_read(borrowed_fd fd, void* buf, size_t len) { static __inline__ int adb_read(int fd, void* buf, size_t len)
return TEMP_FAILURE_RETRY(read(fd.get(), buf, len)); {
return TEMP_FAILURE_RETRY( read( fd, buf, len ) );
} }
// Like unix_read(), but does not handle EINTR. // Like unix_read(), but does not handle EINTR.
static __inline__ int unix_read_interruptible(borrowed_fd fd, void* buf, size_t len) { static __inline__ int unix_read_interruptible(int fd, void* buf, size_t len) {
return read(fd.get(), buf, len); return read(fd, buf, len);
} }
#undef read #undef read
#define read ___xxx_read #define read ___xxx_read
static __inline__ int adb_write(borrowed_fd fd, const void* buf, size_t len) { static __inline__ int adb_write(int fd, const void* buf, size_t len)
return TEMP_FAILURE_RETRY(write(fd.get(), buf, len)); {
return TEMP_FAILURE_RETRY( write( fd, buf, len ) );
} }
#undef write #undef write
#define write ___xxx_write #define write ___xxx_write
static __inline__ int64_t adb_lseek(borrowed_fd fd, int64_t pos, int where) { static __inline__ int64_t adb_lseek(int fd, int64_t pos, int where) {
#if defined(__APPLE__) #if defined(__APPLE__)
return lseek(fd.get(), pos, where); return lseek(fd, pos, where);
#else #else
return lseek64(fd.get(), pos, where); return lseek64(fd, pos, where);
#endif #endif
} }
#undef lseek #undef lseek
#define lseek ___xxx_lseek #define lseek ___xxx_lseek
static __inline__ int adb_unlink(const char* path) { static __inline__ int adb_unlink(const char* path)
return unlink(path); {
return unlink(path);
} }
#undef unlink #undef unlink
#define unlink ___xxx_unlink #define unlink ___xxx_unlink
static __inline__ int adb_creat(const char* path, int mode) { static __inline__ int adb_creat(const char* path, int mode)
int fd = TEMP_FAILURE_RETRY(creat(path, mode)); {
int fd = TEMP_FAILURE_RETRY( creat( path, mode ) );
if (fd < 0) return -1; if ( fd < 0 )
return -1;
close_on_exec(fd); close_on_exec(fd);
return fd; return fd;
} }
#undef creat #undef creat
#define creat ___xxx_creat #define creat ___xxx_creat
static __inline__ int unix_isatty(borrowed_fd fd) { static __inline__ int unix_isatty(int fd) {
return isatty(fd.get()); return isatty(fd);
} }
#define isatty ___xxx_isatty #define isatty ___xxx_isatty
// Helper for network_* functions. // Helper for network_* functions.
inline int _fd_set_error_str(int fd, std::string* error) { inline int _fd_set_error_str(int fd, std::string* error) {
if (fd == -1) { if (fd == -1) {
*error = strerror(errno); *error = strerror(errno);
} }
return fd; return fd;
} }
inline int network_inaddr_any_server(int port, int type, std::string* error) { inline int network_inaddr_any_server(int port, int type, std::string* error) {
return _fd_set_error_str(socket_inaddr_any_server(port, type), error); return _fd_set_error_str(socket_inaddr_any_server(port, type), error);
} }
inline int network_local_client(const char* name, int namespace_id, int type, std::string* error) { inline int network_local_client(const char* name, int namespace_id, int type, std::string* error) {
@ -485,21 +498,22 @@ inline int network_local_server(const char* name, int namespace_id, int type, st
int network_connect(const std::string& host, int port, int type, int timeout, std::string* error); int network_connect(const std::string& host, int port, int type, int timeout, std::string* error);
static __inline__ int adb_socket_accept(borrowed_fd serverfd, struct sockaddr* addr, static __inline__ int adb_socket_accept(int serverfd, struct sockaddr* addr, socklen_t *addrlen)
socklen_t* addrlen) { {
int fd; int fd;
fd = TEMP_FAILURE_RETRY(accept(serverfd.get(), addr, addrlen)); fd = TEMP_FAILURE_RETRY( accept( serverfd, addr, addrlen ) );
if (fd >= 0) close_on_exec(fd); if (fd >= 0)
close_on_exec(fd);
return fd; return fd;
} }
#undef accept #undef accept
#define accept ___xxx_accept #define accept ___xxx_accept
inline int adb_socket_get_local_port(borrowed_fd fd) { inline int adb_socket_get_local_port(int fd) {
return socket_get_local_port(fd.get()); return socket_get_local_port(fd);
} }
// Operate on a file descriptor returned from unix_open() or a well-known file // Operate on a file descriptor returned from unix_open() or a well-known file
@ -510,10 +524,10 @@ inline int adb_socket_get_local_port(borrowed_fd fd) {
// Windows implementations (in the ifdef above and in sysdeps_win32.cpp) call // Windows implementations (in the ifdef above and in sysdeps_win32.cpp) call
// into the C Runtime and its configurable CR/LF translation (which is settable // into the C Runtime and its configurable CR/LF translation (which is settable
// via _setmode()). // via _setmode()).
#define unix_read adb_read #define unix_read adb_read
#define unix_write adb_write #define unix_write adb_write
#define unix_lseek adb_lseek #define unix_lseek adb_lseek
#define unix_close adb_close #define unix_close adb_close
static __inline__ int adb_thread_setname(const std::string& name) { static __inline__ int adb_thread_setname(const std::string& name) {
#ifdef __APPLE__ #ifdef __APPLE__
@ -528,31 +542,34 @@ static __inline__ int adb_thread_setname(const std::string& name) {
#endif #endif
} }
static __inline__ int adb_setsockopt(borrowed_fd fd, int level, int optname, const void* optval, static __inline__ int adb_setsockopt( int fd, int level, int optname, const void* optval, socklen_t optlen )
socklen_t optlen) { {
return setsockopt(fd.get(), level, optname, optval, optlen); return setsockopt( fd, level, optname, optval, optlen );
} }
#undef setsockopt #undef setsockopt
#define setsockopt ___xxx_setsockopt #define setsockopt ___xxx_setsockopt
static __inline__ int unix_socketpair(int d, int type, int protocol, int sv[2]) { static __inline__ int unix_socketpair( int d, int type, int protocol, int sv[2] )
return socketpair(d, type, protocol, sv); {
return socketpair( d, type, protocol, sv );
} }
static __inline__ int adb_socketpair(int sv[2]) { static __inline__ int adb_socketpair( int sv[2] )
int rc; {
int rc;
rc = unix_socketpair(AF_UNIX, SOCK_STREAM, 0, sv); rc = unix_socketpair( AF_UNIX, SOCK_STREAM, 0, sv );
if (rc < 0) return -1; if (rc < 0)
return -1;
close_on_exec(sv[0]); close_on_exec( sv[0] );
close_on_exec(sv[1]); close_on_exec( sv[1] );
return 0; return 0;
} }
#undef socketpair #undef socketpair
#define socketpair ___xxx_socketpair #define socketpair ___xxx_socketpair
typedef struct pollfd adb_pollfd; typedef struct pollfd adb_pollfd;
static __inline__ int adb_poll(adb_pollfd* fds, size_t nfds, int timeout) { static __inline__ int adb_poll(adb_pollfd* fds, size_t nfds, int timeout) {
@ -561,12 +578,13 @@ static __inline__ int adb_poll(adb_pollfd* fds, size_t nfds, int timeout) {
#define poll ___xxx_poll #define poll ___xxx_poll
static __inline__ int adb_mkdir(const std::string& path, int mode) { static __inline__ int adb_mkdir(const std::string& path, int mode)
{
return mkdir(path.c_str(), mode); return mkdir(path.c_str(), mode);
} }
#undef mkdir #undef mkdir
#define mkdir ___xxx_mkdir #define mkdir ___xxx_mkdir
static __inline__ int adb_is_absolute_host_path(const char* path) { static __inline__ int adb_is_absolute_host_path(const char* path) {
return path[0] == '/'; return path[0] == '/';
@ -574,15 +592,15 @@ static __inline__ int adb_is_absolute_host_path(const char* path) {
#endif /* !_WIN32 */ #endif /* !_WIN32 */
static inline void disable_tcp_nagle(borrowed_fd fd) { static inline void disable_tcp_nagle(int fd) {
int off = 1; int off = 1;
adb_setsockopt(fd.get(), IPPROTO_TCP, TCP_NODELAY, &off, sizeof(off)); adb_setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &off, sizeof(off));
} }
// Sets TCP socket |fd| to send a keepalive TCP message every |interval_sec| seconds. Set // Sets TCP socket |fd| to send a keepalive TCP message every |interval_sec| seconds. Set
// |interval_sec| to 0 to disable keepalives. If keepalives are enabled, the connection will be // |interval_sec| to 0 to disable keepalives. If keepalives are enabled, the connection will be
// configured to drop after 10 missed keepalives. Returns true on success. // configured to drop after 10 missed keepalives. Returns true on success.
bool set_tcp_keepalive(borrowed_fd fd, int interval_sec); bool set_tcp_keepalive(int fd, int interval_sec);
#if defined(_WIN32) #if defined(_WIN32)
// Win32 defines ERROR, which we don't need, but which conflicts with google3 logging. // Win32 defines ERROR, which we don't need, but which conflicts with google3 logging.

View file

@ -104,13 +104,13 @@ static int _network_loopback_server(bool ipv6, int port, int type, std::string*
socklen_t addrlen = sizeof(addr_storage); socklen_t addrlen = sizeof(addr_storage);
sockaddr* addr = (ipv6 ? loopback_addr6 : loopback_addr4)(&addr_storage, &addrlen, port); sockaddr* addr = (ipv6 ? loopback_addr6 : loopback_addr4)(&addr_storage, &addrlen, port);
if (bind(s.get(), addr, addrlen) != 0) { if (bind(s, addr, addrlen) != 0) {
set_error(error); set_error(error);
return -1; return -1;
} }
if (type == SOCK_STREAM || type == SOCK_SEQPACKET) { if (type == SOCK_STREAM || type == SOCK_SEQPACKET) {
if (listen(s.get(), SOMAXCONN) != 0) { if (listen(s, SOMAXCONN) != 0) {
set_error(error); set_error(error);
return -1; return -1;
} }

View file

@ -18,8 +18,6 @@
#include <sys/types.h> #include <sys/types.h>
#include "adb_unique_fd.h"
#if defined(_WIN32) #if defined(_WIN32)
// Layout of this struct must match struct WSABUF (verified via static assert in sysdeps_win32.cpp) // Layout of this struct must match struct WSABUF (verified via static assert in sysdeps_win32.cpp)
@ -28,15 +26,13 @@ struct adb_iovec {
void* iov_base; void* iov_base;
}; };
ssize_t adb_writev(borrowed_fd fd, const adb_iovec* iov, int iovcnt); ssize_t adb_writev(int fd, const adb_iovec* iov, int iovcnt);
#else #else
#include <sys/uio.h> #include <sys/uio.h>
using adb_iovec = struct iovec; using adb_iovec = struct iovec;
inline ssize_t adb_writev(borrowed_fd fd, const adb_iovec* iov, int iovcnt) { #define adb_writev writev
return writev(fd.get(), iov, iovcnt);
}
#endif #endif

View file

@ -16,7 +16,7 @@
#include "sysdeps.h" #include "sysdeps.h"
bool set_tcp_keepalive(borrowed_fd fd, int interval_sec) { bool set_tcp_keepalive(int fd, int interval_sec) {
int enable = (interval_sec > 0); int enable = (interval_sec > 0);
if (adb_setsockopt(fd, SOL_SOCKET, SO_KEEPALIVE, &enable, sizeof(enable))) { if (adb_setsockopt(fd, SOL_SOCKET, SO_KEEPALIVE, &enable, sizeof(enable))) {
return false; return false;

View file

@ -145,14 +145,16 @@ static std::mutex& _win32_lock = *new std::mutex();
static FHRec _win32_fhs[ WIN32_MAX_FHS ]; static FHRec _win32_fhs[ WIN32_MAX_FHS ];
static int _win32_fh_next; // where to start search for free FHRec static int _win32_fh_next; // where to start search for free FHRec
static FH _fh_from_int(borrowed_fd bfd, const char* func) { static FH
FH f; _fh_from_int( int fd, const char* func )
{
FH f;
int fd = bfd.get();
fd -= WIN32_FH_BASE; fd -= WIN32_FH_BASE;
if (fd < 0 || fd >= WIN32_MAX_FHS) { if (fd < 0 || fd >= WIN32_MAX_FHS) {
D("_fh_from_int: invalid fd %d passed to %s", fd + WIN32_FH_BASE, func); D( "_fh_from_int: invalid fd %d passed to %s", fd + WIN32_FH_BASE,
func );
errno = EBADF; errno = EBADF;
return nullptr; return nullptr;
} }
@ -160,7 +162,8 @@ static FH _fh_from_int(borrowed_fd bfd, const char* func) {
f = &_win32_fhs[fd]; f = &_win32_fhs[fd];
if (f->used == 0) { if (f->used == 0) {
D("_fh_from_int: invalid fd %d passed to %s", fd + WIN32_FH_BASE, func); D( "_fh_from_int: invalid fd %d passed to %s", fd + WIN32_FH_BASE,
func );
errno = EBADF; errno = EBADF;
return nullptr; return nullptr;
} }
@ -168,15 +171,20 @@ static FH _fh_from_int(borrowed_fd bfd, const char* func) {
return f; return f;
} }
static int _fh_to_int(FH f) {
static int
_fh_to_int( FH f )
{
if (f && f->used && f >= _win32_fhs && f < _win32_fhs + WIN32_MAX_FHS) if (f && f->used && f >= _win32_fhs && f < _win32_fhs + WIN32_MAX_FHS)
return (int)(f - _win32_fhs) + WIN32_FH_BASE; return (int)(f - _win32_fhs) + WIN32_FH_BASE;
return -1; return -1;
} }
static FH _fh_alloc(FHClass clazz) { static FH
FH f = nullptr; _fh_alloc( FHClass clazz )
{
FH f = nullptr;
std::lock_guard<std::mutex> lock(_win32_lock); std::lock_guard<std::mutex> lock(_win32_lock);
@ -198,7 +206,10 @@ static FH _fh_alloc(FHClass clazz) {
return nullptr; return nullptr;
} }
static int _fh_close(FH f) {
static int
_fh_close( FH f )
{
// Use lock so that closing only happens once and so that _fh_alloc can't // Use lock so that closing only happens once and so that _fh_alloc can't
// allocate a FH that we're in the middle of closing. // allocate a FH that we're in the middle of closing.
std::lock_guard<std::mutex> lock(_win32_lock); std::lock_guard<std::mutex> lock(_win32_lock);
@ -445,7 +456,7 @@ int adb_creat(const char* path, int mode) {
return _fh_to_int(f); return _fh_to_int(f);
} }
int adb_read(borrowed_fd fd, void* buf, int len) { int adb_read(int fd, void* buf, int len) {
FH f = _fh_from_int(fd, __func__); FH f = _fh_from_int(fd, __func__);
if (f == nullptr) { if (f == nullptr) {
@ -456,7 +467,7 @@ int adb_read(borrowed_fd fd, void* buf, int len) {
return f->clazz->_fh_read(f, buf, len); return f->clazz->_fh_read(f, buf, len);
} }
int adb_write(borrowed_fd fd, const void* buf, int len) { int adb_write(int fd, const void* buf, int len) {
FH f = _fh_from_int(fd, __func__); FH f = _fh_from_int(fd, __func__);
if (f == nullptr) { if (f == nullptr) {
@ -467,7 +478,7 @@ int adb_write(borrowed_fd fd, const void* buf, int len) {
return f->clazz->_fh_write(f, buf, len); return f->clazz->_fh_write(f, buf, len);
} }
ssize_t adb_writev(borrowed_fd fd, const adb_iovec* iov, int iovcnt) { ssize_t adb_writev(int fd, const adb_iovec* iov, int iovcnt) {
FH f = _fh_from_int(fd, __func__); FH f = _fh_from_int(fd, __func__);
if (f == nullptr) { if (f == nullptr) {
@ -478,7 +489,7 @@ ssize_t adb_writev(borrowed_fd fd, const adb_iovec* iov, int iovcnt) {
return f->clazz->_fh_writev(f, iov, iovcnt); return f->clazz->_fh_writev(f, iov, iovcnt);
} }
int64_t adb_lseek(borrowed_fd fd, int64_t pos, int where) { int64_t adb_lseek(int fd, int64_t pos, int where) {
FH f = _fh_from_int(fd, __func__); FH f = _fh_from_int(fd, __func__);
if (!f) { if (!f) {
errno = EBADF; errno = EBADF;
@ -962,11 +973,11 @@ int adb_register_socket(SOCKET s) {
} }
#undef accept #undef accept
int adb_socket_accept(borrowed_fd serverfd, struct sockaddr* addr, socklen_t* addrlen) { int adb_socket_accept(int serverfd, struct sockaddr* addr, socklen_t* addrlen) {
FH serverfh = _fh_from_int(serverfd, __func__); FH serverfh = _fh_from_int(serverfd, __func__);
if (!serverfh || serverfh->clazz != &_fh_socket_class) { if (!serverfh || serverfh->clazz != &_fh_socket_class) {
D("adb_socket_accept: invalid fd %d", serverfd.get()); D("adb_socket_accept: invalid fd %d", serverfd);
errno = EBADF; errno = EBADF;
return -1; return -1;
} }
@ -981,7 +992,7 @@ int adb_socket_accept(borrowed_fd serverfd, struct sockaddr* addr, socklen_t* ad
fh->fh_socket = accept(serverfh->fh_socket, addr, addrlen); fh->fh_socket = accept(serverfh->fh_socket, addr, addrlen);
if (fh->fh_socket == INVALID_SOCKET) { if (fh->fh_socket == INVALID_SOCKET) {
const DWORD err = WSAGetLastError(); const DWORD err = WSAGetLastError();
LOG(ERROR) << "adb_socket_accept: accept on fd " << serverfd.get() LOG(ERROR) << "adb_socket_accept: accept on fd " << serverfd
<< " failed: " + android::base::SystemErrorCodeToString(err); << " failed: " + android::base::SystemErrorCodeToString(err);
_socket_set_errno(err); _socket_set_errno(err);
return -1; return -1;
@ -989,16 +1000,16 @@ int adb_socket_accept(borrowed_fd serverfd, struct sockaddr* addr, socklen_t* ad
const int fd = _fh_to_int(fh.get()); const int fd = _fh_to_int(fh.get());
snprintf(fh->name, sizeof(fh->name), "%d(accept:%s)", fd, serverfh->name); snprintf(fh->name, sizeof(fh->name), "%d(accept:%s)", fd, serverfh->name);
D("adb_socket_accept on fd %d returns fd %d", serverfd.get(), fd); D("adb_socket_accept on fd %d returns fd %d", serverfd, fd);
fh.release(); fh.release();
return fd; return fd;
} }
int adb_setsockopt(borrowed_fd fd, int level, int optname, const void* optval, socklen_t optlen) { int adb_setsockopt(int fd, int level, int optname, const void* optval, socklen_t optlen) {
FH fh = _fh_from_int(fd, __func__); FH fh = _fh_from_int(fd, __func__);
if (!fh || fh->clazz != &_fh_socket_class) { if (!fh || fh->clazz != &_fh_socket_class) {
D("adb_setsockopt: invalid fd %d", fd.get()); D("adb_setsockopt: invalid fd %d", fd);
errno = EBADF; errno = EBADF;
return -1; return -1;
} }
@ -1011,7 +1022,7 @@ int adb_setsockopt(borrowed_fd fd, int level, int optname, const void* optval, s
setsockopt(fh->fh_socket, level, optname, reinterpret_cast<const char*>(optval), optlen); setsockopt(fh->fh_socket, level, optname, reinterpret_cast<const char*>(optval), optlen);
if (result == SOCKET_ERROR) { if (result == SOCKET_ERROR) {
const DWORD err = WSAGetLastError(); const DWORD err = WSAGetLastError();
D("adb_setsockopt: setsockopt on fd %d level %d optname %d failed: %s\n", fd.get(), level, D("adb_setsockopt: setsockopt on fd %d level %d optname %d failed: %s\n", fd, level,
optname, android::base::SystemErrorCodeToString(err).c_str()); optname, android::base::SystemErrorCodeToString(err).c_str());
_socket_set_errno(err); _socket_set_errno(err);
result = -1; result = -1;
@ -1019,11 +1030,11 @@ int adb_setsockopt(borrowed_fd fd, int level, int optname, const void* optval, s
return result; return result;
} }
static int adb_getsockname(borrowed_fd fd, struct sockaddr* sockaddr, socklen_t* optlen) { int adb_getsockname(int fd, struct sockaddr* sockaddr, socklen_t* optlen) {
FH fh = _fh_from_int(fd, __func__); FH fh = _fh_from_int(fd, __func__);
if (!fh || fh->clazz != &_fh_socket_class) { if (!fh || fh->clazz != &_fh_socket_class) {
D("adb_getsockname: invalid fd %d", fd.get()); D("adb_getsockname: invalid fd %d", fd);
errno = EBADF; errno = EBADF;
return -1; return -1;
} }
@ -1031,7 +1042,7 @@ static int adb_getsockname(borrowed_fd fd, struct sockaddr* sockaddr, socklen_t*
int result = getsockname(fh->fh_socket, sockaddr, optlen); int result = getsockname(fh->fh_socket, sockaddr, optlen);
if (result == SOCKET_ERROR) { if (result == SOCKET_ERROR) {
const DWORD err = WSAGetLastError(); const DWORD err = WSAGetLastError();
D("adb_getsockname: setsockopt on fd %d failed: %s\n", fd.get(), D("adb_getsockname: setsockopt on fd %d failed: %s\n", fd,
android::base::SystemErrorCodeToString(err).c_str()); android::base::SystemErrorCodeToString(err).c_str());
_socket_set_errno(err); _socket_set_errno(err);
result = -1; result = -1;
@ -1039,7 +1050,7 @@ static int adb_getsockname(borrowed_fd fd, struct sockaddr* sockaddr, socklen_t*
return result; return result;
} }
int adb_socket_get_local_port(borrowed_fd fd) { int adb_socket_get_local_port(int fd) {
sockaddr_storage addr_storage; sockaddr_storage addr_storage;
socklen_t addr_len = sizeof(addr_storage); socklen_t addr_len = sizeof(addr_storage);
@ -1057,11 +1068,11 @@ int adb_socket_get_local_port(borrowed_fd fd) {
return ntohs(reinterpret_cast<sockaddr_in*>(&addr_storage)->sin_port); return ntohs(reinterpret_cast<sockaddr_in*>(&addr_storage)->sin_port);
} }
int adb_shutdown(borrowed_fd fd, int direction) { int adb_shutdown(int fd, int direction) {
FH f = _fh_from_int(fd, __func__); FH f = _fh_from_int(fd, __func__);
if (!f || f->clazz != &_fh_socket_class) { if (!f || f->clazz != &_fh_socket_class) {
D("adb_shutdown: invalid fd %d", fd.get()); D("adb_shutdown: invalid fd %d", fd);
errno = EBADF; errno = EBADF;
return -1; return -1;
} }
@ -1069,7 +1080,7 @@ int adb_shutdown(borrowed_fd fd, int direction) {
D("adb_shutdown: %s", f->name); D("adb_shutdown: %s", f->name);
if (shutdown(f->fh_socket, direction) == SOCKET_ERROR) { if (shutdown(f->fh_socket, direction) == SOCKET_ERROR) {
const DWORD err = WSAGetLastError(); const DWORD err = WSAGetLastError();
D("socket shutdown fd %d failed: %s", fd.get(), D("socket shutdown fd %d failed: %s", fd,
android::base::SystemErrorCodeToString(err).c_str()); android::base::SystemErrorCodeToString(err).c_str());
_socket_set_errno(err); _socket_set_errno(err);
return -1; return -1;
@ -1127,12 +1138,12 @@ fail:
return -1; return -1;
} }
bool set_file_block_mode(borrowed_fd fd, bool block) { bool set_file_block_mode(int fd, bool block) {
FH fh = _fh_from_int(fd, __func__); FH fh = _fh_from_int(fd, __func__);
if (!fh || !fh->used) { if (!fh || !fh->used) {
errno = EBADF; errno = EBADF;
D("Setting nonblocking on bad file descriptor %d", fd.get()); D("Setting nonblocking on bad file descriptor %d", fd);
return false; return false;
} }
@ -1141,22 +1152,22 @@ bool set_file_block_mode(borrowed_fd fd, bool block) {
if (ioctlsocket(fh->u.socket, FIONBIO, &x) != 0) { if (ioctlsocket(fh->u.socket, FIONBIO, &x) != 0) {
int error = WSAGetLastError(); int error = WSAGetLastError();
_socket_set_errno(error); _socket_set_errno(error);
D("Setting %d nonblocking failed (%d)", fd.get(), error); D("Setting %d nonblocking failed (%d)", fd, error);
return false; return false;
} }
return true; return true;
} else { } else {
errno = ENOTSOCK; errno = ENOTSOCK;
D("Setting nonblocking on non-socket %d", fd.get()); D("Setting nonblocking on non-socket %d", fd);
return false; return false;
} }
} }
bool set_tcp_keepalive(borrowed_fd fd, int interval_sec) { bool set_tcp_keepalive(int fd, int interval_sec) {
FH fh = _fh_from_int(fd, __func__); FH fh = _fh_from_int(fd, __func__);
if (!fh || fh->clazz != &_fh_socket_class) { if (!fh || fh->clazz != &_fh_socket_class) {
D("set_tcp_keepalive(%d) failed: invalid fd", fd.get()); D("set_tcp_keepalive(%d) failed: invalid fd", fd);
errno = EBADF; errno = EBADF;
return false; return false;
} }
@ -1170,7 +1181,7 @@ bool set_tcp_keepalive(borrowed_fd fd, int interval_sec) {
if (WSAIoctl(fh->fh_socket, SIO_KEEPALIVE_VALS, &keepalive, sizeof(keepalive), nullptr, 0, if (WSAIoctl(fh->fh_socket, SIO_KEEPALIVE_VALS, &keepalive, sizeof(keepalive), nullptr, 0,
&bytes_returned, nullptr, nullptr) != 0) { &bytes_returned, nullptr, nullptr) != 0) {
const DWORD err = WSAGetLastError(); const DWORD err = WSAGetLastError();
D("set_tcp_keepalive(%d) failed: %s", fd.get(), D("set_tcp_keepalive(%d) failed: %s", fd,
android::base::SystemErrorCodeToString(err).c_str()); android::base::SystemErrorCodeToString(err).c_str());
_socket_set_errno(err); _socket_set_errno(err);
return false; return false;
@ -1217,12 +1228,12 @@ bool set_tcp_keepalive(borrowed_fd fd, int interval_sec) {
// Returns a console HANDLE if |fd| is a console, otherwise returns nullptr. // Returns a console HANDLE if |fd| is a console, otherwise returns nullptr.
// If a valid HANDLE is returned and |mode| is not null, |mode| is also filled // If a valid HANDLE is returned and |mode| is not null, |mode| is also filled
// with the console mode. Requires GENERIC_READ access to the underlying HANDLE. // with the console mode. Requires GENERIC_READ access to the underlying HANDLE.
static HANDLE _get_console_handle(borrowed_fd fd, DWORD* mode = nullptr) { static HANDLE _get_console_handle(int fd, DWORD* mode=nullptr) {
// First check isatty(); this is very fast and eliminates most non-console // First check isatty(); this is very fast and eliminates most non-console
// FDs, but returns 1 for both consoles and character devices like NUL. // FDs, but returns 1 for both consoles and character devices like NUL.
#pragma push_macro("isatty") #pragma push_macro("isatty")
#undef isatty #undef isatty
if (!isatty(fd.get())) { if (!isatty(fd)) {
return nullptr; return nullptr;
} }
#pragma pop_macro("isatty") #pragma pop_macro("isatty")
@ -1230,7 +1241,7 @@ static HANDLE _get_console_handle(borrowed_fd fd, DWORD* mode = nullptr) {
// To differentiate between character devices and consoles we need to get // To differentiate between character devices and consoles we need to get
// the underlying HANDLE and use GetConsoleMode(), which is what requires // the underlying HANDLE and use GetConsoleMode(), which is what requires
// GENERIC_READ permissions. // GENERIC_READ permissions.
const intptr_t intptr_handle = _get_osfhandle(fd.get()); const intptr_t intptr_handle = _get_osfhandle(fd);
if (intptr_handle == -1) { if (intptr_handle == -1) {
return nullptr; return nullptr;
} }
@ -1254,7 +1265,7 @@ static HANDLE _get_console_handle(FILE* const stream) {
return _get_console_handle(fd); return _get_console_handle(fd);
} }
int unix_isatty(borrowed_fd fd) { int unix_isatty(int fd) {
return _get_console_handle(fd) ? 1 : 0; return _get_console_handle(fd) ? 1 : 0;
} }
@ -1634,7 +1645,7 @@ static char _get_decimal_char() {
// Prefix the len bytes in buf with the escape character, and then return the // Prefix the len bytes in buf with the escape character, and then return the
// new buffer length. // new buffer length.
static size_t _escape_prefix(char* const buf, const size_t len) { size_t _escape_prefix(char* const buf, const size_t len) {
// If nothing to prefix, don't do anything. We might be called with // If nothing to prefix, don't do anything. We might be called with
// len == 0, if alt was held down with a dead key which produced nothing. // len == 0, if alt was held down with a dead key which produced nothing.
if (len == 0) { if (len == 0) {
@ -2062,7 +2073,7 @@ void stdin_raw_restore() {
} }
// Called by 'adb shell' and 'adb exec-in' (via unix_read()) to read from stdin. // Called by 'adb shell' and 'adb exec-in' (via unix_read()) to read from stdin.
int unix_read_interruptible(borrowed_fd fd, void* buf, size_t len) { int unix_read_interruptible(int fd, void* buf, size_t len) {
if ((fd == STDIN_FILENO) && (_console_handle != nullptr)) { if ((fd == STDIN_FILENO) && (_console_handle != nullptr)) {
// If it is a request to read from stdin, and stdin_raw_init() has been // If it is a request to read from stdin, and stdin_raw_init() has been
// called, and it successfully configured the console, then read from // called, and it successfully configured the console, then read from
@ -2082,7 +2093,7 @@ int unix_read_interruptible(borrowed_fd fd, void* buf, size_t len) {
// plain read() in favor of unix_read() or adb_read(). // plain read() in favor of unix_read() or adb_read().
#pragma push_macro("read") #pragma push_macro("read")
#undef read #undef read
return read(fd.get(), buf, len); return read(fd, buf, len);
#pragma pop_macro("read") #pragma pop_macro("read")
} }
} }

View file

@ -29,7 +29,7 @@
namespace android { namespace android {
namespace base { namespace base {
ssize_t SendFileDescriptorVector(borrowed_fd sockfd, const void* data, size_t len, ssize_t SendFileDescriptorVector(int sockfd, const void* data, size_t len,
const std::vector<int>& fds) { const std::vector<int>& fds) {
size_t cmsg_space = CMSG_SPACE(sizeof(int) * fds.size()); size_t cmsg_space = CMSG_SPACE(sizeof(int) * fds.size());
size_t cmsg_len = CMSG_LEN(sizeof(int) * fds.size()); size_t cmsg_len = CMSG_LEN(sizeof(int) * fds.size());
@ -67,10 +67,10 @@ ssize_t SendFileDescriptorVector(borrowed_fd sockfd, const void* data, size_t le
int flags = 0; int flags = 0;
#endif #endif
return TEMP_FAILURE_RETRY(sendmsg(sockfd.get(), &msg, flags)); return TEMP_FAILURE_RETRY(sendmsg(sockfd, &msg, flags));
} }
ssize_t ReceiveFileDescriptorVector(borrowed_fd sockfd, void* data, size_t len, size_t max_fds, ssize_t ReceiveFileDescriptorVector(int sockfd, void* data, size_t len, size_t max_fds,
std::vector<unique_fd>* fds) { std::vector<unique_fd>* fds) {
fds->clear(); fds->clear();
@ -98,7 +98,7 @@ ssize_t ReceiveFileDescriptorVector(borrowed_fd sockfd, void* data, size_t len,
flags |= MSG_CMSG_CLOEXEC | MSG_NOSIGNAL; flags |= MSG_CMSG_CLOEXEC | MSG_NOSIGNAL;
#endif #endif
ssize_t rc = TEMP_FAILURE_RETRY(recvmsg(sockfd.get(), &msg, flags)); ssize_t rc = TEMP_FAILURE_RETRY(recvmsg(sockfd, &msg, flags));
if (rc == -1) { if (rc == -1) {
return -1; return -1;

View file

@ -176,20 +176,20 @@ namespace base {
// Versions of standard library APIs that support UTF-8 strings. // Versions of standard library APIs that support UTF-8 strings.
using namespace android::base::utf8; using namespace android::base::utf8;
bool ReadFdToString(borrowed_fd fd, std::string* content) { bool ReadFdToString(int fd, std::string* content) {
content->clear(); content->clear();
// Although original we had small files in mind, this code gets used for // Although original we had small files in mind, this code gets used for
// very large files too, where the std::string growth heuristics might not // very large files too, where the std::string growth heuristics might not
// be suitable. https://code.google.com/p/android/issues/detail?id=258500. // be suitable. https://code.google.com/p/android/issues/detail?id=258500.
struct stat sb; struct stat sb;
if (fstat(fd.get(), &sb) != -1 && sb.st_size > 0) { if (fstat(fd, &sb) != -1 && sb.st_size > 0) {
content->reserve(sb.st_size); content->reserve(sb.st_size);
} }
char buf[BUFSIZ]; char buf[BUFSIZ];
ssize_t n; ssize_t n;
while ((n = TEMP_FAILURE_RETRY(read(fd.get(), &buf[0], sizeof(buf)))) > 0) { while ((n = TEMP_FAILURE_RETRY(read(fd, &buf[0], sizeof(buf)))) > 0) {
content->append(buf, n); content->append(buf, n);
} }
return (n == 0) ? true : false; return (n == 0) ? true : false;
@ -206,11 +206,11 @@ bool ReadFileToString(const std::string& path, std::string* content, bool follow
return ReadFdToString(fd, content); return ReadFdToString(fd, content);
} }
bool WriteStringToFd(const std::string& content, borrowed_fd fd) { bool WriteStringToFd(const std::string& content, int fd) {
const char* p = content.data(); const char* p = content.data();
size_t left = content.size(); size_t left = content.size();
while (left > 0) { while (left > 0) {
ssize_t n = TEMP_FAILURE_RETRY(write(fd.get(), p, left)); ssize_t n = TEMP_FAILURE_RETRY(write(fd, p, left));
if (n == -1) { if (n == -1) {
return false; return false;
} }
@ -269,11 +269,11 @@ bool WriteStringToFile(const std::string& content, const std::string& path,
return WriteStringToFd(content, fd) || CleanUpAfterFailedWrite(path); return WriteStringToFd(content, fd) || CleanUpAfterFailedWrite(path);
} }
bool ReadFully(borrowed_fd fd, void* data, size_t byte_count) { bool ReadFully(int fd, void* data, size_t byte_count) {
uint8_t* p = reinterpret_cast<uint8_t*>(data); uint8_t* p = reinterpret_cast<uint8_t*>(data);
size_t remaining = byte_count; size_t remaining = byte_count;
while (remaining > 0) { while (remaining > 0) {
ssize_t n = TEMP_FAILURE_RETRY(read(fd.get(), p, remaining)); ssize_t n = TEMP_FAILURE_RETRY(read(fd, p, remaining));
if (n <= 0) return false; if (n <= 0) return false;
p += n; p += n;
remaining -= n; remaining -= n;
@ -284,14 +284,14 @@ bool ReadFully(borrowed_fd fd, void* data, size_t byte_count) {
#if defined(_WIN32) #if defined(_WIN32)
// Windows implementation of pread. Note that this DOES move the file descriptors read position, // Windows implementation of pread. Note that this DOES move the file descriptors read position,
// but it does so atomically. // but it does so atomically.
static ssize_t pread(borrowed_fd fd, void* data, size_t byte_count, off64_t offset) { static ssize_t pread(int fd, void* data, size_t byte_count, off64_t offset) {
DWORD bytes_read; DWORD bytes_read;
OVERLAPPED overlapped; OVERLAPPED overlapped;
memset(&overlapped, 0, sizeof(OVERLAPPED)); memset(&overlapped, 0, sizeof(OVERLAPPED));
overlapped.Offset = static_cast<DWORD>(offset); overlapped.Offset = static_cast<DWORD>(offset);
overlapped.OffsetHigh = static_cast<DWORD>(offset >> 32); overlapped.OffsetHigh = static_cast<DWORD>(offset >> 32);
if (!ReadFile(reinterpret_cast<HANDLE>(_get_osfhandle(fd.get())), data, if (!ReadFile(reinterpret_cast<HANDLE>(_get_osfhandle(fd)), data, static_cast<DWORD>(byte_count),
static_cast<DWORD>(byte_count), &bytes_read, &overlapped)) { &bytes_read, &overlapped)) {
// In case someone tries to read errno (since this is masquerading as a POSIX call) // In case someone tries to read errno (since this is masquerading as a POSIX call)
errno = EIO; errno = EIO;
return -1; return -1;
@ -300,10 +300,10 @@ static ssize_t pread(borrowed_fd fd, void* data, size_t byte_count, off64_t offs
} }
#endif #endif
bool ReadFullyAtOffset(borrowed_fd fd, void* data, size_t byte_count, off64_t offset) { bool ReadFullyAtOffset(int fd, void* data, size_t byte_count, off64_t offset) {
uint8_t* p = reinterpret_cast<uint8_t*>(data); uint8_t* p = reinterpret_cast<uint8_t*>(data);
while (byte_count > 0) { while (byte_count > 0) {
ssize_t n = TEMP_FAILURE_RETRY(pread(fd.get(), p, byte_count, offset)); ssize_t n = TEMP_FAILURE_RETRY(pread(fd, p, byte_count, offset));
if (n <= 0) return false; if (n <= 0) return false;
p += n; p += n;
byte_count -= n; byte_count -= n;
@ -312,11 +312,11 @@ bool ReadFullyAtOffset(borrowed_fd fd, void* data, size_t byte_count, off64_t of
return true; return true;
} }
bool WriteFully(borrowed_fd fd, const void* data, size_t byte_count) { bool WriteFully(int fd, const void* data, size_t byte_count) {
const uint8_t* p = reinterpret_cast<const uint8_t*>(data); const uint8_t* p = reinterpret_cast<const uint8_t*>(data);
size_t remaining = byte_count; size_t remaining = byte_count;
while (remaining > 0) { while (remaining > 0) {
ssize_t n = TEMP_FAILURE_RETRY(write(fd.get(), p, remaining)); ssize_t n = TEMP_FAILURE_RETRY(write(fd, p, remaining));
if (n == -1) return false; if (n == -1) return false;
p += n; p += n;
remaining -= n; remaining -= n;

View file

@ -51,20 +51,20 @@ namespace base {
// Note that the write can return short if the socket type is SOCK_STREAM. When // Note that the write can return short if the socket type is SOCK_STREAM. When
// this happens, file descriptors are still sent to the other end, but with // this happens, file descriptors are still sent to the other end, but with
// truncated data. For this reason, using SOCK_SEQPACKET or SOCK_DGRAM is recommended. // truncated data. For this reason, using SOCK_SEQPACKET or SOCK_DGRAM is recommended.
ssize_t SendFileDescriptorVector(borrowed_fd sock, const void* data, size_t len, ssize_t SendFileDescriptorVector(int sock, const void* data, size_t len,
const std::vector<int>& fds); const std::vector<int>& fds);
// Receive file descriptors from a Unix domain socket. // Receive file descriptors from a Unix domain socket.
// //
// If more FDs (or bytes, for datagram sockets) are received than expected, // If more FDs (or bytes, for datagram sockets) are received than expected,
// -1 is returned with errno set to EMSGSIZE, and all received FDs are thrown away. // -1 is returned with errno set to EMSGSIZE, and all received FDs are thrown away.
ssize_t ReceiveFileDescriptorVector(borrowed_fd sock, void* data, size_t len, size_t max_fds, ssize_t ReceiveFileDescriptorVector(int sock, void* data, size_t len, size_t max_fds,
std::vector<android::base::unique_fd>* fds); std::vector<android::base::unique_fd>* fds);
// Helper for SendFileDescriptorVector that constructs a std::vector for you, e.g.: // Helper for SendFileDescriptorVector that constructs a std::vector for you, e.g.:
// SendFileDescriptors(sock, "foo", 3, std::move(fd1), std::move(fd2)) // SendFileDescriptors(sock, "foo", 3, std::move(fd1), std::move(fd2))
template <typename... Args> template <typename... Args>
ssize_t SendFileDescriptors(borrowed_fd sock, const void* data, size_t len, Args&&... sent_fds) { ssize_t SendFileDescriptors(int sock, const void* data, size_t len, Args&&... sent_fds) {
// Do not allow implicit conversion to int: people might try to do something along the lines of: // Do not allow implicit conversion to int: people might try to do something along the lines of:
// SendFileDescriptors(..., std::move(a_unique_fd)) // SendFileDescriptors(..., std::move(a_unique_fd))
// and be surprised when the unique_fd isn't closed afterwards. // and be surprised when the unique_fd isn't closed afterwards.
@ -79,7 +79,7 @@ ssize_t SendFileDescriptors(borrowed_fd sock, const void* data, size_t len, Args
// If fewer file descriptors are received than requested, -1 is returned with errno set to ENOMSG. // If fewer file descriptors are received than requested, -1 is returned with errno set to ENOMSG.
// In both cases, all arguments are cleared and any received FDs are thrown away. // In both cases, all arguments are cleared and any received FDs are thrown away.
template <typename... Args> template <typename... Args>
ssize_t ReceiveFileDescriptors(borrowed_fd sock, void* data, size_t len, Args&&... received_fds) { ssize_t ReceiveFileDescriptors(int sock, void* data, size_t len, Args&&... received_fds) {
std::vector<unique_fd*> fds; std::vector<unique_fd*> fds;
Append(fds, std::forward<Args>(received_fds)...); Append(fds, std::forward<Args>(received_fds)...);

View file

@ -21,8 +21,7 @@
#include <string> #include <string>
#include "android-base/macros.h" #include <android-base/macros.h>
#include "android-base/unique_fd.h"
#include "android-base/off64_t.h" #include "android-base/off64_t.h"
#if !defined(_WIN32) && !defined(O_BINARY) #if !defined(_WIN32) && !defined(O_BINARY)
@ -78,13 +77,13 @@ class TemporaryDir {
namespace android { namespace android {
namespace base { namespace base {
bool ReadFdToString(borrowed_fd fd, std::string* content); bool ReadFdToString(int fd, std::string* content);
bool ReadFileToString(const std::string& path, std::string* content, bool ReadFileToString(const std::string& path, std::string* content,
bool follow_symlinks = false); bool follow_symlinks = false);
bool WriteStringToFile(const std::string& content, const std::string& path, bool WriteStringToFile(const std::string& content, const std::string& path,
bool follow_symlinks = false); bool follow_symlinks = false);
bool WriteStringToFd(const std::string& content, borrowed_fd fd); bool WriteStringToFd(const std::string& content, int fd);
#if !defined(_WIN32) #if !defined(_WIN32)
bool WriteStringToFile(const std::string& content, const std::string& path, bool WriteStringToFile(const std::string& content, const std::string& path,
@ -92,7 +91,7 @@ bool WriteStringToFile(const std::string& content, const std::string& path,
bool follow_symlinks = false); bool follow_symlinks = false);
#endif #endif
bool ReadFully(borrowed_fd fd, void* data, size_t byte_count); bool ReadFully(int fd, void* data, size_t byte_count);
// Reads `byte_count` bytes from the file descriptor at the specified offset. // Reads `byte_count` bytes from the file descriptor at the specified offset.
// Returns false if there was an IO error or EOF was reached before reading `byte_count` bytes. // Returns false if there was an IO error or EOF was reached before reading `byte_count` bytes.
@ -102,9 +101,9 @@ bool ReadFully(borrowed_fd fd, void* data, size_t byte_count);
// get modified. This means that ReadFullyAtOffset can be used concurrently with other calls to the // get modified. This means that ReadFullyAtOffset can be used concurrently with other calls to the
// same function, but concurrently seeking or reading incrementally can lead to unexpected // same function, but concurrently seeking or reading incrementally can lead to unexpected
// behavior. // behavior.
bool ReadFullyAtOffset(borrowed_fd fd, void* data, size_t byte_count, off64_t offset); bool ReadFullyAtOffset(int fd, void* data, size_t byte_count, off64_t offset);
bool WriteFully(borrowed_fd fd, const void* data, size_t byte_count); bool WriteFully(int fd, const void* data, size_t byte_count);
bool RemoveFileIfExists(const std::string& path, std::string* err = nullptr); bool RemoveFileIfExists(const std::string& path, std::string* err = nullptr);

View file

@ -16,14 +16,13 @@
#pragma once #pragma once
#include "android-base/macros.h"
#include "android-base/off64_t.h"
#include <sys/types.h> #include <sys/types.h>
#include <memory> #include <memory>
#include "android-base/macros.h"
#include "android-base/off64_t.h"
#include "android-base/unique_fd.h"
#if defined(_WIN32) #if defined(_WIN32)
#include <windows.h> #include <windows.h>
#define PROT_READ 1 #define PROT_READ 1
@ -45,8 +44,7 @@ class MappedFile {
* `offset` does not need to be page-aligned. If `PROT_WRITE` is set in `prot`, the mapping * `offset` does not need to be page-aligned. If `PROT_WRITE` is set in `prot`, the mapping
* will be writable, otherwise it will be read-only. Mappings are always `MAP_SHARED`. * will be writable, otherwise it will be read-only. Mappings are always `MAP_SHARED`.
*/ */
static std::unique_ptr<MappedFile> FromFd(borrowed_fd fd, off64_t offset, size_t length, static std::unique_ptr<MappedFile> FromFd(int fd, off64_t offset, size_t length, int prot);
int prot);
/** /**
* Removes the mapping. * Removes the mapping.

View file

@ -103,17 +103,7 @@ class unique_fd_impl final {
void reset(int new_value = -1) { reset(new_value, nullptr); } void reset(int new_value = -1) { reset(new_value, nullptr); }
int get() const { return fd_; } int get() const { return fd_; }
#if !defined(ANDROID_BASE_UNIQUE_FD_DISABLE_IMPLICIT_CONVERSION)
// unique_fd's operator int is dangerous, but we have way too much code that
// depends on it, so make this opt-in at first.
operator int() const { return get(); } // NOLINT operator int() const { return get(); } // NOLINT
#endif
bool operator>=(int rhs) const { return get() >= rhs; }
bool operator<(int rhs) const { return get() < rhs; }
bool operator==(int rhs) const { return get() == rhs; }
bool operator!=(int rhs) const { return get() != rhs; }
// Catch bogus error checks (i.e.: "!fd" instead of "fd != -1"). // Catch bogus error checks (i.e.: "!fd" instead of "fd != -1").
bool operator!() const = delete; bool operator!() const = delete;
@ -256,22 +246,6 @@ inline DIR* Fdopendir(unique_fd&& ufd) {
#endif // !defined(_WIN32) #endif // !defined(_WIN32)
// A wrapper type that can be implicitly constructed from either int or unique_fd.
struct borrowed_fd {
/* implicit */ borrowed_fd(int fd) : fd_(fd) {}
template <typename T>
/* implicit */ borrowed_fd(const unique_fd_impl<T>& ufd) : fd_(ufd.get()) {}
int get() const { return fd_; }
bool operator>=(int rhs) const { return get() >= rhs; }
bool operator<(int rhs) const { return get() < rhs; }
bool operator==(int rhs) const { return get() == rhs; }
bool operator!=(int rhs) const { return get() != rhs; }
private:
int fd_ = -1;
};
} // namespace base } // namespace base
} // namespace android } // namespace android

View file

@ -18,8 +18,6 @@
#include <errno.h> #include <errno.h>
#include "android-base/unique_fd.h"
namespace android { namespace android {
namespace base { namespace base {
@ -33,8 +31,7 @@ static off64_t InitPageSize() {
#endif #endif
} }
std::unique_ptr<MappedFile> MappedFile::FromFd(borrowed_fd fd, off64_t offset, size_t length, std::unique_ptr<MappedFile> MappedFile::FromFd(int fd, off64_t offset, size_t length, int prot) {
int prot) {
static off64_t page_size = InitPageSize(); static off64_t page_size = InitPageSize();
size_t slop = offset % page_size; size_t slop = offset % page_size;
off64_t file_offset = offset - slop; off64_t file_offset = offset - slop;
@ -42,7 +39,7 @@ std::unique_ptr<MappedFile> MappedFile::FromFd(borrowed_fd fd, off64_t offset, s
#if defined(_WIN32) #if defined(_WIN32)
HANDLE handle = HANDLE handle =
CreateFileMapping(reinterpret_cast<HANDLE>(_get_osfhandle(fd.get())), nullptr, CreateFileMapping(reinterpret_cast<HANDLE>(_get_osfhandle(fd)), nullptr,
(prot & PROT_WRITE) ? PAGE_READWRITE : PAGE_READONLY, 0, 0, nullptr); (prot & PROT_WRITE) ? PAGE_READWRITE : PAGE_READONLY, 0, 0, nullptr);
if (handle == nullptr) { if (handle == nullptr) {
// http://b/119818070 "app crashes when reading asset of zero length". // http://b/119818070 "app crashes when reading asset of zero length".
@ -61,7 +58,7 @@ std::unique_ptr<MappedFile> MappedFile::FromFd(borrowed_fd fd, off64_t offset, s
return std::unique_ptr<MappedFile>( return std::unique_ptr<MappedFile>(
new MappedFile{static_cast<char*>(base), length, slop, handle}); new MappedFile{static_cast<char*>(base), length, slop, handle});
#else #else
void* base = mmap(nullptr, file_length, prot, MAP_SHARED, fd.get(), file_offset); void* base = mmap(nullptr, file_length, prot, MAP_SHARED, fd, file_offset);
if (base == MAP_FAILED) { if (base == MAP_FAILED) {
// http://b/119818070 "app crashes when reading asset of zero length". // http://b/119818070 "app crashes when reading asset of zero length".
// mmap fails with EINVAL for a zero length region. // mmap fails with EINVAL for a zero length region.