gateway.c

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<sys/socket.h>
#include<arpa/inet.h> //inet_addr
#include<unistd.h>
#include<stdint.h>
#include<pthread.h>
#include<gateway_protocol.h>
#include<sys/time.h>
#include<libpq-fe.h>
#include"base64.h"
#include<math.h>
#include<signal.h>

#define TIMEDATE_LENGTH		32

typedef struct {
	uint32_t utc;
	char timedate[TIMEDATE_LENGTH];

	float dht22_t_esp;
	float dht22_h_esp;
	float sht85_t_esp;
	float sht85_h_esp;
	float hih8121_t_esp;
	float hih8121_h_esp;
	float tmp36_0_esp;
	float tmp36_1_esp;
	float tmp36_2_esp;
	float hih4030_esp;
	float hh10d_esp;

	float dht22_t_mkr;
	float dht22_h_mkr;
	float sht85_t_mkr;
	float sht85_h_mkr;
	float hih8121_t_mkr;
	float hih8121_h_mkr;
	float hh10d_mkr;

	float dht22_t_wis;
	float dht22_h_wis;
	float sht85_t_wis;
	float sht85_h_wis;
	float hih8121_t_wis;
	float hih8121_h_wis;
	float tmp102_wis;
	float hh10d_wis;
} sensor_data_t;

void * connection_handler (void *args);
uint8_t gateway_protocol_data_send_payload_decode(sensor_data_t *sensor_data, const uint8_t *payload, const uint8_t payload_length);
void filter_nans(char *com);
void packet_encode(
	uint8_t dev_id, 
	gateway_protocol_packet_type_t p_type, 
	uint8_t payload_length,
	uint8_t *payload,
	uint8_t *packet_length,
	uint8_t *packet);
uint8_t packet_decode(
	uint8_t *dev_id,
	gateway_protocol_packet_type_t *ptype,
	uint8_t *payload_length,
	uint8_t *payload,
	uint8_t packet_length,
	uint8_t *packet);

void ctrc_handler (int sig);
static volatile uint8_t working = 1;

int main (int argc, char **argv) {
	int server_desc, client_desc;
       	struct sockaddr_in server, client;
	socklen_t client_socklen;
	uint8_t buf[1024];
	uint8_t buf_len = 0;
	uint8_t payload[256];
	uint8_t payload_length = 0;
	int sock_len;
	PGresult *res;
	
	signal(SIGINT, ctrc_handler);

	PGconn *conn = PQconnectdb("user=root dbname=gateway");
	if (PQstatus(conn) == CONNECTION_BAD) {
		printf("connection to db error: %s\n", PQerrorMessage(conn));
		return EXIT_FAILURE;
	}

	if ((server_desc = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)) < 0) {
		perror("socket creation error");
		return EXIT_FAILURE;
	}

	server.sin_family 	= AF_INET;
	server.sin_port		= htons(9043);
	server.sin_addr.s_addr 	= INADDR_ANY;

	if (bind(server_desc, (struct sockaddr *) &server, sizeof(server)) < 0) {
		perror("binding error");
		return EXIT_FAILURE;
	}

	uint8_t dev_id = 0xFF;
	gateway_protocol_packet_type_t packet_type;
	
	while (working) {
		buf_len = 0;
		printf("listenninig...\n");
		if ((buf_len = recvfrom(server_desc, (char *)buf, 1024, MSG_WAITALL, (struct sockaddr *)&client, &sock_len)) < 0) {
			perror("socket receive error");
		}

		printf("packet received!\n");

		for(uint8_t i = 0; i < buf_len; i++) {
			printf("%02X : ", buf[i]);
		}
		printf("\n");
	
		if (gateway_protocol_packet_decode(
					&dev_id,
					&packet_type,
					&payload_length, payload,
					buf_len, buf))
		{
			if (packet_type == GATEWAY_PROTOCOL_PACKET_TYPE_TIME_REQ) {
				printf("TIME REQ received\n");
				struct timeval tv;
				
				gettimeofday(&tv, NULL);
				
				packet_encode (
					dev_id,
					GATEWAY_PROTOCOL_PACKET_TYPE_TIME_SEND,
					sizeof(tv.tv_sec), (uint8_t *)&tv.tv_sec,
					&buf_len, buf
				);
					
				for(uint8_t i = 0; i < buf_len; i++) {
					printf("%02X : ", buf[i]);
				}
				printf("\n");
			} else if (packet_type == GATEWAY_PROTOCOL_PACKET_TYPE_DATA_SEND) {
				sensor_data_t sensor_data;
				printf("DATA SEND received\n");
		        	if (gateway_protocol_data_send_payload_decode(&sensor_data, payload, payload_length)) {
					snprintf(buf, sizeof(buf), "INSERT INTO esp32 ("
						"utc, timedate, "
						"dht22_t_esp, dht22_h_esp, "
						"sht85_t_esp, sht85_h_esp, "
						"hih8121_t_esp, hih8121_h_esp, "
						"tmp36_0_esp, tmp36_1_esp, tmp36_2_esp, "
						"hih4030_esp, "
						"hh10d_esp, "
						"dht22_t_mkr, dht22_h_mkr, "
						"sht85_t_mkr, sht85_h_mkr, "
						"hih8121_t_mkr, hih8121_h_mkr, "
						"hh10d_mkr, "
						"dht22_t_wis, dht22_h_wis, "
						"sht85_t_wis, sht85_h_wis, "
						"hih8121_t_wis, hih8121_h_wis, "
						"tmp102_wis, "
						"hh10d_wis) "
						"VALUES("
						"%lu, '%s', "
						"%.2f, %.2f, "  // esp
						"%.2f, %.2f, "
						"%.2f, %.2f, "
						"%.2f, %.2f, %.2f, "
						"%.2f, "
						"%.2f, "
						"%.2f, %.2f, "  // mkr
						"%.2f, %.2f, "
						"%.2f, %.2f, "
						"%.2f, "
						"%.2f, %.2f, "  // wis
						"%.2f, %.2f, "
						"%.2f, %.2f, "
						"%.2f, "
						"%.2f)",
						sensor_data.utc, sensor_data.timedate,
						sensor_data.dht22_t_esp, sensor_data.dht22_h_esp,
						sensor_data.sht85_t_esp, sensor_data.sht85_h_esp,
						sensor_data.hih8121_t_esp, sensor_data.hih8121_h_esp,
						sensor_data.tmp36_0_esp, sensor_data.tmp36_1_esp, sensor_data.tmp36_2_esp,
						sensor_data.hih4030_esp,
						sensor_data.hh10d_esp,
						sensor_data.dht22_t_mkr, sensor_data.dht22_h_mkr,
						sensor_data.sht85_t_mkr, sensor_data.sht85_h_mkr,
						sensor_data.hih8121_t_mkr, sensor_data.hih8121_h_mkr,
						sensor_data.hh10d_mkr,
						sensor_data.dht22_t_wis, sensor_data.dht22_h_wis,
						sensor_data.sht85_t_wis, sensor_data.sht85_h_wis,
						sensor_data.hih8121_t_wis, sensor_data.hih8121_h_wis,
						sensor_data.tmp102_wis,
						sensor_data.hh10d_wis
					);
					filter_nans(buf);
					printf("%s\n", buf);
					res = PQexec(conn, buf);
					if (PQresultStatus(res) == PGRES_COMMAND_OK) {
						PQclear(res);
						//fprintf(stderr, "%s\n", PQerrorMessage(conn));

						sprintf(buf, "SELECT * FROM pend_msgs WHERE dev_id = %d", dev_id);
						res = PQexec(conn, buf);
						if (PQresultStatus(res) == PGRES_TUPLES_OK && PQntuples(res)) {
							buf[3] = GATEWAY_PROTOCOL_STAT_ACK_PEND;
							printf("ACK_PEND prepared\n");
						} else {
							buf[3] = GATEWAY_PROTOCOL_STAT_ACK;
							printf("ACK prepared\n");
						}

						buf[0] = dev_id;				
						buf[1] = GATEWAY_PROTOCOL_PACKET_TYPE_STAT;
						buf[2] = 1;
						buf_len = 4;
					} else {
						fprintf(stderr, "database error : %s\n", PQerrorMessage(conn));
					}
					PQclear(res);
				} else {
					gateway_protocol_stat_t s_type = GATEWAY_PROTOCOL_STAT_NACK;
					packet_encode (
						dev_id,
						GATEWAY_PROTOCOL_PACKET_TYPE_STAT,
						1, (uint8_t *)&s_type,
						&buf_len, buf);
					/*
					buf[0] = dev_id;				
					buf[1] = GATEWAY_PROTOCOL_PACKET_TYPE_STAT;
					buf[2] = 1;
					buf[3] = GATEWAY_PROTOCOL_STAT_NACK;
					buf_len = 4;
					*/
					fprintf(stderr, "payload decode error\n");
				}
			} else if (packet_type == GATEWAY_PROTOCOL_PACKET_TYPE_PEND_REQ) {
				sprintf(buf, "SELECT * FROM pend_msgs WHERE dev_id = %d", dev_id);
				res = PQexec(conn, buf);
				if (PQresultStatus(res) == PGRES_TUPLES_OK && PQntuples(res)) {
					char msg_cont[150];
					strncpy(msg_cont, PQgetvalue(res, 0, 1), sizeof(msg_cont));
					printf("PEND_SEND prepared : %s\n", PQgetvalue(res, 0, 1));

					base64_decode(PQgetvalue(res, 0, 1), strlen(PQgetvalue(res, 0, 1)), payload);
					payload_length = BASE64_DECODE_OUT_SIZE(strlen(PQgetvalue(res, 0, 1)));
					PQclear(res);
					printf("prepared to send %d bytes\n", payload_length);
					
					packet_encode(
						dev_id, 
						GATEWAY_PROTOCOL_PACKET_TYPE_PEND_SEND,
						payload_length, payload,
						&buf_len, buf);

					printf("packet %d bytes\n", buf_len);
					for(uint8_t i = 0; i < buf_len; i++) {
						printf("%02X : ", buf[i]);
					}
					printf("\n");
				
					if (sendto(server_desc, (char *) buf, buf_len, 0, (struct sockaddr *)&client, sock_len) < 0) {
						perror("sendto error");
					}

					if ((buf_len = recvfrom(server_desc, (char *)buf, 1024, MSG_WAITALL, (struct sockaddr *)&client, &sock_len)) < 0) {
						perror("socket receive error");
					}

					if (packet_decode(
						&dev_id, 
						&packet_type,
						&payload_length, payload,
						buf_len, buf)) 
					{
						if (packet_type == GATEWAY_PROTOCOL_PACKET_TYPE_STAT &&
							payload_length == 1 &&
							payload[0] == GATEWAY_PROTOCOL_STAT_ACK)
						{
							sprintf(buf, "DELETE FROM pend_msgs WHERE dev_id = %d AND msg = '%s'", dev_id, msg_cont);
							printf("%s", buf);
							res = PQexec(conn, buf);

							if (PQresultStatus(res) != PGRES_COMMAND_OK) {
								fprintf(stderr, "error db deleting : %s", PQerrorMessage(conn));
							}
							PQclear(res);
							printf("ACK received\n");
						}
					}
					continue;
				}
			} else {
				fprintf(stderr, "packet type error : %02X\n", packet_type);
			}

			if (sendto(server_desc, (char *) buf, buf_len, 0, (struct sockaddr *)&client, sock_len) < 0) {
				perror("sendto error");
			}
		} else {
			fprintf(stderr, "packet decode error (type : %02X, pck_len = %d, pay_len = %d\n", 
					packet_type, buf_len, payload_length);
		}
	}

	close(server_desc);
	PQfinish(conn);

	return EXIT_SUCCESS;
}

void ctrc_handler (int sig) {
	working = 0;
}

uint8_t gateway_protocol_data_send_payload_decode(sensor_data_t *sensor_data, const uint8_t *payload, const uint8_t payload_length) {
	uint8_t p_len = 0;

	memcpy(&sensor_data->utc, &payload[p_len], sizeof(sensor_data->utc));
	p_len += sizeof(sensor_data->utc);

	memcpy(&sensor_data->timedate, &payload[p_len], sizeof(sensor_data->timedate));
	//p_len += sizeof(sensor_data->timedate);
	p_len += TIMEDATE_LENGTH;


	memcpy(&sensor_data->dht22_t_esp, &payload[p_len], sizeof(sensor_data->dht22_t_esp));
	p_len += sizeof(sensor_data->dht22_t_esp);

	memcpy(&sensor_data->dht22_h_esp, &payload[p_len], sizeof(sensor_data->dht22_h_esp));
	p_len += sizeof(sensor_data->dht22_h_esp);

	memcpy(&sensor_data->sht85_t_esp, &payload[p_len], sizeof(sensor_data->sht85_t_esp));
	p_len += sizeof(sensor_data->sht85_t_esp);

	memcpy(&sensor_data->sht85_h_esp, &payload[p_len], sizeof(sensor_data->sht85_h_esp));
	p_len += sizeof(sensor_data->sht85_h_esp);

	memcpy(&sensor_data->hih8121_t_esp, &payload[p_len], sizeof(sensor_data->hih8121_t_esp));
	p_len += sizeof(sensor_data->hih8121_t_esp);

	memcpy(&sensor_data->hih8121_h_esp, &payload[p_len], sizeof(sensor_data->hih8121_h_esp));
	p_len += sizeof(sensor_data->hih8121_h_esp);

	memcpy(&sensor_data->tmp36_0_esp, &payload[p_len], sizeof(sensor_data->tmp36_0_esp));
	p_len += sizeof(sensor_data->tmp36_0_esp);

	memcpy(&sensor_data->tmp36_1_esp, &payload[p_len], sizeof(sensor_data->tmp36_1_esp));
	p_len += sizeof(sensor_data->tmp36_1_esp);

	memcpy(&sensor_data->tmp36_2_esp, &payload[p_len], sizeof(sensor_data->tmp36_2_esp));
	p_len += sizeof(sensor_data->tmp36_2_esp);

	memcpy(&sensor_data->hih4030_esp, &payload[p_len], sizeof(sensor_data->hih4030_esp));
	p_len += sizeof(sensor_data->hih4030_esp);

	memcpy(&sensor_data->hh10d_esp, &payload[p_len], sizeof(sensor_data->hh10d_esp));
	p_len += sizeof(sensor_data->hh10d_esp);


	memcpy(&sensor_data->dht22_t_mkr, &payload[p_len], sizeof(sensor_data->dht22_t_mkr));
	p_len += sizeof(sensor_data->dht22_t_mkr);

	memcpy(&sensor_data->dht22_h_mkr, &payload[p_len], sizeof(sensor_data->dht22_h_mkr));
	p_len += sizeof(sensor_data->dht22_h_mkr);

	memcpy(&sensor_data->sht85_t_mkr, &payload[p_len], sizeof(sensor_data->sht85_t_mkr));
	p_len += sizeof(sensor_data->sht85_t_mkr);

	memcpy(&sensor_data->sht85_h_mkr, &payload[p_len], sizeof(sensor_data->sht85_h_mkr));
	p_len += sizeof(sensor_data->sht85_h_esp);

	memcpy(&sensor_data->hih8121_t_mkr, &payload[p_len], sizeof(sensor_data->hih8121_t_mkr));
	p_len += sizeof(sensor_data->hih8121_t_mkr);

	memcpy(&sensor_data->hih8121_h_mkr, &payload[p_len], sizeof(sensor_data->hih8121_h_mkr));
	p_len += sizeof(sensor_data->hih8121_h_mkr);

	memcpy(&sensor_data->hh10d_mkr, &payload[p_len], sizeof(sensor_data->hh10d_mkr));
	p_len += sizeof(sensor_data->hh10d_mkr);

	
	memcpy(&sensor_data->dht22_t_wis, &payload[p_len], sizeof(sensor_data->dht22_t_wis));
	p_len += sizeof(sensor_data->dht22_t_wis);

	memcpy(&sensor_data->dht22_h_wis, &payload[p_len], sizeof(sensor_data->dht22_h_wis));
	p_len += sizeof(sensor_data->dht22_h_wis);

	memcpy(&sensor_data->sht85_t_wis, &payload[p_len], sizeof(sensor_data->sht85_t_wis));
	p_len += sizeof(sensor_data->sht85_t_wis);

	memcpy(&sensor_data->sht85_h_wis, &payload[p_len], sizeof(sensor_data->sht85_h_wis));
	p_len += sizeof(sensor_data->sht85_h_wis);

	memcpy(&sensor_data->hih8121_t_wis, &payload[p_len], sizeof(sensor_data->hih8121_t_wis));
	p_len += sizeof(sensor_data->hih8121_t_wis);

	memcpy(&sensor_data->hih8121_h_wis, &payload[p_len], sizeof(sensor_data->hih8121_h_wis));
	p_len += sizeof(sensor_data->hih8121_h_wis);

	memcpy(&sensor_data->tmp102_wis, &payload[p_len], sizeof(sensor_data->tmp102_wis));
	p_len += sizeof(sensor_data->tmp102_wis);

	memcpy(&sensor_data->hh10d_wis, &payload[p_len], sizeof(sensor_data->hh10d_wis));
	p_len += sizeof(sensor_data->hh10d_wis);

	//printf("p_len = %d, payload_length = %d\n", p_len, payload_length);
				

	return (p_len == payload_length);
}


void filter_nans(char *com) {
	char *pchr;
	const char nanstr[] = "'NaN'";

	while((pchr = strstr(com, "nan"))) {
		memmove(&pchr[5], &pchr[3], strlen(pchr)+1);
		memcpy(pchr, nanstr, sizeof(nanstr)-1);
	}
	while(pchr = strchr(com, '-')) {
		memmove(pchr, &pchr[1], strlen(pchr));
	}
}


void packet_encode(
	uint8_t dev_id, 
	gateway_protocol_packet_type_t p_type, 
	uint8_t payload_length,
	uint8_t *payload,
	uint8_t *packet_length,
	uint8_t *packet) 
{
	*packet_length = 0;

	packet[*packet_length] = dev_id;
	(*packet_length)++;

	packet[*packet_length] = p_type;
	(*packet_length)++;

	packet[*packet_length] = payload_length;
	(*packet_length)++;

	memcpy(&packet[*packet_length], payload, payload_length);
	*packet_length += payload_length;
}

uint8_t packet_decode(
	uint8_t *dev_id,
	gateway_protocol_packet_type_t *ptype,
	uint8_t *payload_length,
	uint8_t *payload,
	uint8_t packet_length,
	uint8_t *packet)
{
	uint8_t p_len = 0;

	*dev_id = packet[p_len];
	p_len++;

	*ptype = (gateway_protocol_packet_type_t) packet[p_len];
	p_len++;

	*payload_length = packet[p_len];
	p_len++;

	memcpy(payload, &packet[p_len], *payload_length);
	p_len += *payload_length;

	return p_len == packet_length;
}



/* connection handler for multithreading version */
#ifdef MULTITHREADING_VER
void *connection_handler(void *args) {
	int client_desc = *(int *)args;
	
	uint8_t buf[128] = "";
	uint8_t buf_len = 0;
	uint8_t payload[128];
	uint8_t payload_length = 0;

	if ((buf_len = recv(client_desc, buf, sizeof(buf), 0)) > 0) {
		uint8_t dev_id = 0xFF;
		gateway_protocol_packet_type_t packet_type;
		
		if (gateway_protocol_packet_decode(
					&dev_id,
					&packet_type,
					&payload_length, payload,
					buf_len, buf))
		{
			if (packet_type == GATEWAY_PROTOCOL_PACKET_TYPE_TIME_REQ) {
				printf("TIME REQ received\n");
				struct timeval tv;
				buf_len = 0;
				
				buf[0] = dev_id;
				buf_len++;

				buf[1] = GATEWAY_PROTOCOL_PACKET_TYPE_TIME_SEND;
				buf_len++;

				gettimeofday(&tv, NULL);
				memcpy(&buf[buf_len], &tv.tv_sec, sizeof(uint32_t));
				buf_len += sizeof(uint32_t);

				write(client_desc, buf, buf_len);
			} else {
				perror("packet type error");
			}
		} else {
			perror("packet decode error");
		}
	}

}
#endif