c++用文件流实现文件拷贝

之前用c++作后台的时候碰到一个问题：返回一张图片数据。查了一些资料终于解决了。核心就是文件流怎么读写的问题，已经将问题简化为文件拷贝，代码如下：

#include 
#include 
using namespace std;
int main()
{
    cout << "Content-Type:image/png\r\n\r\n";
    fstream file;
    file.open("1.png", ios::binary|ios::in|ios::ate);   //打开时指针在文件尾
    int length = file.tellg();
    char* imgData = new char[length];
    file.seekg(0);
    file.read(imgData, length);  //二进制只能用这个读
    
    fstream file2;
    file2.open("2.png", ios::binary|ios::out);
    file2.write(imgData, length);   //二进制只能用这个写
    cout << "ok";
    
    return 0;
}

结果：

原来是不存在2.png的

----------------------------------------华丽的分割线---------------------------------------------
事实上，以上代码非常辣鸡，存在以下问题：

• 使用了int, 不支持2G以上的文件
• 使用了new申请内存并且没有释放，可能造成程序崩溃
• 没有封装，不方便使用

现修改如下：

#include 
#include 
#include 
using namespace std;

/* 复制文件
 * @参数 src - 源文件名
 * @参数 dest - 目标文件名，如果目标文件已存在，则覆盖
 * @返回 true | false 代表拷贝是否成功
*/
bool copyFile(string src, string dest) {
    ifstream is(src, ios::binary);
    if (is.fail()) {
        return false;
    }

    ofstream os(dest, ios::binary);
    if (os.fail()) {
        return false;
    }

    is.seekg(0, ios::end);
    long long length = is.tellg();  // C++ 支持的最大索引位置
    is.seekg(0);
    char buf[2048];
    while (length > 0)
    {
        int bufSize = length >= 2048 ? 2048 : length;
        is.read(buf, bufSize);
        os.write(buf, bufSize);
        length -= bufSize;
    }

    is.close();
    os.close();
    return true;
}


int main() {
    if (copyFile("./ubuntu-19.10-live-server-amd64.iso", "./test_cp.iso")) {
        cout << "复制文件成功" << endl;
    }
    else {
        cout << "复制文件失败" << endl;
     }
    return 0;
}

拷贝大文件轻轻松松。