pipes.cpp

来自「D-ITG2.4源代码」· C++ 代码 · 共 169 行

 /*	Component of the D-ITG 2.4 Platform
 *
 * 	
 *	copyright	: (C) 2004  	by Stefano Avallone, Alessio Botta, Donato Emma, 
 *					Salvatore Guadagno, Antonio Pescape'
 *					DIS Dipartimento di Informatica e Sistemistica				 
 *					(Computer Science Department)
 *					University of Naples "Federico II"	
 *	email:		: {stavallo, pescape}@unina.it, {abotta, demma, sguadagno}@napoli.consorzio-cini.it
 *
 *	This program is free software; you can redistribute it and/or modify
 *	it under the terms of the GNU General Public License as published by
 *	the Free Software Foundation; either version 2 of the License, or
 *      (at your option) any later version.
 */
 
 


#include "../common/ITG.h"
#include <unistd.h>
#include "pipes.h"

#ifdef WIN32
int numPipes = 0;
HANDLE mutex_numPipes;
#endif

#ifdef LINUX_OS
int sendPipeMsg(int fd[2], pipeMsg *msg)
{
	write(fd[1], msg, sizeof(pipeMsg));
	return(0);
}

int recvPipeMsg(int fd[2], pipeMsg *msg)
{
	read(fd[0], msg, sizeof(pipeMsg));
	return(0);
}

int createNewPipe(int fd[2])
{
	return(pipe(fd));
}

int closePipe(int fd[2])
{
	close(fd[1]);
	close(fd[0]);
	return(0);

}
#endif

#ifdef WIN32
int sendPipeMsg(HANDLE pipe[3], pipeMsg *msg) 
{
	DWORD written = 0;
	DWORD err;
	HANDLE fd = pipe[0];
	HANDLE event = pipe[1];

	if (WriteFile(fd, msg, sizeof(pipeMsg), &written, NULL) == 0) {
                printf("Error writing in pipe, with written = %ld\n",written); 
                err = GetLastError(); 
                printf("GetLastError returned %ld\n", err); 
		return(-1);
	}
	if (written != sizeof(pipeMsg)) {
		printf("Error writing in pipe: written != sizeof(pipeMsg)\n");
		return(-1);
	}

#ifdef DEBUG
	printf("sendPipeMsg: written: %d\n", written);
#endif
	SetEvent(event);
	return(0);
}

int recvPipeMsg(HANDLE pipe[3], pipeMsg *msg) 
{
	DWORD read = 0;
	DWORD err;

	
	if (ReadFile(pipe[2], msg, sizeof(pipeMsg) , &read , NULL) == 0) {
		printf("Error reading from pipe, with read = %ld\n", read);
                err = GetLastError(); 
                printf("GetLastError returned %ld\n", err); 
		return(-1);
	}
	if (read != sizeof(pipeMsg)) {
		printf("Error reading from pipe: read != sizeof(pipeMsg)\n");
		return(-1);
	}

	return(0);
}


int createNewPipe(HANDLE pipe[3])
{
	int num;
	char pipename[30];
	HANDLE hpipe, namedPipe;
	WSAEVENT event;
	char stringa[30];
	
	MUTEX_THREAD_LOCK(mutex_numPipes);
	num = numPipes++;
	MUTEX_THREAD_UNLOCK(mutex_numPipes);

	
	strcpy(pipename,"\\\\.\\pipe\\");
	strcat(pipename, nameProgram);
	 sprintf(stringa, "%d", num);
	strcat(pipename, stringa);

#ifdef DEBUG
	printf("pipename: %s\n", pipename);
#endif

	
	namedPipe = CreateNamedPipe(
		(LPTSTR)pipename,			
		PIPE_ACCESS_DUPLEX,		
		PIPE_TYPE_BYTE |		
		PIPE_READMODE_BYTE |		
		PIPE_WAIT,			
		PIPE_UNLIMITED_INSTANCES,	
		PIPE_BUFSIZE,			
		PIPE_BUFSIZE,			
		PIPE_TIMEOUT,			
		NULL				
	);
	if (namedPipe == INVALID_HANDLE_VALUE) {
		printf("Error in CreateNamedPipe()\n");
		return (-1);
	}

	
	hpipe = CreateFile(pipename, GENERIC_READ | GENERIC_WRITE, 0, NULL, CREATE_NEW, 0, 0);
	if (hpipe == INVALID_HANDLE_VALUE) {
    		printf("Error in CreateFile()\n");
		return (-1);
	}

	
	event = CreateEvent(NULL, FALSE, FALSE, NULL);
	ResetEvent(event);

	pipe[0] = hpipe;  
	pipe[1] = event;  
	pipe[2] = namedPipe; 
	return(0);
}

int closePipe(HANDLE pipe[3])
{
	CloseHandle(pipe[2]);
	return 0;
}

#endif