C实现扩大共享内存大小不丢失数据的demo

以下是一个简单的 C 语言程序，用于演示如何在扩大共享内存大小时不丢失数据：

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

#define SHM_SIZE 1024  // 初始共享内存大小为 1024 字节

int main()
{
    int shmid;
    char *shmaddr;
    int size = SHM_SIZE;

    // 创建共享内存对象
    shmid = shmget(IPC_PRIVATE, size, IPC_CREAT | 0666);
    if (shmid == -1) {
        perror("shmget");
        exit(1);
    }

    // 将共享内存对象附加到进程的地址空间中
    shmaddr = shmat(shmid, NULL, 0);
    if (shmaddr == (char *)-1) {
        perror("shmat");
        exit(1);
    }

    // 输出共享内存的大小和地址
    printf("Shared memory size: %d bytes\n", size);
    printf("Shared memory address: %p\n", shmaddr);

    // 写入一些数据到共享内存中
    char *data = "Hello, world!";
    strncpy(shmaddr, data, strlen(data));
    printf("Write data to shared memory: %s\n", shmaddr);

    // 等待一段时间，然后重新调整共享内存的大小
    sleep(5);
    size *= 2;
    shmid = shmget(IPC_PRIVATE, size, IPC_CREAT | 0666);
    if (shmid == -1) {
        perror("shmget");
        exit(1);
    }
    char *new_shmaddr = shmat(shmid, NULL, 0);
    if (new_shmaddr == (char *)-1) {
        perror("shmat");
        exit(1);
    }

    // 复制之前的数据到新的共享内存中
    strncpy(new_shmaddr, shmaddr, strlen(shmaddr));

    // 输出重新调整后的共享内存的大小和地址
    printf("Shared memory size: %d bytes\n", size);
    printf("Shared memory address: %p\n", new_shmaddr);
    printf("Read data from shared memory: %s\n", new_shmaddr);

    // 将共享内存对象从进程的地址空间中分离
    if (shmdt(shmaddr) == -1) {
        perror("shmdt");
        exit(1);
    }

    // 附加新的共享内存到进程的地址空间中
    shmaddr = new_shmaddr;

    // 写入一些新的数据到共享内存中
    char *new_data = "Welcome to ChitChat!";
    strncpy(shmaddr + strlen(data), new_data, strlen(new_data));
    printf("Write new data to shared memory: %s\n", shmaddr);

    // 将共享内存对象从进程的地址空间中分离
    if (shmdt(shmaddr) == -1) {
        perror("shmdt");
        exit(1);
    }

    // 删除共享内存对象
    if (shmctl(shmid, IPC_RMID, NULL) == -1) {
        perror("shmctl");
        exit(1);
    }

    return 0;
}

这个程序首先创建一个大小为 1024 字节的共享内存对象，然后向共享内存中写入一些数据，并输出数据。接着，程序等待 5 秒钟后将共享内存的大小扩大一倍，重新创建一个新的共享内存对象，并复制之前的数据到新的共享内存中，并输出新的共享内存的大小和地址以及数据。然后，程序将新的共享内存对象附加到进程的地址空间中，并向共享内存中写入一些新的数据，并输出数据。最后，程序将共享内存对象从进程的地址空间中分离并删除共享内存对象。可以看到，在扩大共享内存大小时，程序通过复制之前的数据到新的共享内存中，确保了数据不会丢失。