08d926f5d462b700d0888e754472a8e8546013d2
[gapil.git] / TCP_echod.c
1 /* TCP_echod.c
2  * 
3  * Copyright (C) 2001-2003 Simone Piccardi
4  * 
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License as published by
7  * the Free Software Foundation; either version 2 of the License, or (at
8  * your option) any later version.
9  * 
10  * This program is distributed in the hope that it will be useful, but
11  * WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13  * General Public License for more details.
14  * 
15  * You should have received a copy of the GNU General Public License
16  * along with this program; if not, write to the Free Software
17  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
18  */
19 /****************************************************************
20  *
21  * Program echod 
22  * Elementary TCP server for echo service (port 7)
23  *
24  * Author: Simone Piccardi
25  * Jun. 2001
26  *
27  * Usage: echod -h give all info
28  *
29  * $Id: TCP_echod.c,v 1.4 2003/06/18 21:19:24 piccardi Exp $ 
30  *
31  ****************************************************************/
32 /* 
33  * Include needed headers
34  */
35 #include <sys/types.h>   /* predefined types */
36 #include <unistd.h>      /* include unix standard library */
37 #include <arpa/inet.h>   /* IP addresses conversion utiliites */
38 #include <sys/socket.h>  /* socket library */
39 #include <stdio.h>       /* include standard I/O library */
40 #include <time.h>
41 #include <syslog.h>      /* syslog system functions */
42 #include <signal.h>      /* signal functions */
43 #include <errno.h>       /* error code */
44 #include "Gapil.h"
45
46 #define BACKLOG 10
47 #define MAXLINE 256
48 int demonize  = 1;  /* daemon use option: default is daemon */
49 int debugging = 0;  /* debug info printing option: default is no debug */
50 /* Subroutines declaration */
51 void usage(void);
52 void ServEcho(int sockfd);
53 void PrintErr(char * error);
54 /* Program beginning */
55 int main(int argc, char *argv[])
56 {
57 /* 
58  * Variables definition  
59  */
60     int list_fd, conn_fd;
61     int waiting;
62     pid_t pid;
63     struct sockaddr_in serv_add;
64     /*
65      * Input section: decode parameters passed in the calling 
66      * Use getopt function
67      */
68     int i;
69     opterr = 0;  /* don't want writing to stderr */
70     while ( (i = getopt(argc, argv, "hdiw:")) != -1) {
71         switch (i) {
72         /* 
73          * Handling options 
74          */ 
75         case 'h':  
76             printf("Wrong -h option use\n");
77             usage();
78             return(0);
79             break;
80         case 'i':
81             demonize = 0;
82             break;
83         case 'd':
84             debugging = 1;
85             break;
86         case 'w':
87             waiting = strtol(optarg, NULL, 10);
88             break;
89         case '?':   /* unrecognized options */
90             printf("Unrecognized options -%c\n",optopt);
91             usage();
92         default:    /* should not reached */
93             usage();
94         }
95     }
96     /* ***********************************************************
97      * 
98      *           Options processing completed
99      *
100      *                Main code beginning
101      * 
102      * ***********************************************************/
103     /* install SIGCHLD handler */
104     Signal(SIGCHLD, HandSigCHLD);  /* establish handler */
105     /* create socket */
106     if ( (list_fd = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
107         perror("Socket creation error");
108         exit(1);
109     }
110     /* initialize address */
111     memset((void *)&serv_add, 0, sizeof(serv_add)); /* clear server address */
112     serv_add.sin_family = AF_INET;                  /* address type is INET */
113     serv_add.sin_port = htons(7);                   /* echo port is 7 */
114     serv_add.sin_addr.s_addr = htonl(INADDR_ANY);   /* connect from anywhere */
115     /* bind socket */
116     if (bind(list_fd, (struct sockaddr *)&serv_add, sizeof(serv_add)) < 0) {
117         perror("bind error");
118         exit(1);
119     }
120     /* release privileges and go daemon */
121     if (setgid(65534) !=0) { /* first give away group privileges */
122         perror("cannot give away group privileges");
123         exit(1);
124     }
125     if (setuid(65534) !=0) { /* and only after user ... */
126         perror("cannot give away user privileges");
127         exit(1);
128     }
129     if (demonize) {          /* go daemon */
130         openlog(argv[0], 0, LOG_DAEMON); /* open logging */
131         if (daemon(0, 0) != 0) {
132             perror("cannot start as daemon");
133             exit(1);
134         }
135     }
136     /* main body */
137     if (listen(list_fd, BACKLOG) < 0 ) {
138         PrintErr("listen error");
139         exit(1);
140     }
141     if (waiting) sleep(waiting);
142     /* handle echo to client */
143     while (1) {
144         /* accept connection */
145         while (((conn_fd = accept(list_fd, NULL, NULL)) < 0) 
146                && (errno == EINTR)); 
147         if ( conn_fd < 0) {
148             PrintErr("accept error");
149             exit(1);
150         }
151         /* fork to handle connection */
152         if ( (pid = fork()) < 0 ){
153             PrintErr("fork error");
154             exit(1);
155         }
156         if (pid == 0) {      /* child */
157             close(list_fd);          /* close listening socket */   
158             ServEcho(conn_fd);       /* handle echo */
159             exit(0);
160         } else {             /* parent */
161             close(conn_fd);          /* close connected socket */
162         }
163     }
164     /* normal exit, never reached */
165     exit(0);
166 }
167 /*
168  * routine to print usage info and exit
169  */
170 void usage(void) {
171     printf("Elementary echo server\n");
172     printf("Usage:\n");
173     printf("  echod [-h] \n");
174     printf("  -h           print this help\n");
175     printf("  -d           print debug info\n");
176     printf("  -i           use interactively\n");
177     exit(1);
178 }
179 /*
180  * routine to handle echo for connection
181  */
182 void ServEcho(int sockfd) {
183     char buffer[MAXLINE];
184     int nread, nwrite;
185     char debug[MAXLINE+20];
186     int size;
187     /* main loop, reading 0 char means client close connection */
188     while ( (nread = read(sockfd, buffer, MAXLINE)) != 0) {
189         nwrite = FullWrite(sockfd, buffer, nread);
190         if (debugging) {
191             buffer[nread] = 0;
192             snprintf(debug, MAXLINE+20, "Letti %d byte, %s", nread, buffer);
193             if (demonize) {          /* go daemon */
194                 syslog(LOG_DEBUG, debug);
195             } else {
196                 printf("%s", debug);
197             }
198         }
199     }
200     return;
201 }
202 /*
203  * routine to print error on stout or syslog
204  */
205 void PrintErr(char * error) {
206     if (demonize) {          /* go daemon */
207         syslog(LOG_ERR, error);
208     } else {
209         perror(error);
210     }
211     return;
212 }