am f03397fa: Merge changes If5359c26,I5d09be41 (56935417) · Commits · e / os / android_system_core

libnetd_client/NetdClient.cpp

+39 −0

Original line number	Diff line number	Diff line
		@@ -18,12 +18,22 @@
		#include "netd_client/FwmarkCommands.h"

		#include <sys/socket.h>
		#include <unistd.h>

		namespace {

		int closeFdAndRestoreErrno(int fd) {
		int error = errno;
		close(fd);
		errno = error;
		return -1;
		}

		typedef int (ConnectFunctionType)(int, const sockaddr, socklen_t);
		typedef int (AcceptFunctionType)(int, sockaddr, socklen_t*);

		ConnectFunctionType libcConnect = 0;
		AcceptFunctionType libcAccept = 0;

		int netdClientConnect(int sockfd, const sockaddr* addr, socklen_t addrlen) {
		if (FwmarkClient::shouldSetFwmark(sockfd, addr)) {
		@@ -35,6 +45,28 @@ int netdClientConnect(int sockfd, const sockaddr* addr, socklen_t addrlen) {
		return libcConnect(sockfd, addr, addrlen);
		}

		int netdClientAccept(int sockfd, sockaddr* addr, socklen_t* addrlen) {
		int acceptedSocket = libcAccept(sockfd, addr, addrlen);
		if (acceptedSocket == -1) {
		return -1;
		}
		sockaddr socketAddress;
		if (!addr) {
		socklen_t socketAddressLen = sizeof(socketAddress);
		if (getsockname(acceptedSocket, &socketAddress, &socketAddressLen) == -1) {
		return closeFdAndRestoreErrno(acceptedSocket);
		}
		addr = &socketAddress;
		}
		if (FwmarkClient::shouldSetFwmark(acceptedSocket, addr)) {
		char data[] = {FWMARK_COMMAND_ON_ACCEPT};
		if (!FwmarkClient().send(data, sizeof(data), acceptedSocket)) {
		return closeFdAndRestoreErrno(acceptedSocket);
		}
		}
		return acceptedSocket;
		}

		} // namespace

		extern "C" void netdClientInitConnect(ConnectFunctionType* function) {
		@@ -43,3 +75,10 @@ extern "C" void netdClientInitConnect(ConnectFunctionType* function) {
		*function = netdClientConnect;
		}
		}

		extern "C" void netdClientInitAccept(AcceptFunctionType* function) {
		if (function && *function) {
		libcAccept = *function;
		*function = netdClientAccept;
		}
		}