#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <sys/sem.h>
#include <unistd.h>
#define SHM_SIZE 1024
// 信号量操作函数
union semun {
int val;
struct semid_ds *buf;
unsigned short *array;
struct seminfo *__buf;
};
void semaphore_p(int semid) {
struct sembuf sem_op;
sem_op.sem_num = 0;
sem_op.sem_op = -1;
sem_op.sem_flg = 0;
semop(semid, &sem_op, 1);
}
void semaphore_v(int semid) {
struct sembuf sem_op;
sem_op.sem_num = 0;
sem_op.sem_op = 1;
sem_op.sem_flg = 0;
semop(semid, &sem_op, 1);
}
int main() {
key_t key;
int shmid, semid;
char *shmaddr;
// 创建共享内存的键值
if ((key = ftok(".", 'a')) == -1) {
perror("ftok");
exit(1);
}
// 创建共享内存
if ((shmid = shmget(key, SHM_SIZE, IPC_CREAT | 0666)) == -1) {
perror("shmget");
exit(1);
}
// 共享内存映射到进程地址空间
if ((shmaddr = (char *)shmat(shmid, 0, 0)) == (void *)-1) {
perror("shmat");
exit(1);
}
// 创建信号量
if ((semid = semget(key, 1, IPC_CREAT | 0666)) == -1) {
perror("semget");
exit(1);
}
// 初始化信号量值为0
union semun sem_union;
sem_union.val = 0;
if (semctl(semid, 0, SETVAL, sem_union) == -1) {
perror("semctl");
exit(1);
}
// 创建子进程
pid_t pid = fork();
if (pid == -1) {
perror("fork");
exit(1);
} else if (pid == 0) { // 子进程,负责写
char message[] = "Hello, shared memory!";
strcpy(shmaddr, message);
printf("子进程写入: %s\n", message);
// 写入完成,释放信号量
semaphore_v(semid);
} else { // 父进程,负责读
// 等待信号量
semaphore_p(semid);
printf("父进程读取: %s\n", shmaddr);
// 分离共享内存
if (shmdt(shmaddr) == -1) {
perror("shmdt");
exit(1);
}
// 删除共享内存
if (shmctl(shmid, IPC_RMID, 0) == -1) {
perror("shmctl");
exit(1);
}
// 删除信号量
if (semctl(semid, 0, IPC_RMID, 0) == -1) {
perror("semctl");
exit(1);
}
}
return 0;
}
关键步骤说明:
- 生成共享内存键值:使用
ftok
函数生成一个唯一的键值,用于创建共享内存和信号量。
- 创建共享内存:调用
shmget
函数创建共享内存段,并获取其标识符shmid
。
- 映射共享内存:使用
shmat
函数将共享内存段映射到进程的地址空间,返回共享内存的地址shmaddr
。
- 创建信号量:通过
semget
函数创建一个信号量,用于同步读写操作,获取信号量标识符semid
。
- 初始化信号量:利用
semctl
函数将信号量初始值设为0。
- 创建子进程:使用
fork
函数创建子进程,子进程负责写入共享内存,父进程负责读取共享内存。
- 子进程写操作:子进程将字符串写入共享内存后,调用
semaphore_v
函数释放信号量,表示数据已写入。
- 父进程读操作:父进程调用
semaphore_p
函数等待信号量,当信号量可用时,从共享内存读取数据并打印,之后分离并删除共享内存以及信号量。