代码改变世界

解决ERROR_INVALID_USER_BUFFER

2013-11-16 20:57  DVwei  阅读(2019)  评论(0编辑  收藏  举报

  我用CSocket编写了一个HTTP下载程序,接收数据大概如下:

//......
//use CSocket class
//send request...
//......
int recevieSize(1);
while(recevieSize > 0)
{
    ZeroMemory(buff, BUFFSIZE + 1);
    receviedSize = socket.Receive(buff, BUFFSIZE);
    if(receivedSize > 0)
    {
       //use CFile class
       downloadFile.Write(buff, receviedSize);
       //some handle...
    }
 } //some handle...

  这段代码看上去毫无问题,但是运行过程中,在下载文件时,途中经常会出现ERROR_INVALID_USER_BUFFER错误,根据MSDN上的解释:“The supplied user buffer is not valid for the requested operation.

  于是我尝试加大了缓冲区,但是问题依旧。这让我百思不得其解,上CSDN发帖问,也没人回答。于是我用CInternetSession、CHttpFile重写了这部分。这时候运行起来就毫无问题了,这让我更觉得奇怪了。

  后来根据这个stackoverflow上的这个帖子(http://stackoverflow.com/questions/12643441/win-32-writefile-access-viloation-and-error-1784),再仔细调试一下这两种代码,发现区别就在于两种代码每次写入文件的byets,因为CHttpFile::Read()每次返回固定的bytes,而CSocket::Receive()每次接收的bytes是不确定的,所以使用CSocket::Receive()接收数据每次写入文件的bytes也就不一样,根据帖子上的描述,问题应该就在这里了。

于是把代码改为这样:

//......
//use CSocket class
//send request...
//......
int recevieSize(1);
int readSize;
while(recevieSize > 0)
{
    ZeroMemory(buff, BUFFSIZE + 1);
    //downloadSize是此次任务需要下载的总bytes大小
    readSize = downloadSize> BUFFSIZE ? BUFFSIZE : downloadSize;
    receviedSize = socket.Receive(buff, readSize);
readSize
-= receviedSize; while(readSize && (receviedSize > 0)) { int rs = socket.Receive(buff + receivedSize, readSize); if(rs == 0)//connection has been close break; else if(rs == SOCKET_ERROR) { receviedSize = rs; break; } receivedSize += rs; readSize -= rs; } if(receivedSize > 0) { //use CFile class downloadFile.Write(buff, receviedSize); //some handle... } } //some handle...

  这样就保证了每次写入文件的bytes数,这样运行起来就不会出现ERROR_INVALID_USER_BUFFER了。但是为什么每次写入相同的bytes数就没问题呢?目前我也还没完全搞懂。

readSize