1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486 |
- /*
- This file is part of GNUnet.
- Copyright (C) 2001-2016 GNUnet e.V.
- GNUnet is free software: you can redistribute it and/or modify it
- under the terms of the GNU Affero General Public License as published
- by the Free Software Foundation, either version 3 of the License,
- or (at your option) any later version.
- GNUnet is distributed in the hope that it will be useful, but
- WITHOUT ANY WARRANTY; without even the implied warranty of
- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- Affero General Public License for more details.
-
- You should have received a copy of the GNU Affero General Public License
- along with this program. If not, see <http://www.gnu.org/licenses/>.
- SPDX-License-Identifier: AGPL3.0-or-later
- */
- /**
- * @file peerinfo/gnunet-service-peerinfo.c
- * @brief maintains list of known peers
- *
- * Code to maintain the list of currently known hosts (in memory
- * structure of data/hosts/).
- *
- * @author Christian Grothoff
- */
- #include "platform.h"
- #include "gnunet_util_lib.h"
- #include "gnunet_hello_lib.h"
- #include "gnunet_protocols.h"
- #include "gnunet_statistics_service.h"
- #include "peerinfo.h"
- /**
- * How often do we scan the HOST_DIR for new entries?
- */
- #define DATA_HOST_FREQ GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MINUTES, 15)
- /**
- * How often do we discard old entries in data/hosts/?
- */
- #define DATA_HOST_CLEAN_FREQ GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MINUTES, 60)
- /**
- * In-memory cache of known hosts.
- */
- struct HostEntry
- {
- /**
- * Identity of the peer.
- */
- struct GNUNET_PeerIdentity identity;
- /**
- * Hello for the peer (can be NULL)
- */
- struct GNUNET_HELLO_Message *hello;
- /**
- * Friend only hello for the peer (can be NULL)
- */
- struct GNUNET_HELLO_Message *friend_only_hello;
- };
- /**
- * Result of reading a file
- */
- struct ReadHostFileContext
- {
- /**
- * Hello for the peer (can be NULL)
- */
- struct GNUNET_HELLO_Message *hello;
- /**
- * Friend only hello for the peer (can be NULL)
- */
- struct GNUNET_HELLO_Message *friend_only_hello;
- };
- /**
- * The in-memory list of known hosts, mapping of
- * host IDs to 'struct HostEntry*' values.
- */
- static struct GNUNET_CONTAINER_MultiPeerMap *hostmap;
- /**
- * Clients to immediately notify about all changes.
- */
- static struct GNUNET_NotificationContext *notify_list;
- /**
- * Clients to immediately notify about all changes,
- * even for friend-only HELLOs.
- */
- static struct GNUNET_NotificationContext *notify_friend_only_list;
- /**
- * Directory where the hellos are stored in (peerinfo/)
- */
- static char *networkIdDirectory;
- /**
- * Handle for reporting statistics.
- */
- static struct GNUNET_STATISTICS_Handle *stats;
- /**
- * Handle for task to run #cron_clean_data_hosts()
- */
- static struct GNUNET_SCHEDULER_Task *cron_clean;
- /**
- * Handle for task to run #cron_scan_directory_data_hosts()
- */
- static struct GNUNET_SCHEDULER_Task *cron_scan;
- /**
- * Notify all clients in the notify list about the
- * given host entry changing.
- *
- * @param he entry of the host for which we generate a notification
- * @param include_friend_only create public of friend-only message
- * @return generated notification message
- */
- static struct InfoMessage *
- make_info_message (const struct HostEntry *he,
- int include_friend_only)
- {
- struct InfoMessage *im;
- struct GNUNET_HELLO_Message *src;
- size_t hs;
- if (GNUNET_YES == include_friend_only)
- src = he->friend_only_hello;
- else
- src = he->hello;
- hs = (NULL == src) ? 0 : GNUNET_HELLO_size (src);
- im = GNUNET_malloc (sizeof (struct InfoMessage) + hs);
- im->header.size = htons (hs + sizeof (struct InfoMessage));
- im->header.type = htons (GNUNET_MESSAGE_TYPE_PEERINFO_INFO);
- im->peer = he->identity;
- GNUNET_memcpy (&im[1],
- src,
- hs);
- return im;
- }
- /**
- * Address iterator that causes expired entries to be discarded.
- *
- * @param cls pointer to the current time
- * @param address the address
- * @param expiration expiration time for the address
- * @return #GNUNET_NO if expiration smaller than the current time
- */
- static int
- discard_expired (void *cls,
- const struct GNUNET_HELLO_Address *address,
- struct GNUNET_TIME_Absolute expiration)
- {
- const struct GNUNET_TIME_Absolute *now = cls;
- if (now->abs_value_us > expiration.abs_value_us)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- _("Removing expired address of transport `%s'\n"),
- address->transport_name);
- return GNUNET_NO;
- }
- return GNUNET_OK;
- }
- /**
- * Address iterator that counts the remaining addresses.
- *
- * @param cls pointer to the counter
- * @param address the address
- * @param expiration expiration time for the address
- * @return #GNUNET_OK (always)
- */
- static int
- count_addresses (void *cls,
- const struct GNUNET_HELLO_Address *address,
- struct GNUNET_TIME_Absolute expiration)
- {
- unsigned int *cnt = cls;
- (void) address;
- (void) expiration;
- (*cnt)++;
- return GNUNET_OK;
- }
- /**
- * Get the filename under which we would store the GNUNET_HELLO_Message
- * for the given host and protocol.
- *
- * @param id peer for which we need the filename for the HELLO
- * @return filename of the form DIRECTORY/HOSTID
- */
- static char *
- get_host_filename (const struct GNUNET_PeerIdentity *id)
- {
- char *fn;
- if (NULL == networkIdDirectory)
- return NULL;
- GNUNET_asprintf (&fn,
- "%s%s%s",
- networkIdDirectory,
- DIR_SEPARATOR_STR,
- GNUNET_i2s_full (id));
- return fn;
- }
- /**
- * Broadcast information about the given entry to all
- * clients that care.
- *
- * @param entry entry to broadcast about
- */
- static void
- notify_all (struct HostEntry *entry)
- {
- struct InfoMessage *msg_pub;
- struct InfoMessage *msg_friend;
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Notifying all clients about peer `%s'\n",
- GNUNET_i2s(&entry->identity));
- msg_pub = make_info_message (entry,
- GNUNET_NO);
- GNUNET_notification_context_broadcast (notify_list,
- &msg_pub->header,
- GNUNET_NO);
- GNUNET_free (msg_pub);
- msg_friend = make_info_message (entry,
- GNUNET_YES);
- GNUNET_notification_context_broadcast (notify_friend_only_list,
- &msg_friend->header,
- GNUNET_NO);
- GNUNET_free (msg_friend);
- }
- /**
- * Bind a host address (hello) to a hostId.
- *
- * @param peer the peer for which this is a hello
- * @param hello the verified (!) hello message
- */
- static void
- update_hello (const struct GNUNET_PeerIdentity *peer,
- const struct GNUNET_HELLO_Message *hello);
- /**
- * Try to read the HELLOs in the given filename and discard expired
- * addresses. Removes the file if one the HELLO is malformed. If all
- * addresses are expired, the HELLO is also removed (but the HELLO
- * with the public key is still returned if it was found and valid).
- * The file can contain multiple HELLO messages.
- *
- * @param fn name of the file
- * @param unlink_garbage if #GNUNET_YES, try to remove useless files
- * @param r ReadHostFileContext to store the resutl
- */
- static void
- read_host_file (const char *fn,
- int unlink_garbage,
- struct ReadHostFileContext *r)
- {
- char buffer[GNUNET_MAX_MESSAGE_SIZE - 1] GNUNET_ALIGN;
- ssize_t size_total;
- struct GNUNET_TIME_Absolute now;
- unsigned int left;
- const struct GNUNET_HELLO_Message *hello;
- struct GNUNET_HELLO_Message *hello_clean;
- size_t read_pos;
- uint16_t size_hello;
- r->friend_only_hello = NULL;
- r->hello = NULL;
- if (GNUNET_YES != GNUNET_DISK_file_test (fn))
- return;
- size_total = GNUNET_DISK_fn_read (fn,
- buffer,
- sizeof (buffer));
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Read %d bytes from `%s'\n",
- (int) size_total,
- fn);
- if ( (size_total < 0) ||
- (((size_t) size_total) < sizeof (struct GNUNET_MessageHeader)) )
- {
- GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
- _("Failed to parse HELLO in file `%s': %s\n"),
- fn,
- "File has invalid size");
- if ( (GNUNET_YES == unlink_garbage) &&
- (0 != UNLINK (fn)) &&
- (ENOENT != errno) )
- GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_WARNING,
- "unlink",
- fn);
- return;
- }
- read_pos = 0;
- while (read_pos < (size_t) size_total)
- {
- hello = (const struct GNUNET_HELLO_Message *) &buffer[read_pos];
- size_hello = GNUNET_HELLO_size (hello);
- if ( (0 == size_hello) ||
- (((size_t) size_total) - read_pos < size_hello) )
- {
- GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
- _("Failed to parse HELLO in file `%s'\n"),
- fn);
- if (0 == read_pos)
- {
- if ((GNUNET_YES == unlink_garbage) &&
- (0 != UNLINK (fn)) &&
- (ENOENT != errno) )
- GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_WARNING,
- "unlink",
- fn);
- }
- else
- {
- if ( (GNUNET_YES == unlink_garbage) &&
- (0 != TRUNCATE (fn, read_pos)) &&
- (ENOENT != errno) )
- GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_WARNING,
- "truncate",
- fn);
- }
- return;
- }
- now = GNUNET_TIME_absolute_get ();
- hello_clean = GNUNET_HELLO_iterate_addresses (hello,
- GNUNET_YES,
- &discard_expired,
- &now);
- if (NULL == hello_clean)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
- _("Failed to parse HELLO in file `%s'\n"),
- fn);
- if ((GNUNET_YES == unlink_garbage) &&
- (0 != UNLINK (fn)) &&
- (ENOENT != errno) )
- GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_WARNING,
- "unlink",
- fn);
- return;
- }
- left = 0;
- (void) GNUNET_HELLO_iterate_addresses (hello_clean,
- GNUNET_NO,
- &count_addresses,
- &left);
- if (0 == left)
- {
- GNUNET_free (hello_clean);
- break;
- }
- if (GNUNET_NO == GNUNET_HELLO_is_friend_only (hello_clean))
- {
- if (NULL == r->hello)
- r->hello = hello_clean;
- else
- {
- GNUNET_break (0);
- GNUNET_free (r->hello);
- r->hello = hello_clean;
- }
- }
- else
- {
- if (NULL == r->friend_only_hello)
- r->friend_only_hello = hello_clean;
- else
- {
- GNUNET_break (0);
- GNUNET_free (r->friend_only_hello);
- r->friend_only_hello = hello_clean;
- }
- }
- read_pos += size_hello;
- }
- if (0 == left)
- {
- /* no addresses left, remove from disk */
- if ( (GNUNET_YES == unlink_garbage) &&
- (0 != UNLINK (fn)) )
- GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_WARNING,
- "unlink",
- fn);
- }
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Found `%s' and `%s' HELLO message in file\n",
- (NULL != r->hello) ? "public" : "NON-public",
- (NULL != r->friend_only_hello) ? "friend only" : "NO friend only");
- }
- /**
- * Add a host to the list and notify clients about this event
- *
- * @param identity the identity of the host
- * @return the HostEntry
- */
- static struct HostEntry *
- add_host_to_known_hosts (const struct GNUNET_PeerIdentity *identity)
- {
- struct HostEntry *entry;
- struct ReadHostFileContext r;
- char *fn;
- entry = GNUNET_CONTAINER_multipeermap_get (hostmap,
- identity);
- if (NULL == entry)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Adding new peer `%s'\n",
- GNUNET_i2s (identity));
- GNUNET_STATISTICS_update (stats,
- gettext_noop ("# peers known"),
- 1,
- GNUNET_NO);
- entry = GNUNET_new (struct HostEntry);
- entry->identity = *identity;
- GNUNET_assert (GNUNET_OK ==
- GNUNET_CONTAINER_multipeermap_put (hostmap,
- &entry->identity,
- entry,
- GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY));
- notify_all (entry);
- fn = get_host_filename (identity);
- if (NULL != fn)
- {
- read_host_file (fn,
- GNUNET_YES,
- &r);
- if (NULL != r.hello)
- update_hello (identity,
- r.hello);
- if (NULL != r.friend_only_hello)
- update_hello (identity,
- r.friend_only_hello);
- GNUNET_free_non_null (r.hello);
- GNUNET_free_non_null (r.friend_only_hello);
- GNUNET_free (fn);
- }
- }
- return entry;
- }
- /**
- * Remove a file that should not be there. LOG
- * success or failure.
- *
- * @param fullname name of the file to remove
- */
- static void
- remove_garbage (const char *fullname)
- {
- if (0 == UNLINK (fullname))
- GNUNET_log (GNUNET_ERROR_TYPE_WARNING | GNUNET_ERROR_TYPE_BULK,
- _("File `%s' in directory `%s' does not match naming convention. Removed.\n"),
- fullname,
- networkIdDirectory);
- else
- GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_ERROR | GNUNET_ERROR_TYPE_BULK,
- "unlink",
- fullname);
- }
- /**
- * Closure for #hosts_directory_scan_callback().
- */
- struct DirScanContext
- {
- /**
- * #GNUNET_YES if we should remove files that are broken,
- * #GNUNET_NO if the directory we are iterating over should
- * be treated as read-only by us.
- */
- int remove_files;
- /**
- * Counter for the number of (valid) entries found, incremented
- * by one for each match.
- */
- unsigned int matched;
- };
- /**
- * Function that is called on each HELLO file in a particular directory.
- * Try to parse the file and add the HELLO to our list.
- *
- * @param cls pointer to 'unsigned int' to increment for each file, or NULL
- * if the file is from a read-only, read-once resource directory
- * @param fullname name of the file to parse
- * @return #GNUNET_OK (continue iteration)
- */
- static int
- hosts_directory_scan_callback (void *cls,
- const char *fullname)
- {
- struct DirScanContext *dsc = cls;
- struct GNUNET_PeerIdentity identity;
- struct ReadHostFileContext r;
- const char *filename;
- struct GNUNET_PeerIdentity id_public;
- struct GNUNET_PeerIdentity id_friend;
- struct GNUNET_PeerIdentity id;
- if (GNUNET_YES != GNUNET_DISK_file_test (fullname))
- return GNUNET_OK; /* ignore non-files */
- filename = strrchr (fullname,
- DIR_SEPARATOR);
- if ( (NULL == filename) ||
- (1 > strlen (filename)) )
- filename = fullname;
- else
- filename ++;
- read_host_file (fullname,
- dsc->remove_files,
- &r);
- if ( (NULL == r.hello) &&
- (NULL == r.friend_only_hello))
- return GNUNET_OK;
- if (NULL != r.friend_only_hello)
- {
- if (GNUNET_OK !=
- GNUNET_HELLO_get_id (r.friend_only_hello,
- &id_friend))
- {
- if (GNUNET_YES == dsc->remove_files)
- remove_garbage (fullname);
- return GNUNET_OK;
- }
- id = id_friend;
- }
- if (NULL != r.hello)
- {
- if (GNUNET_OK !=
- GNUNET_HELLO_get_id (r.hello,
- &id_public))
- {
- if (GNUNET_YES == dsc->remove_files)
- remove_garbage (fullname);
- return GNUNET_OK;
- }
- id = id_public;
- }
- if ( (NULL != r.hello) &&
- (NULL != r.friend_only_hello) &&
- (0 != memcmp (&id_friend,
- &id_public,
- sizeof (id_friend))) )
- {
- /* HELLOs are not for the same peer */
- GNUNET_break (0);
- if (GNUNET_YES == dsc->remove_files)
- remove_garbage (fullname);
- return GNUNET_OK;
- }
- if (GNUNET_OK ==
- GNUNET_CRYPTO_eddsa_public_key_from_string (filename,
- strlen (filename),
- &identity.public_key))
- {
- if (0 != memcmp (&id, &identity, sizeof (id_friend)))
- {
- /* HELLOs are not for the same peer */
- GNUNET_break (0);
- if (GNUNET_YES == dsc->remove_files)
- remove_garbage (fullname);
- return GNUNET_OK;
- }
- }
- /* ok, found something valid, remember HELLO */
- add_host_to_known_hosts (&id);
- if (NULL != r.hello)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Updating peer `%s' public HELLO \n",
- GNUNET_i2s (&id));
- update_hello (&id, r.hello);
- GNUNET_free (r.hello);
- }
- if (NULL != r.friend_only_hello)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Updating peer `%s' friend only HELLO \n",
- GNUNET_i2s (&id));
- update_hello (&id, r.friend_only_hello);
- GNUNET_free (r.friend_only_hello);
- }
- dsc->matched++;
- return GNUNET_OK;
- }
- /**
- * Call this method periodically to scan data/hosts for new hosts.
- *
- * @param cls unused
- */
- static void
- cron_scan_directory_data_hosts (void *cls)
- {
- static unsigned int retries;
- struct DirScanContext dsc;
- (void) cls;
- cron_scan = NULL;
- if (GNUNET_SYSERR ==
- GNUNET_DISK_directory_create (networkIdDirectory))
- {
- cron_scan = GNUNET_SCHEDULER_add_delayed_with_priority (DATA_HOST_FREQ,
- GNUNET_SCHEDULER_PRIORITY_IDLE,
- &cron_scan_directory_data_hosts, NULL);
- return;
- }
- dsc.matched = 0;
- dsc.remove_files = GNUNET_YES;
- GNUNET_log (GNUNET_ERROR_TYPE_INFO | GNUNET_ERROR_TYPE_BULK,
- _("Scanning directory `%s'\n"),
- networkIdDirectory);
- GNUNET_DISK_directory_scan (networkIdDirectory,
- &hosts_directory_scan_callback,
- &dsc);
- if ( (0 == dsc.matched) &&
- (0 == (++retries & 31)) )
- GNUNET_log (GNUNET_ERROR_TYPE_WARNING | GNUNET_ERROR_TYPE_BULK,
- _("Still no peers found in `%s'!\n"),
- networkIdDirectory);
- cron_scan = GNUNET_SCHEDULER_add_delayed_with_priority (DATA_HOST_FREQ,
- GNUNET_SCHEDULER_PRIORITY_IDLE,
- &cron_scan_directory_data_hosts,
- NULL);
- }
- /**
- * Update the HELLO of a friend by merging the addresses.
- *
- * @param hello original hello
- * @param friend_hello hello with additional addresses
- * @return merged HELLO
- */
- static struct GNUNET_HELLO_Message *
- update_friend_hello (const struct GNUNET_HELLO_Message *hello,
- const struct GNUNET_HELLO_Message *friend_hello)
- {
- struct GNUNET_HELLO_Message * res;
- struct GNUNET_HELLO_Message * tmp;
- struct GNUNET_PeerIdentity pid;
- if (NULL != friend_hello)
- {
- res = GNUNET_HELLO_merge (hello,
- friend_hello);
- GNUNET_assert (GNUNET_YES ==
- GNUNET_HELLO_is_friend_only (res));
- return res;
- }
- if (GNUNET_OK !=
- GNUNET_HELLO_get_id (hello, &pid))
- {
- GNUNET_break (0);
- return NULL;
- }
- tmp = GNUNET_HELLO_create (&pid.public_key,
- NULL,
- NULL,
- GNUNET_YES);
- res = GNUNET_HELLO_merge (hello, tmp);
- GNUNET_free (tmp);
- GNUNET_assert (GNUNET_YES == GNUNET_HELLO_is_friend_only (res));
- return res;
- }
- /**
- * Bind a host address (hello) to a hostId.
- *
- * @param peer the peer for which this is a hello
- * @param hello the verified (!) hello message
- */
- static void
- update_hello (const struct GNUNET_PeerIdentity *peer,
- const struct GNUNET_HELLO_Message *hello)
- {
- char *fn;
- struct HostEntry *host;
- struct GNUNET_HELLO_Message *mrg;
- struct GNUNET_HELLO_Message **dest;
- struct GNUNET_TIME_Absolute delta;
- unsigned int cnt;
- unsigned int size;
- int friend_hello_type;
- int store_hello;
- int store_friend_hello;
- unsigned int pos;
- char *buffer;
- host = GNUNET_CONTAINER_multipeermap_get (hostmap, peer);
- GNUNET_assert (NULL != host);
- friend_hello_type = GNUNET_HELLO_is_friend_only (hello);
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Updating %s HELLO for `%s'\n",
- (GNUNET_YES == friend_hello_type) ? "friend-only" : "public",
- GNUNET_i2s (peer));
- dest = NULL;
- if (GNUNET_YES == friend_hello_type)
- {
- dest = &host->friend_only_hello;
- }
- else
- {
- dest = &host->hello;
- }
- if (NULL == (*dest))
- {
- (*dest) = GNUNET_malloc (GNUNET_HELLO_size (hello));
- GNUNET_memcpy ((*dest), hello, GNUNET_HELLO_size (hello));
- }
- else
- {
- mrg = GNUNET_HELLO_merge ((*dest),
- hello);
- delta = GNUNET_HELLO_equals (mrg,
- (*dest),
- GNUNET_TIME_absolute_get ());
- if (delta.abs_value_us == GNUNET_TIME_UNIT_FOREVER_ABS.abs_value_us)
- {
- /* no differences, just ignore the update */
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "No change in %s HELLO for `%s'\n",
- (GNUNET_YES == friend_hello_type) ? "friend-only" : "public",
- GNUNET_i2s (peer));
- GNUNET_free (mrg);
- return;
- }
- GNUNET_free ((*dest));
- (*dest) = mrg;
- }
- if ( (NULL != (host->hello)) &&
- (GNUNET_NO == friend_hello_type) )
- {
- /* Update friend only hello */
- mrg = update_friend_hello (host->hello,
- host->friend_only_hello);
- if (NULL != host->friend_only_hello)
- GNUNET_free (host->friend_only_hello);
- host->friend_only_hello = mrg;
- }
- if (NULL != host->hello)
- GNUNET_assert ((GNUNET_NO ==
- GNUNET_HELLO_is_friend_only (host->hello)));
- if (NULL != host->friend_only_hello)
- GNUNET_assert ((GNUNET_YES ==
- GNUNET_HELLO_is_friend_only (host->friend_only_hello)));
- fn = get_host_filename (peer);
- if ( (NULL != fn) &&
- (GNUNET_OK ==
- GNUNET_DISK_directory_create_for_file (fn)) )
- {
- store_hello = GNUNET_NO;
- size = 0;
- cnt = 0;
- if (NULL != host->hello)
- (void) GNUNET_HELLO_iterate_addresses (host->hello,
- GNUNET_NO,
- &count_addresses,
- &cnt);
- if (cnt > 0)
- {
- store_hello = GNUNET_YES;
- size += GNUNET_HELLO_size (host->hello);
- }
- cnt = 0;
- if (NULL != host->friend_only_hello)
- (void) GNUNET_HELLO_iterate_addresses (host->friend_only_hello,
- GNUNET_NO,
- &count_addresses,
- &cnt);
- store_friend_hello = GNUNET_NO;
- if (0 < cnt)
- {
- store_friend_hello = GNUNET_YES;
- size += GNUNET_HELLO_size (host->friend_only_hello);
- }
- if ( (GNUNET_NO == store_hello) &&
- (GNUNET_NO == store_friend_hello) )
- {
- /* no valid addresses, don't put HELLO on disk; in fact,
- if one exists on disk, remove it */
- (void) UNLINK (fn);
- }
- else
- {
- buffer = GNUNET_malloc (size);
- pos = 0;
- if (GNUNET_YES == store_hello)
- {
- GNUNET_memcpy (buffer,
- host->hello,
- GNUNET_HELLO_size (host->hello));
- pos += GNUNET_HELLO_size (host->hello);
- }
- if (GNUNET_YES == store_friend_hello)
- {
- GNUNET_memcpy (&buffer[pos],
- host->friend_only_hello,
- GNUNET_HELLO_size (host->friend_only_hello));
- pos += GNUNET_HELLO_size (host->friend_only_hello);
- }
- GNUNET_assert (pos == size);
- if (GNUNET_SYSERR == GNUNET_DISK_fn_write (fn, buffer, size,
- GNUNET_DISK_PERM_USER_READ |
- GNUNET_DISK_PERM_USER_WRITE |
- GNUNET_DISK_PERM_GROUP_READ |
- GNUNET_DISK_PERM_OTHER_READ))
- GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_WARNING, "write", fn);
- else
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Stored %s %s HELLO in %s with total size %u\n",
- (GNUNET_YES == store_friend_hello) ? "friend-only": "",
- (GNUNET_YES == store_hello) ? "public": "",
- fn,
- size);
- GNUNET_free (buffer);
- }
- }
- GNUNET_free_non_null (fn);
- notify_all (host);
- }
- /**
- * Closure for #add_to_tc()
- */
- struct TransmitContext
- {
- /**
- * Client to transmit to
- */
- struct GNUNET_SERVICE_Client *client;
- /**
- * Include friend only HELLOs #GNUNET_YES or #GNUNET_NO
- */
- int friend_only;
- };
- /**
- * Do transmit info about peer to given host.
- *
- * @param cls NULL to hit all hosts, otherwise specifies a particular target
- * @param key hostID
- * @param value information to transmit
- * @return #GNUNET_YES (continue to iterate)
- */
- static int
- add_to_tc (void *cls,
- const struct GNUNET_PeerIdentity *key,
- void *value)
- {
- struct TransmitContext *tc = cls;
- struct HostEntry *pos = value;
- struct InfoMessage *im;
- uint16_t hs;
- struct GNUNET_MQ_Envelope *env;
- hs = 0;
- if ( (NULL != pos->hello) &&
- (GNUNET_NO == tc->friend_only) )
- {
- /* Copy public HELLO */
- hs = GNUNET_HELLO_size (pos->hello);
- GNUNET_assert (hs < GNUNET_MAX_MESSAGE_SIZE -
- sizeof (struct InfoMessage));
- env = GNUNET_MQ_msg_extra (im,
- hs,
- GNUNET_MESSAGE_TYPE_PEERINFO_INFO);
- GNUNET_memcpy (&im[1],
- pos->hello,
- hs);
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Sending public HELLO with size %u for peer `%s'\n",
- hs,
- GNUNET_i2s (key));
- }
- else if ( (NULL != pos->friend_only_hello) &&
- (GNUNET_YES == tc->friend_only) )
- {
- /* Copy friend only HELLO */
- hs = GNUNET_HELLO_size (pos->friend_only_hello);
- GNUNET_assert (hs < GNUNET_MAX_MESSAGE_SIZE -
- sizeof (struct InfoMessage));
- env = GNUNET_MQ_msg_extra (im,
- hs,
- GNUNET_MESSAGE_TYPE_PEERINFO_INFO);
- GNUNET_memcpy (&im[1],
- pos->friend_only_hello,
- hs);
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Sending friend-only HELLO with size %u for peer `%s'\n",
- hs,
- GNUNET_i2s (key));
- }
- else
- {
- env = GNUNET_MQ_msg (im,
- GNUNET_MESSAGE_TYPE_PEERINFO_INFO);
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Adding no HELLO for peer `%s'\n",
- GNUNET_i2s (key));
- }
- im->peer = pos->identity;
- GNUNET_MQ_send (GNUNET_SERVICE_client_get_mq (tc->client),
- env);
- return GNUNET_YES;
- }
- /**
- * @brief delete expired HELLO entries in directory
- *
- * @param cls pointer to current time (`struct GNUNET_TIME_Absolute *`)
- * @param fn filename to test to see if the HELLO expired
- * @return #GNUNET_OK (continue iteration)
- */
- static int
- discard_hosts_helper (void *cls,
- const char *fn)
- {
- struct GNUNET_TIME_Absolute *now = cls;
- char buffer[GNUNET_MAX_MESSAGE_SIZE - 1] GNUNET_ALIGN;
- const struct GNUNET_HELLO_Message *hello;
- struct GNUNET_HELLO_Message *new_hello;
- int read_size;
- unsigned int cur_hello_size;
- unsigned int new_hello_size;
- int read_pos;
- int write_pos;
- unsigned int cnt;
- char *writebuffer;
- uint64_t fsize;
-
- if (GNUNET_OK !=
- GNUNET_DISK_file_size (fn,
- &fsize,
- GNUNET_YES,
- GNUNET_YES))
- {
- GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_WARNING |
- GNUNET_ERROR_TYPE_BULK,
- "fstat",
- fn);
- return GNUNET_OK;
- }
- read_size = GNUNET_DISK_fn_read (fn,
- buffer,
- sizeof (buffer));
- if ( (read_size < (int) sizeof (struct GNUNET_MessageHeader)) ||
- (fsize > GNUNET_MAX_MESSAGE_SIZE) )
- {
- if (0 != UNLINK (fn))
- GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_WARNING |
- GNUNET_ERROR_TYPE_BULK,
- "unlink",
- fn);
- return GNUNET_OK;
- }
- writebuffer = GNUNET_malloc (read_size);
- read_pos = 0;
- write_pos = 0;
- while (read_pos < read_size)
- {
- /* Check each HELLO */
- hello = (const struct GNUNET_HELLO_Message *) &buffer[read_pos];
- cur_hello_size = GNUNET_HELLO_size (hello);
- if (0 == cur_hello_size)
- {
- /* Invalid data, discard */
- if (0 != UNLINK (fn))
- GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_WARNING |
- GNUNET_ERROR_TYPE_BULK,
- "unlink",
- fn);
- GNUNET_free (writebuffer);
- return GNUNET_OK;
- }
- new_hello = GNUNET_HELLO_iterate_addresses (hello,
- GNUNET_YES,
- &discard_expired,
- now);
- cnt = 0;
- if (NULL != new_hello)
- (void) GNUNET_HELLO_iterate_addresses (hello,
- GNUNET_NO,
- &count_addresses,
- &cnt);
- if ( (NULL != new_hello) && (0 < cnt) )
- {
- /* Store new HELLO to write it when done */
- new_hello_size = GNUNET_HELLO_size (new_hello);
- GNUNET_memcpy (&writebuffer[write_pos],
- new_hello,
- new_hello_size);
- write_pos += new_hello_size;
- }
- read_pos += cur_hello_size;
- GNUNET_free_non_null (new_hello);
- }
- if (0 < write_pos)
- {
- GNUNET_DISK_fn_write (fn,
- writebuffer,
- write_pos,
- GNUNET_DISK_PERM_USER_READ |
- GNUNET_DISK_PERM_USER_WRITE |
- GNUNET_DISK_PERM_GROUP_READ |
- GNUNET_DISK_PERM_OTHER_READ);
- }
- else if (0 != UNLINK (fn))
- GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_WARNING |
- GNUNET_ERROR_TYPE_BULK,
- "unlink", fn);
- GNUNET_free (writebuffer);
- return GNUNET_OK;
- }
- /**
- * Call this method periodically to scan peerinfo/ for ancient
- * HELLOs to expire.
- *
- * @param cls unused
- */
- static void
- cron_clean_data_hosts (void *cls)
- {
- struct GNUNET_TIME_Absolute now;
- (void) cls;
- cron_clean = NULL;
- now = GNUNET_TIME_absolute_get ();
- GNUNET_log (GNUNET_ERROR_TYPE_INFO | GNUNET_ERROR_TYPE_BULK,
- _("Cleaning up directory `%s'\n"),
- networkIdDirectory);
- GNUNET_DISK_directory_scan (networkIdDirectory,
- &discard_hosts_helper,
- &now);
- cron_clean = GNUNET_SCHEDULER_add_delayed (DATA_HOST_CLEAN_FREQ,
- &cron_clean_data_hosts,
- NULL);
- }
- /**
- * Check HELLO-message.
- *
- * @param cls identification of the client
- * @param hello the actual message
- * @return #GNUNET_OK if @a hello is well-formed
- */
- static int
- check_hello (void *cls,
- const struct GNUNET_HELLO_Message *hello)
- {
- struct GNUNET_PeerIdentity pid;
- (void) cls;
- if (GNUNET_OK !=
- GNUNET_HELLO_get_id (hello,
- &pid))
- {
- GNUNET_break (0);
- return GNUNET_SYSERR;
- }
- return GNUNET_OK;
- }
- /**
- * Handle HELLO-message.
- *
- * @param cls identification of the client
- * @param hello the actual message
- */
- static void
- handle_hello (void *cls,
- const struct GNUNET_HELLO_Message *hello)
- {
- struct GNUNET_SERVICE_Client *client = cls;
- struct GNUNET_PeerIdentity pid;
- GNUNET_assert (GNUNET_OK ==
- GNUNET_HELLO_get_id (hello,
- &pid));
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "HELLO message received for peer `%s'\n",
- GNUNET_i2s (&pid));
- add_host_to_known_hosts (&pid);
- update_hello (&pid,
- hello);
- GNUNET_SERVICE_client_continue (client);
- }
- /**
- * Handle GET-message.
- *
- * @param cls identification of the client
- * @param lpm the actual message
- */
- static void
- handle_get (void *cls,
- const struct ListPeerMessage *lpm)
- {
- struct GNUNET_SERVICE_Client *client = cls;
- struct TransmitContext tcx;
- struct GNUNET_MessageHeader *msg;
- struct GNUNET_MQ_Envelope *env;
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "GET message received for peer `%s'\n",
- GNUNET_i2s (&lpm->peer));
- tcx.friend_only = ntohl (lpm->include_friend_only);
- tcx.client = client;
- GNUNET_CONTAINER_multipeermap_get_multiple (hostmap,
- &lpm->peer,
- &add_to_tc,
- &tcx);
- env = GNUNET_MQ_msg (msg,
- GNUNET_MESSAGE_TYPE_PEERINFO_INFO_END);
- GNUNET_MQ_send (GNUNET_SERVICE_client_get_mq (client),
- env);
- GNUNET_SERVICE_client_continue (client);
- }
- /**
- * Handle GET-ALL-message.
- *
- * @param cls identification of the client
- * @param lapm the actual message
- */
- static void
- handle_get_all (void *cls,
- const struct ListAllPeersMessage *lapm)
- {
- struct GNUNET_SERVICE_Client *client = cls;
- struct TransmitContext tcx;
- struct GNUNET_MQ_Envelope *env;
- struct GNUNET_MessageHeader *msg;
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "GET_ALL message received\n");
- tcx.friend_only = ntohl (lapm->include_friend_only);
- tcx.client = client;
- GNUNET_CONTAINER_multipeermap_iterate (hostmap,
- &add_to_tc,
- &tcx);
- env = GNUNET_MQ_msg (msg,
- GNUNET_MESSAGE_TYPE_PEERINFO_INFO_END);
- GNUNET_MQ_send (GNUNET_SERVICE_client_get_mq (client),
- env);
- GNUNET_SERVICE_client_continue (client);
- }
- /**
- * Handle NOTIFY-message.
- *
- * @param cls identification of the client
- * @param nm the actual message
- */
- static void
- handle_notify (void *cls,
- const struct NotifyMessage *nm)
- {
- struct GNUNET_SERVICE_Client *client = cls;
- struct GNUNET_MQ_Handle *mq;
- struct TransmitContext tcx;
- struct GNUNET_MQ_Envelope *env;
- struct GNUNET_MessageHeader *msg;
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "NOTIFY message received\n");
- mq = GNUNET_SERVICE_client_get_mq (client);
- GNUNET_SERVICE_client_mark_monitor (client);
- if (ntohl (nm->include_friend_only))
- GNUNET_notification_context_add (notify_friend_only_list,
- mq);
- else
- GNUNET_notification_context_add (notify_list,
- mq);
- tcx.friend_only = ntohl (nm->include_friend_only);
- tcx.client = client;
- GNUNET_CONTAINER_multipeermap_iterate (hostmap,
- &add_to_tc,
- &tcx);
- env = GNUNET_MQ_msg (msg,
- GNUNET_MESSAGE_TYPE_PEERINFO_INFO_END);
- GNUNET_MQ_send (GNUNET_SERVICE_client_get_mq (client),
- env);
- GNUNET_SERVICE_client_continue (client);
- }
- /**
- * Client connect callback
- *
- * @param cls unused
- * @param client server client
- * @param mq for @a client
- * @return @a client
- */
- static void *
- client_connect_cb (void *cls,
- struct GNUNET_SERVICE_Client *client,
- struct GNUNET_MQ_Handle *mq)
- {
- (void) cls;
- (void) mq;
- return client;
- }
- /**
- * Client disconnect callback
- *
- * @param cls unused
- * @param client server client
- * @param app_ctx should be @a client
- */
- static void
- client_disconnect_cb (void *cls,
- struct GNUNET_SERVICE_Client *client,
- void *app_ctx)
- {
- (void) cls;
- GNUNET_assert (app_ctx == client);
- }
- /**
- * Release memory taken by a host entry.
- *
- * @param cls NULL
- * @param key key of the host entry
- * @param value the `struct HostEntry` to free
- * @return #GNUNET_YES (continue to iterate)
- */
- static int
- free_host_entry (void *cls,
- const struct GNUNET_PeerIdentity *key,
- void *value)
- {
- struct HostEntry *he = value;
- (void) cls;
- (void) key;
- GNUNET_free_non_null (he->hello);
- GNUNET_free_non_null (he->friend_only_hello);
- GNUNET_free (he);
- return GNUNET_YES;
- }
- /**
- * Clean up our state. Called during shutdown.
- *
- * @param cls unused
- */
- static void
- shutdown_task (void *cls)
- {
- (void) cls;
- GNUNET_notification_context_destroy (notify_list);
- notify_list = NULL;
- GNUNET_notification_context_destroy (notify_friend_only_list);
- notify_friend_only_list = NULL;
- GNUNET_CONTAINER_multipeermap_iterate (hostmap,
- &free_host_entry,
- NULL);
- GNUNET_CONTAINER_multipeermap_destroy (hostmap);
- if (NULL != stats)
- {
- GNUNET_STATISTICS_destroy (stats,
- GNUNET_NO);
- stats = NULL;
- }
- if (NULL != cron_clean)
- {
- GNUNET_SCHEDULER_cancel (cron_clean);
- cron_clean = NULL;
- }
- if (NULL != cron_scan)
- {
- GNUNET_SCHEDULER_cancel (cron_scan);
- cron_scan = NULL;
- }
- if (NULL != networkIdDirectory)
- {
- GNUNET_free (networkIdDirectory);
- networkIdDirectory = NULL;
- }
- }
- /**
- * Start up peerinfo service.
- *
- * @param cls closure
- * @param cfg configuration to use
- * @param service the initialized service
- */
- static void
- run (void *cls,
- const struct GNUNET_CONFIGURATION_Handle *cfg,
- struct GNUNET_SERVICE_Handle *service)
- {
- char *peerdir;
- char *ip;
- struct DirScanContext dsc;
- int noio;
- int use_included;
- (void) cls;
- (void) service;
- hostmap
- = GNUNET_CONTAINER_multipeermap_create (1024,
- GNUNET_YES);
- stats
- = GNUNET_STATISTICS_create ("peerinfo",
- cfg);
- notify_list
- = GNUNET_notification_context_create (0);
- notify_friend_only_list
- = GNUNET_notification_context_create (0);
- noio = GNUNET_CONFIGURATION_get_value_yesno (cfg,
- "peerinfo",
- "NO_IO");
- use_included
- = GNUNET_CONFIGURATION_get_value_yesno (cfg,
- "peerinfo",
- "USE_INCLUDED_HELLOS");
- if (GNUNET_SYSERR == use_included)
- use_included = GNUNET_NO;
- GNUNET_SCHEDULER_add_shutdown (&shutdown_task,
- NULL);
- if (GNUNET_YES != noio)
- {
- GNUNET_assert (GNUNET_OK ==
- GNUNET_CONFIGURATION_get_value_filename (cfg,
- "peerinfo",
- "HOSTS",
- &networkIdDirectory));
- if (GNUNET_OK !=
- GNUNET_DISK_directory_create (networkIdDirectory))
- {
- GNUNET_SCHEDULER_shutdown ();
- return;
- }
- cron_scan
- = GNUNET_SCHEDULER_add_with_priority (GNUNET_SCHEDULER_PRIORITY_IDLE,
- &cron_scan_directory_data_hosts,
- NULL);
- cron_clean
- = GNUNET_SCHEDULER_add_with_priority (GNUNET_SCHEDULER_PRIORITY_IDLE,
- &cron_clean_data_hosts,
- NULL);
- if (GNUNET_YES == use_included)
- {
- ip = GNUNET_OS_installation_get_path (GNUNET_OS_IPK_DATADIR);
- GNUNET_asprintf (&peerdir,
- "%shellos",
- ip);
- GNUNET_free (ip);
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- _("Importing HELLOs from `%s'\n"),
- peerdir);
- dsc.matched = 0;
- dsc.remove_files = GNUNET_NO;
- GNUNET_DISK_directory_scan (peerdir,
- &hosts_directory_scan_callback,
- &dsc);
- GNUNET_free (peerdir);
- }
- else
- {
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- _("Skipping import of included HELLOs\n"));
- }
- }
- }
- /**
- * Define "main" method using service macro.
- */
- GNUNET_SERVICE_MAIN
- ("peerinfo",
- GNUNET_SERVICE_OPTION_NONE,
- &run,
- &client_connect_cb,
- &client_disconnect_cb,
- NULL,
- GNUNET_MQ_hd_var_size (hello,
- GNUNET_MESSAGE_TYPE_HELLO,
- struct GNUNET_HELLO_Message,
- NULL),
- GNUNET_MQ_hd_fixed_size (get,
- GNUNET_MESSAGE_TYPE_PEERINFO_GET,
- struct ListPeerMessage,
- NULL),
- GNUNET_MQ_hd_fixed_size (get_all,
- GNUNET_MESSAGE_TYPE_PEERINFO_GET_ALL,
- struct ListAllPeersMessage,
- NULL),
- GNUNET_MQ_hd_fixed_size (notify,
- GNUNET_MESSAGE_TYPE_PEERINFO_NOTIFY,
- struct NotifyMessage,
- NULL),
- GNUNET_MQ_handler_end ());
- /* end of gnunet-service-peerinfo.c */
|