home *** CD-ROM | disk | FTP | other *** search
/ Dream 52 / Amiga_Dream_52.iso / Linux / Magazine / wwwoffle-2.1.tar.gz / wwwoffle-2.1 / sockets.c < prev    next >
C/C++ Source or Header  |  1997-12-29  |  7KB  |  330 lines
  1. /***************************************
  2.   $Header: /home/amb/wwwoffle/RCS/sockets.c 2.5 1997/12/29 17:15:48 amb Exp $
  3.  
  4.   WWWOFFLE - World Wide Web Offline Explorer - Version 2.0a.
  5.   Socket manipulation routines.
  6.   ******************/ /******************
  7.   Written by Andrew M. Bishop
  8.  
  9.   This file Copyright 1996,97 Andrew M. Bishop
  10.   It may be distributed under the GNU Public License, version 2, or
  11.   any higher version.  See section COPYING of the GNU Public license
  12.   for conditions under which this file may be redistributed.
  13.   ***************************************/
  14.  
  15.  
  16. #include <stdio.h>
  17. #include <string.h>
  18.  
  19. #include <errno.h>
  20. #include <unistd.h>
  21.  
  22. #include <netdb.h>
  23. #include <sys/param.h>
  24. #include <sys/types.h>
  25. #include <sys/stat.h>
  26. #include <sys/socket.h>
  27. #include <netinet/in.h>
  28.  
  29. #include "errors.h"
  30. #include "sockets.h"
  31.  
  32.  
  33. /*++++++++++++++++++++++++++++++++++++++
  34.   Opens a socket for a client.
  35.  
  36.   int OpenClientSocket Returns the socket file descriptor.
  37.  
  38.   char* host The name of the remote host.
  39.  
  40.   int port The socket number.
  41.   ++++++++++++++++++++++++++++++++++++++*/
  42.  
  43. int OpenClientSocket(char* host, int port)
  44. {
  45.  int s;
  46.  int retval,err=0;
  47.  struct sockaddr_in server;
  48.  struct hostent* hp;
  49.  int retries=4;
  50.  
  51.  server.sin_family=AF_INET;
  52.  server.sin_port=htons((unsigned short)port);
  53.  
  54.  hp=gethostbyname(host);
  55.  if(!hp)
  56.    {errno=-1;PrintMessage(Warning,"Unknown host '%s' for server [%!s].",host);return(-1);}
  57.  memcpy((char*)&server.sin_addr,(char*)hp->h_addr,sizeof(server.sin_addr));
  58.  
  59.  do{
  60.     s=socket(PF_INET,SOCK_STREAM,0);
  61.     if(s==-1)
  62.       {PrintMessage(Warning,"Cannot create client socket [%!s].");return(-1);}
  63.  
  64.     retval=connect(s,(struct sockaddr *)&server,sizeof(server));
  65.     if(retval==-1)
  66.       {
  67.        err=errno;
  68.        if(--retries && errno==ECONNREFUSED)
  69.           PrintMessage(Inform,"Connect fail [%!s]; trying again.");
  70.        else
  71.           PrintMessage(Warning,"Connect fail [%!s].");
  72.        close(s);
  73.        s=-1;
  74.        sleep(1);
  75.       }
  76.    }
  77.  while(retval==-1 && retries && err==ECONNREFUSED);
  78.  
  79.  return(s);
  80. }
  81.  
  82.  
  83. /*++++++++++++++++++++++++++++++++++++++
  84.   Opens a socket for a server.
  85.  
  86.   int OpenServerSocket Returns a socket to be a server.
  87.  
  88.   int port The port number to use.
  89.   ++++++++++++++++++++++++++++++++++++++*/
  90.  
  91. int OpenServerSocket(int port)
  92. {
  93.  int s;
  94.  int retval;
  95.  struct sockaddr_in server;
  96.  int reuse_addr=1;
  97.  
  98.  s=socket(PF_INET,SOCK_STREAM,0);
  99.  if(s==-1)
  100.    {PrintMessage(Warning,"Cannot create server socket [%!s].");return(-1);}
  101.  
  102.  setsockopt(s,SOL_SOCKET,SO_REUSEADDR,&reuse_addr,sizeof(reuse_addr));
  103.  
  104.  server.sin_family=AF_INET;
  105.  server.sin_addr.s_addr=INADDR_ANY;
  106.  server.sin_port=htons((unsigned int)port);
  107.  
  108.  retval=bind(s,(struct sockaddr*)&server,sizeof(server));
  109.  if(retval==-1)
  110.    {PrintMessage(Warning,"Failed to bind server socket [%!s].");return(-1);}
  111.  
  112.  listen(s,4);
  113.  
  114.  return(s);
  115. }
  116.  
  117.  
  118. /*++++++++++++++++++++++++++++++++++++++
  119.   Waits for a connection on the socket.
  120.  
  121.   int AcceptConnect returns a socket with a remote connection on the end.
  122.  
  123.   int socket Specifies the socket to check for.
  124.   ++++++++++++++++++++++++++++++++++++++*/
  125.  
  126. int AcceptConnect(int socket)
  127. {
  128.  int s;
  129.  
  130.  s=accept(socket,(struct sockaddr*)0,(int*)0);
  131.  
  132.  if(s==-1)
  133.     PrintMessage(Warning,"Accept on server socket failed [%!s].");
  134.  
  135.  return(s);
  136. }
  137.  
  138.  
  139. /*++++++++++++++++++++++++++++++++++++++
  140.   Determines the hostname and port number used for a socket on the other end.
  141.  
  142.   int SocketRemoteName Returns 0 on success and -1 on failure.
  143.  
  144.   int socket Specifies the socket to check.
  145.  
  146.   char **name Returns the hostname.
  147.  
  148.   int *ip Returns the IP address.
  149.  
  150.   int *port Returns the port number.
  151.   ++++++++++++++++++++++++++++++++++++++*/
  152.  
  153. int SocketRemoteName(int socket,char **name,int *ip,int *port)
  154. {
  155.  struct sockaddr_in server;
  156.  int length=sizeof(server),retval;
  157.  static char host[MAXHOSTNAMELEN];
  158.  struct hostent* hp=NULL;
  159.  
  160.  retval=getpeername(socket,(struct sockaddr*)&server,&length);
  161.  if(retval==-1)
  162.     PrintMessage(Warning,"Failed to get socket peername [%!s].");
  163.  else
  164.    {
  165.     hp=gethostbyaddr((char*)&server.sin_addr,sizeof(server.sin_addr),AF_INET);
  166.     if(hp)
  167.        strcpy(host,hp->h_name);
  168.     else
  169.       {
  170.        unsigned char addr[4];
  171.        int i;
  172.  
  173.        for(i=0;i<4;i++)
  174.           addr[i]=(server.sin_addr.s_addr>>(8*i))%256;
  175.  
  176.        sprintf(host,"%d.%d.%d.%d",addr[0],addr[1],addr[2],addr[3]);
  177.       }
  178.  
  179.     if(name)
  180.        *name=host;
  181.     if(ip)
  182.        *ip=ntohl(server.sin_addr.s_addr);
  183.     if(port)
  184.        *port=ntohs(server.sin_port);
  185.    }
  186.  
  187.  return(retval);
  188. }
  189.  
  190.  
  191. /*++++++++++++++++++++++++++++++++++++++
  192.   Determines the hostname and port number used for a socket on this end.
  193.  
  194.   int SocketLocalName Returns 0 on success and -1 on failure.
  195.  
  196.   int socket Specifies the socket to check.
  197.  
  198.   char **name Returns the hostname.
  199.  
  200.   int *ip Returns the IP address.
  201.  
  202.   int *port Returns the port number.
  203.   ++++++++++++++++++++++++++++++++++++++*/
  204.  
  205. int SocketLocalName(int socket,char **name,int *ip,int *port)
  206. {
  207.  struct sockaddr_in server;
  208.  int length=sizeof(server),retval;
  209.  static char host[MAXHOSTNAMELEN];
  210.  struct hostent* hp=NULL;
  211.  
  212.  retval=getsockname(socket,(struct sockaddr*)&server,&length);
  213.  if(retval==-1)
  214.     PrintMessage(Warning,"Failed to get socket name [%!s].");
  215.  else
  216.    {
  217.     hp=gethostbyaddr((char*)&server.sin_addr,sizeof(server.sin_addr),AF_INET);
  218.     if(hp)
  219.        strcpy(host,hp->h_name);
  220.     else
  221.       {
  222.        unsigned char addr[4];
  223.        int i;
  224.  
  225.        for(i=0;i<4;i++)
  226.           addr[i]=(server.sin_addr.s_addr>>(8*i))%256;
  227.  
  228.        sprintf(host,"%d.%d.%d.%d",addr[0],addr[1],addr[2],addr[3]);
  229.       }
  230.  
  231.     if(name)
  232.        *name=host;
  233.     if(ip)
  234.        *ip=ntohl(server.sin_addr.s_addr);
  235.     if(port)
  236.        *port=ntohs(server.sin_port);
  237.    }
  238.  
  239.  return(retval);
  240. }
  241.  
  242.  
  243. /*++++++++++++++++++++++++++++++++++++++
  244.   Closes a previously opened socket.
  245.  
  246.   int CloseSocket Returns 0 on success, -1 on error.
  247.  
  248.   int socket The socket to close
  249.   ++++++++++++++++++++++++++++++++++++++*/
  250.  
  251. int CloseSocket(int socket)
  252. {
  253.  int retval=close(socket);
  254.  
  255.  if(retval==-1)
  256.     PrintMessage(Warning,"Socket close failed [%!s].");
  257.  
  258.  return(retval);
  259. }
  260.  
  261.  
  262. /*++++++++++++++++++++++++++++++++++++++
  263.   Get the Fully Qualified Domain Name for the local host.
  264.  
  265.   char *GetFQDN Return the FQDN.
  266.   ++++++++++++++++++++++++++++++++++++++*/
  267.  
  268. char *GetFQDN(void)
  269. {
  270.  static char fqdn[256],**h;
  271.  struct hostent* hp;
  272.  int type=hp->h_addrtype;
  273.  int length=hp->h_length;
  274.  char addr[sizeof(struct in_addr)];
  275.  
  276.  /* Try gethostname(). */
  277.  
  278.  if(gethostname(fqdn,255)==-1)
  279.    {PrintMessage(Warning,"Failed to get hostname [%!s].");return(NULL);}
  280.  
  281.  if(strchr(fqdn,'.'))
  282.     return(fqdn);
  283.  
  284.  /* Try gethostbyname(). */
  285.  
  286.  hp=gethostbyname(fqdn);
  287.  if(!hp)
  288.    {errno=-1;PrintMessage(Warning,"Can't get IP address for host [%!s].");return(NULL);}
  289.  
  290.  if(strchr(hp->h_name,'.'))
  291.    {
  292.     strcpy(fqdn,hp->h_name);
  293.     return(fqdn);
  294.    }
  295.  
  296.  for(h=hp->h_aliases;*h;h++)
  297.     if(strchr(*h,'.'))
  298.       {
  299.        strcpy(fqdn,*h);
  300.        return(fqdn);
  301.       }
  302.  
  303.  /* Try gethostbyaddr(). */
  304.  
  305.  type=hp->h_addrtype;
  306.  length=hp->h_length;
  307.  memcpy(addr,(char*)hp->h_addr,sizeof(struct in_addr));
  308.  
  309.  hp=gethostbyaddr(addr,length,type);
  310.  if(!hp)
  311.    {errno=-1;PrintMessage(Warning,"Can't get hostname for IP address [%!s].");return(NULL);}
  312.  
  313.  if(strchr(hp->h_name,'.'))
  314.    {
  315.     strcpy(fqdn,hp->h_name);
  316.     return(fqdn);
  317.    }
  318.  
  319.  for(h=hp->h_aliases;*h;h++)
  320.     if(strchr(*h,'.'))
  321.       {
  322.        strcpy(fqdn,*h);
  323.        return(fqdn);
  324.       }
  325.  
  326.  strcpy(fqdn,hp->h_name);
  327.  
  328.  return(fqdn);
  329. }
  330.