agent_data.cpp File Reference

#include "support/configCosmos.h"
#include <sys/stat.h>
#include <stdio.h>
#include "agent/agentclass.h"
#include "physics/physicslib.h"
#include "support/jsonlib.h"
#include "support/datalib.h"
#include <cstring>
#include <fstream>
#include <iostream>

Include dependency graph for agent_data.cpp:

Macros
#define	MAXBUFFERSIZE   256
#define	REQUEST_RUN_PROGRAM   0

Functions
int	agent_data ()
int32_t	request_login (string &request, string &response, Agent *)
int32_t	request_getnodelist (string &request, string &response, Agent *)
int32_t	request_log (string &request, string &response, Agent *)
int	main (int, char **)

Variables
std::vector< agentstruc >	agents
agentstruc	tempagent
std::vector< cosmosstruc >	nodes
char	tempname [100]
char	agentname [COSMOS_MAX_NAME+1] = "data"
std::string	dataDir
int	waitsec = 5
Agent *	agent
int	usercount =0
char	username [1000][50]
char	password [1000][50]
int32_t	diskfree
int32_t	stateflag
int32_t	myport

Macro Definition Documentation

#define MAXBUFFERSIZE   256

#define REQUEST_RUN_PROGRAM   0

Function Documentation

int agent_data

(

)

{

// Start performing the body of the agent
while(agent->running())
    {
    // Gather system information


    COSMOS_USLEEP(10); // no support in win
    }
return 0;
}

int32_t request_login	(	string &	request,
		string &	response,
		Agent *
	)

{
char user[COSMOS_MAX_NAME+1];
char pass[COSMOS_MAX_NAME+1];
int i;
bool login;

login=false;
user[0] = 0;    // These need to be reset, otherwise, logging in with no name lets these retain the previous users name/password.
pass[0] = 0;

sscanf(request.c_str(),"%*s %s %s",user,pass);

for(i=0;i<usercount;i++)
    {   //match given data against what is stored in the array
    if((strcmp(username[i],user)==0) && strcmp(password[i],pass)==0){
        login=true;
        break;
        }
    }

if (login==true)
    response = ("yes");
else
    response = ("no");
return 0;
}

int32_t request_getnodelist	(	string &	request,
		string &	response,
		Agent *
	)

{
uint32_t i;

response.clear();
for (i=0; i<nodes.size(); i++)
    {
    response += ' ' + nodes[i].node.name + ' ' + to_signed(nodes[i].node.type);
    }
return 0;
}

int32_t request_log	(	string &	request,
		string &	response,
		Agent *
	)

{
FILE *log;
uint16_t i;
double utc;

for (i=0; i<request.length(); ++i)
{
    if (request[i] == '{')
        break;
}

json_parse(&request[i], agent->cinfo); //cinfo->stat.agent.user
                                                    // .event
                                                    // .utc ->date
                                                    // .node
utc = (int)(agent->cinfo->event[0].utc);
log = data_open(data_type_path(agent->cinfo->event[0].node, (char *)"data", (char *)"outgoing", utc, (char*)"request_log"), (char*)"a");
fprintf(log,"%s\n",&request[i]);
fclose(log);
return 0;
}

int main	(	int	,
		char **
	)

{
int32_t iretn;
uint32_t i;
char input[100];
FILE *fd;

// Setting the directory where the data is
if (set_cosmosnodes(dataDir) < 0)
{
    printf("Couldn't find Nodes directory\n");
    exit (1);
}

// check if we are already running the agent
if ((iretn=agent->check_agent((char *)"hmoc", agentname, waitsec)) > 0)
    exit (iretn);

// Load user list
usercount = 0;
sprintf(input,"%s/login.dat",dataDir.c_str());

if ((fd = fopen(input,"r")) != NULL)
    {
    while(!feof(fd))
        {//read strings into array
        if (fgets(input,100,fd) != NULL)
            {
            sscanf(input, "%s %s",username[usercount],password[usercount]);
            usercount++;
            }
        }
    fclose(fd);
    }

// Load Node list
data_get_nodes(nodes);

for (i=0; i<nodes.size(); ++i)
    {
#if defined(COSMOS_WIN_OS)
    char command_line[100];
    sprintf(command_line, "agent_node %s", (const char*)nodes[i].node.name);

    STARTUPINFOA si;
    PROCESS_INFORMATION pi;

    ZeroMemory( &si, sizeof(si) );
    si.cb = sizeof(si);
    ZeroMemory( &pi, sizeof(pi) );

#if defined(COSMOS_WIN_BUILD_MSVC)
        if (CreateProcess(NULL, (LPWSTR) command_line, NULL, NULL, FALSE, 0, NULL, NULL, &si, &pi))
#else
        if (CreateProcessA(NULL, (LPSTR) command_line, NULL, NULL, FALSE, 0, NULL, NULL, &si, &pi))
#endif
    {
        strcpy(tempagent.beat.node, nodes[i].node.name);
        tempagent.pid = pi.dwProcessId;
        agents.push_back(tempagent);
        CloseHandle( pi.hProcess );
        CloseHandle( pi.hThread );
    }
#else
    int32_t pid =fork();
    switch(pid)
    {
    case -1:
        strcpy(tempagent.beat.node, nodes[i].node.name);
        tempagent.pid=pid;
        agents.push_back(tempagent);
        break;
    case 0:
        int devn;
        devn = open("/dev/null",O_RDWR);
        dup2(STDIN_FILENO, devn);
        dup2(STDOUT_FILENO, devn);
        dup2(STDERR_FILENO, devn);
        close(devn);
        execl("agent_node", (const char*)nodes[i].node.name,(char*)nullptr);

        fflush(stdout);
        exit(0);
        break;
    default:
        break;  
        }
#endif
    }
// Initialize the Agent
// near future: support cubesat space protocol
// port number = 0 in this case, automatic assignment of port
if (!(agent = new Agent("hmoc", agentname, 1., MAXBUFFERSIZE)))
    exit (iretn);

// Add additional requests
if ((iretn=agent->add_request("login",request_login)))
    exit (iretn);
if ((iretn=agent->add_request("getnodelist",request_getnodelist)))
    exit (iretn);
if ((iretn=agent->add_request("log",request_log)))
    exit (iretn);

// Start our own thread
iretn = agent_data();
}

Variable Documentation

std::vector<agentstruc> agents

agentstruc tempagent

std::vector<cosmosstruc> nodes

char tempname[100]

char agentname[COSMOS_MAX_NAME+1] = "data"

std::string dataDir

int waitsec = 5

Agent* agent

int usercount =0

char username[1000][50]

char password[1000][50]

int32_t diskfree

int32_t stateflag

int32_t myport