进程间的协作一直是做产品的永恒需求和必需技术,但是却是一片技术灰色地带,这里我们基于哈工大的操作系统实验,用实际代码讲解如何完成内存共享,和信号量协作
先给出一套完整可用的样例程序:
生产者producer.c:
#include
#include
#include
#include
#include
#include
#include
#include
#include
#define PRODUCT 1000
#define KEYS 0x345379
#define SIZE 4096
#define SHM_EXIST_ERROR -1int main()
{
int shmid,*myArray,tmp_counter,id_counter =0;
key_t mem_key;
sem_t *shop_sig,*consumer_sig;
int * tests;
mem_key=KEYS;
shmid = shmget(mem_key,SIZE,0666);
if(shmid != SHM_EXIST_ERROR)
{
tests = (int*)shmat(shmid,NULL,0);
if ( tests!= (void *)-1)
{
shmdt(tests);
shmctl(shmid,IPC_RMID,0) ;
}
}
shmid=shmget(mem_key,SIZE,IPC_CREAT|0666|IPC_EXCL);
if(shmid==SHM_EXIST_ERROR)
{printf("ERROR: shmget");return 0;}
myArray=(int *)shmat(shmid,NULL,0);
shop_sig = sem_open("shop_signal",O_CREAT,0644,1);
consumer_sig = sem_open("consumer_signal",O_CREAT,0644,0);
for(tmp_counter=1;tmp_counter<11;tmp_counter++)
{
myArray[tmp_counter]=-1;
}
myArray[0]=0;
while(id_counter++
{
sem_wait(shop_sig);
if(myArray[0]<10)
{
myArray[myArray[0]+1]=id_counter;
myArray[0]++;
}
sem_post(consumer_sig);
}
myArray[0]=-1;
sem_post(consumer_sig);
printf("shop is quittingn");
sem_close(shop_sig);sem_close(consumer_sig);
return 0;
}
消费者进程 consumer.c
#include
#include
#include
#include
#include
#include
#include
#include
#include
#define PRODUCT 1000
#define KEYS 0x345379
#define SIZE 4096
#define SHM_EXIST_ERROR -1int main()
{
int shmid,*myArray,tmp_counter;
key_t mem_key;
sem_t *shop_sig,*consumer_sig;
mem_key=KEYS;
shmid=shmget(mem_key,SIZE,0666);
if(shmid == SHM_EXIST_ERROR)
{
printf("ERROR: shmget failuren");return 0;
}
myArray=(int *)shmat(shmid,NULL,0);
shop_sig = sem_open("shop_signal",O_CREAT,0644,1);
consumer_sig = sem_open("consumer_signal",O_CREAT,0644,0);
tmp_counter=0;
while(myArray[0]!=-1)
{
sem_wait(consumer_sig);
if(myArray[0]>0)
{
printf("con: %dn",myArray[myArray[0]]);
myArray[0]--;
}
sem_post(shop_sig);
}
printf("consumer is quittingn");
sem_close(shop_sig);sem_close(consumer_sig);
sem_unlink("shop_signal");sem_unlink("consumer_signal");
shmdt(myArray);
shmctl(shmid,IPC_RMID,NULL);
return 0;
}
控制台编译语法:
gcc -o producer producer.c -lpthread
gcc -o consumer consumer.c -lpthread
控制台运行语法:
sudo ./producer &
sudo ./consumer
因为涉及高权限的信号和内存操作,所以需要superUser权限来do (这就是sudo)
控制台&符号会将程序置于后台运行,这样就可以运行两个程序了,而且似乎可以递归地运行多个(不同机器表现略不同)。
注:样例代码请同学们严重修改后再使用。
想学技术的小伙伴可以继续看:
我们现在开始讲技术,上述涉及两种技术:1,信号量 2,共享内存 我们一一道来
一,信号量
信号量是:请见百度百科
信号量的抽象用法:请见百度百科
linux信号量具体实现:
struct _sem {
__uint32_t _magic;
struct _usem _kern;
};
typedef struct _sem sem_t;
semaphore 单词是打信号和打旗语的意思sem_t * sem_ open(const char * name , int flag, mode_t authority_mode, int init_value );
int sem_wait(sem_t * sem);
int sem_post(sem_t * sem);
int sem_close(sem_t* sem);
int sem_unlink(const char * sem_name);sem_open返回一个有名信号量的系统分配指针,参数字符串name 是名字,
flag是创建方式【O_CREAT:没有则创建,有则返回; O_EXCL:没有就报错;仅给权限值:0666、 有就返回,没有就报错】,
mode是权限模式(四个用户态的权限,一般用0666最高或0644中等),value是信号的初始值,仅在创建新信号量时用于赋值。
sem_wait 等待指定信号量大于0,这时信号量减一。
sem_opst 指定信号量值加一。
sem_close 保存指定信号量状态到系统文件,断开信号量连接,不删除信号量,可以再次open不创建新的,并直接读取原有信号值。
sem_unlink 从系统删除指定名字信号量的存储文件,清空值和连接,下次使用同名信号量时open函数将重建文件并初始化值。
二,共享内存
key_t 是一个宏定义,原型是signed_Int32
size_t 是一个宏定义 ,原型是 自由机器整形Integer 就是…整形key_t ftok( const char * fname, int id );
int shmget(key_t key,size_t size,int authority_mode); shared memory get
void * shmat(int shmid,const void *shmaddr , int location); shared memory allocate
shmdt(void * memory); shared memory delete
shmctl(int shmid , int cmd , struct shmid_ds *buf); shared memory change table link
ftok 根据你给出的存在的文件路径找到文件的INODE,取出节点编号加上参数2,返回给你。一般较少使用这个函数
注意:你可以自己定义key ,没有任何问题,定义的key 可以任意赋值,建议大于10000,不要碰受保护的共享内存。
shmget 根据你给出的key 和大小新建或返回内存索引号码。其中参数3 mode 【IPC_CREAT:没有则创建,有则返回 ; IPC_EXCL:没有就报错】。注意:UBUNTU-linux是带有严格权限的用户系统,所以ubuntu下用get时,参数三要加上权限才能允许执行,例如shmget(mykey,4096,IPC_CREAT0666);
注意:用shmget(mykey,4096,06xx)形式的语句打开一块确定存在的内存时,权限标志要和申请时一致,或者权限更小,否则报错。
shmat 根据内存索引号返回可用内存地址指针(强转即可) ,shmaddr是你设置的物理地址,设置为null让系统自动分配,location是返回指针相对于内存0的偏移值,一般给0
shmdt 保存指定内存数据到系统文件,断开内存连接,不释放内存,可以再次open不创建新的,并直接读取上次的内存数据。
shmctl 改变内存状态,常用于释放共享内存,参数2是操作类型
【IPC_STAT:得到共享内存的状态,把共享内存的shmid_ds结构复制到buf中
IPC_SET:改变共享内存的状态,把buf所指的shmid_ds结构中的uid、gid、mode复制到共享内存的shmid_ds结构内
IPC_RMID 从系统释放指定索引ID的内存,清空备份文件和连接,下次使用同KEY内存时将重建共享文件和内存并初始化内存。】
参数3给NULL;
从shmat得到指针后,可以像数组一样使用,例如:
int * mymem =(int *) shamt(shmid,NULL,0);
mymem[i]=i;
struct Node * mynode = (struct Node * ) shmat(shmid,NULL,0);
mynode->data = 1;讲技术,说人话
Aurora 极光城