Habitat-lora-gateway/server.c

#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <ctype.h>
#include <stdio.h>              // Standard input/output definitions
#include <string.h>             // String function definitions
#include <fcntl.h>              // File control definitions
#include <unistd.h>             // UNIX standard function definitions
#include <errno.h>              // Error number definitions
#include <termios.h>            // POSIX terminal control definitions
#include <stdint.h>
#include <stdlib.h>
#include <dirent.h>
#include <math.h>
#include <pthread.h>
#include <wiringPi.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#include "server.h"
#include "config.h"
#include "global.h"

extern bool run;

void ProcessJSONClientLine(int connfd, char *line)
{	
	line[strcspn(line, "\r\n")] = '\0';		// Get rid of CR LF
	
	LogMessage("Received %s from JSON client\n", line);
	
	if (strchr(line, '=') == NULL)
	{
		// Request or command
		
		if (strcasecmp(line, "settings") == 0)
		{
			int Index;
			char SettingName[64], SettingValue[256], packet[4096];
			
			LogMessage("Responding to settings request\n");
			
			Index = 0;
			packet[0] = '\0';
			
			while (SettingAsString(Index, SettingName, sizeof(SettingName), SettingValue, sizeof(SettingValue)))
			{
				char temp[300];
				
				sprintf(temp, "{\"class\":\"SET\",\"set\":\"%s\",\"val\":%s}\r\n", SettingName, SettingValue);
				
				if ((strlen(temp) + strlen(packet)) < sizeof(packet))
				{
					strcat(packet, temp);
				}
				
				Index++;
			}

			send(connfd, packet, strlen(packet), MSG_NOSIGNAL);
		}
		else if (strcasecmp(line, "save") == 0)
		{
			LogMessage("Saving Settings\n");
			SaveConfigFile();
		}
	}
	else
	{
		// Setting
		char *setting, *value, *saveptr;
	
		setting = strtok_r(line, "=", &saveptr);
		value = strtok_r( NULL, "\n", &saveptr);
		
		SetConfigValue(setting, value);
	}
}

int SendJSON(int connfd)
{
	int Channel, port_closed;
    char sendBuff[1025];
	
	port_closed = 0;
    memset( sendBuff, '0', sizeof( sendBuff ) );
	
	for (Channel=0; Channel<=1; Channel++)
	{
		if ( Config.EnableDev )
		{
			sprintf(sendBuff, "{\"class\":\"POSN\",\"index\":%d,\"payload\":\"%s\",\"time\":\"%s\",\"lat\":%.5lf,\"lon\":%.5lf,\"alt\":%d,\"rate\":%.1lf,\"predlat\":%.5lf,\"predlon\":%.5lf,\"speed\":%d,"
							  "\"head\":%d,\"cda\":%.2lf,\"pls\":%.1lf,\"pt\":%d,\"ca\":%d,\"ct\":%d,\"as\":%.1lf,\"ad\":%d,\"sl\":%d,\"sr\":%d,\"st\":%d,\"gr\":%.2lf,\"fm\":%d}\r\n",
						Channel,
						Config.LoRaDevices[Channel].Payload,
						Config.LoRaDevices[Channel].Time,
						Config.LoRaDevices[Channel].Latitude,
						Config.LoRaDevices[Channel].Longitude,
						Config.LoRaDevices[Channel].Altitude,
						Config.LoRaDevices[Channel].AscentRate,
						Config.LoRaDevices[Channel].PredictedLatitude,
						Config.LoRaDevices[Channel].PredictedLongitude,
						Config.LoRaDevices[Channel].Speed,
						
						Config.LoRaDevices[Channel].Heading,
						Config.LoRaDevices[Channel].cda,
						Config.LoRaDevices[Channel].PredictedLandingSpeed,
						Config.LoRaDevices[Channel].PredictedTime,
						Config.LoRaDevices[Channel].CompassActual,
						Config.LoRaDevices[Channel].CompassTarget,
						Config.LoRaDevices[Channel].AirSpeed,
						Config.LoRaDevices[Channel].AirDirection,
						Config.LoRaDevices[Channel].ServoLeft,
						Config.LoRaDevices[Channel].ServoRight,
						Config.LoRaDevices[Channel].ServoTime,
						Config.LoRaDevices[Channel].GlideRatio,
						Config.LoRaDevices[Channel].FlightMode);
		}
		else
		{
			sprintf(sendBuff, "{\"class\":\"POSN\",\"index\":%d,\"payload\":\"%s\",\"time\":\"%s\",\"lat\":%.5lf,\"lon\":%.5lf,\"alt\":%d,\"rate\":%.1lf}\r\n",
						Channel,
						Config.LoRaDevices[Channel].Payload,
						Config.LoRaDevices[Channel].Time,
						Config.LoRaDevices[Channel].Latitude,
						Config.LoRaDevices[Channel].Longitude,
						Config.LoRaDevices[Channel].Altitude,
						Config.LoRaDevices[Channel].AscentRate);
		}

		if ( !run )
		{
			port_closed = 1;
		}
		else if ( send(connfd, sendBuff, strlen(sendBuff), MSG_NOSIGNAL ) <= 0 )
		{
			LogMessage( "Disconnected from client\n" );
			port_closed = 1;
		}
	}
	
	return port_closed;
}
void *ServerLoop( void *some_void_ptr )
{
    struct sockaddr_in serv_addr;
	struct TServerInfo *ServerInfo;
	
	ServerInfo = (struct TServerInfo *)some_void_ptr;

    ServerInfo->sockfd = socket( AF_INET, SOCK_STREAM, 0 );
    memset( &serv_addr, '0', sizeof( serv_addr ) );

    serv_addr.sin_family = AF_INET;
    serv_addr.sin_addr.s_addr = htonl( INADDR_ANY );
    serv_addr.sin_port = htons(ServerInfo->Port);

    LogMessage( "Listening on JSON port %d\n", ServerInfo->Port);

    if (setsockopt(ServerInfo->sockfd, SOL_SOCKET, SO_REUSEADDR, &(int){1}, sizeof(int)) < 0)
	{
        LogMessage( "setsockopt(SO_REUSEADDR) failed" );
    }

    if (bind(ServerInfo->sockfd, (struct sockaddr *) &serv_addr, sizeof(serv_addr)) < 0)
    {
        LogMessage( "Server failed errno %d\n", errno );
        exit( -1 );
    }

    listen(ServerInfo->sockfd, 10);
	
    while (run)
    {
		int SendEveryMS = 1000;
		int MSPerLoop=100;
        int ms=0;
		int connfd;
		
		Config.EnableDev=1;

		fcntl(ServerInfo->sockfd, F_SETFL, fcntl(ServerInfo->sockfd, F_GETFL) & ~O_NONBLOCK);	// Blocking mode so we wait for a connection

        connfd = accept(ServerInfo->sockfd, ( struct sockaddr * ) NULL, NULL );	// Wait for connection

        LogMessage( "Connected to client\n");
		ServerInfo->Connected = 1;

		fcntl(connfd, F_SETFL, fcntl(ServerInfo->sockfd, F_GETFL) | O_NONBLOCK);	// Non-blocking, so we don't block on receiving any commands from client

        while (ServerInfo->Connected)
        {
			char packet[4096];
			int bytecount;

			ms += MSPerLoop;
			
			// Listen part
			bytecount = -1;
				
			while ((bytecount = recv(connfd, packet, sizeof(packet), 0)) > 0)
			{
				char *line, *saveptr;
				
				packet[bytecount] = 0;
				
				// JSON server
				line = strtok_r(packet, "\n", &saveptr);
				while (line)
				{
					ProcessJSONClientLine(connfd, line);
					line = strtok_r( NULL, "\n", &saveptr);
				}
			}
			
			if (bytecount == 0)
			{
				// -1 is no more data, 0 means port closed
				LogMessage("Disconnected from client\n");
				ServerInfo->Connected = 0;
			}

			if (ServerInfo->Connected)
			{
				// Send to JSON client
				if (ms >= SendEveryMS)
				{
					if (SendJSON(connfd))
					{
						ServerInfo->Connected = 0;
					}
				}
			}
			
			delay(MSPerLoop);
        }

        close(connfd);
    }

    return NULL;
}