我正在使用Winsocket编写一个小型服务器/客户端文件传输器,它基本上可以正常工作,除了我收到文件并将其写入我的硬盘后我不能再接收套接字上的消息。
传输代码就像这样:
long size = GetFileSize(hFile, NULL);
TransmitFile(_socket, hFile, size, 0, NULL, NULL, TF_DISCONNECT);
ok = ::recv(_socket, cantwortfilename, 100, 0); // getting a confirmation (1)
cantwortfilename[ok] = '\0';
cout << cantwortfilename << endl;
char test[] = "ok!";
::send(_socket, test, strlen(test), 0); // to test if server receives again (2)
我用0而不是大小尝试了相同的结果。
现在到服务器端的接收:
HANDLE hFile = CreateFile(filepathlong, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
while (1)
{
ReadFile((HANDLE)_socket, &buffer, MAX_PATH, &dwbr, &AsyncInfo);
GetOverlappedResult((HANDLE)_socket, &AsyncInfo, &dwbr, TRUE);
if (dwbr == 0) break;
WriteFile(hFile, buffer, dwbr, &dwbr, NULL);
}
char test[] = "alles ok!";
::send(_socket, test, strlen(test), 0); // sending the confirmation (1)
CloseHandle(hFile);
i = ::recv(_socket, test, 10, 0); // receiving the test (2)
test[i] = '\0';
cout << "empfangen:" << test << endl;
据我所知,文件的传输工作正常(试过rar jpg .h)和
::send(_socket, test, strlen(test), 0); // sending the confirmation (1)
也很好。
但之后的接收给了我什么?或者空的东西? 因为recv没有阻止该程序,我会猜测是空的。 但是&#34;我&#34;我放弃时会是0。
检查我是否在while(1)循环中发生了某种错误,我尝试了另一种接收文件的方法。
第二次尝试:
int r;
ofstream file(filepath, std::ios::out | std::ios::binary | std::ios::trunc);
char *memblock;
int size = 7766; // was the size of the file i was testing with
memblock = new char[size];
memset(memblock, 0, size);
if (file.is_open()){
//while memory blocks are still being received
while ((r = recv(_socket, memblock, 256, 0)) != 0)
{
//if there's a socket error, abort
if (r == SOCKET_ERROR)
{
cout << "error" << endl;
}
//write client's file blocks to file on server
file.write(memblock, r);
}
delete[] memblock;
//finished sending memory blocks; file is completely transferred.
file.close();
}
之后,再次发送一个具有相同结果的recv。 该文件有效但接收再次让我感到空虚。
所以有人能告诉我为什么以及如何解决这个问题?如果可能,尽可能改变最少?
谢谢, 马丁
编辑:现在正在为我工作的代码:
char csize[256];
rc = recv(_socket, csize, 256, 0);
csize[rc] = '\0';
int size = atoi(csize);
int bytes_read = 0, len = 0;
int r;
ofstream file(filepath, std::ios::out | std::ios::binary | std::ios::trunc);
char *memblock;
memblock = new char[size];
memset(memblock, 0, size);
if (file.is_open()){
while (bytes_read < size){
len = recv(_socket, memblock + bytes_read, size - bytes_read, 0);
bytes_read += len;
}
file.write(memblock, size);
delete[] memblock;
file.close();
}
答案 0 :(得分:0)
您的发射机正在使用TransmitFile()
标志呼叫TF_DSCONNECT
。这将在文件发送完毕后关闭套接字。 TransmitFile()
退出后,发送器无法再发送任何数据(我不确定TransmitFile()
是否只接近半双工接近发送方向,但接收方向保持打开状态,或者它在发送和接收方向上进行全双工关闭 - 我会谨慎行事并假设后者。
根据您展示的内容,您需要使用shutdown()
和closesocket()
代替TF_DISCONNECT
,这样您就可以更好地控制断开连接实际发生的时间和方式。
此外,TransmitFile()
不发送文件大小,只发送文件数据,因此您的接收方无法知道文件数据何时实际完成接收,除非它等待超时(这不是可靠的解决方案)。您需要更改发送器以在发送文件数据之前发送文件大小,然后更改接收器以首先读取文件大小,然后循环直到它收到指定的字节数。