1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/shm.h>
#include <sys/sem.h>
#include <sys/ipc.h>
#include <unistd.h>
#include <sys/wait.h>
#define SHM_KEY 9999
#define SEM_KEY 8888
#define SHM_SIZE 256
// 信号量操作
void sem_wait(int semid) {
struct sembuf op = {0, -1, 0}; // P 操作
semop(semid, &op, 1);
}
void sem_signal(int semid) {
struct sembuf op = {0, 1, 0}; // V 操作
semop(semid, &op, 1);
}
int main() {
int shmid, semid;
char *shm_ptr;
// 创建共享内存
shmid = shmget(SHM_KEY, SHM_SIZE, 0666 | IPC_CREAT);
shm_ptr = shmat(shmid, NULL, 0);
// 创建信号量(初始值为 1)
semid = semget(SEM_KEY, 1, 0666 | IPC_CREAT);
semctl(semid, 0, SETVAL, 1);
pid_t pid = fork();
if (pid == 0) {
// 子进程:写入
sem_wait(semid); // 获取锁
strcpy(shm_ptr, "Message from child");
printf("Child: wrote '%s'\n", shm_ptr);
sem_signal(semid); // 释放锁
exit(0);
} else {
// 父进程:读取
wait(NULL); // 等待子进程写入
sem_wait(semid);
printf("Parent: read '%s'\n", shm_ptr);
sem_signal(semid);
// 清理
shmdt(shm_ptr);
shmctl(shmid, IPC_RMID, NULL);
semctl(semid, 0, IPC_RMID);
}
return 0;
}
|