tarvfswriter.cpp@ 78365

Last change on this file since 78365 was 76553, checked in by vboxsync, 6 years ago
scm --update-copyright-year
Property svn:eol-style set to `native` Property svn:keywords set to `Author Date Id Revision`
File size: 74.3 KB

Line
1	/* $Id: tarvfswriter.cpp 76553 2019-01-01 01:45:53Z vboxsync $ */
2	/** @file
3	* IPRT - TAR Virtual Filesystem, Writer.
4	*/
5
6	/*
7	* Copyright (C) 2010-2019 Oracle Corporation
8	*
9	* This file is part of VirtualBox Open Source Edition (OSE), as
10	* available from http://www.virtualbox.org. This file is free software;
11	* you can redistribute it and/or modify it under the terms of the GNU
12	* General Public License (GPL) as published by the Free Software
13	* Foundation, in version 2 as it comes in the "COPYING" file of the
14	* VirtualBox OSE distribution. VirtualBox OSE is distributed in the
15	* hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
16	*
17	* The contents of this file may alternatively be used under the terms
18	* of the Common Development and Distribution License Version 1.0
19	* (CDDL) only, as it comes in the "COPYING.CDDL" file of the
20	* VirtualBox OSE distribution, in which case the provisions of the
21	* CDDL are applicable instead of those of the GPL.
22	*
23	* You may elect to license modified versions of this file under the
24	* terms and conditions of either the GPL or the CDDL or both.
25	*/
26
27
28	/*********************************************************************************************************************************
29	* Header Files *
30	*********************************************************************************************************************************/
31	#include "internal/iprt.h"
32	#include <iprt/zip.h>
33
34	#include <iprt/asm.h>
35	#include <iprt/assert.h>
36	#include <iprt/err.h>
37	#include <iprt/file.h>
38	#include <iprt/mem.h>
39	#include <iprt/path.h>
40	#include <iprt/string.h>
41	#include <iprt/vfs.h>
42	#include <iprt/vfslowlevel.h>
43	#include <iprt/zero.h>
44
45	#include "tar.h"
46
47
48	/*********************************************************************************************************************************
49	* Defined Constants And Macros *
50	*********************************************************************************************************************************/
51	/** The TAR block size we're using in this implementation.
52	* @remarks Should technically be user configurable, but we don't currently need that. */
53	#define RTZIPTAR_BLOCKSIZE sizeof(RTZIPTARHDR)
54
55	/** Minimum file size we consider for sparse files. */
56	#define RTZIPTAR_MIN_SPARSE _64K
57
58
59	/*********************************************************************************************************************************
60	* Structures and Typedefs *
61	*********************************************************************************************************************************/
62	/**
63	* A data span descriptor in a sparse file.
64	*/
65	typedef struct RTZIPTARSPARSESPAN
66	{
67	/** Byte offset into the file of the data. */
68	uint64_t off;
69	/** Number of bytes of data, rounded up to a multiple of blocksize. */
70	uint64_t cb;
71	} RTZIPTARSPARSESPAN;
72	/** Pointer to a data span. */
73	typedef RTZIPTARSPARSESPAN *PRTZIPTARSPARSESPAN;
74	/** Pointer to a const data span. */
75	typedef RTZIPTARSPARSESPAN const *PCRTZIPTARSPARSESPAN;
76
77	/**
78	* Chunk of TAR sparse file data spans.
79	*/
80	typedef struct RTZIPTARSPARSECHUNK
81	{
82	/** List entry. */
83	RTLISTNODE Entry;
84	/** Array of data spans. */
85	RTZIPTARSPARSESPAN aSpans[63];
86	} RTZIPTARSPARSECHUNK;
87	AssertCompile(sizeof(RTZIPTARSPARSECHUNK) <= 1024);
88	AssertCompile(sizeof(RTZIPTARSPARSECHUNK) >= 1008);
89	/** Pointer to a chunk of TAR data spans. */
90	typedef RTZIPTARSPARSECHUNK *PRTZIPTARSPARSECHUNK;
91	/** Pointer to a const chunk of TAR data spans. */
92	typedef RTZIPTARSPARSECHUNK const *PCRTZIPTARSPARSECHUNK;
93
94	/**
95	* TAR sparse file info.
96	*/
97	typedef struct RTZIPTARSPARSE
98	{
99	/** Number of data bytes (real size). */
100	uint64_t cbDataSpans;
101	/** Number of data spans. */
102	uint32_t cDataSpans;
103	/** The index of the next span in the tail chunk (to avoid modulus 63). */
104	uint32_t iNextSpan;
105	/** Head of the data span chunk list (PRTZIPTARSPARSECHUNK). */
106	RTLISTANCHOR ChunkHead;
107	} RTZIPTARSPARSE;
108	/** Pointer to TAR sparse file info. */
109	typedef RTZIPTARSPARSE *PRTZIPTARSPARSE;
110	/** Pointer to a const TAR sparse file info. */
111	typedef RTZIPTARSPARSE const *PCRTZIPTARSPARSE;
112
113
114	/** Pointer to a the private data of a TAR filesystem stream. */
115	typedef struct RTZIPTARFSSTREAMWRITER *PRTZIPTARFSSTREAMWRITER;
116
117
118	/**
119	* Instance data for a file or I/O stream returned by
120	* RTVFSFSSTREAMOPS::pfnPushFile.
121	*/
122	typedef struct RTZIPTARFSSTREAMWRITERPUSH
123	{
124	/** Pointer to the parent FS stream writer instance.
125	* This is set to NULL should the push object live longer than the stream. */
126	PRTZIPTARFSSTREAMWRITER pParent;
127	/** The header offset, UINT64_MAX if non-seekable output. */
128	uint64_t offHdr;
129	/** The data offset, UINT64_MAX if non-seekable output. */
130	uint64_t offData;
131	/** The current I/O stream position (relative to offData). */
132	uint64_t offCurrent;
133	/** The expected size amount of file content, or max file size if open-ended. */
134	uint64_t cbExpected;
135	/** The current amount of file content written. */
136	uint64_t cbCurrent;
137	/** Object info copy for rtZipTarWriterPush_QueryInfo. */
138	RTFSOBJINFO ObjInfo;
139	/** Set if open-ended file size requiring a tar header update when done. */
140	bool fOpenEnded;
141	} RTZIPTARFSSTREAMWRITERPUSH;
142	/** Pointer to a push I/O instance. */
143	typedef RTZIPTARFSSTREAMWRITERPUSH *PRTZIPTARFSSTREAMWRITERPUSH;
144
145
146	/**
147	* Tar filesystem stream private data.
148	*/
149	typedef struct RTZIPTARFSSTREAMWRITER
150	{
151	/** The output I/O stream. */
152	RTVFSIOSTREAM hVfsIos;
153	/** Non-nil if the output is a file. */
154	RTVFSFILE hVfsFile;
155
156	/** The current push file. NULL if none. */
157	PRTZIPTARFSSTREAMWRITERPUSH pPush;
158
159	/** The TAR format. */
160	RTZIPTARFORMAT enmFormat;
161	/** Set if we've encountered a fatal error. */
162	int rcFatal;
163	/** Flags. */
164	uint32_t fFlags;
165
166	/** Number of bytes written. */
167	uint64_t cbWritten;
168
169	/** @name Attribute overrides.
170	* @{
171	*/
172	RTUID uidOwner; /*< Owner, NIL_RTUID if no change. /
173	char pszOwner; /< Owner, NULL if no change. /
174	RTGID gidGroup; /*< Group, NIL_RTGID if no change. /
175	char pszGroup; /< Group, NULL if no change. /
176	char pszPrefix; /< Path prefix, NULL if no change. /
177	size_t cchPrefix; /*< The length of pszPrefix. /
178	PRTTIMESPEC pModTime; /*< Modification time, NULL of no change. /
179	RTTIMESPEC ModTime; /*< pModTime points to this. /
180	RTFMODE fFileModeAndMask; /*< File mode AND mask. /
181	RTFMODE fFileModeOrMask; /*< File mode OR mask. /
182	RTFMODE fDirModeAndMask; /*< Directory mode AND mask. /
183	RTFMODE fDirModeOrMask; /*< Directory mode OR mask. /
184	/** @} */
185
186
187	/** Number of headers returned by rtZipTarFssWriter_ObjInfoToHdr. */
188	uint32_t cHdrs;
189	/** Header buffers returned by rtZipTarFssWriter_ObjInfoToHdr. */
190	RTZIPTARHDR aHdrs[3];
191	} RTZIPTARFSSTREAMWRITER;
192
193
194	/*********************************************************************************************************************************
195	* Internal Functions *
196	*********************************************************************************************************************************/
197	static DECLCALLBACK(int) rtZipTarWriterPush_Seek(void *pvThis, RTFOFF offSeek, unsigned uMethod, PRTFOFF poffActual);
198	static int rtZipTarFssWriter_CompleteCurrentPushFile(PRTZIPTARFSSTREAMWRITER pThis);
199	static int rtZipTarFssWriter_AddFile(PRTZIPTARFSSTREAMWRITER pThis, const char *pszPath, RTVFSIOSTREAM hVfsIos,
200	PCRTFSOBJINFO pObjInfo, const char pszOwnerNm, const char pszGroupNm);
201
202
203	/**
204	* Calculates the header checksum and stores it in the chksum field.
205	*
206	* @returns IPRT status code.
207	* @param pHdr The header.
208	*/
209	static int rtZipTarFssWriter_ChecksumHdr(PRTZIPTARHDR pHdr)
210	{
211	int32_t iUnsignedChksum;
212	rtZipTarCalcChkSum(pHdr, &iUnsignedChksum, NULL);
213
214	int rc = RTStrFormatU32(pHdr->Common.chksum, sizeof(pHdr->Common.chksum), iUnsignedChksum,
215	8 /uBase/, -1 /cchWidth/, sizeof(pHdr->Common.chksum) - 1, RTSTR_F_ZEROPAD \| RTSTR_F_PRECISION);
216	AssertRCReturn(rc, VERR_TAR_NUM_VALUE_TOO_LARGE);
217	return VINF_SUCCESS;
218	}
219
220
221
222	/**
223	* Formats a 12 character wide file offset or size field.
224	*
225	* This is mainly used for RTZIPTARHDR::Common.size, but also for formatting the
226	* sparse map.
227	*
228	* @returns IPRT status code.
229	* @param pach12Field The 12 character wide destination field.
230	* @param off The offset to set.
231	*/
232	static int rtZipTarFssWriter_FormatOffset(char pach12Field[12], uint64_t off)
233	{
234	/*
235	* Is the size small enough for the standard octal string encoding?
236	*
237	* Note! We could actually use the terminator character as well if we liked,
238	* but let not do that as it's easier to test this way.
239	*/
240	if (off < _4G * 2U)
241	{
242	int rc = RTStrFormatU64(pach12Field, 12, off, 8 /uBase/, -1 /cchWidth/, 12 - 1, RTSTR_F_ZEROPAD \| RTSTR_F_PRECISION);
243	AssertRCReturn(rc, rc);
244	}
245	/*
246	* No, use the base 256 extension. Set the highest bit of the left most
247	* character. We don't deal with negatives here, cause the size have to
248	* be greater than zero.
249	*
250	* Note! The base-256 extension are never used by gtar or libarchive
251	* with the "ustar \0" format version, only the later
252	* "ustar\000" version. However, this shouldn't cause much
253	* trouble as they are not picky about what they read.
254	*/
255	/** @todo above note is wrong: GNU tar only uses base-256 with the GNU tar
256	* format, i.e. "ustar \0", see create.c line 303 in v1.29. */
257	else
258	{
259	size_t cchField = 12 - 1;
260	unsigned char puchField = (unsigned char )pach12Field;
261	puchField[0] = 0x80;
262	do
263	{
264	puchField[cchField--] = off & 0xff;
265	off >>= 8;
266	} while (cchField);
267	}
268
269	return VINF_SUCCESS;
270	}
271
272
273	/**
274	* Creates one or more tar headers for the object.
275	*
276	* Returns RTZIPTARFSSTREAMWRITER::aHdrs and RTZIPTARFSSTREAMWRITER::cHdrs.
277	*
278	* @returns IPRT status code.
279	* @param pThis The TAR writer instance.
280	* @param pszPath The path to the file.
281	* @param hVfsIos The I/O stream of the file.
282	* @param fFlags The RTVFSFSSTREAMOPS::pfnAdd flags.
283	* @param pObjInfo The object information.
284	* @param pszOwnerNm The owner name.
285	* @param pszGroupNm The group name.
286	* @param chType The tar record type, UINT8_MAX for default.
287	*/
288	static int rtZipTarFssWriter_ObjInfoToHdr(PRTZIPTARFSSTREAMWRITER pThis, const char *pszPath, PCRTFSOBJINFO pObjInfo,
289	const char pszOwnerNm, const char pszGroupNm, uint8_t chType)
290	{
291	pThis->cHdrs = 0;
292	RT_ZERO(pThis->aHdrs[0]);
293
294	/*
295	* The path name first. Make sure to flip DOS slashes.
296	*/
297	size_t cchPath = strlen(pszPath);
298	if (cchPath < sizeof(pThis->aHdrs[0].Common.name))
299	{
300	memcpy(pThis->aHdrs[0].Common.name, pszPath, cchPath + 1);
301	#if RTPATH_STYLE != RTPATH_STR_F_STYLE_UNIX
302	char *pszDosSlash = strchr(pThis->aHdrs[0].Common.name, '\\');
303	while (pszDosSlash)
304	{
305	*pszDosSlash = '/';
306	pszDosSlash = strchr(pszDosSlash + 1, '\\');
307	}
308	#endif
309	}
310	else
311	{
312	/** @todo implement gnu and pax long name extensions. */
313	return VERR_TAR_NAME_TOO_LONG;
314	}
315
316	/*
317	* File mode. ASSUME that the unix part of the IPRT mode mask is
318	* compatible with the TAR/Unix world.
319	*/
320	uint32_t uValue = pObjInfo->Attr.fMode & RTFS_UNIX_MASK;
321	if (RTFS_IS_DIRECTORY(pObjInfo->Attr.fMode))
322	uValue = (uValue & pThis->fDirModeAndMask) \| pThis->fDirModeOrMask;
323	else
324	uValue = (uValue & pThis->fFileModeAndMask) \| pThis->fFileModeOrMask;
325	int rc = RTStrFormatU32(pThis->aHdrs[0].Common.mode, sizeof(pThis->aHdrs[0].Common.mode), uValue, 8 /uBase/,
326	-1 /cchWidth/, sizeof(pThis->aHdrs[0].Common.mode) - 1, RTSTR_F_ZEROPAD \| RTSTR_F_PRECISION);
327	AssertRCReturn(rc, VERR_TAR_NUM_VALUE_TOO_LARGE);
328
329	/*
330	* uid & gid. Just guard against NIL values as they won't fit.
331	*/
332	uValue = pThis->uidOwner != NIL_RTUID ? pThis->uidOwner
333	: pObjInfo->Attr.u.Unix.uid != NIL_RTUID ? pObjInfo->Attr.u.Unix.uid : 0;
334	rc = RTStrFormatU32(pThis->aHdrs[0].Common.uid, sizeof(pThis->aHdrs[0].Common.uid), uValue,
335	8 /uBase/, -1 /cchWidth/, sizeof(pThis->aHdrs[0].Common.uid) - 1, RTSTR_F_ZEROPAD \| RTSTR_F_PRECISION);
336	AssertRCReturn(rc, VERR_TAR_NUM_VALUE_TOO_LARGE);
337
338	uValue = pThis->gidGroup != NIL_RTGID ? pThis->gidGroup
339	: pObjInfo->Attr.u.Unix.gid != NIL_RTGID ? pObjInfo->Attr.u.Unix.gid : 0;
340	rc = RTStrFormatU32(pThis->aHdrs[0].Common.gid, sizeof(pThis->aHdrs[0].Common.gid), uValue,
341	8 /uBase/, -1 /cchWidth/, sizeof(pThis->aHdrs[0].Common.gid) - 1, RTSTR_F_ZEROPAD \| RTSTR_F_PRECISION);
342	AssertRCReturn(rc, VERR_TAR_NUM_VALUE_TOO_LARGE);
343
344	/*
345	* The file size.
346	*/
347	rc = rtZipTarFssWriter_FormatOffset(pThis->aHdrs[0].Common.size, pObjInfo->cbObject);
348	AssertRCReturn(rc, rc);
349
350	/*
351	* Modification time relative to unix epoc.
352	*/
353	rc = RTStrFormatU64(pThis->aHdrs[0].Common.mtime, sizeof(pThis->aHdrs[0].Common.mtime),
354	RTTimeSpecGetSeconds(pThis->pModTime ? pThis->pModTime : &pObjInfo->ModificationTime),
355	8 /uBase/, -1 /cchWidth/, sizeof(pThis->aHdrs[0].Common.mtime) - 1, RTSTR_F_ZEROPAD \| RTSTR_F_PRECISION);
356	AssertRCReturn(rc, rc);
357
358	/* Skipping checksum for now */
359
360	/*
361	* The type flag.
362	*/
363	if (chType == UINT8_MAX)
364	switch (pObjInfo->Attr.fMode & RTFS_TYPE_MASK)
365	{
366	case RTFS_TYPE_FIFO: chType = RTZIPTAR_TF_FIFO; break;
367	case RTFS_TYPE_DEV_CHAR: chType = RTZIPTAR_TF_CHR; break;
368	case RTFS_TYPE_DIRECTORY: chType = RTZIPTAR_TF_DIR; break;
369	case RTFS_TYPE_DEV_BLOCK: chType = RTZIPTAR_TF_BLK; break;
370	case RTFS_TYPE_FILE: chType = RTZIPTAR_TF_NORMAL; break;
371	case RTFS_TYPE_SYMLINK: chType = RTZIPTAR_TF_SYMLINK; break;
372	case RTFS_TYPE_SOCKET: chType = RTZIPTAR_TF_FIFO; break;
373	case RTFS_TYPE_WHITEOUT: AssertFailedReturn(VERR_WRONG_TYPE);
374	}
375	pThis->aHdrs[0].Common.typeflag = chType;
376
377	/* No link name, at least not for now. Caller might set it. */
378
379	/*
380	* Set TAR record magic and version.
381	*/
382	if (pThis->enmFormat == RTZIPTARFORMAT_GNU)
383	memcpy(pThis->aHdrs[0].Gnu.magic, RTZIPTAR_GNU_MAGIC, sizeof(pThis->aHdrs[0].Gnu.magic));
384	else if ( pThis->enmFormat == RTZIPTARFORMAT_USTAR
385	\|\| pThis->enmFormat == RTZIPTARFORMAT_PAX)
386	{
387	memcpy(pThis->aHdrs[0].Common.magic, RTZIPTAR_USTAR_MAGIC, sizeof(pThis->aHdrs[0].Common.magic));
388	memcpy(pThis->aHdrs[0].Common.version, RTZIPTAR_USTAR_VERSION, sizeof(pThis->aHdrs[0].Common.version));
389	}
390	else
391	AssertFailedReturn(VERR_INTERNAL_ERROR_4);
392
393	/*
394	* Owner and group names. Silently truncate them for now.
395	*/
396	RTStrCopy(pThis->aHdrs[0].Common.uname, sizeof(pThis->aHdrs[0].Common.uname), pThis->pszOwner ? pThis->pszOwner : pszOwnerNm);
397	RTStrCopy(pThis->aHdrs[0].Common.gname, sizeof(pThis->aHdrs[0].Common.uname), pThis->pszGroup ? pThis->pszGroup : pszGroupNm);
398
399	/*
400	* Char/block device numbers.
401	*/
402	if ( RTFS_IS_DEV_BLOCK(pObjInfo->Attr.fMode)
403	\|\| RTFS_IS_DEV_CHAR(pObjInfo->Attr.fMode) )
404	{
405	rc = RTStrFormatU32(pThis->aHdrs[0].Common.devmajor, sizeof(pThis->aHdrs[0].Common.devmajor),
406	RTDEV_MAJOR(pObjInfo->Attr.u.Unix.Device),
407	8 /uBase/, -1 /cchWidth/, sizeof(pThis->aHdrs[0].Common.devmajor) - 1, RTSTR_F_ZEROPAD \| RTSTR_F_PRECISION);
408	AssertRCReturn(rc, VERR_TAR_NUM_VALUE_TOO_LARGE);
409
410	rc = RTStrFormatU32(pThis->aHdrs[0].Common.devminor, sizeof(pThis->aHdrs[0].Common.devmajor),
411	RTDEV_MINOR(pObjInfo->Attr.u.Unix.Device),
412	8 /uBase/, -1 /cchWidth/, sizeof(pThis->aHdrs[0].Common.devmajor) - 1, RTSTR_F_ZEROPAD \| RTSTR_F_PRECISION);
413	AssertRCReturn(rc, VERR_TAR_NUM_VALUE_TOO_LARGE);
414	}
415
416	#if 0 /** @todo why doesn't this work? */
417	/*
418	* Set GNU specific stuff.
419	*/
420	if (pThis->enmFormat == RTZIPTARFORMAT_GNU)
421	{
422	rc = RTStrFormatU64(pThis->aHdrs[0].Gnu.ctime, sizeof(pThis->aHdrs[0].Gnu.ctime),
423	RTTimeSpecGetSeconds(&pObjInfo->ChangeTime),
424	8 /uBase/, -1 /cchWidth/, sizeof(pThis->aHdrs[0].Gnu.ctime) - 1, RTSTR_F_ZEROPAD \| RTSTR_F_PRECISION);
425	AssertRCReturn(rc, rc);
426
427	rc = RTStrFormatU64(pThis->aHdrs[0].Gnu.atime, sizeof(pThis->aHdrs[0].Gnu.atime),
428	RTTimeSpecGetSeconds(&pObjInfo->ChangeTime),
429	8 /uBase/, -1 /cchWidth/, sizeof(pThis->aHdrs[0].Gnu.atime) - 1, RTSTR_F_ZEROPAD \| RTSTR_F_PRECISION);
430	AssertRCReturn(rc, rc);
431	}
432	#endif
433
434	/*
435	* Finally the checksum.
436	*/
437	pThis->cHdrs = 1;
438	return rtZipTarFssWriter_ChecksumHdr(&pThis->aHdrs[0]);
439	}
440
441
442
443
444	/**
445	* @interface_method_impl{RTVFSOBJOPS,pfnClose}
446	*/
447	static DECLCALLBACK(int) rtZipTarWriterPush_Close(void *pvThis)
448	{
449	PRTZIPTARFSSTREAMWRITERPUSH pPush = (PRTZIPTARFSSTREAMWRITERPUSH)pvThis;
450	PRTZIPTARFSSTREAMWRITER pParent = pPush->pParent;
451	if (pParent)
452	{
453	if (pParent->pPush == pPush)
454	rtZipTarFssWriter_CompleteCurrentPushFile(pParent);
455	else
456	AssertFailedStmt(pPush->pParent = NULL);
457	}
458	return VINF_SUCCESS;
459	}
460
461
462	/**
463	* @interface_method_impl{RTVFSOBJOPS,pfnQueryInfo}
464	*/
465	static DECLCALLBACK(int) rtZipTarWriterPush_QueryInfo(void *pvThis, PRTFSOBJINFO pObjInfo, RTFSOBJATTRADD enmAddAttr)
466	{
467	PRTZIPTARFSSTREAMWRITERPUSH pPush = (PRTZIPTARFSSTREAMWRITERPUSH)pvThis;
468
469	/* Basic info (w/ additional unix attribs). */
470	*pObjInfo = pPush->ObjInfo;
471	pObjInfo->cbObject = pPush->cbCurrent;
472	pObjInfo->cbAllocated = RT_ALIGN_64(pPush->cbCurrent, RTZIPTAR_BLOCKSIZE);
473
474	/* Additional info. */
475	switch (enmAddAttr)
476	{
477	case RTFSOBJATTRADD_NOTHING:
478	case RTFSOBJATTRADD_UNIX:
479	Assert(pObjInfo->Attr.enmAdditional == RTFSOBJATTRADD_UNIX);
480	break;
481
482	case RTFSOBJATTRADD_UNIX_OWNER:
483	pObjInfo->Attr.u.UnixOwner.uid = pPush->ObjInfo.Attr.u.Unix.uid;
484	if (pPush->pParent)
485	strcpy(pObjInfo->Attr.u.UnixOwner.szName, pPush->pParent->aHdrs[0].Common.uname);
486	else
487	pObjInfo->Attr.u.UnixOwner.szName[0] = '\0';
488	pObjInfo->Attr.enmAdditional = enmAddAttr;
489	break;
490
491	case RTFSOBJATTRADD_UNIX_GROUP:
492	pObjInfo->Attr.u.UnixGroup.gid = pPush->ObjInfo.Attr.u.Unix.gid;
493	if (pPush->pParent)
494	strcpy(pObjInfo->Attr.u.UnixGroup.szName, pPush->pParent->aHdrs[0].Common.uname);
495	else
496	pObjInfo->Attr.u.UnixGroup.szName[0] = '\0';
497	pObjInfo->Attr.enmAdditional = enmAddAttr;
498	break;
499
500	case RTFSOBJATTRADD_EASIZE:
501	pObjInfo->Attr.u.EASize.cb = 0;
502	pObjInfo->Attr.enmAdditional = enmAddAttr;
503	break;
504
505	default:
506	AssertFailed();
507	}
508
509	return VINF_SUCCESS;
510	}
511
512
513	/**
514	* @interface_method_impl{RTVFSIOSTREAMOPS,pfnRead}
515	*/
516	static DECLCALLBACK(int) rtZipTarWriterPush_Read(void pvThis, RTFOFF off, PCRTSGBUF pSgBuf, bool fBlocking, size_t pcbRead)
517	{
518	/* No read support, sorry. */
519	RT_NOREF(pvThis, off, pSgBuf, fBlocking, pcbRead);
520	AssertFailed();
521	return VERR_ACCESS_DENIED;
522	}
523
524
525	/**
526	* @interface_method_impl{RTVFSIOSTREAMOPS,pfnWrite}
527	*/
528	static DECLCALLBACK(int) rtZipTarWriterPush_Write(void pvThis, RTFOFF off, PCRTSGBUF pSgBuf, bool fBlocking, size_t pcbWritten)
529	{
530	PRTZIPTARFSSTREAMWRITERPUSH pPush = (PRTZIPTARFSSTREAMWRITERPUSH)pvThis;
531	PRTZIPTARFSSTREAMWRITER pParent = pPush->pParent;
532	AssertPtrReturn(pParent, VERR_WRONG_ORDER);
533
534	int rc = pParent->rcFatal;
535	AssertRCReturn(rc, rc);
536
537	/*
538	* Single segment at a time.
539	*/
540	Assert(pSgBuf->cSegs == 1);
541	size_t cbToWrite = pSgBuf->paSegs[0].cbSeg;
542	void const *pvToWrite = pSgBuf->paSegs[0].pvSeg;
543
544	/*
545	* Hopefully we don't need to seek. But if we do, let the seek method do
546	* it as it's not entirely trivial.
547	*/
548	if ( off < 0
549	\|\| (uint64_t)off == pPush->offCurrent)
550	rc = VINF_SUCCESS;
551	else
552	rc = rtZipTarWriterPush_Seek(pvThis, off, RTFILE_SEEK_BEGIN, NULL);
553	if (RT_SUCCESS(rc))
554	{
555	Assert(pPush->offCurrent <= pPush->cbExpected);
556	Assert(pPush->offCurrent <= pPush->cbCurrent);
557	AssertMsgReturn(cbToWrite <= pPush->cbExpected - pPush->offCurrent,
558	("offCurrent=%#RX64 + cbToWrite=%#zx = %#RX64; cbExpected=%RX64\n",
559	pPush->offCurrent, cbToWrite, pPush->offCurrent + cbToWrite, pPush->cbExpected),
560	VERR_DISK_FULL);
561	size_t cbWritten = 0;
562	rc = RTVfsIoStrmWrite(pParent->hVfsIos, pvToWrite, cbToWrite, fBlocking, &cbWritten);
563	if (RT_SUCCESS(rc))
564	{
565	pPush->offCurrent += cbWritten;
566	if (pPush->offCurrent > pPush->cbCurrent)
567	{
568	pParent->cbWritten = pPush->offCurrent - pPush->cbCurrent;
569	pPush->cbCurrent = pPush->offCurrent;
570	}
571	if (pcbWritten)
572	*pcbWritten = cbWritten;
573	}
574	}
575
576	/*
577	* Fatal errors get down here, non-fatal ones returns earlier.
578	*/
579	if (RT_SUCCESS(rc))
580	return VINF_SUCCESS;
581	pParent->rcFatal = rc;
582	return rc;
583	}
584
585
586	/**
587	* @interface_method_impl{RTVFSIOSTREAMOPS,pfnFlush}
588	*/
589	static DECLCALLBACK(int) rtZipTarWriterPush_Flush(void *pvThis)
590	{
591	PRTZIPTARFSSTREAMWRITERPUSH pPush = (PRTZIPTARFSSTREAMWRITERPUSH)pvThis;
592	PRTZIPTARFSSTREAMWRITER pParent = pPush->pParent;
593	AssertPtrReturn(pParent, VERR_WRONG_ORDER);
594	int rc = pParent->rcFatal;
595	if (RT_SUCCESS(rc))
596	pParent->rcFatal = rc = RTVfsIoStrmFlush(pParent->hVfsIos);
597	return rc;
598	}
599
600
601	/**
602	* @interface_method_impl{RTVFSIOSTREAMOPS,pfnPollOne}
603	*/
604	static DECLCALLBACK(int) rtZipTarWriterPush_PollOne(void *pvThis, uint32_t fEvents, RTMSINTERVAL cMillies, bool fIntr,
605	uint32_t *pfRetEvents)
606	{
607	PRTZIPTARFSSTREAMWRITERPUSH pPush = (PRTZIPTARFSSTREAMWRITERPUSH)pvThis;
608	PRTZIPTARFSSTREAMWRITER pParent = pPush->pParent;
609	AssertPtrReturn(pParent, VERR_WRONG_ORDER);
610	return RTVfsIoStrmPoll(pParent->hVfsIos, fEvents, cMillies, fIntr, pfRetEvents);
611	}
612
613
614	/**
615	* @interface_method_impl{RTVFSIOSTREAMOPS,pfnTell}
616	*/
617	static DECLCALLBACK(int) rtZipTarWriterPush_Tell(void *pvThis, PRTFOFF poffActual)
618	{
619	PRTZIPTARFSSTREAMWRITERPUSH pPush = (PRTZIPTARFSSTREAMWRITERPUSH)pvThis;
620	*poffActual = (RTFOFF)pPush->offCurrent;
621	return VINF_SUCCESS;
622	}
623
624
625	/**
626	* @interface_method_impl{RTVFSIOSTREAMOPS,pfnSkip}
627	*/
628	static DECLCALLBACK(int) rtZipTarWriterPush_Skip(void *pvThis, RTFOFF cb)
629	{
630	RT_NOREF(pvThis, cb);
631	AssertFailed();
632	return VERR_ACCESS_DENIED;
633	}
634
635
636	/**
637	* @interface_method_impl{RTVFSOBJSETOPS,pfnMode}
638	*/
639	static DECLCALLBACK(int) rtZipTarWriterPush_SetMode(void *pvThis, RTFMODE fMode, RTFMODE fMask)
640	{
641	RT_NOREF(pvThis, fMode, fMask);
642	AssertFailed();
643	return VERR_ACCESS_DENIED;
644	}
645
646
647	/**
648	* @interface_method_impl{RTVFSOBJSETOPS,pfnSetTimes}
649	*/
650	static DECLCALLBACK(int) rtZipTarWriterPush_SetTimes(void *pvThis, PCRTTIMESPEC pAccessTime, PCRTTIMESPEC pModificationTime,
651	PCRTTIMESPEC pChangeTime, PCRTTIMESPEC pBirthTime)
652	{
653	RT_NOREF(pvThis, pAccessTime, pModificationTime, pChangeTime, pBirthTime);
654	AssertFailed();
655	return VERR_ACCESS_DENIED;
656	}
657
658
659	/**
660	* @interface_method_impl{RTVFSOBJSETOPS,pfnSetOwner}
661	*/
662	static DECLCALLBACK(int) rtZipTarWriterPush_SetOwner(void *pvThis, RTUID uid, RTGID gid)
663	{
664	RT_NOREF(pvThis, uid, gid);
665	AssertFailed();
666	return VERR_ACCESS_DENIED;
667	}
668
669
670	/**
671	* @interface_method_impl{RTVFSFILEOPS,pfnSeek}
672	*/
673	static DECLCALLBACK(int) rtZipTarWriterPush_Seek(void *pvThis, RTFOFF offSeek, unsigned uMethod, PRTFOFF poffActual)
674	{
675	PRTZIPTARFSSTREAMWRITERPUSH pPush = (PRTZIPTARFSSTREAMWRITERPUSH)pvThis;
676	PRTZIPTARFSSTREAMWRITER pParent = pPush->pParent;
677	AssertPtrReturn(pParent, VERR_WRONG_ORDER);
678
679	int rc = pParent->rcFatal;
680	AssertRCReturn(rc, rc);
681	Assert(pPush->offCurrent <= pPush->cbCurrent);
682
683	/*
684	* Calculate the new file offset.
685	*/
686	RTFOFF offNewSigned;
687	switch (uMethod)
688	{
689	case RTFILE_SEEK_BEGIN:
690	offNewSigned = offSeek;
691	break;
692	case RTFILE_SEEK_CURRENT:
693	offNewSigned = pPush->offCurrent + offSeek;
694	break;
695	case RTFILE_SEEK_END:
696	offNewSigned = pPush->cbCurrent + offSeek;
697	break;
698	default:
699	AssertFailedReturn(VERR_INVALID_PARAMETER);
700	}
701
702	/*
703	* Check the new file offset against expectations.
704	*/
705	AssertMsgReturn(offNewSigned >= 0, ("offNewSigned=%RTfoff\n", offNewSigned), VERR_NEGATIVE_SEEK);
706
707	uint64_t offNew = (uint64_t)offNewSigned;
708	AssertMsgReturn(offNew <= pPush->cbExpected, ("offNew=%#RX64 cbExpected=%#Rx64\n", offNew, pPush->cbExpected), VERR_SEEK);
709
710	/*
711	* Any change at all? We can always hope...
712	*/
713	if (offNew == pPush->offCurrent)
714	{ }
715	/*
716	* Gap that needs zero filling?
717	*/
718	else if (offNew > pPush->cbCurrent)
719	{
720	if (pPush->offCurrent != pPush->cbCurrent)
721	{
722	AssertReturn(pParent->hVfsFile != NIL_RTVFSFILE, VERR_NOT_A_FILE);
723	rc = RTVfsFileSeek(pParent->hVfsFile, pPush->offData + pPush->cbCurrent, RTFILE_SEEK_BEGIN, NULL);
724	if (RT_FAILURE(rc))
725	return pParent->rcFatal = rc;
726	pPush->offCurrent = pPush->cbCurrent;
727	}
728
729	uint64_t cbToZero = offNew - pPush->cbCurrent;
730	rc = RTVfsIoStrmZeroFill(pParent->hVfsIos, cbToZero);
731	if (RT_FAILURE(rc))
732	return pParent->rcFatal = rc;
733	pParent->cbWritten += cbToZero;
734	pPush->cbCurrent = pPush->offCurrent = offNew;
735	}
736	/*
737	* Just change the file position to somewhere we've already written.
738	*/
739	else
740	{
741	AssertReturn(pParent->hVfsFile != NIL_RTVFSFILE, VERR_NOT_A_FILE);
742	rc = RTVfsFileSeek(pParent->hVfsFile, pPush->offData + offNew, RTFILE_SEEK_BEGIN, NULL);
743	if (RT_FAILURE(rc))
744	return pParent->rcFatal = rc;
745	pPush->offCurrent = offNew;
746	}
747	Assert(pPush->offCurrent <= pPush->cbCurrent);
748
749	if (poffActual)
750	*poffActual = pPush->offCurrent;
751	return VINF_SUCCESS;
752	}
753
754
755	/**
756	* @interface_method_impl{RTVFSFILEOPS,pfnQuerySize}
757	*/
758	static DECLCALLBACK(int) rtZipTarWriterPush_QuerySize(void pvThis, uint64_t pcbFile)
759	{
760	PRTZIPTARFSSTREAMWRITERPUSH pPush = (PRTZIPTARFSSTREAMWRITERPUSH)pvThis;
761	*pcbFile = pPush->cbCurrent;
762	return VINF_SUCCESS;
763	}
764
765
766	/**
767	* TAR writer push I/O stream operations.
768	*/
769	DECL_HIDDEN_CONST(const RTVFSIOSTREAMOPS) g_rtZipTarWriterIoStrmOps =
770	{
771	{ /* Obj */
772	RTVFSOBJOPS_VERSION,
773	RTVFSOBJTYPE_IO_STREAM,
774	"TAR push I/O Stream",
775	rtZipTarWriterPush_Close,
776	rtZipTarWriterPush_QueryInfo,
777	RTVFSOBJOPS_VERSION
778	},
779	RTVFSIOSTREAMOPS_VERSION,
780	RTVFSIOSTREAMOPS_FEAT_NO_SG,
781	rtZipTarWriterPush_Read,
782	rtZipTarWriterPush_Write,
783	rtZipTarWriterPush_Flush,
784	rtZipTarWriterPush_PollOne,
785	rtZipTarWriterPush_Tell,
786	rtZipTarWriterPush_Skip,
787	NULL /ZeroFill/,
788	RTVFSIOSTREAMOPS_VERSION,
789	};
790
791
792	/**
793	* TAR writer push file operations.
794	*/
795	DECL_HIDDEN_CONST(const RTVFSFILEOPS) g_rtZipTarWriterFileOps =
796	{
797	{ /* Stream */
798	{ /* Obj */
799	RTVFSOBJOPS_VERSION,
800	RTVFSOBJTYPE_FILE,
801	"TAR push file",
802	rtZipTarWriterPush_Close,
803	rtZipTarWriterPush_QueryInfo,
804	RTVFSOBJOPS_VERSION
805	},
806	RTVFSIOSTREAMOPS_VERSION,
807	RTVFSIOSTREAMOPS_FEAT_NO_SG,
808	rtZipTarWriterPush_Read,
809	rtZipTarWriterPush_Write,
810	rtZipTarWriterPush_Flush,
811	rtZipTarWriterPush_PollOne,
812	rtZipTarWriterPush_Tell,
813	rtZipTarWriterPush_Skip,
814	NULL /ZeroFill/,
815	RTVFSIOSTREAMOPS_VERSION,
816	},
817	RTVFSFILEOPS_VERSION,
818	0,
819	{ /* ObjSet */
820	RTVFSOBJSETOPS_VERSION,
821	RT_UOFFSETOF(RTVFSFILEOPS, ObjSet) - RT_UOFFSETOF(RTVFSFILEOPS, Stream.Obj),
822	rtZipTarWriterPush_SetMode,
823	rtZipTarWriterPush_SetTimes,
824	rtZipTarWriterPush_SetOwner,
825	RTVFSOBJSETOPS_VERSION
826	},
827	rtZipTarWriterPush_Seek,
828	rtZipTarWriterPush_QuerySize,
829	NULL /SetSize/,
830	NULL /QueryMaxSize/,
831	RTVFSFILEOPS_VERSION
832	};
833
834
835
836	/**
837	* Checks rcFatal and completes any current push file.
838	*
839	* On return the output stream position will be at the next header location.
840	*
841	* After this call, the push object no longer can write anything.
842	*
843	* @returns IPRT status code.
844	* @param pThis The TAR writer instance.
845	*/
846	static int rtZipTarFssWriter_CompleteCurrentPushFile(PRTZIPTARFSSTREAMWRITER pThis)
847	{
848	/*
849	* Check if there is a push file pending, remove it if there is.
850	* We also check for fatal errors at this point so the caller doesn't need to.
851	*/
852	PRTZIPTARFSSTREAMWRITERPUSH pPush = pThis->pPush;
853	if (!pPush)
854	{
855	AssertRC(pThis->rcFatal);
856	return pThis->rcFatal;
857	}
858
859	pThis->pPush = NULL;
860	pPush->pParent = NULL;
861
862	int rc = pThis->rcFatal;
863	AssertRCReturn(rc, rc);
864
865	/*
866	* Do we need to update the header. pThis->aHdrs[0] will retain the current
867	* content at pPush->offHdr and we only need to update the size.
868	*/
869	if (pPush->fOpenEnded)
870	{
871	rc = rtZipTarFssWriter_FormatOffset(pThis->aHdrs[0].Common.size, pPush->cbCurrent);
872	if (RT_SUCCESS(rc))
873	rc = rtZipTarFssWriter_ChecksumHdr(&pThis->aHdrs[0]);
874	if (RT_SUCCESS(rc))
875	{
876	rc = RTVfsFileWriteAt(pThis->hVfsFile, pPush->offHdr, &pThis->aHdrs[0], sizeof(pThis->aHdrs[0]), NULL);
877	if (RT_SUCCESS(rc))
878	rc = RTVfsFileSeek(pThis->hVfsFile, pPush->offData + pPush->cbCurrent, RTFILE_SEEK_BEGIN, NULL);
879	}
880	}
881	/*
882	* Check that we've received all the data we were promissed in the PushFile
883	* call, fail if we weren't.
884	*/
885	else
886	AssertMsgStmt(pPush->cbCurrent == pPush->cbExpected,
887	("cbCurrent=%#RX64 cbExpected=%#RX64\n", pPush->cbCurrent, pPush->cbExpected),
888	rc = VERR_BUFFER_UNDERFLOW);
889	if (RT_SUCCESS(rc))
890	{
891	/*
892	* Do zero padding if necessary.
893	*/
894	if (pPush->cbCurrent & (RTZIPTAR_BLOCKSIZE - 1))
895	{
896	size_t cbToZero = RTZIPTAR_BLOCKSIZE - (pPush->cbCurrent & (RTZIPTAR_BLOCKSIZE - 1));
897	rc = RTVfsIoStrmWrite(pThis->hVfsIos, g_abRTZero4K, cbToZero, true /fBlocking/, NULL);
898	if (RT_SUCCESS(rc))
899	pThis->cbWritten += cbToZero;
900	}
901	}
902
903	if (RT_SUCCESS(rc))
904	return VINF_SUCCESS;
905	pThis->rcFatal = rc;
906	return rc;
907	}
908
909
910	/**
911	* Allocates a buffer for transfering file data.
912	*
913	* @note Will use the 3rd TAR header as fallback buffer if we're out of
914	* memory!
915	*
916	* @returns Pointer to buffer (won't ever fail).
917	* @param pThis The TAR writer instance.
918	* @param pcbBuf Where to return the buffer size. This will be a
919	* multiple of the TAR block size.
920	* @param ppvFree Where to return the pointer to pass to RTMemTmpFree
921	* when done with the buffer.
922	* @param cbFile The file size. Used as a buffer size hint.
923	*/
924	static uint8_t rtZipTarFssWrite_AllocBuf(PRTZIPTARFSSTREAMWRITER pThis, size_t pcbBuf, void **ppvFree, uint64_t cbObject)
925	{
926	uint8_t *pbBuf;
927
928	/*
929	* If this is a large file, try for a large buffer with 16KB alignment.
930	*/
931	if (cbObject >= _64M)
932	{
933	pbBuf = (uint8_t *)RTMemTmpAlloc(_2M + _16K - 1);
934	if (pbBuf)
935	{
936	*pcbBuf = _2M;
937	*ppvFree = pbBuf;
938	return RT_ALIGN_PT(pbBuf, _16K, uint8_t *);
939	}
940	}
941	/*
942	* 4KB aligned 512KB buffer if larger 512KB or larger.
943	*/
944	else if (cbObject >= _512K)
945	{
946	pbBuf = (uint8_t *)RTMemTmpAlloc(_512K + _4K - 1);
947	if (pbBuf)
948	{
949	*pcbBuf = _512K;
950	*ppvFree = pbBuf;
951	return RT_ALIGN_PT(pbBuf, _4K, uint8_t *);
952	}
953	}
954	/*
955	* Otherwise a 4KB aligned 128KB buffer.
956	*/
957	else
958	{
959	pbBuf = (uint8_t *)RTMemTmpAlloc(_128K + _4K - 1);
960	if (pbBuf)
961	{
962	*pcbBuf = _128K;
963	*ppvFree = pbBuf;
964	return RT_ALIGN_PT(pbBuf, _4K, uint8_t *);
965	}
966	}
967
968	/*
969	* If allocation failed, fallback on a 16KB buffer without any extra alignment.
970	*/
971	pbBuf = (uint8_t *)RTMemTmpAlloc(_16K);
972	if (pbBuf)
973	{
974	*pcbBuf = _16K;
975	*ppvFree = pbBuf;
976	return pbBuf;
977	}
978
979	/*
980	* Final fallback, 512KB buffer using the 3rd header.
981	*/
982	AssertCompile(RT_ELEMENTS(pThis->aHdrs) >= 3);
983	*pcbBuf = sizeof(pThis->aHdrs[2]);
984	*ppvFree = NULL;
985	return (uint8_t *)&pThis->aHdrs[2];
986	}
987
988
989	/**
990	* Frees the sparse info for a TAR file.
991	*
992	* @param pSparse The sparse info to free.
993	*/
994	static void rtZipTarFssWriter_SparseInfoDestroy(PRTZIPTARSPARSE pSparse)
995	{
996	PRTZIPTARSPARSECHUNK pCur;
997	PRTZIPTARSPARSECHUNK pNext;
998	RTListForEachSafe(&pSparse->ChunkHead, pCur, pNext, RTZIPTARSPARSECHUNK, Entry)
999	RTMemTmpFree(pCur);
1000	RTMemTmpFree(pSparse);
1001	}
1002
1003
1004	/**
1005	* Adds a data span to the sparse info.
1006	*
1007	* @returns VINF_SUCCESS or VINF_NO_TMP_MEMORY.
1008	* @param pSparse The sparse info to free.
1009	* @param offSpan Offset of the span.
1010	* @param cbSpan Number of bytes.
1011	*/
1012	static int rtZipTarFssWriter_SparseInfoAddSpan(PRTZIPTARSPARSE pSparse, uint64_t offSpan, uint64_t cbSpan)
1013	{
1014	/*
1015	* Get the chunk we're adding it to.
1016	*/
1017	PRTZIPTARSPARSECHUNK pChunk;
1018	if (pSparse->iNextSpan != 0)
1019	{
1020	pChunk = RTListGetLast(&pSparse->ChunkHead, RTZIPTARSPARSECHUNK, Entry);
1021	Assert(pSparse->iNextSpan < RT_ELEMENTS(pChunk->aSpans));
1022	}
1023	else
1024	{
1025	pChunk = (PRTZIPTARSPARSECHUNK)RTMemTmpAllocZ(sizeof(*pChunk));
1026	if (!pChunk)
1027	return VERR_NO_TMP_MEMORY;
1028	RTListAppend(&pSparse->ChunkHead, &pChunk->Entry);
1029	}
1030
1031	/*
1032	* Append it.
1033	*/
1034	pSparse->cDataSpans += 1;
1035	pSparse->cbDataSpans += cbSpan;
1036	pChunk->aSpans[pSparse->iNextSpan].cb = cbSpan;
1037	pChunk->aSpans[pSparse->iNextSpan].off = offSpan;
1038	if (++pSparse->iNextSpan >= RT_ELEMENTS(pChunk->aSpans))
1039	pSparse->iNextSpan = 0;
1040	return VINF_SUCCESS;
1041	}
1042
1043
1044	/**
1045	* Scans the input stream recording non-zero blocks.
1046	*/
1047	static int rtZipTarFssWriter_ScanSparseFile(PRTZIPTARFSSTREAMWRITER pThis, RTVFSFILE hVfsFile, uint64_t cbFile,
1048	size_t cbBuf, uint8_t pbBuf, PRTZIPTARSPARSE ppSparse)
1049	{
1050	RT_NOREF(pThis);
1051
1052	/*
1053	* Create an empty sparse info bundle.
1054	*/
1055	PRTZIPTARSPARSE pSparse = (PRTZIPTARSPARSE)RTMemTmpAlloc(sizeof(*pSparse));
1056	AssertReturn(pSparse, VERR_NO_MEMORY);
1057	pSparse->cbDataSpans = 0;
1058	pSparse->cDataSpans = 0;
1059	pSparse->iNextSpan = 0;
1060	RTListInit(&pSparse->ChunkHead);
1061
1062	/*
1063	* Scan the file from the start.
1064	*/
1065	int rc = RTVfsFileSeek(hVfsFile, 0, RTFILE_SEEK_BEGIN, NULL);
1066	if (RT_SUCCESS(rc))
1067	{
1068	bool fZeroSpan = false;
1069	uint64_t offSpan = 0;
1070	uint64_t cbSpan = 0;
1071
1072	for (uint64_t off = 0; off < cbFile;)
1073	{
1074	uint64_t cbLeft = cbFile - off;
1075	size_t cbToRead = cbLeft >= cbBuf ? cbBuf : (size_t)cbLeft;
1076	rc = RTVfsFileRead(hVfsFile, pbBuf, cbToRead, NULL);
1077	if (RT_FAILURE(rc))
1078	break;
1079	size_t cBlocks = cbToRead / RTZIPTAR_BLOCKSIZE;
1080
1081	/* Zero pad the final buffer to a multiple of the blocksize. */
1082	if (!(cbToRead & (RTZIPTAR_BLOCKSIZE - 1)))
1083	{ /* likely */ }
1084	else
1085	{
1086	AssertBreakStmt(cbLeft == cbToRead, rc = VERR_INTERNAL_ERROR_3);
1087	RT_BZERO(&pbBuf[cbToRead], RTZIPTAR_BLOCKSIZE - (cbToRead & (RTZIPTAR_BLOCKSIZE - 1)));
1088	cBlocks++;
1089	}
1090
1091	/*
1092	* Process the blocks we've just read one by one.
1093	*/
1094	uint8_t const *pbBlock = pbBuf;
1095	for (size_t iBlock = 0; iBlock < cBlocks; iBlock++)
1096	{
1097	bool fZeroBlock = ASMMemIsZero(pbBlock, RTZIPTAR_BLOCKSIZE);
1098	if (fZeroBlock == fZeroSpan)
1099	cbSpan += RTZIPTAR_BLOCKSIZE;
1100	else
1101	{
1102	if (!fZeroSpan && cbSpan)
1103	{
1104	rc = rtZipTarFssWriter_SparseInfoAddSpan(pSparse, offSpan, cbSpan);
1105	if (RT_FAILURE(rc))
1106	break;
1107	}
1108	fZeroSpan = fZeroBlock;
1109	offSpan = off;
1110	cbSpan = RTZIPTAR_BLOCKSIZE;
1111	}
1112
1113	/* next block. */
1114	pbBlock += RTZIPTAR_BLOCKSIZE;
1115	off += RTZIPTAR_BLOCKSIZE;
1116	}
1117	}
1118
1119	/*
1120	* Deal with the final span. If we've got zeros thowards the end, we
1121	* must add a zero byte data span at the end.
1122	*/
1123	if (RT_SUCCESS(rc))
1124	{
1125	if (!fZeroSpan && cbSpan)
1126	{
1127	if (cbFile & (RTZIPTAR_BLOCKSIZE - 1))
1128	{
1129	Assert(!(cbSpan & (RTZIPTAR_BLOCKSIZE - 1)));
1130	cbSpan -= RTZIPTAR_BLOCKSIZE;
1131	cbSpan \|= cbFile & (RTZIPTAR_BLOCKSIZE - 1);
1132	}
1133	rc = rtZipTarFssWriter_SparseInfoAddSpan(pSparse, offSpan, cbSpan);
1134	}
1135	if (RT_SUCCESS(rc))
1136	rc = rtZipTarFssWriter_SparseInfoAddSpan(pSparse, cbFile, 0);
1137	}
1138	}
1139
1140	if (RT_SUCCESS(rc))
1141	{
1142	/*
1143	* Return the file back to the start position before we return so that we
1144	* can segue into the regular rtZipTarFssWriter_AddFile without further ado.
1145	*/
1146	rc = RTVfsFileSeek(hVfsFile, 0, RTFILE_SEEK_BEGIN, NULL);
1147	if (RT_SUCCESS(rc))
1148	{
1149	*ppSparse = pSparse;
1150	return VINF_SUCCESS;
1151	}
1152	}
1153
1154	rtZipTarFssWriter_SparseInfoDestroy(pSparse);
1155	*ppSparse = NULL;
1156	return rc;
1157	}
1158
1159
1160	/**
1161	* Writes GNU the sparse file headers.
1162	*
1163	* @returns IPRT status code.
1164	* @param pThis The TAR writer instance.
1165	* @param pszPath The path to the file.
1166	* @param pObjInfo The object information.
1167	* @param pszOwnerNm The owner name.
1168	* @param pszGroupNm The group name.
1169	* @param pSparse The sparse file info.
1170	*/
1171	static int rtZipTarFssWriter_WriteGnuSparseHeaders(PRTZIPTARFSSTREAMWRITER pThis, const char *pszPath, PCRTFSOBJINFO pObjInfo,
1172	const char pszOwnerNm, const char pszGroupNm, PCRTZIPTARSPARSE pSparse)
1173	{
1174	/*
1175	* Format the first header.
1176	*/
1177	int rc = rtZipTarFssWriter_ObjInfoToHdr(pThis, pszPath, pObjInfo, pszOwnerNm, pszGroupNm, RTZIPTAR_TF_GNU_SPARSE);
1178	AssertRCReturn(rc, rc);
1179	AssertReturn(pThis->cHdrs == 1, VERR_INTERNAL_ERROR_2);
1180
1181	/* data size. */
1182	rc = rtZipTarFssWriter_FormatOffset(pThis->aHdrs[0].Common.size, pSparse->cbDataSpans);
1183	AssertRCReturn(rc, rc);
1184
1185	/* realsize. */
1186	rc = rtZipTarFssWriter_FormatOffset(pThis->aHdrs[0].Gnu.realsize, pObjInfo->cbObject);
1187	AssertRCReturn(rc, rc);
1188
1189	Assert(pThis->aHdrs[0].Gnu.isextended == 0);
1190
1191	/*
1192	* Walk the sparse spans, fill and write headers one by one.
1193	*/
1194	PRTZIPTARGNUSPARSE paSparse = &pThis->aHdrs[0].Gnu.sparse[0];
1195	uint32_t cSparse = RT_ELEMENTS(pThis->aHdrs[0].Gnu.sparse);
1196	uint32_t iSparse = 0;
1197
1198	PRTZIPTARSPARSECHUNK const pLastChunk = RTListGetLast(&pSparse->ChunkHead, RTZIPTARSPARSECHUNK, Entry);
1199	PRTZIPTARSPARSECHUNK pChunk;
1200	RTListForEach(&pSparse->ChunkHead, pChunk, RTZIPTARSPARSECHUNK, Entry)
1201	{
1202	uint32_t cSpans = pChunk != pLastChunk \|\| pSparse->iNextSpan == 0
1203	? RT_ELEMENTS(pChunk->aSpans) : pSparse->iNextSpan;
1204	for (uint32_t iSpan = 0; iSpan < cSpans; iSpan++)
1205	{
1206	/* Flush the header? */
1207	if (iSparse >= cSparse)
1208	{
1209	if (cSparse != RT_ELEMENTS(pThis->aHdrs[0].Gnu.sparse))
1210	pThis->aHdrs[0].GnuSparse.isextended = 1; /* more headers to come */
1211	else
1212	{
1213	pThis->aHdrs[0].Gnu.isextended = 1; /* more headers to come */
1214	rc = rtZipTarFssWriter_ChecksumHdr(&pThis->aHdrs[0]);
1215	}
1216	if (RT_SUCCESS(rc))
1217	rc = RTVfsIoStrmWrite(pThis->hVfsIos, &pThis->aHdrs[0], sizeof(pThis->aHdrs[0]), true /fBlocking/, NULL);
1218	if (RT_FAILURE(rc))
1219	return rc;
1220	RT_ZERO(pThis->aHdrs[0]);
1221	cSparse = RT_ELEMENTS(pThis->aHdrs[0].GnuSparse.sp);
1222	iSparse = 0;
1223	paSparse = &pThis->aHdrs[0].GnuSparse.sp[0];
1224	}
1225
1226	/* Append sparse data segment. */
1227	rc = rtZipTarFssWriter_FormatOffset(paSparse[iSparse].offset, pChunk->aSpans[iSpan].off);
1228	AssertRCReturn(rc, rc);
1229	rc = rtZipTarFssWriter_FormatOffset(paSparse[iSparse].numbytes, pChunk->aSpans[iSpan].cb);
1230	AssertRCReturn(rc, rc);
1231	iSparse++;
1232	}
1233	}
1234
1235	/*
1236	* The final header.
1237	*/
1238	if (iSparse != 0)
1239	{
1240	if (cSparse != RT_ELEMENTS(pThis->aHdrs[0].Gnu.sparse))
1241	Assert(pThis->aHdrs[0].GnuSparse.isextended == 0);
1242	else
1243	{
1244	Assert(pThis->aHdrs[0].Gnu.isextended == 0);
1245	rc = rtZipTarFssWriter_ChecksumHdr(&pThis->aHdrs[0]);
1246	}
1247	if (RT_SUCCESS(rc))
1248	rc = RTVfsIoStrmWrite(pThis->hVfsIos, &pThis->aHdrs[0], sizeof(pThis->aHdrs[0]), true /fBlocking/, NULL);
1249	}
1250	pThis->cHdrs = 0;
1251	return rc;
1252	}
1253
1254
1255	/**
1256	* Adds a potentially sparse file to the output.
1257	*
1258	* @returns IPRT status code.
1259	* @param pThis The TAR writer instance.
1260	* @param pszPath The path to the file.
1261	* @param hVfsFile The potentially sparse file.
1262	* @param hVfsIos The I/O stream of the file. Same as @a hVfsFile.
1263	* @param pObjInfo The object information.
1264	* @param pszOwnerNm The owner name.
1265	* @param pszGroupNm The group name.
1266	*/
1267	static int rtZipTarFssWriter_AddFileSparse(PRTZIPTARFSSTREAMWRITER pThis, const char *pszPath, RTVFSFILE hVfsFile,
1268	RTVFSIOSTREAM hVfsIos, PCRTFSOBJINFO pObjInfo,
1269	const char pszOwnerNm, const char pszGroupNm)
1270	{
1271	/*
1272	* Scan the input file to locate all zero blocks.
1273	*/
1274	void *pvBufFree;
1275	size_t cbBuf;
1276	uint8_t *pbBuf = rtZipTarFssWrite_AllocBuf(pThis, &cbBuf, &pvBufFree, pObjInfo->cbObject);
1277
1278	PRTZIPTARSPARSE pSparse;
1279	int rc = rtZipTarFssWriter_ScanSparseFile(pThis, hVfsFile, pObjInfo->cbObject, cbBuf, pbBuf, &pSparse);
1280	if (RT_SUCCESS(rc))
1281	{
1282	/*
1283	* If there aren't at least 2 zero blocks in the file, don't bother
1284	* doing the sparse stuff and store it as a normal file.
1285	*/
1286	if (pSparse->cbDataSpans + RTZIPTAR_BLOCKSIZE > (uint64_t)pObjInfo->cbObject)
1287	{
1288	rtZipTarFssWriter_SparseInfoDestroy(pSparse);
1289	RTMemTmpFree(pvBufFree);
1290	return rtZipTarFssWriter_AddFile(pThis, pszPath, hVfsIos, pObjInfo, pszOwnerNm, pszGroupNm);
1291	}
1292
1293	/*
1294	* Produce and write the headers.
1295	*/
1296	if (pThis->enmFormat == RTZIPTARFORMAT_GNU)
1297	rc = rtZipTarFssWriter_WriteGnuSparseHeaders(pThis, pszPath, pObjInfo, pszOwnerNm, pszGroupNm, pSparse);
1298	else
1299	AssertStmt(pThis->enmFormat != RTZIPTARFORMAT_GNU, rc = VERR_NOT_IMPLEMENTED);
1300	if (RT_SUCCESS(rc))
1301	{
1302	/*
1303	* Write the file bytes.
1304	*/
1305	PRTZIPTARSPARSECHUNK const pLastChunk = RTListGetLast(&pSparse->ChunkHead, RTZIPTARSPARSECHUNK, Entry);
1306	PRTZIPTARSPARSECHUNK pChunk;
1307	RTListForEach(&pSparse->ChunkHead, pChunk, RTZIPTARSPARSECHUNK, Entry)
1308	{
1309	uint32_t cSpans = pChunk != pLastChunk \|\| pSparse->iNextSpan == 0
1310	? RT_ELEMENTS(pChunk->aSpans) : pSparse->iNextSpan;
1311	for (uint32_t iSpan = 0; iSpan < cSpans; iSpan++)
1312	{
1313	rc = RTVfsFileSeek(hVfsFile, pChunk->aSpans[iSpan].off, RTFILE_SEEK_BEGIN, NULL);
1314	if (RT_FAILURE(rc))
1315	break;
1316	uint64_t cbLeft = pChunk->aSpans[iSpan].cb;
1317	Assert( !(cbLeft & (RTZIPTAR_BLOCKSIZE - 1))
1318	\|\| (iSpan + 1 == cSpans && pChunk == pLastChunk));
1319	while (cbLeft > 0)
1320	{
1321	size_t cbToRead = cbLeft >= cbBuf ? cbBuf : (size_t)cbLeft;
1322	rc = RTVfsFileRead(hVfsFile, pbBuf, cbToRead, NULL);
1323	if (RT_SUCCESS(rc))
1324	{
1325	rc = RTVfsIoStrmWrite(pThis->hVfsIos, pbBuf, cbToRead, true /fBlocking/, NULL);
1326	if (RT_SUCCESS(rc))
1327	{
1328	pThis->cbWritten += cbToRead;
1329	cbLeft -= cbToRead;
1330	continue;
1331	}
1332	}
1333	break;
1334	}
1335	if (RT_FAILURE(rc))
1336	break;
1337	}
1338	}
1339
1340	/*
1341	* Do the zero padding.
1342	*/
1343	if ( RT_SUCCESS(rc)
1344	&& (pSparse->cbDataSpans & (RTZIPTAR_BLOCKSIZE - 1)))
1345	{
1346	size_t cbToZero = RTZIPTAR_BLOCKSIZE - (pSparse->cbDataSpans & (RTZIPTAR_BLOCKSIZE - 1));
1347	rc = RTVfsIoStrmWrite(pThis->hVfsIos, g_abRTZero4K, cbToZero, true /fBlocking/, NULL);
1348	if (RT_SUCCESS(rc))
1349	pThis->cbWritten += cbToZero;
1350	}
1351	}
1352
1353	if (RT_FAILURE(rc))
1354	pThis->rcFatal = rc;
1355	rtZipTarFssWriter_SparseInfoDestroy(pSparse);
1356	}
1357	RTMemTmpFree(pvBufFree);
1358	return rc;
1359	}
1360
1361
1362	/**
1363	* Adds an I/O stream of indeterminate length to the TAR file.
1364	*
1365	* This requires the output to be seekable, i.e. a file, because we need to go
1366	* back and update @c size field of the TAR header after pumping all the data
1367	* bytes thru and establishing the file length.
1368	*
1369	* @returns IPRT status code.
1370	* @param pThis The TAR writer instance.
1371	* @param pszPath The path to the file.
1372	* @param hVfsIos The I/O stream of the file.
1373	* @param pObjInfo The object information.
1374	* @param pszOwnerNm The owner name.
1375	* @param pszGroupNm The group name.
1376	*/
1377	static int rtZipTarFssWriter_AddFileStream(PRTZIPTARFSSTREAMWRITER pThis, const char *pszPath, RTVFSIOSTREAM hVfsIos,
1378	PCRTFSOBJINFO pObjInfo, const char pszOwnerNm, const char pszGroupNm)
1379	{
1380	AssertReturn(pThis->hVfsFile != NIL_RTVFSFILE, VERR_NOT_A_FILE);
1381
1382	/*
1383	* Append the header.
1384	*/
1385	int rc = rtZipTarFssWriter_ObjInfoToHdr(pThis, pszPath, pObjInfo, pszOwnerNm, pszGroupNm, UINT8_MAX);
1386	if (RT_SUCCESS(rc))
1387	{
1388	RTFOFF const offHdr = RTVfsFileTell(pThis->hVfsFile);
1389	if (offHdr >= 0)
1390	{
1391	rc = RTVfsIoStrmWrite(pThis->hVfsIos, pThis->aHdrs, pThis->cHdrs * sizeof(pThis->aHdrs[0]), true /fBlocking/, NULL);
1392	if (RT_SUCCESS(rc))
1393	{
1394	pThis->cbWritten += pThis->cHdrs * sizeof(pThis->aHdrs[0]);
1395
1396	/*
1397	* Transfer the bytes.
1398	*/
1399	void *pvBufFree;
1400	size_t cbBuf;
1401	uint8_t *pbBuf = rtZipTarFssWrite_AllocBuf(pThis, &cbBuf, &pvBufFree,
1402	pObjInfo->cbObject > 0 && pObjInfo->cbObject != RTFOFF_MAX
1403	? pObjInfo->cbObject : _1G);
1404
1405	uint64_t cbReadTotal = 0;
1406	for (;;)
1407	{
1408	size_t cbRead = 0;
1409	int rc2 = rc = RTVfsIoStrmRead(hVfsIos, pbBuf, cbBuf, true /fBlocking/, &cbRead);
1410	if (RT_SUCCESS(rc))
1411	{
1412	cbReadTotal += cbRead;
1413	rc = RTVfsIoStrmWrite(pThis->hVfsIos, pbBuf, cbRead, true /fBlocking/, NULL);
1414	if (RT_SUCCESS(rc))
1415	{
1416	pThis->cbWritten += cbRead;
1417	if (rc2 != VINF_EOF)
1418	continue;
1419	}
1420	}
1421	Assert(rc != VERR_EOF /* expecting VINF_EOF! */);
1422	break;
1423	}
1424
1425	RTMemTmpFree(pvBufFree);
1426
1427	/*
1428	* Do the zero padding.
1429	*/
1430	if ((cbReadTotal & (RTZIPTAR_BLOCKSIZE - 1)) && RT_SUCCESS(rc))
1431	{
1432	size_t cbToZero = RTZIPTAR_BLOCKSIZE - (cbReadTotal & (RTZIPTAR_BLOCKSIZE - 1));
1433	rc = RTVfsIoStrmWrite(pThis->hVfsIos, g_abRTZero4K, cbToZero, true /fBlocking/, NULL);
1434	if (RT_SUCCESS(rc))
1435	pThis->cbWritten += cbToZero;
1436	}
1437
1438	/*
1439	* Update the header. We ASSUME that aHdr[0] is unmodified
1440	* from before the data pumping above and just update the size.
1441	*/
1442	if ((RTFOFF)cbReadTotal != pObjInfo->cbObject && RT_SUCCESS(rc))
1443	{
1444	RTFOFF const offRestore = RTVfsFileTell(pThis->hVfsFile);
1445	if (offRestore >= 0)
1446	{
1447	rc = rtZipTarFssWriter_FormatOffset(pThis->aHdrs[0].Common.size, cbReadTotal);
1448	if (RT_SUCCESS(rc))
1449	rc = rtZipTarFssWriter_ChecksumHdr(&pThis->aHdrs[0]);
1450	if (RT_SUCCESS(rc))
1451	{
1452	rc = RTVfsFileWriteAt(pThis->hVfsFile, offHdr, &pThis->aHdrs[0], sizeof(pThis->aHdrs[0]), NULL);
1453	if (RT_SUCCESS(rc))
1454	rc = RTVfsFileSeek(pThis->hVfsFile, offRestore, RTFILE_SEEK_BEGIN, NULL);
1455	}
1456	}
1457	else
1458	rc = (int)offRestore;
1459	}
1460
1461	if (RT_SUCCESS(rc))
1462	return VINF_SUCCESS;
1463	}
1464	}
1465	else
1466	rc = (int)offHdr;
1467	pThis->rcFatal = rc;
1468	}
1469	return rc;
1470	}
1471
1472
1473	/**
1474	* Adds a file to the stream.
1475	*
1476	* @returns IPRT status code.
1477	* @param pThis The TAR writer instance.
1478	* @param pszPath The path to the file.
1479	* @param hVfsIos The I/O stream of the file.
1480	* @param fFlags The RTVFSFSSTREAMOPS::pfnAdd flags.
1481	* @param pObjInfo The object information.
1482	* @param pszOwnerNm The owner name.
1483	* @param pszGroupNm The group name.
1484	*/
1485	static int rtZipTarFssWriter_AddFile(PRTZIPTARFSSTREAMWRITER pThis, const char *pszPath, RTVFSIOSTREAM hVfsIos,
1486	PCRTFSOBJINFO pObjInfo, const char pszOwnerNm, const char pszGroupNm)
1487	{
1488	/*
1489	* Append the header.
1490	*/
1491	int rc = rtZipTarFssWriter_ObjInfoToHdr(pThis, pszPath, pObjInfo, pszOwnerNm, pszGroupNm, UINT8_MAX);
1492	if (RT_SUCCESS(rc))
1493	{
1494	rc = RTVfsIoStrmWrite(pThis->hVfsIos, pThis->aHdrs, pThis->cHdrs * sizeof(pThis->aHdrs[0]), true /fBlocking/, NULL);
1495	if (RT_SUCCESS(rc))
1496	{
1497	pThis->cbWritten += pThis->cHdrs * sizeof(pThis->aHdrs[0]);
1498
1499	/*
1500	* Copy the bytes. Padding the last buffer to a multiple of 512.
1501	*/
1502	void *pvBufFree;
1503	size_t cbBuf;
1504	uint8_t *pbBuf = rtZipTarFssWrite_AllocBuf(pThis, &cbBuf, &pvBufFree, pObjInfo->cbObject);
1505
1506	uint64_t cbLeft = pObjInfo->cbObject;
1507	while (cbLeft > 0)
1508	{
1509	size_t cbRead = cbLeft > cbBuf ? cbBuf : (size_t)cbLeft;
1510	rc = RTVfsIoStrmRead(hVfsIos, pbBuf, cbRead, true /fBlocking/, NULL);
1511	if (RT_FAILURE(rc))
1512	break;
1513
1514	size_t cbToWrite = cbRead;
1515	if (cbRead & (RTZIPTAR_BLOCKSIZE - 1))
1516	{
1517	size_t cbToZero = RTZIPTAR_BLOCKSIZE - (cbRead & (RTZIPTAR_BLOCKSIZE - 1));
1518	memset(&pbBuf[cbRead], 0, cbToZero);
1519	cbToWrite += cbToZero;
1520	}
1521
1522	rc = RTVfsIoStrmWrite(pThis->hVfsIos, pbBuf, cbToWrite, true /fBlocking/, NULL);
1523	if (RT_FAILURE(rc))
1524	break;
1525	pThis->cbWritten += cbToWrite;
1526	cbLeft -= cbRead;
1527	}
1528
1529	RTMemTmpFree(pvBufFree);
1530
1531	if (RT_SUCCESS(rc))
1532	return VINF_SUCCESS;
1533	}
1534	pThis->rcFatal = rc;
1535	}
1536	return rc;
1537	}
1538
1539
1540	/**
1541	* Adds a symbolic link to the stream.
1542	*
1543	* @returns IPRT status code.
1544	* @param pThis The TAR writer instance.
1545	* @param pszPath The path to the object.
1546	* @param hVfsSymlink The symbolic link object to add.
1547	* @param pObjInfo The object information.
1548	* @param pszOwnerNm The owner name.
1549	* @param pszGroupNm The group name.
1550	*/
1551	static int rtZipTarFssWriter_AddSymlink(PRTZIPTARFSSTREAMWRITER pThis, const char *pszPath, RTVFSSYMLINK hVfsSymlink,
1552	PCRTFSOBJINFO pObjInfo, const char pszOwnerNm, const char pszGroupNm)
1553	{
1554	/*
1555	* Read the symlink target first and check that it's not too long.
1556	* Flip DOS slashes.
1557	*/
1558	char szTarget[RTPATH_MAX];
1559	int rc = RTVfsSymlinkRead(hVfsSymlink, szTarget, sizeof(szTarget));
1560	if (RT_SUCCESS(rc))
1561	{
1562	#if RTPATH_STYLE != RTPATH_STR_F_STYLE_UNIX
1563	char *pszDosSlash = strchr(szTarget, '\\');
1564	while (pszDosSlash)
1565	{
1566	*pszDosSlash = '/';
1567	pszDosSlash = strchr(pszDosSlash + 1, '\\');
1568	}
1569	#endif
1570	size_t cchTarget = strlen(szTarget);
1571	if (cchTarget < sizeof(pThis->aHdrs[0].Common.linkname))
1572	{
1573	/*
1574	* Create a header, add the link target and push it out.
1575	*/
1576	rc = rtZipTarFssWriter_ObjInfoToHdr(pThis, pszPath, pObjInfo, pszOwnerNm, pszGroupNm, UINT8_MAX);
1577	if (RT_SUCCESS(rc))
1578	{
1579	memcpy(pThis->aHdrs[0].Common.linkname, szTarget, cchTarget + 1);
1580	rc = rtZipTarFssWriter_ChecksumHdr(&pThis->aHdrs[0]);
1581	if (RT_SUCCESS(rc))
1582	{
1583	rc = RTVfsIoStrmWrite(pThis->hVfsIos, pThis->aHdrs, pThis->cHdrs * sizeof(pThis->aHdrs[0]),
1584	true /fBlocking/, NULL);
1585	if (RT_SUCCESS(rc))
1586	{
1587	pThis->cbWritten += pThis->cHdrs * sizeof(pThis->aHdrs[0]);
1588	return VINF_SUCCESS;
1589	}
1590	pThis->rcFatal = rc;
1591	}
1592	}
1593	}
1594	else
1595	{
1596	/** @todo implement gnu and pax long name extensions. */
1597	rc = VERR_TAR_NAME_TOO_LONG;
1598	}
1599	}
1600	return rc;
1601	}
1602
1603
1604	/**
1605	* Adds a simple object to the stream.
1606	*
1607	* Simple objects only contains metadata, no actual data bits. Directories,
1608	* devices, fifos, sockets and such.
1609	*
1610	* @returns IPRT status code.
1611	* @param pThis The TAR writer instance.
1612	* @param pszPath The path to the object.
1613	* @param pObjInfo The object information.
1614	* @param pszOwnerNm The owner name.
1615	* @param pszGroupNm The group name.
1616	*/
1617	static int rtZipTarFssWriter_AddSimpleObject(PRTZIPTARFSSTREAMWRITER pThis, const char *pszPath, PCRTFSOBJINFO pObjInfo,
1618	const char pszOwnerNm, const char pszGroupNm)
1619	{
1620	int rc = rtZipTarFssWriter_ObjInfoToHdr(pThis, pszPath, pObjInfo, pszOwnerNm, pszGroupNm, UINT8_MAX);
1621	if (RT_SUCCESS(rc))
1622	{
1623	rc = RTVfsIoStrmWrite(pThis->hVfsIos, pThis->aHdrs, pThis->cHdrs * sizeof(pThis->aHdrs[0]), true /fBlocking/, NULL);
1624	if (RT_SUCCESS(rc))
1625	{
1626	pThis->cbWritten += pThis->cHdrs * sizeof(pThis->aHdrs[0]);
1627	return VINF_SUCCESS;
1628	}
1629	pThis->rcFatal = rc;
1630	}
1631	return rc;
1632	}
1633
1634
1635	/**
1636	* @interface_method_impl{RTVFSOBJOPS,pfnClose}
1637	*/
1638	static DECLCALLBACK(int) rtZipTarFssWriter_Close(void *pvThis)
1639	{
1640	PRTZIPTARFSSTREAMWRITER pThis = (PRTZIPTARFSSTREAMWRITER)pvThis;
1641
1642	rtZipTarFssWriter_CompleteCurrentPushFile(pThis);
1643
1644	RTVfsIoStrmRelease(pThis->hVfsIos);
1645	pThis->hVfsIos = NIL_RTVFSIOSTREAM;
1646
1647	if (pThis->hVfsFile != NIL_RTVFSFILE)
1648	{
1649	RTVfsFileRelease(pThis->hVfsFile);
1650	pThis->hVfsFile = NIL_RTVFSFILE;
1651	}
1652
1653	if (pThis->pszOwner)
1654	{
1655	RTStrFree(pThis->pszOwner);
1656	pThis->pszOwner = NULL;
1657	}
1658	if (pThis->pszGroup)
1659	{
1660	RTStrFree(pThis->pszGroup);
1661	pThis->pszGroup = NULL;
1662	}
1663	if (pThis->pszPrefix)
1664	{
1665	RTStrFree(pThis->pszPrefix);
1666	pThis->pszPrefix = NULL;
1667	}
1668
1669	return VINF_SUCCESS;
1670	}
1671
1672
1673	/**
1674	* @interface_method_impl{RTVFSOBJOPS,pfnQueryInfo}
1675	*/
1676	static DECLCALLBACK(int) rtZipTarFssWriter_QueryInfo(void *pvThis, PRTFSOBJINFO pObjInfo, RTFSOBJATTRADD enmAddAttr)
1677	{
1678	PRTZIPTARFSSTREAMWRITER pThis = (PRTZIPTARFSSTREAMWRITER)pvThis;
1679	/* Take the lazy approach here, with the sideffect of providing some info
1680	that is actually kind of useful. */
1681	return RTVfsIoStrmQueryInfo(pThis->hVfsIos, pObjInfo, enmAddAttr);
1682	}
1683
1684
1685	/**
1686	* @interface_method_impl{RTVFSFSSTREAMOPS,pfnAdd}
1687	*/
1688	static DECLCALLBACK(int) rtZipTarFssWriter_Add(void pvThis, const char pszPath, RTVFSOBJ hVfsObj, uint32_t fFlags)
1689	{
1690	PRTZIPTARFSSTREAMWRITER pThis = (PRTZIPTARFSSTREAMWRITER)pvThis;
1691
1692	/*
1693	* Before we continue we must complete any current push file and check rcFatal.
1694	*/
1695	int rc = rtZipTarFssWriter_CompleteCurrentPushFile(pThis);
1696	if (RT_FAILURE(rc))
1697	return rc;
1698
1699	/*
1700	* Query information about the object.
1701	*/
1702	RTFSOBJINFO ObjInfo;
1703	rc = RTVfsObjQueryInfo(hVfsObj, &ObjInfo, RTFSOBJATTRADD_UNIX);
1704	AssertRCReturn(rc, rc);
1705
1706	RTFSOBJINFO ObjOwnerName;
1707	rc = RTVfsObjQueryInfo(hVfsObj, &ObjOwnerName, RTFSOBJATTRADD_UNIX_OWNER);
1708	if (RT_FAILURE(rc) \|\| ObjOwnerName.Attr.u.UnixOwner.szName[0] == '\0')
1709	strcpy(ObjOwnerName.Attr.u.UnixOwner.szName, "someone");
1710
1711	RTFSOBJINFO ObjGrpName;
1712	rc = RTVfsObjQueryInfo(hVfsObj, &ObjGrpName, RTFSOBJATTRADD_UNIX_GROUP);
1713	if (RT_FAILURE(rc) \|\| ObjGrpName.Attr.u.UnixGroup.szName[0] == '\0')
1714	strcpy(ObjGrpName.Attr.u.UnixGroup.szName, "somegroup");
1715
1716	/*
1717	* Do type specific handling. File have several options and variations to
1718	* take into account, thus the mess.
1719	*/
1720	if (RTFS_IS_FILE(ObjInfo.Attr.fMode))
1721	{
1722	RTVFSIOSTREAM hVfsIos = RTVfsObjToIoStream(hVfsObj);
1723	AssertReturn(hVfsIos != NIL_RTVFSIOSTREAM, VERR_WRONG_TYPE);
1724
1725	if (fFlags & RTVFSFSSTRM_ADD_F_STREAM)
1726	rc = rtZipTarFssWriter_AddFileStream(pThis, pszPath, hVfsIos, &ObjInfo,
1727	ObjOwnerName.Attr.u.UnixOwner.szName, ObjGrpName.Attr.u.UnixOwner.szName);
1728	else if ( !(pThis->fFlags & RTZIPTAR_C_SPARSE)
1729	\|\| ObjInfo.cbObject < RTZIPTAR_MIN_SPARSE)
1730	rc = rtZipTarFssWriter_AddFile(pThis, pszPath, hVfsIos, &ObjInfo,
1731	ObjOwnerName.Attr.u.UnixOwner.szName, ObjGrpName.Attr.u.UnixOwner.szName);
1732	else
1733	{
1734	RTVFSFILE hVfsFile = RTVfsObjToFile(hVfsObj);
1735	if (hVfsFile != NIL_RTVFSFILE)
1736	{
1737	rc = rtZipTarFssWriter_AddFileSparse(pThis, pszPath, hVfsFile, hVfsIos, &ObjInfo,
1738	ObjOwnerName.Attr.u.UnixOwner.szName, ObjGrpName.Attr.u.UnixOwner.szName);
1739	RTVfsFileRelease(hVfsFile);
1740	}
1741	else
1742	rc = rtZipTarFssWriter_AddFile(pThis, pszPath, hVfsIos, &ObjInfo,
1743	ObjOwnerName.Attr.u.UnixOwner.szName, ObjGrpName.Attr.u.UnixOwner.szName);
1744	}
1745	RTVfsIoStrmRelease(hVfsIos);
1746	}
1747	else if (RTFS_IS_SYMLINK(ObjInfo.Attr.fMode))
1748	{
1749	RTVFSSYMLINK hVfsSymlink = RTVfsObjToSymlink(hVfsObj);
1750	AssertReturn(hVfsSymlink != NIL_RTVFSSYMLINK, VERR_WRONG_TYPE);
1751	rc = rtZipTarFssWriter_AddSymlink(pThis, pszPath, hVfsSymlink, &ObjInfo,
1752	ObjOwnerName.Attr.u.UnixOwner.szName, ObjGrpName.Attr.u.UnixOwner.szName);
1753	RTVfsSymlinkRelease(hVfsSymlink);
1754	}
1755	else
1756	rc = rtZipTarFssWriter_AddSimpleObject(pThis, pszPath, &ObjInfo,
1757	ObjOwnerName.Attr.u.UnixOwner.szName, ObjGrpName.Attr.u.UnixOwner.szName);
1758
1759	return rc;
1760	}
1761
1762
1763	/**
1764	* @interface_method_impl{RTVFSFSSTREAMOPS,pfnPushFile}
1765	*/
1766	static DECLCALLBACK(int) rtZipTarFssWriter_PushFile(void pvThis, const char pszPath, uint64_t cbFile, PCRTFSOBJINFO paObjInfo,
1767	uint32_t cObjInfo, uint32_t fFlags, PRTVFSIOSTREAM phVfsIos)
1768	{
1769	PRTZIPTARFSSTREAMWRITER pThis = (PRTZIPTARFSSTREAMWRITER)pvThis;
1770
1771	/*
1772	* We can only deal with output of indeterminate length if the output is
1773	* seekable (see also rtZipTarFssWriter_AddFileStream).
1774	*/
1775	AssertReturn(cbFile != UINT64_MAX \|\| pThis->hVfsFile != NIL_RTVFSFILE, VERR_NOT_A_FILE);
1776	AssertReturn(RT_BOOL(cbFile == UINT64_MAX) == RT_BOOL(fFlags & RTVFSFSSTRM_ADD_F_STREAM), VERR_INVALID_FLAGS);
1777
1778	/*
1779	* Before we continue we must complete any current push file and check rcFatal.
1780	*/
1781	int rc = rtZipTarFssWriter_CompleteCurrentPushFile(pThis);
1782	if (RT_FAILURE(rc))
1783	return rc;
1784
1785	/*
1786	* If no object info was provideded, fake up some.
1787	*/
1788	const char *pszOwnerNm = "someone";
1789	const char *pszGroupNm = "somegroup";
1790	RTFSOBJINFO ObjInfo;
1791	if (cObjInfo == 0)
1792	{
1793	/* Fake up a info. */
1794	RT_ZERO(ObjInfo);
1795	ObjInfo.cbObject = cbFile != UINT64_MAX ? cbFile : 0;
1796	ObjInfo.cbAllocated = cbFile != UINT64_MAX ? RT_ALIGN_64(cbFile, RTZIPTAR_BLOCKSIZE) : UINT64_MAX;
1797	RTTimeNow(&ObjInfo.ModificationTime);
1798	ObjInfo.BirthTime = ObjInfo.ModificationTime;
1799	ObjInfo.ChangeTime = ObjInfo.ModificationTime;
1800	ObjInfo.AccessTime = ObjInfo.ModificationTime;
1801	ObjInfo.Attr.fMode = RTFS_TYPE_FILE \| 0666;
1802	ObjInfo.Attr.enmAdditional = RTFSOBJATTRADD_UNIX;
1803	ObjInfo.Attr.u.Unix.uid = NIL_RTUID;
1804	ObjInfo.Attr.u.Unix.gid = NIL_RTGID;
1805	ObjInfo.Attr.u.Unix.cHardlinks = 1;
1806	//ObjInfo.Attr.u.Unix.INodeIdDevice = 0;
1807	//ObjInfo.Attr.u.Unix.INodeId = 0;
1808	//ObjInfo.Attr.u.Unix.fFlags = 0;
1809	//ObjInfo.Attr.u.Unix.GenerationId = 0;
1810	//ObjInfo.Attr.u.Unix.Device = 0;
1811	}
1812	else
1813	{
1814	/* Make a copy of the object info and adjust the size, if necessary. */
1815	ObjInfo = paObjInfo[0];
1816	Assert(ObjInfo.Attr.enmAdditional == RTFSOBJATTRADD_UNIX);
1817	Assert(RTFS_IS_FILE(ObjInfo.Attr.fMode));
1818	if ((uint64_t)ObjInfo.cbObject != cbFile)
1819	{
1820	ObjInfo.cbObject = cbFile != UINT64_MAX ? cbFile : 0;
1821	ObjInfo.cbAllocated = cbFile != UINT64_MAX ? RT_ALIGN_64(cbFile, RTZIPTAR_BLOCKSIZE) : UINT64_MAX;
1822	}
1823
1824	/* Lookup the group and user names. */
1825	for (uint32_t i = 0; i < cObjInfo; i++)
1826	if ( paObjInfo[i].Attr.enmAdditional == RTFSOBJATTRADD_UNIX_OWNER
1827	&& paObjInfo[i].Attr.u.UnixOwner.szName[0] != '\0')
1828	pszOwnerNm = paObjInfo[i].Attr.u.UnixOwner.szName;
1829	else if ( paObjInfo[i].Attr.enmAdditional == RTFSOBJATTRADD_UNIX_GROUP
1830	&& paObjInfo[i].Attr.u.UnixGroup.szName[0] != '\0')
1831	pszGroupNm = paObjInfo[i].Attr.u.UnixGroup.szName;
1832	}
1833
1834	/*
1835	* Create an I/O stream object for the caller to use.
1836	*/
1837	RTFOFF const offHdr = RTVfsIoStrmTell(pThis->hVfsIos);
1838	AssertReturn(offHdr >= 0, (int)offHdr);
1839
1840	PRTZIPTARFSSTREAMWRITERPUSH pPush;
1841	RTVFSIOSTREAM hVfsIos;
1842	if (pThis->hVfsFile == NIL_RTVFSFILE)
1843	{
1844	rc = RTVfsNewIoStream(&g_rtZipTarWriterIoStrmOps, sizeof(*pPush), RTFILE_O_WRITE, NIL_RTVFS, NIL_RTVFSLOCK,
1845	&hVfsIos, (void **)&pPush);
1846	if (RT_FAILURE(rc))
1847	return rc;
1848	}
1849	else
1850	{
1851	RTVFSFILE hVfsFile;
1852	rc = RTVfsNewFile(&g_rtZipTarWriterFileOps, sizeof(*pPush), RTFILE_O_WRITE, NIL_RTVFS, NIL_RTVFSLOCK,
1853	&hVfsFile, (void **)&pPush);
1854	if (RT_FAILURE(rc))
1855	return rc;
1856	hVfsIos = RTVfsFileToIoStream(hVfsFile);
1857	RTVfsFileRelease(hVfsFile);
1858	}
1859	pPush->pParent = NULL;
1860	pPush->cbExpected = cbFile;
1861	pPush->offHdr = (uint64_t)offHdr;
1862	pPush->offData = 0;
1863	pPush->offCurrent = 0;
1864	pPush->cbCurrent = 0;
1865	pPush->ObjInfo = ObjInfo;
1866	pPush->fOpenEnded = cbFile == UINT64_MAX;
1867
1868	/*
1869	* Produce and write file headers.
1870	*/
1871	rc = rtZipTarFssWriter_ObjInfoToHdr(pThis, pszPath, &ObjInfo, pszOwnerNm, pszGroupNm, RTZIPTAR_TF_NORMAL);
1872	if (RT_SUCCESS(rc))
1873	{
1874	size_t cbHdrs = pThis->cHdrs * sizeof(pThis->aHdrs[0]);
1875	rc = RTVfsIoStrmWrite(pThis->hVfsIos, pThis->aHdrs, cbHdrs, true /fBlocking/, NULL);
1876	if (RT_SUCCESS(rc))
1877	{
1878	pThis->cbWritten += cbHdrs;
1879
1880	/*
1881	* Complete the object and return.
1882	*/
1883	pPush->offData = pPush->offHdr + cbHdrs;
1884	if (cbFile == UINT64_MAX)
1885	pPush->cbExpected = (uint64_t)(RTFOFF_MAX - _4K) - pPush->offData;
1886	pPush->pParent = pThis;
1887	pThis->pPush = pPush;
1888
1889	*phVfsIos = hVfsIos;
1890	return VINF_SUCCESS;
1891	}
1892	pThis->rcFatal = rc;
1893	}
1894
1895	RTVfsIoStrmRelease(hVfsIos);
1896	return rc;
1897	}
1898
1899
1900	/**
1901	* @interface_method_impl{RTVFSFSSTREAMOPS,pfnEnd}
1902	*/
1903	static DECLCALLBACK(int) rtZipTarFssWriter_End(void *pvThis)
1904	{
1905	PRTZIPTARFSSTREAMWRITER pThis = (PRTZIPTARFSSTREAMWRITER)pvThis;
1906
1907	/*
1908	* Make sure to complete any pending push file and that rcFatal is fine.
1909	*/
1910	int rc = rtZipTarFssWriter_CompleteCurrentPushFile(pThis);
1911	if (RT_SUCCESS(rc))
1912	{
1913	/*
1914	* There are supposed to be two zero headers at the end of the archive.
1915	* GNU tar may write more because of the way it does buffering,
1916	* libarchive OTOH writes exactly two.
1917	*/
1918	rc = RTVfsIoStrmWrite(pThis->hVfsIos, g_abRTZero4K, RTZIPTAR_BLOCKSIZE * 2, true /fBlocking/, NULL);
1919	if (RT_SUCCESS(rc))
1920	{
1921	pThis->cbWritten += RTZIPTAR_BLOCKSIZE * 2;
1922
1923	/*
1924	* Flush the output.
1925	*/
1926	rc = RTVfsIoStrmFlush(pThis->hVfsIos);
1927	if (RT_SUCCESS(rc))
1928	return rc;
1929	}
1930	pThis->rcFatal = rc;
1931	}
1932	return rc;
1933	}
1934
1935
1936	/**
1937	* Tar filesystem stream operations.
1938	*/
1939	static const RTVFSFSSTREAMOPS g_rtZipTarFssOps =
1940	{
1941	{ /* Obj */
1942	RTVFSOBJOPS_VERSION,
1943	RTVFSOBJTYPE_FS_STREAM,
1944	"TarFsStreamWriter",
1945	rtZipTarFssWriter_Close,
1946	rtZipTarFssWriter_QueryInfo,
1947	RTVFSOBJOPS_VERSION
1948	},
1949	RTVFSFSSTREAMOPS_VERSION,
1950	0,
1951	NULL,
1952	rtZipTarFssWriter_Add,
1953	rtZipTarFssWriter_PushFile,
1954	rtZipTarFssWriter_End,
1955	RTVFSFSSTREAMOPS_VERSION
1956	};
1957
1958
1959	RTDECL(int) RTZipTarFsStreamToIoStream(RTVFSIOSTREAM hVfsIosOut, RTZIPTARFORMAT enmFormat,
1960	uint32_t fFlags, PRTVFSFSSTREAM phVfsFss)
1961	{
1962	/*
1963	* Input validation.
1964	*/
1965	AssertPtrReturn(phVfsFss, VERR_INVALID_HANDLE);
1966	*phVfsFss = NIL_RTVFSFSSTREAM;
1967	AssertPtrReturn(hVfsIosOut, VERR_INVALID_HANDLE);
1968	AssertReturn(enmFormat > RTZIPTARFORMAT_INVALID && enmFormat < RTZIPTARFORMAT_END, VERR_INVALID_PARAMETER);
1969	AssertReturn(!(fFlags & ~RTZIPTAR_C_VALID_MASK), VERR_INVALID_FLAGS);
1970
1971	if (enmFormat == RTZIPTARFORMAT_DEFAULT)
1972	enmFormat = RTZIPTARFORMAT_GNU;
1973	AssertReturn( enmFormat == RTZIPTARFORMAT_GNU
1974	\|\| enmFormat == RTZIPTARFORMAT_USTAR
1975	, VERR_NOT_IMPLEMENTED); /* Only implementing GNU and USTAR output at the moment. */
1976
1977	uint32_t cRefs = RTVfsIoStrmRetain(hVfsIosOut);
1978	AssertReturn(cRefs != UINT32_MAX, VERR_INVALID_HANDLE);
1979
1980	/*
1981	* Retain the input stream and create a new filesystem stream handle.
1982	*/
1983	PRTZIPTARFSSTREAMWRITER pThis;
1984	RTVFSFSSTREAM hVfsFss;
1985	int rc = RTVfsNewFsStream(&g_rtZipTarFssOps, sizeof(pThis), NIL_RTVFS, NIL_RTVFSLOCK, false /fReadOnly*/,
1986	&hVfsFss, (void **)&pThis);
1987	if (RT_SUCCESS(rc))
1988	{
1989	pThis->hVfsIos = hVfsIosOut;
1990	pThis->hVfsFile = RTVfsIoStrmToFile(hVfsIosOut);
1991
1992	pThis->enmFormat = enmFormat;
1993	pThis->fFlags = fFlags;
1994	pThis->rcFatal = VINF_SUCCESS;
1995
1996	pThis->uidOwner = NIL_RTUID;
1997	pThis->pszOwner = NULL;
1998	pThis->gidGroup = NIL_RTGID;
1999	pThis->pszGroup = NULL;
2000	pThis->pszPrefix = NULL;
2001	pThis->pModTime = NULL;
2002	pThis->fFileModeAndMask = ~(RTFMODE)0;
2003	pThis->fFileModeOrMask = 0;
2004	pThis->fDirModeAndMask = ~(RTFMODE)0;
2005	pThis->fDirModeOrMask = 0;
2006
2007	*phVfsFss = hVfsFss;
2008	return VINF_SUCCESS;
2009	}
2010
2011	RTVfsIoStrmRelease(hVfsIosOut);
2012	return rc;
2013	}
2014
2015
2016	RTDECL(int) RTZipTarFsStreamSetOwner(RTVFSFSSTREAM hVfsFss, RTUID uid, const char *pszOwner)
2017	{
2018	PRTZIPTARFSSTREAMWRITER pThis = (PRTZIPTARFSSTREAMWRITER)RTVfsFsStreamToPrivate(hVfsFss, &g_rtZipTarFssOps);
2019	AssertReturn(pThis, VERR_WRONG_TYPE);
2020
2021	pThis->uidOwner = uid;
2022	if (pThis->pszOwner)
2023	{
2024	RTStrFree(pThis->pszOwner);
2025	pThis->pszOwner = NULL;
2026	}
2027	if (pszOwner)
2028	{
2029	pThis->pszOwner = RTStrDup(pszOwner);
2030	AssertReturn(pThis->pszOwner, VERR_NO_STR_MEMORY);
2031	}
2032
2033	return VINF_SUCCESS;
2034	}
2035
2036
2037	RTDECL(int) RTZipTarFsStreamSetGroup(RTVFSFSSTREAM hVfsFss, RTGID gid, const char *pszGroup)
2038	{
2039	PRTZIPTARFSSTREAMWRITER pThis = (PRTZIPTARFSSTREAMWRITER)RTVfsFsStreamToPrivate(hVfsFss, &g_rtZipTarFssOps);
2040	AssertReturn(pThis, VERR_WRONG_TYPE);
2041
2042	pThis->gidGroup = gid;
2043	if (pThis->pszGroup)
2044	{
2045	RTStrFree(pThis->pszGroup);
2046	pThis->pszGroup = NULL;
2047	}
2048	if (pszGroup)
2049	{
2050	pThis->pszGroup = RTStrDup(pszGroup);
2051	AssertReturn(pThis->pszGroup, VERR_NO_STR_MEMORY);
2052	}
2053
2054	return VINF_SUCCESS;
2055	}
2056
2057
2058	RTDECL(int) RTZipTarFsStreamSetPrefix(RTVFSFSSTREAM hVfsFss, const char *pszPrefix)
2059	{
2060	PRTZIPTARFSSTREAMWRITER pThis = (PRTZIPTARFSSTREAMWRITER)RTVfsFsStreamToPrivate(hVfsFss, &g_rtZipTarFssOps);
2061	AssertReturn(pThis, VERR_WRONG_TYPE);
2062	AssertReturn(!pszPrefix \|\| *pszPrefix, VERR_INVALID_NAME);
2063
2064	if (pThis->pszPrefix)
2065	{
2066	RTStrFree(pThis->pszPrefix);
2067	pThis->pszPrefix = NULL;
2068	pThis->cchPrefix = 0;
2069	}
2070	if (pszPrefix)
2071	{
2072	/*
2073	* Make a copy of the prefix, make sure it ends with a slash,
2074	* then flip DOS slashes.
2075	*/
2076	size_t cchPrefix = strlen(pszPrefix);
2077	char *pszCopy = RTStrAlloc(cchPrefix + 3);
2078	AssertReturn(pszCopy, VERR_NO_STR_MEMORY);
2079	memcpy(pszCopy, pszPrefix, cchPrefix + 1);
2080
2081	RTPathEnsureTrailingSeparator(pszCopy, cchPrefix + 3);
2082
2083	#if RTPATH_STYLE != RTPATH_STR_F_STYLE_UNIX
2084	char *pszDosSlash = strchr(pszCopy, '\\');
2085	while (pszDosSlash)
2086	{
2087	*pszDosSlash = '/';
2088	pszDosSlash = strchr(pszDosSlash + 1, '\\');
2089	}
2090	#endif
2091
2092	pThis->cchPrefix = cchPrefix + strlen(&pszCopy[cchPrefix]);
2093	pThis->pszPrefix = pszCopy;
2094	}
2095
2096	return VINF_SUCCESS;
2097	}
2098
2099
2100	RTDECL(int) RTZipTarFsStreamSetModTime(RTVFSFSSTREAM hVfsFss, PCRTTIMESPEC pModificationTime)
2101	{
2102	PRTZIPTARFSSTREAMWRITER pThis = (PRTZIPTARFSSTREAMWRITER)RTVfsFsStreamToPrivate(hVfsFss, &g_rtZipTarFssOps);
2103	AssertReturn(pThis, VERR_WRONG_TYPE);
2104
2105	if (pModificationTime)
2106	{
2107	pThis->ModTime = *pModificationTime;
2108	pThis->pModTime = &pThis->ModTime;
2109	}
2110	else
2111	pThis->pModTime = NULL;
2112
2113	return VINF_SUCCESS;
2114	}
2115
2116
2117	RTDECL(int) RTZipTarFsStreamSetFileMode(RTVFSFSSTREAM hVfsFss, RTFMODE fAndMode, RTFMODE fOrMode)
2118	{
2119	PRTZIPTARFSSTREAMWRITER pThis = (PRTZIPTARFSSTREAMWRITER)RTVfsFsStreamToPrivate(hVfsFss, &g_rtZipTarFssOps);
2120	AssertReturn(pThis, VERR_WRONG_TYPE);
2121
2122	pThis->fFileModeAndMask = fAndMode \| ~RTFS_UNIX_ALL_PERMS;
2123	pThis->fFileModeOrMask = fOrMode & RTFS_UNIX_ALL_PERMS;
2124	return VINF_SUCCESS;
2125	}
2126
2127
2128	RTDECL(int) RTZipTarFsStreamSetDirMode(RTVFSFSSTREAM hVfsFss, RTFMODE fAndMode, RTFMODE fOrMode)
2129	{
2130	PRTZIPTARFSSTREAMWRITER pThis = (PRTZIPTARFSSTREAMWRITER)RTVfsFsStreamToPrivate(hVfsFss, &g_rtZipTarFssOps);
2131	AssertReturn(pThis, VERR_WRONG_TYPE);
2132
2133	pThis->fDirModeAndMask = fAndMode \| ~RTFS_UNIX_ALL_PERMS;
2134	pThis->fDirModeOrMask = fOrMode & RTFS_UNIX_ALL_PERMS;
2135	return VINF_SUCCESS;
2136	}
2137

Note: See TracBrowser for help on using the repository browser.

source: vbox/trunk/src/VBox/Runtime/common/zip/tarvfswriter.cpp@ 78365

Download in other formats: