/* * (C) 2013 by Christian Hesse * * This software may be used and distributed according to the terms * of the GNU General Public License, incorporated herein by reference. */ #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include "config.h" /* services */ struct services { /* true if host/service is online */ uint8_t online; /* unix timestamp of last bad request */ __time_t bad; }; /* hosts */ struct hosts { /* host name */ char * host; /* online status and bad time for services */ struct services pacserve; struct services pacdbserve; /* pointer to next struct element */ struct hosts * next; }; /* global variables */ struct hosts * hosts = NULL; static AvahiSimplePoll *simple_poll = NULL; /*** get_fqdn ***/ char * get_fqdn(const char * hostname, const char * domainname) { char * name; name = malloc(strlen(hostname) + strlen(domainname) + 2 /* '.' and null char */); sprintf(name, "%s.%s", hostname, domainname); return name; } /*** browse_callback *** * Called whenever a new services becomes available on the LAN or is removed from the LAN */ static void browse_callback(AvahiServiceBrowser *b, AvahiIfIndex interface, AvahiProtocol protocol, AvahiBrowserEvent event, const char *name, const char *type, const char *domain, AVAHI_GCC_UNUSED AvahiLookupResultFlags flags, void* userdata) { struct hosts * tmphosts = hosts; char * host; assert(b); switch (event) { case AVAHI_BROWSER_FAILURE: fprintf(stderr, "%s\n", avahi_strerror(avahi_client_errno(avahi_service_browser_get_client(b)))); avahi_simple_poll_quit(simple_poll); return; case AVAHI_BROWSER_NEW: host = get_fqdn(name, domain); # if defined DEBUG fprintf(stderr, "NEW: service '%s' of type '%s' in domain '%s'\n", name, type, domain); # endif if (flags & AVAHI_LOOKUP_RESULT_LOCAL) goto out; while (tmphosts->host != NULL) { if (strcmp(tmphosts->host, host) == 0) { /* host already exists */ printf("Updating service %s on %s\n", type, host); goto update; } tmphosts = tmphosts->next; } /* host not found, adding a new one */ printf("Adding host %s with service %s\n", host, type); tmphosts->host = strdup(host); tmphosts->pacserve.online = 0; tmphosts->pacserve.bad = 0; tmphosts->pacdbserve.online = 0; tmphosts->pacdbserve.bad = 0; tmphosts->next = realloc(tmphosts->next, sizeof(struct hosts)); tmphosts->next->host = NULL; tmphosts->next->next = NULL; update: if (strcmp(type, PACSERVE) == 0) { tmphosts->pacserve.online = 1; tmphosts->pacserve.bad = 0; } else if (strcmp(type, PACDBSERVE) == 0) { tmphosts->pacdbserve.online = 1; tmphosts->pacdbserve.bad = 0; } out: free(host); break; case AVAHI_BROWSER_REMOVE: host = get_fqdn(name, domain); # if defined DEBUG fprintf(stderr, "REMOVE: service '%s' of type '%s' in domain '%s'\n", name, type, domain); # endif while (tmphosts->host != NULL) { if (strcmp(tmphosts->host, host) == 0) { printf("Marking service %s on host %s offline\n", type, host); if (strcmp(type, PACSERVE) == 0) { tmphosts->pacserve.online = 0; } else if (strcmp(type, PACDBSERVE) == 0) { tmphosts->pacdbserve.online = 0; } break; } tmphosts = tmphosts->next; } free(host); break; case AVAHI_BROWSER_ALL_FOR_NOW: case AVAHI_BROWSER_CACHE_EXHAUSTED: break; } } /*** client_callback ***/ static void client_callback(AvahiClient *c, AvahiClientState state, AVAHI_GCC_UNUSED void * userdata) { assert(c); if (state == AVAHI_CLIENT_FAILURE) { fprintf(stderr, "Server connection failure: %s\n", avahi_strerror(avahi_client_errno(c))); avahi_simple_poll_quit(simple_poll); } } /*** get_http_code ***/ int get_http_code(const char * host, const uint16_t port, const char * url, int * http_code, int * last_modified) { CURL *curl; CURLcode res; curl_global_init(CURL_GLOBAL_ALL); if ((curl = curl_easy_init()) != NULL) { curl_easy_setopt(curl, CURLOPT_URL, url); /* example.com is redirected, so we tell libcurl to follow redirection */ curl_easy_setopt(curl, CURLOPT_FOLLOWLOCATION, 1L); /* set user agent */ curl_easy_setopt(curl, CURLOPT_USERAGENT, "pacredir/" VERSION); /* do not receive body */ curl_easy_setopt(curl, CURLOPT_NOBODY, 1); /* ask for filetime */ curl_easy_setopt(curl, CURLOPT_FILETIME, 1); /* set connection timeout to 2 seconds * if the host needs longer we do not want to use it anyway ;) */ curl_easy_setopt(curl, CURLOPT_CONNECTTIMEOUT, 2); /* perform the request */ if (curl_easy_perform(curl) != CURLE_OK) { fprintf(stderr, "Could not connect to server %s on port %d.\n", host, port); *http_code = 0; *last_modified = 0; return EXIT_FAILURE; } /* get http code */ if ((res = curl_easy_getinfo(curl, CURLINFO_RESPONSE_CODE, http_code)) != CURLE_OK) { fprintf(stderr, "curl_easy_getinfo() failed: %s\n", curl_easy_strerror(res)); return EXIT_FAILURE; } /* get last modified time */ if (*http_code == MHD_HTTP_OK) { if ((res = curl_easy_getinfo(curl, CURLINFO_FILETIME, last_modified)) != CURLE_OK) { fprintf(stderr, "curl_easy_getinfo() failed: %s\n", curl_easy_strerror(res)); return EXIT_FAILURE; } } else *last_modified = 0; /* always cleanup */ curl_easy_cleanup(curl); } /* we're done with libcurl, so clean it up */ curl_global_cleanup(); return EXIT_SUCCESS; } /*** ahc_echo *** * called whenever a http request is received */ static int ahc_echo(void * cls, struct MHD_Connection * connection, const char * uri, const char * method, const char * version, const char * upload_data, size_t * upload_data_size, void ** ptr) { static int dummy; struct MHD_Response * response; int ret; struct hosts * tmphosts = hosts; char * url = NULL, * url_recent = NULL, * page; const char * basename; struct timeval tv; struct stat fst; char * filename; int http_code, recent = 0; int last_modified, last_modified_recent = 0; /* we want to filename, not the path */ basename = uri; while (strstr(basename, "/") != NULL) basename = strstr(basename, "/") + 1; if (strcmp(method, "GET") != 0) return MHD_NO; /* unexpected method */ if (&dummy != *ptr) { /* The first time only the headers are valid, * do not respond in the first round... */ *ptr = &dummy; return MHD_YES; } if (*upload_data_size != 0) return MHD_NO; /* upload data in a GET!? */ /* clear context pointer */ *ptr = NULL; /* process db file request */ if (strlen(basename) > 3 && strcmp(basename + strlen(basename) - 3, ".db") == 0) { /* get timestamp of local file */ filename = malloc(strlen(SYNCPATH) + strlen(basename) + 2); sprintf(filename, SYNCPATH "/%s", basename); bzero(&fst, sizeof(fst)); if (stat(filename, &fst) != 0) printf("stat() failed\n"); else last_modified_recent = fst.st_mtime; free(filename); /* try to find a server with most recent file */ while (tmphosts->host != NULL) { gettimeofday(&tv, NULL); /* skip host if offline or had a bad request within last BADTIME seconds */ if (tmphosts->pacdbserve.online == 0 || tmphosts->pacdbserve.bad + BADTIME > tv.tv_sec) { tmphosts = tmphosts->next; continue; } url = realloc(url, 10 + strlen(tmphosts->host) + (log10(PORT_PACDBSERVE)+1) + strlen(basename)); sprintf(url, "http://%s:%d/%s", tmphosts->host, PORT_PACDBSERVE, basename); printf("Trying %s\n", url); if (get_http_code(tmphosts->host, PORT_PACDBSERVE, url, &http_code, &last_modified) == EXIT_FAILURE) tmphosts->pacdbserve.bad = tv.tv_sec; else if (http_code == MHD_HTTP_OK && last_modified > last_modified_recent) { if (recent > 0) free(url_recent); last_modified_recent = last_modified; url_recent = url; url = NULL; recent++; } tmphosts = tmphosts->next; } if (url != NULL) { free(url); url = NULL; } if (recent > 0) { http_code = MHD_HTTP_OK; url = url_recent; } else http_code = 0; /* process package file request */ } else { /* try to find a server */ while (tmphosts->host != NULL) { gettimeofday(&tv, NULL); /* skip host if offline or had a bad request within last BADTIME seconds */ if (tmphosts->pacserve.online == 0 || tmphosts->pacserve.bad + BADTIME > tv.tv_sec) { tmphosts = tmphosts->next; continue; } url = realloc(url, 10 + strlen(tmphosts->host) + (log10(PORT_PACSERVE)+1) + strlen(basename)); sprintf(url, "http://%s:%d/%s", tmphosts->host, PORT_PACSERVE, basename); printf("Trying %s\n", url); if (get_http_code(tmphosts->host, PORT_PACSERVE, url, &http_code, &last_modified) == EXIT_FAILURE) tmphosts->pacserve.bad = tv.tv_sec; else if (http_code == MHD_HTTP_OK) break; tmphosts = tmphosts->next; } } /* give response */ if (http_code == MHD_HTTP_OK) { printf("Redirecting to %s\n", url); page = malloc(strlen(PAGE307) + strlen(url) + strlen(basename) + 1); sprintf(page, PAGE307, url, basename + 1); response = MHD_create_response_from_data(strlen(page), (void*) page, MHD_NO, MHD_NO); ret = MHD_add_response_header(response, "Location", url); ret = MHD_queue_response(connection, MHD_HTTP_TEMPORARY_REDIRECT, response); } else { printf("File %s not found, giving up.\n", basename); page = malloc(strlen(PAGE404) + strlen(basename) + 1); sprintf(page, PAGE404, basename + 1); response = MHD_create_response_from_data(strlen(page), (void*) page, MHD_NO, MHD_NO); ret = MHD_queue_response(connection, MHD_HTTP_NOT_FOUND, response); } MHD_destroy_response(response); free(page); if (url != NULL) free(url); return ret; } /*** sigterm_callback ***/ void sigterm_callback(int signal) { avahi_simple_poll_quit(simple_poll); } /*** sighup_callback ***/ void sighup_callback(int signal) { struct hosts * tmphosts = hosts; printf("Received SIGHUP, marking all hosts offline.\n"); while (tmphosts->host != NULL) { tmphosts->pacserve.online = 0; tmphosts->pacdbserve.online = 0; tmphosts = tmphosts->next; } } /*** main ***/ int main(int argc, char ** argv) { AvahiClient *client = NULL; AvahiServiceBrowser *pacserve = NULL, *pacdbserve = NULL; int error; int ret = 1; struct MHD_Daemon * mhd; struct hosts * tmphosts; struct sockaddr_in address; printf("Starting pacredir/" VERSION "\n"); /* allocate first struct element as dummy */ hosts = malloc(sizeof(struct hosts)); hosts->host = NULL; hosts->pacserve.online = 0; hosts->pacserve.bad = 0; hosts->pacdbserve.online = 0; hosts->pacdbserve.bad = 0; hosts->next = NULL; /* allocate main loop object */ if (!(simple_poll = avahi_simple_poll_new())) { fprintf(stderr, "Failed to create simple poll object.\n"); goto fail; } /* allocate a new client */ client = avahi_client_new(avahi_simple_poll_get(simple_poll), 0, client_callback, NULL, &error); /* check wether creating the client object succeeded */ if (!client) { fprintf(stderr, "Failed to create client: %s\n", avahi_strerror(error)); goto fail; } /* create the service browser for PACSERVE */ if ((pacserve = avahi_service_browser_new(client, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, PACSERVE, NULL, 0, browse_callback, client)) == NULL) { fprintf(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_UNSPEC, PACDBSERVE, NULL, 0, browse_callback, client)) == NULL) { fprintf(stderr, "Failed to create service browser: %s\n", avahi_strerror(avahi_client_errno(client))); goto fail; } /* prepare struct to make microhttpd listen on localhost only */ address.sin_family = AF_INET; address.sin_port = htons(PORT_PACREDIR); 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) { fprintf(stderr, "Could not start daemon on port %d.\n", PORT_PACREDIR); return EXIT_FAILURE; } /* register signal callbacks */ signal(SIGTERM, sigterm_callback); signal(SIGHUP, sighup_callback); /* run the main loop */ avahi_simple_poll_loop(simple_poll); MHD_stop_daemon(mhd); ret = EXIT_SUCCESS; fail: /* Cleanup things */ while(hosts->host != NULL) { free(hosts->host); tmphosts = hosts->next; free(hosts); hosts = tmphosts; } if (pacdbserve) avahi_service_browser_free(pacdbserve); if (pacserve) avahi_service_browser_free(pacdbserve); if (client) avahi_client_free(client); if (simple_poll) avahi_simple_poll_free(simple_poll); return ret; } // vim: set syntax=c: