TinyFrame/demo/demo.c

//
// Created by MightyPork on 2017/10/15.
//

#include "demo.h"

// those magic defines are needed so we can use clone()
#define _GNU_SOURCE
#define __USE_GNU
#include <sched.h>

#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <string.h>
#include <arpa/inet.h>
#include <signal.h>

volatile int sockfd = -1;
volatile bool conn_disband = false;

/**
 * Close socket
 */
void demo_disconn(void)
{
	conn_disband = true;
	if (sockfd >= 0) close(sockfd);
}

/**
 * Demo WriteImpl - send stuff to our peer
 *
 * @param buff
 * @param len
 */
void TF_WriteImpl(const uint8_t *buff, size_t len)
{
	printf("\033[32mTF_WriteImpl - sending frame:\033[0m\n");
	dumpFrame(buff, len);
	usleep(1000);

	if (sockfd != -1) {
		write(sockfd, buff, len);
	} else {
		printf("\nNo peer!\n");
	}
}


/**
 * Client bg thread
 *
 * @param unused
 * @return unused
 */
static int demo_client(void* unused)
{
	(void)unused;

	ssize_t n = 0;
	uint8_t recvBuff[1024];
	struct sockaddr_in serv_addr;

	printf("\n--- STARTING CLIENT! ---\n");

	memset(recvBuff, '0', sizeof(recvBuff));
	if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
		printf("\n Error : Could not create socket \n");
		return false;
	}

	memset(&serv_addr, '0', sizeof(serv_addr));

	serv_addr.sin_family = AF_INET;
	serv_addr.sin_port = htons(PORT);

	if (inet_pton(AF_INET, "127.0.0.1", &serv_addr.sin_addr) <= 0) {
		printf("\n inet_pton error occured\n");
		return false;
	}

	if (connect(sockfd, (struct sockaddr *) &serv_addr, sizeof(serv_addr)) < 0) {
		printf("\n Error : Connect Failed \n");
		perror("PERROR ");
		return false;
	}

	printf("\n Child Process \n");

	while ((n = read(sockfd, recvBuff, sizeof(recvBuff) - 1)) > 0) {
		printf("\033[36m--- RX %ld bytes ---\033[0m\n", n);
		dumpFrame(recvBuff, (size_t) n);
		TF_Accept(recvBuff, (size_t) n);
	}
	return 0;
}

/**
 * Server bg thread
 *
 * @param unused
 * @return unused
 */
static int demo_server(void* unused)
{
	(void)unused;
	ssize_t n;
	int listenfd = 0;
	uint8_t recvBuff[1024];
	struct sockaddr_in serv_addr;
	int option;

	printf("\n--- STARTING SERVER! ---\n");

	listenfd = socket(AF_INET, SOCK_STREAM, 0);
	memset(&serv_addr, '0', sizeof(serv_addr));

	option = 1;
	setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, (char*)&option, sizeof(option));

	serv_addr.sin_family = AF_INET;
	serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);
	serv_addr.sin_port = htons(PORT);

	if (bind(listenfd, (struct sockaddr *) &serv_addr, sizeof(serv_addr)) < 0) {
		perror("Failed to bind");
		return 1;
	}

	if (listen(listenfd, 10) < 0) {
		perror("Failed to listen");
		return 1;
	}

	while (1) {
		printf("\nWaiting for client...\n");
		sockfd = accept(listenfd, (struct sockaddr *) NULL, NULL);
		setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, (char*)&option, sizeof(option));
		printf("\nClient connected\n");
		conn_disband = false;

		while ((n = read(sockfd, recvBuff, sizeof(recvBuff) - 1)) > 0 && !conn_disband) {
			printf("\033[36m--- RX %ld bytes ---\033[0m\n", n);
			dumpFrame(recvBuff, n);
			TF_Accept(recvBuff, (size_t) n);
		}

		if (n < 0) {
			printf("\n Read error \n");
		}

		printf("Closing socket\n");
		close(sockfd);
		sockfd = -1;
	}
	return 0;
}

/**
 * Trap - clean up
 *
 * @param sig - signal that caused this
 */
static void signal_handler(int sig)
{
	(void)sig;
	printf("Shutting down...");
	demo_disconn();

	exit(sig); // pass the signal through - this is nonstandard behavior but useful for debugging
}

/**
 * Sleaping Beauty's fave function
 */
void demo_sleep(void)
{
	while(1) usleep(10);
}

/**
 * Start the background thread
 *
 * Slave is started first and doesn't normally init transactions - but it could
 *
 * @param peer what peer we are
 */
void demo_init(TF_Peer peer)
{
	signal(SIGTERM, signal_handler);
	signal(SIGINT, signal_handler);

	int retc;
	void *stack = malloc(8192);
	if (stack == NULL) {
		perror("Oh fuck");
		signal_handler(9);
		return;
	}

	printf("Starting %s...\n", peer == TF_MASTER ? "MASTER" : "SLAVE");

	// CLONE_VM    --- share heap
	// CLONE_FILES --- share stdout and stderr
	if (peer == TF_MASTER) {
		retc = clone(&demo_client, (char *)stack+8192, CLONE_VM|CLONE_FILES, 0);
	} else {
		retc = clone(&demo_server, (char *)stack+8192, CLONE_VM|CLONE_FILES, 0);
	}

	if (retc == 0) {
		perror("Clone fail");
		signal_handler(9);
		return;
	}

	printf("Thread started\n");
}