rendered paste body#include <sys/socket.h> /* socket definitions */
#include <netinet/in.h>
#include <netinet/tcp.h>
#include <sys/types.h> /* socket types */
#include <arpa/inet.h> /* inet (3) funtions */
#include <unistd.h> /* misc. UNIX functions */
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <time.h>
#include <ctype.h>
#include <limits.h>
#include <stdarg.h>
#include <signal.h>
#include <pthread.h>
#include <mysql/mysql.h>
/* Helper Functions */
#include "helper.h"
/* Configuration */
#include "../c.config/database.h"
#include "../c.config/errormsg.h"
/* Standard Configuration */
#define BLISTENER_VERSION "1.1.2"
#define BLISTENER_PORT (10100)
#define MAX_LINE (512)
void * tcp_service(void *arg)
{
MYSQL mysql;
MYSQL_RES *res;
MYSQL_ROW row;
pthread_t tid;
int conn_s = (int)arg;
char buffer[MAX_LINE], sqlquery[MAX_LINE];
pthread_detach(tid);
/* Initialize MySQL code */
if( !mysql_init(&mysql) ) {
fprintf(stderr, "bListener.(Thread).(MySQL): %s\n", ERR_MYSQL_NOINIT);
exit(EXIT_FAILURE);
} else {
if( !mysql_real_connect(&mysql,CONF_MYSQL_HOSTNAME,CONF_MYSQL_USERNAME,CONF_MYSQL_PASSWORD,CONF_MYSQL_DATABASE,0,NULL,0) ){
fprintf(stderr, "bListener.(Thread).(MySQL): %s\n", ERR_MYSQL_NOCONNECT);
exit(EXIT_FAILURE);
} else {
fprintf(stdout,"bListener.(Thread).(MySQL): Connected to %s, Database: %s\n",CONF_MYSQL_HOSTNAME,CONF_MYSQL_DATABASE);
}
}
fprintf(stdout, "bListener.(Thread).(ClientConn): Connection Opened\n");
while(Readline(conn_s, buffer, MAX_LINE-1)) {
sprintf(sqlquery, "INSERT INTO `%s`.`datalog` VALUES (%s,NULL);", CONF_MYSQL_DATABASE, buffer);
if (mysql_query(&mysql, sqlquery)) {
fprintf(stderr, "bListener.(Thread).(MySQL): %s\n -- %s\n", mysql_error(&mysql), sqlquery);
} else {
fprintf(stdout, "bListener.(Thread).(MySQL): Data successfully entered into realtime database\n");
}
Writeline(conn_s, "220- RECV\n", strlen("220- RECV\n"));
}
fprintf(stdout, "bListener.(Thread).(ClientConn): Connection Closed\n");
mysql_close(&mysql);
fprintf(stderr, "bListener.(Thread).(MySQL): Disconnected from %s, Database: %s\n", CONF_MYSQL_HOSTNAME,CONF_MYSQL_DATABASE);
if ( close(conn_s) < 0 ) {
fprintf(stderr, "bListener.(Thread).(ClientConn): %s\n",ERR_UNABLE_TO_CLOSE);
exit(EXIT_FAILURE);
}
fprintf(stdout, "---------------------------------------------------------------------------\n");
pthread_exit(NULL);
}
void sighandle(int signo) {
if ((signo==SIGTERM) || (signo==SIGINT)) {
exit(0);
}
if (signo == SIGHUP) {
}
if (signo == SIGALRM) {
}
}
int main(int argc, char **argv) {
int list_s; /* listening socket */
int conn_s; /* connection socket */
struct sockaddr_in servaddr; /* socket address structure */
pthread_t tid;
int yes = 1, err;
fprintf(stdout, "bListener.(System): Initialising...\n");
if (signal(SIGPIPE, sighandle) == SIG_ERR)
fprintf(stderr, "bListener.(System): Can't catch SIGPIPE signals\n");
if (signal(SIGTERM, sighandle) == SIG_ERR)
fprintf(stderr, "bListener.(System): Can't catch SIGTERM signals\n");
if (signal(SIGINT, sighandle) == SIG_ERR)
fprintf(stderr, "bListener.(System): Can't catch SIGINT signals\n");
if (signal(SIGALRM, sighandle) == SIG_ERR)
fprintf(stderr, "bListener.(System): Can't catch SIGALRM signals\n");
if (signal(SIGHUP, sighandle) == SIG_ERR)
fprintf(stderr, "bListener.(System): Can't catch SIGHUP signals\n");
/* Create the listening socket */
if ( (list_s = socket(AF_INET, SOCK_STREAM, 0)) < 0 ) {
fprintf(stderr, "bListener.(TCP): %s",ERR_UNABLE_TO_MKSOCK);
exit(EXIT_FAILURE);
}
/* Set all bytes in socket address structure to zero, and fill in the relevant data members */
memset(&servaddr, 0, sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
servaddr.sin_port = htons(BLISTENER_PORT);
/* Bind our socket addresss to the listening socket, and call listen() */
if ( bind(list_s, (struct sockaddr *) &servaddr, sizeof(servaddr)) < 0 ) {
fprintf(stderr, "bListener.(TCP): %s\n", ERR_UNABLE_TO_BIND);
exit(EXIT_FAILURE);
}
if ( listen(list_s, LISTENQ) < 0 ) {
fprintf(stderr, "bListener.(TCP): %s\n", ERR_UNABLE_TO_LISTEN);
exit(EXIT_FAILURE);
}
/* If we have gotten to this point, I think we have initialised everything! */
fprintf(stdout, "bListener.(System): Initialised...\n");
/* Enter an infinite loop to respond to client requests */
while ( 1 ) {
if ( (conn_s = accept(list_s, NULL, NULL) ) < 0 ) {
fprintf(stderr, "bListener.(TCP): %s\n", ERR_UNABLE_TO_ACCEPT);
exit(EXIT_FAILURE);
}
err = pthread_create(&tid, NULL, tcp_service, (void *)conn_s);
if (err != 0)
{
fprintf(stderr, "bListener.(Thread): Error Starting Thread, Message: %d\n", err);
}
}
exit(EXIT_SUCCESS);
}