Category

Development Platform
Home > SourceCode/Document > Linux-Unix program > Linux Network > View Code
daytimed.c
Package: linux-socket-program.rar　[view]
Upload User: wei_4586
Upload Date: 2008-05-28
Package Size: 18k
Code Size: 2k
Category:

Linux Network

Development Platform:

Unix_Linux

daytimed.c：Code Content
  1. /* daytimed.c - main */
  3. #include <sys/types.h>
  4. #include <sys/socket.h>
  5. #include <sys/time.h>
  6. #include <netinet/in.h>
  8. #include <unistd.h>
  9. #include <stdio.h>
  10. #include <string.h>
  12. extern int errno;
  14. int daytime(char buf[]);
  15. int errexit(const char *format, ...);
  16. int passiveTCP(const char *service, int qlen);
  17. int passiveUDP(const char *service);
  19. #define MAX(x, y) ((x) > (y) ? (x) : (y))
  21. #define QLEN  32
  23. #define LINELEN 128
  25. /*------------------------------------------------------------------------
  26.  * main - Iterative server for DAYTIME service
  27.  *------------------------------------------------------------------------
  28.  */
  29. int
  30. main(int argc, char *argv[])
  31. {
  32. char *service = "daytime"; /* service name or port number */
  33. char buf[LINELEN+1]; /* buffer for one line of text */
  34. struct sockaddr_in fsin; /* the request from address */
  35. unsigned int alen; /* from-address length */
  36. int tsock;  /* TCP master socket */
  37. int usock; /* UDP socket */
  38. int nfds;
  39. fd_set rfds; /* readable file descriptors */
  41. switch (argc) {
  42. case 1:
  43. break;
  44. case 2:
  45. service = argv[1];
  46. break;
  47. default:
  48. errexit("usage: daytimed [port]n");
  49. }
  51. tsock = passiveTCP(service, QLEN);
  52. usock = passiveUDP(service);
  53. nfds = MAX(tsock, usock) + 1; /* bit number of max fd */
  55. FD_ZERO(&rfds);
  57. while (1) {
  58. FD_SET(tsock, &rfds);
  59. FD_SET(usock, &rfds);
  61. if (select(nfds, &rfds, (fd_set *)0, (fd_set *)0,
  62. (struct timeval *)0) < 0)
  63. errexit("select error: %sn", strerror(errno));
  64. if (FD_ISSET(tsock, &rfds)) {
  65. int ssock; /* TCP slave socket */
  67. alen = sizeof(fsin);
  68. ssock = accept(tsock, (struct sockaddr *)&fsin,
  69. &alen);
  70. if (ssock < 0)
  71. errexit("accept failed: %sn",
  72. strerror(errno));
  73. daytime(buf);
  74. (void) write(ssock, buf, strlen(buf));
  75. (void) close(ssock);
  76. }
  77. if (FD_ISSET(usock, &rfds)) {
  78. alen = sizeof(fsin);
  79. if (recvfrom(usock, buf, sizeof(buf), 0,
  80. (struct sockaddr *)&fsin, &alen) < 0)
  81. errexit("recvfrom: %sn",
  82. strerror(errno));
  83. daytime(buf);
  84. (void) sendto(usock, buf, strlen(buf), 0,
  85. (struct sockaddr *)&fsin, sizeof(fsin));
  86. }
  87. }
  88. }
  90. /*------------------------------------------------------------------------
  91.  * daytime - fill the given buffer with the time of day
  92.  *------------------------------------------------------------------------
  93.  */
  94. int
  95. daytime(char buf[])
  96. {
  97. char *ctime();
  98. time_t now;
  100. (void) time(&now);
  101. sprintf(buf, "%s", ctime(&now));
  102. }