Linux进程通信——共享内存

/*
* 进程间通信之信号量，代码源自《Linux程序设计》
*/
#include <iostream>
#include <string>
#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
// 使用信号量必须包含该文件
#include <sys/shm.h>

using namespace std;

// shmid_ds这个结构体已经在系统中定义了，所以不需要再次定义
struct shmid_ds;

#define TEXT_SZ (2028)
#define BUF_SIZE (1024)

// 用户自定义的一个数据结构，同于进行共享
struct shared_use_st
{
int written;    // 数据是否已经被写入的标志（等于1表示数据已经被写入，等于0表示没有被写）
char text[TEXT_SZ];
};

// 生产者进程
int producer()
{
int running = 1;
// 共享内存的地址
    void* shm = 0;
struct shared_use_st* shared;
// 输入缓冲区
 char buffer[BUF_SIZE + 1] = {0};

// 共享内存对象的id
int shm_id;

// 创建或者打开共享内存
shm_id = shmget((key_t)1234,sizeof(struct shared_use_st),0666|IPC_CREAT);

if(shm_id == -1)
{
return -1;
}

// 把共享内存连接到当前进程的地址空间中
shm = shmat(shm_id,0,0);

if((void*)-1 == shm)
{
return -1;
}

// 使用共享内存，即把共享内存当成一块用户自定义的数据，可以直接进行读写
shared = (struct shared_use_st*)shm;

while (running)
{
// 消费者还没有读取完数据，需要等待
while (shared->written == 1) {
sleep(1);
}

// 用户输入一些数据
cout << "Please input something:";
fgets(buffer,BUF_SIZE,stdin);

// 写到共享内存中
strcpy(shared->text,buffer);
shared->written = 1;
// 如果用户写入的是end，那么表示退出
if(strcmp(buffer,"end") == 0)
{
running = 0;
}
}

// 把共享内存从进程的地址空间中分离
if(shmdt(shm) == -1)
{
return -1;
}
sleep(2);

return 0;
}

int main(int argc,char* argv[])
{
return producer();
}

/*
* 进程间通信之信号量，代码源自《Linux程序设计》
*/
#include <iostream>
#include <string>
#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
// 使用信号量必须包含该文件
#include <sys/shm.h>

using namespace std;

// shmid_ds这个结构体已经在系统中定义了，所以不需要再次定义
struct shmid_ds;

#define TEXT_SZ (2028)
#define BUF_SIZE (1024)

// 用户自定义的一个数据结构，同于进行共享
struct shared_use_st
{
int written;    // 数据是否已经被写入的标志（等于1表示数据已经被写入，等于0表示没有被写）
char text[TEXT_SZ];
};

// 消费者进程
int consumer()
{
// 循环标志
int running = 1;
// 共享内存的地址
void* shm = NULL;
// 用户的数据结构的指针
struct shared_use_st* shared;

// 共享内存的id
int shm_id;

// 创建或者打开共享内存，创建的共享内存的大小是 sizeof(struct shared_use_st)
shm_id = shmget((key_t)1234,sizeof(struct shared_use_st),0666|IPC_CREAT);

if(-1 == shm_id)
{
return -1;
}

// 把共享内存连接到当前进程地址空间上
shm = shmat(shm_id,0,0);
if((void*)-1 == shm)
{
return -1;
}

// 使用共享内存，把共享内存当作一块用户数据
shared = (struct shared_use_st*)shm;

shared->written = 0;

while (running)
{
// 如果数据已经被写入了，那么可以进行读取
if(shared->written > 0)
{
cout << shared->text << endl;
sleep(1);
// 重置写入标志
shared->written = 0;
// 当被写入end的时候表示退出
if(0 == strcmp(shared->text,"end"))
{
running = 0;
}
}
else
{
sleep(1);
}
}

// 把共享内存从当前进程的地址空间中分离
if(shmdt(shm) ==1)
{
return -1;
}

// 删除共享内存
if(shmctl(shm_id,IPC_RMID,0) == -1)
{
return -1;
}

return 0;
}

int main(int argc,char* argv[])
{
return consumer();
}