// // 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 #include #include #include #include #include #include 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"); }