使用信号量机制来模拟实现 PV 操作

/********** sem_pv.h *************************/
#include <sys/types.h>
#include <sys/ipc.h>
#include <sys/sem.h>
#include <stdlib.h>
#include <stdio.h>
typedef union semun{
int val;
struct semid_ds *buf;
unsigned short *array;
}semun;
//初始化信号量
int sem_init(int key)
{
int semid;
semun arg;

if((semid=semget(key,1,0660|IPC_CREAT|IPC_EXCL))<0) //创建包含 1 个信号量
的信号量对象
{
perror(“sem_init:semget error”);
exit(1);
}
arg.val=1;
if(semctl(semid,0,SETVAL,arg)<0)
{
perror(“sem_init:semctl error”);
exit(2);
};
return semid;
}
//信号量的 P 操作
int sem_p(int semid)
{
struct sembuf pbuf;
pbuf.sem_num=0;
pbuf.sem_op=-1;
pbuf.sem_flg=SEM_UNDO;
if(semop(semid,&pbuf,1)-1)
{
perror(“sem_p:semop error”)exit(3);
}
return 0;
}
//信号量的 V 操作
int sem_v(int semid)
{
struct sembuf vbuf;
vbuf.sem_num=0;
vbuf.sem_op=1;
vbuf.sem_flg=SEM_UNDO;
if(semop(semid,&vbuf,1)-1)
{
perror(“sem_v:semop error”)exit(3);
}
return 0;
}
//删除信号量
void sem_rmv(int semid)
{
if(semctl(semid,0,IPC_RMID)<-1)
{
perror(“sem_rmv:semctl error”);
exit(2);
}
}

/********** test.c *************/
#include <unistd.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <sys/wait.h>
#include <string.h>
#include<stdio.h>
#include “sem_pv.h”
int main(void)
{
key_t key;
int pid,fd,semid,n;
char str[80];
key=ftok(".",5);
semid=sem_init(key);
fd=open(“test.txt”,O_RDWR|O_CREAT|O_TRUNC,0644);
while((pid=fork())-1); /´´½¨×Ó½ø³Ì/
if(pid0) /×Ó½ø³Ì/
{
sleep(1);
sem_p(semid);
lseek(fd,SEEK_SET,0);
read(fd,str,sizeof(str));
sem_v(semid);
printf(“child:read str from test file:%s\n”,str);
exit(0);
}
else /¸¸½ø³Ì/
{
sem_p(semid);
printf(“parent:please enter a str for test file(strlen<80):\n”);
gets(str);
n=strlen(str);
lseek(fd, SEEK_SET,0);
write(fd,str,n);
sem_v(semid);
wait(0);
close(fd);
sem_rmv(semid);
exit(0);
}
}