Chinaunix首页 | 论坛 | 博客

平静 思考 行动dara.blog.chinaunix.net

首页　| 　博文目录　| 　关于我

czysocket_dara

  • 博客访问： 468102
  • 博文数量： 62
  • 博客积分： 1742
  • 博客等级： 中尉
  • 技术积分： 859
  • 用 户 组： 普通用户
  • 注册时间： 2010-11-06 00:13
个人简介

这是一句很长很长而且又很啰嗦并且很无聊的废话...

文章分类

全部博文（62）

文章存档

2013年（1）

2012年（13）

2011年（48）

我的朋友
最近访客
推荐博文
相关博文
Semaphore Select Fifo

分类： LINUX

2011-12-31 20:44:28

常用代码,贴着以备查用.

Write
  1. int send_login_msg(p_Login_Msg p_lm){
  2.     int key, fifo_fd, len, pro_lock, tw_frequency = MAX_TRY_WRITE;
  3.     fd_set rfds;
  4.     struct timeval tv;
  5.     p_Fifo_Msg p_fm;
  6.     char buf[1024];
  7.     union semun sem_arg;

  9.     key = ftok(SEM, 'a');
  10.     pro_lock = semget(key, 1, (IPC_CREAT | 0666));
  11.     if(pro_lock < 0){
  12.         DBG("Error:Semget Call Failure!\n");
  13.         return 0;
  14.     }

  16.     if(access(FIFO, F_OK)){
  17.         DBG("Bengin Init Sem And Mk Fifo!\n");

  19.         sem_arg.val = 1;
  20.         if(semctl(pro_lock, 0, SETVAL, sem_arg) < 0){
  21.             DBG("Error:Init Sem Value Failure!\n");
  22.             return 0;
  23.         }
  24.         if((mkfifo(FIFO, 0777) < 0) && (errno != EEXIST)){
  25.             DBG("Error:Cannot Create FIFO File!\n");
  26.             return 0;
  27.         }
  28.     }

  30.     fifo_fd = open(FIFO, (O_RDWR | O_NONBLOCK), 0);
  31.     if(fifo_fd < 0){
  32.         DBG("Error:Fifo Open Failure!\n");
  33.         return 0;
  34.     }

  36.     p_fm = (p_Fifo_Msg)buf;
  37.     p_fm->fm_head.ft = FIFO_LOGIN;
  38.     p_fm->fm_head.msg_len = sizeof(Login_Msg);
  39.     memcpy(&p_fm->body, p_lm, sizeof(Login_Msg));

  41. try_write_again:
  42.     Pro_Lock(pro_lock, apps_thread_exit);
  43.     len = write(fifo_fd, buf, (sizeof(Fifo_Msg_Head) + p_fm->fm_head.msg_len));
  44.     Pro_UnLock(pro_lock, apps_thread_exit);
  45.     if(len < 0){
  46.         if(errno == EAGAIN){
  47.             if(tw_frequency--){
  48.                 usleep(500000);
  49.                 goto try_write_again;
  50.             }else{
  51.                 DBG("Error: Send Login Msg Fialure!\n");
  52.             }
  53.         }
  54.     }

  56. apps_thread_exit:
  57.     close(fifo_fd);

  59.     return 0;
  60. }
Read
  1. void *apps_thread(void *arg){
  2.     int key, fifo_fd, len, pro_lock;
  3.     fd_set rfds;
  4.     struct timeval tv;
  5.     p_Fifo_Msg p_fm;
  6.     char buf[1024];
  7.     union semun sem_arg;

  9.     key = ftok(SEM, 'a');
  10.     pro_lock = semget(key, 1, (IPC_CREAT | 0666));
  11.     if(pro_lock < 0){
  12.         DBG("Error:Semget Call Failure!\n");
  13.         return 0;
  14.     }

  16.     if(access(FIFO, F_OK)){
  17.         DBG("Bengin Init Sem And Mk Fifo!\n");

  19.         sem_arg.val = 1;
  20.         if(semctl(pro_lock, 0, SETVAL, sem_arg) < 0){
  21.             DBG("Error:Init Sem Value Failure!\n");
  22.             goto apps_thread_exit_sem;
  23.         }
  24.         if((mkfifo(FIFO, 0777) < 0) && (errno != EEXIST)){
  25.             DBG("Error:Cannot Create FIFO File!\n");
  26.             goto apps_thread_exit_sem;
  27.         }
  28.     }

  30.     fifo_fd = open(FIFO, (O_RDWR | O_NONBLOCK), 0);
  31.     if(fifo_fd < 0){
  32.         DBG("Error:Fifo Open Failure!\n");
  33.         goto apps_thread_exit_sem;
  34.     }

  36.     p_fm = (p_Fifo_Msg)buf;
  37.     while(1){
  38.         FD_ZERO(&rfds);
  39.         FD_SET(fifo_fd, &rfds);
  40.         tv.tv_sec = SEL_TO;
  41.         tv.tv_usec = 0;

  43.         if(select(fifo_fd + 1, &rfds, 0, 0, &tv) < 0){
  44.             DBG("Error:Select Call Failure!\n");
  45.             goto apps_thread_exit;
  46.         }

  48.         if(FD_ISSET(fifo_fd, &rfds)){
  49.             memset(buf, 0, sizeof(buf));
  50.             
  51.             Pro_Lock(pro_lock, apps_thread_exit);
  52.             len = read(fifo_fd, buf, sizeof(Fifo_Msg_Head));
  53.             if(len != sizeof(Fifo_Msg_Head)){
  54.                 DBG("Error:Cant Not Read Msg Head!\n");
  55.                 Pro_UnLock(pro_lock, apps_thread_exit);
  56.                 goto apps_thread_exit;
  57.             }

  59.             len = read(fifo_fd, (buf + sizeof(Fifo_Msg_Head)), p_fm->fm_head.msg_len);
  60.             if(len != p_fm->fm_head.msg_len){
  61.                 DBG("Error:Read Msg Body Failure!\n");
  62.                 Pro_UnLock(pro_lock, apps_thread_exit);
  63.                 goto apps_thread_exit;
  64.             }
  65.             Pro_UnLock(pro_lock, apps_thread_exit);

  67.             switch(p_fm->fm_head.ft){
  68.                 case FIFO_APPS:
  69.                     handle_apps_msg(p_fm);
  70.                     break;
  71.                 case FIFO_LOGIN:
  72.                     handle_login_msg(p_fm);
  73.                     break;
  74.                 case FIFO_GPS:
  75.                     handle_gps_msg(p_fm);
  76.                     break;
  77.                 case FIFO_CGI:
  78.                     handle_cgi_msg(p_fm);
  79.                     break;
  80.                 default:
  81.                     break;
  82.             }
  83.         }
  84.     }

  86. apps_thread_exit:
  87.     close(fifo_fd);
  88. apps_thread_exit_sem:
  89.     unlink(FIFO);
  90.     if(semctl(pro_lock, 0, IPC_RMID, sem_arg) < 0){
  91.         DBG("Error:Rm Sem Vaule Failure!\n");
  92.     }

  94.     return 0;
  95. }
  1. #define Pro_Lock(semid, err_goto) do{ \
  2.                                         struct sembuf sem_b; \
  3.                                         sem_b.sem_num = 0; \
  4.                                         sem_b.sem_op = -1; \
  5.                                         sem_b.sem_flg = SEM_UNDO; \
  6.                                         if(semop((semid), &sem_b, 1) < 0){ \
  7.                                             DBG("%s->%s->%d:Error:Sem P Ctl Fialure!\n", __FILE__, __FUNCTION__, __LINE__); \
  8.                                             goto err_goto; \
  9.                                         } \
  10.                                     }while(0) \

  12. #define Pro_UnLock(semid, err_goto) do{ \
  13.                                         struct sembuf sem_b; \
  14.                                         sem_b.sem_num = 0; \
  15.                                         sem_b.sem_op = 1; \
  16.                                         sem_b.sem_flg = SEM_UNDO; \
  17.                                         if(semop((semid), &sem_b, 1) < 0){ \
  18.                                             DBG("%s->%s->%d:Error:Sem V Ctl Fialure!\n", __FILE__, __FUNCTION__, __LINE__); \
  19.                                             goto err_goto; \
  20.                                         } \
  21.                                     }while(0) \

阅读(2462) | 评论(0) | 转发(0) |
0

上一篇：TCP/IP

下一篇：TCP的Send和Recv

给主人留下些什么吧！~~
关于我们 | 关于IT168 | 联系方式 | 广告合作 | 法律声明 | 免费注册

Copyright 2001-2010 ChinaUnix.net All Rights Reserved 北京皓辰网域网络信息技术有限公司. 版权所有

感谢所有关心和支持过ChinaUnix的朋友们

16024965号-6