论坛徽章:: 0

电梯直达

1楼 [收藏(0)] [报告]

发表于 2009-09-03 08:55 |只看该作者 |倒序浏览

20可用积分

近来移植AIX下的多路复用代码到LINUX下，我在AIX使用的是poll()函数来实现Socket和消息队列句柄的监控，但发现在Linux下好像不支持对消息队列句柄的监控哦！
不知道哪位大侠知道是不是确实如此啊？
代码如下：

  int iRet;
msgbuf_t tMsg;

#ifdef AIX
unsigned int  nfds,nmsgs;
#else
struct pollfd tPoll[2];
#endif

while(1)
{
#ifdef AIX
   tPollList.fds.fd=iSock;
   tPollList.fds.events=POLLIN;

   tPollList.msgs.msgid=iMsgFrom;
   tPollList.msgs.events=POLLIN;

   nfds =1;
   nmsgs =1;

   Debug(2,"Begin polling ...\n");
   iRet=poll(&tPollList,(nmsgs<<16)|(nfds),-1);
#else
   tPoll[0].fd = iMsgFrom;
   tPoll[0].events = POLLIN;
   tPoll[0].revents = 0;

   tPoll[1].fd = iSock;
   tPoll[1].events = POLLIN;
   tPoll[1].revents = 0;

   Debug(2,"Begin polling ...\n");
   iRet=poll(tPoll,2,-1);
Debug(1,"iRet[%d][%x][%x]\n",iRet,tPoll[0].revents,tPoll[1].revents);

#endif
Debug(1,"iRet[%d]\n",iRet);

   if( iRet < 0 && errno != EINTR )
   {
      Debug(1,"Polling error [%d][%d][%s] !\n",iRet,errno,strerror(errno));
      continue;
   }

#ifdef AIX
   if( tPollList.msgs.revents & POLLIN )
#else
   if( tPoll[0].revents & POLLIN )
#endif
   {
DEBUG(1);
      GetDate(szDate,YYMD|SYM_HYPHEN);
      GetTime(szTime,SYM_COLON);
      Debug(2,"\n%s %s ",szDate,szTime);
      if( (iLen=msgrcv(iMsgFrom,&tMsg,MSG_LEN,0,~IPC_NOWAIT)) < 0 )
      {
         DEBUG(1);
         Debug(1,"接收数据错误! [%d][%s]\n",iLen,strerror(errno));
         break;
      }

      szBuffer[0] = iLen / 256;
      szBuffer[1] = iLen % 256;
      memcpy(&szBuffer[2],&tMsg.szText[0],iLen);
      iLen += 2; /* add length */
      Debug(2,"Receive from MQ[%x][%d] and Send to socket.len=[%d]\n",
         kFrom,tMsg.lType,iLen);
      DebugHex(3,&szBuffer[0],iLen);

      if( (iLen=write(iSock,szBuffer,iLen)) == -1 )
      {
         Debug(1,"Send to socket! errno=[%d][%s]\n",errno,strerror(errno));
         break;
      }
   }

#ifdef AIX
   if( tPollList.fds.revents & POLLIN )
#else
   if( tPoll[1].revents & POLLIN )
#endif
   {
DEBUG(1);
      GetDate(szDate,YYMD|SYM_HYPHEN);
      GetTime(szTime,SYM_COLON);
      Debug(2,"\n%s %s ",szDate,szTime);

      iLen=ReadSock(iSock,tMsg.szText,2,0,1,100);
      if( iLen == 0 )
      {
         DEBUG(5);
         continue;
      }
      else if( iLen == -1 )
      {
         DEBUG(5);
         break;
      }
      Debug(2,"len=[%d] ",iLen);
      if( iLen < 5 )
      {
         Debug(1,"Data error![%d]\n",iLen);
         DebugHex(1,&tMsg.szText[0],iLen);
         continue;
      }

      tMsg.lType = 1;

      Debug(2,"Receive from socket and Send to MQ[%x][%d].len=[%d]\n",
         kTo,tMsg.lType,iLen);
      DebugHex(3,tMsg.szText,iLen);

      if( msgsnd(iMsgTo,&tMsg,iLen,IPC_NOWAIT) < 0 )
      {
         Debug(1,"Send date to message queue [%x] error![%s]\n",kTo,strerror(errno));
         break;
      }
   }

   break;
}

GetDate(szDate,YYMD|SYM_HYPHEN);
GetTime(szTime,SYM_COLON);
Debug(2,"%s %s 通讯故障!\n",szDate,szTime);