ACE 学习笔记（一）

今天终于把尘封了好久的《C++网络编程》给搬出来，打算认真研究一下。

 

兴冲冲地把第四章 ACE_Message_Block 的示例输入进去测试，竟然发现有BUG，最后的数据会丢失一段。

 

#include <ace/ACE.h> #include <ace/OS.h> #include <ace/Message_Block.h> int main(int, ACE_TCHAR *[]) { ACE_Message_Block *head = new ACE_Message_Block(BUFSIZ); ACE_Message_Block *mblk = head; for (;;) { ssize_t nbytes = ACE::read_n(ACE_STDIN, mblk->wr_ptr(), mblk->size()); if(nbytes <= 0) break; mblk->wr_ptr(bytesTran); mblk->cont(new ACE_Message_Block(BUFSIZ)); mblk = mblk->cont(); } for (mblk = head; mblk != 0; mblk = mblk->cont()) { ACE::write_n(ACE_STDOUT, mblk->rd_ptr(), mblk->length()); } head->release(); return 0; }

 

仔细研究文档才明白，read_n 返回 0 或者 -1的时候，也可能有数据被读入的，这个时候要依赖 read_n 的参数4来检查。

下面是修改的代码，终于正常了。

 

#include <ace/ACE.h> #include <ace/OS.h> #include <ace/Message_Block.h> int main(int, ACE_TCHAR *[]) { ACE_Message_Block *head = new ACE_Message_Block(BUFSIZ); ACE_Message_Block *mblk = head; for (;;) { // 加上这个变量。 size_t bytesTran = 0; // 增加参数4 ssize_t nbytes = ACE::read_n(ACE_STDIN, mblk->wr_ptr(), mblk->size(), &bytesTran ); mblk->wr_ptr(bytesTran); // 这一句挪到上面来。 if(nbytes <= 0) // 注意：返回0的时候可能已读入数据，需要通过read_n的第四个参数判断。 break; mblk->cont(new ACE_Message_Block(BUFSIZ)); mblk = mblk->cont(); } for (mblk = head; mblk != 0; mblk = mblk->cont()) { ACE::write_n(ACE_STDOUT, mblk->rd_ptr(), mblk->length()); } head->release(); return 0; }