1 #include "resolve.h"

     2 

     3 #include <sys/types.h>

     4 #include <sys/socket.h>

     5 #include <string.h>

     6 #include <stdio.h>

     7 

     8 err_t resolve_addr (struct resolve_result *res, const char *node, const char *service, int socktype, int ai_flags, error_t *err)

     9 {

    10     struct addrinfo hints, *ai;

    11     int ret;

    12 

    13     // build hints

    14     memset(&hints, 0, sizeof(hints));

    15     hints.ai_flags = ai_flags;

    16     hints.ai_family = AF_UNSPEC;

    17     hints.ai_socktype = socktype;

    18 

    19     // resolve (blocking)

    20     if ((ret = getaddrinfo(node, service, &hints, &ai)))

    21         RETURN_SET_ERROR_EXTRA(err, ERR_GETADDRINFO, ret);

    22 

    23     // no results?

    24     if (!ai)

    25         RETURN_SET_ERROR(err, ERR_GETADDRINFO_EMPTY);

    26     

    27     // store

    28     res->list = res->item = ai;

    29 

    30     // ok

    31     return SUCCESS;

    32 }

    33 

    34 void resolve_result_init (struct resolve_result *res)

    35 {

    36     res->list = res->item = NULL;

    37 }

    38 

    39 struct addrinfo* resolve_result_next (struct resolve_result *res)

    40 {

    41     if (!res->item) {

    42         // no items left

    43         return NULL;

    44 

    45     } else {

    46         // ...remember the current item

    47         struct addrinfo *ai = res->item;

    48 

    49         if (res->item)

    50             // advance item to the next one

    51             res->item = res->item->ai_next;

    52 

    53         // return the current one

    54         return ai;

    55     }

    56 }

    57 

    58 void resolve_result_deinit (struct resolve_result *res)

    59 {

    60     if (res->list)

    61         // free them all

    62         freeaddrinfo(res->list);

    63     

    64     // invalidate

    65     res->list = res->item = NULL;

    66 }

    67 

    68 const char * resolve_addr_text (const struct addrinfo *addr)

    69 {

    70     static char text[1024];

    71     char host[NI_MAXHOST], service[NI_MAXSERV];

    72     int ret;

    73 

    74     // lookup the reverse nameinfo

    75     if ((ret = getnameinfo(

    76                 addr->ai_addr, addr->ai_addrlen, 

    77                 host, sizeof(host), service, sizeof(service), 

    78                 NI_NUMERICHOST | NI_NUMERICSERV

    79     ))) {

    80         strcpy(host, "???");

    81         strcpy(service, "???");

    82     }

    83 

    84     // format message

    85     snprintf(text, sizeof(text), "[%s]:%s", host, service);

    86 

    87     // return static pointer

    88     return text;

    89 }