linux 下多程序的同步
linux 多程序的同步:linux多程序我實現同步操作,操作單個訊號量已經不能實現,對多程序的通訊可以採取訊號集的方式,一個訊號集包含了多個訊號量。
首先通過semget()建立訊號量。例如:semid = semget(SEMKEY,2,0600|IFLAGS;
然後對訊號集中各個訊號量賦初值:
semctl(semid,0,SETVAL,0);
semctl(semid,1,SETVAL,1);
也可以通過 set_sem_value(semid,0,1),對訊號集為semid重的第一個訊號量賦初始值為1
void set_sem_value(int semid,int num ,int val)
{
union semun semun_info;
semun_info.val=val;
semctl(semid,0,SETVAL,semun_info);
}
然後就是類似於PV原語的操作。
注意一下下面的這個機構體:
struct sembuf p1 = {0,-1,0},p2 = {1,-1,0},
v1 = {0,1,0},v2 = {1,1,0};
然後利用 int semop(int semid, struct sembuf *sops, unsigned nsops); 進行pv操作!。下面是具體的程式碼:
/**************************************produce.c*******************************/
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <fcntl.h>
#include <sys/time.h>
#include <sys/mman.h>
#include <sys/ioctl.h>
#include <sys/types.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <sys/sem.h>
#include<errno.h>
#define SHMKEY (key_t) 0x100
#define SEMKEY (key_t) 0x200
#define IFLAGS (IPC_CREAT|IPC_EXCL)
#define ERR ((struct databuf *) -1)
struct databuf{
int d_buf[10];
};
static int shmid,semid;
pr_error(char *mess)
{
perror(mess);
exit(1);
}
getseg(struct databuf* *pdata)
{
/*取得共享記憶體的key,如果還沒建立的話,就建立(IPC_CREAT)*/
if((shmid =shmget(SHMKEY,sizeof(struct databuf),0600|IFLAGS)) < 0)
{
if( errno == EEXIST)
{
shmid=shmget(SHMKEY,sizeof(struct databuf),0600|IPC_CREAT);
if(shmid <0)
pr_error("shmget");
}
else
pr_error("shmget");
}
/*取得共享記憶體的指標,使用共享記憶體時可以和malloc分配的記憶體一樣*/
if((*pdata = (struct databuf *)(shmat(shmid,0,0))) == ERR)
pr_error("shmat");
}
int getsem()
{ /*建立訊號量(且稱為A和B)*/
if((semid = semget(SEMKEY,2,0600|IFLAGS)) < 0)
{
if(errno == EEXIST)
{
semid=semget(SEMKEY,2,0600|IPC_CREAT);
if( semid <0)
pr_error("semget");
else
/*加上這一句,如果已存在,不再初始化訊號量*/
return semid;
}
else
pr_error("semget");
}
/*設定訊號量A初值為0*/
if(semctl(semid,0,SETVAL,0) < 0)
pr_error("semctl");
/*設定訊號量B初值為1*/
if(semctl(semid,1,SETVAL,1) < 0)
pr_error("semctl");
return(semid);
}
void remove_s()
{
/*刪除共享記憶體*/
if(shmctl(shmid,IPC_RMID,NULL) < 0)
pr_error("shmctl");
/*刪除訊號量*/
if(semctl(semid,IPC_RMID,NULL) < 0)
pr_error("semctl");
}
main()
{
int semid;
struct databuf *buf;
semid = getsem();
getseg(&buf);
writer(semid,buf);
remove_s();
exit(0);
}
struct sembuf p1 = {0,-1,0},p2 = {1,-1,0},
v1 = {0,1,0},v2 = {1,1,0};
writer(int semid,struct databuf *buf)
{
int i,j;
for(i = 0;i < 10;i++){
/*P(B) B=0;準備寫緩衝區*/
printf("next loop\n");
semop(semid,&p2,1);
/*寫共享記憶體*/
for(j = 0;j < 10;j++){
buf -> d_buf[j] = 10 * i + j;
printf("buf->d_buf[%d]=%d\n",j,buf -> d_buf[j] );
}
sleep(5);
/*V(A) A==1,喚醒消費者程序,讀緩衝區*/
semop(semid,&v1,1);
}
return;
}
/****************************consumer.c***************************/
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <fcntl.h>
#include <sys/time.h>
#include <sys/mman.h>
#include <sys/ioctl.h>
#include <sys/types.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <sys/sem.h>
#include<errno.h>
#define SHMKEY (key_t) 0x100
#define SEMKEY (key_t) 0x200
#define IFLAGS (IPC_CREAT|IPC_EXCL)
#define ERR ((struct databuf *) -1)
struct databuf{
int d_buf[10];
};
static int shmid,semid;
pr_error(char *mess)
{
perror(mess);
exit(1);
}
getseg(struct databuf* *pdata)
{
/*取得共享記憶體的key,如果還沒建立的話,就建立(IPC_CREAT)*/
if((shmid =shmget(SHMKEY,sizeof(struct databuf),0600|IFLAGS)) < 0)
{
if( errno == EEXIST)
{
shmid=shmget(SHMKEY,sizeof(struct databuf),0600|IPC_CREAT);
if(shmid <0)
pr_error("shmget");
}
else
pr_error("shmget");
}
/*取得共享記憶體的指標,使用共享記憶體時可以和malloc分配的記憶體一樣*/
if((*pdata = (struct databuf *)(shmat(shmid,0,0))) == ERR)
pr_error("shmat");
}
int getsem()
{ /*建立訊號量(且稱為A和B)*/
if((semid = semget(SEMKEY,2,0600|IFLAGS)) < 0)
{
if(errno == EEXIST)
{
semid=semget(SEMKEY,2,0600|IPC_CREAT);
if( semid <0)
pr_error("semget");
else
/*加上這一句,如果已存在,不再初始化訊號量*/
return semid;
}
else
pr_error("semget");
}
/*設定訊號量A初值為0*/
if(semctl(semid,0,SETVAL,0) < 0)
pr_error("semctl");
/*設定訊號量B初值為1*/
if(semctl(semid,1,SETVAL,1) < 0)
pr_error("semctl");
return(semid);
}
main()
{
int semid;
struct databuf *buf;
semid = getsem();
getseg(&buf);
reader(semid,buf);
exit(0);
}
struct sembuf p1 = {0,-1,0},p2 = {1,-1,0},
v1 = {0,1,0},v2 = {1,1,0};
reader(int semid, struct databuf *buf)
{
int i,j;
for(i = 0;i <10;i++){
/*P(A) A==-1,如果沒資料的話,blocking,有的話,讀*/
printf("next loop\n");
semop(semid,&p1,1);
for(j = 0;j < 10;j++){
printf(" %d ",buf -> d_buf[j]);
}
printf("\n");
semop(semid,&v2,1);
/*V(B) B++,喚醒寫程序*/
}
return;
}