时间戳

首页 > 解决方案 > Postgres 用户定义类型和正确分配内存

c - Postgres 用户定义类型和正确分配内存

问题描述

我正在为 postgres 编写一个扩展，其中包括创建一个新的可变长度基类型，但是我在理解 SET_VARSIZE 的语义时遇到了一些困难。

以下面的示例代码为例，这并不能准确地反映我的用例，但它说明了这一点。

typedef struct another_struct 
{
  char *a;
  char *b;
} another_struct;

typedef struct test_struct
{
  char vl_len_[4];
  another_struct *data;
} test_struct;

1）在为test_struct的新实例分配内存时，大概我可以简单地做以下事情，它会考虑到变长成员vl_len_的大小？

test_struct *t = palloc0(sizeof(struct test_struct));

2) 因为 another_struct 的两个成员都有可变长度，我假设我还需要跟踪为这两个字段分配了多少内存，以便将正确的长度传递给 SET_VARSIZE？

3) 调用 SET_VARSIZE 时是否还需要考虑 another_struct 指针的大小？

我认为对 SET_VARSIZE 的最终调用看起来像这样

SET_VARSIZE(t, sizeof(struct test_struct) + sizeof(struct another_struct) + a_and_b_length);

这接近正确吗？对任何错误表示歉意，我对 C 编程相当陌生。

谢谢

标签： cpostgresqlmemory-management

解决方案

仅使用常规指针是无法做到的。所有内存都必须在 size 成员之后。

typedef struct another_struct 
{
  char a[FLEXIBLE_ARRAY_MEMBER];
  char b[FLEXIBLE_ARRAY_MEMBER];
} another_struct;

typedef struct test_struct
{
  char vl_len_[4];
  another_struct data;
} test_struct;

你现在必须知道数据的实际大小。

test_struct *t = palloc0(VARHDRSZ + size_of_a + size_of_b);
SET_VARSIZE(t, VARHDRSZ + size_of_a + size_of_b );
memcpy(t->data.a, src_a, size_of_a);
memcpy(t->data.b, src_b, size_of_b);

文档中的示例只有一个字段，所以我只是假设这种方式有效。间接可能还有其他问题。来源：https ://www.postgresql.org/docs/current/xfunc-c.html

推荐阅读