123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442 |
- /*
- This file is part of GNUnet.
- Copyright (C) 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 util/service.c
- * @brief functions related to starting services (redesign)
- * @author Christian Grothoff
- * @author Florian Dold
- */
- #include "platform.h"
- #include "gnunet_util_lib.h"
- #include "gnunet_protocols.h"
- #include "gnunet_constants.h"
- #include "gnunet_resolver_service.h"
- #include "speedup.h"
- #if HAVE_MALLINFO
- #include <malloc.h>
- #include "gauger.h"
- #endif
- #define LOG(kind, ...) GNUNET_log_from (kind, "util-service", __VA_ARGS__)
- #define LOG_STRERROR(kind, syscall) \
- GNUNET_log_from_strerror (kind, "util-service", syscall)
- #define LOG_STRERROR_FILE(kind, syscall, filename) \
- GNUNET_log_from_strerror_file (kind, "util-service", syscall, filename)
- /**
- * Information the service tracks per listen operation.
- */
- struct ServiceListenContext
- {
- /**
- * Kept in a DLL.
- */
- struct ServiceListenContext *next;
- /**
- * Kept in a DLL.
- */
- struct ServiceListenContext *prev;
- /**
- * Service this listen context belongs to.
- */
- struct GNUNET_SERVICE_Handle *sh;
- /**
- * Socket we are listening on.
- */
- struct GNUNET_NETWORK_Handle *listen_socket;
- /**
- * Task scheduled to do the listening.
- */
- struct GNUNET_SCHEDULER_Task *listen_task;
- };
- /**
- * Reasons why we might be suspended.
- */
- enum SuspendReason
- {
- /**
- * We are running normally.
- */
- SUSPEND_STATE_NONE = 0,
- /**
- * Application requested it.
- */
- SUSPEND_STATE_APP = 1,
- /**
- * OS ran out of file descriptors.
- */
- SUSPEND_STATE_EMFILE = 2,
- /**
- * Both reasons, APP and EMFILE apply.
- */
- SUSPEND_STATE_APP_AND_EMFILE = 3,
- /**
- * Suspension because service was permanently shutdown.
- */
- SUSPEND_STATE_SHUTDOWN = 4
- };
- /**
- * Handle to a service.
- */
- struct GNUNET_SERVICE_Handle
- {
- /**
- * Our configuration.
- */
- const struct GNUNET_CONFIGURATION_Handle *cfg;
- /**
- * Name of our service.
- */
- const char *service_name;
- /**
- * Main service-specific task to run.
- */
- GNUNET_SERVICE_InitCallback service_init_cb;
- /**
- * Function to call when clients connect.
- */
- GNUNET_SERVICE_ConnectHandler connect_cb;
- /**
- * Function to call when clients disconnect / are disconnected.
- */
- GNUNET_SERVICE_DisconnectHandler disconnect_cb;
- /**
- * Closure for @e service_init_cb, @e connect_cb, @e disconnect_cb.
- */
- void *cb_cls;
- /**
- * DLL of listen sockets used to accept new connections.
- */
- struct ServiceListenContext *slc_head;
- /**
- * DLL of listen sockets used to accept new connections.
- */
- struct ServiceListenContext *slc_tail;
- /**
- * Our clients, kept in a DLL.
- */
- struct GNUNET_SERVICE_Client *clients_head;
- /**
- * Our clients, kept in a DLL.
- */
- struct GNUNET_SERVICE_Client *clients_tail;
- /**
- * Message handlers to use for all clients.
- */
- struct GNUNET_MQ_MessageHandler *handlers;
- /**
- * Closure for @e task.
- */
- void *task_cls;
- /**
- * IPv4 addresses that are not allowed to connect.
- */
- struct GNUNET_STRINGS_IPv4NetworkPolicy *v4_denied;
- /**
- * IPv6 addresses that are not allowed to connect.
- */
- struct GNUNET_STRINGS_IPv6NetworkPolicy *v6_denied;
- /**
- * IPv4 addresses that are allowed to connect (if not
- * set, all are allowed).
- */
- struct GNUNET_STRINGS_IPv4NetworkPolicy *v4_allowed;
- /**
- * IPv6 addresses that are allowed to connect (if not
- * set, all are allowed).
- */
- struct GNUNET_STRINGS_IPv6NetworkPolicy *v6_allowed;
- /**
- * Do we require a matching UID for UNIX domain socket connections?
- * #GNUNET_NO means that the UID does not have to match (however,
- * @e match_gid may still impose other access control checks).
- */
- int match_uid;
- /**
- * Do we require a matching GID for UNIX domain socket connections?
- * Ignored if @e match_uid is #GNUNET_YES. Note that this is about
- * checking that the client's UID is in our group OR that the
- * client's GID is our GID. If both "match_gid" and @e match_uid are
- * #GNUNET_NO, all users on the local system have access.
- */
- int match_gid;
- /**
- * Are we suspended, and if so, why?
- */
- enum SuspendReason suspend_state;
- /**
- * Our options.
- */
- enum GNUNET_SERVICE_Options options;
- /**
- * If we are daemonizing, this FD is set to the
- * pipe to the parent. Send '.' if we started
- * ok, '!' if not. -1 if we are not daemonizing.
- */
- int ready_confirm_fd;
- /**
- * Overall success/failure of the service start.
- */
- int ret;
- /**
- * If #GNUNET_YES, consider unknown message types an error where the
- * client is disconnected.
- */
- int require_found;
- };
- /**
- * Handle to a client that is connected to a service.
- */
- struct GNUNET_SERVICE_Client
- {
- /**
- * Kept in a DLL.
- */
- struct GNUNET_SERVICE_Client *next;
- /**
- * Kept in a DLL.
- */
- struct GNUNET_SERVICE_Client *prev;
- /**
- * Service that this client belongs to.
- */
- struct GNUNET_SERVICE_Handle *sh;
- /**
- * Socket of this client.
- */
- struct GNUNET_NETWORK_Handle *sock;
- /**
- * Message queue for the client.
- */
- struct GNUNET_MQ_Handle *mq;
- /**
- * Tokenizer we use for processing incoming data.
- */
- struct GNUNET_MessageStreamTokenizer *mst;
- /**
- * Task that warns about missing calls to
- * #GNUNET_SERVICE_client_continue().
- */
- struct GNUNET_SCHEDULER_Task *warn_task;
- /**
- * Task run to finish dropping the client after the stack has
- * properly unwound.
- */
- struct GNUNET_SCHEDULER_Task *drop_task;
- /**
- * Task that receives data from the client to
- * pass it to the handlers.
- */
- struct GNUNET_SCHEDULER_Task *recv_task;
- /**
- * Task that transmit data to the client.
- */
- struct GNUNET_SCHEDULER_Task *send_task;
- /**
- * Pointer to the message to be transmitted by @e send_task.
- */
- const struct GNUNET_MessageHeader *msg;
- /**
- * User context value, value returned from
- * the connect callback.
- */
- void *user_context;
- /**
- * Time when we last gave a message from this client
- * to the application.
- */
- struct GNUNET_TIME_Absolute warn_start;
- /**
- * Current position in @e msg at which we are transmitting.
- */
- size_t msg_pos;
- /**
- * Persist the file handle for this client no matter what happens,
- * force the OS to close once the process actually dies. Should only
- * be used in special cases!
- */
- int persist;
- /**
- * Is this client a 'monitor' client that should not be counted
- * when deciding on destroying the server during soft shutdown?
- * (see also #GNUNET_SERVICE_start)
- */
- int is_monitor;
- /**
- * Are we waiting for the application to call #GNUNET_SERVICE_client_continue()?
- */
- int needs_continue;
- /**
- * Type of last message processed (for warn_no_receive_done).
- */
- uint16_t warn_type;
- };
- /**
- * Check if any of the clients we have left are unrelated to
- * monitoring.
- *
- * @param sh service to check clients for
- * @return #GNUNET_YES if we have non-monitoring clients left
- */
- static int
- have_non_monitor_clients (struct GNUNET_SERVICE_Handle *sh)
- {
- for (struct GNUNET_SERVICE_Client *client = sh->clients_head; NULL != client;
- client = client->next)
- {
- if (client->is_monitor)
- continue;
- return GNUNET_YES;
- }
- return GNUNET_NO;
- }
- /**
- * Suspend accepting connections from the listen socket temporarily.
- * Resume activity using #do_resume.
- *
- * @param sh service to stop accepting connections.
- * @param sr reason for suspending accepting connections
- */
- static void
- do_suspend (struct GNUNET_SERVICE_Handle *sh, enum SuspendReason sr)
- {
- struct ServiceListenContext *slc;
- GNUNET_assert (0 == (sh->suspend_state & sr));
- sh->suspend_state |= sr;
- for (slc = sh->slc_head; NULL != slc; slc = slc->next)
- {
- if (NULL != slc->listen_task)
- {
- GNUNET_SCHEDULER_cancel (slc->listen_task);
- slc->listen_task = NULL;
- }
- }
- }
- /**
- * Shutdown task triggered when a service should be terminated.
- * This considers active clients and the service options to see
- * how this specific service is to be terminated, and depending
- * on this proceeds with the shutdown logic.
- *
- * @param cls our `struct GNUNET_SERVICE_Handle`
- */
- static void
- service_shutdown (void *cls)
- {
- struct GNUNET_SERVICE_Handle *sh = cls;
- switch (sh->options & GNUNET_SERVICE_OPTION_SHUTDOWN_BITMASK)
- {
- case GNUNET_SERVICE_OPTION_NONE:
- GNUNET_SERVICE_shutdown (sh);
- break;
- case GNUNET_SERVICE_OPTION_MANUAL_SHUTDOWN:
- /* This task should never be run if we are using
- the manual shutdown. */
- GNUNET_assert (0);
- break;
- case GNUNET_SERVICE_OPTION_SOFT_SHUTDOWN:
- if (0 == (sh->suspend_state & SUSPEND_STATE_SHUTDOWN))
- do_suspend (sh, SUSPEND_STATE_SHUTDOWN);
- if (GNUNET_NO == have_non_monitor_clients (sh))
- GNUNET_SERVICE_shutdown (sh);
- break;
- }
- }
- /**
- * Check if the given IP address is in the list of IP addresses.
- *
- * @param list a list of networks
- * @param add the IP to check (in network byte order)
- * @return #GNUNET_NO if the IP is not in the list, #GNUNET_YES if it it is
- */
- static int
- check_ipv4_listed (const struct GNUNET_STRINGS_IPv4NetworkPolicy *list,
- const struct in_addr *add)
- {
- unsigned int i;
- if (NULL == list)
- return GNUNET_NO;
- i = 0;
- while ((0 != list[i].network.s_addr) || (0 != list[i].netmask.s_addr))
- {
- if ((add->s_addr & list[i].netmask.s_addr) ==
- (list[i].network.s_addr & list[i].netmask.s_addr))
- return GNUNET_YES;
- i++;
- }
- return GNUNET_NO;
- }
- /**
- * Check if the given IP address is in the list of IP addresses.
- *
- * @param list a list of networks
- * @param ip the IP to check (in network byte order)
- * @return #GNUNET_NO if the IP is not in the list, #GNUNET_YES if it it is
- */
- static int
- check_ipv6_listed (const struct GNUNET_STRINGS_IPv6NetworkPolicy *list,
- const struct in6_addr *ip)
- {
- unsigned int i;
- if (NULL == list)
- return GNUNET_NO;
- i = 0;
- NEXT:
- while (0 != GNUNET_is_zero (&list[i].network))
- {
- for (unsigned int j = 0; j < sizeof(struct in6_addr) / sizeof(int); j++)
- if (((((int *) ip)[j] & ((int *) &list[i].netmask)[j])) !=
- (((int *) &list[i].network)[j] & ((int *) &list[i].netmask)[j]))
- {
- i++;
- goto NEXT;
- }
- return GNUNET_YES;
- }
- return GNUNET_NO;
- }
- /**
- * Task run when we are ready to transmit data to the
- * client.
- *
- * @param cls the `struct GNUNET_SERVICE_Client *` to send to
- */
- static void
- do_send (void *cls)
- {
- struct GNUNET_SERVICE_Client *client = cls;
- ssize_t ret;
- size_t left;
- const char *buf;
- LOG (GNUNET_ERROR_TYPE_DEBUG,
- "service: sending message with type %u\n",
- ntohs (client->msg->type));
- client->send_task = NULL;
- buf = (const char *) client->msg;
- left = ntohs (client->msg->size) - client->msg_pos;
- ret = GNUNET_NETWORK_socket_send (client->sock,
- &buf[client->msg_pos],
- left);
- GNUNET_assert (ret <= (ssize_t) left);
- if (0 == ret)
- {
- LOG (GNUNET_ERROR_TYPE_DEBUG, "no data send");
- GNUNET_MQ_inject_error (client->mq, GNUNET_MQ_ERROR_WRITE);
- return;
- }
- if (-1 == ret)
- {
- if ((EAGAIN == errno) || (EINTR == errno))
- {
- /* ignore */
- ret = 0;
- }
- else
- {
- if (EPIPE != errno)
- GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "send");
- LOG (GNUNET_ERROR_TYPE_DEBUG,
- "socket send returned with error code %i",
- errno);
- GNUNET_MQ_inject_error (client->mq, GNUNET_MQ_ERROR_WRITE);
- return;
- }
- }
- if (0 == client->msg_pos)
- {
- GNUNET_MQ_impl_send_in_flight (client->mq);
- }
- client->msg_pos += ret;
- if (left > (size_t) ret)
- {
- GNUNET_assert (NULL == client->drop_task);
- client->send_task =
- GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
- client->sock,
- &do_send,
- client);
- return;
- }
- GNUNET_MQ_impl_send_continue (client->mq);
- }
- /**
- * Signature of functions implementing the sending functionality of a
- * message queue.
- *
- * @param mq the message queue
- * @param msg the message to send
- * @param impl_state our `struct GNUNET_SERVICE_Client *`
- */
- static void
- service_mq_send (struct GNUNET_MQ_Handle *mq,
- const struct GNUNET_MessageHeader *msg,
- void *impl_state)
- {
- struct GNUNET_SERVICE_Client *client = impl_state;
- (void) mq;
- if (NULL != client->drop_task)
- return; /* we're going down right now, do not try to send */
- GNUNET_assert (NULL == client->send_task);
- LOG (GNUNET_ERROR_TYPE_DEBUG,
- "Sending message of type %u and size %u to client\n",
- ntohs (msg->type),
- ntohs (msg->size));
- client->msg = msg;
- client->msg_pos = 0;
- client->send_task = GNUNET_SCHEDULER_add_now (&do_send,
- client);
- }
- /**
- * Implementation function that cancels the currently sent message.
- *
- * @param mq message queue
- * @param impl_state state specific to the implementation
- */
- static void
- service_mq_cancel (struct GNUNET_MQ_Handle *mq, void *impl_state)
- {
- struct GNUNET_SERVICE_Client *client = impl_state;
- (void) mq;
- GNUNET_assert (0 == client->msg_pos);
- client->msg = NULL;
- GNUNET_SCHEDULER_cancel (client->send_task);
- client->send_task = NULL;
- }
- /**
- * Generic error handler, called with the appropriate
- * error code and the same closure specified at the creation of
- * the message queue.
- * Not every message queue implementation supports an error handler.
- *
- * @param cls closure with our `struct GNUNET_SERVICE_Client`
- * @param error error code
- */
- static void
- service_mq_error_handler (void *cls, enum GNUNET_MQ_Error error)
- {
- struct GNUNET_SERVICE_Client *client = cls;
- struct GNUNET_SERVICE_Handle *sh = client->sh;
- if ((GNUNET_MQ_ERROR_NO_MATCH == error) && (GNUNET_NO == sh->require_found))
- {
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "No handler for message of type %u found\n",
- (unsigned int) client->warn_type);
- GNUNET_SERVICE_client_continue (client);
- return; /* ignore error */
- }
- GNUNET_SERVICE_client_drop (client);
- }
- /**
- * Task run to warn about missing calls to #GNUNET_SERVICE_client_continue().
- *
- * @param cls our `struct GNUNET_SERVICE_Client *` to process more requests from
- */
- static void
- warn_no_client_continue (void *cls)
- {
- struct GNUNET_SERVICE_Client *client = cls;
- GNUNET_break (
- 0 !=
- client->warn_type); /* type should never be 0 here, as we don't use 0 */
- client->warn_task = GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_MINUTES,
- &warn_no_client_continue,
- client);
- LOG (
- GNUNET_ERROR_TYPE_WARNING,
- _ (
- "Processing code for message of type %u did not call `GNUNET_SERVICE_client_continue' after %s\n"),
- (unsigned int) client->warn_type,
- GNUNET_STRINGS_relative_time_to_string (GNUNET_TIME_absolute_get_duration (
- client->warn_start),
- GNUNET_YES));
- }
- /**
- * Functions with this signature are called whenever a
- * complete message is received by the tokenizer for a client.
- *
- * Do not call #GNUNET_MST_destroy() from within
- * the scope of this callback.
- *
- * @param cls closure with the `struct GNUNET_SERVICE_Client *`
- * @param message the actual message
- * @return #GNUNET_OK on success, #GNUNET_SYSERR if the client was dropped
- */
- static int
- service_client_mst_cb (void *cls, const struct GNUNET_MessageHeader *message)
- {
- struct GNUNET_SERVICE_Client *client = cls;
- LOG (GNUNET_ERROR_TYPE_DEBUG,
- "Received message of type %u and size %u from client\n",
- ntohs (message->type),
- ntohs (message->size));
- GNUNET_assert (GNUNET_NO == client->needs_continue);
- client->needs_continue = GNUNET_YES;
- client->warn_type = ntohs (message->type);
- client->warn_start = GNUNET_TIME_absolute_get ();
- GNUNET_assert (NULL == client->warn_task);
- client->warn_task = GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_MINUTES,
- &warn_no_client_continue,
- client);
- GNUNET_MQ_inject_message (client->mq, message);
- if (NULL != client->drop_task)
- return GNUNET_SYSERR;
- return GNUNET_OK;
- }
- /**
- * A client sent us data. Receive and process it. If we are done,
- * reschedule this task.
- *
- * @param cls the `struct GNUNET_SERVICE_Client` that sent us data.
- */
- static void
- service_client_recv (void *cls)
- {
- struct GNUNET_SERVICE_Client *client = cls;
- int ret;
- client->recv_task = NULL;
- ret = GNUNET_MST_read (client->mst, client->sock, GNUNET_NO, GNUNET_YES);
- if (GNUNET_SYSERR == ret)
- {
- /* client closed connection (or IO error) */
- if (NULL == client->drop_task)
- {
- GNUNET_assert (GNUNET_NO == client->needs_continue);
- GNUNET_SERVICE_client_drop (client);
- }
- return;
- }
- if (GNUNET_NO == ret)
- return; /* more messages in buffer, wait for application
- to be done processing */
- GNUNET_assert (GNUNET_OK == ret);
- if (GNUNET_YES == client->needs_continue)
- return;
- if (NULL != client->recv_task)
- return;
- /* MST needs more data, re-schedule read job */
- client->recv_task =
- GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
- client->sock,
- &service_client_recv,
- client);
- }
- /**
- * We have successfully accepted a connection from a client. Now
- * setup the client (with the scheduler) and tell the application.
- *
- * @param sh service that accepted the client
- * @param sock socket associated with the client
- */
- static void
- start_client (struct GNUNET_SERVICE_Handle *sh,
- struct GNUNET_NETWORK_Handle *csock)
- {
- struct GNUNET_SERVICE_Client *client;
- client = GNUNET_new (struct GNUNET_SERVICE_Client);
- GNUNET_CONTAINER_DLL_insert (sh->clients_head, sh->clients_tail, client);
- client->sh = sh;
- client->sock = csock;
- client->mq = GNUNET_MQ_queue_for_callbacks (&service_mq_send,
- NULL,
- &service_mq_cancel,
- client,
- sh->handlers,
- &service_mq_error_handler,
- client);
- client->mst = GNUNET_MST_create (&service_client_mst_cb, client);
- if (NULL != sh->connect_cb)
- client->user_context = sh->connect_cb (sh->cb_cls, client, client->mq);
- GNUNET_MQ_set_handlers_closure (client->mq, client->user_context);
- client->recv_task =
- GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
- client->sock,
- &service_client_recv,
- client);
- }
- /**
- * We have a client. Accept the incoming socket(s) (and reschedule
- * the listen task).
- *
- * @param cls the `struct ServiceListenContext` of the ready listen socket
- */
- static void
- accept_client (void *cls)
- {
- struct ServiceListenContext *slc = cls;
- struct GNUNET_SERVICE_Handle *sh = slc->sh;
- slc->listen_task = NULL;
- while (1)
- {
- struct GNUNET_NETWORK_Handle *sock;
- const struct sockaddr_in *v4;
- const struct sockaddr_in6 *v6;
- struct sockaddr_storage sa;
- socklen_t addrlen;
- int ok;
- addrlen = sizeof(sa);
- sock = GNUNET_NETWORK_socket_accept (slc->listen_socket,
- (struct sockaddr *) &sa,
- &addrlen);
- if (NULL == sock)
- {
- if (EMFILE == errno)
- do_suspend (sh, SUSPEND_STATE_EMFILE);
- else if (EAGAIN != errno)
- GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "accept");
- break;
- }
- switch (sa.ss_family)
- {
- case AF_INET:
- GNUNET_assert (addrlen == sizeof(struct sockaddr_in));
- v4 = (const struct sockaddr_in *) &sa;
- ok = (((NULL == sh->v4_allowed) ||
- (check_ipv4_listed (sh->v4_allowed, &v4->sin_addr))) &&
- ((NULL == sh->v4_denied) ||
- (! check_ipv4_listed (sh->v4_denied, &v4->sin_addr))));
- break;
- case AF_INET6:
- GNUNET_assert (addrlen == sizeof(struct sockaddr_in6));
- v6 = (const struct sockaddr_in6 *) &sa;
- ok = (((NULL == sh->v6_allowed) ||
- (check_ipv6_listed (sh->v6_allowed, &v6->sin6_addr))) &&
- ((NULL == sh->v6_denied) ||
- (! check_ipv6_listed (sh->v6_denied, &v6->sin6_addr))));
- break;
- case AF_UNIX:
- ok = GNUNET_OK; /* controlled using file-system ACL now */
- break;
- default:
- LOG (GNUNET_ERROR_TYPE_WARNING,
- _ ("Unknown address family %d\n"),
- sa.ss_family);
- return;
- }
- if (! ok)
- {
- LOG (GNUNET_ERROR_TYPE_DEBUG,
- "Service rejected incoming connection from %s due to policy.\n",
- GNUNET_a2s ((const struct sockaddr *) &sa, addrlen));
- GNUNET_break (GNUNET_OK == GNUNET_NETWORK_socket_close (sock));
- continue;
- }
- LOG (GNUNET_ERROR_TYPE_DEBUG,
- "Service accepted incoming connection from %s.\n",
- GNUNET_a2s ((const struct sockaddr *) &sa, addrlen));
- start_client (slc->sh, sock);
- }
- if (0 != sh->suspend_state)
- return;
- slc->listen_task =
- GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
- slc->listen_socket,
- &accept_client,
- slc);
- }
- /**
- * Resume accepting connections from the listen socket.
- *
- * @param sh service to resume accepting connections.
- * @param sr reason that is no longer causing the suspension,
- * or #SUSPEND_STATE_NONE on first startup
- */
- static void
- do_resume (struct GNUNET_SERVICE_Handle *sh, enum SuspendReason sr)
- {
- struct ServiceListenContext *slc;
- GNUNET_assert ((SUSPEND_STATE_NONE == sr) || (0 != (sh->suspend_state & sr)));
- sh->suspend_state -= sr;
- if (SUSPEND_STATE_NONE != sh->suspend_state)
- return;
- for (slc = sh->slc_head; NULL != slc; slc = slc->next)
- {
- GNUNET_assert (NULL == slc->listen_task);
- slc->listen_task =
- GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
- slc->listen_socket,
- &accept_client,
- slc);
- }
- }
- /**
- * First task run by any service. Initializes our shutdown task,
- * starts the listening operation on our listen sockets and launches
- * the custom logic of the application service.
- *
- * @param cls our `struct GNUNET_SERVICE_Handle`
- */
- static void
- service_main (void *cls)
- {
- struct GNUNET_SERVICE_Handle *sh = cls;
- if (GNUNET_SERVICE_OPTION_MANUAL_SHUTDOWN !=
- (sh->options & GNUNET_SERVICE_OPTION_SHUTDOWN_BITMASK))
- GNUNET_SCHEDULER_add_shutdown (&service_shutdown, sh);
- do_resume (sh, SUSPEND_STATE_NONE);
- if (-1 != sh->ready_confirm_fd)
- {
- GNUNET_break (1 == write (sh->ready_confirm_fd, ".", 1));
- GNUNET_break (0 == close (sh->ready_confirm_fd));
- sh->ready_confirm_fd = -1;
- }
- if (NULL != sh->service_init_cb)
- sh->service_init_cb (sh->cb_cls, sh->cfg, sh);
- }
- /**
- * Parse an IPv4 access control list.
- *
- * @param ret location where to write the ACL (set)
- * @param sh service context to use to get the configuration
- * @param option name of the ACL option to parse
- * @return #GNUNET_SYSERR on parse error, #GNUNET_OK on success (including
- * no ACL configured)
- */
- static int
- process_acl4 (struct GNUNET_STRINGS_IPv4NetworkPolicy **ret,
- struct GNUNET_SERVICE_Handle *sh,
- const char *option)
- {
- char *opt;
- if (! GNUNET_CONFIGURATION_have_value (sh->cfg, sh->service_name, option))
- {
- *ret = NULL;
- return GNUNET_OK;
- }
- GNUNET_break (GNUNET_OK ==
- GNUNET_CONFIGURATION_get_value_string (sh->cfg,
- sh->service_name,
- option,
- &opt));
- if (NULL == (*ret = GNUNET_STRINGS_parse_ipv4_policy (opt)))
- {
- LOG (GNUNET_ERROR_TYPE_WARNING,
- _ ("Could not parse IPv4 network specification `%s' for `%s:%s'\n"),
- opt,
- sh->service_name,
- option);
- GNUNET_free (opt);
- return GNUNET_SYSERR;
- }
- GNUNET_free (opt);
- return GNUNET_OK;
- }
- /**
- * Parse an IPv6 access control list.
- *
- * @param ret location where to write the ACL (set)
- * @param sh service context to use to get the configuration
- * @param option name of the ACL option to parse
- * @return #GNUNET_SYSERR on parse error, #GNUNET_OK on success (including
- * no ACL configured)
- */
- static int
- process_acl6 (struct GNUNET_STRINGS_IPv6NetworkPolicy **ret,
- struct GNUNET_SERVICE_Handle *sh,
- const char *option)
- {
- char *opt;
- if (! GNUNET_CONFIGURATION_have_value (sh->cfg, sh->service_name, option))
- {
- *ret = NULL;
- return GNUNET_OK;
- }
- GNUNET_break (GNUNET_OK ==
- GNUNET_CONFIGURATION_get_value_string (sh->cfg,
- sh->service_name,
- option,
- &opt));
- if (NULL == (*ret = GNUNET_STRINGS_parse_ipv6_policy (opt)))
- {
- LOG (GNUNET_ERROR_TYPE_WARNING,
- _ ("Could not parse IPv6 network specification `%s' for `%s:%s'\n"),
- opt,
- sh->service_name,
- option);
- GNUNET_free (opt);
- return GNUNET_SYSERR;
- }
- GNUNET_free (opt);
- return GNUNET_OK;
- }
- /**
- * Add the given UNIX domain path as an address to the
- * list (as the first entry).
- *
- * @param saddrs array to update
- * @param saddrlens where to store the address length
- * @param unixpath path to add
- */
- static void
- add_unixpath (struct sockaddr **saddrs,
- socklen_t *saddrlens,
- const char *unixpath)
- {
- #ifdef AF_UNIX
- struct sockaddr_un *un;
- un = GNUNET_new (struct sockaddr_un);
- un->sun_family = AF_UNIX;
- GNUNET_strlcpy (un->sun_path, unixpath, sizeof(un->sun_path));
- #if HAVE_SOCKADDR_UN_SUN_LEN
- un->sun_len = (u_char) sizeof(struct sockaddr_un);
- #endif
- *saddrs = (struct sockaddr *) un;
- *saddrlens = sizeof(struct sockaddr_un);
- #else
- /* this function should never be called
- * unless AF_UNIX is defined! */
- GNUNET_assert (0);
- #endif
- }
- /**
- * Get the list of addresses that a server for the given service
- * should bind to.
- *
- * @param service_name name of the service
- * @param cfg configuration (which specifies the addresses)
- * @param addrs set (call by reference) to an array of pointers to the
- * addresses the server should bind to and listen on; the
- * array will be NULL-terminated (on success)
- * @param addr_lens set (call by reference) to an array of the lengths
- * of the respective `struct sockaddr` struct in the @a addrs
- * array (on success)
- * @return number of addresses found on success,
- * #GNUNET_SYSERR if the configuration
- * did not specify reasonable finding information or
- * if it specified a hostname that could not be resolved;
- * #GNUNET_NO if the number of addresses configured is
- * zero (in this case, `*addrs` and `*addr_lens` will be
- * set to NULL).
- */
- static int
- get_server_addresses (const char *service_name,
- const struct GNUNET_CONFIGURATION_Handle *cfg,
- struct sockaddr ***addrs,
- socklen_t **addr_lens)
- {
- int disablev6;
- struct GNUNET_NETWORK_Handle *desc;
- unsigned long long port;
- char *unixpath;
- struct addrinfo hints;
- struct addrinfo *res;
- struct addrinfo *pos;
- struct addrinfo *next;
- unsigned int i;
- int resi;
- int ret;
- struct sockaddr **saddrs;
- socklen_t *saddrlens;
- char *hostname;
- *addrs = NULL;
- *addr_lens = NULL;
- desc = NULL;
- disablev6 = GNUNET_NO;
- if ((GNUNET_NO == GNUNET_NETWORK_test_pf (PF_INET6)) ||
- (GNUNET_YES ==
- GNUNET_CONFIGURATION_get_value_yesno (cfg, service_name, "DISABLEV6")))
- disablev6 = GNUNET_YES;
- port = 0;
- if (GNUNET_CONFIGURATION_have_value (cfg, service_name, "PORT"))
- {
- if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_number (cfg,
- service_name,
- "PORT",
- &port))
- {
- LOG (GNUNET_ERROR_TYPE_ERROR,
- _ ("Require valid port number for service `%s' in configuration!\n"),
- service_name);
- }
- if (port > 65535)
- {
- LOG (GNUNET_ERROR_TYPE_ERROR,
- _ ("Require valid port number for service `%s' in configuration!\n"),
- service_name);
- return GNUNET_SYSERR;
- }
- }
- if (GNUNET_CONFIGURATION_have_value (cfg, service_name, "BINDTO"))
- {
- GNUNET_break (GNUNET_OK ==
- GNUNET_CONFIGURATION_get_value_string (cfg,
- service_name,
- "BINDTO",
- &hostname));
- }
- else
- hostname = NULL;
- unixpath = NULL;
- #ifdef AF_UNIX
- if ((GNUNET_YES ==
- GNUNET_CONFIGURATION_have_value (cfg, service_name, "UNIXPATH")) &&
- (GNUNET_OK == GNUNET_CONFIGURATION_get_value_filename (cfg,
- service_name,
- "UNIXPATH",
- &unixpath)) &&
- (0 < strlen (unixpath)))
- {
- /* probe UNIX support */
- struct sockaddr_un s_un;
- if (strlen (unixpath) >= sizeof(s_un.sun_path))
- {
- LOG (GNUNET_ERROR_TYPE_WARNING,
- _ ("UNIXPATH `%s' too long, maximum length is %llu\n"),
- unixpath,
- (unsigned long long) sizeof(s_un.sun_path));
- unixpath = GNUNET_NETWORK_shorten_unixpath (unixpath);
- LOG (GNUNET_ERROR_TYPE_INFO, _ ("Using `%s' instead\n"), unixpath);
- }
- if (GNUNET_OK != GNUNET_DISK_directory_create_for_file (unixpath))
- GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_ERROR, "mkdir", unixpath);
- }
- if (NULL != unixpath)
- {
- desc = GNUNET_NETWORK_socket_create (AF_UNIX, SOCK_STREAM, 0);
- if (NULL == desc)
- {
- if ((ENOBUFS == errno) || (ENOMEM == errno) || (ENFILE == errno) ||
- (EACCES == errno))
- {
- LOG_STRERROR (GNUNET_ERROR_TYPE_ERROR, "socket");
- GNUNET_free (hostname);
- GNUNET_free (unixpath);
- return GNUNET_SYSERR;
- }
- LOG (GNUNET_ERROR_TYPE_INFO,
- _ (
- "Disabling UNIX domain socket support for service `%s', failed to create UNIX domain socket: %s\n"),
- service_name,
- strerror (errno));
- GNUNET_free (unixpath);
- unixpath = NULL;
- }
- else
- {
- GNUNET_break (GNUNET_OK == GNUNET_NETWORK_socket_close (desc));
- desc = NULL;
- }
- }
- #endif
- if ((0 == port) && (NULL == unixpath))
- {
- LOG (GNUNET_ERROR_TYPE_ERROR,
- _ (
- "Have neither PORT nor UNIXPATH for service `%s', but one is required\n"),
- service_name);
- GNUNET_free (hostname);
- return GNUNET_SYSERR;
- }
- if (0 == port)
- {
- saddrs = GNUNET_new_array (2, struct sockaddr *);
- saddrlens = GNUNET_new_array (2, socklen_t);
- add_unixpath (saddrs, saddrlens, unixpath);
- GNUNET_free (unixpath);
- GNUNET_free (hostname);
- *addrs = saddrs;
- *addr_lens = saddrlens;
- return 1;
- }
- if (NULL != hostname)
- {
- LOG (GNUNET_ERROR_TYPE_DEBUG,
- "Resolving `%s' since that is where `%s' will bind to.\n",
- hostname,
- service_name);
- memset (&hints, 0, sizeof(struct addrinfo));
- if (disablev6)
- hints.ai_family = AF_INET;
- hints.ai_protocol = IPPROTO_TCP;
- if ((0 != (ret = getaddrinfo (hostname, NULL, &hints, &res))) ||
- (NULL == res))
- {
- LOG (GNUNET_ERROR_TYPE_ERROR,
- _ ("Failed to resolve `%s': %s\n"),
- hostname,
- gai_strerror (ret));
- GNUNET_free (hostname);
- GNUNET_free (unixpath);
- return GNUNET_SYSERR;
- }
- next = res;
- i = 0;
- while (NULL != (pos = next))
- {
- next = pos->ai_next;
- if ((disablev6) && (pos->ai_family == AF_INET6))
- continue;
- i++;
- }
- if (0 == i)
- {
- LOG (GNUNET_ERROR_TYPE_ERROR,
- _ ("Failed to find %saddress for `%s'.\n"),
- disablev6 ? "IPv4 " : "",
- hostname);
- freeaddrinfo (res);
- GNUNET_free (hostname);
- GNUNET_free (unixpath);
- return GNUNET_SYSERR;
- }
- resi = i;
- if (NULL != unixpath)
- resi++;
- saddrs = GNUNET_new_array (resi + 1, struct sockaddr *);
- saddrlens = GNUNET_new_array (resi + 1, socklen_t);
- i = 0;
- if (NULL != unixpath)
- {
- add_unixpath (saddrs, saddrlens, unixpath);
- i++;
- }
- next = res;
- while (NULL != (pos = next))
- {
- next = pos->ai_next;
- if ((disablev6) && (AF_INET6 == pos->ai_family))
- continue;
- if ((IPPROTO_TCP != pos->ai_protocol) && (0 != pos->ai_protocol))
- continue; /* not TCP */
- if ((SOCK_STREAM != pos->ai_socktype) && (0 != pos->ai_socktype))
- continue; /* huh? */
- LOG (GNUNET_ERROR_TYPE_DEBUG,
- "Service `%s' will bind to `%s'\n",
- service_name,
- GNUNET_a2s (pos->ai_addr, pos->ai_addrlen));
- if (AF_INET == pos->ai_family)
- {
- GNUNET_assert (sizeof(struct sockaddr_in) == pos->ai_addrlen);
- saddrlens[i] = pos->ai_addrlen;
- saddrs[i] = GNUNET_malloc (saddrlens[i]);
- GNUNET_memcpy (saddrs[i], pos->ai_addr, saddrlens[i]);
- ((struct sockaddr_in *) saddrs[i])->sin_port = htons (port);
- }
- else
- {
- GNUNET_assert (AF_INET6 == pos->ai_family);
- GNUNET_assert (sizeof(struct sockaddr_in6) == pos->ai_addrlen);
- saddrlens[i] = pos->ai_addrlen;
- saddrs[i] = GNUNET_malloc (saddrlens[i]);
- GNUNET_memcpy (saddrs[i], pos->ai_addr, saddrlens[i]);
- ((struct sockaddr_in6 *) saddrs[i])->sin6_port = htons (port);
- }
- i++;
- }
- GNUNET_free (hostname);
- freeaddrinfo (res);
- resi = i;
- }
- else
- {
- /* will bind against everything, just set port */
- if (disablev6)
- {
- /* V4-only */
- resi = 1;
- if (NULL != unixpath)
- resi++;
- i = 0;
- saddrs = GNUNET_new_array (resi + 1, struct sockaddr *);
- saddrlens = GNUNET_new_array (resi + 1, socklen_t);
- if (NULL != unixpath)
- {
- add_unixpath (saddrs, saddrlens, unixpath);
- i++;
- }
- saddrlens[i] = sizeof(struct sockaddr_in);
- saddrs[i] = GNUNET_malloc (saddrlens[i]);
- #if HAVE_SOCKADDR_IN_SIN_LEN
- ((struct sockaddr_in *) saddrs[i])->sin_len = saddrlens[i];
- #endif
- ((struct sockaddr_in *) saddrs[i])->sin_family = AF_INET;
- ((struct sockaddr_in *) saddrs[i])->sin_port = htons (port);
- }
- else
- {
- /* dual stack */
- resi = 2;
- if (NULL != unixpath)
- resi++;
- saddrs = GNUNET_new_array (resi + 1, struct sockaddr *);
- saddrlens = GNUNET_new_array (resi + 1, socklen_t);
- i = 0;
- if (NULL != unixpath)
- {
- add_unixpath (saddrs, saddrlens, unixpath);
- i++;
- }
- saddrlens[i] = sizeof(struct sockaddr_in6);
- saddrs[i] = GNUNET_malloc (saddrlens[i]);
- #if HAVE_SOCKADDR_IN_SIN_LEN
- ((struct sockaddr_in6 *) saddrs[i])->sin6_len = saddrlens[0];
- #endif
- ((struct sockaddr_in6 *) saddrs[i])->sin6_family = AF_INET6;
- ((struct sockaddr_in6 *) saddrs[i])->sin6_port = htons (port);
- i++;
- saddrlens[i] = sizeof(struct sockaddr_in);
- saddrs[i] = GNUNET_malloc (saddrlens[i]);
- #if HAVE_SOCKADDR_IN_SIN_LEN
- ((struct sockaddr_in *) saddrs[i])->sin_len = saddrlens[1];
- #endif
- ((struct sockaddr_in *) saddrs[i])->sin_family = AF_INET;
- ((struct sockaddr_in *) saddrs[i])->sin_port = htons (port);
- }
- }
- GNUNET_free (unixpath);
- *addrs = saddrs;
- *addr_lens = saddrlens;
- return resi;
- }
- /**
- * Create and initialize a listen socket for the server.
- *
- * @param server_addr address to listen on
- * @param socklen length of @a server_addr
- * @return NULL on error, otherwise the listen socket
- */
- static struct GNUNET_NETWORK_Handle *
- open_listen_socket (const struct sockaddr *server_addr,
- socklen_t socklen)
- {
- struct GNUNET_NETWORK_Handle *sock;
- uint16_t port;
- int eno;
- switch (server_addr->sa_family)
- {
- case AF_INET:
- port = ntohs (((const struct sockaddr_in *) server_addr)->sin_port);
- break;
- case AF_INET6:
- port = ntohs (((const struct sockaddr_in6 *) server_addr)->sin6_port);
- break;
- case AF_UNIX:
- port = 0;
- break;
- default:
- GNUNET_break (0);
- port = 0;
- break;
- }
- sock = GNUNET_NETWORK_socket_create (server_addr->sa_family,
- SOCK_STREAM,
- 0);
- if (NULL == sock)
- {
- LOG_STRERROR (GNUNET_ERROR_TYPE_ERROR,
- "socket");
- errno = 0;
- return NULL;
- }
- /* bind the socket */
- if (GNUNET_OK !=
- GNUNET_NETWORK_socket_bind (sock,
- server_addr,
- socklen))
- {
- eno = errno;
- if (EADDRINUSE != errno)
- {
- /* we don't log 'EADDRINUSE' here since an IPv4 bind may
- * fail if we already took the port on IPv6; if both IPv4 and
- * IPv6 binds fail, then our caller will log using the
- * errno preserved in 'eno' */
- if (0 != port)
- LOG (GNUNET_ERROR_TYPE_ERROR,
- _ ("`%s' failed for port %d (%s).\n"),
- "bind",
- port,
- (AF_INET == server_addr->sa_family) ? "IPv4" : "IPv6");
- else
- LOG_STRERROR (GNUNET_ERROR_TYPE_ERROR, "bind");
- eno = 0;
- }
- else
- {
- if (0 != port)
- LOG (GNUNET_ERROR_TYPE_WARNING,
- _ ("`%s' failed for port %d (%s): address already in use\n"),
- "bind",
- port,
- (AF_INET == server_addr->sa_family) ? "IPv4" : "IPv6");
- else if (AF_UNIX == server_addr->sa_family)
- {
- LOG (GNUNET_ERROR_TYPE_WARNING,
- _ ("`%s' failed for `%s': address already in use\n"),
- "bind",
- GNUNET_a2s (server_addr, socklen));
- }
- }
- GNUNET_break (GNUNET_OK == GNUNET_NETWORK_socket_close (sock));
- errno = eno;
- return NULL;
- }
- if (GNUNET_OK != GNUNET_NETWORK_socket_listen (sock, 5))
- {
- LOG_STRERROR (GNUNET_ERROR_TYPE_ERROR, "listen");
- GNUNET_break (GNUNET_OK == GNUNET_NETWORK_socket_close (sock));
- errno = 0;
- return NULL;
- }
- if (0 != port)
- LOG (GNUNET_ERROR_TYPE_DEBUG,
- "Server starts to listen on port %u.\n",
- port);
- return sock;
- }
- /**
- * Setup service handle
- *
- * Configuration may specify:
- * - PORT (where to bind to for TCP)
- * - UNIXPATH (where to bind to for UNIX domain sockets)
- * - DISABLEV6 (disable support for IPv6, otherwise we use dual-stack)
- * - BINDTO (hostname or IP address to bind to, otherwise we take everything)
- * - ACCEPT_FROM (only allow connections from specified IPv4 subnets)
- * - ACCEPT_FROM6 (only allow connections from specified IPv6 subnets)
- * - REJECT_FROM (disallow allow connections from specified IPv4 subnets)
- * - REJECT_FROM6 (disallow allow connections from specified IPv6 subnets)
- *
- * @param sh service context to initialize
- * @return #GNUNET_OK if configuration succeeded
- */
- static int
- setup_service (struct GNUNET_SERVICE_Handle *sh)
- {
- int tolerant;
- struct GNUNET_NETWORK_Handle **csocks = NULL;
- struct GNUNET_NETWORK_Handle **lsocks;
- const char *nfds;
- unsigned int cnt;
- int flags;
- char dummy[2];
- if (GNUNET_CONFIGURATION_have_value (sh->cfg,
- sh->service_name,
- "TOLERANT"))
- {
- if (GNUNET_SYSERR ==
- (tolerant = GNUNET_CONFIGURATION_get_value_yesno (sh->cfg,
- sh->service_name,
- "TOLERANT")))
- {
- LOG (GNUNET_ERROR_TYPE_ERROR,
- _ ("Specified value for `%s' of service `%s' is invalid\n"),
- "TOLERANT",
- sh->service_name);
- return GNUNET_SYSERR;
- }
- }
- else
- tolerant = GNUNET_NO;
- lsocks = NULL;
- errno = 0;
- if ((NULL != (nfds = getenv ("LISTEN_FDS"))) &&
- (1 == sscanf (nfds, "%u%1s", &cnt, dummy)) && (cnt > 0) &&
- (cnt < FD_SETSIZE) && (cnt + 4 < FD_SETSIZE))
- {
- lsocks = GNUNET_new_array (cnt + 1, struct GNUNET_NETWORK_Handle *);
- while (0 < cnt--)
- {
- flags = fcntl (3 + cnt, F_GETFD);
- if ((flags < 0) || (0 != (flags & FD_CLOEXEC)) ||
- (NULL == (lsocks[cnt] = GNUNET_NETWORK_socket_box_native (3 + cnt))))
- {
- LOG (GNUNET_ERROR_TYPE_ERROR,
- _ (
- "Could not access pre-bound socket %u, will try to bind myself\n"),
- (unsigned int) 3 + cnt);
- cnt++;
- while (NULL != lsocks[cnt])
- GNUNET_break (GNUNET_OK ==
- GNUNET_NETWORK_socket_close (lsocks[cnt++]));
- GNUNET_free (lsocks);
- lsocks = NULL;
- break;
- }
- }
- unsetenv ("LISTEN_FDS");
- }
- if ( (0 != (GNUNET_SERVICE_OPTION_CLOSE_LSOCKS & sh->options)) &&
- (NULL != lsocks) )
- {
- csocks = lsocks;
- lsocks = NULL;
- }
- if (NULL != lsocks)
- {
- /* listen only on inherited sockets if we have any */
- for (struct GNUNET_NETWORK_Handle **ls = lsocks; NULL != *ls; ls++)
- {
- struct ServiceListenContext *slc;
- slc = GNUNET_new (struct ServiceListenContext);
- slc->sh = sh;
- slc->listen_socket = *ls;
- GNUNET_CONTAINER_DLL_insert (sh->slc_head, sh->slc_tail, slc);
- }
- GNUNET_free (lsocks);
- }
- else
- {
- struct sockaddr **addrs;
- socklen_t *addrlens;
- int num;
- num = get_server_addresses (sh->service_name, sh->cfg, &addrs, &addrlens);
- if (GNUNET_SYSERR == num)
- return GNUNET_SYSERR;
- for (int i = 0; i < num; i++)
- {
- struct ServiceListenContext *slc;
- slc = GNUNET_new (struct ServiceListenContext);
- slc->sh = sh;
- slc->listen_socket = open_listen_socket (addrs[i], addrlens[i]);
- GNUNET_free (addrs[i]);
- if (NULL == slc->listen_socket)
- {
- GNUNET_log_strerror (GNUNET_ERROR_TYPE_ERROR, "bind");
- GNUNET_free (slc);
- continue;
- }
- GNUNET_CONTAINER_DLL_insert (sh->slc_head, sh->slc_tail, slc);
- }
- GNUNET_free (addrlens);
- GNUNET_free (addrs);
- if ((0 != num) && (NULL == sh->slc_head))
- {
- /* All attempts to bind failed, hard failure */
- GNUNET_log (
- GNUNET_ERROR_TYPE_ERROR,
- _ (
- "Could not bind to any of the ports I was supposed to, refusing to run!\n"));
- GNUNET_free (csocks);
- return GNUNET_SYSERR;
- }
- }
- if (NULL != csocks)
- {
- /* close inherited sockets to signal parent that we are ready */
- for (struct GNUNET_NETWORK_Handle **ls = csocks; NULL != *ls; ls++)
- GNUNET_NETWORK_socket_close (*ls);
- GNUNET_free (csocks);
- }
- sh->require_found = tolerant ? GNUNET_NO : GNUNET_YES;
- sh->match_uid = GNUNET_CONFIGURATION_get_value_yesno (sh->cfg,
- sh->service_name,
- "UNIX_MATCH_UID");
- sh->match_gid = GNUNET_CONFIGURATION_get_value_yesno (sh->cfg,
- sh->service_name,
- "UNIX_MATCH_GID");
- process_acl4 (&sh->v4_denied, sh, "REJECT_FROM");
- process_acl4 (&sh->v4_allowed, sh, "ACCEPT_FROM");
- process_acl6 (&sh->v6_denied, sh, "REJECT_FROM6");
- process_acl6 (&sh->v6_allowed, sh, "ACCEPT_FROM6");
- return GNUNET_OK;
- }
- /**
- * Get the name of the user that'll be used
- * to provide the service.
- *
- * @param sh service context
- * @return value of the 'USERNAME' option
- */
- static char *
- get_user_name (struct GNUNET_SERVICE_Handle *sh)
- {
- char *un;
- if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_filename (sh->cfg,
- sh->service_name,
- "USERNAME",
- &un))
- return NULL;
- return un;
- }
- /**
- * Set user ID.
- *
- * @param sh service context
- * @return #GNUNET_OK on success, #GNUNET_SYSERR on error
- */
- static int
- set_user_id (struct GNUNET_SERVICE_Handle *sh)
- {
- char *user;
- if (NULL == (user = get_user_name (sh)))
- return GNUNET_OK; /* keep */
- struct passwd *pws;
- errno = 0;
- pws = getpwnam (user);
- if (NULL == pws)
- {
- LOG (GNUNET_ERROR_TYPE_ERROR,
- _ ("Cannot obtain information about user `%s': %s\n"),
- user,
- errno == 0 ? _ ("No such user") : strerror (errno));
- GNUNET_free (user);
- return GNUNET_SYSERR;
- }
- if ((0 != setgid (pws->pw_gid)) || (0 != setegid (pws->pw_gid)) ||
- #if HAVE_INITGROUPS
- (0 != initgroups (user, pws->pw_gid)) ||
- #endif
- (0 != setuid (pws->pw_uid)) || (0 != seteuid (pws->pw_uid)))
- {
- if ((0 != setregid (pws->pw_gid, pws->pw_gid)) ||
- (0 != setreuid (pws->pw_uid, pws->pw_uid)))
- {
- LOG (GNUNET_ERROR_TYPE_ERROR,
- _ ("Cannot change user/group to `%s': %s\n"),
- user,
- strerror (errno));
- GNUNET_free (user);
- return GNUNET_SYSERR;
- }
- }
- GNUNET_free (user);
- return GNUNET_OK;
- }
- /**
- * Get the name of the file where we will
- * write the PID of the service.
- *
- * @param sh service context
- * @return name of the file for the process ID
- */
- static char *
- get_pid_file_name (struct GNUNET_SERVICE_Handle *sh)
- {
- char *pif;
- if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_filename (sh->cfg,
- sh->service_name,
- "PIDFILE",
- &pif))
- return NULL;
- return pif;
- }
- /**
- * Delete the PID file that was created by our parent.
- *
- * @param sh service context
- */
- static void
- pid_file_delete (struct GNUNET_SERVICE_Handle *sh)
- {
- char *pif = get_pid_file_name (sh);
- if (NULL == pif)
- return; /* no PID file */
- if (0 != unlink (pif))
- LOG_STRERROR_FILE (GNUNET_ERROR_TYPE_WARNING, "unlink", pif);
- GNUNET_free (pif);
- }
- /**
- * Detach from terminal.
- *
- * @param sh service context
- * @return #GNUNET_OK on success, #GNUNET_SYSERR on error
- */
- static int
- detach_terminal (struct GNUNET_SERVICE_Handle *sh)
- {
- pid_t pid;
- int nullfd;
- int filedes[2];
- if (0 != pipe (filedes))
- {
- LOG_STRERROR (GNUNET_ERROR_TYPE_ERROR, "pipe");
- return GNUNET_SYSERR;
- }
- pid = fork ();
- if (pid < 0)
- {
- LOG_STRERROR (GNUNET_ERROR_TYPE_ERROR, "fork");
- return GNUNET_SYSERR;
- }
- if (0 != pid)
- {
- /* Parent */
- char c;
- GNUNET_break (0 == close (filedes[1]));
- c = 'X';
- if (1 != read (filedes[0], &c, sizeof(char)))
- LOG_STRERROR (GNUNET_ERROR_TYPE_WARNING, "read");
- fflush (stdout);
- switch (c)
- {
- case '.':
- exit (0);
- case 'I':
- LOG (GNUNET_ERROR_TYPE_INFO,
- _ ("Service process failed to initialize\n"));
- break;
- case 'S':
- LOG (GNUNET_ERROR_TYPE_INFO,
- _ ("Service process could not initialize server function\n"));
- break;
- case 'X':
- LOG (GNUNET_ERROR_TYPE_INFO,
- _ ("Service process failed to report status\n"));
- break;
- }
- exit (1); /* child reported error */
- }
- GNUNET_break (0 == close (0));
- GNUNET_break (0 == close (1));
- GNUNET_break (0 == close (filedes[0]));
- nullfd = open ("/dev/null", O_RDWR | O_APPEND);
- if (nullfd < 0)
- return GNUNET_SYSERR;
- /* set stdin/stdout to /dev/null */
- if ((dup2 (nullfd, 0) < 0) || (dup2 (nullfd, 1) < 0))
- {
- LOG_STRERROR (GNUNET_ERROR_TYPE_ERROR, "dup2");
- (void) close (nullfd);
- return GNUNET_SYSERR;
- }
- (void) close (nullfd);
- /* Detach from controlling terminal */
- pid = setsid ();
- if (-1 == pid)
- LOG_STRERROR (GNUNET_ERROR_TYPE_ERROR, "setsid");
- sh->ready_confirm_fd = filedes[1];
- return GNUNET_OK;
- }
- /**
- * Tear down the service, closing the listen sockets and
- * freeing the ACLs.
- *
- * @param sh handle to the service to tear down.
- */
- static void
- teardown_service (struct GNUNET_SERVICE_Handle *sh)
- {
- struct ServiceListenContext *slc;
- GNUNET_free (sh->v4_denied);
- GNUNET_free (sh->v6_denied);
- GNUNET_free (sh->v4_allowed);
- GNUNET_free (sh->v6_allowed);
- while (NULL != (slc = sh->slc_head))
- {
- GNUNET_CONTAINER_DLL_remove (sh->slc_head, sh->slc_tail, slc);
- if (NULL != slc->listen_task)
- GNUNET_SCHEDULER_cancel (slc->listen_task);
- GNUNET_break (GNUNET_OK ==
- GNUNET_NETWORK_socket_close (slc->listen_socket));
- GNUNET_free (slc);
- }
- }
- /**
- * Function to return link to AGPL source upon request.
- *
- * @param cls closure with the identification of the client
- * @param msg AGPL request
- */
- static void
- return_agpl (void *cls, const struct GNUNET_MessageHeader *msg)
- {
- struct GNUNET_SERVICE_Client *client = cls;
- struct GNUNET_MQ_Handle *mq;
- struct GNUNET_MQ_Envelope *env;
- struct GNUNET_MessageHeader *res;
- size_t slen;
- const struct GNUNET_OS_ProjectData *pd = GNUNET_OS_project_data_get ();
- (void) msg;
- slen = strlen (pd->agpl_url) + 1;
- env = GNUNET_MQ_msg_extra (res, GNUNET_MESSAGE_TYPE_RESPONSE_AGPL, slen);
- memcpy (&res[1], GNUNET_AGPL_URL, slen);
- mq = GNUNET_SERVICE_client_get_mq (client);
- GNUNET_MQ_send (mq, env);
- GNUNET_SERVICE_client_continue (client);
- }
- /**
- * Low-level function to start a service if the scheduler
- * is already running. Should only be used directly in
- * special cases.
- *
- * The function will launch the service with the name @a service_name
- * using the @a service_options to configure its shutdown
- * behavior. When clients connect or disconnect, the respective
- * @a connect_cb or @a disconnect_cb functions will be called. For
- * messages received from the clients, the respective @a handlers will
- * be invoked; for the closure of the handlers we use the return value
- * from the @a connect_cb invocation of the respective client.
- *
- * Each handler MUST call #GNUNET_SERVICE_client_continue() after each
- * message to receive further messages from this client. If
- * #GNUNET_SERVICE_client_continue() is not called within a short
- * time, a warning will be logged. If delays are expected, services
- * should call #GNUNET_SERVICE_client_disable_continue_warning() to
- * disable the warning.
- *
- * Clients sending invalid messages (based on @a handlers) will be
- * dropped. Additionally, clients can be dropped at any time using
- * #GNUNET_SERVICE_client_drop().
- *
- * The service must be stopped using #GNUNET_SERVICE_stop().
- *
- * @param service_name name of the service to run
- * @param cfg configuration to use
- * @param connect_cb function to call whenever a client connects
- * @param disconnect_cb function to call whenever a client disconnects
- * @param cls closure argument for @a connect_cb and @a disconnect_cb
- * @param handlers NULL-terminated array of message handlers for the service,
- * the closure will be set to the value returned by
- * the @a connect_cb for the respective connection
- * @return NULL on error
- */
- struct GNUNET_SERVICE_Handle *
- GNUNET_SERVICE_start (const char *service_name,
- const struct GNUNET_CONFIGURATION_Handle *cfg,
- GNUNET_SERVICE_ConnectHandler connect_cb,
- GNUNET_SERVICE_DisconnectHandler disconnect_cb,
- void *cls,
- const struct GNUNET_MQ_MessageHandler *handlers)
- {
- struct GNUNET_SERVICE_Handle *sh;
- sh = GNUNET_new (struct GNUNET_SERVICE_Handle);
- sh->service_name = service_name;
- sh->cfg = cfg;
- sh->connect_cb = connect_cb;
- sh->disconnect_cb = disconnect_cb;
- sh->cb_cls = cls;
- sh->handlers = GNUNET_MQ_copy_handlers2 (handlers, &return_agpl, NULL);
- if (GNUNET_OK != setup_service (sh))
- {
- GNUNET_free (sh->handlers);
- GNUNET_free (sh);
- return NULL;
- }
- do_resume (sh, SUSPEND_STATE_NONE);
- return sh;
- }
- /**
- * Stops a service that was started with #GNUNET_SERVICE_start().
- *
- * @param srv service to stop
- */
- void
- GNUNET_SERVICE_stop (struct GNUNET_SERVICE_Handle *srv)
- {
- struct GNUNET_SERVICE_Client *client;
- GNUNET_SERVICE_suspend (srv);
- while (NULL != (client = srv->clients_head))
- GNUNET_SERVICE_client_drop (client);
- teardown_service (srv);
- GNUNET_free (srv->handlers);
- GNUNET_free (srv);
- }
- /**
- * Creates the "main" function for a GNUnet service. You
- * should almost always use the #GNUNET_SERVICE_MAIN macro
- * instead of calling this function directly (except
- * for ARM, which should call this function directly).
- *
- * The function will launch the service with the name @a service_name
- * using the @a service_options to configure its shutdown
- * behavior. Once the service is ready, the @a init_cb will be called
- * for service-specific initialization. @a init_cb will be given the
- * service handler which can be used to control the service's
- * availability. When clients connect or disconnect, the respective
- * @a connect_cb or @a disconnect_cb functions will be called. For
- * messages received from the clients, the respective @a handlers will
- * be invoked; for the closure of the handlers we use the return value
- * from the @a connect_cb invocation of the respective client.
- *
- * Each handler MUST call #GNUNET_SERVICE_client_continue() after each
- * message to receive further messages from this client. If
- * #GNUNET_SERVICE_client_continue() is not called within a short
- * time, a warning will be logged. If delays are expected, services
- * should call #GNUNET_SERVICE_client_disable_continue_warning() to
- * disable the warning.
- *
- * Clients sending invalid messages (based on @a handlers) will be
- * dropped. Additionally, clients can be dropped at any time using
- * #GNUNET_SERVICE_client_drop().
- *
- * @param argc number of command-line arguments in @a argv
- * @param argv array of command-line arguments
- * @param service_name name of the service to run
- * @param options options controlling shutdown of the service
- * @param service_init_cb function to call once the service is ready
- * @param connect_cb function to call whenever a client connects
- * @param disconnect_cb function to call whenever a client disconnects
- * @param cls closure argument for @a service_init_cb, @a connect_cb and @a disconnect_cb
- * @param handlers NULL-terminated array of message handlers for the service,
- * the closure will be set to the value returned by
- * the @a connect_cb for the respective connection
- * @return 0 on success, non-zero on error
- */
- int
- GNUNET_SERVICE_run_ (int argc,
- char *const *argv,
- const char *service_name,
- enum GNUNET_SERVICE_Options options,
- GNUNET_SERVICE_InitCallback service_init_cb,
- GNUNET_SERVICE_ConnectHandler connect_cb,
- GNUNET_SERVICE_DisconnectHandler disconnect_cb,
- void *cls,
- const struct GNUNET_MQ_MessageHandler *handlers)
- {
- struct GNUNET_SERVICE_Handle sh;
- #if ENABLE_NLS
- char *path;
- #endif
- char *cfg_filename;
- char *opt_cfg_filename;
- char *loglev;
- const char *xdg;
- char *logfile;
- int do_daemonize;
- unsigned long long skew_offset;
- unsigned long long skew_variance;
- long long clock_offset;
- struct GNUNET_CONFIGURATION_Handle *cfg;
- int ret;
- int err;
- const struct GNUNET_OS_ProjectData *pd = GNUNET_OS_project_data_get ();
- struct GNUNET_GETOPT_CommandLineOption service_options[] = {
- GNUNET_GETOPT_option_cfgfile (&opt_cfg_filename),
- GNUNET_GETOPT_option_flag ('d',
- "daemonize",
- gettext_noop (
- "do daemonize (detach from terminal)"),
- &do_daemonize),
- GNUNET_GETOPT_option_help (NULL),
- GNUNET_GETOPT_option_loglevel (&loglev),
- GNUNET_GETOPT_option_logfile (&logfile),
- GNUNET_GETOPT_option_version (pd->version),
- GNUNET_GETOPT_OPTION_END
- };
- err = 1;
- memset (&sh, 0, sizeof(sh));
- xdg = getenv ("XDG_CONFIG_HOME");
- if (NULL != xdg)
- GNUNET_asprintf (&cfg_filename,
- "%s%s%s",
- xdg,
- DIR_SEPARATOR_STR,
- pd->config_file);
- else
- cfg_filename = GNUNET_strdup (pd->user_config_file);
- sh.ready_confirm_fd = -1;
- sh.options = options;
- sh.cfg = cfg = GNUNET_CONFIGURATION_create ();
- sh.service_init_cb = service_init_cb;
- sh.connect_cb = connect_cb;
- sh.disconnect_cb = disconnect_cb;
- sh.cb_cls = cls;
- sh.handlers = (NULL == pd->agpl_url)
- ? GNUNET_MQ_copy_handlers (handlers)
- : GNUNET_MQ_copy_handlers2 (handlers, &return_agpl, NULL);
- sh.service_name = service_name;
- sh.ret = 0;
- /* setup subsystems */
- loglev = NULL;
- logfile = NULL;
- opt_cfg_filename = NULL;
- do_daemonize = 0;
- #if ENABLE_NLS
- if (NULL != pd->gettext_domain)
- {
- setlocale (LC_ALL, "");
- path = (NULL == pd->gettext_path) ?
- GNUNET_OS_installation_get_path (GNUNET_OS_IPK_LOCALEDIR) :
- GNUNET_strdup (pd->gettext_path);
- if (NULL != path)
- {
- bindtextdomain (pd->gettext_domain, path);
- GNUNET_free (path);
- }
- textdomain (pd->gettext_domain);
- }
- #endif
- ret = GNUNET_GETOPT_run (service_name,
- service_options,
- argc,
- argv);
- if (GNUNET_SYSERR == ret)
- goto shutdown;
- if (GNUNET_NO == ret)
- {
- err = 0;
- goto shutdown;
- }
- if (GNUNET_OK != GNUNET_log_setup (service_name,
- loglev,
- logfile))
- {
- GNUNET_break (0);
- goto shutdown;
- }
- if (NULL != opt_cfg_filename)
- {
- if ((GNUNET_YES != GNUNET_DISK_file_test (opt_cfg_filename)) ||
- (GNUNET_SYSERR == GNUNET_CONFIGURATION_load (cfg, opt_cfg_filename)))
- {
- GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
- _ ("Malformed configuration file `%s', exit ...\n"),
- opt_cfg_filename);
- goto shutdown;
- }
- }
- else
- {
- if (GNUNET_YES == GNUNET_DISK_file_test (cfg_filename))
- {
- if (GNUNET_SYSERR == GNUNET_CONFIGURATION_load (cfg, cfg_filename))
- {
- GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
- _ ("Malformed configuration file `%s', exit ...\n"),
- cfg_filename);
- goto shutdown;
- }
- }
- else
- {
- if (GNUNET_SYSERR == GNUNET_CONFIGURATION_load (cfg, NULL))
- {
- GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
- _ ("Malformed configuration, exit ...\n"));
- goto shutdown;
- }
- }
- }
- if (GNUNET_OK != setup_service (&sh))
- goto shutdown;
- if ((1 == do_daemonize) && (GNUNET_OK != detach_terminal (&sh)))
- {
- GNUNET_break (0);
- goto shutdown;
- }
- if (GNUNET_OK != set_user_id (&sh))
- goto shutdown;
- LOG (GNUNET_ERROR_TYPE_DEBUG,
- "Service `%s' runs with configuration from `%s'\n",
- service_name,
- (NULL != opt_cfg_filename) ? opt_cfg_filename : cfg_filename);
- if ((GNUNET_OK == GNUNET_CONFIGURATION_get_value_number (sh.cfg,
- "TESTING",
- "SKEW_OFFSET",
- &skew_offset)) &&
- (GNUNET_OK == GNUNET_CONFIGURATION_get_value_number (sh.cfg,
- "TESTING",
- "SKEW_VARIANCE",
- &skew_variance)))
- {
- clock_offset = skew_offset - skew_variance;
- GNUNET_TIME_set_offset (clock_offset);
- LOG (GNUNET_ERROR_TYPE_DEBUG, "Skewing clock by %dll ms\n", clock_offset);
- }
- GNUNET_RESOLVER_connect (sh.cfg);
- /* actually run service */
- err = 0;
- GNUNET_SCHEDULER_run (&service_main, &sh);
- /* shutdown */
- if (1 == do_daemonize)
- pid_file_delete (&sh);
- shutdown:
- if (-1 != sh.ready_confirm_fd)
- {
- if (1 != write (sh.ready_confirm_fd, err ? "I" : "S", 1))
- LOG_STRERROR (GNUNET_ERROR_TYPE_WARNING, "write");
- GNUNET_break (0 == close (sh.ready_confirm_fd));
- }
- #if HAVE_MALLINFO
- {
- char *counter;
- if ((GNUNET_YES == GNUNET_CONFIGURATION_have_value (sh.cfg,
- service_name,
- "GAUGER_HEAP")) &&
- (GNUNET_OK == GNUNET_CONFIGURATION_get_value_string (sh.cfg,
- service_name,
- "GAUGER_HEAP",
- &counter)))
- {
- struct mallinfo mi;
- mi = mallinfo ();
- GAUGER (service_name, counter, mi.usmblks, "blocks");
- GNUNET_free (counter);
- }
- }
- #endif
- teardown_service (&sh);
- GNUNET_free (sh.handlers);
- GNUNET_SPEEDUP_stop_ ();
- GNUNET_CONFIGURATION_destroy (cfg);
- GNUNET_free (logfile);
- GNUNET_free (loglev);
- GNUNET_free (cfg_filename);
- GNUNET_free (opt_cfg_filename);
- return err ? GNUNET_SYSERR : sh.ret;
- }
- /**
- * Suspend accepting connections from the listen socket temporarily.
- * Resume activity using #GNUNET_SERVICE_resume.
- *
- * @param sh service to stop accepting connections.
- */
- void
- GNUNET_SERVICE_suspend (struct GNUNET_SERVICE_Handle *sh)
- {
- do_suspend (sh, SUSPEND_STATE_APP);
- }
- /**
- * Resume accepting connections from the listen socket.
- *
- * @param sh service to resume accepting connections.
- */
- void
- GNUNET_SERVICE_resume (struct GNUNET_SERVICE_Handle *sh)
- {
- do_resume (sh, SUSPEND_STATE_APP);
- }
- /**
- * Task run to resume receiving data from the client after
- * the client called #GNUNET_SERVICE_client_continue().
- *
- * @param cls our `struct GNUNET_SERVICE_Client`
- */
- static void
- resume_client_receive (void *cls)
- {
- struct GNUNET_SERVICE_Client *c = cls;
- int ret;
- c->recv_task = NULL;
- /* first, check if there is still something in the buffer */
- ret = GNUNET_MST_next (c->mst, GNUNET_YES);
- if (GNUNET_SYSERR == ret)
- {
- if (NULL == c->drop_task)
- GNUNET_SERVICE_client_drop (c);
- return;
- }
- if (GNUNET_NO == ret)
- return; /* done processing, wait for more later */
- GNUNET_assert (GNUNET_OK == ret);
- if (GNUNET_YES == c->needs_continue)
- return; /* #GNUNET_MST_next() did give a message to the client */
- /* need to receive more data from the network first */
- if (NULL != c->recv_task)
- return;
- c->recv_task = GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
- c->sock,
- &service_client_recv,
- c);
- }
- /**
- * Continue receiving further messages from the given client.
- * Must be called after each message received.
- *
- * @param c the client to continue receiving from
- */
- void
- GNUNET_SERVICE_client_continue (struct GNUNET_SERVICE_Client *c)
- {
- GNUNET_assert (NULL == c->drop_task);
- GNUNET_assert (GNUNET_YES == c->needs_continue);
- GNUNET_assert (NULL == c->recv_task);
- c->needs_continue = GNUNET_NO;
- if (NULL != c->warn_task)
- {
- GNUNET_SCHEDULER_cancel (c->warn_task);
- c->warn_task = NULL;
- }
- c->recv_task = GNUNET_SCHEDULER_add_now (&resume_client_receive, c);
- }
- /**
- * Disable the warning the server issues if a message is not
- * acknowledged in a timely fashion. Use this call if a client is
- * intentionally delayed for a while. Only applies to the current
- * message.
- *
- * @param c client for which to disable the warning
- */
- void
- GNUNET_SERVICE_client_disable_continue_warning (struct GNUNET_SERVICE_Client *c)
- {
- GNUNET_break (NULL != c->warn_task);
- if (NULL != c->warn_task)
- {
- GNUNET_SCHEDULER_cancel (c->warn_task);
- c->warn_task = NULL;
- }
- }
- /**
- * Asynchronously finish dropping the client.
- *
- * @param cls the `struct GNUNET_SERVICE_Client`.
- */
- static void
- finish_client_drop (void *cls)
- {
- struct GNUNET_SERVICE_Client *c = cls;
- struct GNUNET_SERVICE_Handle *sh = c->sh;
- c->drop_task = NULL;
- GNUNET_assert (NULL == c->send_task);
- GNUNET_assert (NULL == c->recv_task);
- GNUNET_assert (NULL == c->warn_task);
- GNUNET_MST_destroy (c->mst);
- GNUNET_MQ_destroy (c->mq);
- if (GNUNET_NO == c->persist)
- {
- GNUNET_break (GNUNET_OK == GNUNET_NETWORK_socket_close (c->sock));
- if ((0 != (SUSPEND_STATE_EMFILE & sh->suspend_state)) &&
- (0 == (SUSPEND_STATE_SHUTDOWN & sh->suspend_state)))
- do_resume (sh, SUSPEND_STATE_EMFILE);
- }
- else
- {
- GNUNET_NETWORK_socket_free_memory_only_ (c->sock);
- }
- GNUNET_free (c);
- if ((0 != (SUSPEND_STATE_SHUTDOWN & sh->suspend_state)) &&
- (GNUNET_NO == have_non_monitor_clients (sh)))
- GNUNET_SERVICE_shutdown (sh);
- }
- /**
- * Ask the server to disconnect from the given client. This is the
- * same as returning #GNUNET_SYSERR within the check procedure when
- * handling a message, wexcept that it allows dropping of a client even
- * when not handling a message from that client. The `disconnect_cb`
- * will be called on @a c even if the application closes the connection
- * using this function.
- *
- * @param c client to disconnect now
- */
- void
- GNUNET_SERVICE_client_drop (struct GNUNET_SERVICE_Client *c)
- {
- struct GNUNET_SERVICE_Handle *sh = c->sh;
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Client dropped: %p (MQ: %p)\n",
- c,
- c->mq);
- #if EXECINFO
- {
- void *backtrace_array[MAX_TRACE_DEPTH];
- int num_backtrace_strings = backtrace (backtrace_array, MAX_TRACE_DEPTH);
- char **backtrace_strings =
- backtrace_symbols (backtrace_array, t->num_backtrace_strings);
- for (unsigned int i = 0; i < num_backtrace_strings; i++)
- LOG (GNUNET_ERROR_TYPE_DEBUG,
- "client drop trace %u: %s\n",
- i,
- backtrace_strings[i]);
- }
- #endif
- if (NULL != c->drop_task)
- {
- /* asked to drop twice! */
- GNUNET_assert (0);
- return;
- }
- GNUNET_CONTAINER_DLL_remove (sh->clients_head,
- sh->clients_tail,
- c);
- if (NULL != sh->disconnect_cb)
- sh->disconnect_cb (sh->cb_cls,
- c,
- c->user_context);
- if (NULL != c->warn_task)
- {
- GNUNET_SCHEDULER_cancel (c->warn_task);
- c->warn_task = NULL;
- }
- if (NULL != c->recv_task)
- {
- GNUNET_SCHEDULER_cancel (c->recv_task);
- c->recv_task = NULL;
- }
- if (NULL != c->send_task)
- {
- GNUNET_SCHEDULER_cancel (c->send_task);
- c->send_task = NULL;
- }
- c->drop_task = GNUNET_SCHEDULER_add_now (&finish_client_drop, c);
- }
- /**
- * Explicitly stops the service.
- *
- * @param sh server to shutdown
- */
- void
- GNUNET_SERVICE_shutdown (struct GNUNET_SERVICE_Handle *sh)
- {
- struct GNUNET_SERVICE_Client *client;
- if (0 == (sh->suspend_state & SUSPEND_STATE_SHUTDOWN))
- do_suspend (sh, SUSPEND_STATE_SHUTDOWN);
- while (NULL != (client = sh->clients_head))
- GNUNET_SERVICE_client_drop (client);
- }
- /**
- * Set the 'monitor' flag on this client. Clients which have been
- * marked as 'monitors' won't prevent the server from shutting down
- * once #GNUNET_SERVICE_stop_listening() has been invoked. The idea is
- * that for "normal" clients we likely want to allow them to process
- * their requests; however, monitor-clients are likely to 'never'
- * disconnect during shutdown and thus will not be considered when
- * determining if the server should continue to exist after
- * shutdown has been triggered.
- *
- * @param c client to mark as a monitor
- */
- void
- GNUNET_SERVICE_client_mark_monitor (struct GNUNET_SERVICE_Client *c)
- {
- c->is_monitor = GNUNET_YES;
- if (((0 != (SUSPEND_STATE_SHUTDOWN & c->sh->suspend_state)) &&
- (GNUNET_NO == have_non_monitor_clients (c->sh))))
- GNUNET_SERVICE_shutdown (c->sh);
- }
- /**
- * Set the persist option on this client. Indicates that the
- * underlying socket or fd should never really be closed. Used for
- * indicating process death.
- *
- * @param c client to persist the socket (never to be closed)
- */
- void
- GNUNET_SERVICE_client_persist (struct GNUNET_SERVICE_Client *c)
- {
- c->persist = GNUNET_YES;
- }
- /**
- * Obtain the message queue of @a c. Convenience function.
- *
- * @param c the client to continue receiving from
- * @return the message queue of @a c
- */
- struct GNUNET_MQ_Handle *
- GNUNET_SERVICE_client_get_mq (struct GNUNET_SERVICE_Client *c)
- {
- return c->mq;
- }
- /* end of service.c */
|