几个问题。
首先,您到处都有不兼容的类型。在行中
my_struct = some_func(my_struct);
my_struct 的类型为struct stu1 **,但some_func 的定义需要struct stu1 * 类型的参数;这两种类型不一样。指向 T 的指针与指向 T 的指针的类型不同。
其次,你的选角体操不会如你所愿。指针类型不会自动兼容;它们的基类型必须兼容,不同的结构类型不兼容,即使它们的布局相同,例如:
struct {int x, int y} foo;
struct {int x, int y} bar;
struct S {int x, int y} blurga;
struct S bletch;
foo、bar 和 bletch 是不同的类型并且不兼容,即使它们的布局相同。 blurga 和 bletch 属于同一类型(结构 S)。如果struct stu2 或struct stu3 的大小与struct stu1 不同,那么您将无法为my_struct2 和my_struct3 分配正确的内存量。
为了清楚起见和您的理智,您应该为每种类型设置不同的分配函数,而不是试图将方形钉子强行插入五边形孔:
struct stu1 **stu1_alloc(size_t count)
{
struct stu1 **arr = malloc(sizeof *arr * count);
if (arr)
{
size_t i;
for (i = 0; i < count; i++)
{
arr[i] = malloc(sizeof *arr[i]);
if (arr[i])
{
// initialize arr[i] as necessary
}
}
}
return arr;
}
struct stu2 **stu2_alloc(size_t count)
{
struct stu2 **arr = malloc(sizeof *arr * count);
if (arr)
{
size_t i;
for (i = 0; i < count; i++)
{
arr[i] = malloc(sizeof *arr[i]);
if (arr[i])
{
// initialize arr[i] as necessary
}
}
}
return arr;
}
struct stu3 **stu3_alloc(size_t count)
{
struct stu3 **arr = malloc(sizeof *arr * count);
if (arr)
{
size_t i;
for (i = 0; i < count; i++)
{
arr[i] = malloc(sizeof *arr[i]);
if (arr[i])
{
// initialize arr[i] as necessary
}
}
}
return arr;
}
int main(void)
{
struct stu1 **my_struct = stu1_alloc(SIZE);
struct stu2 **my_struct2 = stu2_alloc(SIZE2);
struct stu3 **my_struct3 = stu3_alloc(SIZE3);
...
}
是的,三个分配函数之间唯一不同的是类型。但是,如果不同的结构类型有不同的大小或不同的初始化需求,那么这是必要的。
请注意几件事。首先,我没有投射malloc() 的结果。有两个原因。其一,从 C89 及更高版本开始,您不必:malloc() 返回 void * 类型,该类型隐式转换为目标指针类型。两个,更重要的是,如果我忘记包含 stdlib.h 或者在范围内没有 malloc() 的原型,那么放弃该类型将触发“不兼容类型”警告(因为假定未声明的函数返回 @ 987654342@,并且不能将 int 值隐式转换为指针类型)。如果你转换malloc()的返回值,那么你在运行时抑制了警告和风险问题(因为malloc()返回的值将从void *转换为int,然后从int转换为目标指针类型,不保证工作)。
其次,我在对象上使用sizeof,而不是类型。这有两个好处。一,代码更容易阅读。二、如果我改变了对象的类型,我就不必每次调用都回过头来改变malloc()。
如果你真的不想拥有三个独立的分配函数,你可以试试这样的宏魔法:
#define ALLOC_STU(target, size) \
do { \
target = malloc(sizeof *target * size); \
if (target) \
{ \
size_t i; \
for (i = 0; i < size; i++) \
{ \
target[i] = malloc(sizeof *target[i]); \
} \
} \
} while(0)
int main(void)
{
struct stu1 **my_struct;
...
ALLOC_STU(my_struct, SIZE);
...
}
虽然我认为单独的分配函数是更安全的方法。