net.c

#include <stdlib.h>
#include <assert.h>
#include <string.h>
#include "error.h"
#include "net.h"

const struct net_param net_params[] = {
#ifdef USEUNIXSOCK
        { STR_UNIX,  sizeof(STR_UNIX) -1, STR_NAME_UNIX, AF_UNIX,  SOCK_STREAM, IPPROTO_IP   }, 
#endif
#ifdef USETCP4  
        { STR_TCP4,  sizeof(STR_TCP4) -1, STR_NAME_TCP, AF_INET,  SOCK_STREAM,  IPPROTO_TCP  }, 
#endif
#ifdef USEUDP4  
        { STR_UDP4,  sizeof(STR_UDP4) -1, STR_NAME_UDP, AF_INET,  SOCK_DGRAM,   IPPROTO_UDP  }, 
#endif
#ifdef USESCTP4 
        { STR_SCTP4, sizeof(STR_SCTP4)-1, STR_NAME_SCTP, AF_INET,  SOCK_STREAM, IPPROTO_SCTP }, 
#endif
#ifdef USETCP6  
        { STR_TCP6,  sizeof(STR_TCP6) -1, STR_NAME_TCP, AF_INET6, SOCK_STREAM,  IPPROTO_TCP  },
#endif
#ifdef USEUDP6  
        { STR_UDP6,  sizeof(STR_UDP6) -1, STR_NAME_UDP, AF_INET6, SOCK_DGRAM,   IPPROTO_UDP  }, 
#endif
#ifdef USESCTP6 
        { STR_SCTP6, sizeof(STR_SCTP6)-1, STR_NAME_SCTP, AF_INET6, SOCK_STREAM, IPPROTO_SCTP }, 
#endif
        { NULL,      0,                   NULL,          -1,       -1,          -1 }
};

int parsetarget(char* target, struct hostent** host, int* port)
{
int proto = -1;
int index = 0;
while ( net_params[index].m_str )
    {
    if ( !memcmp( target, net_params[index].m_str, net_params[index].m_strlength ) )
         {
         proto = index;
         target += net_params[index].m_strlength;
         break;
         }
    ++index;
    }
if ( proto == -1 )
    {
    fprintf( stderr, "protocol not specified\n" );
    exit( EXIT_FAILURE );
    }

char endchar = ( *target == '[' )? ']' : ':';
if ( *target == '[' )
    ++target;

char* hostend;
for (hostend = target; *hostend && *hostend != endchar; ++hostend)
    ;


if ( !*hostend )
    {
#ifdef USEUNIXSOCK
    if ( net_params[index].m_str == STR_UNIX )
        {
        *host = (struct hostent*)malloc( sizeof(struct hostent) );
        if ( *host == NULL )
            {
            error_malloc( errno );
            exit( EXIT_FAILURE );
            }

        memset( *host, 0, sizeof(struct hostent) );
        (*host)->h_length = strlen( target );
        (*host)->h_addr_list = (char**)malloc( sizeof(char*) );
        if ( (*host)->h_addr_list == NULL )
            {
            error_malloc( errno );
            exit( EXIT_FAILURE );
            }

        *((*host)->h_addr_list) = strdup( target );
        return proto;
        }
    else
#endif
        {
        fprintf( stderr, "port not specified\n" );
        exit( EXIT_FAILURE );
        }
    }
int lenght = hostend - target;
char* hostname = (char*)malloc( lenght + 1 );
if ( hostname == NULL )
    {
    error_malloc( errno );
    exit( EXIT_FAILURE );
    }

memcpy( hostname, target, lenght );
hostname[lenght] = 0;
*host = gethostbyname2( hostname, net_params[index].m_domain );
free( hostname );

if ( *hostend == ']' )
    ++hostend;
++hostend;
struct servent* serv = getservbyname( hostend, net_params[index].m_protoname );
if ( serv )
    *port = serv->s_port;
else
    *port = htons( atoi( hostend ) );
return proto;
}