From 2b1c967df2e4c7d72571b3f3eda5a7306e2d3913 Mon Sep 17 00:00:00 2001 From: Christian Hesse Date: Tue, 2 May 2017 21:27:30 +0200 Subject: (re-)add support for IPv6 --- pacredir.c | 63 ++++++++++++++++++++++++++++++++++++-------------------------- pacredir.h | 8 +++++--- 2 files changed, 42 insertions(+), 29 deletions(-) diff --git a/pacredir.c b/pacredir.c index 3f2939a..9fc08fd 100644 --- a/pacredir.c +++ b/pacredir.c @@ -46,7 +46,7 @@ char * get_fqdn(const char * hostname, const char * domainname) { } /*** get_url ***/ -char * get_url(const char * hostname, const char * address, const uint16_t port, const char * uri) { +char * get_url(const char * hostname, AvahiProtocol proto, const char * address, const uint16_t port, const char * uri) { const char * host; char * url; @@ -55,23 +55,28 @@ char * get_url(const char * hostname, const char * address, const uint16_t port, url = malloc(10 /* static chars of an url & null char */ + strlen(host) + 5 /* max strlen of decimal 16bit value */ + + 2 /* square brackets for IPv6 address */ + strlen(uri)); - sprintf(url, "http://%s:%d/%s", host, port, uri); + + if (*address != 0 && proto == AVAHI_PROTO_INET6) + sprintf(url, "http://[%s]:%d/%s", address, port, uri); + else + sprintf(url, "http://%s:%d/%s", host, port, uri); return url; } /*** add_host ***/ -int add_host(const char * host, const char * address, const uint16_t port, const char * type) { +int add_host(const char * host, AvahiProtocol proto, const char * address, const uint16_t port, const char * type) { struct hosts * tmphosts = hosts; struct request request; while (tmphosts->host != NULL) { - if (strcmp(tmphosts->host, host) == 0) { + if (strcmp(tmphosts->host, host) == 0 && tmphosts->proto == proto) { /* host already exists */ if (verbose > 0) - write_log(stdout, "Updating service %s (port %d) on host %s\n", - type, port, host); + write_log(stdout, "Updating service %s (port %d) on host %s (%s)\n", + type, port, host, avahi_proto_to_string(proto)); goto update; } tmphosts = tmphosts->next; @@ -79,10 +84,12 @@ int add_host(const char * host, const char * address, const uint16_t port, const /* host not found, adding a new one */ if (verbose > 0) - write_log(stdout, "Adding host %s with service %s (port %d)\n", - host, type, port); + write_log(stdout, "Adding host %s (%s) with service %s (port %d)\n", + host, avahi_proto_to_string(proto), type, port); tmphosts->host = strdup(host); + tmphosts->proto = AVAHI_PROTO_UNSPEC; + *tmphosts->address = 0; tmphosts->pacserve.port = 0; tmphosts->pacserve.online = 0; @@ -99,10 +106,9 @@ int add_host(const char * host, const char * address, const uint16_t port, const tmphosts->next->next = NULL; update: + tmphosts->proto = proto; if (address != NULL) memcpy(tmphosts->address, address, AVAHI_ADDRESS_STR_MAX); - else - memset(tmphosts->address, 0, AVAHI_ADDRESS_STR_MAX); if (strcmp(type, PACSERVE) == 0) { tmphosts->pacserve.online = 1; @@ -116,7 +122,7 @@ update: /* do a first request and let get_http_code() set the bad status */ request.host = tmphosts->host; - request.url = get_url(request.host, tmphosts->address, request.service->port, ""); + request.url = get_url(request.host, tmphosts->proto, tmphosts->address, request.service->port, ""); request.http_code = 0; request.last_modified = 0; get_http_code(&request); @@ -126,14 +132,14 @@ update: } /*** remove_host ***/ -int remove_host(const char * host, const char * type) { +int remove_host(const char * host, AvahiProtocol proto, const char * type) { struct hosts * tmphosts = hosts; while (tmphosts->host != NULL) { - if (strcmp(tmphosts->host, host) == 0) { + if (strcmp(tmphosts->host, host) == 0 && tmphosts->proto == proto) { if (verbose > 0) - write_log(stdout, "Marking service %s on host %s offline\n", - type, host); + write_log(stdout, "Marking service %s on host %s (%s) offline\n", + type, host, avahi_proto_to_string(proto)); if (strcmp(type, PACSERVE) == 0) { tmphosts->pacserve.online = 0; } else if (strcmp(type, PACDBSERVE) == 0) { @@ -182,7 +188,7 @@ static void resolve_callback(AvahiServiceResolver *r, write_log(stdout, "Found service %s on host %s (%s) on interface %s\n", type, host, ipaddress, intname); - add_host(host, ipaddress, strcmp(type, PACSERVE) == 0 ? PORT_PACSERVE : PORT_PACDBSERVE, type); + add_host(host, protocol, ipaddress, strcmp(type, PACSERVE) == 0 ? PORT_PACSERVE : PORT_PACDBSERVE, type); break; } @@ -233,7 +239,7 @@ static void browse_callback(AvahiServiceBrowser *b, tmp_ignore_interfaces = tmp_ignore_interfaces->next; } - if ((avahi_service_resolver_new(c, interface, protocol, name, type, domain, AVAHI_PROTO_INET, 0, resolve_callback, c)) == NULL) + if ((avahi_service_resolver_new(c, interface, protocol, name, type, domain, protocol, 0, resolve_callback, c)) == NULL) write_log(stderr, "Failed to create resolver for service '%s' of type '%s' in domain '%s': %s\n", name, type, domain, avahi_strerror(avahi_client_errno(c))); out: @@ -248,7 +254,7 @@ out: write_log(stdout, "Service %s on host %s disappeared\n", type, host); - remove_host(host, type); + remove_host(host, protocol, type); free(host); @@ -363,7 +369,7 @@ static int ahc_echo(void * cls, struct hosts * tmphosts = hosts; char * url = NULL, * page; - const char * basename; + const char * basename, * host = NULL; struct timeval tv; struct stat fst; @@ -408,6 +414,7 @@ static int ahc_echo(void * cls, http_code = MHD_HTTP_OK; /* duplicate string so we can free it later */ url = strdup(WEBSITE); + host = "project site"; goto response; } @@ -481,7 +488,7 @@ static int ahc_echo(void * cls, request->service = &(tmphosts->pacdbserve); else request->service = &(tmphosts->pacserve); - request->url = get_url(tmphosts->host, tmphosts->address, request->service->port, basename); + request->url = get_url(tmphosts->host, tmphosts->proto, tmphosts->address, request->service->port, basename); request->http_code = 0; request->last_modified = 0; @@ -528,6 +535,7 @@ static int ahc_echo(void * cls, if (url != NULL) free(url); url = request->url; + host = request->host; http_code = MHD_HTTP_OK; last_modified = request->last_modified; time_total = request->time_total; @@ -539,7 +547,7 @@ static int ahc_echo(void * cls, response: /* give response */ if (http_code == MHD_HTTP_OK) { - write_log(stdout, "Redirecting to %s\n", url); + write_log(stdout, "Redirecting to %s: %s\n", host, url); page = malloc(strlen(PAGE307) + strlen(url) + strlen(basename) + 1); sprintf(page, PAGE307, url, basename); response = MHD_create_response_from_buffer(strlen(page), (void*) page, MHD_RESPMEM_PERSISTENT); @@ -704,7 +712,7 @@ int main(int argc, char ** argv) { *strchr(value, ':') = 0; } else port = PORT_PACSERVE; - add_host(value, NULL, port, PACSERVE); + add_host(value, AVAHI_PROTO_UNSPEC, NULL, port, PACSERVE); value = strtok(NULL, DELIMITER); } free(values); @@ -723,7 +731,7 @@ int main(int argc, char ** argv) { *strchr(value, ':') = 0; } else port = PORT_PACDBSERVE; - add_host(value, NULL, port, PACDBSERVE); + add_host(value, AVAHI_PROTO_UNSPEC, NULL, port, PACDBSERVE); value = strtok(NULL, DELIMITER); } free(values); @@ -746,13 +754,15 @@ int main(int argc, char ** argv) { } /* create the service browser for PACSERVE */ - if ((pacserve = avahi_service_browser_new(client, AVAHI_IF_UNSPEC, AVAHI_PROTO_INET, PACSERVE, NULL, 0, browse_callback, client)) == NULL) { + if ((pacserve = avahi_service_browser_new(client, AVAHI_IF_UNSPEC, + AVAHI_PROTO_UNSPEC, PACSERVE, NULL, 0, browse_callback, client)) == NULL) { write_log(stderr, "Failed to create service browser: %s\n", avahi_strerror(avahi_client_errno(client))); goto fail; } /* create the service browser for PACDBSERVE */ - if ((pacdbserve = avahi_service_browser_new(client, AVAHI_IF_UNSPEC, AVAHI_PROTO_INET, PACDBSERVE, NULL, 0, browse_callback, client)) == NULL) { + if ((pacdbserve = avahi_service_browser_new(client, AVAHI_IF_UNSPEC, + AVAHI_PROTO_UNSPEC, PACDBSERVE, NULL, 0, browse_callback, client)) == NULL) { write_log(stderr, "Failed to create service browser: %s\n", avahi_strerror(avahi_client_errno(client))); goto fail; } @@ -763,7 +773,8 @@ int main(int argc, char ** argv) { inet_pton(AF_INET, "127.0.0.1", &address.sin_addr); /* start http server */ - if ((mhd = MHD_start_daemon(MHD_USE_THREAD_PER_CONNECTION, PORT_PACREDIR, NULL, NULL, &ahc_echo, NULL, MHD_OPTION_SOCK_ADDR, &address, MHD_OPTION_END)) == NULL) { + if ((mhd = MHD_start_daemon(MHD_USE_THREAD_PER_CONNECTION, PORT_PACREDIR, + NULL, NULL, &ahc_echo, NULL, MHD_OPTION_SOCK_ADDR, &address, MHD_OPTION_END)) == NULL) { write_log(stderr, "Could not start daemon on port %d.\n", PORT_PACREDIR); return EXIT_FAILURE; } diff --git a/pacredir.h b/pacredir.h index 2cd8cca..11f890b 100644 --- a/pacredir.h +++ b/pacredir.h @@ -61,6 +61,8 @@ struct services { struct hosts { /* host name */ char * host; + /* protocol (AVAHI_PROTO_INET, AVAHI_PROTO_INET6 or AVAHI_PROTO_UNSPEC) */ + AvahiProtocol proto; /* resolved address */ char address[AVAHI_ADDRESS_STR_MAX]; /* online status and bad time for services */ @@ -99,12 +101,12 @@ int write_log(FILE *stream, const char *format, ...); /* get_fqdn */ char * get_fqdn(const char * hostname, const char * domainname); /* get_url */ -char * get_url(const char * hostname, const char * address, const uint16_t port, const char * uri); +char * get_url(const char * hostname, AvahiProtocol proto, const char * address, const uint16_t port, const char * uri); /* add_host */ -int add_host(const char * host, const char * address, const uint16_t port, const char * type); +int add_host(const char * host, AvahiProtocol proto, const char * address, const uint16_t port, const char * type); /* remove_host */ -int remove_host(const char * host, const char * type); +int remove_host(const char * host, AvahiProtocol proto, const char * type); /* resolve_callback */ static void resolve_callback(AvahiServiceResolver *r, -- cgit v1.2.3-70-g09d2