我正在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));