c - 位打包和动态分配的问题

我在这段代码上遇到了麻烦。

我正在建立一个图像编码器。基本上，我使用图像中的值构建了一个编码数组。该数组称为“代码”，并存储二进制值将是什么的char *表示形式。

此部分读取每个像素的灰度值，在“代码”数组中查找其值，并打包一个二进制值字节（tempString）。读取8个值后，tempString
被添加到已编码的无符号字节数组（encodedString）的末尾。

该程序将运行直到numBytes约为27,000个字节，然后出现段错误。

我知道它的作用很远，但是我希望它在分配内存方面存在明显问题。

    unsigned char* encodedString = malloc(1);
    unsigned char* tempString;
    encodedString[0] = '\0';

    unsigned char packedString = 0;
    int one = 1;
    int zero = 0;
    int width = image->width;
    int height = image->height;
    int row, col, count=0, numBytes=0; //numBytes is the number of already encoded bytes
    for(row = 0; row<height; row++)
    for(col = 0; col<width; col++)
    {
            int value = image->pixel[row][col];    //Gets the pixel value(0-255)
            char* code = codes[value];             //Gets the compression code for the color

            int length = strlen(code);

            for(index=0;index<length;index++)
            {
                    //This loop goes through every character in the code 'string'
                    if(code[index] == '1')
                            packedString = packedString | one;
                    else
                            packedString = packedString | zero;

                    count++;
                    if(count == 8)  //If 8 consecutive values have been read, add to the end of the encoded string
                    {
                            tempString = realloc(encodedString, (strlen(encodedString)+2));
                            if(tempString == NULL)
                                    return NULL;

                            encodedString = tempString;

                            //Add newly formed binary byte to the end of the already encoded string
                            encodedString[numBytes] = packedString;
                            //Add terminating character to very end
                            encodedString[numBytes+1] = '\0';

                            count=0; //reset count
                            numBytes++;
                    }
                    else
                         packedString = packedString << 1;
            }

            *length_of_encoded_string += strlen(codes[value]);

    }

最佳答案

不要在二进制字符串上调用strlen(str)！请改用numBytes。 strlen将返回
它在您的字符串中找到的第一个零的索引。然后，realloc将停止增加字符串的大小，从而在您使用numBytes访问它时导致段错误。