论坛徽章:: 0

电梯直达

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

发表于 2009-03-19 21:14 |只看该作者 |倒序浏览

最近为了移植windows网络SDK代码到Linux下,在保持代码的统一性前提下,模拟了windows WSAEVENT的相关函数, 幸运的是在该SDK中: DWORD
WSAWaitForMultipleEvents(DWORD cEvents, const WSAEVENT FAR *lphEvents, BOOL fWaitAll, DWORD dwTimeout, BOOL fAlertable);
参数fWaitAll, 以及fAlertable都是使用FALSE. 要是有一个TRUE的话就难办了

#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <pthread.h>

#define __USE_XOPEN
#include <sys/poll.h>

#define TRUE 1
#define FALSE 0
#define BOOL int

#define DWORD unsigned int
#define FAR

#define WSA_INVALID_PARAMETER ((DWORD)-3L)
#define WSA_INVALID_EVENT    ((DWORD)-2L)
#define WSA_WAIT_FAILED       ((DWORD)-1L)
#define WSA_WAIT_TIMEOUT       ((DWORD)0)
#define WSA_WAIT_EVENT_0       ((DWORD)1)

#define FLG_WSA_EVENT 0

typedef struct __LNX_EVENT
{
int flags;
int lnx_fd[2];
pthread_mutex_t mutex;
}LNX_EVENT;

typedef void* HANDLE;
typedef LNX_EVENT* WSAEVENT;

HANDLE WSACreateEvent(void)
{
LNX_EVENT* event = (LNX_EVENT*)malloc(sizeof(LNX_EVENT));
int retval = pipe(event->lnx_fd);
if(retval < 0)
{
      free(event);
      return NULL;
}

event->flags = FLG_WSA_EVENT;
pthread_mutex_init(&(event->mutex), NULL);
return (HANDLE)event;
}

BOOL WSASetEvent(HANDLE hEvent)
{
LNX_EVENT* event = (LNX_EVENT*)hEvent;
if(event->flags != FLG_WSA_EVENT)
{
      return FALSE;
}

pthread_mutex_lock(&(event->mutex));
struct pollfd fdarray[2];
memset(fdarray, 0x0, sizeof(struct pollfd) * 2);

fdarray[0].fd = event->lnx_fd[0];
fdarray[0].events = POLLRDNORM;

fdarray[1].fd = event->lnx_fd[1];
fdarray[1].events = POLLWRNORM;

int ret = poll(&fdarray[0], 1, 0);
if(ret > 0 && (fdarray[0].revents & POLLRDNORM))
{
      char buffer[256] = {0};
      read(event->lnx_fd[0], buffer, 1024);
}

ret = poll(&fdarray[1], 1, 0);
if(ret <= 0)
{
      pthread_mutex_unlock(&(event->mutex));
      return FALSE;
}

char w_signal = 'h';
BOOL bRet = write(event->lnx_fd[1], &w_signal, 1) == 1 ? TRUE : FALSE;
pthread_mutex_unlock(&(event->mutex));
return bRet;
}

BOOL WSAResetEvent(HANDLE hEvent)
{
BOOL bRet = FALSE;
LNX_EVENT* event = (LNX_EVENT*)hEvent;
if(event->flags != FLG_WSA_EVENT)
{
      return FALSE;
}

pthread_mutex_lock(&(event->mutex));
struct pollfd fdarray;
memset(&fdarray, 0x0, sizeof(struct pollfd));

fdarray.fd = event->lnx_fd[0];
fdarray.events = POLLRDNORM;

int ret = poll(&fdarray, 1, 0);
if(ret > 0 && (fdarray.revents & POLLRDNORM))
{
      char buffer[256] = {0};
      bRet = read(event->lnx_fd[0], buffer, 1024) > 0 ? TRUE : FALSE;
      fprintf(stderr, "Reset the send buffer: %s\n", buffer);
}

pthread_mutex_unlock(&(event->mutex));
return bRet;
}

BOOL WSACloseEvent(HANDLE hEvent)
{
LNX_EVENT* event = (LNX_EVENT*)hEvent;
if(event->flags != FLG_WSA_EVENT)
{
      return FALSE;
}

close(event->lnx_fd[0]);
close(event->lnx_fd[1]);
pthread_mutex_destroy(&(event->mutex));
free(event);

return TRUE;
}

DWORD WSAWaitForMultipleEvents(DWORD cEvents, const WSAEVENT FAR *lphEvents, BOOL fWaitAll, DWORD dwTimeout, BOOL fAlertable)
{
if(cEvents <= 0 || lphEvents == NULL || fWaitAll == TRUE || fAlertable == TRUE)
{
      return WSA_INVALID_PARAMETER;
}

struct pollfd fdarray[cEvents];
memset(fdarray, 0x0, sizeof(struct pollfd) * cEvents);

int index = 0;
for(; index < cEvents; index++)
{
      if(lphEvents[index]->flags != FLG_WSA_EVENT)
      {
         return WSA_INVALID_EVENT;
      }

      fdarray[index].fd = lphEvents[index]->lnx_fd[0];
      fdarray[index].events = POLLRDNORM;
}

int ret = poll(fdarray, cEvents, dwTimeout);
if(ret > 0)
{
      for(index = 0; index < cEvents; index++)
      {
         if(fdarray[index].revents & POLLRDNORM)
         {
            return WSA_WAIT_EVENT_0 + index;
         }
      }
}
else if(ret == 0)
{
      return WSA_WAIT_TIMEOUT;
}

return WSA_WAIT_FAILED;
}

[ 本帖最后由飞雪横天于 2009-3-19 21:19 编辑 ]

文库|博客

飞雪横天

丰衣足食

论坛徽章:: 0

2楼 [报告]

发表于 2009-03-19 21:17 |只看该作者

WSAWaitForMultipleEvents测试用例

static void* send_event(void* argptr)
{
WSAEVENT event = (WSAEVENT)argptr;
sleep(5);
WSASetEvent(event);
WSASetEvent(event);
pthread_exit(NULL);
return NULL;
}
int main(int argc, char* argv[])
{
HANDLE hEvent1 = WSACreateEvent();
HANDLE hEvent2 = WSACreateEvent();

pthread_t hthread1, hthread2;
pthread_create(&hthread2, NULL, send_event, hEvent2);
pthread_create(&hthread1, NULL, send_event, hEvent1);

WSAEVENT events[2] = {hEvent1, hEvent2};
DWORD ret = WSAWaitForMultipleEvents(2, events, FALSE, 5020, FALSE);
if(ret == WSA_WAIT_EVENT_0)
{
      fprintf(stderr, "wait for two events, and the first event: %p signal\n", hEvent1);
      sleep(1);
      WSAResetEvent(hEvent1);
      goto END_EVENT;
}
else if (ret == WSA_WAIT_EVENT_0 + 1)
{
      fprintf(stderr, "wait for two events, and the second event: %p signal\n", hEvent2);
      sleep(1);
      WSAResetEvent(hEvent2);
      goto END_EVENT;
}
else
{
      fprintf(stderr, "wait for two events(:%p, %p) failed\n", hEvent1, hEvent2);
      goto END_EVENT;
}

END_EVENT:
pthread_join(hthread1, NULL);
pthread_join(hthread2, NULL);

WSACloseEvent(hEvent1);
WSACloseEvent(hEvent2);

return 0;
}

用poll模拟WSAWaitForMultipleEvents的行为, event用pipe来进行模拟. 整个过程比较简单, 欢迎大家轻点拍砖;

实战分享：从技术角度谈机器学习入门| 【大话IT】RadonDB低门槛向MySQL集群下战书 | ChinaUnix打赏功能已上线！ | 新一代分布式关系型数据库RadonDB知多少？

MMMIX

广告杀手

论坛徽章:: 95

3楼 [报告]

发表于 2009-03-19 22:20 |只看该作者

原帖由 飞雪横天 于 2009-3-19 21:14 发表
最近为了移植windows网络SDK代码到Linux下,在保持代码的统一性前提下,模拟了windows WSAEVENT的相关函数, 幸运的是在该SDK中: DWORD
WSAWaitForMultipleEvents(DWORD cEvents, const WSAEVENT FAR *lphEvents, ...

光看这函数名字就能让我做噩梦。。。