进程通信之共享内存

6） 其他说明

例子：

shm_common.h：（公共头文件）

#ifndef SHM_COMMON_H
#define SHM_COMMON_H

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <errno.h>

// 定义用于生成key的文件路径
#define KEY_PATHNAME "shm_key_file"

// 定义用于生成key的子序号
#define KEY_PROJ_ID 'A'

// 定义共享内存的大小
#define SHM_SIZE 1024

#endif // SHM_COMMON_H

writer.c

#include "shm_common.h"

int main() {
    // 1. 使用ftok生成一个唯一的key
    key_t key = ftok(KEY_PATHNAME, KEY_PROJ_ID);
    if (key == -1) {
        perror("ftok");
        exit(EXIT_FAILURE);
    }
    printf("Writer: Generated key = %d\n", key);

    // 2. 使用shmget创建共享内存
    // IPC_CREAT: 如果共享内存不存在，则创建它
    // 0666: 设置权限为所有用户可读可写
    int shmid = shmget(key, SHM_SIZE, IPC_CREAT | 0666);
    if (shmid == -1) {
        perror("shmget");
        exit(EXIT_FAILURE);
    }
    printf("Writer: Shared memory created with ID = %d\n", shmid);

    // 3. 使用shmat将共享内存映射到进程的地址空间
    // shmaddr = NULL: 让内核选择一个合适的地址
    // shmflg = 0: 默认模式（可读可写）
    void *shm_ptr = shmat(shmid, NULL, 0);
    if (shm_ptr == (void *)-1) {
        perror("shmat");
        exit(EXIT_FAILURE);
    }
    printf("Writer: Shared memory attached at address %p\n", shm_ptr);

    // 4. 使用共享内存进行数据交换
    printf("Enter a message to write to shared memory: ");
    // 从标准输入读取一行文本并写入共享内存
    if (fgets((char *)shm_ptr, SHM_SIZE, stdin) == NULL) {
        fprintf(stderr, "Failed to read input.\n");
    } else {
        printf("Writer: Message written successfully.\n");
    }

    // 5. 使用shmdt断开与共享内存的映射
    if (shmdt(shm_ptr) == -1) {
        perror("shmdt");
        exit(EXIT_FAILURE);
    }
    printf("Writer: Detached from shared memory.\n");

    printf("Writer finished. The reader can now access the data.\n");

    return 0;
}

reader.c

#include "shm_common.h"

int main() {
    // 1. 使用与writer相同的参数调用ftok，以获取相同的key
    key_t key = ftok(KEY_PATHNAME, KEY_PROJ_ID);
    if (key == -1) {
        perror("ftok");
        exit(EXIT_FAILURE);
    }
    printf("Reader: Generated key = %d\n", key);

    // 2. 使用shmget获取已存在的共享内存
    // 注意：这里没有使用 IPC_CREAT 标志，因为我们期望共享内存已经被writer创建好了
    int shmid = shmget(key, SHM_SIZE, 0666);
    if (shmid == -1) {
        perror("shmget");
        exit(EXIT_FAILURE);
    }
    printf("Reader: Accessed shared memory with ID = %d\n", shmid);

    // 3. 使用shmat将共享内存映射到进程的地址空间
    void *shm_ptr = shmat(shmid, NULL, 0);
    if (shm_ptr == (void *)-1) {
        perror("shmat");
        exit(EXIT_FAILURE);
    }
    printf("Reader: Shared memory attached at address %p\n", shm_ptr);

    // 4. 读取并打印共享内存中的数据
    printf("Reader: The message from shared memory is: \"%s\"\n", (char *)shm_ptr);

    // 5. 使用shmdt断开与共享内存的映射
    if (shmdt(shm_ptr) == -1) {
        perror("shmdt");
        exit(EXIT_FAILURE);
    }
    printf("Reader: Detached from shared memory.\n");

    // 6. 使用shmctl回收共享内存
    // IPC_RMID: 标记该共享内存段以便在所有进程都分离后删除它
    if (shmctl(shmid, IPC_RMID, NULL) == -1) {
        perror("shmctl");
        exit(EXIT_FAILURE);
    }
    printf("Reader: Shared memory segment marked for deletion.\n");

    return 0;
}

首先在该路径下创建一个空文件，便于实现ftok功能

touch shm_key_file

编译并运行