1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130 |
- /*
- This file is part of GNUnet.
- (C) 2001--2012 Christian Grothoff (and other contributing authors)
- GNUnet is free software; you can redistribute it and/or modify
- it under the terms of the GNU General Public License as published
- by the Free Software Foundation; either version 3, 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
- General Public License for more details.
- You should have received a copy of the GNU General Public License
- along with GNUnet; see the file COPYING. If not, write to the
- Free Software Foundation, Inc., 59 Temple Place - Suite 330,
- Boston, MA 02111-1307, USA.
- */
- /**
- * @file fs/fs_api.c
- * @brief main FS functions (master initialization, serialization, deserialization, shared code)
- * @author Christian Grothoff
- */
- #include "platform.h"
- #include "gnunet_util_lib.h"
- #include "gnunet_fs_service.h"
- #include "fs_api.h"
- #include "fs_tree.h"
- /**
- * How many block requests can we have outstanding in parallel at a time by default?
- */
- #define DEFAULT_MAX_PARALLEL_REQUESTS (1024 * 10)
- /**
- * How many downloads can we have outstanding in parallel at a time by default?
- */
- #define DEFAULT_MAX_PARALLEL_DOWNLOADS 16
- /**
- * Start the given job (send signal, remove from pending queue, update
- * counters and state).
- *
- * @param qe job to start
- */
- static void
- start_job (struct GNUNET_FS_QueueEntry *qe)
- {
- GNUNET_assert (NULL == qe->client);
- qe->client = GNUNET_CLIENT_connect ("fs", qe->h->cfg);
- if (NULL == qe->client)
- {
- GNUNET_break (0);
- return;
- }
- qe->start (qe->cls, qe->client);
- qe->start_times++;
- qe->h->active_blocks += qe->blocks;
- qe->h->active_downloads++;
- qe->start_time = GNUNET_TIME_absolute_get ();
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Starting job %p (%u active)\n",
- qe,
- qe->h->active_downloads);
- GNUNET_CONTAINER_DLL_remove (qe->h->pending_head, qe->h->pending_tail, qe);
- GNUNET_CONTAINER_DLL_insert_after (qe->h->running_head, qe->h->running_tail,
- qe->h->running_tail, qe);
- }
- /**
- * Stop the given job (send signal, remove from active queue, update
- * counters and state).
- *
- * @param qe job to stop
- */
- static void
- stop_job (struct GNUNET_FS_QueueEntry *qe)
- {
- qe->client = NULL;
- qe->stop (qe->cls);
- GNUNET_assert (0 < qe->h->active_downloads);
- qe->h->active_downloads--;
- qe->h->active_blocks -= qe->blocks;
- qe->run_time =
- GNUNET_TIME_relative_add (qe->run_time,
- GNUNET_TIME_absolute_get_duration
- (qe->start_time));
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Stopping job %p (%u active)\n",
- qe,
- qe->h->active_downloads);
- GNUNET_CONTAINER_DLL_remove (qe->h->running_head, qe->h->running_tail, qe);
- GNUNET_CONTAINER_DLL_insert_after (qe->h->pending_head, qe->h->pending_tail,
- qe->h->pending_tail, qe);
- }
- /**
- * Process the jobs in the job queue, possibly starting some
- * and stopping others.
- *
- * @param cls the `struct GNUNET_FS_Handle *`
- * @param tc scheduler context
- */
- static void
- process_job_queue (void *cls,
- const struct GNUNET_SCHEDULER_TaskContext *tc)
- {
- struct GNUNET_FS_Handle *h = cls;
- struct GNUNET_FS_QueueEntry *qe;
- struct GNUNET_FS_QueueEntry *next;
- struct GNUNET_TIME_Relative run_time;
- struct GNUNET_TIME_Relative restart_at;
- struct GNUNET_TIME_Relative rst;
- struct GNUNET_TIME_Absolute end_time;
- unsigned int num_downloads_waiting;
- unsigned int num_downloads_active;
- unsigned int num_downloads_expired;
- unsigned int num_probes_active;
- unsigned int num_probes_waiting;
- unsigned int num_probes_expired;
- int num_probes_change;
- int num_downloads_change;
- int block_limit_hit;
- h->queue_job = GNUNET_SCHEDULER_NO_TASK;
- /* restart_at will be set to the time when it makes sense to
- re-evaluate the job queue (unless, of course, jobs complete
- or are added, then we'll be triggered immediately */
- restart_at = GNUNET_TIME_UNIT_FOREVER_REL;
- /* first, calculate some basic statistics on pending jobs */
- num_probes_waiting = 0;
- num_downloads_waiting = 0;
- for (qe = h->pending_head; NULL != qe; qe = qe->next)
- {
- switch (qe->priority)
- {
- case GNUNET_FS_QUEUE_PRIORITY_PROBE:
- num_probes_waiting++;
- break;
- case GNUNET_FS_QUEUE_PRIORITY_NORMAL:
- num_downloads_waiting++;
- break;
- default:
- GNUNET_break (0);
- break;
- }
- }
- /* now, calculate some basic statistics on running jobs */
- num_probes_active = 0;
- num_probes_expired = 0;
- num_downloads_active = 0;
- num_downloads_expired = 0;
- next = h->running_head;
- while (NULL != (qe = next))
- {
- next = qe->next;
- switch (qe->priority)
- {
- case GNUNET_FS_QUEUE_PRIORITY_PROBE:
- run_time = GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MINUTES, 2);
- end_time = GNUNET_TIME_absolute_add (qe->start_time, run_time);
- rst = GNUNET_TIME_absolute_get_remaining (end_time);
- if (0 == rst.rel_value_us)
- {
- num_probes_expired++;
- stop_job (qe);
- }
- else
- {
- num_probes_active++;
- restart_at = GNUNET_TIME_relative_min (rst, restart_at);
- }
- break;
- case GNUNET_FS_QUEUE_PRIORITY_NORMAL:
- run_time =
- GNUNET_TIME_relative_multiply (h->avg_block_latency,
- qe->blocks * qe->start_times);
- end_time = GNUNET_TIME_absolute_add (qe->start_time, run_time);
- rst = GNUNET_TIME_absolute_get_remaining (end_time);
- if (0 == rst.rel_value_us)
- {
- num_downloads_expired++;
- stop_job (qe);
- }
- else
- {
- num_downloads_active++;
- restart_at = GNUNET_TIME_relative_min (rst, restart_at);
- }
- break;
- default:
- GNUNET_break (0);
- break;
- }
- }
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "PA: %u, PE: %u, PW: %u; DA: %u, DE: %u, DW: %u\n",
- num_probes_active,
- num_probes_expired,
- num_probes_waiting,
- num_downloads_active,
- num_downloads_expired,
- num_downloads_waiting);
- /* calculate start/stop decisions */
- if (h->active_downloads + num_downloads_waiting > h->max_parallel_requests)
- {
- /* stop probes if possible */
- num_probes_change = - num_probes_active;
- num_downloads_change = h->max_parallel_requests - h->active_downloads;
- }
- else
- {
- /* start all downloads */
- num_downloads_change = num_downloads_waiting;
- /* start as many probes as we can */
- num_probes_change = GNUNET_MIN (num_probes_waiting,
- h->max_parallel_requests - (h->active_downloads + num_downloads_waiting));
- }
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Changing %d probes and %d downloads\n",
- num_probes_change,
- num_downloads_change);
- /* actually stop probes */
- next = h->running_head;
- while (NULL != (qe = next))
- {
- next = qe->next;
- if (GNUNET_FS_QUEUE_PRIORITY_PROBE != qe->priority)
- continue;
- if (num_probes_change < 0)
- {
- stop_job (qe);
- num_probes_change++;
- if (0 == num_probes_change)
- break;
- }
- }
- GNUNET_break (0 <= num_probes_change);
- /* start some more tasks if we now have empty slots */
- block_limit_hit = GNUNET_NO;
- next = h->pending_head;
- while ( (NULL != (qe = next)) &&
- ( (num_probes_change > 0) ||
- (num_downloads_change > 0) ) )
- {
- next = qe->next;
- switch (qe->priority)
- {
- case GNUNET_FS_QUEUE_PRIORITY_PROBE:
- if (num_probes_change > 0)
- {
- start_job (qe);
- num_probes_change--;
- run_time = GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MINUTES, 2);
- restart_at = GNUNET_TIME_relative_min (run_time, restart_at);
- }
- break;
- case GNUNET_FS_QUEUE_PRIORITY_NORMAL:
- if ( (num_downloads_change > 0) &&
- ( (qe->blocks + h->active_blocks <= h->max_parallel_requests) ||
- ( (qe->blocks > h->max_parallel_requests) &&
- (0 == h->active_downloads) ) ) )
- {
- start_job (qe);
- num_downloads_change--;
- }
- else if (num_downloads_change > 0)
- block_limit_hit = GNUNET_YES;
- break;
- default:
- GNUNET_break (0);
- break;
- }
- }
- GNUNET_break ( (0 == num_downloads_change) || (GNUNET_YES == block_limit_hit) );
- GNUNET_break (0 == num_probes_change);
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "AD: %u, MP: %u; %d probes and %d downloads to start, will run again in %s\n",
- h->active_downloads,
- h->max_parallel_requests,
- num_probes_change,
- num_downloads_change,
- GNUNET_STRINGS_relative_time_to_string (restart_at, GNUNET_YES));
- /* make sure we run again */
- h->queue_job =
- GNUNET_SCHEDULER_add_delayed (restart_at, &process_job_queue, h);
- }
- /**
- * Add a job to the queue.
- *
- * @param h handle to the overall FS state
- * @param start function to call to begin the job
- * @param stop function to call to pause the job, or on dequeue (if the job was running)
- * @param cls closure for start and stop
- * @param blocks number of blocks this jobs uses
- * @param priority how important is this download
- * @return queue handle
- */
- struct GNUNET_FS_QueueEntry *
- GNUNET_FS_queue_ (struct GNUNET_FS_Handle *h,
- GNUNET_FS_QueueStart start,
- GNUNET_FS_QueueStop stop, void *cls,
- unsigned int blocks,
- enum GNUNET_FS_QueuePriority priority)
- {
- struct GNUNET_FS_QueueEntry *qe;
- qe = GNUNET_new (struct GNUNET_FS_QueueEntry);
- qe->h = h;
- qe->start = start;
- qe->stop = stop;
- qe->cls = cls;
- qe->queue_time = GNUNET_TIME_absolute_get ();
- qe->blocks = blocks;
- qe->priority = priority;
- GNUNET_CONTAINER_DLL_insert_after (h->pending_head, h->pending_tail,
- h->pending_tail, qe);
- if (h->queue_job != GNUNET_SCHEDULER_NO_TASK)
- GNUNET_SCHEDULER_cancel (h->queue_job);
- h->queue_job = GNUNET_SCHEDULER_add_now (&process_job_queue, h);
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Queueing job %p\n",
- qe);
- return qe;
- }
- /**
- * Dequeue a job from the queue.
- *
- * @param qe handle for the job
- */
- void
- GNUNET_FS_dequeue_ (struct GNUNET_FS_QueueEntry *qe)
- {
- struct GNUNET_FS_Handle *h;
- h = qe->h;
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Dequeueing job %p\n",
- qe);
- if (NULL != qe->client)
- stop_job (qe);
- GNUNET_CONTAINER_DLL_remove (h->pending_head, h->pending_tail, qe);
- GNUNET_free (qe);
- if (h->queue_job != GNUNET_SCHEDULER_NO_TASK)
- GNUNET_SCHEDULER_cancel (h->queue_job);
- h->queue_job = GNUNET_SCHEDULER_add_now (&process_job_queue, h);
- }
- /**
- * Create a top-level activity entry.
- *
- * @param h global fs handle
- * @param ssf suspend signal function to use
- * @param ssf_cls closure for @a ssf
- * @return fresh top-level activity handle
- */
- struct TopLevelActivity *
- GNUNET_FS_make_top (struct GNUNET_FS_Handle *h,
- SuspendSignalFunction ssf,
- void *ssf_cls)
- {
- struct TopLevelActivity *ret;
- ret = GNUNET_new (struct TopLevelActivity);
- ret->ssf = ssf;
- ret->ssf_cls = ssf_cls;
- GNUNET_CONTAINER_DLL_insert (h->top_head, h->top_tail, ret);
- return ret;
- }
- /**
- * Destroy a top-level activity entry.
- *
- * @param h global fs handle
- * @param top top level activity entry
- */
- void
- GNUNET_FS_end_top (struct GNUNET_FS_Handle *h,
- struct TopLevelActivity *top)
- {
- GNUNET_CONTAINER_DLL_remove (h->top_head, h->top_tail, top);
- GNUNET_free (top);
- }
- /**
- * Closure for #GNUNET_FS_data_reader_file_().
- */
- struct FileInfo
- {
- /**
- * Name of the file to read.
- */
- char *filename;
- /**
- * File descriptor, NULL if it has not yet been opened.
- */
- struct GNUNET_DISK_FileHandle *fd;
- };
- /**
- * Function that provides data by reading from a file.
- *
- * @param cls closure with the `struct FileInfo *`
- * @param offset offset to read from; it is possible
- * that the caller might need to go backwards
- * a bit at times; set to `UINT64_MAX` to tell
- * the reader that we won't be reading for a while
- * (used to close the file descriptor but NOT fully
- * clean up the reader's state); in this case,
- * a value of '0' for @a max should be ignored
- * @param max maximum number of bytes that should be
- * copied to @a buf; readers are not allowed
- * to provide less data unless there is an error;
- * a value of "0" will be used at the end to allow
- * the reader to clean up its internal state
- * @param buf where the reader should write the data
- * @param emsg location for the reader to store an error message
- * @return number of bytes written, usually @a max, 0 on error
- */
- size_t
- GNUNET_FS_data_reader_file_ (void *cls,
- uint64_t offset,
- size_t max,
- void *buf,
- char **emsg)
- {
- struct FileInfo *fi = cls;
- ssize_t ret;
- if (UINT64_MAX == offset)
- {
- if (NULL != fi->fd)
- {
- GNUNET_DISK_file_close (fi->fd);
- fi->fd = NULL;
- }
- return 0;
- }
- if (0 == max)
- {
- if (NULL != fi->fd)
- GNUNET_DISK_file_close (fi->fd);
- GNUNET_free (fi->filename);
- GNUNET_free (fi);
- return 0;
- }
- if (NULL == fi->fd)
- {
- fi->fd =
- GNUNET_DISK_file_open (fi->filename,
- GNUNET_DISK_OPEN_READ,
- GNUNET_DISK_PERM_NONE);
- if (NULL == fi->fd)
- {
- GNUNET_asprintf (emsg,
- _("Could not open file `%s': %s"),
- fi->filename,
- STRERROR (errno));
- return 0;
- }
- }
- if ( (GNUNET_SYSERR ==
- GNUNET_DISK_file_seek (fi->fd, offset, GNUNET_DISK_SEEK_SET)) ||
- (-1 == (ret = GNUNET_DISK_file_read (fi->fd, buf, max))) )
- {
- GNUNET_asprintf (emsg,
- _("Could not read file `%s': %s"),
- fi->filename,
- STRERROR (errno));
- return 0;
- }
- if (ret != max)
- {
- GNUNET_asprintf (emsg,
- _("Short read reading from file `%s'!"),
- fi->filename);
- return 0;
- }
- return max;
- }
- /**
- * Create the closure for the #GNUNET_FS_data_reader_file_() callback.
- *
- * @param filename file to read
- * @return closure to use, NULL on error
- */
- void *
- GNUNET_FS_make_file_reader_context_ (const char *filename)
- {
- struct FileInfo *fi;
- fi = GNUNET_new (struct FileInfo);
- fi->filename = GNUNET_STRINGS_filename_expand (filename);
- if (NULL == fi->filename)
- {
- GNUNET_free (fi);
- return NULL;
- }
- return fi;
- }
- /**
- * Function that provides data by copying from a buffer.
- *
- * @param cls closure (points to the buffer)
- * @param offset offset to read from; it is possible
- * that the caller might need to go backwards
- * a bit at times; set to `UINT64_MAX` to tell
- * the reader that we won't be reading for a while
- * (used to close the file descriptor but NOT fully
- * clean up the reader's state); in this case,
- * a value of '0' for @a max should be ignored
- * @param max maximum number of bytes that should be
- * copied to @a buf; readers are not allowed
- * to provide less data unless there is an error;
- * a value of "0" will be used at the end to allow
- * the reader to clean up its internal state
- * @param buf where the reader should write the data
- * @param emsg location for the reader to store an error message
- * @return number of bytes written, usually @a max, 0 on error
- */
- size_t
- GNUNET_FS_data_reader_copy_ (void *cls,
- uint64_t offset,
- size_t max,
- void *buf,
- char **emsg)
- {
- char *data = cls;
- if (UINT64_MAX == offset)
- return 0;
- if (0 == max)
- {
- GNUNET_free_non_null (data);
- return 0;
- }
- memcpy (buf, &data[offset], max);
- return max;
- }
- /**
- * Return the full filename where we would store state information
- * (for serialization/deserialization).
- *
- * @param h master context
- * @param ext component of the path
- * @param ent entity identifier (or emtpy string for the directory)
- * @return NULL on error
- */
- static char *
- get_serialization_file_name (struct GNUNET_FS_Handle *h,
- const char *ext,
- const char *ent)
- {
- char *basename;
- char *ret;
- if (0 == (h->flags & GNUNET_FS_FLAGS_PERSISTENCE))
- return NULL; /* persistence not requested */
- if (GNUNET_OK !=
- GNUNET_CONFIGURATION_get_value_filename (h->cfg, "fs", "STATE_DIR",
- &basename))
- return NULL;
- GNUNET_asprintf (&ret, "%s%s%s%s%s%s%s", basename, DIR_SEPARATOR_STR,
- h->client_name, DIR_SEPARATOR_STR, ext, DIR_SEPARATOR_STR,
- ent);
- GNUNET_free (basename);
- return ret;
- }
- /**
- * Return the full filename where we would store state information
- * (for serialization/deserialization) that is associated with a
- * parent operation.
- *
- * @param h master context
- * @param ext component of the path
- * @param uni name of the parent operation
- * @param ent entity identifier (or emtpy string for the directory)
- * @return NULL on error
- */
- static char *
- get_serialization_file_name_in_dir (struct GNUNET_FS_Handle *h,
- const char *ext,
- const char *uni,
- const char *ent)
- {
- char *basename;
- char *ret;
- if (0 == (h->flags & GNUNET_FS_FLAGS_PERSISTENCE))
- return NULL; /* persistence not requested */
- if (GNUNET_OK !=
- GNUNET_CONFIGURATION_get_value_filename (h->cfg, "fs", "STATE_DIR",
- &basename))
- return NULL;
- GNUNET_asprintf (&ret, "%s%s%s%s%s%s%s.dir%s%s", basename, DIR_SEPARATOR_STR,
- h->client_name, DIR_SEPARATOR_STR, ext, DIR_SEPARATOR_STR,
- uni, DIR_SEPARATOR_STR, ent);
- GNUNET_free (basename);
- return ret;
- }
- /**
- * Return a read handle for deserialization.
- *
- * @param h master context
- * @param ext component of the path
- * @param ent entity identifier (or emtpy string for the directory)
- * @return NULL on error
- */
- static struct GNUNET_BIO_ReadHandle *
- get_read_handle (struct GNUNET_FS_Handle *h,
- const char *ext,
- const char *ent)
- {
- char *fn;
- struct GNUNET_BIO_ReadHandle *ret;
- fn = get_serialization_file_name (h, ext, ent);
- if (NULL == fn)
- return NULL;
- ret = GNUNET_BIO_read_open (fn);
- GNUNET_free (fn);
- return ret;
- }
- /**
- * Return a write handle for serialization.
- *
- * @param h master context
- * @param ext component of the path
- * @param ent entity identifier (or emtpy string for the directory)
- * @return NULL on error
- */
- static struct GNUNET_BIO_WriteHandle *
- get_write_handle (struct GNUNET_FS_Handle *h,
- const char *ext,
- const char *ent)
- {
- char *fn;
- struct GNUNET_BIO_WriteHandle *ret;
- fn = get_serialization_file_name (h, ext, ent);
- if (NULL == fn)
- return NULL;
- ret = GNUNET_BIO_write_open (fn);
- GNUNET_break (NULL != ret);
- GNUNET_free (fn);
- return ret;
- }
- /**
- * Return a write handle for serialization.
- *
- * @param h master context
- * @param ext component of the path
- * @param uni name of parent
- * @param ent entity identifier (or emtpy string for the directory)
- * @return NULL on error
- */
- static struct GNUNET_BIO_WriteHandle *
- get_write_handle_in_dir (struct GNUNET_FS_Handle *h, const char *ext,
- const char *uni, const char *ent)
- {
- char *fn;
- struct GNUNET_BIO_WriteHandle *ret;
- fn = get_serialization_file_name_in_dir (h, ext, uni, ent);
- if (NULL == fn)
- return NULL;
- ret = GNUNET_BIO_write_open (fn);
- GNUNET_free (fn);
- return ret;
- }
- /**
- * Remove serialization/deserialization file from disk.
- *
- * @param h master context
- * @param ext component of the path
- * @param ent entity identifier
- */
- void
- GNUNET_FS_remove_sync_file_ (struct GNUNET_FS_Handle *h,
- const char *ext,
- const char *ent)
- {
- char *filename;
- if ((NULL == ent) || (0 == strlen (ent)))
- {
- GNUNET_break (0);
- return;
- }
- filename = get_serialization_file_name (h, ext, ent);
- if (NULL != filename)
- {
- if ( (0 != UNLINK (filename)) &&
- (ENOENT != errno) )
- GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_WARNING, "unlink", filename);
- GNUNET_free (filename);
- }
- }
- /**
- * Remove serialization/deserialization file from disk.
- *
- * @param h master context
- * @param ext component of the path
- * @param uni parent name
- * @param ent entity identifier
- */
- static void
- remove_sync_file_in_dir (struct GNUNET_FS_Handle *h,
- const char *ext,
- const char *uni, const char *ent)
- {
- char *filename;
- if ((NULL == ent) || (0 == strlen (ent)))
- {
- GNUNET_break (0);
- return;
- }
- filename = get_serialization_file_name_in_dir (h, ext, uni, ent);
- if (NULL == filename)
- return;
- if (0 != UNLINK (filename))
- GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_WARNING, "unlink", filename);
- GNUNET_free (filename);
- }
- /**
- * Remove serialization/deserialization directory from disk.
- *
- * @param h master context
- * @param ext component of the path
- * @param uni unique name of parent
- */
- void
- GNUNET_FS_remove_sync_dir_ (struct GNUNET_FS_Handle *h,
- const char *ext,
- const char *uni)
- {
- char *dn;
- if (NULL == uni)
- return;
- dn = get_serialization_file_name_in_dir (h, ext, uni, "");
- if (NULL == dn)
- return;
- if ((GNUNET_YES == GNUNET_DISK_directory_test (dn, GNUNET_YES)) &&
- (GNUNET_OK != GNUNET_DISK_directory_remove (dn)))
- GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_WARNING, "rmdir", dn);
- GNUNET_free (dn);
- }
- /**
- * Serialize a start-time. Since we use start-times to
- * calculate the duration of some operation, we actually
- * do not serialize the absolute time but the (relative)
- * duration since the start time. When we then
- * deserialize the start time, we take the current time and
- * subtract that duration so that we get again an absolute
- * time stamp that will result in correct performance
- * calculations.
- *
- * @param wh handle for writing
- * @param timestamp time to serialize
- * @return #GNUNET_OK on success
- */
- static int
- write_start_time (struct GNUNET_BIO_WriteHandle *wh,
- struct GNUNET_TIME_Absolute timestamp)
- {
- struct GNUNET_TIME_Relative dur;
- dur = GNUNET_TIME_absolute_get_duration (timestamp);
- return GNUNET_BIO_write_int64 (wh, dur.rel_value_us);
- }
- /**
- * Deserialize a start-time. Since we use start-times to
- * calculate the duration of some operation, we actually
- * do not serialize the absolute time but the (relative)
- * duration since the start time. Thus, when we then
- * deserialize the start time, we take the current time and
- * subtract that duration so that we get again an absolute
- * time stamp that will result in correct performance
- * calculations.
- *
- * @param rh handle for reading
- * @param timestamp where to write the deserialized timestamp
- * @return #GNUNET_OK on success
- */
- static int
- read_start_time (struct GNUNET_BIO_ReadHandle *rh,
- struct GNUNET_TIME_Absolute *timestamp)
- {
- struct GNUNET_TIME_Relative dur;
- if (GNUNET_OK != GNUNET_BIO_read_int64 (rh, &dur.rel_value_us))
- return GNUNET_SYSERR;
- *timestamp = GNUNET_TIME_absolute_subtract (GNUNET_TIME_absolute_get (), dur);
- return GNUNET_OK;
- }
- /**
- * Using the given serialization filename, try to deserialize
- * the file-information tree associated with it.
- *
- * @param h master context
- * @param filename name of the file (without directory) with
- * the infromation
- * @return NULL on error
- */
- static struct GNUNET_FS_FileInformation *
- deserialize_file_information (struct GNUNET_FS_Handle *h,
- const char *filename);
- /**
- * Using the given serialization filename, try to deserialize
- * the file-information tree associated with it.
- *
- * @param h master context
- * @param fn name of the file (without directory) with
- * the infromation
- * @param rh handle for reading
- * @return NULL on error
- */
- static struct GNUNET_FS_FileInformation *
- deserialize_fi_node (struct GNUNET_FS_Handle *h,
- const char *fn,
- struct GNUNET_BIO_ReadHandle *rh)
- {
- struct GNUNET_FS_FileInformation *ret;
- struct GNUNET_FS_FileInformation *nxt;
- char b;
- char *ksks;
- char *chks;
- char *skss;
- char *filename;
- uint32_t dsize;
- if (GNUNET_OK != GNUNET_BIO_read (rh, "status flag", &b, sizeof (b)))
- {
- GNUNET_break (0);
- return NULL;
- }
- ret = GNUNET_new (struct GNUNET_FS_FileInformation);
- ret->h = h;
- ksks = NULL;
- chks = NULL;
- skss = NULL;
- filename = NULL;
- if ((GNUNET_OK != GNUNET_BIO_read_meta_data (rh, "metadata", &ret->meta)) ||
- (GNUNET_OK != GNUNET_BIO_read_string (rh, "ksk-uri", &ksks, 32 * 1024)) ||
- ( (NULL != ksks) &&
- ( (NULL == (ret->keywords = GNUNET_FS_uri_parse (ksks, NULL))) ||
- (GNUNET_YES != GNUNET_FS_uri_test_ksk (ret->keywords)) ) ) ||
- (GNUNET_OK != GNUNET_BIO_read_string (rh, "chk-uri", &chks, 1024)) ||
- ( (NULL != chks) &&
- ( (NULL == (ret->chk_uri = GNUNET_FS_uri_parse (chks, NULL))) ||
- (GNUNET_YES != GNUNET_FS_uri_test_chk (ret->chk_uri))) ) ||
- (GNUNET_OK != GNUNET_BIO_read_string (rh, "sks-uri", &skss, 1024)) ||
- ( (NULL != skss) &&
- ( (NULL == (ret->sks_uri = GNUNET_FS_uri_parse (skss, NULL))) ||
- (GNUNET_YES != GNUNET_FS_uri_test_sks (ret->sks_uri))) ) ||
- (GNUNET_OK != read_start_time (rh, &ret->start_time)) ||
- (GNUNET_OK != GNUNET_BIO_read_string (rh, "emsg", &ret->emsg, 16 * 1024))
- || (GNUNET_OK !=
- GNUNET_BIO_read_string (rh, "fn", &ret->filename, 16 * 1024)) ||
- (GNUNET_OK !=
- GNUNET_BIO_read_int64 (rh, &ret->bo.expiration_time.abs_value_us)) ||
- (GNUNET_OK != GNUNET_BIO_read_int32 (rh, &ret->bo.anonymity_level)) ||
- (GNUNET_OK != GNUNET_BIO_read_int32 (rh, &ret->bo.content_priority)) ||
- (GNUNET_OK != GNUNET_BIO_read_int32 (rh, &ret->bo.replication_level)))
- {
- GNUNET_break (0);
- goto cleanup;
- }
- switch (b)
- {
- case 0: /* file-insert */
- if (GNUNET_OK != GNUNET_BIO_read_int64 (rh, &ret->data.file.file_size))
- {
- GNUNET_break (0);
- goto cleanup;
- }
- ret->is_directory = GNUNET_NO;
- ret->data.file.do_index = GNUNET_NO;
- ret->data.file.have_hash = GNUNET_NO;
- ret->data.file.index_start_confirmed = GNUNET_NO;
- if (GNUNET_NO == ret->is_published)
- {
- if (NULL == ret->filename)
- {
- ret->data.file.reader = &GNUNET_FS_data_reader_copy_;
- ret->data.file.reader_cls =
- GNUNET_malloc_large (ret->data.file.file_size);
- if (ret->data.file.reader_cls == NULL)
- goto cleanup;
- if (GNUNET_OK !=
- GNUNET_BIO_read (rh, "file-data", ret->data.file.reader_cls,
- ret->data.file.file_size))
- {
- GNUNET_break (0);
- goto cleanup;
- }
- }
- else
- {
- ret->data.file.reader = &GNUNET_FS_data_reader_file_;
- ret->data.file.reader_cls =
- GNUNET_FS_make_file_reader_context_ (ret->filename);
- }
- }
- break;
- case 1: /* file-index, no hash */
- if (NULL == ret->filename)
- {
- GNUNET_break (0);
- goto cleanup;
- }
- if (GNUNET_OK != GNUNET_BIO_read_int64 (rh, &ret->data.file.file_size))
- {
- GNUNET_break (0);
- goto cleanup;
- }
- ret->is_directory = GNUNET_NO;
- ret->data.file.do_index = GNUNET_YES;
- ret->data.file.have_hash = GNUNET_NO;
- ret->data.file.index_start_confirmed = GNUNET_NO;
- ret->data.file.reader = &GNUNET_FS_data_reader_file_;
- ret->data.file.reader_cls =
- GNUNET_FS_make_file_reader_context_ (ret->filename);
- break;
- case 2: /* file-index-with-hash */
- if (NULL == ret->filename)
- {
- GNUNET_break (0);
- goto cleanup;
- }
- if ((GNUNET_OK != GNUNET_BIO_read_int64 (rh, &ret->data.file.file_size)) ||
- (GNUNET_OK !=
- GNUNET_BIO_read (rh, "fileid", &ret->data.file.file_id,
- sizeof (struct GNUNET_HashCode))))
- {
- GNUNET_break (0);
- goto cleanup;
- }
- ret->is_directory = GNUNET_NO;
- ret->data.file.do_index = GNUNET_YES;
- ret->data.file.have_hash = GNUNET_YES;
- ret->data.file.index_start_confirmed = GNUNET_NO;
- ret->data.file.reader = &GNUNET_FS_data_reader_file_;
- ret->data.file.reader_cls =
- GNUNET_FS_make_file_reader_context_ (ret->filename);
- break;
- case 3: /* file-index-with-hash-confirmed */
- if (NULL == ret->filename)
- {
- GNUNET_break (0);
- goto cleanup;
- }
- if ((GNUNET_OK != GNUNET_BIO_read_int64 (rh, &ret->data.file.file_size)) ||
- (GNUNET_OK !=
- GNUNET_BIO_read (rh, "fileid", &ret->data.file.file_id,
- sizeof (struct GNUNET_HashCode))))
- {
- GNUNET_break (0);
- goto cleanup;
- }
- ret->is_directory = GNUNET_NO;
- ret->data.file.do_index = GNUNET_YES;
- ret->data.file.have_hash = GNUNET_YES;
- ret->data.file.index_start_confirmed = GNUNET_YES;
- ret->data.file.reader = &GNUNET_FS_data_reader_file_;
- ret->data.file.reader_cls =
- GNUNET_FS_make_file_reader_context_ (ret->filename);
- break;
- case 4: /* directory */
- ret->is_directory = GNUNET_YES;
- if ((GNUNET_OK != GNUNET_BIO_read_int32 (rh, &dsize)) ||
- (GNUNET_OK != GNUNET_BIO_read_int64 (rh, &ret->data.dir.contents_completed)) ||
- (GNUNET_OK != GNUNET_BIO_read_int64 (rh, &ret->data.dir.contents_size)) ||
- (NULL == (ret->data.dir.dir_data = GNUNET_malloc_large (dsize))) ||
- (GNUNET_OK !=
- GNUNET_BIO_read (rh, "dir-data", ret->data.dir.dir_data, dsize)) ||
- (GNUNET_OK !=
- GNUNET_BIO_read_string (rh, "ent-filename", &filename, 16 * 1024)))
- {
- GNUNET_break (0);
- goto cleanup;
- }
- ret->data.dir.dir_size = (uint32_t) dsize;
- if (NULL != filename)
- {
- ret->data.dir.entries = deserialize_file_information (h, filename);
- GNUNET_free (filename);
- filename = NULL;
- nxt = ret->data.dir.entries;
- while (NULL != nxt)
- {
- nxt->dir = ret;
- nxt = nxt->next;
- }
- }
- break;
- default:
- GNUNET_break (0);
- goto cleanup;
- }
- ret->serialization = GNUNET_strdup (fn);
- if (GNUNET_OK !=
- GNUNET_BIO_read_string (rh, "nxt-filename", &filename, 16 * 1024))
- {
- GNUNET_break (0);
- goto cleanup;
- }
- if (NULL != filename)
- {
- ret->next = deserialize_file_information (h, filename);
- GNUNET_free (filename);
- filename = NULL;
- }
- GNUNET_free_non_null (ksks);
- GNUNET_free_non_null (skss);
- GNUNET_free_non_null (chks);
- return ret;
- cleanup:
- GNUNET_free_non_null (ksks);
- GNUNET_free_non_null (chks);
- GNUNET_free_non_null (skss);
- GNUNET_free_non_null (filename);
- GNUNET_FS_file_information_destroy (ret, NULL, NULL);
- return NULL;
- }
- /**
- * Using the given serialization filename, try to deserialize
- * the file-information tree associated with it.
- *
- * @param h master context
- * @param filename name of the file (without directory) with
- * the infromation
- * @return NULL on error
- */
- static struct GNUNET_FS_FileInformation *
- deserialize_file_information (struct GNUNET_FS_Handle *h,
- const char *filename)
- {
- struct GNUNET_FS_FileInformation *ret;
- struct GNUNET_BIO_ReadHandle *rh;
- char *emsg;
- char *fn;
- rh = get_read_handle (h, GNUNET_FS_SYNC_PATH_FILE_INFO, filename);
- if (NULL == rh)
- return NULL;
- ret = deserialize_fi_node (h, filename, rh);
- if (GNUNET_OK != GNUNET_BIO_read_close (rh, &emsg))
- {
- GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
- _("Failed to resume publishing information `%s': %s\n"),
- filename, emsg);
- GNUNET_free (emsg);
- }
- if (NULL == ret)
- {
- fn = get_serialization_file_name (h, GNUNET_FS_SYNC_PATH_FILE_INFO, filename);
- if (NULL != fn)
- {
- if (0 != UNLINK (fn))
- GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_WARNING, "unlink", fn);
- GNUNET_free (fn);
- }
- }
- return ret;
- }
- /**
- * Given a serialization name (full absolute path), return the
- * basename of the file (without the path), which must only
- * consist of the 6 random characters.
- *
- * @param fullname name to extract the basename from
- * @return copy of the basename, NULL on error
- */
- static char *
- get_serialization_short_name (const char *fullname)
- {
- const char *end;
- const char *nxt;
- end = NULL;
- nxt = fullname;
- /* FIXME: we could do this faster since we know
- * the length of 'end'... */
- while ('\0' != *nxt)
- {
- if (DIR_SEPARATOR == *nxt)
- end = nxt + 1;
- nxt++;
- }
- if ((NULL == end) || (0 == strlen (end)))
- {
- GNUNET_break (0);
- return NULL;
- }
- GNUNET_break (6 == strlen (end));
- return GNUNET_strdup (end);
- }
- /**
- * Create a new random name for serialization. Also checks if persistence
- * is enabled and returns NULL if not.
- *
- * @param h master context
- * @param ext component of the path
- * @return NULL on errror
- */
- static char *
- make_serialization_file_name (struct GNUNET_FS_Handle *h,
- const char *ext)
- {
- char *fn;
- char *dn;
- char *ret;
- if (0 == (h->flags & GNUNET_FS_FLAGS_PERSISTENCE))
- return NULL; /* persistence not requested */
- dn = get_serialization_file_name (h, ext, "");
- if (NULL == dn)
- return NULL;
- if (GNUNET_OK != GNUNET_DISK_directory_create_for_file (dn))
- {
- GNUNET_free (dn);
- return NULL;
- }
- fn = GNUNET_DISK_mktemp (dn);
- GNUNET_free (dn);
- if (NULL == fn)
- return NULL; /* epic fail */
- ret = get_serialization_short_name (fn);
- GNUNET_free (fn);
- return ret;
- }
- /**
- * Create a new random name for serialization. Also checks if persistence
- * is enabled and returns NULL if not.
- *
- * @param h master context
- * @param ext component of the path
- * @param uni name of parent
- * @return NULL on errror
- */
- static char *
- make_serialization_file_name_in_dir (struct GNUNET_FS_Handle *h,
- const char *ext,
- const char *uni)
- {
- char *fn;
- char *dn;
- char *ret;
- if (0 == (h->flags & GNUNET_FS_FLAGS_PERSISTENCE))
- return NULL; /* persistence not requested */
- dn = get_serialization_file_name_in_dir (h, ext, uni, "");
- if (NULL == dn)
- return NULL;
- if (GNUNET_OK != GNUNET_DISK_directory_create_for_file (dn))
- {
- GNUNET_free (dn);
- return NULL;
- }
- fn = GNUNET_DISK_mktemp (dn);
- GNUNET_free (dn);
- if (NULL == fn)
- return NULL; /* epic fail */
- ret = get_serialization_short_name (fn);
- GNUNET_free (fn);
- return ret;
- }
- /**
- * Copy all of the data from the reader to the write handle.
- *
- * @param wh write handle
- * @param fi file with reader
- * @return #GNUNET_OK on success
- */
- static int
- copy_from_reader (struct GNUNET_BIO_WriteHandle *wh,
- struct GNUNET_FS_FileInformation *fi)
- {
- char buf[32 * 1024];
- uint64_t off;
- size_t ret;
- size_t left;
- char *emsg;
- emsg = NULL;
- off = 0;
- while (off < fi->data.file.file_size)
- {
- left = GNUNET_MIN (sizeof (buf), fi->data.file.file_size - off);
- ret =
- fi->data.file.reader (fi->data.file.reader_cls, off, left, buf, &emsg);
- if (0 == ret)
- {
- GNUNET_free (emsg);
- return GNUNET_SYSERR;
- }
- if (GNUNET_OK != GNUNET_BIO_write (wh, buf, ret))
- return GNUNET_SYSERR;
- off += ret;
- }
- return GNUNET_OK;
- }
- /**
- * Create a temporary file on disk to store the current
- * state of @a fi in.
- *
- * @param fi file information to sync with disk
- */
- void
- GNUNET_FS_file_information_sync_ (struct GNUNET_FS_FileInformation *fi)
- {
- char *fn;
- struct GNUNET_BIO_WriteHandle *wh;
- char b;
- char *ksks;
- char *chks;
- char *skss;
- if (NULL == fi->serialization)
- fi->serialization =
- make_serialization_file_name (fi->h, GNUNET_FS_SYNC_PATH_FILE_INFO);
- if (NULL == fi->serialization)
- return;
- wh = get_write_handle (fi->h, GNUNET_FS_SYNC_PATH_FILE_INFO,
- fi->serialization);
- if (NULL == wh)
- {
- GNUNET_free (fi->serialization);
- fi->serialization = NULL;
- return;
- }
- if (GNUNET_YES == fi->is_directory)
- b = 4;
- else if (GNUNET_YES == fi->data.file.index_start_confirmed)
- b = 3;
- else if (GNUNET_YES == fi->data.file.have_hash)
- b = 2;
- else if (GNUNET_YES == fi->data.file.do_index)
- b = 1;
- else
- b = 0;
- if (NULL != fi->keywords)
- ksks = GNUNET_FS_uri_to_string (fi->keywords);
- else
- ksks = NULL;
- if (NULL != fi->chk_uri)
- chks = GNUNET_FS_uri_to_string (fi->chk_uri);
- else
- chks = NULL;
- if (NULL != fi->sks_uri)
- skss = GNUNET_FS_uri_to_string (fi->sks_uri);
- else
- skss = NULL;
- if ((GNUNET_OK != GNUNET_BIO_write (wh, &b, sizeof (b))) ||
- (GNUNET_OK != GNUNET_BIO_write_meta_data (wh, fi->meta)) ||
- (GNUNET_OK != GNUNET_BIO_write_string (wh, ksks)) ||
- (GNUNET_OK != GNUNET_BIO_write_string (wh, chks)) ||
- (GNUNET_OK != GNUNET_BIO_write_string (wh, skss)) ||
- (GNUNET_OK != write_start_time (wh, fi->start_time)) ||
- (GNUNET_OK != GNUNET_BIO_write_string (wh, fi->emsg)) ||
- (GNUNET_OK != GNUNET_BIO_write_string (wh, fi->filename)) ||
- (GNUNET_OK !=
- GNUNET_BIO_write_int64 (wh, fi->bo.expiration_time.abs_value_us)) ||
- (GNUNET_OK != GNUNET_BIO_write_int32 (wh, fi->bo.anonymity_level)) ||
- (GNUNET_OK != GNUNET_BIO_write_int32 (wh, fi->bo.content_priority)) ||
- (GNUNET_OK != GNUNET_BIO_write_int32 (wh, fi->bo.replication_level)))
- {
- GNUNET_break (0);
- goto cleanup;
- }
- GNUNET_free_non_null (chks);
- chks = NULL;
- GNUNET_free_non_null (ksks);
- ksks = NULL;
- GNUNET_free_non_null (skss);
- skss = NULL;
- switch (b)
- {
- case 0: /* file-insert */
- if (GNUNET_OK != GNUNET_BIO_write_int64 (wh, fi->data.file.file_size))
- {
- GNUNET_break (0);
- goto cleanup;
- }
- if ((GNUNET_NO == fi->is_published) && (NULL == fi->filename))
- if (GNUNET_OK != copy_from_reader (wh, fi))
- {
- GNUNET_break (0);
- goto cleanup;
- }
- break;
- case 1: /* file-index, no hash */
- if (NULL == fi->filename)
- {
- GNUNET_break (0);
- goto cleanup;
- }
- if (GNUNET_OK != GNUNET_BIO_write_int64 (wh, fi->data.file.file_size))
- {
- GNUNET_break (0);
- goto cleanup;
- }
- break;
- case 2: /* file-index-with-hash */
- case 3: /* file-index-with-hash-confirmed */
- if (NULL == fi->filename)
- {
- GNUNET_break (0);
- goto cleanup;
- }
- if ((GNUNET_OK != GNUNET_BIO_write_int64 (wh, fi->data.file.file_size)) ||
- (GNUNET_OK !=
- GNUNET_BIO_write (wh, &fi->data.file.file_id,
- sizeof (struct GNUNET_HashCode))))
- {
- GNUNET_break (0);
- goto cleanup;
- }
- break;
- case 4: /* directory */
- if ( (NULL != fi->data.dir.entries) &&
- (NULL == fi->data.dir.entries->serialization) )
- GNUNET_FS_file_information_sync_ (fi->data.dir.entries);
- if ((GNUNET_OK != GNUNET_BIO_write_int32 (wh, fi->data.dir.dir_size)) ||
- (GNUNET_OK != GNUNET_BIO_write_int64 (wh, fi->data.dir.contents_completed)) ||
- (GNUNET_OK != GNUNET_BIO_write_int64 (wh, fi->data.dir.contents_size)) ||
- (GNUNET_OK !=
- GNUNET_BIO_write (wh, fi->data.dir.dir_data,
- (uint32_t) fi->data.dir.dir_size)) ||
- (GNUNET_OK !=
- GNUNET_BIO_write_string (wh,
- (fi->data.dir.entries ==
- NULL) ? NULL : fi->data.dir.
- entries->serialization)))
- {
- GNUNET_break (0);
- goto cleanup;
- }
- break;
- default:
- GNUNET_assert (0);
- goto cleanup;
- }
- if ( (NULL != fi->next) &&
- (NULL == fi->next->serialization) )
- GNUNET_FS_file_information_sync_ (fi->next);
- if (GNUNET_OK !=
- GNUNET_BIO_write_string (wh,
- (fi->next !=
- NULL) ? fi->next->serialization : NULL))
- {
- GNUNET_break (0);
- goto cleanup;
- }
- if (GNUNET_OK != GNUNET_BIO_write_close (wh))
- {
- wh = NULL;
- GNUNET_break (0);
- goto cleanup;
- }
- return; /* done! */
- cleanup:
- if (NULL != wh)
- (void) GNUNET_BIO_write_close (wh);
- GNUNET_free_non_null (chks);
- GNUNET_free_non_null (ksks);
- GNUNET_free_non_null (skss);
- fn = get_serialization_file_name (fi->h, GNUNET_FS_SYNC_PATH_FILE_INFO,
- fi->serialization);
- if (NULL != fn)
- {
- if (0 != UNLINK (fn))
- GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_WARNING, "unlink", fn);
- GNUNET_free (fn);
- }
- GNUNET_free (fi->serialization);
- fi->serialization = NULL;
- }
- /**
- * Find the entry in the file information struct where the
- * serialization filename matches the given name.
- *
- * @param pos file information to search
- * @param srch filename to search for
- * @return NULL if srch was not found in this subtree
- */
- static struct GNUNET_FS_FileInformation *
- find_file_position (struct GNUNET_FS_FileInformation *pos,
- const char *srch)
- {
- struct GNUNET_FS_FileInformation *r;
- while (NULL != pos)
- {
- if (0 == strcmp (srch, pos->serialization))
- return pos;
- if ( (GNUNET_YES == pos->is_directory) &&
- (NULL != (r = find_file_position (pos->data.dir.entries, srch))) )
- return r;
- pos = pos->next;
- }
- return NULL;
- }
- /**
- * Signal the FS's progress function that we are resuming
- * an upload.
- *
- * @param cls closure (of type `struct GNUNET_FS_PublishContext *`, for the parent (!))
- * @param fi the entry in the publish-structure
- * @param length length of the file or directory
- * @param meta metadata for the file or directory (can be modified)
- * @param uri pointer to the keywords that will be used for this entry (can be modified)
- * @param bo block options (can be modified)
- * @param do_index should we index?
- * @param client_info pointer to client context set upon creation (can be modified)
- * @return #GNUNET_OK to continue (always)
- */
- static int
- fip_signal_resume (void *cls,
- struct GNUNET_FS_FileInformation *fi,
- uint64_t length,
- struct GNUNET_CONTAINER_MetaData *meta,
- struct GNUNET_FS_Uri **uri,
- struct GNUNET_FS_BlockOptions *bo,
- int *do_index,
- void **client_info)
- {
- struct GNUNET_FS_PublishContext *pc = cls;
- struct GNUNET_FS_ProgressInfo pi;
- if (GNUNET_YES == pc->skip_next_fi_callback)
- {
- pc->skip_next_fi_callback = GNUNET_NO;
- return GNUNET_OK;
- }
- pi.status = GNUNET_FS_STATUS_PUBLISH_RESUME;
- pi.value.publish.specifics.resume.message = fi->emsg;
- pi.value.publish.specifics.resume.chk_uri = fi->chk_uri;
- *client_info = GNUNET_FS_publish_make_status_ (&pi, pc, fi, 0);
- if (GNUNET_YES == GNUNET_FS_meta_data_test_for_directory (meta))
- {
- /* process entries in directory */
- pc->skip_next_fi_callback = GNUNET_YES;
- GNUNET_FS_file_information_inspect (fi, &fip_signal_resume, pc);
- }
- return GNUNET_OK;
- }
- /**
- * Function called with a filename of serialized publishing operation
- * to deserialize.
- *
- * @param cls the `struct GNUNET_FS_Handle *`
- * @param filename complete filename (absolute path)
- * @return #GNUNET_OK (continue to iterate)
- */
- static int
- deserialize_publish_file (void *cls,
- const char *filename)
- {
- struct GNUNET_FS_Handle *h = cls;
- struct GNUNET_BIO_ReadHandle *rh;
- struct GNUNET_FS_PublishContext *pc;
- int32_t options;
- int32_t all_done;
- int32_t have_ns;
- char *fi_root;
- struct GNUNET_CRYPTO_EcdsaPrivateKey ns;
- char *fi_pos;
- char *emsg;
- pc = GNUNET_new (struct GNUNET_FS_PublishContext);
- pc->h = h;
- pc->serialization = get_serialization_short_name (filename);
- fi_root = NULL;
- fi_pos = NULL;
- rh = GNUNET_BIO_read_open (filename);
- if (NULL == rh)
- {
- GNUNET_break (0);
- goto cleanup;
- }
- if ((GNUNET_OK != GNUNET_BIO_read_string (rh, "publish-nid", &pc->nid, 1024))
- || (GNUNET_OK !=
- GNUNET_BIO_read_string (rh, "publish-nuid", &pc->nuid, 1024)) ||
- (GNUNET_OK != GNUNET_BIO_read_int32 (rh, &options)) ||
- (GNUNET_OK != GNUNET_BIO_read_int32 (rh, &all_done)) ||
- (GNUNET_OK != GNUNET_BIO_read_int32 (rh, &have_ns)) ||
- (GNUNET_OK !=
- GNUNET_BIO_read_string (rh, "publish-firoot", &fi_root, 128)) ||
- (GNUNET_OK != GNUNET_BIO_read_string (rh, "publish-fipos", &fi_pos, 128))
- || ( (GNUNET_YES == have_ns) &&
- (GNUNET_OK != GNUNET_BIO_read (rh, "publish-ns", &ns, sizeof (ns)))) )
- {
- GNUNET_break (0);
- goto cleanup;
- }
- pc->options = options;
- pc->all_done = all_done;
- if (NULL == fi_root)
- {
- GNUNET_break (0);
- goto cleanup;
- }
- pc->fi = deserialize_file_information (h, fi_root);
- if (NULL == pc->fi)
- {
- GNUNET_break (0);
- goto cleanup;
- }
- if (GNUNET_YES == have_ns)
- {
- pc->ns = GNUNET_new (struct GNUNET_CRYPTO_EcdsaPrivateKey);
- *pc->ns = ns;
- }
- if ((0 == (pc->options & GNUNET_FS_PUBLISH_OPTION_SIMULATE_ONLY)) &&
- (GNUNET_YES != pc->all_done))
- {
- pc->dsh = GNUNET_DATASTORE_connect (h->cfg);
- if (NULL == pc->dsh)
- goto cleanup;
- }
- if (NULL != fi_pos)
- {
- pc->fi_pos = find_file_position (pc->fi, fi_pos);
- GNUNET_free (fi_pos);
- fi_pos = NULL;
- if (NULL == pc->fi_pos)
- {
- /* failed to find position for resuming, outch! Will start from root! */
- GNUNET_break (0);
- if (GNUNET_YES != pc->all_done)
- pc->fi_pos = pc->fi;
- }
- }
- GNUNET_free (fi_root);
- fi_root = NULL;
- /* generate RESUME event(s) */
- GNUNET_FS_file_information_inspect (pc->fi, &fip_signal_resume, pc);
- /* re-start publishing (if needed)... */
- if (GNUNET_YES != pc->all_done)
- {
- GNUNET_assert (GNUNET_SCHEDULER_NO_TASK == pc->upload_task);
- pc->upload_task =
- GNUNET_SCHEDULER_add_with_priority
- (GNUNET_SCHEDULER_PRIORITY_BACKGROUND,
- &GNUNET_FS_publish_main_, pc);
- }
- if (GNUNET_OK != GNUNET_BIO_read_close (rh, &emsg))
- {
- GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
- _("Failure while resuming publishing operation `%s': %s\n"),
- filename, emsg);
- GNUNET_free (emsg);
- }
- pc->top = GNUNET_FS_make_top (h, &GNUNET_FS_publish_signal_suspend_, pc);
- return GNUNET_OK;
- cleanup:
- GNUNET_free_non_null (pc->nid);
- GNUNET_free_non_null (pc->nuid);
- GNUNET_free_non_null (fi_root);
- GNUNET_free_non_null (fi_pos);
- if ((NULL != rh) && (GNUNET_OK != GNUNET_BIO_read_close (rh, &emsg)))
- {
- GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
- _("Failed to resume publishing operation `%s': %s\n"), filename,
- emsg);
- GNUNET_free (emsg);
- }
- if (NULL != pc->fi)
- GNUNET_FS_file_information_destroy (pc->fi, NULL, NULL);
- if (0 != UNLINK (filename))
- GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_WARNING, "unlink", filename);
- GNUNET_free (pc->serialization);
- GNUNET_free (pc);
- return GNUNET_OK;
- }
- /**
- * Synchronize this publishing struct with its mirror
- * on disk. Note that all internal FS-operations that change
- * publishing structs should already call "sync" internally,
- * so this function is likely not useful for clients.
- *
- * @param pc the struct to sync
- */
- void
- GNUNET_FS_publish_sync_ (struct GNUNET_FS_PublishContext *pc)
- {
- struct GNUNET_BIO_WriteHandle *wh;
- int32_t have_ns;
- if (NULL == pc->serialization)
- pc->serialization =
- make_serialization_file_name (pc->h,
- GNUNET_FS_SYNC_PATH_MASTER_PUBLISH);
- if (NULL == pc->serialization)
- return;
- if (NULL == pc->fi)
- return;
- if (NULL == pc->fi->serialization)
- {
- GNUNET_break (0);
- return;
- }
- wh = get_write_handle (pc->h, GNUNET_FS_SYNC_PATH_MASTER_PUBLISH,
- pc->serialization);
- if (NULL == wh)
- {
- GNUNET_break (0);
- goto cleanup;
- }
- have_ns = (NULL != pc->ns) ? GNUNET_YES : GNUNET_NO;
- if ((GNUNET_OK != GNUNET_BIO_write_string (wh, pc->nid)) ||
- (GNUNET_OK != GNUNET_BIO_write_string (wh, pc->nuid)) ||
- (GNUNET_OK != GNUNET_BIO_write_int32 (wh, pc->options)) ||
- (GNUNET_OK != GNUNET_BIO_write_int32 (wh, pc->all_done)) ||
- (GNUNET_OK != GNUNET_BIO_write_int32 (wh, have_ns)) ||
- (GNUNET_OK != GNUNET_BIO_write_string (wh, pc->fi->serialization)) ||
- (GNUNET_OK !=
- GNUNET_BIO_write_string (wh,
- (NULL == pc->fi_pos) ? NULL : pc->fi_pos->serialization)) ||
- ( (NULL != pc->ns) &&
- (GNUNET_OK != GNUNET_BIO_write (wh,
- pc->ns,
- sizeof (struct GNUNET_CRYPTO_EcdsaPrivateKey)) ) ))
- {
- GNUNET_break (0);
- goto cleanup;
- }
- if (GNUNET_OK != GNUNET_BIO_write_close (wh))
- {
- wh = NULL;
- GNUNET_break (0);
- goto cleanup;
- }
- return;
- cleanup:
- if (NULL != wh)
- (void) GNUNET_BIO_write_close (wh);
- GNUNET_FS_remove_sync_file_ (pc->h, GNUNET_FS_SYNC_PATH_MASTER_PUBLISH,
- pc->serialization);
- GNUNET_free (pc->serialization);
- pc->serialization = NULL;
- }
- /**
- * Synchronize this unindex struct with its mirror
- * on disk. Note that all internal FS-operations that change
- * publishing structs should already call "sync" internally,
- * so this function is likely not useful for clients.
- *
- * @param uc the struct to sync
- */
- void
- GNUNET_FS_unindex_sync_ (struct GNUNET_FS_UnindexContext *uc)
- {
- struct GNUNET_BIO_WriteHandle *wh;
- char *uris;
- if (NULL == uc->serialization)
- uc->serialization =
- make_serialization_file_name (uc->h,
- GNUNET_FS_SYNC_PATH_MASTER_UNINDEX);
- if (NULL == uc->serialization)
- return;
- wh = get_write_handle (uc->h, GNUNET_FS_SYNC_PATH_MASTER_UNINDEX,
- uc->serialization);
- if (NULL == wh)
- {
- GNUNET_break (0);
- goto cleanup;
- }
- if (NULL != uc->ksk_uri)
- uris = GNUNET_FS_uri_to_string (uc->ksk_uri);
- else
- uris = NULL;
- if ((GNUNET_OK != GNUNET_BIO_write_string (wh, uc->filename)) ||
- (GNUNET_OK != GNUNET_BIO_write_int64 (wh, uc->file_size)) ||
- (GNUNET_OK != write_start_time (wh, uc->start_time)) ||
- (GNUNET_OK != GNUNET_BIO_write_int32 (wh, (uint32_t) uc->state)) ||
- (GNUNET_OK !=
- GNUNET_BIO_write (wh, &uc->chk, sizeof (struct ContentHashKey))) ||
- (GNUNET_OK != GNUNET_BIO_write_string (wh, uris)) ||
- (GNUNET_OK != GNUNET_BIO_write_int32 (wh, (uint32_t) uc->ksk_offset)) ||
- ((uc->state == UNINDEX_STATE_FS_NOTIFY) &&
- (GNUNET_OK !=
- GNUNET_BIO_write (wh, &uc->file_id, sizeof (struct GNUNET_HashCode)))) ||
- ((uc->state == UNINDEX_STATE_ERROR) &&
- (GNUNET_OK != GNUNET_BIO_write_string (wh, uc->emsg))))
- {
- GNUNET_break (0);
- goto cleanup;
- }
- if (GNUNET_OK != GNUNET_BIO_write_close (wh))
- {
- wh = NULL;
- GNUNET_break (0);
- goto cleanup;
- }
- return;
- cleanup:
- if (NULL != wh)
- (void) GNUNET_BIO_write_close (wh);
- GNUNET_FS_remove_sync_file_ (uc->h, GNUNET_FS_SYNC_PATH_MASTER_UNINDEX,
- uc->serialization);
- GNUNET_free (uc->serialization);
- uc->serialization = NULL;
- }
- /**
- * Serialize a download request.
- *
- * @param wh handle for writing the download request to disk
- * @param dr the the request to write to disk
- * @return #GNUNET_YES on success, #GNUNET_NO on error
- */
- static int
- write_download_request (struct GNUNET_BIO_WriteHandle *wh,
- struct DownloadRequest *dr)
- {
- unsigned int i;
- if ((GNUNET_OK != GNUNET_BIO_write_int32 (wh, dr->state)) ||
- (GNUNET_OK != GNUNET_BIO_write_int64 (wh, dr->offset)) ||
- (GNUNET_OK != GNUNET_BIO_write_int32 (wh, dr->num_children)) ||
- (GNUNET_OK != GNUNET_BIO_write_int32 (wh, dr->depth)))
- return GNUNET_NO;
- if ((BRS_CHK_SET == dr->state) &&
- (GNUNET_OK !=
- GNUNET_BIO_write (wh, &dr->chk, sizeof (struct ContentHashKey))))
- return GNUNET_NO;
- for (i = 0; i < dr->num_children; i++)
- if (GNUNET_NO == write_download_request (wh, dr->children[i]))
- return GNUNET_NO;
- return GNUNET_YES;
- }
- /**
- * Read a download request tree.
- *
- * @param rh cadet to read from
- * @return value the download request read from disk, NULL on error
- */
- static struct DownloadRequest *
- read_download_request (struct GNUNET_BIO_ReadHandle *rh)
- {
- struct DownloadRequest *dr;
- unsigned int i;
- dr = GNUNET_new (struct DownloadRequest);
- if ((GNUNET_OK != GNUNET_BIO_read_int32 (rh, &dr->state)) ||
- (GNUNET_OK != GNUNET_BIO_read_int64 (rh, &dr->offset)) ||
- (GNUNET_OK != GNUNET_BIO_read_int32 (rh, &dr->num_children)) ||
- (dr->num_children > CHK_PER_INODE) ||
- (GNUNET_OK != GNUNET_BIO_read_int32 (rh, &dr->depth)) ||
- ( (0 == dr->depth) &&
- (dr->num_children > 0) ) ||
- ((dr->depth > 0) && (0 == dr->num_children)))
- {
- GNUNET_break (0);
- dr->num_children = 0;
- goto cleanup;
- }
- if (dr->num_children > 0)
- dr->children =
- GNUNET_malloc (dr->num_children * sizeof (struct DownloadRequest *));
- switch (dr->state)
- {
- case BRS_INIT:
- case BRS_RECONSTRUCT_DOWN:
- case BRS_RECONSTRUCT_META_UP:
- case BRS_RECONSTRUCT_UP:
- break;
- case BRS_CHK_SET:
- if (GNUNET_OK !=
- GNUNET_BIO_read (rh, "chk", &dr->chk, sizeof (struct ContentHashKey)))
- goto cleanup;
- break;
- case BRS_DOWNLOAD_DOWN:
- case BRS_DOWNLOAD_UP:
- case BRS_ERROR:
- break;
- default:
- GNUNET_break (0);
- goto cleanup;
- }
- for (i = 0; i < dr->num_children; i++)
- {
- if (NULL == (dr->children[i] = read_download_request (rh)))
- goto cleanup;
- dr->children[i]->parent = dr;
- }
- return dr;
- cleanup:
- GNUNET_FS_free_download_request_ (dr);
- return NULL;
- }
- /**
- * Compute the name of the sync file (or directory) for the given download
- * context.
- *
- * @param dc download context to compute for
- * @param uni unique filename to use, use "" for the directory name
- * @param ext extension to use, use ".dir" for our own subdirectory
- * @return the expanded file name, NULL for none
- */
- static char *
- get_download_sync_filename (struct GNUNET_FS_DownloadContext *dc,
- const char *uni,
- const char *ext)
- {
- char *par;
- char *epar;
- if (dc->parent == NULL)
- return get_serialization_file_name (dc->h,
- (dc->search != NULL) ?
- GNUNET_FS_SYNC_PATH_CHILD_DOWNLOAD :
- GNUNET_FS_SYNC_PATH_MASTER_DOWNLOAD,
- uni);
- if (NULL == dc->parent->serialization)
- return NULL;
- par = get_download_sync_filename (dc->parent, dc->parent->serialization, "");
- if (NULL == par)
- return NULL;
- GNUNET_asprintf (&epar, "%s.dir%s%s%s", par, DIR_SEPARATOR_STR, uni, ext);
- GNUNET_free (par);
- return epar;
- }
- /**
- * Synchronize this download struct with its mirror
- * on disk. Note that all internal FS-operations that change
- * publishing structs should already call "sync" internally,
- * so this function is likely not useful for clients.
- *
- * @param dc the struct to sync
- */
- void
- GNUNET_FS_download_sync_ (struct GNUNET_FS_DownloadContext *dc)
- {
- struct GNUNET_BIO_WriteHandle *wh;
- char *uris;
- char *fn;
- char *dir;
- if (0 != (dc->options & GNUNET_FS_DOWNLOAD_IS_PROBE))
- return; /* we don't sync probes */
- if (NULL == dc->serialization)
- {
- dir = get_download_sync_filename (dc, "", "");
- if (NULL == dir)
- return;
- if (GNUNET_OK != GNUNET_DISK_directory_create_for_file (dir))
- {
- GNUNET_free (dir);
- return;
- }
- fn = GNUNET_DISK_mktemp (dir);
- GNUNET_free (dir);
- if (NULL == fn)
- return;
- dc->serialization = get_serialization_short_name (fn);
- }
- else
- {
- fn = get_download_sync_filename (dc, dc->serialization, "");
- if (NULL == fn)
- {
- GNUNET_free (dc->serialization);
- dc->serialization = NULL;
- GNUNET_free (fn);
- return;
- }
- }
- wh = GNUNET_BIO_write_open (fn);
- if (NULL == wh)
- {
- GNUNET_free (dc->serialization);
- dc->serialization = NULL;
- GNUNET_free (fn);
- return;
- }
- GNUNET_assert ((GNUNET_YES == GNUNET_FS_uri_test_chk (dc->uri)) ||
- (GNUNET_YES == GNUNET_FS_uri_test_loc (dc->uri)));
- uris = GNUNET_FS_uri_to_string (dc->uri);
- if ((GNUNET_OK != GNUNET_BIO_write_string (wh, uris)) ||
- (GNUNET_OK != GNUNET_BIO_write_meta_data (wh, dc->meta)) ||
- (GNUNET_OK != GNUNET_BIO_write_string (wh, dc->emsg)) ||
- (GNUNET_OK != GNUNET_BIO_write_string (wh, dc->filename)) ||
- (GNUNET_OK != GNUNET_BIO_write_string (wh, dc->temp_filename)) ||
- (GNUNET_OK != GNUNET_BIO_write_int64 (wh, dc->old_file_size)) ||
- (GNUNET_OK != GNUNET_BIO_write_int64 (wh, dc->offset)) ||
- (GNUNET_OK != GNUNET_BIO_write_int64 (wh, dc->length)) ||
- (GNUNET_OK != GNUNET_BIO_write_int64 (wh, dc->completed)) ||
- (GNUNET_OK != write_start_time (wh, dc->start_time)) ||
- (GNUNET_OK != GNUNET_BIO_write_int32 (wh, dc->anonymity)) ||
- (GNUNET_OK != GNUNET_BIO_write_int32 (wh, (uint32_t) dc->options)) ||
- (GNUNET_OK != GNUNET_BIO_write_int32 (wh, (uint32_t) dc->has_finished)))
- {
- GNUNET_break (0);
- goto cleanup;
- }
- if (NULL == dc->emsg)
- {
- GNUNET_assert (dc->top_request != NULL);
- if (GNUNET_YES != write_download_request (wh, dc->top_request))
- {
- GNUNET_break (0);
- goto cleanup;
- }
- }
- GNUNET_free_non_null (uris);
- uris = NULL;
- if (GNUNET_OK != GNUNET_BIO_write_close (wh))
- {
- wh = NULL;
- GNUNET_break (0);
- goto cleanup;
- }
- GNUNET_free (fn);
- return;
- cleanup:
- if (NULL != wh)
- (void) GNUNET_BIO_write_close (wh);
- GNUNET_free_non_null (uris);
- if (0 != UNLINK (fn))
- GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_WARNING, "unlink", fn);
- GNUNET_free (fn);
- GNUNET_free (dc->serialization);
- dc->serialization = NULL;
- }
- /**
- * Synchronize this search result with its mirror
- * on disk. Note that all internal FS-operations that change
- * publishing structs should already call "sync" internally,
- * so this function is likely not useful for clients.
- *
- * @param sr the struct to sync
- */
- void
- GNUNET_FS_search_result_sync_ (struct GNUNET_FS_SearchResult *sr)
- {
- struct GNUNET_BIO_WriteHandle *wh;
- char *uris;
- if (NULL == sr->sc)
- return;
- uris = NULL;
- if (NULL == sr->serialization)
- sr->serialization =
- make_serialization_file_name_in_dir (sr->h,
- (sr->sc->psearch_result ==
- NULL) ?
- GNUNET_FS_SYNC_PATH_MASTER_SEARCH :
- GNUNET_FS_SYNC_PATH_CHILD_SEARCH,
- sr->sc->serialization);
- if (NULL == sr->serialization)
- return;
- wh = get_write_handle_in_dir (sr->h,
- (sr->sc->psearch_result ==
- NULL) ? GNUNET_FS_SYNC_PATH_MASTER_SEARCH :
- GNUNET_FS_SYNC_PATH_CHILD_SEARCH,
- sr->sc->serialization, sr->serialization);
- if (NULL == wh)
- {
- GNUNET_break (0);
- goto cleanup;
- }
- uris = GNUNET_FS_uri_to_string (sr->uri);
- if ((GNUNET_OK != GNUNET_BIO_write_string (wh, uris)) ||
- (GNUNET_OK !=
- GNUNET_BIO_write_string (wh,
- sr->download !=
- NULL ? sr->download->serialization : NULL)) ||
- (GNUNET_OK !=
- GNUNET_BIO_write_string (wh,
- sr->update_search !=
- NULL ? sr->update_search->serialization : NULL))
- || (GNUNET_OK != GNUNET_BIO_write_meta_data (wh, sr->meta)) ||
- (GNUNET_OK != GNUNET_BIO_write (wh, &sr->key, sizeof (struct GNUNET_HashCode)))
- || (GNUNET_OK != GNUNET_BIO_write_int32 (wh, sr->mandatory_missing)) ||
- (GNUNET_OK != GNUNET_BIO_write_int32 (wh, sr->optional_support)) ||
- (GNUNET_OK != GNUNET_BIO_write_int32 (wh, sr->availability_success)) ||
- (GNUNET_OK != GNUNET_BIO_write_int32 (wh, sr->availability_trials)) )
- {
- GNUNET_break (0);
- goto cleanup;
- }
- if ( (NULL != sr->uri) &&
- (GNUNET_FS_URI_KSK == sr->sc->uri->type) &&
- (GNUNET_OK != GNUNET_BIO_write (wh, sr->keyword_bitmap,
- (sr->sc->uri->data.ksk.keywordCount + 7) / 8)) )
- {
- GNUNET_break (0);
- goto cleanup;
- }
- if (GNUNET_OK != GNUNET_BIO_write_close (wh))
- {
- wh = NULL;
- GNUNET_break (0);
- goto cleanup;
- }
- GNUNET_free_non_null (uris);
- return;
- cleanup:
- GNUNET_free_non_null (uris);
- if (NULL != wh)
- (void) GNUNET_BIO_write_close (wh);
- remove_sync_file_in_dir (sr->h,
- (NULL == sr->sc->psearch_result)
- ? GNUNET_FS_SYNC_PATH_MASTER_SEARCH
- : GNUNET_FS_SYNC_PATH_CHILD_SEARCH,
- sr->sc->serialization, sr->serialization);
- GNUNET_free (sr->serialization);
- sr->serialization = NULL;
- }
- /**
- * Synchronize this search struct with its mirror
- * on disk. Note that all internal FS-operations that change
- * publishing structs should already call "sync" internally,
- * so this function is likely not useful for clients.
- *
- * @param sc the struct to sync
- */
- void
- GNUNET_FS_search_sync_ (struct GNUNET_FS_SearchContext *sc)
- {
- struct GNUNET_BIO_WriteHandle *wh;
- char *uris;
- char in_pause;
- const char *category;
- category =
- (NULL == sc->psearch_result)
- ? GNUNET_FS_SYNC_PATH_MASTER_SEARCH
- : GNUNET_FS_SYNC_PATH_CHILD_SEARCH;
- if (NULL == sc->serialization)
- sc->serialization = make_serialization_file_name (sc->h, category);
- if (NULL == sc->serialization)
- return;
- uris = NULL;
- wh = get_write_handle (sc->h, category, sc->serialization);
- if (NULL == wh)
- {
- GNUNET_break (0);
- goto cleanup;
- }
- GNUNET_assert ((GNUNET_YES == GNUNET_FS_uri_test_ksk (sc->uri)) ||
- (GNUNET_YES == GNUNET_FS_uri_test_sks (sc->uri)));
- uris = GNUNET_FS_uri_to_string (sc->uri);
- in_pause = (sc->task != GNUNET_SCHEDULER_NO_TASK) ? 'r' : '\0';
- if ((GNUNET_OK != GNUNET_BIO_write_string (wh, uris)) ||
- (GNUNET_OK != write_start_time (wh, sc->start_time)) ||
- (GNUNET_OK != GNUNET_BIO_write_string (wh, sc->emsg)) ||
- (GNUNET_OK != GNUNET_BIO_write_int32 (wh, (uint32_t) sc->options)) ||
- (GNUNET_OK != GNUNET_BIO_write (wh, &in_pause, sizeof (in_pause))) ||
- (GNUNET_OK != GNUNET_BIO_write_int32 (wh, sc->anonymity)))
- {
- GNUNET_break (0);
- goto cleanup;
- }
- GNUNET_free (uris);
- uris = NULL;
- if (GNUNET_OK != GNUNET_BIO_write_close (wh))
- {
- wh = NULL;
- GNUNET_break (0);
- goto cleanup;
- }
- return;
- cleanup:
- if (NULL != wh)
- (void) GNUNET_BIO_write_close (wh);
- GNUNET_free_non_null (uris);
- GNUNET_FS_remove_sync_file_ (sc->h, category, sc->serialization);
- GNUNET_free (sc->serialization);
- sc->serialization = NULL;
- }
- /**
- * Function called with a filename of serialized unindexing operation
- * to deserialize.
- *
- * @param cls the `struct GNUNET_FS_Handle *`
- * @param filename complete filename (absolute path)
- * @return #GNUNET_OK (continue to iterate)
- */
- static int
- deserialize_unindex_file (void *cls,
- const char *filename)
- {
- struct GNUNET_FS_Handle *h = cls;
- struct GNUNET_BIO_ReadHandle *rh;
- struct GNUNET_FS_UnindexContext *uc;
- struct GNUNET_FS_ProgressInfo pi;
- char *emsg;
- char *uris;
- uint32_t state;
- uc = GNUNET_new (struct GNUNET_FS_UnindexContext);
- uc->h = h;
- uc->serialization = get_serialization_short_name (filename);
- rh = GNUNET_BIO_read_open (filename);
- if (NULL == rh)
- {
- GNUNET_break (0);
- goto cleanup;
- }
- uris = NULL;
- if ((GNUNET_OK !=
- GNUNET_BIO_read_string (rh, "unindex-fn", &uc->filename, 10 * 1024)) ||
- (GNUNET_OK != GNUNET_BIO_read_int64 (rh, &uc->file_size)) ||
- (GNUNET_OK != read_start_time (rh, &uc->start_time)) ||
- (GNUNET_OK != GNUNET_BIO_read_int32 (rh, &state)) ||
- (GNUNET_OK != GNUNET_BIO_read (rh, "uri", &uc->chk, sizeof (struct ContentHashKey))) ||
- (GNUNET_OK != GNUNET_BIO_read_string (rh, "unindex-kskuri", &uris, 10 * 1024)) ||
- (GNUNET_OK != GNUNET_BIO_read_int32 (rh, &uc->ksk_offset)) )
- {
- GNUNET_free_non_null (uris);
- GNUNET_break (0);
- goto cleanup;
- }
- if (NULL != uris)
- {
- uc->ksk_uri = GNUNET_FS_uri_parse (uris, &emsg);
- GNUNET_free (uris);
- if (NULL == uc->ksk_uri)
- {
- GNUNET_break (0);
- GNUNET_free_non_null (emsg);
- goto cleanup;
- }
- }
- if ( (uc->ksk_offset > 0) &&
- ( (NULL == uc->ksk_uri) ||
- (uc->ksk_offset > uc->ksk_uri->data.ksk.keywordCount) ) )
- {
- GNUNET_break (0);
- goto cleanup;
- }
- uc->state = (enum UnindexState) state;
- switch (state)
- {
- case UNINDEX_STATE_HASHING:
- break;
- case UNINDEX_STATE_FS_NOTIFY:
- if (GNUNET_OK !=
- GNUNET_BIO_read (rh, "unindex-hash", &uc->file_id,
- sizeof (struct GNUNET_HashCode)))
- {
- GNUNET_break (0);
- goto cleanup;
- }
- break;
- case UNINDEX_STATE_DS_REMOVE:
- case UNINDEX_STATE_EXTRACT_KEYWORDS:
- case UNINDEX_STATE_DS_REMOVE_KBLOCKS:
- break;
- case UNINDEX_STATE_COMPLETE:
- break;
- case UNINDEX_STATE_ERROR:
- if (GNUNET_OK !=
- GNUNET_BIO_read_string (rh, "unindex-emsg", &uc->emsg, 10 * 1024))
- {
- GNUNET_break (0);
- goto cleanup;
- }
- break;
- default:
- GNUNET_break (0);
- goto cleanup;
- }
- uc->top = GNUNET_FS_make_top (h, &GNUNET_FS_unindex_signal_suspend_, uc);
- pi.status = GNUNET_FS_STATUS_UNINDEX_RESUME;
- pi.value.unindex.specifics.resume.message = uc->emsg;
- GNUNET_FS_unindex_make_status_ (&pi, uc,
- (uc->state ==
- UNINDEX_STATE_COMPLETE) ? uc->file_size : 0);
- switch (uc->state)
- {
- case UNINDEX_STATE_HASHING:
- uc->fhc =
- GNUNET_CRYPTO_hash_file (GNUNET_SCHEDULER_PRIORITY_IDLE, uc->filename,
- HASHING_BLOCKSIZE,
- &GNUNET_FS_unindex_process_hash_, uc);
- break;
- case UNINDEX_STATE_FS_NOTIFY:
- uc->state = UNINDEX_STATE_HASHING;
- GNUNET_FS_unindex_process_hash_ (uc, &uc->file_id);
- break;
- case UNINDEX_STATE_DS_REMOVE:
- GNUNET_FS_unindex_do_remove_ (uc);
- break;
- case UNINDEX_STATE_EXTRACT_KEYWORDS:
- GNUNET_FS_unindex_do_extract_keywords_ (uc);
- break;
- case UNINDEX_STATE_DS_REMOVE_KBLOCKS:
- GNUNET_FS_unindex_do_remove_kblocks_ (uc);
- break;
- case UNINDEX_STATE_COMPLETE:
- case UNINDEX_STATE_ERROR:
- /* no need to resume any operation, we were done */
- break;
- default:
- break;
- }
- if (GNUNET_OK != GNUNET_BIO_read_close (rh, &emsg))
- {
- GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
- _("Failure while resuming unindexing operation `%s': %s\n"),
- filename, emsg);
- GNUNET_free (emsg);
- }
- return GNUNET_OK;
- cleanup:
- GNUNET_free_non_null (uc->filename);
- if ((NULL != rh) && (GNUNET_OK != GNUNET_BIO_read_close (rh, &emsg)))
- {
- GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
- _("Failed to resume unindexing operation `%s': %s\n"),
- filename,
- emsg);
- GNUNET_free (emsg);
- }
- if (NULL != uc->serialization)
- GNUNET_FS_remove_sync_file_ (h, GNUNET_FS_SYNC_PATH_MASTER_UNINDEX,
- uc->serialization);
- GNUNET_free_non_null (uc->serialization);
- GNUNET_free (uc);
- return GNUNET_OK;
- }
- /**
- * Deserialize a download.
- *
- * @param h overall context
- * @param rh file to deserialize from
- * @param parent parent download
- * @param search associated search
- * @param serialization name under which the search was serialized
- */
- static void
- deserialize_download (struct GNUNET_FS_Handle *h,
- struct GNUNET_BIO_ReadHandle *rh,
- struct GNUNET_FS_DownloadContext *parent,
- struct GNUNET_FS_SearchResult *search,
- const char *serialization);
- /**
- * Deserialize a search.
- *
- * @param h overall context
- * @param rh file to deserialize from
- * @param psearch_result parent search result
- * @param serialization name under which the search was serialized
- */
- static struct GNUNET_FS_SearchContext *
- deserialize_search (struct GNUNET_FS_Handle *h,
- struct GNUNET_BIO_ReadHandle *rh,
- struct GNUNET_FS_SearchResult *psearch_result,
- const char *serialization);
- /**
- * Function called with a filename of serialized search result
- * to deserialize.
- *
- * @param cls the `struct GNUNET_FS_SearchContext *`
- * @param filename complete filename (absolute path)
- * @return #GNUNET_OK (continue to iterate)
- */
- static int
- deserialize_search_result (void *cls,
- const char *filename)
- {
- struct GNUNET_FS_SearchContext *sc = cls;
- char *ser;
- char *uris;
- char *emsg;
- char *download;
- char *update_srch;
- struct GNUNET_BIO_ReadHandle *rh;
- struct GNUNET_BIO_ReadHandle *drh;
- struct GNUNET_FS_SearchResult *sr;
- ser = get_serialization_short_name (filename);
- rh = GNUNET_BIO_read_open (filename);
- if (NULL == rh)
- {
- if (NULL != ser)
- {
- remove_sync_file_in_dir (sc->h,
- (NULL == sc->psearch_result)
- ? GNUNET_FS_SYNC_PATH_MASTER_SEARCH
- : GNUNET_FS_SYNC_PATH_CHILD_SEARCH,
- sc->serialization, ser);
- GNUNET_free (ser);
- }
- return GNUNET_OK;
- }
- emsg = NULL;
- uris = NULL;
- download = NULL;
- update_srch = NULL;
- sr = GNUNET_new (struct GNUNET_FS_SearchResult);
- sr->h = sc->h;
- sr->sc = sc;
- sr->serialization = ser;
- if ((GNUNET_OK != GNUNET_BIO_read_string (rh, "result-uri", &uris, 10 * 1024))
- || (NULL == (sr->uri = GNUNET_FS_uri_parse (uris, &emsg))) ||
- (GNUNET_OK != GNUNET_BIO_read_string (rh, "download-lnk", &download, 16))
- || (GNUNET_OK !=
- GNUNET_BIO_read_string (rh, "search-lnk", &update_srch, 16)) ||
- (GNUNET_OK != GNUNET_BIO_read_meta_data (rh, "result-meta", &sr->meta)) ||
- (GNUNET_OK !=
- GNUNET_BIO_read (rh, "result-key", &sr->key, sizeof (struct GNUNET_HashCode)))
- || (GNUNET_OK != GNUNET_BIO_read_int32 (rh, &sr->mandatory_missing)) ||
- (GNUNET_OK != GNUNET_BIO_read_int32 (rh, &sr->optional_support)) ||
- (GNUNET_OK != GNUNET_BIO_read_int32 (rh, &sr->availability_success)) ||
- (GNUNET_OK != GNUNET_BIO_read_int32 (rh, &sr->availability_trials)))
- {
- GNUNET_break (0);
- goto cleanup;
- }
- if (GNUNET_FS_URI_KSK == sr->sc->uri->type)
- {
- sr->keyword_bitmap = GNUNET_malloc ((sr->sc->uri->data.ksk.keywordCount + 7) / 8); /* round up, count bits */
- if (GNUNET_OK != GNUNET_BIO_read (rh, "keyword-bitmap",
- sr->keyword_bitmap,
- (sr->sc->uri->data.ksk.keywordCount + 7) / 8))
- {
- GNUNET_break (0);
- goto cleanup;
- }
- }
- GNUNET_free (uris);
- if (NULL != download)
- {
- drh = get_read_handle (sc->h, GNUNET_FS_SYNC_PATH_CHILD_DOWNLOAD, download);
- if (NULL != drh)
- {
- deserialize_download (sc->h, drh, NULL, sr, download);
- if (GNUNET_OK != GNUNET_BIO_read_close (drh, &emsg))
- {
- GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
- _("Failed to resume sub-download `%s': %s\n"),
- download,
- emsg);
- GNUNET_free (emsg);
- }
- }
- GNUNET_free (download);
- }
- if (NULL != update_srch)
- {
- drh =
- get_read_handle (sc->h, GNUNET_FS_SYNC_PATH_CHILD_SEARCH, update_srch);
- if (NULL != drh)
- {
- deserialize_search (sc->h, drh, sr, update_srch);
- if (GNUNET_OK != GNUNET_BIO_read_close (drh, &emsg))
- {
- GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
- _("Failed to resume sub-search `%s': %s\n"),
- update_srch,
- emsg);
- GNUNET_free (emsg);
- }
- }
- GNUNET_free (update_srch);
- }
- GNUNET_break (GNUNET_YES ==
- GNUNET_CONTAINER_multihashmap_put (sc->master_result_map,
- &sr->key, sr,
- GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE));
- if (GNUNET_OK != GNUNET_BIO_read_close (rh, &emsg))
- {
- GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
- _("Failure while resuming search operation `%s': %s\n"),
- filename, emsg);
- GNUNET_free (emsg);
- }
- return GNUNET_OK;
- cleanup:
- GNUNET_free_non_null (download);
- GNUNET_free_non_null (emsg);
- GNUNET_free_non_null (uris);
- GNUNET_free_non_null (update_srch);
- if (NULL != sr->uri)
- GNUNET_FS_uri_destroy (sr->uri);
- if (NULL != sr->meta)
- GNUNET_CONTAINER_meta_data_destroy (sr->meta);
- GNUNET_free (sr->serialization);
- GNUNET_free (sr);
- if (GNUNET_OK != GNUNET_BIO_read_close (rh, &emsg))
- {
- GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
- _("Failure while resuming search operation `%s': %s\n"),
- filename, emsg);
- GNUNET_free (emsg);
- }
- return GNUNET_OK;
- }
- /**
- * Send the 'resume' signal to the callback; also actually
- * resume the download (put it in the queue). Does this
- * recursively for the top-level download and all child
- * downloads.
- *
- * @param dc download to resume
- */
- static void
- signal_download_resume (struct GNUNET_FS_DownloadContext *dc)
- {
- struct GNUNET_FS_DownloadContext *dcc;
- struct GNUNET_FS_ProgressInfo pi;
- pi.status = GNUNET_FS_STATUS_DOWNLOAD_RESUME;
- pi.value.download.specifics.resume.meta = dc->meta;
- pi.value.download.specifics.resume.message = dc->emsg;
- GNUNET_FS_download_make_status_ (&pi, dc);
- dcc = dc->child_head;
- while (NULL != dcc)
- {
- signal_download_resume (dcc);
- dcc = dcc->next;
- }
- if (NULL != dc->pending_head)
- GNUNET_FS_download_start_downloading_ (dc);
- }
- /**
- * Signal resuming of a search to our clients (for the
- * top level search and all sub-searches).
- *
- * @param sc search being resumed
- */
- static void
- signal_search_resume (struct GNUNET_FS_SearchContext *sc);
- /**
- * Iterator over search results signaling resume to the client for
- * each result.
- *
- * @param cls closure, the `struct GNUNET_FS_SearchContext *`
- * @param key current key code
- * @param value value in the hash map, the `struct GNUNET_FS_SearchResult *`
- * @return #GNUNET_YES (we should continue to iterate)
- */
- static int
- signal_result_resume (void *cls,
- const struct GNUNET_HashCode *key,
- void *value)
- {
- struct GNUNET_FS_SearchContext *sc = cls;
- struct GNUNET_FS_ProgressInfo pi;
- struct GNUNET_FS_SearchResult *sr = value;
- if (0 == sr->mandatory_missing)
- {
- pi.status = GNUNET_FS_STATUS_SEARCH_RESUME_RESULT;
- pi.value.search.specifics.resume_result.meta = sr->meta;
- pi.value.search.specifics.resume_result.uri = sr->uri;
- pi.value.search.specifics.resume_result.result = sr;
- pi.value.search.specifics.resume_result.availability_rank =
- 2 * sr->availability_success - sr->availability_trials;
- pi.value.search.specifics.resume_result.availability_certainty =
- sr->availability_trials;
- pi.value.search.specifics.resume_result.applicability_rank =
- sr->optional_support;
- sr->client_info = GNUNET_FS_search_make_status_ (&pi, sc->h, sc);
- }
- if (NULL != sr->download)
- {
- signal_download_resume (sr->download);
- }
- else
- {
- GNUNET_FS_search_start_probe_ (sr);
- }
- if (NULL != sr->update_search)
- signal_search_resume (sr->update_search);
- return GNUNET_YES;
- }
- /**
- * Free memory allocated by the search context and its children
- *
- * @param sc search context to free
- */
- static void
- free_search_context (struct GNUNET_FS_SearchContext *sc);
- /**
- * Iterator over search results freeing each.
- *
- * @param cls closure, the `struct GNUNET_FS_SearchContext *`
- * @param key current key code
- * @param value value in the hash map, the `struct GNUNET_FS_SearchResult *`
- * @return #GNUNET_YES (we should continue to iterate)
- */
- static int
- free_result (void *cls,
- const struct GNUNET_HashCode *key,
- void *value)
- {
- struct GNUNET_FS_SearchResult *sr = value;
- if (NULL != sr->update_search)
- {
- free_search_context (sr->update_search);
- GNUNET_assert (NULL == sr->update_search);
- }
- GNUNET_CONTAINER_meta_data_destroy (sr->meta);
- GNUNET_FS_uri_destroy (sr->uri);
- GNUNET_free (sr);
- return GNUNET_YES;
- }
- /**
- * Free memory allocated by the search context and its children
- *
- * @param sc search context to free
- */
- static void
- free_search_context (struct GNUNET_FS_SearchContext *sc)
- {
- if (NULL != sc->serialization)
- {
- GNUNET_FS_remove_sync_file_ (sc->h,
- (sc->psearch_result ==
- NULL) ? GNUNET_FS_SYNC_PATH_MASTER_SEARCH :
- GNUNET_FS_SYNC_PATH_CHILD_SEARCH,
- sc->serialization);
- GNUNET_FS_remove_sync_dir_ (sc->h,
- (sc->psearch_result ==
- NULL) ? GNUNET_FS_SYNC_PATH_MASTER_SEARCH :
- GNUNET_FS_SYNC_PATH_CHILD_SEARCH,
- sc->serialization);
- }
- GNUNET_free_non_null (sc->serialization);
- GNUNET_free_non_null (sc->emsg);
- if (NULL != sc->uri)
- GNUNET_FS_uri_destroy (sc->uri);
- if (NULL != sc->master_result_map)
- {
- GNUNET_CONTAINER_multihashmap_iterate (sc->master_result_map, &free_result,
- sc);
- GNUNET_CONTAINER_multihashmap_destroy (sc->master_result_map);
- }
- GNUNET_free (sc);
- }
- /**
- * Function called with a filename of serialized sub-download
- * to deserialize.
- *
- * @param cls the `struct GNUNET_FS_DownloadContext *` (parent)
- * @param filename complete filename (absolute path)
- * @return #GNUNET_OK (continue to iterate)
- */
- static int
- deserialize_subdownload (void *cls,
- const char *filename)
- {
- struct GNUNET_FS_DownloadContext *parent = cls;
- char *ser;
- char *emsg;
- struct GNUNET_BIO_ReadHandle *rh;
- ser = get_serialization_short_name (filename);
- rh = GNUNET_BIO_read_open (filename);
- if (NULL == rh)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
- _("Failed to resume sub-download `%s': could not open file `%s'\n"),
- ser,
- filename);
- GNUNET_free (ser);
- return GNUNET_OK;
- }
- deserialize_download (parent->h, rh, parent, NULL, ser);
- if (GNUNET_OK != GNUNET_BIO_read_close (rh, &emsg))
- {
- GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
- _("Failed to resume sub-download `%s': %s\n"),
- ser,
- emsg);
- GNUNET_free (emsg);
- }
- GNUNET_free (ser);
- return GNUNET_OK;
- }
- /**
- * Free this download context and all of its descendants.
- * (only works during deserialization since not all possible
- * state it taken care of).
- *
- * @param dc context to free
- */
- static void
- free_download_context (struct GNUNET_FS_DownloadContext *dc)
- {
- struct GNUNET_FS_DownloadContext *dcc;
- if (NULL != dc->meta)
- GNUNET_CONTAINER_meta_data_destroy (dc->meta);
- if (NULL != dc->uri)
- GNUNET_FS_uri_destroy (dc->uri);
- GNUNET_free_non_null (dc->temp_filename);
- GNUNET_free_non_null (dc->emsg);
- GNUNET_free_non_null (dc->filename);
- GNUNET_free_non_null (dc->serialization);
- while (NULL != (dcc = dc->child_head))
- {
- GNUNET_CONTAINER_DLL_remove (dc->child_head,
- dc->child_tail,
- dcc);
- free_download_context (dcc);
- }
- GNUNET_FS_free_download_request_ (dc->top_request);
- if (NULL != dc->active)
- GNUNET_CONTAINER_multihashmap_destroy (dc->active);
- GNUNET_free (dc);
- }
- /**
- * Deserialize a download.
- *
- * @param h overall context
- * @param rh file to deserialize from
- * @param parent parent download
- * @param search associated search
- * @param serialization name under which the search was serialized
- */
- static void
- deserialize_download (struct GNUNET_FS_Handle *h,
- struct GNUNET_BIO_ReadHandle *rh,
- struct GNUNET_FS_DownloadContext *parent,
- struct GNUNET_FS_SearchResult *search,
- const char *serialization)
- {
- struct GNUNET_FS_DownloadContext *dc;
- char *emsg;
- char *uris;
- char *dn;
- uint32_t options;
- uint32_t status;
- uris = NULL;
- emsg = NULL;
- dc = GNUNET_new (struct GNUNET_FS_DownloadContext);
- dc->parent = parent;
- dc->h = h;
- dc->serialization = GNUNET_strdup (serialization);
- if ((GNUNET_OK !=
- GNUNET_BIO_read_string (rh, "download-uri", &uris, 10 * 1024)) ||
- (NULL == (dc->uri = GNUNET_FS_uri_parse (uris, &emsg))) ||
- ((GNUNET_YES != GNUNET_FS_uri_test_chk (dc->uri)) &&
- (GNUNET_YES != GNUNET_FS_uri_test_loc (dc->uri))) ||
- (GNUNET_OK != GNUNET_BIO_read_meta_data (rh, "download-meta", &dc->meta))
- || (GNUNET_OK !=
- GNUNET_BIO_read_string (rh, "download-emsg", &dc->emsg, 10 * 1024)) ||
- (GNUNET_OK !=
- GNUNET_BIO_read_string (rh, "download-fn", &dc->filename, 10 * 1024)) ||
- (GNUNET_OK !=
- GNUNET_BIO_read_string (rh, "download-tfn", &dc->temp_filename,
- 10 * 1024)) ||
- (GNUNET_OK != GNUNET_BIO_read_int64 (rh, &dc->old_file_size)) ||
- (GNUNET_OK != GNUNET_BIO_read_int64 (rh, &dc->offset)) ||
- (GNUNET_OK != GNUNET_BIO_read_int64 (rh, &dc->length)) ||
- (GNUNET_OK != GNUNET_BIO_read_int64 (rh, &dc->completed)) ||
- (GNUNET_OK != read_start_time (rh, &dc->start_time)) ||
- (GNUNET_OK != GNUNET_BIO_read_int32 (rh, &dc->anonymity)) ||
- (GNUNET_OK != GNUNET_BIO_read_int32 (rh, &options)) ||
- (GNUNET_OK != GNUNET_BIO_read_int32 (rh, &status)))
- {
- GNUNET_break (0);
- goto cleanup;
- }
- dc->options = (enum GNUNET_FS_DownloadOptions) options;
- dc->active =
- GNUNET_CONTAINER_multihashmap_create (1 + 2 * (dc->length / DBLOCK_SIZE), GNUNET_NO);
- dc->has_finished = (int) status;
- dc->treedepth =
- GNUNET_FS_compute_depth (GNUNET_FS_uri_chk_get_file_size (dc->uri));
- if (GNUNET_FS_uri_test_loc (dc->uri))
- GNUNET_assert (GNUNET_OK ==
- GNUNET_FS_uri_loc_get_peer_identity (dc->uri, &dc->target));
- if (NULL == dc->emsg)
- {
- dc->top_request = read_download_request (rh);
- if (NULL == dc->top_request)
- {
- GNUNET_break (0);
- goto cleanup;
- }
- }
- dn = get_download_sync_filename (dc, dc->serialization, ".dir");
- if (NULL != dn)
- {
- if (GNUNET_YES == GNUNET_DISK_directory_test (dn, GNUNET_YES))
- GNUNET_DISK_directory_scan (dn, &deserialize_subdownload, dc);
- GNUNET_free (dn);
- }
- if (NULL != parent)
- {
- GNUNET_CONTAINER_DLL_insert (parent->child_head, parent->child_tail, dc);
- }
- if (NULL != search)
- {
- dc->search = search;
- search->download = dc;
- }
- if ((NULL == parent) && (NULL == search))
- {
- dc->top =
- GNUNET_FS_make_top (dc->h, &GNUNET_FS_download_signal_suspend_, dc);
- signal_download_resume (dc);
- }
- GNUNET_free (uris);
- dc->task = GNUNET_SCHEDULER_add_now (&GNUNET_FS_download_start_task_, dc);
- return;
- cleanup:
- GNUNET_free_non_null (uris);
- GNUNET_free_non_null (emsg);
- free_download_context (dc);
- }
- /**
- * Signal resuming of a search to our clients (for the
- * top level search and all sub-searches).
- *
- * @param sc search being resumed
- */
- static void
- signal_search_resume (struct GNUNET_FS_SearchContext *sc)
- {
- struct GNUNET_FS_ProgressInfo pi;
- pi.status = GNUNET_FS_STATUS_SEARCH_RESUME;
- pi.value.search.specifics.resume.message = sc->emsg;
- pi.value.search.specifics.resume.is_paused =
- (NULL == sc->client) ? GNUNET_YES : GNUNET_NO;
- sc->client_info = GNUNET_FS_search_make_status_ (&pi, sc->h, sc);
- GNUNET_CONTAINER_multihashmap_iterate (sc->master_result_map,
- &signal_result_resume, sc);
- }
- /**
- * Deserialize a search.
- *
- * @param h overall context
- * @param rh file to deserialize from
- * @param psearch_result parent search result
- * @param serialization name under which the search was serialized
- */
- static struct GNUNET_FS_SearchContext *
- deserialize_search (struct GNUNET_FS_Handle *h,
- struct GNUNET_BIO_ReadHandle *rh,
- struct GNUNET_FS_SearchResult *psearch_result,
- const char *serialization)
- {
- struct GNUNET_FS_SearchContext *sc;
- char *emsg;
- char *uris;
- char *dn;
- uint32_t options;
- char in_pause;
- if ((NULL != psearch_result) && (NULL != psearch_result->update_search))
- {
- GNUNET_break (0);
- return NULL;
- }
- uris = NULL;
- emsg = NULL;
- sc = GNUNET_new (struct GNUNET_FS_SearchContext);
- if (NULL != psearch_result)
- {
- sc->psearch_result = psearch_result;
- psearch_result->update_search = sc;
- }
- sc->h = h;
- sc->serialization = GNUNET_strdup (serialization);
- if ((GNUNET_OK !=
- GNUNET_BIO_read_string (rh, "search-uri", &uris, 10 * 1024)) ||
- (NULL == (sc->uri = GNUNET_FS_uri_parse (uris, &emsg))) ||
- ((GNUNET_YES != GNUNET_FS_uri_test_ksk (sc->uri)) &&
- (GNUNET_YES != GNUNET_FS_uri_test_sks (sc->uri))) ||
- (GNUNET_OK != read_start_time (rh, &sc->start_time)) ||
- (GNUNET_OK !=
- GNUNET_BIO_read_string (rh, "search-emsg", &sc->emsg, 10 * 1024)) ||
- (GNUNET_OK != GNUNET_BIO_read_int32 (rh, &options)) ||
- (GNUNET_OK !=
- GNUNET_BIO_read (rh, "search-pause", &in_pause, sizeof (in_pause))) ||
- (GNUNET_OK != GNUNET_BIO_read_int32 (rh, &sc->anonymity)))
- {
- GNUNET_break (0);
- goto cleanup;
- }
- sc->options = (enum GNUNET_FS_SearchOptions) options;
- sc->master_result_map = GNUNET_CONTAINER_multihashmap_create (16, GNUNET_NO);
- dn = get_serialization_file_name_in_dir (h,
- (NULL == sc->psearch_result) ?
- GNUNET_FS_SYNC_PATH_MASTER_SEARCH :
- GNUNET_FS_SYNC_PATH_CHILD_SEARCH,
- sc->serialization, "");
- if (NULL != dn)
- {
- if (GNUNET_YES == GNUNET_DISK_directory_test (dn, GNUNET_YES))
- GNUNET_DISK_directory_scan (dn, &deserialize_search_result, sc);
- GNUNET_free (dn);
- }
- if (('\0' == in_pause) &&
- (GNUNET_OK != GNUNET_FS_search_start_searching_ (sc)))
- {
- GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
- _("Could not resume running search, will resume as paused search\n"));
- }
- signal_search_resume (sc);
- GNUNET_free (uris);
- return sc;
- cleanup:
- GNUNET_free_non_null (emsg);
- free_search_context (sc);
- GNUNET_free_non_null (uris);
- return NULL;
- }
- /**
- * Function called with a filename of serialized search operation
- * to deserialize.
- *
- * @param cls the `struct GNUNET_FS_Handle *`
- * @param filename complete filename (absolute path)
- * @return #GNUNET_OK (continue to iterate)
- */
- static int
- deserialize_search_file (void *cls,
- const char *filename)
- {
- struct GNUNET_FS_Handle *h = cls;
- char *ser;
- char *emsg;
- struct GNUNET_BIO_ReadHandle *rh;
- struct GNUNET_FS_SearchContext *sc;
- struct stat buf;
- if (0 != STAT (filename, &buf))
- {
- GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_WARNING, "stat", filename);
- return GNUNET_OK;
- }
- if (S_ISDIR (buf.st_mode))
- return GNUNET_OK; /* skip directories */
- ser = get_serialization_short_name (filename);
- rh = GNUNET_BIO_read_open (filename);
- if (NULL == rh)
- {
- if (NULL != ser)
- {
- GNUNET_FS_remove_sync_file_ (h, GNUNET_FS_SYNC_PATH_MASTER_SEARCH, ser);
- GNUNET_free (ser);
- }
- return GNUNET_OK;
- }
- sc = deserialize_search (h, rh, NULL, ser);
- if (NULL != sc)
- sc->top = GNUNET_FS_make_top (h, &GNUNET_FS_search_signal_suspend_, sc);
- GNUNET_free (ser);
- if (GNUNET_OK != GNUNET_BIO_read_close (rh, &emsg))
- {
- GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
- _("Failure while resuming search operation `%s': %s\n"),
- filename, emsg);
- GNUNET_free (emsg);
- }
- return GNUNET_OK;
- }
- /**
- * Function called with a filename of serialized download operation
- * to deserialize.
- *
- * @param cls the `struct GNUNET_FS_Handle *`
- * @param filename complete filename (absolute path)
- * @return #GNUNET_OK (continue to iterate)
- */
- static int
- deserialize_download_file (void *cls, const char *filename)
- {
- struct GNUNET_FS_Handle *h = cls;
- char *ser;
- char *emsg;
- struct GNUNET_BIO_ReadHandle *rh;
- ser = get_serialization_short_name (filename);
- rh = GNUNET_BIO_read_open (filename);
- if (NULL == rh)
- {
- if (0 != UNLINK (filename))
- GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_WARNING, "unlink", filename);
- GNUNET_free (ser);
- return GNUNET_OK;
- }
- deserialize_download (h, rh, NULL, NULL, ser);
- GNUNET_free (ser);
- if (GNUNET_OK != GNUNET_BIO_read_close (rh, &emsg))
- {
- GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
- _("Failure while resuming download operation `%s': %s\n"),
- filename, emsg);
- GNUNET_free (emsg);
- }
- return GNUNET_OK;
- }
- /**
- * Deserialize informatin about pending operations.
- *
- * @param master_path which master directory should be scanned
- * @param proc function to call for each entry (will get 'h' for 'cls')
- * @param h the 'struct GNUNET_FS_Handle*'
- */
- static void
- deserialization_master (const char *master_path, GNUNET_FileNameCallback proc,
- struct GNUNET_FS_Handle *h)
- {
- char *dn;
- dn = get_serialization_file_name (h, master_path, "");
- if (NULL == dn)
- return;
- if (GNUNET_YES == GNUNET_DISK_directory_test (dn, GNUNET_YES))
- GNUNET_DISK_directory_scan (dn, proc, h);
- GNUNET_free (dn);
- }
- /**
- * Setup a connection to the file-sharing service.
- *
- * @param cfg configuration to use
- * @param client_name unique identifier for this client
- * @param upcb function to call to notify about FS actions
- * @param upcb_cls closure for @a upcb
- * @param flags specific attributes for fs-operations
- * @param ... list of optional options, terminated with #GNUNET_FS_OPTIONS_END
- * @return NULL on error
- */
- struct GNUNET_FS_Handle *
- GNUNET_FS_start (const struct GNUNET_CONFIGURATION_Handle *cfg,
- const char *client_name,
- GNUNET_FS_ProgressCallback upcb,
- void *upcb_cls,
- enum GNUNET_FS_Flags flags, ...)
- {
- struct GNUNET_FS_Handle *ret;
- enum GNUNET_FS_OPTIONS opt;
- va_list ap;
- ret = GNUNET_new (struct GNUNET_FS_Handle);
- ret->cfg = cfg;
- ret->client_name = GNUNET_strdup (client_name);
- ret->upcb = upcb;
- ret->upcb_cls = upcb_cls;
- ret->flags = flags;
- ret->max_parallel_downloads = DEFAULT_MAX_PARALLEL_DOWNLOADS;
- ret->max_parallel_requests = DEFAULT_MAX_PARALLEL_REQUESTS;
- ret->avg_block_latency = GNUNET_TIME_UNIT_MINUTES; /* conservative starting point */
- va_start (ap, flags);
- while (GNUNET_FS_OPTIONS_END != (opt = va_arg (ap, enum GNUNET_FS_OPTIONS)))
- {
- switch (opt)
- {
- case GNUNET_FS_OPTIONS_DOWNLOAD_PARALLELISM:
- ret->max_parallel_downloads = va_arg (ap, unsigned int);
- break;
- case GNUNET_FS_OPTIONS_REQUEST_PARALLELISM:
- ret->max_parallel_requests = va_arg (ap, unsigned int);
- break;
- default:
- GNUNET_break (0);
- GNUNET_free (ret->client_name);
- GNUNET_free (ret);
- va_end (ap);
- return NULL;
- }
- }
- va_end (ap);
- if (0 != (GNUNET_FS_FLAGS_PERSISTENCE & flags))
- {
- deserialization_master (GNUNET_FS_SYNC_PATH_MASTER_PUBLISH,
- &deserialize_publish_file, ret);
- deserialization_master (GNUNET_FS_SYNC_PATH_MASTER_SEARCH,
- &deserialize_search_file, ret);
- deserialization_master (GNUNET_FS_SYNC_PATH_MASTER_DOWNLOAD,
- &deserialize_download_file, ret);
- deserialization_master (GNUNET_FS_SYNC_PATH_MASTER_UNINDEX,
- &deserialize_unindex_file, ret);
- }
- return ret;
- }
- /**
- * Close our connection with the file-sharing service.
- * The callback given to GNUNET_FS_start will no longer be
- * called after this function returns.
- *
- * @param h handle that was returned from #GNUNET_FS_start()
- */
- void
- GNUNET_FS_stop (struct GNUNET_FS_Handle *h)
- {
- while (h->top_head != NULL)
- h->top_head->ssf (h->top_head->ssf_cls);
- if (h->queue_job != GNUNET_SCHEDULER_NO_TASK)
- GNUNET_SCHEDULER_cancel (h->queue_job);
- GNUNET_free (h->client_name);
- GNUNET_free (h);
- }
- /* end of fs_api.c */
|