コード例 #1
0
ファイル: srun.c プロジェクト: labmec/oopar
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;
}
コード例 #2
0
ファイル: userManager.c プロジェクト: macDemarco/Firewall
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;
	}
}