从 base64 C++ 解码和保存图像文件

Decoding and saving image files from base64 C++

我正在尝试用 C++ 编写一个程序,它可以将图像编码为 base64,也可以将 base64 解码为图像。我相信编码器功能工作正常,一些网站可以采用我生成的 base64 代码并将其解码成图像,但出于某种原因,一旦我将 base64 解码为字符串,然后将其写入文件并将其另存为png 它说它不能在图像查看器中打开。

我确认正在写入新文件的字符串与现有文件完全相同(在文本编辑器中打开时),但由于某种原因,新文件无法打开,但现有的可以。我什至尝试过在文本编辑器中创建一个新文件,然后将旧文件中的文本复制到其中,但它仍然无法在图像查看器中打开。

我相信编码函数和 base64 解码函数都可以正常工作。我认为问题出在图像解码功能上。

图像编码功能

string base64_encode_image(const string& path) {
    vector<char> temp;

    std::ifstream infile;
    infile.open(path, ios::binary);     // Open file in binary mode
    if (infile.is_open()) {
        while (!infile.eof()) {
            char c = (char)infile.get();
            temp.push_back(c);
        }
        infile.close();
    }
    else return "File could not be opened";
    string ret(temp.begin(), temp.end() - 1);
    ret = base64_encode((unsigned const char*)ret.c_str(), ret.size());

    return ret;
}

图像解码功能

void base64_decode_image(const string& input) {
    ofstream outfile;
    outfile.open("test.png", ofstream::out);

    string temp = base64_decode(input);

    outfile.write(temp.c_str(), temp.size());
    outfile.close();
    cout << "file saved" << endl;
}

编码函数base64

string base64_encode(unsigned const char* input, unsigned const int len) {
    string ret;
    size_t i = 0;
    unsigned char bytes[3];
    unsigned char sextets[4];

    while (i <= (len - 3)) {
        bytes[0] = *(input++);
        bytes[1] = *(input++);
        bytes[2] = *(input++);

        sextets[0] = (bytes[0] & 0xfc) >> 2;    // Cuts last two bits off of first byte
        sextets[1] = ((bytes[0] & 0x03) << 4) + ((bytes[1] & 0xf0) >> 4);   // Takes last two bits from first byte and adds it to first 4 bits of 2nd byte
        sextets[2] = ((bytes[1] & 0x0f) << 2) + ((bytes[2] & 0xc0) >> 6);   // Takes last 4 bits of 2nd byte and adds it to first 2 bits of third byte
        sextets[3] = bytes[2] & 0x3f;   // takes last 6 bits of third byte

        for (size_t j = 0; j < 4; ++j) {
            ret += base64_chars[sextets[j]];
        }

        i += 3; // increases to go to third byte
    }

    if (i != len) {
        size_t k = 0;
        size_t j = len - i; // Find index of last byte
        while (k < j) {     // Sets first bytes
            bytes[k] = *(input++);
            ++k;
        }

        while (j < 3) {     // Set last bytes to 0x00
            bytes[j] = '[=12=]';
            ++j;
        }

        sextets[0] = (bytes[0] & 0xfc) >> 2;    // Cuts last two bits off of first byte
        sextets[1] = ((bytes[0] & 0x03) << 4) + ((bytes[1] & 0xf0) >> 4);   // Takes last two bits from first byte and adds it to first 4 bits of 2nd byte
        sextets[2] = ((bytes[1] & 0x0f) << 2) + ((bytes[2] & 0xc0) >> 6);   // Takes last 4 bits of 2nd byte and adds it to first 2 bits of third byte
        // No last one is needed, because if there were 4, then (i == len) == true

        for (j = 0; j < (len - i) + 1; ++j) {   // Gets sextets that include data
            ret += base64_chars[sextets[j]];    // Appends them to string
        }

        while ((j++) < 4)   // Appends remaining ='s
            ret += '=';

    }

    return ret;

}

解码函数base64

string base64_decode(const string& input) {
    string ret;
    size_t i = 0;
    unsigned char bytes[3];
    unsigned char sextets[4];

    while (i < input.size() && input[i] != '=') {
        size_t j = i % 4;   // index per sextet
        if (is_base64(input[i])) sextets[j] = input[i++];       // set sextets with characters from string
        else { cerr << "Non base64 string included in input (possibly newline)" << endl; return ""; }
        if (i % 4 == 0) {
            for (j = 0; j < 4; ++j)             // Using j as a seperate index (not the same as it was originally used as, will later be reset)
                sextets[j] = indexof(base64_chars, strlen(base64_chars), sextets[j]);   // Change value to indicies of b64 characters and not ascii characters

            bytes[0] = (sextets[0] << 2) + ((sextets[1] & 0x30) >> 4);  // Similar bitshifting to before
            bytes[1] = ((sextets[1] & 0x0f) << 4) + ((sextets[2] & 0x3c) >> 2);
            bytes[2] = ((sextets[2] & 0x03) << 6) + sextets[3];

            for (j = 0; j < 3; ++j)     // Using j seperately again to iterate through bytes and adding them to full string
                ret += bytes[j];
        }
    }

    if (i % 4 != 0) {
        for (size_t j = 0; j < (i % 4); ++j)
            sextets[j] = indexof(base64_chars, strlen(base64_chars), sextets[j]);

        bytes[0] = (sextets[0] << 2) + ((sextets[1] & 0x30) >> 4);  // Similar bitshifting to before
        bytes[1] = ((sextets[1] & 0x0f) << 4) + ((sextets[2] & 0x3c) >> 2);
        for (size_t j = 0; j < (i % 4) - 1; ++j)
            ret += bytes[j];        // Add final bytes
    }
    return ret;
}

当我尝试打开图像解码功能生成的文件时,提示文件格式不受支持,或者文件已损坏。 我要解码的编码函数生成的 base64 在此 link https://pastebin.com/S5D90Fs8

当您在 base64_decode_image 中打开 outfile 时,您在读取图像时不会像在 base64_encode_image 中那样指定 ofstream::binary 标志。如果没有该标志,您将在文本模式下编写,这可能会改变您正在编写的数据(调整换行符时)。