linux 共享内存与信号量,Linux共享内存+信号量编程示例

写在前面：最近被一本超级垃圾的书折磨够呛，书中的例子全是错的，不过幸好我不是初学者的水平。有几个例子还不错，但是书上的代码编译都不过，我折腾了很长时间才修改正确，发上来留着以后复习用。

/**

*描述:通过共享内存进行进程间通信,使用信号量来同步.

*直接运行程序会启动服务端,会打印出shmid

*然后再启动并添加shmid这个参数即可启动客户端,然后即可进行通信

*Created on: 2010-4-29

*Author: QQwen

*开发环境: fc9 + eclipse c/c++

*/

#include

#include

#include

#include

#include

#include

#include

#include

#defineSHMDATASIZE 1000

#defineBUFFERSIZE (SHMDATASIZE -sizeof(int))

#defineSN_EMPTY 0

#defineSN_FULL 1

#defineTRUE 1

intdeleteSemid = 0;

//必须自己定义一个union,否则编译不过

unionsemun

{

intval;

structsemid_ds *buf;

unsignedshortint*array;

structseminfo *__buf;

};

voidserver();

voidclient(intshmid);

voiddelete();

voidsigdelete(intsignum);

voidlocksem(intsemid,intsemnum);

voidunlocksem(intsemid,intsemnum);

//void waitzero(int semid, int semnum);

voidclientwrite(intshmid,intsemid,char*buffer);

intsafesemget(key_t key,intnsems,intsemflg);

intsafesemctl(intsemid,intsemnum,intcmd,unionsemun arg);

intsafesemop(intsemid,structsembuf *sops,unsignednsops);

intsafeshmget(key_t key,intsize,intshmflg);

void* safeshmat(intshmid,constvoid*shmaddr,intshmflg);

intsafeshmctl(intshmid,intcmd,structshmid_ds *buf);

intmain(intargc,char*argv[])

{

if(argc < 2)

{//一个参数启动server

server();

}

else

{//多个参数启动client

client(atoi(argv[1]));//atoi把字符串转为整形数

}

return0;

}

/**

*服务器端

*/

voidserver()

{

unionsemun sunion;//与semctl中的cmd参数有关

intsemid, shmid;

void*shmdata;

char*buffer;

semid = safesemget(IPC_PRIVATE, 2, SHM_R | SHM_W);//创建一个信号集

deleteSemid = semid;

atexit(&delete);//注册终止函数

signal(SIGINT, &sigdelete);//设置某一信号的对应动作,程序终止(interrupt)信号,在用户键入INTR字符(通常是Ctrl-C)时发出

sunion.val= 1;//将一个二元信号量初始化为1

safesemctl(semid, SN_EMPTY, SETVAL, sunion);//SETVAL设置信号量集中的一个单独的信号量的值

sunion.val= 0;//将一个二元信号量赋值为0

safesemctl(semid, SN_FULL, SETVAL, sunion);

shmid = safeshmget(IPC_PRIVATE, SHMDATASIZE, IPC_CREAT | SHM_R | SHM_W);//创建共享内存

shmdata = safeshmat(shmid, 0, 0);//将该共享内存映射进进程的虚存空间

safeshmctl(shmid, IPC_RMID, NULL);//将该共享内存标志为已销毁的，这样在使用完毕后，将被自动销毁

*(int*) shmdata = semid;//将信号量的标识符写入共享内存，以通知其它的进程

buffer = shmdata +sizeof(int);//buf数据的起始地址,因为第一个地址装载了semid

printf("server is running with SHM id ** %d **\n", shmid);

while(TRUE)

{

printf("waiting until full... ");

fflush(stdout);

locksem(semid, SN_FULL);//获得共享资源

printf("done. \n");

printf("message received: %s.\n", buffer);

unlocksem(semid, SN_EMPTY);

}

}

voidclient(intshmid)

{

intsemid;

void*shmdata;

char*buffer;

shmdata = safeshmat(shmid, 0, 0);//将该共享内存映射进进程的虚存空间,这时共享内存的第一个数据是之前sercer写入的

semid = *(int*) shmdata;//获得server创建的共享内存

buffer = shmdata +sizeof(int);//buf数据的起始地址,因为第一个地址装载了semid

printf("client operational: shm id is %d,sem id is %d\n", shmid, semid);

while(TRUE)

{

charinput[3];

printf("\n menu \n 1.Send a message\n");

printf("2.Exit\n");

fgets(input,sizeof(input), stdin);

switch(input[0])

{

case'1':

clientwrite(shmid, semid, buffer);

break;

case'2':

exit(0);

break;

}

}

}

/**

*信号量集得创建与打开

* @para key_t key表示所创建或打开信号量集的键。

* @para int nsems表示创建的信号量集中的信号量的个数，该参数只在创建信号量集时有效。

* @para int semflg表示调用函数的操作类型，也可用于设置信号量集的访问权限，两者通过or表示

* @return如果成功，则返回信号量集的IPC标识符。如果失败，则返回-1

*/

intsafesemget(key_t key,intnsems,intsemflg)

{

intretval;

if((retval = semget(key, nsems, semflg)) == -1)

{

printf("semget error: %s. \n", sys_errlist[errno]);

exit(254);

}

returnretval;

}

/**

*程序终止(interrupt)信号对应的函数

*/

voidsigdelete(intsignum)

{

exit(0);

}

/**

*@para int semid信号集的标识符，即是信号表的索引。

*@para int semnum信号集的索引，用来存取信号集内的某个信号

*@para int cmd需要执行的命令

*@para union semun arg

*@return返回值：如果成功，则为一个正数。如果失败，则为-1

*/

intsafesemctl(intsemid,intsemnum,intcmd,unionsemun arg)

{

intretval;

if((retval = semctl(semid, semnum, cmd, arg)) == -1)

{

printf("semctl error: %s. \n", sys_errlist[errno]);

exit(254);

}

returnretval;

}

/**

*创建一块新的共享内存

* @para key_t key标识共享内存的键值

* @para int size建立共享内存的长度(byte)

* @para int shmflg标志

* @retval成功返回共享内存的标识符；不成功返回-1

*/

intsafeshmget(key_t key,intsize,intshmflg)

{

intretval;

if((retval = shmget(key, size, shmflg)) == -1)

{

printf("shmget error: %s. \n", sys_errlist[errno]);

exit(254);

}

returnretval;

}

/**

*共享内存区对象映射到调用进程的地址空间

* @para int shmid共享内存id

* @para const void *shmaddr共享内存的起始地址(附加到进程的地址空间)

* @para int shmflg标志

*/

void* safeshmat(intshmid,constvoid*shmaddr,intshmflg)

{

void* retval;

if((retval = shmat(shmid, shmaddr, shmflg)) == (void*) -1)

{

printf("shmat error: %s. \n", sys_errlist[errno]);

exit(254);

}

returnretval;

}

/**

*对共享内存的具体控制操作

* @para int shmid共享内存的引用标示符

*/

intsafeshmctl(intshmid,intcmd,structshmid_ds *buf)

{

intretval;

if((retval = shmctl(shmid, cmd, buf)) == -1)

{

printf("shmctl error: %s. \n", sys_errlist[errno]);

exit(254);

}

returnretval;

}

/**

*锁定共享内存

* @para int semid信号量集的引用id

* @para int semnum要操作的信号量

*/

voidlocksem(intsemid,intsemnum)

{

structsembuf sb;//指定调用semop函数所做操作

sb.sem_num= semnum;//指定要操作的信号量

sb.sem_op= -1;//要执行的操作,<0表示进程希望使用资源

sb.sem_flg= SEM_UNDO;//标志

safesemop(semid, &sb, 1);

}

voidunlocksem(intsemid,intsemnum)

{

structsembuf sb;//指定调用semop函数所做操作

sb.sem_num= semnum;//指定要操作的信号量

sb.sem_op= 1;//要执行的操作,>0表示對進程的资源使用完畢,交回该资源

sb.sem_flg= SEM_UNDO;//标志

safesemop(semid, &sb, 1);

}

/*

void waitzero(int semid,int semnum)

{

struct sembuf sb;//指定调用semop函数所做操作

sb.sem_num=semnum;//指定要操作的信号量

sb.sem_op=0;//需要等待直至sem_base.semval变为0

sb.sem_flg=0;//标志

safesemop(semid,&sb,1);

}

*/

/**

*对信号量的操作

* @para int semid信号量集的引用id

* @para struct sembuf *sops用于指定调用semop函数所做的操作

* @para unsigned nsops指定操作函数的个数

*/

intsafesemop(intsemid,structsembuf *sops,unsignednsops)

{

intretval;

if((retval = semop(semid, sops, nsops)) == -1)

{

printf("semop error: %s. \n", sys_errlist[errno]);

exit(254);

}

returnretval;

}

voidclientwrite(intshmid,intsemid,char*buffer)

{

printf("waiting until empty...");

fflush(stdout);

locksem(semid, SN_EMPTY);//这个过程是在等待共享内存资源

printf("done.\n");

printf("enter message: ");

fgets(buffer, BUFFERSIZE, stdin);

unlocksem(semid, SN_FULL);

}

voiddelete()

{

printf("\n master exiting; deleting semaphore %d. \n", deleteSemid);

if((semctl(deleteSemid, 0, IPC_RMID, 0)) == -1)//IPC_RMID删除该信号量

printf("error releasing semaphore. \n");

}