fixed send(from) recv(from) bug, added dgramrcv.c
[my-code/ivac.git] / datagram.c
1 /* datagram.c - datagram sockets server
2  *
3  * author: hackbard
4  *
5  */
6
7 #include <stdio.h>
8 #include <stdlib.h>
9
10 /* memset */
11 #include <string.h>
12
13 /* socket and bind stuff */
14 #include <sys/types.h>
15 #include <sys/socket.h>
16
17 /* sockkaddr_in */
18 #include <netinet/in.h>
19
20 /* inet_ntoa ... */
21 #include <arpa/inet.h>
22
23 /* errno stuff ... */
24 #include <errno.h>
25
26 /* read, close */
27 #include <unistd.h>
28
29 /* open */
30 #include <sys/stat.h>
31 #include <fcntl.h>
32
33 int main(int argc, char *argv[]) {
34   int send_fd,cmd_fd;
35   struct sockaddr_in local_addr, remote_addr;
36   socklen_t remote_addr_len,optlen;
37   int send_bytes, read_bytes;
38
39   if(argc!=4) {
40     printf("usage: %s \"cmd command\" <target-ip/broadcast-addr> <port>\n",argv[0]);
41     exit(1);
42   }
43
44   if((send_fd=socket(AF_INET,SOCK_DGRAM,0)) == -1) {
45     printf("can't open socket.\n");
46     exit(1);
47   }
48
49   optlen=1;
50   setsockopt(send_fd,SOL_SOCKET,SO_BROADCAST,NULL,optlen);
51  
52   memset(&local_addr,0,sizeof(local_addr));
53   local_addr.sin_family=AF_INET;
54   local_addr.sin_port=htons(atoi(argv[3]));
55   local_addr.sin_addr.s_addr=htonl(INADDR_ANY);
56
57   if(bind(send_fd,(struct sockaddr *)&local_addr,sizeof(local_addr))==-1) {
58     printf("unable to bind on port %d.\n",atoi(argv[3]));
59     perror("bind");
60     exit(1);
61   }
62
63   remote_addr_len=sizeof(remote_addr); 
64   memset(&remote_addr,0,sizeof(remote_addr));
65   remote_addr.sin_family=AF_INET;
66   remote_addr.sin_port=htons(atoi(argv[3]));
67   remote_addr.sin_addr.s_addr=inet_addr(argv[2]);
68
69   if((cmd_fd=open(argv[1],O_RDONLY))<0) {
70     printf("unable to open file descriptor for %s.\n",argv[1]);
71     exit(1);
72   }
73
74   /* send stuff .... */
75   read_bytes=1;
76   while(read_bytes>0) {
77     unsigned char buf[1000];
78     read_bytes=read(cmd_fd,buf,sizeof(buf));
79     send_bytes=sendto(send_fd,buf,read_bytes,0,(struct sockaddr *)&remote_addr,remote_addr_len);
80   }
81
82   close(send_fd);
83   printf("connection closed ...\n");
84   printf("%d from %d total bytes sent.\n",send_bytes,read_bytes);
85
86   return 0;
87 }