加载中…
个人资料
微博
  • 博客等级：
  • 博客积分：
  • 博客访问：
  • 关注人气：
  • 获赠金笔：0支
  • 赠出金笔：0支
  • 荣誉徽章：
正文 字体大小：大 中 小

AF_LOCAL和AF_INET两个套接口同时监听

(2013-02-28 16:38:17)
标签：

af_local

af_inet

 分类： 网络编程
  1. #include <stdlib.h>
  2. #include <stdio.h>
  3. #include <errno.h>
  4. #include <string.h>
  5. #include <unistd.h>
  6. #include <netdb.h>
  7. #include <sys/socket.h>
  8. #include <netinet/in.h>
  9. #include <sys/types.h>
  10. #include <arpa/inet.h>

  12. #define SA struct sockaddr

  14. int main(int argc, char **argv)
  15. {
  16.     int use_tcp = 1;
  17.     int use_unix = 1;

  19.     int fdmax = 0;

  21.     int rt = 0;
  22.     int fdtcp, fdunix;

  24.     if (!getenv("USEUNIX"))
  25.         use_unix = 0;
  26.     if (!getenv("USETCP"))
  27.         use_tcp = 0;

  29.     if (use_tcp) {
  30.         // Enable TCP server socket
  31.         fdtcp = socket(AF_INET, SOCK_STREAM, 0);
  32.         if (fdtcp < 0) {
  33.             perror("socket( TCP )");
  34.         }
  35.         fdmax = fdmax > fdtcp ? fdmax : fdtcp;
  36.         struct sockaddr_in servaddr;
  37.         bzero(&servaddr, sizeof(servaddr));
  38.         servaddr.sin_family = AF_INET;
  39.         servaddr.sin_port = htons(1235);
  40.         int reuseaddr = 1;
  41.         socklen_t reuseaddr_len = sizeof(reuseaddr);
  42.         rt = setsockopt(fdtcp, SOL_SOCKET, SO_REUSEADDR, &reuseaddr,
  43.                         reuseaddr_len);
  44.         rt = bind(fdtcp, (SA *) & servaddr, sizeof(servaddr));
  45.         if (rt < 0) {
  46.             perror("bind( TCP )");
  47.             exit(1);
  48.         }
  49.         listen(fdtcp, 20);
  50.     }

  52.     if (use_unix) {
  53.         // Get ready the UNIX Domain Socket
  54.         fdunix = socket(AF_UNIX, SOCK_STREAM, 0);
  55.         if (fdunix < 0) {
  56.             perror("socket( UNIX )");
  57.             exit(1);
  58.         }
  59.         fdmax = fdmax > fdunix ? fdmax : fdunix;
  60.         struct sockaddr_un unixaddr;
  61.         bzero(&unixaddr, sizeof(unixaddr));
  62.         unixaddr.sun_family = AF_UNIX;
  63.         strncpy(&unixaddr.sun_path[0], "/tmp/test-unix.0", 17);
  64.         rt = bind(fdunix, (SA *) & unixaddr, sizeof(unixaddr));
  65.         //unlink( "/tmp/test-unix.0" );
  66.         if (rt < 0) {
  67.             perror("bind( UNIX )");
  68.             exit(1);
  69.         }

  71.         rt = listen(fdunix, 20);
  72.         if (rt < 0) {
  73.             perror("listen( UNIX )");
  74.             exit(1);
  75.         }
  76.     }

  78.     int selectrt = 0;
  79.     while (use_tcp || use_unix) {
  80.         fd_set ssset;
  81.         FD_ZERO(&ssset);
  82.         if (use_tcp)
  83.             FD_SET(fdtcp, &ssset);
  84.         if (use_unix)
  85.             FD_SET(fdunix, &ssset);

  87.         selectrt = select(fdmax + 1, &ssset, NULL, NULL, NULL);
  88.         while (selectrt > 0) {
  89.             if (use_tcp) {
  90.                 if (FD_ISSET(fdtcp, &ssset)) {
  91.                     printf("TCP Server is ready.\n");
  92.                     struct sockaddr_in caddr;
  93.                     bzero(&caddr, sizeof(caddr));
  94.                     socklen_t clen = sizeof(caddr);
  95.                     int client = accept(fdtcp, (SA *) & caddr, &clen);
  96.                     write(client, "HELO", 5);
  97.                     close(client);
  98.                     --selectrt;
  99.                 }
  100.             }
  101.             if (use_unix) {
  102.                 if (FD_ISSET(fdunix, &ssset)) {
  103.                     printf("UNIX Server is ready.\n");
  104.                     struct sockaddr_un caddr;
  105.                     bzero(&caddr, sizeof(caddr));
  106.                     socklen_t clen = sizeof(caddr);
  107.                     int client = accept(fdunix, (SA *) & caddr, &clen);
  108.                     rt = write(client, "HELO", 5);
  109.                     printf("UNIX Socket wrote %d bytes off.\n", rt);
  110.                     close(client);
  111.                     --selectrt;
  112.                 }
  113.             }
  114.         }
  115.         if ( selectrt < 0 ) {
  116.           perror( "select()" );
  117.           exit( 1 );
  118.         }
  119.         FD_ZERO(&ssset);
  120.     }
  121. }


  1. $ make sockd
  2. cc     sockd.c   -o sockd
  3. $ env USETCP='' USEUNIX='' ./sockd
  4. UNIX Server is ready.
  5. UNIX Socket wrote 5 bytes off.
  6. UNIX Server is ready.
  7. UNIX Socket wrote 5 bytes off.
  8. TCP Server is ready.
另外一边
  1. $ ./sockclient /tmp/test-unix.0
  2. UNIX Socket read 5 bytes.
  3. HELO
  4. $ ./sockclient /tmp/test-unix.0
  5. UNIX Socket read 5 bytes.
  6. HELO
  7. $ nc localhost 1235 -vvv
  8. gydebian [127.0.0.1] 1235 (?) open
  9. HELO sent 0, rcvd 5

喜欢

0

阅读┊ 收藏 ┊ 喜欢▼ ┊打印┊举报/Report
  

新浪BLOG意见反馈留言板　欢迎批评指正

新浪简介 | About Sina | 广告服务 | 联系我们 | 招聘信息 | 网站律师 | SINA English | 产品答疑

Copyright © 1996 - 2022 SINA Corporation, All Rights Reserved

新浪公司 版权所有