C语言 环境变量替换方案中的缓冲区溢出问题



每当环境变量的值大于此方法中的键时,我就会得到缓冲区溢出。目标是为令牌动态分配的二维数组的一部分。每当我将作为环境变量的令牌替换为比它长的值时,它都会流入下一个令牌。我尝试添加一个 realloc 来尝试修复它,但它不起作用或导致段错误。

如果有人有任何建议或可以指出我忽略的东西,我将不胜感激,因为我有一种感觉,无论如何当我发现它时,我会踢自己。

方法是:

void envReplace(ENV *evlist, char *Target)
{
    if (Target[0] == '@')
    {
        memmove(Target, Target+1, strlen(Target));
        for(q = 0; q<16; q++)
        {
            if(evlist[q].envVariable!=NULL)
            {
                if(strcmp(Target, evlist[q].envVariable)==0)
                {
                    //this is where I'd add the realloc as realloc(Target, strlen(evlist[q].Value))
                    strcpy(Target, evlist[q].Value);
                    return;
                }
            }
        }
                printf("Variable not foundn");
    }
    else
    {
        printf("A value that didn't start with @ was an argumentn");
        return;
    }
}

数据结构 ENV 为:

typedef struct envStorage
{
    char *envVariable;
    char *Value;
}ENV;

你可以像这样传递指向指针的指针,然后你可以在函数内部调用 realloc(),修改原始指针。

void func (char **to_change) {
    // Changes the char *target in main()
    *to_change = (char *) realloc(*to_change, 20);
    sprintf (*to_change, "Blablablablablan");
}
int main (int argc, char **argv) {
    char *target = (char *) malloc (10);
    func(&target);
    printf(target);
    free(target);
}

在此示例中,func()写入main()函数中的原始指针char *target

代码中发生的情况是,您realloc()并分配给指针的副本。调用 envReplace() 时会复制指针。当此函数返回时,原始指针包含旧的内存地址,该地址不再是有效的分配内存(它被realloc()释放)。

最新更新