Malloc &初始化-宏

Question

是否可以重写以下代码，使其符合 ISO C 标准？ 以下宏正在执行 malloc & init 指定类型和值。

当前代码适用于 gcc 编译器 (u这是一个 gcc 扩展），但它不是标准的。如果我使用 -pedantic，我会收到警告。

#ifdef __GNUC__

#define NM_CVPTR(type, value) \
    ({ \
        type * var = NULL; \
        var = nm_malloc(sizeof(*var)); \
        *var = value; \
        (const void*) var; \
    }) \

#define NM_VPTR(type, value) \
    ({ \
        type * var = NULL; \
        var = nm_malloc(sizeof(*var)); \
        *var = value; \
        (void*) var; \
    }) \

#define NM_PTR(type, value) \
    ({ \
        type * var = NULL; \
        var = nm_malloc(sizeof(*var)); \
        *var = value; \
        (type *) var; \
    }) \

#endif

Answer 1

这可以使用 逗号运算符 来完成，但在标准 C 中你将无法将变量声明为表达式的一部分，因此您必须将 var 的替换名称传递给宏：

// C - comma operator but not able to declare the storage during the
// expression.
#define NM_PTR(type, var, value) \
    (var = nm_malloc(sizeof(*var)), \
    *var = value, \
    (type * var))

Answer 2

您可以使用memcpy分配一个值，然后返回指针。下面使用两个不同的版本，具体取决于您的初始值是否是原始类型（整数、浮点、指针...）或者是否是struct。值版本使用复合文字(type){ (value) }，因此仅在C99中有效。

#include <stdlib.h>
#include <string.h>

static inline
void* memcpy_safe(void* target, void const* source, size_t n) {
  if (target) memcpy(target, source, n);
  return target;
}

#define NM_PTR_RVALUE(type, rvalue)                                     \
  ((type*)memcpy_safe(malloc(sizeof(type)), &(type){ (rvalue) }, sizeof(type)))

#define NM_PTR_LVALUE(type, lvalue)                                     \
  ((type*)memcpy_safe(malloc(sizeof(type)), &(lvalue), sizeof(type)))

typedef struct {
  int a;
} hoi;

hoi H7 = {.a = 7};

int main() {
  int* a = NM_PTR_RVALUE(int, 7);
  hoi* b = NM_PTR_LVALUE(hoi, H7);
}

（此处添加了使用内联函数的 NULL 检查，尽管最初并未要求这样做。）

顺便说一句，在 C++ 中，与 C 不同，赋值运算符 = 返回左值，因此对于 C++，您可能可以用它玩游戏。