]>
git.saurik.com Git - apt.git/blob - apt-pkg/acquire.cc
cb32e8f2ba2e6dda4623d1e3c0133517f34db94a
1 // -*- mode: cpp; mode: fold -*-
3 // $Id: acquire.cc,v 1.50 2004/03/17 05:17:11 mdz Exp $
4 /* ######################################################################
6 Acquire - File Acquiration
8 The core element for the schedule system is the concept of a named
9 queue. Each queue is unique and each queue has a name derived from the
10 URI. The degree of paralization can be controlled by how the queue
11 name is derived from the URI.
13 ##################################################################### */
15 // Include Files /*{{{*/
18 #include <apt-pkg/acquire.h>
19 #include <apt-pkg/acquire-item.h>
20 #include <apt-pkg/acquire-worker.h>
21 #include <apt-pkg/configuration.h>
22 #include <apt-pkg/error.h>
23 #include <apt-pkg/strutl.h>
24 #include <apt-pkg/fileutl.h>
42 #include <sys/select.h>
51 // Acquire::pkgAcquire - Constructor /*{{{*/
52 // ---------------------------------------------------------------------
53 /* We grab some runtime state from the configuration space */
54 pkgAcquire::pkgAcquire() : LockFD(-1), d(NULL
), Queues(0), Workers(0), Configs(0), Log(NULL
), ToFetch(0),
55 Debug(_config
->FindB("Debug::pkgAcquire",false)),
60 pkgAcquire::pkgAcquire(pkgAcquireStatus
*Progress
) : LockFD(-1), d(NULL
), Queues(0), Workers(0),
61 Configs(0), Log(NULL
), ToFetch(0),
62 Debug(_config
->FindB("Debug::pkgAcquire",false)),
68 void pkgAcquire::Initialize()
70 string
const Mode
= _config
->Find("Acquire::Queue-Mode","host");
71 if (strcasecmp(Mode
.c_str(),"host") == 0)
72 QueueMode
= QueueHost
;
73 if (strcasecmp(Mode
.c_str(),"access") == 0)
74 QueueMode
= QueueAccess
;
76 // chown the auth.conf file as it will be accessed by our methods
77 std::string
const SandboxUser
= _config
->Find("APT::Sandbox::User");
78 if (getuid() == 0 && SandboxUser
.empty() == false) // if we aren't root, we can't chown, so don't try it
80 struct passwd
const * const pw
= getpwnam(SandboxUser
.c_str());
81 struct group
const * const gr
= getgrnam("root");
82 if (pw
!= NULL
&& gr
!= NULL
)
84 std::string
const AuthConf
= _config
->FindFile("Dir::Etc::netrc");
85 if(AuthConf
.empty() == false && RealFileExists(AuthConf
) &&
86 chown(AuthConf
.c_str(), pw
->pw_uid
, gr
->gr_gid
) != 0)
87 _error
->WarningE("SetupAPTPartialDirectory", "chown to %s:root of file %s failed", SandboxUser
.c_str(), AuthConf
.c_str());
92 // Acquire::GetLock - lock directory and prepare for action /*{{{*/
93 static bool SetupAPTPartialDirectory(std::string
const &grand
, std::string
const &parent
)
95 std::string
const partial
= parent
+ "partial";
96 mode_t
const mode
= umask(S_IWGRP
| S_IWOTH
);
97 bool const creation_fail
= (CreateAPTDirectoryIfNeeded(grand
, partial
) == false &&
98 CreateAPTDirectoryIfNeeded(parent
, partial
) == false);
100 if (creation_fail
== true)
103 std::string
const SandboxUser
= _config
->Find("APT::Sandbox::User");
104 if (getuid() == 0 && SandboxUser
.empty() == false) // if we aren't root, we can't chown, so don't try it
106 struct passwd
const * const pw
= getpwnam(SandboxUser
.c_str());
107 struct group
const * const gr
= getgrnam("root");
108 if (pw
!= NULL
&& gr
!= NULL
)
110 // chown the partial dir
111 if(chown(partial
.c_str(), pw
->pw_uid
, gr
->gr_gid
) != 0)
112 _error
->WarningE("SetupAPTPartialDirectory", "chown to %s:root of directory %s failed", SandboxUser
.c_str(), partial
.c_str());
115 if (chmod(partial
.c_str(), 0700) != 0)
116 _error
->WarningE("SetupAPTPartialDirectory", "chmod 0700 of directory %s failed", partial
.c_str());
120 bool pkgAcquire::Setup(pkgAcquireStatus
*Progress
, string
const &Lock
)
125 string
const listDir
= _config
->FindDir("Dir::State::lists");
126 if (SetupAPTPartialDirectory(_config
->FindDir("Dir::State"), listDir
) == false)
127 return _error
->Errno("Acquire", _("List directory %spartial is missing."), listDir
.c_str());
128 string
const archivesDir
= _config
->FindDir("Dir::Cache::Archives");
129 if (SetupAPTPartialDirectory(_config
->FindDir("Dir::Cache"), archivesDir
) == false)
130 return _error
->Errno("Acquire", _("Archives directory %spartial is missing."), archivesDir
.c_str());
133 return GetLock(Lock
);
135 bool pkgAcquire::GetLock(std::string
const &Lock
)
137 if (Lock
.empty() == true)
140 // check for existence and possibly create auxiliary directories
141 string
const listDir
= _config
->FindDir("Dir::State::lists");
142 string
const archivesDir
= _config
->FindDir("Dir::Cache::Archives");
146 if (SetupAPTPartialDirectory(_config
->FindDir("Dir::State"), listDir
) == false)
147 return _error
->Errno("Acquire", _("List directory %spartial is missing."), listDir
.c_str());
149 if (Lock
== archivesDir
)
151 if (SetupAPTPartialDirectory(_config
->FindDir("Dir::Cache"), archivesDir
) == false)
152 return _error
->Errno("Acquire", _("Archives directory %spartial is missing."), archivesDir
.c_str());
155 if (_config
->FindB("Debug::NoLocking", false) == true)
158 // Lock the directory this acquire object will work in
161 LockFD
= ::GetLock(flCombine(Lock
, "lock"));
163 return _error
->Error(_("Unable to lock directory %s"), Lock
.c_str());
168 // Acquire::~pkgAcquire - Destructor /*{{{*/
169 // ---------------------------------------------------------------------
170 /* Free our memory, clean up the queues (destroy the workers) */
171 pkgAcquire::~pkgAcquire()
180 MethodConfig
*Jnk
= Configs
;
181 Configs
= Configs
->Next
;
186 // Acquire::Shutdown - Clean out the acquire object /*{{{*/
187 // ---------------------------------------------------------------------
189 void pkgAcquire::Shutdown()
191 while (Items
.empty() == false)
193 if (Items
[0]->Status
== Item::StatFetching
)
194 Items
[0]->Status
= Item::StatError
;
201 Queues
= Queues
->Next
;
206 // Acquire::Add - Add a new item /*{{{*/
207 // ---------------------------------------------------------------------
208 /* This puts an item on the acquire list. This list is mainly for tracking
210 void pkgAcquire::Add(Item
*Itm
)
212 Items
.push_back(Itm
);
215 // Acquire::Remove - Remove a item /*{{{*/
216 // ---------------------------------------------------------------------
217 /* Remove an item from the acquire list. This is usually not used.. */
218 void pkgAcquire::Remove(Item
*Itm
)
222 for (ItemIterator I
= Items
.begin(); I
!= Items
.end();)
234 // Acquire::Add - Add a worker /*{{{*/
235 // ---------------------------------------------------------------------
236 /* A list of workers is kept so that the select loop can direct their FD
238 void pkgAcquire::Add(Worker
*Work
)
240 Work
->NextAcquire
= Workers
;
244 // Acquire::Remove - Remove a worker /*{{{*/
245 // ---------------------------------------------------------------------
246 /* A worker has died. This can not be done while the select loop is running
247 as it would require that RunFds could handling a changing list state and
249 void pkgAcquire::Remove(Worker
*Work
)
254 Worker
**I
= &Workers
;
258 *I
= (*I
)->NextAcquire
;
260 I
= &(*I
)->NextAcquire
;
264 // Acquire::Enqueue - Queue an URI for fetching /*{{{*/
265 // ---------------------------------------------------------------------
266 /* This is the entry point for an item. An item calls this function when
267 it is constructed which creates a queue (based on the current queue
268 mode) and puts the item in that queue. If the system is running then
269 the queue might be started. */
270 void pkgAcquire::Enqueue(ItemDesc
&Item
)
272 // Determine which queue to put the item in
273 const MethodConfig
*Config
;
274 string Name
= QueueName(Item
.URI
,Config
);
275 if (Name
.empty() == true)
278 // Find the queue structure
280 for (; I
!= 0 && I
->Name
!= Name
; I
= I
->Next
);
283 I
= new Queue(Name
,this);
291 // See if this is a local only URI
292 if (Config
->LocalOnly
== true && Item
.Owner
->Complete
== false)
293 Item
.Owner
->Local
= true;
294 Item
.Owner
->Status
= Item::StatIdle
;
296 // Queue it into the named queue
303 clog
<< "Fetching " << Item
.URI
<< endl
;
304 clog
<< " to " << Item
.Owner
->DestFile
<< endl
;
305 clog
<< " Queue is: " << Name
<< endl
;
309 // Acquire::Dequeue - Remove an item from all queues /*{{{*/
310 // ---------------------------------------------------------------------
311 /* This is called when an item is finished being fetched. It removes it
312 from all the queues */
313 void pkgAcquire::Dequeue(Item
*Itm
)
318 clog
<< "Dequeuing " << Itm
->DestFile
<< endl
;
320 for (; I
!= 0; I
= I
->Next
)
326 clog
<< "Dequeued from " << I
->Name
<< endl
;
334 // Acquire::QueueName - Return the name of the queue for this URI /*{{{*/
335 // ---------------------------------------------------------------------
336 /* The string returned depends on the configuration settings and the
337 method parameters. Given something like http://foo.org/bar it can
338 return http://foo.org or http */
339 string
pkgAcquire::QueueName(string Uri
,MethodConfig
const *&Config
)
343 Config
= GetConfig(U
.Access
);
347 /* Single-Instance methods get exactly one queue per URI. This is
348 also used for the Access queue method */
349 if (Config
->SingleInstance
== true || QueueMode
== QueueAccess
)
352 string AccessSchema
= U
.Access
+ ':',
353 FullQueueName
= AccessSchema
+ U
.Host
;
354 unsigned int Instances
= 0, SchemaLength
= AccessSchema
.length();
357 for (; I
!= 0; I
= I
->Next
) {
358 // if the queue already exists, re-use it
359 if (I
->Name
== FullQueueName
)
360 return FullQueueName
;
362 if (I
->Name
.compare(0, SchemaLength
, AccessSchema
) == 0)
367 clog
<< "Found " << Instances
<< " instances of " << U
.Access
<< endl
;
370 if (Instances
>= (unsigned int)_config
->FindI("Acquire::QueueHost::Limit",10))
373 return FullQueueName
;
376 // Acquire::GetConfig - Fetch the configuration information /*{{{*/
377 // ---------------------------------------------------------------------
378 /* This locates the configuration structure for an access method. If
379 a config structure cannot be found a Worker will be created to
381 pkgAcquire::MethodConfig
*pkgAcquire::GetConfig(string Access
)
383 // Search for an existing config
385 for (Conf
= Configs
; Conf
!= 0; Conf
= Conf
->Next
)
386 if (Conf
->Access
== Access
)
389 // Create the new config class
390 Conf
= new MethodConfig
;
391 Conf
->Access
= Access
;
392 Conf
->Next
= Configs
;
395 // Create the worker to fetch the configuration
397 if (Work
.Start() == false)
400 /* if a method uses DownloadLimit, we switch to SingleInstance mode */
401 if(_config
->FindI("Acquire::"+Access
+"::Dl-Limit",0) > 0)
402 Conf
->SingleInstance
= true;
407 // Acquire::SetFds - Deal with readable FDs /*{{{*/
408 // ---------------------------------------------------------------------
409 /* Collect FDs that have activity monitors into the fd sets */
410 void pkgAcquire::SetFds(int &Fd
,fd_set
*RSet
,fd_set
*WSet
)
412 for (Worker
*I
= Workers
; I
!= 0; I
= I
->NextAcquire
)
414 if (I
->InReady
== true && I
->InFd
>= 0)
418 FD_SET(I
->InFd
,RSet
);
420 if (I
->OutReady
== true && I
->OutFd
>= 0)
424 FD_SET(I
->OutFd
,WSet
);
429 // Acquire::RunFds - Deal with active FDs /*{{{*/
430 // ---------------------------------------------------------------------
431 /* Dispatch active FDs over to the proper workers. It is very important
432 that a worker never be erased while this is running! The queue class
433 should never erase a worker except during shutdown processing. */
434 void pkgAcquire::RunFds(fd_set
*RSet
,fd_set
*WSet
)
436 for (Worker
*I
= Workers
; I
!= 0; I
= I
->NextAcquire
)
438 if (I
->InFd
>= 0 && FD_ISSET(I
->InFd
,RSet
) != 0)
440 if (I
->OutFd
>= 0 && FD_ISSET(I
->OutFd
,WSet
) != 0)
445 // Acquire::Run - Run the fetch sequence /*{{{*/
446 // ---------------------------------------------------------------------
447 /* This runs the queues. It manages a select loop for all of the
448 Worker tasks. The workers interact with the queues and items to
449 manage the actual fetch. */
450 static void CheckDropPrivsMustBeDisabled(pkgAcquire
const &Fetcher
)
455 std::string SandboxUser
= _config
->Find("APT::Sandbox::User");
456 if (SandboxUser
.empty())
459 struct passwd
const * const pw
= getpwnam(SandboxUser
.c_str());
463 if (setegid(pw
->pw_gid
) != 0)
464 _error
->Errno("setegid", "setegid %u failed", pw
->pw_gid
);
465 if (seteuid(pw
->pw_uid
) != 0)
466 _error
->Errno("seteuid", "seteuid %u failed", pw
->pw_uid
);
468 bool dropPrivs
= true;
469 for (pkgAcquire::ItemCIterator I
= Fetcher
.ItemsBegin();
470 I
!= Fetcher
.ItemsEnd() && dropPrivs
== true; ++I
)
472 if ((*I
)->DestFile
.empty())
475 // we check directory instead of file as the file might or might not
476 // exist already as a link or not which complicates everything…
477 std::string dirname
= flNotFile((*I
)->DestFile
);
478 if (DirectoryExists(dirname
))
481 continue; // assume it is created correctly by the acquire system
483 if (faccessat(AT_FDCWD
, dirname
.c_str(), R_OK
| W_OK
| X_OK
, AT_EACCESS
| AT_SYMLINK_NOFOLLOW
) != 0)
486 _error
->WarningE("pkgAcquire::Run", _("Can't drop privileges for downloading as file '%s' couldn't be accessed by user '%s'."),
487 (*I
)->DestFile
.c_str(), SandboxUser
.c_str());
488 _config
->Set("APT::Sandbox::User", "");
494 _error
->Errno("seteuid", "seteuid %u failed", 0);
496 _error
->Errno("setegid", "setegid %u failed", 0);
498 pkgAcquire::RunResult
pkgAcquire::Run(int PulseIntervall
)
500 CheckDropPrivsMustBeDisabled(*this);
504 for (Queue
*I
= Queues
; I
!= 0; I
= I
->Next
)
510 bool WasCancelled
= false;
512 // Run till all things have been acquired
515 tv
.tv_usec
= PulseIntervall
;
523 SetFds(Highest
,&RFds
,&WFds
);
528 Res
= select(Highest
+1,&RFds
,&WFds
,0,&tv
);
530 while (Res
< 0 && errno
== EINTR
);
534 _error
->Errno("select","Select has failed");
539 if (_error
->PendingError() == true)
542 // Timeout, notify the log class
543 if (Res
== 0 || (Log
!= 0 && Log
->Update
== true))
545 tv
.tv_usec
= PulseIntervall
;
546 for (Worker
*I
= Workers
; I
!= 0; I
= I
->NextAcquire
)
548 if (Log
!= 0 && Log
->Pulse(this) == false)
559 // Shut down the acquire bits
561 for (Queue
*I
= Queues
; I
!= 0; I
= I
->Next
)
564 // Shut down the items
565 for (ItemIterator I
= Items
.begin(); I
!= Items
.end(); ++I
)
568 if (_error
->PendingError())
575 // Acquire::Bump - Called when an item is dequeued /*{{{*/
576 // ---------------------------------------------------------------------
577 /* This routine bumps idle queues in hopes that they will be able to fetch
579 void pkgAcquire::Bump()
581 for (Queue
*I
= Queues
; I
!= 0; I
= I
->Next
)
585 // Acquire::WorkerStep - Step to the next worker /*{{{*/
586 // ---------------------------------------------------------------------
587 /* Not inlined to advoid including acquire-worker.h */
588 pkgAcquire::Worker
*pkgAcquire::WorkerStep(Worker
*I
)
590 return I
->NextAcquire
;
593 // Acquire::Clean - Cleans a directory /*{{{*/
594 // ---------------------------------------------------------------------
595 /* This is a bit simplistic, it looks at every file in the dir and sees
596 if it is part of the download set. */
597 bool pkgAcquire::Clean(string Dir
)
599 // non-existing directories are by definition clean…
600 if (DirectoryExists(Dir
) == false)
604 return _error
->Error(_("Clean of %s is not supported"), Dir
.c_str());
606 DIR *D
= opendir(Dir
.c_str());
608 return _error
->Errno("opendir",_("Unable to read %s"),Dir
.c_str());
610 string StartDir
= SafeGetCWD();
611 if (chdir(Dir
.c_str()) != 0)
614 return _error
->Errno("chdir",_("Unable to change to %s"),Dir
.c_str());
617 for (struct dirent
*Dir
= readdir(D
); Dir
!= 0; Dir
= readdir(D
))
620 if (strcmp(Dir
->d_name
,"lock") == 0 ||
621 strcmp(Dir
->d_name
,"partial") == 0 ||
622 strcmp(Dir
->d_name
,".") == 0 ||
623 strcmp(Dir
->d_name
,"..") == 0)
626 // Look in the get list
627 ItemCIterator I
= Items
.begin();
628 for (; I
!= Items
.end(); ++I
)
629 if (flNotDir((*I
)->DestFile
) == Dir
->d_name
)
632 // Nothing found, nuke it
633 if (I
== Items
.end())
638 if (chdir(StartDir
.c_str()) != 0)
639 return _error
->Errno("chdir",_("Unable to change to %s"),StartDir
.c_str());
643 // Acquire::TotalNeeded - Number of bytes to fetch /*{{{*/
644 // ---------------------------------------------------------------------
645 /* This is the total number of bytes needed */
646 APT_PURE
unsigned long long pkgAcquire::TotalNeeded()
648 unsigned long long Total
= 0;
649 for (ItemCIterator I
= ItemsBegin(); I
!= ItemsEnd(); ++I
)
650 Total
+= (*I
)->FileSize
;
654 // Acquire::FetchNeeded - Number of bytes needed to get /*{{{*/
655 // ---------------------------------------------------------------------
656 /* This is the number of bytes that is not local */
657 APT_PURE
unsigned long long pkgAcquire::FetchNeeded()
659 unsigned long long Total
= 0;
660 for (ItemCIterator I
= ItemsBegin(); I
!= ItemsEnd(); ++I
)
661 if ((*I
)->Local
== false)
662 Total
+= (*I
)->FileSize
;
666 // Acquire::PartialPresent - Number of partial bytes we already have /*{{{*/
667 // ---------------------------------------------------------------------
668 /* This is the number of bytes that is not local */
669 APT_PURE
unsigned long long pkgAcquire::PartialPresent()
671 unsigned long long Total
= 0;
672 for (ItemCIterator I
= ItemsBegin(); I
!= ItemsEnd(); ++I
)
673 if ((*I
)->Local
== false)
674 Total
+= (*I
)->PartialSize
;
678 // Acquire::UriBegin - Start iterator for the uri list /*{{{*/
679 // ---------------------------------------------------------------------
681 pkgAcquire::UriIterator
pkgAcquire::UriBegin()
683 return UriIterator(Queues
);
686 // Acquire::UriEnd - End iterator for the uri list /*{{{*/
687 // ---------------------------------------------------------------------
689 pkgAcquire::UriIterator
pkgAcquire::UriEnd()
691 return UriIterator(0);
694 // Acquire::MethodConfig::MethodConfig - Constructor /*{{{*/
695 // ---------------------------------------------------------------------
697 pkgAcquire::MethodConfig::MethodConfig() : d(NULL
), Next(0), SingleInstance(false),
698 Pipeline(false), SendConfig(false), LocalOnly(false), NeedsCleanup(false),
703 // Queue::Queue - Constructor /*{{{*/
704 // ---------------------------------------------------------------------
706 pkgAcquire::Queue::Queue(string
const &name
,pkgAcquire
* const owner
) : d(NULL
), Next(0),
707 Name(name
), Items(0), Workers(0), Owner(owner
), PipeDepth(0), MaxPipeDepth(1)
711 // Queue::~Queue - Destructor /*{{{*/
712 // ---------------------------------------------------------------------
714 pkgAcquire::Queue::~Queue()
726 // Queue::Enqueue - Queue an item to the queue /*{{{*/
727 // ---------------------------------------------------------------------
729 bool pkgAcquire::Queue::Enqueue(ItemDesc
&Item
)
732 // move to the end of the queue and check for duplicates here
733 HashStringList
const hsl
= Item
.Owner
->GetExpectedHashes();
734 for (; *I
!= 0; I
= &(*I
)->Next
)
735 if (Item
.URI
== (*I
)->URI
|| hsl
== (*I
)->Owner
->GetExpectedHashes())
737 if (_config
->FindB("Debug::pkgAcquire::Worker",false) == true)
738 std::cerr
<< " @ Queue: Action combined for " << Item
.URI
<< " and " << (*I
)->URI
<< std::endl
;
739 (*I
)->Owners
.push_back(Item
.Owner
);
740 Item
.Owner
->Status
= (*I
)->Owner
->Status
;
745 QItem
*Itm
= new QItem
;
750 Item
.Owner
->QueueCounter
++;
751 if (Items
->Next
== 0)
756 // Queue::Dequeue - Remove an item from the queue /*{{{*/
757 // ---------------------------------------------------------------------
758 /* We return true if we hit something */
759 bool pkgAcquire::Queue::Dequeue(Item
*Owner
)
761 if (Owner
->Status
== pkgAcquire::Item::StatFetching
)
762 return _error
->Error("Tried to dequeue a fetching object");
769 if (Owner
== (*I
)->Owner
)
773 Owner
->QueueCounter
--;
784 // Queue::Startup - Start the worker processes /*{{{*/
785 // ---------------------------------------------------------------------
786 /* It is possible for this to be called with a pre-existing set of
788 bool pkgAcquire::Queue::Startup()
793 pkgAcquire::MethodConfig
*Cnf
= Owner
->GetConfig(U
.Access
);
797 Workers
= new Worker(this,Cnf
,Owner
->Log
);
799 if (Workers
->Start() == false)
802 /* When pipelining we commit 10 items. This needs to change when we
803 added other source retry to have cycle maintain a pipeline depth
805 if (Cnf
->Pipeline
== true)
806 MaxPipeDepth
= _config
->FindI("Acquire::Max-Pipeline-Depth",10);
814 // Queue::Shutdown - Shutdown the worker processes /*{{{*/
815 // ---------------------------------------------------------------------
816 /* If final is true then all workers are eliminated, otherwise only workers
817 that do not need cleanup are removed */
818 bool pkgAcquire::Queue::Shutdown(bool Final
)
820 // Delete all of the workers
821 pkgAcquire::Worker
**Cur
= &Workers
;
824 pkgAcquire::Worker
*Jnk
= *Cur
;
825 if (Final
== true || Jnk
->GetConf()->NeedsCleanup
== false)
827 *Cur
= Jnk
->NextQueue
;
832 Cur
= &(*Cur
)->NextQueue
;
838 // Queue::FindItem - Find a URI in the item list /*{{{*/
839 // ---------------------------------------------------------------------
841 pkgAcquire::Queue::QItem
*pkgAcquire::Queue::FindItem(string URI
,pkgAcquire::Worker
*Owner
)
843 for (QItem
*I
= Items
; I
!= 0; I
= I
->Next
)
844 if (I
->URI
== URI
&& I
->Worker
== Owner
)
849 // Queue::ItemDone - Item has been completed /*{{{*/
850 // ---------------------------------------------------------------------
851 /* The worker signals this which causes the item to be removed from the
852 queue. If this is the last queue instance then it is removed from the
854 bool pkgAcquire::Queue::ItemDone(QItem
*Itm
)
857 for (QItem::owner_iterator O
= Itm
->Owners
.begin(); O
!= Itm
->Owners
.end(); ++O
)
859 if ((*O
)->Status
== pkgAcquire::Item::StatFetching
)
860 (*O
)->Status
= pkgAcquire::Item::StatDone
;
863 if (Itm
->Owner
->QueueCounter
<= 1)
864 Owner
->Dequeue(Itm
->Owner
);
874 // Queue::Cycle - Queue new items into the method /*{{{*/
875 // ---------------------------------------------------------------------
876 /* This locates a new idle item and sends it to the worker. If pipelining
877 is enabled then it keeps the pipe full. */
878 bool pkgAcquire::Queue::Cycle()
880 if (Items
== 0 || Workers
== 0)
884 return _error
->Error("Pipedepth failure");
886 // Look for a queable item
888 while (PipeDepth
< (signed)MaxPipeDepth
)
890 for (; I
!= 0; I
= I
->Next
)
891 if (I
->Owner
->Status
== pkgAcquire::Item::StatIdle
)
894 // Nothing to do, queue is idle.
899 for (QItem::owner_iterator O
= I
->Owners
.begin(); O
!= I
->Owners
.end(); ++O
)
900 (*O
)->Status
= pkgAcquire::Item::StatFetching
;
902 if (Workers
->QueueItem(I
) == false)
909 // Queue::Bump - Fetch any pending objects if we are idle /*{{{*/
910 // ---------------------------------------------------------------------
911 /* This is called when an item in multiple queues is dequeued */
912 void pkgAcquire::Queue::Bump()
917 HashStringList
pkgAcquire::Queue::QItem::GetExpectedHashes() const /*{{{*/
919 /* each Item can have multiple owners and each owner might have different
920 hashes, even if that is unlikely in practice and if so at least some
921 owners will later fail. There is one situation through which is not a
922 failure and still needs this handling: Two owners who expect the same
923 file, but one owner only knows the SHA1 while the other only knows SHA256. */
924 HashStringList superhsl
;
925 for (pkgAcquire::Queue::QItem::owner_iterator O
= Owners
.begin(); O
!= Owners
.end(); ++O
)
927 HashStringList
const hsl
= (*O
)->GetExpectedHashes();
928 if (hsl
.usable() == false)
930 if (superhsl
.usable() == false)
934 // we merge both lists - if we find disagreement send no hashes
935 HashStringList::const_iterator hs
= hsl
.begin();
936 for (; hs
!= hsl
.end(); ++hs
)
937 if (superhsl
.push_back(*hs
) == false)
949 APT_PURE
unsigned long long pkgAcquire::Queue::QItem::GetMaximumSize() const /*{{{*/
951 unsigned long long Maximum
= std::numeric_limits
<unsigned long long>::max();
952 for (pkgAcquire::Queue::QItem::owner_iterator O
= Owners
.begin(); O
!= Owners
.end(); ++O
)
954 if ((*O
)->FileSize
== 0)
956 Maximum
= std::min(Maximum
, (*O
)->FileSize
);
958 if (Maximum
== std::numeric_limits
<unsigned long long>::max())
963 void pkgAcquire::Queue::QItem::SyncDestinationFiles() const /*{{{*/
965 /* ensure that the first owner has the best partial file of all and
966 the rest have (potentially dangling) symlinks to it so that
967 everything (like progress reporting) finds it easily */
968 std::string superfile
= Owner
->DestFile
;
970 for (pkgAcquire::Queue::QItem::owner_iterator O
= Owners
.begin(); O
!= Owners
.end(); ++O
)
972 if ((*O
)->DestFile
== superfile
)
975 if (lstat((*O
)->DestFile
.c_str(),&file
) == 0)
977 if ((file
.st_mode
& S_IFREG
) == 0)
978 unlink((*O
)->DestFile
.c_str());
979 else if (supersize
< file
.st_size
)
981 supersize
= file
.st_size
;
982 unlink(superfile
.c_str());
983 rename((*O
)->DestFile
.c_str(), superfile
.c_str());
986 unlink((*O
)->DestFile
.c_str());
987 if (symlink(superfile
.c_str(), (*O
)->DestFile
.c_str()) != 0)
989 ; // not a problem per-se and no real alternative
995 std::string
pkgAcquire::Queue::QItem::Custom600Headers() const /*{{{*/
997 /* The others are relatively easy to merge, but this one?
998 Lets not merge and see how far we can run with it…
999 Likely, nobody will ever notice as all the items will
1000 be of the same class and hence generate the same headers. */
1001 return Owner
->Custom600Headers();
1005 // AcquireStatus::pkgAcquireStatus - Constructor /*{{{*/
1006 // ---------------------------------------------------------------------
1008 pkgAcquireStatus::pkgAcquireStatus() : d(NULL
), Percent(-1), Update(true), MorePulses(false)
1013 // AcquireStatus::Pulse - Called periodically /*{{{*/
1014 // ---------------------------------------------------------------------
1015 /* This computes some internal state variables for the derived classes to
1016 use. It generates the current downloaded bytes and total bytes to download
1017 as well as the current CPS estimate. */
1018 bool pkgAcquireStatus::Pulse(pkgAcquire
*Owner
)
1025 // Compute the total number of bytes to fetch
1026 unsigned int Unknown
= 0;
1027 unsigned int Count
= 0;
1028 bool UnfetchedReleaseFiles
= false;
1029 for (pkgAcquire::ItemCIterator I
= Owner
->ItemsBegin();
1030 I
!= Owner
->ItemsEnd();
1034 if ((*I
)->Status
== pkgAcquire::Item::StatDone
)
1037 // Totally ignore local items
1038 if ((*I
)->Local
== true)
1041 // see if the method tells us to expect more
1042 TotalItems
+= (*I
)->ExpectedAdditionalItems
;
1044 // check if there are unfetched Release files
1045 if ((*I
)->Complete
== false && (*I
)->ExpectedAdditionalItems
> 0)
1046 UnfetchedReleaseFiles
= true;
1048 TotalBytes
+= (*I
)->FileSize
;
1049 if ((*I
)->Complete
== true)
1050 CurrentBytes
+= (*I
)->FileSize
;
1051 if ((*I
)->FileSize
== 0 && (*I
)->Complete
== false)
1055 // Compute the current completion
1056 unsigned long long ResumeSize
= 0;
1057 for (pkgAcquire::Worker
*I
= Owner
->WorkersBegin(); I
!= 0;
1058 I
= Owner
->WorkerStep(I
))
1060 if (I
->CurrentItem
!= 0 && I
->CurrentItem
->Owner
->Complete
== false)
1062 CurrentBytes
+= I
->CurrentSize
;
1063 ResumeSize
+= I
->ResumePoint
;
1065 // Files with unknown size always have 100% completion
1066 if (I
->CurrentItem
->Owner
->FileSize
== 0 &&
1067 I
->CurrentItem
->Owner
->Complete
== false)
1068 TotalBytes
+= I
->CurrentSize
;
1072 // Normalize the figures and account for unknown size downloads
1073 if (TotalBytes
<= 0)
1075 if (Unknown
== Count
)
1076 TotalBytes
= Unknown
;
1078 // Wha?! Is not supposed to happen.
1079 if (CurrentBytes
> TotalBytes
)
1080 CurrentBytes
= TotalBytes
;
1083 if (_config
->FindB("Debug::acquire::progress", false) == true)
1084 std::clog
<< " Bytes: "
1085 << SizeToStr(CurrentBytes
) << " / " << SizeToStr(TotalBytes
)
1089 struct timeval NewTime
;
1090 gettimeofday(&NewTime
,0);
1091 if ((NewTime
.tv_sec
- Time
.tv_sec
== 6 && NewTime
.tv_usec
> Time
.tv_usec
) ||
1092 NewTime
.tv_sec
- Time
.tv_sec
> 6)
1094 double Delta
= NewTime
.tv_sec
- Time
.tv_sec
+
1095 (NewTime
.tv_usec
- Time
.tv_usec
)/1000000.0;
1097 // Compute the CPS value
1101 CurrentCPS
= ((CurrentBytes
- ResumeSize
) - LastBytes
)/Delta
;
1102 LastBytes
= CurrentBytes
- ResumeSize
;
1103 ElapsedTime
= (unsigned long long)Delta
;
1107 double const OldPercent
= Percent
;
1108 // calculate the percentage, if we have too little data assume 1%
1109 if (TotalBytes
> 0 && UnfetchedReleaseFiles
)
1112 // use both files and bytes because bytes can be unreliable
1113 Percent
= (0.8 * (CurrentBytes
/float(TotalBytes
)*100.0) +
1114 0.2 * (CurrentItems
/float(TotalItems
)*100.0));
1115 double const DiffPercent
= Percent
- OldPercent
;
1116 if (DiffPercent
< 0.001 && _config
->FindB("Acquire::Progress::Diffpercent", false) == true)
1119 int fd
= _config
->FindI("APT::Status-Fd",-1);
1122 ostringstream status
;
1125 long i
= CurrentItems
< TotalItems
? CurrentItems
+ 1 : CurrentItems
;
1126 unsigned long long ETA
= 0;
1128 ETA
= (TotalBytes
- CurrentBytes
) / CurrentCPS
;
1130 // only show the ETA if it makes sense
1131 if (ETA
> 0 && ETA
< 172800 /* two days */ )
1132 snprintf(msg
,sizeof(msg
), _("Retrieving file %li of %li (%s remaining)"), i
, TotalItems
, TimeToStr(ETA
).c_str());
1134 snprintf(msg
,sizeof(msg
), _("Retrieving file %li of %li"), i
, TotalItems
);
1136 // build the status str
1137 status
<< "dlstatus:" << i
1138 << ":" << std::setprecision(3) << Percent
1142 std::string
const dlstatus
= status
.str();
1143 FileFd::Write(fd
, dlstatus
.c_str(), dlstatus
.size());
1149 // AcquireStatus::Start - Called when the download is started /*{{{*/
1150 // ---------------------------------------------------------------------
1151 /* We just reset the counters */
1152 void pkgAcquireStatus::Start()
1154 gettimeofday(&Time
,0);
1155 gettimeofday(&StartTime
,0);
1166 // AcquireStatus::Stop - Finished downloading /*{{{*/
1167 // ---------------------------------------------------------------------
1168 /* This accurately computes the elapsed time and the total overall CPS. */
1169 void pkgAcquireStatus::Stop()
1171 // Compute the CPS and elapsed time
1172 struct timeval NewTime
;
1173 gettimeofday(&NewTime
,0);
1175 double Delta
= NewTime
.tv_sec
- StartTime
.tv_sec
+
1176 (NewTime
.tv_usec
- StartTime
.tv_usec
)/1000000.0;
1178 // Compute the CPS value
1182 CurrentCPS
= FetchedBytes
/Delta
;
1183 LastBytes
= CurrentBytes
;
1184 ElapsedTime
= (unsigned long long)Delta
;
1187 // AcquireStatus::Fetched - Called when a byte set has been fetched /*{{{*/
1188 // ---------------------------------------------------------------------
1189 /* This is used to get accurate final transfer rate reporting. */
1190 void pkgAcquireStatus::Fetched(unsigned long long Size
,unsigned long long Resume
)
1192 FetchedBytes
+= Size
- Resume
;
1196 pkgAcquire::UriIterator::UriIterator(pkgAcquire::Queue
*Q
) : d(NULL
), CurQ(Q
), CurItem(0)
1198 while (CurItem
== 0 && CurQ
!= 0)
1200 CurItem
= CurQ
->Items
;
1205 APT_CONST
pkgAcquire::UriIterator::~UriIterator() {}
1206 APT_CONST
pkgAcquire::MethodConfig::~MethodConfig() {}
1207 APT_CONST
pkgAcquireStatus::~pkgAcquireStatus() {}