Question

我正在开发一个基于POSIX的程序，它在两个或多个进程之间共享一个内存对象，即一个服务器 - 客户端程序。

我在程序的服务器端遇到了一些问题。我知道如何使用mmap()映射内存以及如何在两个不同的进程/程序之间使用Object。但是，在向共享内存对象添加整数数组时遇到问题。

我可以使用sprintf()将内容打印到共享内存对象，然后通过简单地说ptr += strlen(whatstringiamworkingwith);

来增加（重新分配）共享内存对象的空间

但是，当我在C中包含一个整数数组时，我不知道如何重新为对象分配内存。这是我的程序：

#include <stdio.h>
#include <stdlib.h>
#include <sys/shm.h>
#include <sys/stat.h>
#include <sys/mman.h>
#include <fcntl.h>
#include <string.h>
#include <unistd.h>

int *calcCollatz(int n) {
    int solution;

    int *collatzSeq = malloc(sizeof(int));
    int j = 0;

    while (n != 1) {
        if (n % 2 == 0) {
            solution = n / 2;
            n = solution;
            collatzSeq[j] = n;
            j++;
            return collatzSeq;
        } else {
            solution = (3 * n) + 1;
            n = solution;
            collatzSeq[j] = n;
            j++;
            return collatzSeq;
        }
    }
}

int main(int argc, char *argv[])
{
    const int SIZE = 4096;          //size in bytes of Shared Memory Object
    const char *sharedObj = "Shm";  //name of the Shared memory Object
    int shm_fd;                     //Shared memory file descriptor
    void *ptrShm; //Pointer to shared memory object

    shm_fd = shm_open(sharedObj, O_CREAT | O_RDWR, 0666); //Create Shared Memory Object
    ftruncate(shm_fd, SIZE);  //configure the size of the shared memory object
                              //Map the shared memory object in the space of the process
    ptrShm = mmap(0, SIZE, PROT_WRITE, MAP_SHARED, shm_fd, 0);

    if (ptrShm == MAP_FAILED) {
        printf("Map Failed\n");
        return -1;
    }

    sprintf(ptrShm, argv[1]);
    ptrShm += strlen(argv[1]);

    sprintf(ptrShm, calcCollatz(atoi(argv[1])));
    ptrShm += sizeof(calcCollatz(atoi(argv[1])));


    printf("Writing the sequence to a shared memory object!\n");
    return 0;
}

我是否在正确的轨道上？

Answer 1

您的代码中存在一些错误和问题。

ptr += strlen(whatstringiamworkingwith);正在推进指针，而不是重新分配记忆。
如果我执行您的程序，执行sprintf(ptrShm, argv[1]);会很危险像这样：./yourprogram %s那么你将有%s未定义的行为将出现说明符，但指向char*以满足说明符的指针是失踪。永远不要将用户填写的变量作为printf＆amp;的格式传递公司所以这样做：
```
sprintf(ptrShm, "%s", argv[1]);
```
Sames用于此行：sprintf(ptrShm, calcCollatz(atoi(argv[1])));

你不应该检查是否有足够的命令行参数在程序开头添加：
```
if(argc != 2)
{
    fprintf(stderr, "usage: %s data\n", argv[0]);
    return 1;
}
```
错误消息应打印到stderr，而不是执行 printf("Map Failed\n");你应该这样做：
```
fprintf(stderr, "Map failed\n");
```
但这不是很有用，你还应该打印errno的值像这样：
```
fprintf(stderr, "Map failed: %s\n", strerrno(errno));
// or
perror("Map failed");
```
您没有检查返回值shm_open。
退出代码是带符号的8位值，因此执行return -1相当于 return 255。
如shm_open手册页中所述：


man shm_open

共享内存对象应该由/somename形式的名称标识;也就是说，一个以null结尾的字符串          NAME_MAX（即255个）字符由初始斜杠组成，后跟一个或多个字符，其中没有一个是斜杠。

所以正确的名称应该是
```
const char *sharedObj = "/Shm";
```
sizeof(calcCollatz(atoi(argv[1])));与sizeof(int*)相同表示指向int的指针的大小。并且因为sizeof被评估编译时，该函数实际上从未执行过。您可能想要按数字调整大小的项目。不仅ptr += size是不正确的呼叫，你得到了信息完全错误。

您的calcCollatz也是错误的，您只为一个单独分配空间 int，为什么要为单个int动态分配空间呢？你也正在做return collatzSeq，再次，为什么甚至打扰为单个int动态分配空间？我认为这就是你的意思寻找：
```
int *calcCollatz(int n, size_t *len) {
    if(len == NULL)
        return NULL;

    int *collatzSeq = NULL, *tmp = NULL;
    *len = 0;

    while (n != 1) {
        // better than n % 2 == 0
        if (n & 1 == 0)
            n /= 2;
        else
            n = 3 * n + 1;

        tmp = realloc(collatzSeq, (*len + 1) * sizeof *collatzSeq);
        if(tmp == NULL)
        {
            free(collatzSeq);
            return NULL;
        }
        collatzSeq = tmp;

        collatzSeq[(*len)++] = n;
    }

    return collatzSeq;
}
```

我真的不明白为什么要调整mmap内存的大小。通常你马上知道你需要多少内存，以及是否需要动态存储生成数组，然后计算你需要的内存量，然后创建共享内存，而不是相反：

size_t base = 4096;
size_t arrlen;

int *arr = calcCollatz(atoi(argv[1]), &arrlen);
if(arr == NULL)
{
    fprintf(stderr, "failed to do calculation\n");
    return 1;
}

size_t size = base + arrlen;

shm_fd = shm_open(sharedObj, O_CREAT | O_RDWR, 0666);
if(shm_fd == -1)
{
    fprintf(stderr, "Failed to open shared memory: %s\n", strerror(errno));
    return 1;
}
ftruncate(shm_fd, size);

ptrShm = mmap(0, size, PROT_WRITE, MAP_SHARED, shm_fd, 0);

if (ptrShm == MAP_FAILED) {
    fprintf(stderr, "Map Failed: %s\n", strerror(errno));
    return 1;
}

printf("Writing the sequence to a shared memory object!\n");
sprintf(ptrShm, "%s", argv[1]);

// note that void* arithmetic is a GNU extension
// otherwise you have to cast to (char*) before doing the
// the arithmetic
memcpy(ptrShm + base, arr, arrlen * sizeof *arr);

free(arr);
close(shm_fd);
return 0;

但是如果你以后坚持调整mmap内存的大小，那么就可以使用了 mremap，但请注意，这只适用于GNU / Linux和您在加入#define _GNU_SOURCE之前必须添加sys/mman.h：

size_t size = 4096;

// create shared memory + mmap
...

// exapanding mmap memory
size_t arrlen;

int *arr = calcCollatz(atoi(argv[1]), &arrlen);
if(arr == NULL)
{
    fprintf(stderr, "failed to do calculation\n");
    close(shm_fd);
    return 1;
}

size_t newsize = size + arrlen * sizeof *arr;

void *newptrShm = mremap(ptrShm, size, newsize, MREMAP_MAYMOVE);

if(newptrShm == MAP_FAILED)
{
    fprintf(stderr, "Failed to expand memory: %s\n", strerror(errno));
    free(arr);
    close(shm_fd);
    return 1;
}

ptrShm = newptrShm;

// resize shared file size
ftruncate(shm_fd, newsize);

memcpy(ptrShm + base, arr, arrlen * sizeof *arr);

free(arr);

...

另见：Fast resize of a mmap file和 How to portably extend a file accessed using mmap()

将一维数组添加到C

1 个答案: