c-realloc()在重复调用后失败



我正在C中实现一些多项式算法。我使用动态结构来存储整数因子和多项式次数。除了其他函数外,我还需要运算p[X]*X,所以我试图实现某种右移。

但在几次转换后,realloc()使我的程序崩溃。在这个例子中,它是在第三次调用时,但如果我尝试切换2和4次,它会在第二次调用后崩溃。

/* Simple polynom type with variable length and degree n-1. */
typedef struct {
    int n;
    int *factors;
} polynom_t;

polynom_t *poly_init(int n) {
    polynom_t *p_new = malloc(sizeof(polynom_t));
    p_new->n = n;
    p_new->factors = calloc(n, sizeof(int));
    return p_new;
}
void poly_clear(polynom_t *p) {
    free(p->factors);
    free(p);
    p = NULL;
}

void poly_set(polynom_t *p, int a[], int len){
    memcpy(p->factors, a, sizeof(int)*p->n);
}

void _poly_rsz(polynom_t *p, int n) {
    if (n != p->n) {
        p->n = n;
        // This realloc() seems to fail
        p->factors = realloc(p->factors, sizeof(int) * n);
    }
}

void _poly_lsr(polynom_t *p, int i) {
    _poly_rsz(p, p->n + i);
    memmove(p->factors + i, p->factors, sizeof(int)*(p->n));
    memset(p->factors, 0, sizeof(int)*i);
}

int main(int argc, char **argv) {
    polynom_t *p2 = poly_init(11);
    int a2[11] = {1, 2, 0, 2, 2, 1, 0, 2, 1, 2, 0};
    poly_set(p2, a2, 11);
    _poly_lsr(p2, 1);  // works as expected
    _poly_lsr(p2, 1);  // works as expected
    _poly_lsr(p2, 1);  // crash
    poly_clear(p2);
    return 0;
}

这里的代码块有问题:

void _poly_lsr(polynom_t *p, int i) {
    _poly_rsz(p, p->n + i);
    memmove(p->factors + i, p->factors, sizeof(int)*(p->n));  // the problem is here!
    memset(p->factors, 0, sizeof(int)*i);
}

当你调整多项式的大小时,你会重置它的计数,这意味着,当你加1时,你的多项式数组的边界溢出了1。要修复,只需从memmove计数中减去i

memmove(p->factors + i, p->factors, sizeof(int)*(p->n - i));

相关内容

  • 没有找到相关文章

最新更新