int main(int argc, char *argv[])
{
// estruturas para comunicacao
struct sockaddr_in sAddr, mAddr, cAddr; // estruturas de endereco em modo servidor e cliente
int sSocket, cSocket, mSocket; // sSocket - modo servidor - ouve
// cSocket - comunicacao em modo servidor - comunica-se com cliente
// mSocket - modo mensageiro - para encaminhar mensagens
struct hostent *host;
char hostName[256];
int ret, nbytes, length, bufsize, tmpPort, tmpRank;
struct srunEnvelope msgSend, msgRecv; // envelope para comunicacao
// final das estruturas
// manipulacao de parametros
int np, i, j, binIndex;
char hostFile[256] = "hostfile.oopar";
char bin[256], binArgs[256] = "";
// variaveis de controle
int condition=STILL_RUNNING, // condicao para continuar o loop
totalNp=0; // numero total de processos que enviaram o finalize - quando for igual ao NP, termina
//---Tratamento dos parametros de linha de comando----------
if(argc < 4)
help();
if(strcmp(argv[1],"-np")!=0)
help();
np = atoi(argv[2]);
if(np<=0)
help();
if(strcmp(argv[3],"-hostfile")!=0)
{
binIndex=3;
}
else
{
if(argc<6)
help();
strcpy(hostFile,argv[4]);
binIndex=5;
}
if(argc > binIndex+1)
{
if(argc < binIndex+3)
help();
if(strcmp(argv[binIndex+1],"-args")!=0)
help();
strcpy(binArgs,argv[binIndex+2]);
}
strcpy(bin,argv[binIndex]);
printf("-np %d\n",np);
printf("-hostfile %s\n", hostFile);
printf("%s",bin);
printf(" %s\n\n",binArgs);
//-------------
parseHostfile(hostFile);
#if DEBUG
showHosts();
#endif
// preparacao do socket para ficar ouvindo em modo servidor
// cria socket
if ( (sSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) < 0 )
{
fprintf(stderr,"SRUN Error: Error creating listening socket\n");
exit(1);
}
int yes=1;
if (setsockopt(sSocket, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(int)) < 0)
{
fprintf(stderr,"SRUN Error: Setsockopt\n");
exit(1);
}
sAddr.sin_family = AF_INET;
sAddr.sin_port = htons(SRUN_PORT);
sAddr.sin_addr.s_addr = INADDR_ANY;
memset(&(sAddr.sin_zero), '\0', 8);
//bind da porta
if (bind(sSocket, (struct sockaddr *)&sAddr, sizeof(struct sockaddr)) < 0)
{
fprintf(stderr,"SRUN Error: Error on binding listening address %d \n", SRUN_PORT);
perror("");
close(sSocket);
exit(1);
}
// listen da fila
if (listen(sSocket,MAX_QUEUE) < 0)
{
fprintf(stderr,"SRUN Error: Error on defining listening queue\n");
perror("");
close(sSocket);
exit(1);
}
//----------------------------------------
// FASE 1
// entrar em contato com cada um dos spds dos participantes para iniciar o processo de startup
for(i=0; i<np; i++)
{
if ((mSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) < 0)
{
fprintf(stderr,"SRUN Error: Could not create socket\n");
exit(1);
}
mAddr.sin_family = AF_INET;
mAddr.sin_port = htons(SPD_PORT);
host = gethostbyname(pTable[i].hostName);
if (host == NULL)
{
fprintf(stderr,"SRUN Error: Could not resolve %s host\n",pTable[i].hostName);
exit(1);
}
memcpy(&mAddr.sin_addr, host->h_addr, host->h_length);
//mAddr.sin_addr.s_addr = inet_addr(pTable[i].hostName);
printf("Conectando com SPD em %s %d\n", pTable[i].hostName, SPD_PORT); fflush(stdout);
if (connect(mSocket,(struct sockaddr*) &mAddr, sizeof(struct sockaddr_in)) < 0)
{
fprintf(stderr,"SRUN Error: Startup Error. Could not connect to host %s\n",pTable[i].hostName);
exit(1);
}
// FIXME - completar path do binario.
// Token separador dos elementos do binario e seus argumentos eh o espaco em branco
msgSend.msgType = STARTUP;
strcpy(msgSend.message,bin);
strcat(msgSend.message," ");
strcat(msgSend.message,binArgs);
printf("Msg: %s\n",msgSend.message); fflush(stdout);
if (send(mSocket, (char*)&msgSend, sizeof(struct srunEnvelope), 0) < 0)
{
fprintf(stderr,"SRUN Error: Error sending startup message\n");
exit(1);
}
if (recv(mSocket, (char*)&msgRecv, sizeof(struct srunEnvelope), 0) < 0)
{
fprintf(stderr,"SRUN Error: Error receiving por number form SPD\n");
exit(1);
}
// unmarshalling do numero da porta na mensagem
if (msgRecv.msgType == SETPORT)
{
sscanf(msgRecv.message,"%d",&pTable[i].port);
printf("SRUN - Recebi mensagem de Port : %d\n",pTable[i].port); fflush(stdout);
}
else
{
fprintf(stderr,"SRUN Error: Protocol error\n");
exit(1);
}
close(mSocket);
}
// --Notificacao de todos os demais processos para a cria'c~ao da tabela de processos.
// descomentar quando houver o INIT
// passo 1 - criacao da lista de processos
// tamanho do buffer = quantidade de processos
bufsize = np * sizeof(struct processTable);
msgSend.msgType = NOTIFYALL;
// passo 2 - enviar a estrutura para os participantes
printf("FASE 2 -----------\n");fflush(stdout);
showHosts();
for(i=0; i<np; i++)
{
sprintf(msgSend.message, "%d %d %d" , bufsize, np, i);
if ((mSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) < 0)
{
fprintf(stderr,"SRUN Error: Could not create socket\n");
exit(1);
}
mAddr.sin_family = AF_INET;
mAddr.sin_port = htons(pTable[i].port);
host = gethostbyname(pTable[i].hostName);
if (host == NULL)
{
fprintf(stderr,"SRUN Error: Could not resolve %s host\n",pTable[i].hostName);
exit(1);
}
memcpy(&mAddr.sin_addr, host->h_addr, host->h_length);
//mAddr.sin_addr.s_addr = inet_addr(pTable[i].hostName);
printf("Conectando com PROCESSO em %s %d\n", pTable[i].hostName, pTable[i].port); fflush(stdout);
while (connect(mSocket,(struct sockaddr*) &mAddr, sizeof(struct sockaddr_in)) < 0) usleep(250000);
//essa parte tem que ser espera ocupada para garantir a sincronizacao entre as entidades
/*{
fprintf(stderr,"SRUN Error: Startup Error. Could not connect to host %s\n",pTable[i].hostName);
exit(1);
}*/
printf("passou - %d %d\n", msgSend.msgType, bufsize);fflush(stdout);
if (send(mSocket, (char*)&msgSend, sizeof(struct srunEnvelope), 0) < 0)
{
fprintf(stderr,"SRUN Error: Error sending notify message\n");
exit(1);
}
if (send(mSocket, (char*)&pTable, bufsize, 0) < 0)
{
fprintf(stderr," SRUN Error: Error sending process table configuration message\n");
exit(1);
}
close(mSocket);
}
// FASE 3
// loop para aguardar mensagens
printf("Aguardando mensagens de alguem \n"); fflush(stdout);
while(condition == STILL_RUNNING)
{
length = sizeof(struct sockaddr_in);
cSocket = accept(sSocket, (struct sockaddr*) &cAddr, &length);
if (nbytes = recv(cSocket, (char*)&msgRecv, sizeof(struct srunEnvelope), 0) < 0)
{
fprintf(stderr,"SRUN Error: Error receiving message\n");
exit(1);
}
printf("SRUN -> Received %d\n", msgRecv.msgType); fflush(stdout);
switch (msgRecv.msgType)
{
case NEWPROC:
// --> FASE 1 -- enviar a mensagem ao SPD responsavel pela criacao do respectivo processo
if (mSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP) < 0)
{
fprintf(stderr,"SRUN Error: Could not create socket\n");
exit(1);
}
mAddr.sin_family = AF_INET;
mAddr.sin_port = htons(SPD_PORT);
host = gethostbyname(pTable[np].hostName);
if (host == NULL)
{
fprintf(stderr,"SRUN Error: Could not resolve %s host\n",pTable[np].hostName);
exit(1);
}
memcpy(&mAddr.sin_addr, host->h_addr_list[0], host->h_length);
if (connect(mSocket,(struct sockaddr*) &mAddr, sizeof(struct sockaddr_in)) < 0)
{
fprintf(stderr,"SRUN Error: Startup Error. Could not connect to host %d\n",pTable[i].hostName);
exit(1);
}
// FIXME - completar path do binario.
// Token separador dos elementos do binario e seus argumentos eh o espaco em branco
msgSend.msgType = STARTUP;
strcpy(msgSend.message,bin);
strcpy(msgSend.message," ");
strcpy(msgSend.message,binArgs);
if (send(mSocket, (char*)&msgSend, sizeof(struct srunEnvelope), 0) < 0)
{
fprintf(stderr,"SRUN Error: Error sending startup message\n");
exit(1);
}
if (recv(mSocket, (char*)&msgRecv, sizeof(struct srunEnvelope), 0) < 0)
{
fprintf(stderr,"SRUN Error: Error receiving por number form SPD\n");
exit(1);
}
// unmarshalling do numero da porta na mensagem
if (msgRecv.msgType == SETPORT)
sscanf(msgRecv.message,"%d",&pTable[np].port);
else
{
fprintf(stderr,"SRUN Error: Protocol error\n");
exit(1);
}
np++; //incremento o numero de processos
close(cSocket);
close(mSocket);
break;
case FINALIZE:
totalNp++;
if (totalNp == np)
condition=FINISHED;
close(cSocket);
break;
default:
fprintf(stderr,"SRUN Error: Protocol error - Invalid Message\n");
exit(1);
}
}
printf("Saindo do SRUN...\n"); fflush(stdout);
if(close(sSocket))
perror("Error in closing socket: ");
return 0;
}
int main(int argc, const char * argv[])
{
Bool result = TRUE;
if ((argc < 2) || (argc > 3))
{
printf(USAGE);
return -1;
}
if (strcmp(argv[1], LOAD_RULES) == 0)
{
if (argc != 3)
{
printf("A path to a rules file must be specified in order to load rules.\n");
return -1;
}
else
{
result = loadRules(argv[2]);
if (!result)
{
return -1;
}
return 0;
}
}
else if (strcmp(argv[1], LOAD_HOSTS) == 0)
{
if (argc != 3)
{
printf("A path to a hosts file must be specified in order to load hosts.\n");
return -1;
}
else
{
result = loadHosts(argv[2]);
if (!result)
{
return -1;
}
return 0;
}
}
else if (argc != 2)
{
printf("Any action other than %s, %s must not take any parameters.\n", LOAD_RULES, LOAD_HOSTS);
return -1;
}
if (strcmp(argv[1], ACTIVATE) == 0)
{
result = activate();
}
else if (strcmp(argv[1], DEACTIVATE) == 0)
{
result = deactivate();
}
else if (strcmp(argv[1], SHOW_RULES) == 0)
{
result = showRules();
}
else if (strcmp(argv[1], CLEAR_RULES) == 0)
{
result = clearRules();
}
else if (strcmp(argv[1], SHOW_LOG) == 0)
{
result = showLog();
}
else if (strcmp(argv[1], CLEAR_LOG) == 0)
{
result = clearLog();
}
else if (strcmp(argv[1], SHOW_CONNECTION_TABLE) == 0)
{
result = showConnectionTable();
}
else if (strcmp(argv[1], SHOW_HOSTS) == 0)
{
result = showHosts();
}
else
{
printf("The only valid actions are: %s, %s, %s, %s, %s, %s, %s, %s\n",
ACTIVATE, DEACTIVATE, SHOW_RULES, CLEAR_RULES, LOAD_RULES, SHOW_LOG,
CLEAR_LOG, SHOW_CONNECTION_TABLE, SHOW_HOSTS, LOAD_HOSTS);
}
if (result)
{
return 0;
}
else
{
return -1;
}
}