[apt.git] / apt-pkg / policy.cc

// -*- mode: cpp; mode: fold -*-
// Description								/*{{{*/
// $Id: policy.cc,v 1.2 2001/02/20 07:03:17 jgg Exp $
/* ######################################################################

   Package Version Policy implementation
   
   This is just a really simple wrapper around pkgVersionMatch with
   some added goodies to manage the list of things..
   
   Priority Table:
   
   1000 -> inf = Downgradeable priorities
   1000        = The 'no downgrade' pseduo-status file
   100 -> 1000 = Standard priorities
   990         = Config file override package files
   989         = Start for preference auto-priorities
   500         = Default package files
   100         = The status file
   0 -> 100    = NotAutomatic sources like experimental
   -inf -> 0   = Never selected   
   
   ##################################################################### */
									/*}}}*/
// Include Files							/*{{{*/
#ifdef __GNUG__
#pragma implementation "apt-pkg/policy.h"
#endif
#include <apt-pkg/policy.h>
#include <apt-pkg/configuration.h>
#include <apt-pkg/tagfile.h>
#include <apt-pkg/strutl.h>
#include <apt-pkg/error.h>
#include <apt-pkg/sptr.h>
    
#include <apti18n.h>
									/*}}}*/

// Policy::Init - Startup and bind to a cache				/*{{{*/
// ---------------------------------------------------------------------
/* Set the defaults for operation. The default mode with no loaded policy
   file matches the V0 policy engine. */
pkgPolicy::pkgPolicy(pkgCache *Owner) : Pins(0), PFPriority(0), Cache(Owner)
{
   PFPriority = new signed short[Owner->Head().PackageFileCount];
   Pins = new Pin[Owner->Head().PackageCount];

   for (unsigned long I = 0; I != Owner->Head().PackageCount; I++)
      Pins[I].Type = pkgVersionMatch::None;

   // The config file has a master override.
   string DefRel = _config->Find("APT::Default-Release");
   if (DefRel.empty() == false)
      CreatePin(pkgVersionMatch::Release,"",DefRel,990);
      
   InitDefaults();
}
									/*}}}*/
// Policy::InitDefaults - Compute the default selections		/*{{{*/
// ---------------------------------------------------------------------
/* */
bool pkgPolicy::InitDefaults()
{   
   // Initialize the priorities based on the status of the package file
   for (pkgCache::PkgFileIterator I = Cache->FileBegin(); I != Cache->FileEnd(); I++)
   {
      PFPriority[I->ID] = 500;
      if ((I->Flags & pkgCache::Flag::NotSource) == pkgCache::Flag::NotSource)
	 PFPriority[I->ID] = 100;
      else
	 if ((I->Flags & pkgCache::Flag::NotAutomatic) == pkgCache::Flag::NotAutomatic)
	    PFPriority[I->ID] = 1;
   }

   // Apply the defaults..
   SPtr<bool> Fixed = new bool[Cache->HeaderP->PackageFileCount];
   memset(Fixed,0,sizeof(*Fixed)*Cache->HeaderP->PackageFileCount);
   signed Cur = 989;
   StatusOverride = false;
   for (vector<Pin>::const_iterator I = Defaults.begin(); I != Defaults.end();
	I++, Cur--)
   {
      pkgVersionMatch Match(I->Data,I->Type);
      for (pkgCache::PkgFileIterator F = Cache->FileBegin(); F != Cache->FileEnd(); F++)
      {
/* hmm?
 if ((F->Flags & pkgCache::Flag::NotSource) == pkgCache::Flag::NotSource)
	    continue;*/

	 if (Match.FileMatch(F) == true && Fixed[F->ID] == false)
	 {
	    if (I->Priority != 0 && I->Priority > 0)
	       Cur = I->Priority;
	    
	    if (I->Priority < 0)
	       PFPriority[F->ID] =  I->Priority;
	    else
	       PFPriority[F->ID] = Cur;
	    
	    if (PFPriority[F->ID] > 1000)
	       StatusOverride = true;
	    
	    Fixed[F->ID] = true;
	 }      
      }      
   }

   if (_config->FindB("Debug::pkgPolicy",false) == true)
      for (pkgCache::PkgFileIterator F = Cache->FileBegin(); F != Cache->FileEnd(); F++)
	 cout << "Prio of " << F.FileName() << ' ' << PFPriority[F->ID] << endl; 
   
   return true;   
}
									/*}}}*/
// Policy::GetCandidateVer - Get the candidate install version		/*{{{*/
// ---------------------------------------------------------------------
/* Evaluate the package pins and the default list to deteremine what the
   best package is. */
pkgCache::VerIterator pkgPolicy::GetCandidateVer(pkgCache::PkgIterator Pkg)
{
   const Pin &PPkg = Pins[Pkg->ID];
   
   // Look for a package pin and evaluate it.
   signed Max = 0;
   pkgCache::VerIterator Pref(*Cache);
   if (PPkg.Type != pkgVersionMatch::None)
   {
      pkgVersionMatch Match(PPkg.Data,PPkg.Type);
      Pref = Match.Find(Pkg);
      Max = PPkg.Priority;
      if (PPkg.Priority == 0)
	 Max = 989;
   }
      
   /* Falling through to the default version.. Setting Max to zero
      effectively excludes everything <= 0 which are the non-automatic
      priorities.. The status file is given a prio of 100 which will exclude
      not-automatic sources, except in a single shot not-installed mode.
      The second pseduo-status file is at prio 1000, above which will permit
      the user to force-downgrade things.
      
      The user pin is subject to the same priority rules as default 
      selections. Thus there are two ways to create a pin - a pin that
      tracks the default when the default is taken away, and a permanent
      pin that stays at that setting.
    */
   for (pkgCache::VerIterator Ver = Pkg.VersionList(); Ver.end() == false; Ver++)
   {      
      for (pkgCache::VerFileIterator VF = Ver.FileList(); VF.end() == false; VF++)
      {
	 signed Prio = PFPriority[VF.File()->ID];
	 if (Prio > Max)
	 {
	    Pref = Ver;
	    Max = Prio;
	 }	 
      }      
      
      if (Pkg.CurrentVer() == Ver && Max < 1000)
      {
	 /* Elevate our current selection (or the status file itself)
	    to the Pseudo-status priority. */
	 if (Pref.end() == true)
	    Pref = Ver;
	 Max = 1000;
	 
	 // Fast path optimize.
	 if (StatusOverride == false)
	    break;
      }            
   }
   
   return Pref;
}
									/*}}}*/
// Policy::CreatePin - Create an entry in the pin table..		/*{{{*/
// ---------------------------------------------------------------------
/* For performance we have 3 tables, the default table, the main cache
   table (hashed to the cache). A blank package name indicates the pin
   belongs to the default table. Order of insertion matters here, the
   earlier defaults override later ones. */
void pkgPolicy::CreatePin(pkgVersionMatch::MatchType Type,string Name,
			  string Data,signed short Priority)
{
   pkgCache::PkgIterator Pkg = Cache->FindPkg(Name);
   Pin *P = 0;
   
   if (Name.empty() == true)
      P = Defaults.insert(Defaults.end());
   else
   {
      // Get a spot to put the pin
      if (Pkg.end() == true)
      {
	 // Check the unmatched table
	 for (vector<PkgPin>::iterator I = Unmatched.begin(); 
	      I != Unmatched.end() && P == 0; I++)
	    if (I->Pkg == Name)
	       P = I;
	 
	 if (P == 0)
	    P = Unmatched.insert(Unmatched.end());      
      }
      else
      {
	 P = Pins + Pkg->ID;
      }      
   }
   
   // Set..
   P->Type = Type;
   P->Priority = Priority;
   P->Data = Data;
}
									/*}}}*/

// ReadPinFile - Load the pin file into a Policy			/*{{{*/
// ---------------------------------------------------------------------
/* I'd like to see the preferences file store more than just pin information
   but right now that is the only stuff I have to store. Later there will
   have to be some kind of combined super parser to get the data into all
   the right classes.. */
bool ReadPinFile(pkgPolicy &Plcy,string File)
{
   if (File.empty() == true)
      File = _config->FindFile("Dir::Etc::Preferences");

   if (FileExists(File) == false)
      return true;
   
   FileFd Fd(File,FileFd::ReadOnly);
   pkgTagFile TF(&Fd);
   if (_error->PendingError() == true)
      return false;
   
   pkgTagSection Tags;
   while (TF.Step(Tags) == true)
   {
      string Name = Tags.FindS("Package");
      if (Name.empty() == true)
	 return _error->Error(_("Invalid record in the preferences file, no Package header"));
      if (Name == "*")
	 Name = string();
      
      const char *Start;
      const char *End;
      if (Tags.Find("Pin",Start,End) == false)
	 continue;
	 
      const char *Word = Start;
      for (; Word != End && isspace(*Word) == 0; Word++);

      // Parse the type..
      pkgVersionMatch::MatchType Type;
      if (stringcasecmp(Start,Word,"version") == 0 && Name.empty() == false)
	 Type = pkgVersionMatch::Version;
      else if (stringcasecmp(Start,Word,"release") == 0)
	 Type = pkgVersionMatch::Release;
      else if (stringcasecmp(Start,Word,"origin") == 0)
	 Type = pkgVersionMatch::Origin;
      else
      {
	 _error->Warning(_("Did not understand pin type %s"),string(Start,Word).c_str());
	 continue;
      }
      for (; Word != End && isspace(*Word) != 0; Word++);

      Plcy.CreatePin(Type,Name,string(Word,End),
		     Tags.FindI("Pin-Priority"));
   }

   Plcy.InitDefaults();
   return true;
}
									/*}}}*/
Commit	Line	Data
b2e465d6 AL	1	// -- mode: cpp; mode: fold --
	2	// Description /{{{/
	3	// $Id: policy.cc,v 1.2 2001/02/20 07:03:17 jgg Exp $
	4	/* ######################################################################
	5
	6	Package Version Policy implementation
	7
	8	This is just a really simple wrapper around pkgVersionMatch with
	9	some added goodies to manage the list of things..
	10
	11	Priority Table:
	12
	13	1000 -> inf = Downgradeable priorities
	14	1000 = The 'no downgrade' pseduo-status file
	15	100 -> 1000 = Standard priorities
	16	990 = Config file override package files
	17	989 = Start for preference auto-priorities
	18	500 = Default package files
	19	100 = The status file
	20	0 -> 100 = NotAutomatic sources like experimental
	21	-inf -> 0 = Never selected
	22
	23	##################################################################### */
	24	/}}}/
	25	// Include Files /{{{/
	26	#ifdef __GNUG__
	27	#pragma implementation "apt-pkg/policy.h"
	28	#endif
	29	#include <apt-pkg/policy.h>
	30	#include <apt-pkg/configuration.h>
	31	#include <apt-pkg/tagfile.h>
	32	#include <apt-pkg/strutl.h>
	33	#include <apt-pkg/error.h>
	34	#include <apt-pkg/sptr.h>
	35
	36	#include <apti18n.h>
	37	/}}}/
	38
	39	// Policy::Init - Startup and bind to a cache /{{{/
	40	// ---------------------------------------------------------------------
	41	/* Set the defaults for operation. The default mode with no loaded policy
	42	file matches the V0 policy engine. */
	43	pkgPolicy::pkgPolicy(pkgCache *Owner) : Pins(0), PFPriority(0), Cache(Owner)
	44	{
	45	PFPriority = new signed short[Owner->Head().PackageFileCount];
	46	Pins = new Pin[Owner->Head().PackageCount];
	47
	48	for (unsigned long I = 0; I != Owner->Head().PackageCount; I++)
	49	Pins[I].Type = pkgVersionMatch::None;
	50
	51	// The config file has a master override.
	52	string DefRel = _config->Find("APT::Default-Release");
	53	if (DefRel.empty() == false)
	54	CreatePin(pkgVersionMatch::Release,"",DefRel,990);
	55
	56	InitDefaults();
	57	}
	58	/}}}/
	59	// Policy::InitDefaults - Compute the default selections /{{{/
	60	// ---------------------------------------------------------------------
	61	/* */
	62	bool pkgPolicy::InitDefaults()
	63	{
	64	// Initialize the priorities based on the status of the package file
65	for (pkgCache::PkgFileIterator I = Cache->FileBegin(); I != Cache->FileEnd(); I++)
66	{
67	PFPriority[I->ID] = 500;
68	if ((I->Flags & pkgCache::Flag::NotSource) == pkgCache::Flag::NotSource)
69	PFPriority[I->ID] = 100;
70	else
71	if ((I->Flags & pkgCache::Flag::NotAutomatic) == pkgCache::Flag::NotAutomatic)
72	PFPriority[I->ID] = 1;
73	}
74
75	// Apply the defaults..
76	SPtr<bool> Fixed = new bool[Cache->HeaderP->PackageFileCount];
77	memset(Fixed,0,sizeof(Fixed)Cache->HeaderP->PackageFileCount);
78	signed Cur = 989;
79	StatusOverride = false;
80	for (vector<Pin>::const_iterator I = Defaults.begin(); I != Defaults.end();
81	I++, Cur--)
82	{
83	pkgVersionMatch Match(I->Data,I->Type);
84	for (pkgCache::PkgFileIterator F = Cache->FileBegin(); F != Cache->FileEnd(); F++)
85	{
86	/* hmm?
87	if ((F->Flags & pkgCache::Flag::NotSource) == pkgCache::Flag::NotSource)
88	continue;*/
89
90	if (Match.FileMatch(F) == true && Fixed[F->ID] == false)
91	{
92	if (I->Priority != 0 && I->Priority > 0)
93	Cur = I->Priority;
94
95	if (I->Priority < 0)
96	PFPriority[F->ID] = I->Priority;
97	else
98	PFPriority[F->ID] = Cur;
99
100	if (PFPriority[F->ID] > 1000)
101	StatusOverride = true;
102
103	Fixed[F->ID] = true;
104	}
105	}
106	}
107
108	if (_config->FindB("Debug::pkgPolicy",false) == true)
109	for (pkgCache::PkgFileIterator F = Cache->FileBegin(); F != Cache->FileEnd(); F++)
110	cout << "Prio of " << F.FileName() << ' ' << PFPriority[F->ID] << endl;
111
112	return true;
113	}
114	/}}}/
115	// Policy::GetCandidateVer - Get the candidate install version /{{{/
116	// ---------------------------------------------------------------------
117	/* Evaluate the package pins and the default list to deteremine what the
118	best package is. */
119	pkgCache::VerIterator pkgPolicy::GetCandidateVer(pkgCache::PkgIterator Pkg)
120	{
121	const Pin &PPkg = Pins[Pkg->ID];
122
123	// Look for a package pin and evaluate it.
124	signed Max = 0;
125	pkgCache::VerIterator Pref(*Cache);
126	if (PPkg.Type != pkgVersionMatch::None)
127	{
128	pkgVersionMatch Match(PPkg.Data,PPkg.Type);
129	Pref = Match.Find(Pkg);
130	Max = PPkg.Priority;
131	if (PPkg.Priority == 0)
132	Max = 989;
133	}
134
135	/* Falling through to the default version.. Setting Max to zero
136	effectively excludes everything <= 0 which are the non-automatic
137	priorities.. The status file is given a prio of 100 which will exclude
138	not-automatic sources, except in a single shot not-installed mode.
139	The second pseduo-status file is at prio 1000, above which will permit
140	the user to force-downgrade things.
141
142	The user pin is subject to the same priority rules as default
143	selections. Thus there are two ways to create a pin - a pin that
144	tracks the default when the default is taken away, and a permanent
145	pin that stays at that setting.
146	*/
147	for (pkgCache::VerIterator Ver = Pkg.VersionList(); Ver.end() == false; Ver++)
148	{
149	for (pkgCache::VerFileIterator VF = Ver.FileList(); VF.end() == false; VF++)
150	{
151	signed Prio = PFPriority[VF.File()->ID];
152	if (Prio > Max)
153	{
154	Pref = Ver;
155	Max = Prio;
156	}
157	}
158
159	if (Pkg.CurrentVer() == Ver && Max < 1000)
160	{
161	/* Elevate our current selection (or the status file itself)
162	to the Pseudo-status priority. */
163	if (Pref.end() == true)
164	Pref = Ver;
165	Max = 1000;
166
167	// Fast path optimize.
168	if (StatusOverride == false)
169	break;
170	}
171	}
172
173	return Pref;
174	}
175	/}}}/
176	// Policy::CreatePin - Create an entry in the pin table.. /{{{/
177	// ---------------------------------------------------------------------
178	/* For performance we have 3 tables, the default table, the main cache
179	table (hashed to the cache). A blank package name indicates the pin
180	belongs to the default table. Order of insertion matters here, the
181	earlier defaults override later ones. */
182	void pkgPolicy::CreatePin(pkgVersionMatch::MatchType Type,string Name,
183	string Data,signed short Priority)
184	{
185	pkgCache::PkgIterator Pkg = Cache->FindPkg(Name);
186	Pin *P = 0;
187
188	if (Name.empty() == true)
189	P = Defaults.insert(Defaults.end());
190	else
191	{
192	// Get a spot to put the pin
193	if (Pkg.end() == true)
194	{
195	// Check the unmatched table
196	for (vector<PkgPin>::iterator I = Unmatched.begin();
197	I != Unmatched.end() && P == 0; I++)
198	if (I->Pkg == Name)
199	P = I;
200
201	if (P == 0)
202	P = Unmatched.insert(Unmatched.end());
203	}
204	else
205	{
206	P = Pins + Pkg->ID;
207	}
208	}
209
210	// Set..
211	P->Type = Type;
212	P->Priority = Priority;
213	P->Data = Data;
214	}
215	/}}}/
216
217	// ReadPinFile - Load the pin file into a Policy /{{{/
218	// ---------------------------------------------------------------------
219	/* I'd like to see the preferences file store more than just pin information
220	but right now that is the only stuff I have to store. Later there will
221	have to be some kind of combined super parser to get the data into all
222	the right classes.. */
223	bool ReadPinFile(pkgPolicy &Plcy,string File)
224	{
225	if (File.empty() == true)
226	File = _config->FindFile("Dir::Etc::Preferences");
227
228	if (FileExists(File) == false)
229	return true;
230
231	FileFd Fd(File,FileFd::ReadOnly);
232	pkgTagFile TF(&Fd);
233	if (_error->PendingError() == true)
234	return false;
235
236	pkgTagSection Tags;
237	while (TF.Step(Tags) == true)
238	{
239	string Name = Tags.FindS("Package");
240	if (Name.empty() == true)
241	return _error->Error(_("Invalid record in the preferences file, no Package header"));
242	if (Name == "*")
243	Name = string();
244
245	const char *Start;
246	const char *End;
247	if (Tags.Find("Pin",Start,End) == false)
248	continue;
249
250	const char *Word = Start;
251	for (; Word != End && isspace(*Word) == 0; Word++);
252
253	// Parse the type..
254	pkgVersionMatch::MatchType Type;
255	if (stringcasecmp(Start,Word,"version") == 0 && Name.empty() == false)
256	Type = pkgVersionMatch::Version;
257	else if (stringcasecmp(Start,Word,"release") == 0)
258	Type = pkgVersionMatch::Release;
259	else if (stringcasecmp(Start,Word,"origin") == 0)
260	Type = pkgVersionMatch::Origin;
261	else
262	{
263	_error->Warning(_("Did not understand pin type %s"),string(Start,Word).c_str());
264	continue;
265	}
266	for (; Word != End && isspace(*Word) != 0; Word++);
267
268	Plcy.CreatePin(Type,Name,string(Word,End),
269	Tags.FindI("Pin-Priority"));
270	}
271
272	Plcy.InitDefaults();
273	return true;
274	}
275	/}}}/