修改ACE自带的模拟sendfile()实现

wishel

ace/OS_NS_sys_sendfile.cpp中sendfile_emulation的实现有些问题，unix系统ACE_OS::mmap的offset参数必须是内存页面大小的整数倍，否则会出错。虽然常用的是0，但有时用到非0值的时候就要注意了。

简单修改了下。（假设内存页大小为4k）

在windows上可以用TransmitFile()来模拟。具体实现相当tricky，详细说明见代码注释。

1. // $Id: OS_NS_sys_sendfile.cpp 84216 2009-01-22 18:34:40Z johnnyw $
3. #include "ace/OS_NS_sys_sendfile.h"
4. #include "ace/OS_NS_sys_mman.h"
5. #include "ace/OS_NS_unistd.h"
7. #if defined (ACE_WIN32) || defined (HPUX)
8. # include "ace/OS_NS_sys_socket.h"
9. #else
10. # include "ace/OS_NS_unistd.h"
11. #endif  /* ACE_WIN32 || HPUX */
13. #ifndef ACE_HAS_INLINED_OSCALLS
14. # include "ace/OS_NS_sys_sendfile.inl"
15. #endif  /* ACE_HAS_INLINED_OSCALLS */
17. ACE_BEGIN_VERSIONED_NAMESPACE_DECL
19. #if defined ACE_HAS_SENDFILE && ACE_HAS_SENDFILE == 0
21. ssize_t
22. ACE_OS::sendfile_emulation (ACE_HANDLE out_fd,
23.                        ACE_HANDLE in_fd,
24.                        off_t * offset,
25.                        size_t count)
26. {
27.      // @@ Is it possible to inline a call to ::TransmitFile() on
28.      //    MS Windows instead of emulating here?
30.      // @@ We may want set up a signal lease (or oplock) if supported by
31.      //    the platform so that we don't get a bus error if the mmap()ed
32.      //    file is truncated.
33. #if defined (ACE_WIN32)
34.      // 0表示取系统默认值，也可自定义大小
35.      static const size_t BytesPerSend = 0;
37.      if (offset != NULL)
38.          ACE_OS::lseek(in_fd, (ACE_OFF_T)offset, SEEK_SET);
40.      // 记录发送前文件指针位置
41.      ACE_OFF_T currentOffset = ACE_OS::lseek(in_fd, 0, SEEK_CUR);
43.      // 当前文件指针+count不得超过文件尾，否则TransmitFile出错WSAEINVAL。
44.      // 此时需修改count
45.      ACE_OFF_T eofPosition = ACE_OS::lseek(in_fd, 0, SEEK_END);
46.      ACE_OS::lseek(in_fd, currentOffset, SEEK_SET);
47.      size_t bytesLeft = (size_t)(eofPosition - currentOffset);
48.      if (bytesLeft < count)
49.          count = bytesLeft;
51.      // 发送从当前文件指针开始的count个字节的数据
52.      // count==0表示传送从当前文件指针到文件尾的所有数据
53.      BOOL suc = ::TransmitFile((SOCKET)out_fd, in_fd, count, BytesPerSend, 0, 0, TF_USE_DEFAULT_WORKER);
54.      if (suc == FALSE)  
55.          return -1;
57.      // TransmitFile()的文件指针自动设置不可依赖（仅在第一次调用时前进BytesPerSend，
58.      // 而不是实际发送值count，而且以后再次调用时不再前进）。需要自己控制绝对值
59.      currentOffset += count;
60.      ACE_OS::lseek(in_fd, currentOffset, SEEK_SET);
62.      return count;
63. #endif
64.      const size_t alignmentGranularity = 4 * 1024; // page size on unix
66.      off_t adjustedOffset = (*offset / (alignmentGranularity)) * alignmentGranularity;
68.      void * const buf =
69.          ACE_OS::mmap (0, count, PROT_READ, MAP_SHARED, in_fd, adjustedOffset);
71.      if (buf == MAP_FAILED)
72.          return -1;
74.      size_t memoryOffset = *offset % (alignmentGranularity);
75.      void* readPtr = (char*)buf + memoryOffset;
76.      size_t countLeft = count - memoryOffset;
77. #if defined (HPUX)
78.      ssize_t const r =
79.          ACE_OS::send (out_fd, static_cast<const char *> (readPtr), countLeft);
80. #else
81.      ssize_t const r = ACE_OS::write (out_fd, readPtr, countLeft);
82. #endif /* HPUX */
84.      (void) ACE_OS::munmap (buf, count);
86.      if (r > 0)
87.          *offset += static_cast<off_t> (r);
89.      return r;
90. }
92. #endif  /* ACE_HAS_SENDFILE==0 */
94. ACE_END_VERSIONED_NAMESPACE_DECL

复制代码

[ 本帖最后由 wishel 于 2009-11-10 14:53 编辑 ]

winston

你完全可以提交patch给开发组的。

wishel

原帖由 winston 于 2009-11-10 17:34 发表
你完全可以提交patch给开发组的。

我也是这么想，不过要把中文注释翻译成英文比较累哈哈。
再继续测一段时间吧，没问题的话就提交。

wishel

遇到新难题：

Sendfile有个重要语义TransimitFile难以模拟：
当作用nonblock socket时，sendfile发送部分请求量（填满socket 发送缓冲）后返回发送量。
但TransimitFile却会block，一直到请求的发送量全部发送完毕才返回。