xref: /haiku/src/kits/package/hpkg/ReaderImplBase.cpp (revision 6a89a36aa0ce6a53f9758ba767863ac6b4c6156c) 
1 /*
2  * Copyright 2009-2014, Ingo Weinhold, ingo_weinhold@gmx.de.
3  * Copyright 2011, Oliver Tappe <zooey@hirschkaefer.de>
4  * Distributed under the terms of the MIT License.
5  */
6 
7 
8 #include <package/hpkg/ReaderImplBase.h>
9 
10 #include <errno.h>
11 #include <stdio.h>
12 #include <stdlib.h>
13 #include <string.h>
14 #include <unistd.h>
15 
16 #include <algorithm>
17 #include <new>
18 
19 #include <ByteOrder.h>
20 #include <DataIO.h>
21 
22 #include <package/hpkg/HPKGDefsPrivate.h>
23 
24 #include <package/hpkg/PackageFileHeapReader.h>
25 #include <ZlibDecompressor.h>
26 
27 
28 namespace BPackageKit {
29 
30 namespace BHPKG {
31 
32 namespace BPrivate {
33 
34 
35 static const size_t kScratchBufferSize = 64 * 1024;
36 
37 static const uint16 kAttributeTypes[B_HPKG_ATTRIBUTE_ID_ENUM_COUNT] = {
38 	#define B_DEFINE_HPKG_ATTRIBUTE(id, type, name, constant)	\
39 		B_HPKG_ATTRIBUTE_TYPE_##type,
40 	#include <package/hpkg/PackageAttributes.h>
41 	#undef B_DEFINE_HPKG_ATTRIBUTE
42 };
43 
44 // #pragma mark - AttributeHandlerContext
45 
46 
47 ReaderImplBase::AttributeHandlerContext::AttributeHandlerContext(
48 	BErrorOutput* errorOutput, BPackageContentHandler* packageContentHandler,
49 	BHPKGPackageSectionID section, bool ignoreUnknownAttributes)
50 	:
51 	errorOutput(errorOutput),
52 	packageContentHandler(packageContentHandler),
53 	hasLowLevelHandler(false),
54 	ignoreUnknownAttributes(ignoreUnknownAttributes),
55 	section(section)
56 {
57 }
58 
59 
60 ReaderImplBase::AttributeHandlerContext::AttributeHandlerContext(
61 	BErrorOutput* errorOutput, BLowLevelPackageContentHandler* lowLevelHandler,
62 	BHPKGPackageSectionID section, bool ignoreUnknownAttributes)
63 	:
64 	errorOutput(errorOutput),
65 	lowLevelHandler(lowLevelHandler),
66 	hasLowLevelHandler(true),
67 	ignoreUnknownAttributes(ignoreUnknownAttributes),
68 	section(section)
69 {
70 }
71 
72 
73 void
74 ReaderImplBase::AttributeHandlerContext::ErrorOccurred()
75 {
76 	if (hasLowLevelHandler)
77 		lowLevelHandler->HandleErrorOccurred();
78 	else
79 		packageContentHandler->HandleErrorOccurred();
80 }
81 
82 
83 // #pragma mark - AttributeHandler
84 
85 
86 ReaderImplBase::AttributeHandler::~AttributeHandler()
87 {
88 }
89 
90 
91 void
92 ReaderImplBase::AttributeHandler::SetLevel(int level)
93 {
94 	fLevel = level;
95 }
96 
97 
98 status_t
99 ReaderImplBase::AttributeHandler::HandleAttribute(
100 	AttributeHandlerContext* context, uint8 id, const AttributeValue& value,
101 	AttributeHandler** _handler)
102 {
103 	return B_OK;
104 }
105 
106 
107 status_t
108 ReaderImplBase::AttributeHandler::NotifyDone(
109 	AttributeHandlerContext* context)
110 {
111 	return B_OK;
112 }
113 
114 
115 status_t
116 ReaderImplBase::AttributeHandler::Delete(AttributeHandlerContext* context)
117 {
118 	delete this;
119 	return B_OK;
120 }
121 
122 
123 // #pragma mark - PackageInfoAttributeHandlerBase
124 
125 
126 ReaderImplBase::PackageInfoAttributeHandlerBase
127 	::PackageInfoAttributeHandlerBase(
128 		BPackageInfoAttributeValue& packageInfoValue)
129 	:
130 	fPackageInfoValue(packageInfoValue)
131 {
132 }
133 
134 
135 status_t
136 ReaderImplBase::PackageInfoAttributeHandlerBase::NotifyDone(
137 	AttributeHandlerContext* context)
138 {
139 	status_t error = context->packageContentHandler->HandlePackageAttribute(
140 		fPackageInfoValue);
141 	fPackageInfoValue.Clear();
142 	return error;
143 }
144 
145 
146 // #pragma mark - PackageVersionAttributeHandler
147 
148 
149 ReaderImplBase::PackageVersionAttributeHandler::PackageVersionAttributeHandler(
150 	BPackageInfoAttributeValue& packageInfoValue,
151 	BPackageVersionData& versionData, bool notify)
152 	:
153 	super(packageInfoValue),
154 	fPackageVersionData(versionData),
155 	fNotify(notify)
156 {
157 }
158 
159 
160 status_t
161 ReaderImplBase::PackageVersionAttributeHandler::HandleAttribute(
162 	AttributeHandlerContext* context, uint8 id, const AttributeValue& value,
163 	AttributeHandler** _handler)
164 {
165 	switch (id) {
166 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_VERSION_MINOR:
167 			fPackageVersionData.minor = value.string;
168 			break;
169 
170 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_VERSION_MICRO:
171 			fPackageVersionData.micro = value.string;
172 			break;
173 
174 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_VERSION_PRE_RELEASE:
175 			fPackageVersionData.preRelease = value.string;
176 			break;
177 
178 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_VERSION_REVISION:
179 			fPackageVersionData.revision = value.unsignedInt;
180 			break;
181 
182 		default:
183 			if (context->ignoreUnknownAttributes)
184 				break;
185 
186 			context->errorOutput->PrintError("Error: Invalid package "
187 				"attribute section: unexpected package attribute id %d "
188 				"encountered when parsing package version\n", id);
189 			return B_BAD_DATA;
190 	}
191 
192 	return B_OK;
193 }
194 
195 
196 status_t
197 ReaderImplBase::PackageVersionAttributeHandler::NotifyDone(
198 	AttributeHandlerContext* context)
199 {
200 	if (!fNotify)
201 		return B_OK;
202 
203 	fPackageInfoValue.attributeID = B_PACKAGE_INFO_VERSION;
204 	return super::NotifyDone(context);
205 }
206 
207 
208 // #pragma mark - PackageResolvableAttributeHandler
209 
210 
211 ReaderImplBase::PackageResolvableAttributeHandler
212 	::PackageResolvableAttributeHandler(
213 		BPackageInfoAttributeValue& packageInfoValue)
214 	:
215 	super(packageInfoValue)
216 {
217 }
218 
219 
220 status_t
221 ReaderImplBase::PackageResolvableAttributeHandler::HandleAttribute(
222 	AttributeHandlerContext* context, uint8 id, const AttributeValue& value,
223 	AttributeHandler** _handler)
224 {
225 	switch (id) {
226 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_VERSION_MAJOR:
227 			fPackageInfoValue.resolvable.haveVersion = true;
228 			fPackageInfoValue.resolvable.version.major = value.string;
229 			if (_handler != NULL) {
230 				*_handler
231 					= new(std::nothrow) PackageVersionAttributeHandler(
232 						fPackageInfoValue,
233 						fPackageInfoValue.resolvable.version, false);
234 				if (*_handler == NULL)
235 					return B_NO_MEMORY;
236 			}
237 			break;
238 
239 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_PROVIDES_COMPATIBLE:
240 			fPackageInfoValue.resolvable.haveCompatibleVersion = true;
241 			fPackageInfoValue.resolvable.compatibleVersion.major = value.string;
242 			if (_handler != NULL) {
243 				*_handler
244 					= new(std::nothrow) PackageVersionAttributeHandler(
245 						fPackageInfoValue,
246 						fPackageInfoValue.resolvable.compatibleVersion, false);
247 				if (*_handler == NULL)
248 					return B_NO_MEMORY;
249 			}
250 			break;
251 
252 		default:
253 			if (context->ignoreUnknownAttributes)
254 				break;
255 
256 			context->errorOutput->PrintError("Error: Invalid package "
257 				"attribute section: unexpected package attribute id %d "
258 				"encountered when parsing package resolvable\n", id);
259 			return B_BAD_DATA;
260 	}
261 
262 	return B_OK;
263 }
264 
265 
266 // #pragma mark - PackageResolvableExpressionAttributeHandler
267 
268 
269 ReaderImplBase::PackageResolvableExpressionAttributeHandler
270 	::PackageResolvableExpressionAttributeHandler(
271 		BPackageInfoAttributeValue& packageInfoValue)
272 	:
273 	super(packageInfoValue)
274 {
275 }
276 
277 
278 status_t
279 ReaderImplBase::PackageResolvableExpressionAttributeHandler::HandleAttribute(
280 	AttributeHandlerContext* context, uint8 id, const AttributeValue& value,
281 	AttributeHandler** _handler)
282 {
283 	switch (id) {
284 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_RESOLVABLE_OPERATOR:
285 			if (value.unsignedInt >= B_PACKAGE_RESOLVABLE_OP_ENUM_COUNT) {
286 				context->errorOutput->PrintError(
287 					"Error: Invalid package attribute section: invalid "
288 					"package resolvable operator %lld encountered\n",
289 					value.unsignedInt);
290 				return B_BAD_DATA;
291 			}
292 			fPackageInfoValue.resolvableExpression.op
293 				= (BPackageResolvableOperator)value.unsignedInt;
294 			break;
295 
296 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_VERSION_MAJOR:
297 			fPackageInfoValue.resolvableExpression.haveOpAndVersion = true;
298 			fPackageInfoValue.resolvableExpression.version.major
299 				= value.string;
300 			if (_handler != NULL) {
301 				*_handler
302 					= new(std::nothrow) PackageVersionAttributeHandler(
303 						fPackageInfoValue,
304 						fPackageInfoValue.resolvableExpression.version,
305 						false);
306 				if (*_handler == NULL)
307 					return B_NO_MEMORY;
308 			}
309 			return B_OK;
310 
311 		default:
312 			if (context->ignoreUnknownAttributes)
313 				break;
314 
315 			context->errorOutput->PrintError("Error: Invalid package "
316 				"attribute section: unexpected package attribute id %d "
317 				"encountered when parsing package resolvable-expression\n",
318 				id);
319 			return B_BAD_DATA;
320 	}
321 
322 	return B_OK;
323 }
324 
325 
326 // #pragma mark - GlobalWritableFileInfoAttributeHandler
327 
328 
329 ReaderImplBase::GlobalWritableFileInfoAttributeHandler
330 	::GlobalWritableFileInfoAttributeHandler(
331 		BPackageInfoAttributeValue& packageInfoValue)
332 	:
333 	super(packageInfoValue)
334 {
335 }
336 
337 
338 status_t
339 ReaderImplBase::GlobalWritableFileInfoAttributeHandler::HandleAttribute(
340 	AttributeHandlerContext* context, uint8 id, const AttributeValue& value,
341 	AttributeHandler** _handler)
342 {
343 	switch (id) {
344 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_WRITABLE_FILE_UPDATE_TYPE:
345 			if (value.unsignedInt >= B_WRITABLE_FILE_UPDATE_TYPE_ENUM_COUNT) {
346 				context->errorOutput->PrintError(
347 					"Error: Invalid package attribute section: invalid "
348 					"global settings file update type %" B_PRIu64
349 					" encountered\n", value.unsignedInt);
350 				return B_BAD_DATA;
351 			}
352 			fPackageInfoValue.globalWritableFileInfo.updateType
353 				= (BWritableFileUpdateType)value.unsignedInt;
354 			break;
355 
356 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_IS_WRITABLE_DIRECTORY:
357 			fPackageInfoValue.globalWritableFileInfo.isDirectory
358 				= value.unsignedInt != 0;
359 			break;
360 
361 		default:
362 			if (context->ignoreUnknownAttributes)
363 				break;
364 
365 			context->errorOutput->PrintError("Error: Invalid package "
366 				"attribute section: unexpected package attribute id %d "
367 				"encountered when parsing global settings file info\n",
368 				id);
369 			return B_BAD_DATA;
370 	}
371 
372 	return B_OK;
373 }
374 
375 
376 // #pragma mark - UserSettingsFileInfoAttributeHandler
377 
378 
379 ReaderImplBase::UserSettingsFileInfoAttributeHandler
380 	::UserSettingsFileInfoAttributeHandler(
381 		BPackageInfoAttributeValue& packageInfoValue)
382 	:
383 	super(packageInfoValue)
384 {
385 }
386 
387 
388 status_t
389 ReaderImplBase::UserSettingsFileInfoAttributeHandler::HandleAttribute(
390 	AttributeHandlerContext* context, uint8 id, const AttributeValue& value,
391 	AttributeHandler** _handler)
392 {
393 	switch (id) {
394 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_SETTINGS_FILE_TEMPLATE:
395 			fPackageInfoValue.userSettingsFileInfo.templatePath = value.string;
396 			break;
397 
398 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_IS_WRITABLE_DIRECTORY:
399 			fPackageInfoValue.userSettingsFileInfo.isDirectory
400 				= value.unsignedInt != 0;
401 			break;
402 
403 		default:
404 			if (context->ignoreUnknownAttributes)
405 				break;
406 
407 			context->errorOutput->PrintError("Error: Invalid package "
408 				"attribute section: unexpected package attribute id %d "
409 				"encountered when parsing user settings file info\n",
410 				id);
411 			return B_BAD_DATA;
412 	}
413 
414 	return B_OK;
415 }
416 
417 
418 // #pragma mark - UserAttributeHandler
419 
420 
421 ReaderImplBase::UserAttributeHandler::UserAttributeHandler(
422 		BPackageInfoAttributeValue& packageInfoValue)
423 	:
424 	super(packageInfoValue),
425 	fGroups()
426 {
427 }
428 
429 
430 status_t
431 ReaderImplBase::UserAttributeHandler::HandleAttribute(
432 	AttributeHandlerContext* context, uint8 id, const AttributeValue& value,
433 	AttributeHandler** _handler)
434 {
435 	switch (id) {
436 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_USER_REAL_NAME:
437 			fPackageInfoValue.user.realName = value.string;
438 			break;
439 
440 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_USER_HOME:
441 			fPackageInfoValue.user.home = value.string;
442 			break;
443 
444 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_USER_SHELL:
445 			fPackageInfoValue.user.shell = value.string;
446 			break;
447 
448 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_USER_GROUP:
449 			if (!fGroups.Add(value.string))
450 				return B_NO_MEMORY;
451 			break;
452 
453 		default:
454 			if (context->ignoreUnknownAttributes)
455 				break;
456 
457 			context->errorOutput->PrintError("Error: Invalid package "
458 				"attribute section: unexpected package attribute id %d "
459 				"encountered when parsing user settings file info\n",
460 				id);
461 			return B_BAD_DATA;
462 	}
463 
464 	return B_OK;
465 }
466 
467 
468 status_t
469 ReaderImplBase::UserAttributeHandler::NotifyDone(
470 	AttributeHandlerContext* context)
471 {
472 	if (!fGroups.IsEmpty()) {
473 		fPackageInfoValue.user.groups = fGroups.Elements();
474 		fPackageInfoValue.user.groupCount = fGroups.Count();
475 	}
476 
477 	return super::NotifyDone(context);
478 }
479 
480 
481 // #pragma mark - PackageAttributeHandler
482 
483 
484 status_t
485 ReaderImplBase::PackageAttributeHandler::HandleAttribute(
486 	AttributeHandlerContext* context, uint8 id, const AttributeValue& value,
487 	AttributeHandler** _handler)
488 {
489 	switch (id) {
490 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_NAME:
491 			fPackageInfoValue.SetTo(B_PACKAGE_INFO_NAME, value.string);
492 			break;
493 
494 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_SUMMARY:
495 			fPackageInfoValue.SetTo(B_PACKAGE_INFO_SUMMARY, value.string);
496 			break;
497 
498 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_DESCRIPTION:
499 			fPackageInfoValue.SetTo(B_PACKAGE_INFO_DESCRIPTION,
500 				value.string);
501 			break;
502 
503 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_VENDOR:
504 			fPackageInfoValue.SetTo(B_PACKAGE_INFO_VENDOR, value.string);
505 			break;
506 
507 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_PACKAGER:
508 			fPackageInfoValue.SetTo(B_PACKAGE_INFO_PACKAGER, value.string);
509 			break;
510 
511 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_BASE_PACKAGE:
512 			fPackageInfoValue.SetTo(B_PACKAGE_INFO_BASE_PACKAGE, value.string);
513 			break;
514 
515 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_FLAGS:
516 			fPackageInfoValue.SetTo(B_PACKAGE_INFO_FLAGS,
517 				(uint32)value.unsignedInt);
518 			break;
519 
520 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_ARCHITECTURE:
521 			if (value.unsignedInt
522 					>= B_PACKAGE_ARCHITECTURE_ENUM_COUNT) {
523 				context->errorOutput->PrintError(
524 					"Error: Invalid package attribute section: "
525 					"Invalid package architecture %lld encountered\n",
526 					value.unsignedInt);
527 				return B_BAD_DATA;
528 			}
529 			fPackageInfoValue.SetTo(B_PACKAGE_INFO_ARCHITECTURE,
530 				(uint8)value.unsignedInt);
531 			break;
532 
533 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_VERSION_MAJOR:
534 			fPackageInfoValue.attributeID = B_PACKAGE_INFO_VERSION;
535 			fPackageInfoValue.version.major = value.string;
536 			if (_handler != NULL) {
537 				*_handler
538 					= new(std::nothrow) PackageVersionAttributeHandler(
539 						fPackageInfoValue, fPackageInfoValue.version, true);
540 				if (*_handler == NULL)
541 					return B_NO_MEMORY;
542 			}
543 			break;
544 
545 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_COPYRIGHT:
546 			fPackageInfoValue.SetTo(B_PACKAGE_INFO_COPYRIGHTS,
547 				value.string);
548 			break;
549 
550 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_LICENSE:
551 			fPackageInfoValue.SetTo(B_PACKAGE_INFO_LICENSES,
552 				value.string);
553 			break;
554 
555 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_URL:
556 			fPackageInfoValue.SetTo(B_PACKAGE_INFO_URLS, value.string);
557 			break;
558 
559 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_SOURCE_URL:
560 			fPackageInfoValue.SetTo(B_PACKAGE_INFO_SOURCE_URLS, value.string);
561 			break;
562 
563 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_PROVIDES:
564 			fPackageInfoValue.resolvable.name = value.string;
565 			fPackageInfoValue.attributeID = B_PACKAGE_INFO_PROVIDES;
566 			if (_handler != NULL) {
567 				*_handler
568 					= new(std::nothrow) PackageResolvableAttributeHandler(
569 						fPackageInfoValue);
570 				if (*_handler == NULL)
571 					return B_NO_MEMORY;
572 			}
573 			break;
574 
575 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_REQUIRES:
576 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_SUPPLEMENTS:
577 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_CONFLICTS:
578 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_FRESHENS:
579 			fPackageInfoValue.resolvableExpression.name = value.string;
580 			switch (id) {
581 				case B_HPKG_ATTRIBUTE_ID_PACKAGE_REQUIRES:
582 					fPackageInfoValue.attributeID = B_PACKAGE_INFO_REQUIRES;
583 					break;
584 
585 				case B_HPKG_ATTRIBUTE_ID_PACKAGE_SUPPLEMENTS:
586 					fPackageInfoValue.attributeID
587 						= B_PACKAGE_INFO_SUPPLEMENTS;
588 					break;
589 
590 				case B_HPKG_ATTRIBUTE_ID_PACKAGE_CONFLICTS:
591 					fPackageInfoValue.attributeID
592 						= B_PACKAGE_INFO_CONFLICTS;
593 					break;
594 
595 				case B_HPKG_ATTRIBUTE_ID_PACKAGE_FRESHENS:
596 					fPackageInfoValue.attributeID = B_PACKAGE_INFO_FRESHENS;
597 					break;
598 			}
599 			if (_handler != NULL) {
600 				*_handler = new(std::nothrow)
601 					PackageResolvableExpressionAttributeHandler(
602 						fPackageInfoValue);
603 				if (*_handler == NULL)
604 					return B_NO_MEMORY;
605 			}
606 			break;
607 
608 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_REPLACES:
609 			fPackageInfoValue.SetTo(B_PACKAGE_INFO_REPLACES, value.string);
610 			break;
611 
612 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_CHECKSUM:
613 			fPackageInfoValue.SetTo(B_PACKAGE_INFO_CHECKSUM, value.string);
614 			break;
615 
616 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_INSTALL_PATH:
617 			fPackageInfoValue.SetTo(B_PACKAGE_INFO_INSTALL_PATH, value.string);
618 			break;
619 
620 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_GLOBAL_WRITABLE_FILE:
621 			fPackageInfoValue.globalWritableFileInfo.path = value.string;
622 			fPackageInfoValue.globalWritableFileInfo.updateType
623 				= B_WRITABLE_FILE_UPDATE_TYPE_ENUM_COUNT;
624 			fPackageInfoValue.attributeID
625 				= B_PACKAGE_INFO_GLOBAL_WRITABLE_FILES;
626 			if (_handler != NULL) {
627 				*_handler
628 					= new(std::nothrow) GlobalWritableFileInfoAttributeHandler(
629 						fPackageInfoValue);
630 				if (*_handler == NULL)
631 					return B_NO_MEMORY;
632 			}
633 			break;
634 
635 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_USER_SETTINGS_FILE:
636 			fPackageInfoValue.userSettingsFileInfo.path = value.string;
637 			fPackageInfoValue.attributeID
638 				= B_PACKAGE_INFO_USER_SETTINGS_FILES;
639 			if (_handler != NULL) {
640 				*_handler
641 					= new(std::nothrow) UserSettingsFileInfoAttributeHandler(
642 						fPackageInfoValue);
643 				if (*_handler == NULL)
644 					return B_NO_MEMORY;
645 			}
646 			break;
647 
648 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_USER:
649 			fPackageInfoValue.user.name = value.string;
650 			fPackageInfoValue.attributeID = B_PACKAGE_INFO_USERS;
651 			if (_handler != NULL) {
652 				*_handler = new(std::nothrow) UserAttributeHandler(
653 					fPackageInfoValue);
654 				if (*_handler == NULL)
655 					return B_NO_MEMORY;
656 			}
657 			break;
658 
659 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_GROUP:
660 			fPackageInfoValue.SetTo(B_PACKAGE_INFO_GROUPS, value.string);
661 			break;
662 
663 		case B_HPKG_ATTRIBUTE_ID_PACKAGE_POST_INSTALL_SCRIPT:
664 			fPackageInfoValue.SetTo(B_PACKAGE_INFO_POST_INSTALL_SCRIPTS,
665 				value.string);
666 			break;
667 
668 		default:
669 			if (context->ignoreUnknownAttributes)
670 				break;
671 
672 			context->errorOutput->PrintError(
673 				"Error: Invalid package attribute section: unexpected "
674 				"package attribute id %d encountered\n", id);
675 			return B_BAD_DATA;
676 	}
677 
678 	// notify unless the current attribute has children, in which case
679 	// the child-handler will notify when it's done
680 	if (_handler == NULL) {
681 		status_t error = context->packageContentHandler
682 			->HandlePackageAttribute(fPackageInfoValue);
683 		fPackageInfoValue.Clear();
684 		if (error != B_OK)
685 			return error;
686 	}
687 
688 	return B_OK;
689 }
690 
691 
692 // #pragma mark - LowLevelAttributeHandler
693 
694 
695 ReaderImplBase::LowLevelAttributeHandler::LowLevelAttributeHandler()
696 	:
697 	fParentToken(NULL),
698 	fToken(NULL),
699 	fID(B_HPKG_ATTRIBUTE_ID_ENUM_COUNT)
700 {
701 }
702 
703 
704 ReaderImplBase::LowLevelAttributeHandler::LowLevelAttributeHandler(uint8 id,
705 	const BPackageAttributeValue& value, void* parentToken, void* token)
706 	:
707 	fParentToken(NULL),
708 	fToken(token),
709 	fID(id),
710 	fValue(value)
711 {
712 }
713 
714 
715 status_t
716 ReaderImplBase::LowLevelAttributeHandler::HandleAttribute(
717 	AttributeHandlerContext* context, uint8 id, const AttributeValue& value,
718 	AttributeHandler** _handler)
719 {
720 	// notify the content handler
721 	void* token;
722 	status_t error = context->lowLevelHandler->HandleAttribute(
723 		(BHPKGAttributeID)id, value, fToken, token);
724 	if (error != B_OK)
725 		return error;
726 
727 	// create a subhandler for the attribute, if it has children
728 	if (_handler != NULL) {
729 		*_handler = new(std::nothrow) LowLevelAttributeHandler(id, value,
730 			fToken, token);
731 		if (*_handler == NULL) {
732 			context->lowLevelHandler->HandleAttributeDone((BHPKGAttributeID)id,
733 				value, fToken, token);
734 			return B_NO_MEMORY;
735 		}
736 		return B_OK;
737 	}
738 
739 	// no children -- just call the done hook
740 	return context->lowLevelHandler->HandleAttributeDone((BHPKGAttributeID)id,
741 		value, fToken, token);
742 }
743 
744 
745 status_t
746 ReaderImplBase::LowLevelAttributeHandler::NotifyDone(
747 	AttributeHandlerContext* context)
748 {
749 	if (fID != B_HPKG_ATTRIBUTE_ID_ENUM_COUNT) {
750 		status_t error = context->lowLevelHandler->HandleAttributeDone(
751 			(BHPKGAttributeID)fID, fValue, fParentToken, fToken);
752 		if (error != B_OK)
753 			return error;
754 	}
755 	return super::NotifyDone(context);
756 }
757 
758 
759 // #pragma mark - ReaderImplBase
760 
761 
762 ReaderImplBase::ReaderImplBase(const char* fileType, BErrorOutput* errorOutput)
763 	:
764 	fPackageAttributesSection("package attributes"),
765 	fFileType(fileType),
766 	fErrorOutput(errorOutput),
767 	fFD(-1),
768 	fOwnsFD(false),
769 	fRawHeapReader(NULL),
770 	fHeapReader(NULL),
771 	fCurrentSection(NULL)
772 {
773 }
774 
775 
776 ReaderImplBase::~ReaderImplBase()
777 {
778 	delete fHeapReader;
779 	if (fRawHeapReader != fHeapReader)
780 		delete fRawHeapReader;
781 
782 	if (fOwnsFD && fFD >= 0)
783 		close(fFD);
784 }
785 
786 
787 uint64
788 ReaderImplBase::UncompressedHeapSize() const
789 {
790 	return fRawHeapReader->UncompressedHeapSize();
791 }
792 
793 
794 BAbstractBufferedDataReader*
795 ReaderImplBase::DetachHeapReader(PackageFileHeapReader** _rawHeapReader)
796 {
797 	BAbstractBufferedDataReader* heapReader = fHeapReader;
798 	fHeapReader = NULL;
799 
800 	if (_rawHeapReader != NULL)
801 		*_rawHeapReader = fRawHeapReader;
802 	fRawHeapReader = NULL;
803 
804 	return heapReader;
805 }
806 
807 
808 status_t
809 ReaderImplBase::InitHeapReader(uint32 compression, uint32 chunkSize,
810 	off_t offset, uint64 compressedSize, uint64 uncompressedSize)
811 {
812 	if (compression != B_HPKG_COMPRESSION_ZLIB) {
813 		fErrorOutput->PrintError("Error: Invalid heap compression\n");
814 		return B_BAD_DATA;
815 	}
816 
817 	fRawHeapReader = new(std::nothrow) PackageFileHeapReader(fErrorOutput, fFD,
818 		offset, compressedSize, uncompressedSize);
819 	if (fRawHeapReader == NULL)
820 		return B_NO_MEMORY;
821 
822 	status_t error = fRawHeapReader->Init();
823 	if (error != B_OK)
824 		return error;
825 
826 	error = CreateCachedHeapReader(fRawHeapReader, fHeapReader);
827 	if (error != B_OK) {
828 		if (error != B_NOT_SUPPORTED)
829 			return error;
830 
831 		fHeapReader = fRawHeapReader;
832 	}
833 
834 	return B_OK;
835 }
836 
837 
838 status_t
839 ReaderImplBase::CreateCachedHeapReader(PackageFileHeapReader* heapReader,
840 	BAbstractBufferedDataReader*& _cachedReader)
841 {
842 	return B_NOT_SUPPORTED;
843 }
844 
845 
846 status_t
847 ReaderImplBase::InitSection(PackageFileSection& section, uint64 endOffset,
848 	uint64 length, uint64 maxSaneLength, uint64 stringsLength,
849 	uint64 stringsCount)
850 {
851 	// check length vs. endOffset
852 	if (length > endOffset) {
853 		ErrorOutput()->PrintError("Error: %s file %s section size is %"
854 			B_PRIu64 " bytes. This is greater than the available space\n",
855 			fFileType, section.name, length);
856 		return B_BAD_DATA;
857 	}
858 
859 	// check sanity length
860 	if (maxSaneLength > 0 && length > maxSaneLength) {
861 		ErrorOutput()->PrintError("Error: %s file %s section size is %"
862 			B_PRIu64 " bytes. This is beyond the reader's sanity limit\n",
863 			fFileType, section.name, length);
864 		return B_NOT_SUPPORTED;
865 	}
866 
867 	// check strings subsection size/count
868 	if ((stringsLength <= 1) != (stringsCount == 0) || stringsLength > length) {
869 		ErrorOutput()->PrintError("Error: strings subsection description of %s "
870 			"file %s section is invalid (%" B_PRIu64 " strings, length: %"
871 			B_PRIu64 ", section length: %" B_PRIu64 ")\n",
872 			fFileType, section.name, stringsCount, stringsLength, length);
873 		return B_BAD_DATA;
874 	}
875 
876 	section.uncompressedLength = length;
877 	section.offset = endOffset - length;
878 	section.currentOffset = 0;
879 	section.stringsLength = stringsLength;
880 	section.stringsCount = stringsCount;
881 
882 	return B_OK;
883 }
884 
885 
886 status_t
887 ReaderImplBase::PrepareSection(PackageFileSection& section)
888 {
889 	// allocate memory for the section data and read it in
890 	section.data = new(std::nothrow) uint8[section.uncompressedLength];
891 	if (section.data == NULL) {
892 		ErrorOutput()->PrintError("Error: Out of memory!\n");
893 		return B_NO_MEMORY;
894 	}
895 
896 	status_t error = ReadSection(section);
897 	if (error != B_OK)
898 		return error;
899 
900 	// parse the section strings
901 	section.currentOffset = 0;
902 	SetCurrentSection(&section);
903 
904 	error = ParseStrings();
905 	if (error != B_OK)
906 		return error;
907 
908 	return B_OK;
909 }
910 
911 
912 status_t
913 ReaderImplBase::ParseStrings()
914 {
915 	// allocate table, if there are any strings
916 	if (fCurrentSection->stringsCount == 0) {
917 		fCurrentSection->currentOffset += fCurrentSection->stringsLength;
918 		return B_OK;
919 	}
920 
921 	fCurrentSection->strings
922 		= new(std::nothrow) char*[fCurrentSection->stringsCount];
923 	if (fCurrentSection->strings == NULL) {
924 		fErrorOutput->PrintError("Error: Out of memory!\n");
925 		return B_NO_MEMORY;
926 	}
927 
928 	// parse the section and fill the table
929 	char* position
930 		= (char*)fCurrentSection->data + fCurrentSection->currentOffset;
931 	char* sectionEnd = position + fCurrentSection->stringsLength;
932 	uint32 index = 0;
933 	while (true) {
934 		if (position >= sectionEnd) {
935 			fErrorOutput->PrintError("Error: Malformed %s strings section\n",
936 				fCurrentSection->name);
937 			return B_BAD_DATA;
938 		}
939 
940 		size_t stringLength = strnlen(position, (char*)sectionEnd - position);
941 
942 		if (stringLength == 0) {
943 			if (position + 1 != sectionEnd) {
944 				fErrorOutput->PrintError(
945 					"Error: %ld excess bytes in %s strings section\n",
946 					sectionEnd - (position + 1), fCurrentSection->name);
947 				return B_BAD_DATA;
948 			}
949 
950 			if (index != fCurrentSection->stringsCount) {
951 				fErrorOutput->PrintError("Error: Invalid %s strings section: "
952 					"Less strings (%lld) than specified in the header (%lld)\n",
953 					fCurrentSection->name, index,
954 					fCurrentSection->stringsCount);
955 				return B_BAD_DATA;
956 			}
957 
958 			fCurrentSection->currentOffset += fCurrentSection->stringsLength;
959 
960 			return B_OK;
961 		}
962 
963 		if (index >= fCurrentSection->stringsCount) {
964 			fErrorOutput->PrintError("Error: Invalid %s strings section: "
965 				"More strings (%lld) than specified in the header (%lld)\n",
966 				fCurrentSection->name, index, fCurrentSection->stringsCount);
967 			return B_BAD_DATA;
968 		}
969 
970 		fCurrentSection->strings[index++] = position;
971 		position += stringLength + 1;
972 	}
973 }
974 
975 
976 status_t
977 ReaderImplBase::ParsePackageAttributesSection(
978 	AttributeHandlerContext* context, AttributeHandler* rootAttributeHandler)
979 {
980 	// parse package attributes
981 	SetCurrentSection(&fPackageAttributesSection);
982 
983 	// init the attribute handler stack
984 	rootAttributeHandler->SetLevel(0);
985 	ClearAttributeHandlerStack();
986 	PushAttributeHandler(rootAttributeHandler);
987 
988 	bool sectionHandled;
989 	status_t error = ParseAttributeTree(context, sectionHandled);
990 	if (error == B_OK && sectionHandled) {
991 		if (fPackageAttributesSection.currentOffset
992 				< fPackageAttributesSection.uncompressedLength) {
993 			fErrorOutput->PrintError("Error: %llu excess byte(s) in package "
994 				"attributes section\n",
995 				fPackageAttributesSection.uncompressedLength
996 					- fPackageAttributesSection.currentOffset);
997 			error = B_BAD_DATA;
998 		}
999 	}
1000 
1001 	SetCurrentSection(NULL);
1002 
1003 	// clean up on error
1004 	if (error != B_OK) {
1005 		context->ErrorOccurred();
1006 		while (AttributeHandler* handler = PopAttributeHandler()) {
1007 			if (handler != rootAttributeHandler)
1008 				handler->Delete(context);
1009 		}
1010 		return error;
1011 	}
1012 
1013 	return B_OK;
1014 }
1015 
1016 
1017 status_t
1018 ReaderImplBase::ParseAttributeTree(AttributeHandlerContext* context,
1019 	bool& _sectionHandled)
1020 {
1021 	if (context->hasLowLevelHandler) {
1022 		bool handleSection = false;
1023 		status_t error = context->lowLevelHandler->HandleSectionStart(
1024 			context->section, handleSection);
1025 		if (error != B_OK)
1026 			return error;
1027 
1028 		if (!handleSection) {
1029 			_sectionHandled = false;
1030 			return B_OK;
1031 		}
1032 	}
1033 
1034 	status_t error = _ParseAttributeTree(context);
1035 
1036 	if (context->hasLowLevelHandler) {
1037 		status_t endError = context->lowLevelHandler->HandleSectionEnd(
1038 			context->section);
1039 		if (error == B_OK)
1040 			error = endError;
1041 	}
1042 
1043 	_sectionHandled = true;
1044 	return error;
1045 }
1046 
1047 
1048 status_t
1049 ReaderImplBase::_Init(int fd, bool keepFD)
1050 {
1051 	fFD = fd;
1052 	fOwnsFD = keepFD;
1053 
1054 	return B_OK;
1055 }
1056 
1057 
1058 status_t
1059 ReaderImplBase::_ParseAttributeTree(AttributeHandlerContext* context)
1060 {
1061 	int level = 0;
1062 
1063 	while (true) {
1064 		uint8 id;
1065 		AttributeValue value;
1066 		bool hasChildren;
1067 		uint64 tag;
1068 
1069 		status_t error = _ReadAttribute(id, value, &hasChildren, &tag);
1070 		if (error != B_OK)
1071 			return error;
1072 
1073 		if (tag == 0) {
1074 			AttributeHandler* handler = PopAttributeHandler();
1075 			error = handler->NotifyDone(context);
1076 			if (error != B_OK)
1077 				return error;
1078 			if (level-- == 0)
1079 				return B_OK;
1080 
1081 			error = handler->Delete(context);
1082 			if (error != B_OK)
1083 				return error;
1084 
1085 			continue;
1086 		}
1087 
1088 		AttributeHandler* childHandler = NULL;
1089 		error = CurrentAttributeHandler()->HandleAttribute(context, id, value,
1090 			hasChildren ? &childHandler : NULL);
1091 		if (error != B_OK)
1092 			return error;
1093 
1094 		// parse children
1095 		if (hasChildren) {
1096 			// create an ignore handler, if necessary
1097 			if (childHandler == NULL) {
1098 				childHandler = new(std::nothrow) IgnoreAttributeHandler;
1099 				if (childHandler == NULL) {
1100 					fErrorOutput->PrintError("Error: Out of memory!\n");
1101 					return B_NO_MEMORY;
1102 				}
1103 			}
1104 
1105 			childHandler->SetLevel(++level);
1106 			PushAttributeHandler(childHandler);
1107 		}
1108 	}
1109 }
1110 
1111 
1112 status_t
1113 ReaderImplBase::_ReadAttribute(uint8& _id, AttributeValue& _value,
1114 	bool* _hasChildren, uint64* _tag)
1115 {
1116 	uint64 tag;
1117 	status_t error = ReadUnsignedLEB128(tag);
1118 	if (error != B_OK)
1119 		return error;
1120 
1121 	if (tag != 0) {
1122 		// get the type
1123 		uint16 type = attribute_tag_type(tag);
1124 		if (type >= B_HPKG_ATTRIBUTE_TYPE_ENUM_COUNT) {
1125 			fErrorOutput->PrintError("Error: Invalid %s section: attribute "
1126 				"type %d not supported!\n", fCurrentSection->name, type);
1127 			return B_BAD_DATA;
1128 		}
1129 
1130 		// get the ID
1131 		_id = attribute_tag_id(tag);
1132 		if (_id < B_HPKG_ATTRIBUTE_ID_ENUM_COUNT) {
1133 			if (type != kAttributeTypes[_id]) {
1134 				fErrorOutput->PrintError("Error: Invalid %s section: "
1135 					"unexpected type %d for attribute id %d (expected %d)!\n",
1136 					fCurrentSection->name, type, _id, kAttributeTypes[_id]);
1137 				return B_BAD_DATA;
1138 			}
1139 		} else if (fMinorFormatVersion <= fCurrentMinorFormatVersion) {
1140 			fErrorOutput->PrintError("Error: Invalid %s section: "
1141 				"attribute id %d not supported!\n", fCurrentSection->name, _id);
1142 			return B_BAD_DATA;
1143 		}
1144 
1145 		// get the value
1146 		error = ReadAttributeValue(type, attribute_tag_encoding(tag),
1147 			_value);
1148 		if (error != B_OK)
1149 			return error;
1150 	}
1151 
1152 	if (_hasChildren != NULL)
1153 		*_hasChildren = attribute_tag_has_children(tag);
1154 	if (_tag != NULL)
1155 		*_tag = tag;
1156 
1157 	return B_OK;
1158 }
1159 
1160 
1161 status_t
1162 ReaderImplBase::ReadAttributeValue(uint8 type, uint8 encoding,
1163 	AttributeValue& _value)
1164 {
1165 	switch (type) {
1166 		case B_HPKG_ATTRIBUTE_TYPE_INT:
1167 		case B_HPKG_ATTRIBUTE_TYPE_UINT:
1168 		{
1169 			uint64 intValue;
1170 			status_t error;
1171 
1172 			switch (encoding) {
1173 				case B_HPKG_ATTRIBUTE_ENCODING_INT_8_BIT:
1174 				{
1175 					uint8 value;
1176 					error = _Read(value);
1177 					intValue = value;
1178 					break;
1179 				}
1180 				case B_HPKG_ATTRIBUTE_ENCODING_INT_16_BIT:
1181 				{
1182 					uint16 value;
1183 					error = _Read(value);
1184 					intValue = B_BENDIAN_TO_HOST_INT16(value);
1185 					break;
1186 				}
1187 				case B_HPKG_ATTRIBUTE_ENCODING_INT_32_BIT:
1188 				{
1189 					uint32 value;
1190 					error = _Read(value);
1191 					intValue = B_BENDIAN_TO_HOST_INT32(value);
1192 					break;
1193 				}
1194 				case B_HPKG_ATTRIBUTE_ENCODING_INT_64_BIT:
1195 				{
1196 					uint64 value;
1197 					error = _Read(value);
1198 					intValue = B_BENDIAN_TO_HOST_INT64(value);
1199 					break;
1200 				}
1201 				default:
1202 				{
1203 					fErrorOutput->PrintError("Error: Invalid %s section: "
1204 						"invalid encoding %d for int value type %d\n",
1205 						fCurrentSection->name, encoding, type);
1206 					return B_BAD_VALUE;
1207 				}
1208 			}
1209 
1210 			if (error != B_OK)
1211 				return error;
1212 
1213 			if (type == B_HPKG_ATTRIBUTE_TYPE_INT)
1214 				_value.SetTo((int64)intValue);
1215 			else
1216 				_value.SetTo(intValue);
1217 
1218 			return B_OK;
1219 		}
1220 
1221 		case B_HPKG_ATTRIBUTE_TYPE_STRING:
1222 		{
1223 			if (encoding == B_HPKG_ATTRIBUTE_ENCODING_STRING_TABLE) {
1224 				uint64 index;
1225 				status_t error = ReadUnsignedLEB128(index);
1226 				if (error != B_OK)
1227 					return error;
1228 
1229 				if (index > fCurrentSection->stringsCount) {
1230 					fErrorOutput->PrintError("Error: Invalid %s section: "
1231 						"string reference (%lld) out of bounds (%lld)\n",
1232 						fCurrentSection->name, index,
1233 						fCurrentSection->stringsCount);
1234 					return B_BAD_DATA;
1235 				}
1236 
1237 				_value.SetTo(fCurrentSection->strings[index]);
1238 			} else if (encoding == B_HPKG_ATTRIBUTE_ENCODING_STRING_INLINE) {
1239 				const char* string;
1240 				status_t error = _ReadString(string);
1241 				if (error != B_OK)
1242 					return error;
1243 
1244 				_value.SetTo(string);
1245 			} else {
1246 				fErrorOutput->PrintError("Error: Invalid %s section: invalid "
1247 					"string encoding (%u)\n", fCurrentSection->name, encoding);
1248 				return B_BAD_DATA;
1249 			}
1250 
1251 			return B_OK;
1252 		}
1253 
1254 		default:
1255 			fErrorOutput->PrintError("Error: Invalid %s section: invalid "
1256 				"value type: %d\n", fCurrentSection->name, type);
1257 			return B_BAD_DATA;
1258 	}
1259 }
1260 
1261 
1262 status_t
1263 ReaderImplBase::ReadUnsignedLEB128(uint64& _value)
1264 {
1265 	uint64 result = 0;
1266 	int shift = 0;
1267 	while (true) {
1268 		uint8 byte;
1269 		status_t error = _Read(byte);
1270 		if (error != B_OK)
1271 			return error;
1272 
1273 		result |= uint64(byte & 0x7f) << shift;
1274 		if ((byte & 0x80) == 0)
1275 			break;
1276 		shift += 7;
1277 	}
1278 
1279 	_value = result;
1280 	return B_OK;
1281 }
1282 
1283 
1284 status_t
1285 ReaderImplBase::_ReadString(const char*& _string, size_t* _stringLength)
1286 {
1287 	const char* string
1288 		= (const char*)fCurrentSection->data + fCurrentSection->currentOffset;
1289 	size_t stringLength = strnlen(string,
1290 		fCurrentSection->uncompressedLength - fCurrentSection->currentOffset);
1291 
1292 	if (stringLength
1293 		== fCurrentSection->uncompressedLength
1294 			- fCurrentSection->currentOffset) {
1295 		fErrorOutput->PrintError(
1296 			"_ReadString(): string extends beyond %s end\n",
1297 			fCurrentSection->name);
1298 		return B_BAD_DATA;
1299 	}
1300 
1301 	_string = string;
1302 	if (_stringLength != NULL)
1303 		*_stringLength = stringLength;
1304 
1305 	fCurrentSection->currentOffset += stringLength + 1;
1306 	return B_OK;
1307 }
1308 
1309 
1310 status_t
1311 ReaderImplBase::_ReadSectionBuffer(void* buffer, size_t size)
1312 {
1313 	if (size > fCurrentSection->uncompressedLength
1314 			- fCurrentSection->currentOffset) {
1315 		fErrorOutput->PrintError(
1316 			"_ReadSectionBuffer(%lu): read beyond %s end\n", size,
1317 			fCurrentSection->name);
1318 		return B_BAD_DATA;
1319 	}
1320 
1321 	memcpy(buffer, fCurrentSection->data + fCurrentSection->currentOffset,
1322 		size);
1323 	fCurrentSection->currentOffset += size;
1324 	return B_OK;
1325 }
1326 
1327 
1328 status_t
1329 ReaderImplBase::ReadBuffer(off_t offset, void* buffer, size_t size)
1330 {
1331 	ssize_t bytesRead = pread(fFD, buffer, size, offset);
1332 	if (bytesRead < 0) {
1333 		fErrorOutput->PrintError("_ReadBuffer(%p, %lu) failed to read data: "
1334 			"%s\n", buffer, size, strerror(errno));
1335 		return errno;
1336 	}
1337 	if ((size_t)bytesRead != size) {
1338 		fErrorOutput->PrintError("_ReadBuffer(%p, %lu) failed to read all "
1339 			"data\n", buffer, size);
1340 		return B_ERROR;
1341 	}
1342 
1343 	return B_OK;
1344 }
1345 
1346 
1347 status_t
1348 ReaderImplBase::ReadSection(const PackageFileSection& section)
1349 {
1350 	return fHeapReader->ReadData(section.offset,
1351 		section.data, section.uncompressedLength);
1352 }
1353 
1354 
1355 }	// namespace BPrivate
1356 
1357 }	// namespace BHPKG
1358 
1359 }	// namespace BPackageKit
1360