Форум программистов [Powered by Invision Power Board]

Версия для печати темы
Нажмите сюда для просмотра этой темы в оригинальном формате

Форум программистов > C/C++: Сети > TransmitFile

Автор: Johen 9.6.2005, 14:16

Есть такая функция в Windows Sockets как TransmitFile.

Описание в MSDN http://msdn.microsoft.com/library/default.asp?url=/library/en-us/winsock/winsock/transmitfile_2.asp.

Вопрос такой: допустим сервер передает клиенту файл, с помощью каких средств клиент принимает файл? Какие функции клиент должен использовать, чтобы нормально этот файл принять?

Желателен пример кода. smile

Автор: UncleBob 10.6.2005, 11:06

Я думаю, что это можно выполнить с помощью обычного метода чтения сокета - recv.

Автор: Johen 14.6.2005, 13:46

А поконкретней можно? Если через recv, то тогда зачем нужна вообще функция TransmitFile()? Можно используя send передавать файло...

Я думаю здесь recv не поможет...

Автор: UncleBob 14.6.2005, 16:26

Цитата(Johen @ 14.6.2005, 13:46)

зачем нужна вообще функция TransmitFile()?

The TransmitFile function transmits file data over a connected socket handle. This function uses the operating system's cache manager to retrieve the file data, and provides high-performance file data transfer over sockets.

Ну и как ты сделаешь то же самое через обычный send?

Добавлено @ 16:26

Цитата(Johen @ 14.6.2005, 13:46)

Я думаю здесь recv не поможет...

Уверен в обратном.

Автор: Johen 17.6.2005, 17:10

Тогда что-то вроде этого:
сервер: fread() -> send() ->
клиент: recv() -> fwrite()

smile

Автор: UncleBob 20.6.2005, 11:01

Цитата(Johen @ 17.6.2005, 17:10)

Тогда что-то вроде этого:

Я бы сделал так:

1. На сервере формируешь заголовок. Для простоты фиксированной длины

Код


#pragma pack(push)
#pragma pack(1)

typedef struct tag_MyFileHeader
{
  // имя файла
  char szFileName[256];    
  // длина
  DWORD dwLength; 
}MyFileHeader;

#pragma pack (pop)

2. Коннектишься, открываешь файл, отсылаешь файл:

Код


int SendFileBySocket (char szFileName, SOCKET hSocket)
{
  BOOL bRes;

  HANDLE hFile = CreateFile (szFileName, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_FLAG_SEQUENTIAL_SCAN, NULL);
  
  if (INVALID_HANDLE_VALUE != hFile)
  {
    MyFileHeader mfh = {0};
    strncpy (mfh.szFileName, szFileName, 256);
    mfh.dwLength = GetFileSize (hFile, NULL);

    TRANSMIT_FILE_BUFFERS tfb = 
    {
      &mfh,         // заголовок
      sizeof (mfh), // длина заголовка
      NULL,         // "хвостовик" не используем
      0             // 
    };

    bRes = TransmitFile (
                          hSocket,        // идентификатор сокета
                          hFile,          // хэндл файла
                          0,              // послать файл целиком
                          0,              // размер посылки - по умолчанию
                          NULL,           // не использовать OVERLAPPED
                          &tfb,           // заголовки
                          TF_DISCONNECT   // разорвать соединение после отсылки
                        ); 

    CloseHandle (hFile);
  }
  else
    return GetLastError();

  return bRes ? ERROR_SUCCESS : WSAGetLastError ();
}

3. Принимаешь файл:

Код


int ReceiveFile (SOCKET hSocket)
{
  MyFileHeader mfh = {0};

  int nReceived = 0;
  int nReceivedHeader = 0;

  while (nReceivedHeader < sizeof (mfh))
  {
    nReceived = recv (hSocket, (char*)&mfh + nReceivedHeader, nReceivedHeader - sizeof (mfh), 0);
    if (SOCKET_ERROR == nReceived)
      return WSAGetLastError ();
    
    if (0 == nReceived)
      return WSAECONNRESET; // хедер не дочитан - сэмулируем обрыв связи

    nReceivedHeader += nReceived;
  }

  HANDLE hFile = CreateFile (mfh.szFileName, GENERIC_WRITE, FILE_SHARE_READ, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_ARCHIVE, NULL);

  if (INVALID_HANDLE_VALUE == hFile)
    return GetLastError ();

  int nRes = ERROR_SUCCESS;
  
  do 
  {
    // Проверяем, хватит ли места на диске. Если нет - зачем принимать файл - лишний траффик
    SetFilePointer (hFile, mfh.dwLength, NULL, FILE_BEGIN);
    if (!SetEndOfFile (hFile))    
    {
      nRes = GetLastError();
      break;
    }

    SetFilePointer (hFile, 0, NULL, FILE_BEGIN);

    int nReceivedFile = 0;
    char TempBuf [4096];

    while (ERROR_SUCCESS == nRes && nReceivedFile < mfh.dwLength)
    {
      nReceived = recv (hSocket, TempBuf, 4096, 0);

      if (SOCKET_ERROR == nReceived)
        nRes = WSAGetLastError ();
      else if (0 == nReceived)
        nRes = WSAECONNRESET; // файл не дочитан - сэмулируем обрыв связи
      else
      {
        DWORD dwWritten;
        if (!WriteFile (hFile, TempBuf, nReceived, &dwWritten, NULL))
          nRes = GetLastError ();

        nReceivedFile += nReceived;
      }
    }

    if (ERROR_SUCCESS == nRes)
    {
      // по идее здесь recv должен вернуть 0, т.к. файл уже прочитан из сокета целиком
      nReceived = recv (hSocket, TempBuf, 1, 0); 

      if (0 != nReceived)
        nRes = WSAEMSGSIZE; // наиболее подходящая по смыслу ошибка
    }
  } 
  while(0);
  
  CloseHandle (hFile);
  
  return nRes;
}

4. Писал не проверяя, мобыть налажал smile