“生产者-消费者”编程

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <fcntl.h>
#include <pthread.h>
#include <errno.h>
#include <semaphore.h>
#include <sys/ipc.h>
#include <memory.h>

#define MYFIFO "myfifo"
#define BUFFER_SIZE 3
#define UNIT_SIZE 5
#define RUN_TIME 30
#define DELAY_TIME_LEVELS 5.0

int fd;
time_t end_time; //?
sem_t mutex,full,avail;

void *producer(void *arg)
{
	int real_write;
	int delay_time=0;
	while(time(NULL)<end_time)
	{
		delay_time=(int)(rand() *DELAY_TIME_LEVELS/(RAND_MAX)/2.0)+1;    //假设生产者速度比消费者平均快1倍
		sleep(delay_time);
                /*P 操作信号量avail  mutex ，信号量减一*/
		sem_wait(&avail);
		sem_wait(&mutex);
		printf("\nProducer: delay %d\n",delay_time);
		/*生产者写入数据*/
		if((real_write=write(fd,"hello",UNIT_SIZE))==-1)
		{
			if(errno==EAGAIN)
				printf("The FIFO has not been read yet.\n");
			
		}
		else
			printf("Write %d to the FIFO\n",real_write);
		/*V 操作信号量full  mutex，信号量加一*/
		sem_post(&full);
		sem_post(&mutex);

	}
	pthread_exit(NULL);
}

void *customer(void *arg)
{	
	unsigned char read_buffer[UNIT_SIZE];
	int real_read;
	int delay_time=0;
	
	while(time(NULL)<end_time)
	{
		delay_time=(int)(rand() *DELAY_TIME_LEVELS/(RAND_MAX))+1;
		sleep(delay_time);

		sem_wait(&full);
		sem_wait(&mutex);
		memset(read_buffer,0,UNIT_SIZE);
		printf("\nCustomer: delay %d\n",delay_time);
		if((real_read=read(fd,read_buffer,UNIT_SIZE))==-1)
		{
			if(errno==EAGAIN)
				printf("No data yet.\n");
			
		}
		
		printf("Read %s from FIFO\n",read_buffer);
		sem_post(&avail);
		sem_post(&mutex);

	}
	pthread_exit(NULL);
}

int main()
{	
	pthread_t thrd_prd_id,thrd_cst_id;
	pthread_t mon_th_id;	
	int ret;
	
	srand(time(NULL));//?
	end_time=time(NULL)+RUN_TIME;

	if((mkfifo(MYFIFO,O_CREAT|O_EXCL)<0)&&(errno !=EEXIST))
	{
		printf("Cannot create fifo\n");
		return errno;
	}

	
	if(fd= open(MYFIFO,O_RDWR)==-1)
	{
		perror("open");
		printf("Open fifo error\n");
		return fd;
	}

	ret =sem_init(&mutex,0,1);                        //缓冲区互斥
	ret +=sem_init(&full,0,0);                           //缓冲区非空单元数
	ret +=sem_init(&avail,0,BUFFER_SIZE);     //缓冲区空单元数
	if(ret!=0)
	{
		printf("Any semaphore initialization failed\n");
		return ret;	
	}

	ret =pthread_create(&thrd_prd_id,NULL,producer,NULL);
	if(ret !=0)
	{
		printf("Create producer thread error\n");
		return ret;
	}
	ret =pthread_create(&thrd_cst_id,NULL,customer,NULL);
	
	if(ret !=0)
	{
		printf("Create customer thread error\n");
		return ret;
	};
	pthread_join(thrd_prd_id,NULL);
	pthread_join(thrd_cst_id,NULL);
	close(fd);
	unlink(MYFIFO);
	return 0;
}