hostage/native/pp.c

#include <errno.h>
#include <netinet/in.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <sys/uio.h>
#include <sys/un.h>
#include <unistd.h>

#define CONTROLLEN CMSG_LEN(sizeof(int))

#define UNIX_PATH "hostage"

#define TCP "TCP"
#define UDP "UDP"

int ipc_sock() {
	int fd;
	struct sockaddr_un addr;

	if ((fd = socket(AF_UNIX, SOCK_STREAM, 0)) == -1) {
		perror("Unable to create local socket");
		return -1;
	}

	memset(&addr, 0, sizeof(addr));
	addr.sun_family = AF_UNIX;
	strncpy(&addr.sun_path[1], UNIX_PATH, strlen(UNIX_PATH));

	if (connect(fd, (struct sockaddr*) &addr,
			sizeof(sa_family_t) + strlen(UNIX_PATH) + 1) == -1) {
		perror("Unable to connect local socket");
		return -1;
	}

	return fd;
}

int net_sock(int type, int port) {
	int fd;
	int reuseaddr = 1;
	struct sockaddr_in addr;

	if ((fd = socket(AF_INET, (type == 1 ? SOCK_STREAM : SOCK_DGRAM), 0)) == -1) {
		perror("Unable to create net socket");
		return -1;
	}

	memset(&addr, 0, sizeof(addr));
	addr.sin_family = AF_INET;
	addr.sin_addr.s_addr = INADDR_ANY;
	addr.sin_port = htons(port);

	if (type == 1) {
		if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &reuseaddr, sizeof(reuseaddr))
				== -1) {
			perror("Unable to set socket options");
			return -1;
		}
	}

	if (bind(fd, (struct sockaddr *) &addr, sizeof(addr)) == -1) {
		perror("Unable to bind net socket");
		return -1;
	}

	if (type == 1) {
		if (listen(fd, 5) == -1) {
			perror("Unable to listen net socket");
			return -1;
		}
	}

	return fd;
}

int send_fd(int fd, int fd_to_send) {
	struct iovec iov[1];
	struct cmsghdr *cmptr;
	struct msghdr msg;
	char buf[2] = "FD";

	iov[0].iov_base = buf;
	iov[0].iov_len = 2;

	cmptr = malloc(CONTROLLEN);
	cmptr->cmsg_level = SOL_SOCKET;
	cmptr->cmsg_type = SCM_RIGHTS;
	cmptr->cmsg_len = CONTROLLEN;

	msg.msg_iov = iov;
	msg.msg_iovlen = 1;
	msg.msg_name = NULL;
	msg.msg_namelen = 0;
	msg.msg_control = cmptr;
	msg.msg_controllen = CONTROLLEN;
	*(int *) CMSG_DATA(cmptr) = fd_to_send;

	if (sendmsg(fd, &msg, 0) == -1) {
		perror("sendmsg failed");
	}

	return 0;
}

int main(int argc, char *argv[]) {
	int type;
	int port;
	int ipc_fd, net_fd;

	if (argc < 3) {
		exit(EXIT_FAILURE);
	}

	if (strncmp(argv[1], TCP) == 0) {
		type = 1;
	} else if (strncmp(argv[1], UDP) == 0) {
		type = 0;
	} else {
		exit(EXIT_FAILURE);
	}

	if ((port = atoi(argv[2])) < 1 || (port = atoi(argv[2])) > 65535) {
		exit(EXIT_FAILURE);
	}

	if ((ipc_fd = ipc_sock()) == -1) {
		close(ipc_fd);
		exit(EXIT_FAILURE);
	}
	printf("ipc_fd: %d\n", ipc_fd);

	if ((net_fd = net_sock(type, port)) == -1) {
		close(ipc_fd);
		close(net_fd);
		exit(EXIT_FAILURE);
	}
	printf("net_fd: %d\n", net_fd);

	int status;
	status = send_fd(ipc_fd, net_fd);
	printf("send_fd: %d\n", status);

	close(ipc_fd);
	close(net_fd);

	if (status == -1) {
		return (EXIT_FAILURE);
	}

	return EXIT_SUCCESS;
}