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 int main(int argc, char *argv[]) {
  30   int send_fd;
  31   struct sockaddr_in local_addr, remote_addr;
  32   socklen_t remote_addr_len;
  33   int send_bytes, read_bytes;
  34
  35   if(argc!=3) {
  36     printf("usage: %s <target-ip/broadcast-addr> <port>\n",argv[0]);
  37     exit(1);
  38   }
  39
  40   if((send_fd=socket(AF_INET,SOCK_DGRAM,0)) == -1) {
  41     printf("can't open socket.\n");
  42     exit(1);
  43   }
  44
  45   // setsockopt(send_fd,SOL_SOCKET,SO_BROADCAST,
  46
  47   memset(&local_addr,0,sizeof(local_addr));
  48   local_addr.sin_family=AF_INET;
  49   local_addr.sin_port=htons(atoi(argv[2]));
  50   local_addr.sin_addr.s_addr=htonl(INADDR_ANY);
  51
  52   if(bind(send_fd,(struct sockaddr *)&local_addr,sizeof(local_addr))==-1) {
  53     printf("unable to bind on port %d.\n",atoi(argv[1]));
  54     perror("bind");
  55     exit(1);
  56   }
  57
  58   remote_addr_len=sizeof(remote_addr);
  59   memset(&remote_addr,0,sizeof(remote_addr));
  60   remote_addr.sin_family=AF_INET;
  61   remote_addr.sin_port=htons(atoi(argv[2]));
  62   remote_addr.sin_addr.s_addr=inet_addr(argv[1]);
  63
  64   /* send stuff .... */
  65   read_bytes=1;
  66   while(read_bytes>0) {
  67     unsigned char buf[1000];
  68     read_bytes=read(0,buf,sizeof(buf));
  69     send_bytes=sendto(send_fd,buf,sizeof(buf),0,(struct sockaddr *)&remote_addr,remote_addr_len);
  70   }
  71
  72   close(send_fd);
  73   printf("connection closed ...\n");
  74   printf("%d from %d total bytes sent.\n",send_bytes,read_bytes);
  75
  76   return 0;
  77 }