Piccole modifiche
[gapil.git] / sources / TCP_echo_second.c
1 /* TCP_echo_second.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 TCP_echo_second.c
22  * Simple TCP client for echo service (port 7)
23  * This version handle server closing connection
24  *
25  * Author: Simone Piccardi
26  * Jun. 2001
27  *
28  * Usage: echo -h give all info's
29  *
30  ****************************************************************/
31 /* 
32  * Include needed headers
33  */
34 #include <sys/types.h>   /* primitive system data types */
35 #include <unistd.h>      /* unix standard library */
36 #include <arpa/inet.h>   /* IP addresses conversion utilities */
37 #include <sys/socket.h>  /* socket constants, types and functions */
38 #include <stdio.h>       /* standard I/O library */
39 #include <errno.h>       /* error definitions and routines */
40 #include <string.h>      /* C strings library */
41
42 #define MAXLINE 256
43 void usage(void);
44 void ClientEcho(FILE * filein, int socket);
45 void SigTERM_hand(int sig);
46
47 /* Program begin */
48 int main(int argc, char *argv[])
49 {
50 /* 
51  * Variables definition  
52  */
53     int sock, i;
54     int reset = 0;
55     struct sockaddr_in serv_add;
56     struct linger ling;
57     /*
58      * Input section: decode parameters passed in the calling 
59      * Use getopt function
60      */
61     opterr = 0;  /* don't want writing to stderr */
62     while ( (i = getopt(argc, argv, "hr")) != -1) {
63         switch (i) {
64         /* 
65          * Handling options 
66          */ 
67         case 'h':  
68             printf("Wrong -h option use\n");
69             usage();
70             return(1);
71             break;
72         case 'r':
73             reset = 1;
74             break;
75         case '?':   /* unrecognized options */
76             printf("Unrecognized options -%c\n",optopt);
77             usage();
78         default:    /* should not reached */
79             usage();
80         }
81     }
82     /* ***********************************************************
83      * 
84      *           Options processing completed
85      *
86      *                Main code beginning
87      * 
88      * ***********************************************************/
89     /* create socket */
90     if ( (sock = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
91         perror("Socket creation error");
92         return 1;
93     }
94     /* initialize address */
95     memset((void *) &serv_add, 0, sizeof(serv_add)); /* clear server address */
96     serv_add.sin_family = AF_INET;                   /* address type is INET */
97     serv_add.sin_port = htons(7);                    /* echo port is 7 */
98     /* build address using inet_pton */
99     if ( (inet_pton(AF_INET, argv[optind], &serv_add.sin_addr)) <= 0) {
100         perror("Address creation error");
101         return 1;
102     }
103     /* extablish connection */
104     if (connect(sock, (struct sockaddr *)&serv_add, sizeof(serv_add)) < 0) {
105         perror("Connection error");
106         return 1;
107     }
108     /* check if resetting on close is required */
109     if (reset) {
110         printf("Setting reset on close \n");
111         ling.l_onoff = 1;
112         ling.l_linger = 0;      
113         if (setsockopt(sock, SOL_SOCKET, SO_LINGER, &ling, sizeof(ling))) {
114             perror("Cannot set linger");
115             exit(1);
116         }
117     }
118     /* do read/write operations */
119     ClientEcho(stdin, sock);
120     /* normal exit */
121     return 0;
122 }
123 /*
124  * routine to print usage info and exit 
125  */
126 void usage(void) {
127     printf("Take daytime from a remote host \n");
128     printf("Usage:\n");
129     printf("  daytime [-h] [-v] [host in dotted decimal form] \n");
130 //    printf("  -v         set verbosity on\n");
131     printf("  -r           require reset on closing\n");
132     printf("  -h           print this help\n");
133     exit(1);
134 }
135
136 void ClientEcho(FILE * filein, int socket) 
137 {
138     char sendbuff[MAXLINE+1], recvbuff[MAXLINE+1];
139     int nread, nwrite; 
140     while (fgets(sendbuff, MAXLINE, filein) != NULL) {
141         nwrite = FullWrite(socket, sendbuff, strlen(sendbuff)); 
142         if (nwrite < 0) {
143             printf("Errore in scrittura: %s", strerror(errno));
144             return;
145         }
146         nread = read(socket, recvbuff, strlen(sendbuff));
147         if (nread < 0) {
148             printf("Errore in lettura: %s\n", strerror(errno));
149             return;
150         }
151         if (nread == 0) {
152             printf("EOF sul socket\n");
153             return;
154         }
155         recvbuff[nread] = 0;
156         if (fputs(recvbuff, stdout) == EOF) {
157             perror("Errore in scrittura su terminale");
158             return;
159         }
160     }
161     return;
162 }