tar2qfile.c 26 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018
  1. /* $OpenBSD: tar.h,v 1.7 2003/06/02 23:32:09 millert Exp $ */
  2. /* $NetBSD: tar.h,v 1.3 1995/03/21 09:07:51 cgd Exp $ */
  3. /*-
  4. * Copyright (c) 1992 Keith Muller.
  5. * Copyright (c) 1992, 1993
  6. * The Regents of the University of California. All rights reserved.
  7. *
  8. * This code is derived from software contributed to Berkeley by
  9. * Keith Muller of the University of California, San Diego.
  10. *
  11. * Redistribution and use in source and binary forms, with or without
  12. * modification, are permitted provided that the following conditions
  13. * are met:
  14. * 1. Redistributions of source code must retain the above copyright
  15. * notice, this list of conditions and the following disclaimer.
  16. * 2. Redistributions in binary form must reproduce the above copyright
  17. * notice, this list of conditions and the following disclaimer in the
  18. * documentation and/or other materials provided with the distribution.
  19. * 3. Neither the name of the University nor the names of its contributors
  20. * may be used to endorse or promote products derived from this software
  21. * without specific prior written permission.
  22. *
  23. * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  24. * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  25. * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  26. * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  27. * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  28. * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  29. * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  30. * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  31. * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  32. * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  33. * SUCH DAMAGE.
  34. *
  35. * @(#)tar.h 8.2 (Berkeley) 4/18/94
  36. */
  37. #define _GNU_SOURCE /* For O_NOFOLLOW. */
  38. #include <errno.h>
  39. #include <fcntl.h>
  40. #include <sys/time.h>
  41. #include <sys/stat.h>
  42. #include <stdlib.h>
  43. #include <unistd.h>
  44. #include <stdio.h>
  45. #include <libqubes-rpc-filecopy.h>
  46. #include <string.h>
  47. #include <gui-fatal.h>
  48. // #define DEBUG
  49. /***************************************************
  50. * Most routines extracted from the PAX project (tar.c...) *
  51. ***************************************************/
  52. /*
  53. * BSD PAX global data structures and constants.
  54. */
  55. #define MAXBLK 64512 /* MAX blocksize supported (posix SPEC) */
  56. /* WARNING: increasing MAXBLK past 32256 */
  57. /* will violate posix spec. */
  58. #define MAXBLK_POSIX 32256 /* MAX blocksize supported as per POSIX */
  59. #define BLKMULT 512 /* blocksize must be even mult of 512 bytes */
  60. /* Don't even think of changing this */
  61. #define DEVBLK 8192 /* default read blksize for devices */
  62. #define FILEBLK 10240 /* default read blksize for files */
  63. #define PAXPATHLEN 3072 /* maximium path length for pax. MUST be */
  64. /*
  65. * defines and data structures common to all tar formats
  66. */
  67. #define CHK_LEN 8 /* length of checksum field */
  68. #define TNMSZ 100 /* size of name field */
  69. #define NULLCNT 2 /* number of null blocks in trailer */
  70. #define CHK_OFFSET 148 /* start of chksum field */
  71. #define BLNKSUM 256L /* sum of checksum field using ' ' */
  72. /*
  73. * General Defines
  74. */
  75. #define HEX 16
  76. #define OCT 8
  77. #define _PAX_ 1
  78. #define _TFILE_BASE "paxXXXXXXXXXX"
  79. /*
  80. * General Macros
  81. */
  82. #ifndef MIN
  83. #define MIN(a,b) (((a)<(b))?(a):(b))
  84. #endif
  85. #ifndef MAX
  86. #define MAX(a,b) (((a)>(b))?(a):(b))
  87. #endif
  88. #define MAJOR(x) major(x)
  89. #define MINOR(x) minor(x)
  90. #define TODEV(x, y) makedev((x), (y))
  91. /*
  92. * Values used in typeflag field in all tar formats
  93. * (only REGTYPE, LNKTYPE and SYMTYPE are used in old bsd tar headers)
  94. */
  95. #define REGTYPE '0' /* Regular File */
  96. #define AREGTYPE '\0' /* Regular File */
  97. #define LNKTYPE '1' /* Link */
  98. #define SYMTYPE '2' /* Symlink */
  99. #define CHRTYPE '3' /* Character Special File */
  100. #define BLKTYPE '4' /* Block Special File */
  101. #define DIRTYPE '5' /* Directory */
  102. #define FIFOTYPE '6' /* FIFO */
  103. #define CONTTYPE '7' /* high perf file */
  104. /*
  105. * GNU tar compatibility;
  106. */
  107. #define LONGLINKTYPE 'K' /* Long Symlink */
  108. #define LONGNAMETYPE 'L' /* Long File */
  109. #define EXTHEADERTYPE 'x' /* Extended header */
  110. /*
  111. * Pad with a bit mask, much faster than doing a mod but only works on powers
  112. * of 2. Macro below is for block of 512 bytes.
  113. */
  114. #define TAR_PAD(x) ((512 - ((x) & 511)) & 511)
  115. /*
  116. * Data Interchange Format - Extended tar header format - POSIX 1003.1-1990
  117. */
  118. #define TPFSZ 155
  119. #define TMAGIC "ustar" /* ustar and a null */
  120. #define TMAGLEN 6
  121. #define TVERSION "00" /* 00 and no null */
  122. #define TVERSLEN 2
  123. typedef struct {
  124. char name[TNMSZ]; /* name of entry */
  125. char mode[8]; /* mode */
  126. char uid[8]; /* uid */
  127. char gid[8]; /* gid */
  128. char size[12]; /* size */
  129. char mtime[12]; /* modification time */
  130. char chksum[CHK_LEN]; /* checksum */
  131. char typeflag; /* type of file. */
  132. char linkname[TNMSZ]; /* linked to name */
  133. char magic[TMAGLEN]; /* magic cookie */
  134. char version[TVERSLEN]; /* version */
  135. char uname[32]; /* ascii owner name */
  136. char gname[32]; /* ascii group name */
  137. char devmajor[8]; /* major device number */
  138. char devminor[8]; /* minor device number */
  139. char prefix[TPFSZ]; /* linked to name */
  140. } HD_USTAR;
  141. /*
  142. * Routines for manipulating headers, trailers:
  143. * asc_ul()
  144. * tar_trail()
  145. * tar_chksm()
  146. * ustar_id()
  147. */
  148. static unsigned long tar_chksm (char *, int);
  149. char *gnu_hack_string; /* GNU ././@LongLink hackery */
  150. char untrusted_namebuf[MAX_PATH_LENGTH];
  151. int use_seek = 1;
  152. extern int ignore_quota_error;
  153. struct filters {
  154. int filters_count;
  155. char **filters;
  156. int *filters_matches;
  157. int matched_filters;
  158. };
  159. /*
  160. * asc_ul()
  161. * convert hex/octal character string into a u_long. We do not have to
  162. * check for overflow! (the headers in all supported formats are not large
  163. * enough to create an overflow).
  164. * NOTE: strings passed to us are NOT TERMINATED.
  165. * Return:
  166. * unsigned long value
  167. */
  168. u_long
  169. asc_ul (char *str, int len, int base)
  170. {
  171. char *stop;
  172. u_long tval = 0;
  173. stop = str + len;
  174. /*
  175. * skip over leading blanks and zeros
  176. */
  177. while ((str < stop) && ((*str == ' ') || (*str == '0')))
  178. ++str;
  179. /*
  180. * for each valid digit, shift running value (tval) over to next digit
  181. * and add next digit
  182. */
  183. if (base == HEX)
  184. {
  185. while (str < stop)
  186. {
  187. if ((*str >= '0') && (*str <= '9'))
  188. tval = (tval << 4) + (*str++ - '0');
  189. else if ((*str >= 'A') && (*str <= 'F'))
  190. tval = (tval << 4) + 10 + (*str++ - 'A');
  191. else if ((*str >= 'a') && (*str <= 'f'))
  192. tval = (tval << 4) + 10 + (*str++ - 'a');
  193. else
  194. break;
  195. }
  196. }
  197. else
  198. {
  199. while ((str < stop) && (*str >= '0') && (*str <= '7'))
  200. tval = (tval << 3) + (*str++ - '0');
  201. }
  202. return (tval);
  203. }
  204. /*
  205. * tar_trail()
  206. * Called to determine if a header block is a valid trailer. We are passed
  207. * the block, the in_sync flag (which tells us we are in resync mode;
  208. * looking for a valid header), and cnt (which starts at zero) which is
  209. * used to count the number of empty blocks we have seen so far.
  210. * Return:
  211. * 0 if a valid trailer, -1 if not a valid trailer, or 1 if the block
  212. * could never contain a header.
  213. */
  214. int
  215. tar_trail (char *buf,
  216. int in_resync, int *cnt)
  217. {
  218. register int i;
  219. /*
  220. * look for all zero, trailer is two consecutive blocks of zero
  221. */
  222. for (i = 0; i < BLKMULT; ++i)
  223. {
  224. if (buf[i] != '\0')
  225. break;
  226. }
  227. /*
  228. * if not all zero it is not a trailer, but MIGHT be a header.
  229. */
  230. if (i != BLKMULT)
  231. return (-1);
  232. /*
  233. * When given a zero block, we must be careful!
  234. * If we are not in resync mode, check for the trailer. Have to watch
  235. * out that we do not mis-identify file data as the trailer, so we do
  236. * NOT try to id a trailer during resync mode. During resync mode we
  237. * might as well throw this block out since a valid header can NEVER be
  238. * a block of all 0 (we must have a valid file name).
  239. */
  240. if (!in_resync && (++*cnt >= NULLCNT))
  241. return (0);
  242. return (1);
  243. }
  244. /*
  245. * tar_chksm()
  246. * calculate the checksum for a tar block counting the checksum field as
  247. * all blanks (BLNKSUM is that value pre-calculated, the sum of 8 blanks).
  248. * NOTE: we use len to short circuit summing 0's on write since we ALWAYS
  249. * pad headers with 0.
  250. * Return:
  251. * unsigned long checksum
  252. */
  253. static unsigned long
  254. tar_chksm (char *blk, int len)
  255. {
  256. char *stop;
  257. char *pt;
  258. unsigned int chksm = BLNKSUM; /* initial value is checksum field sum */
  259. /*
  260. * add the part of the block before the checksum field
  261. */
  262. pt = blk;
  263. stop = blk + CHK_OFFSET;
  264. while (pt < stop)
  265. chksm += (*pt++ & 0xff);
  266. /*
  267. * move past the checksum field and keep going, spec counts the
  268. * checksum field as the sum of 8 blanks (which is pre-computed as
  269. * BLNKSUM).
  270. * ASSUMED: len is greater than CHK_OFFSET. (len is where our 0 padding
  271. * starts, no point in summing zero's)
  272. */
  273. pt += CHK_LEN;
  274. stop = blk + len;
  275. while (pt < stop)
  276. chksm += (*pt++ & 0xff);
  277. return chksm;
  278. }
  279. /*
  280. * ustar_id()
  281. * determine if a block given to us is a valid ustar header. We have to
  282. * be on the lookout for those pesky blocks of all zero's
  283. * Return:
  284. * 0 if a ustar header, -1 otherwise
  285. */
  286. int
  287. ustar_id (char *blk, size_t size)
  288. {
  289. HD_USTAR *hd;
  290. if (size < BLKMULT)
  291. return (-1);
  292. hd = (HD_USTAR *) blk;
  293. /*
  294. * check for block of zero's first, a simple and fast test then check
  295. * ustar magic cookie. We should use TMAGLEN, but some USTAR archive
  296. * programs are fouled up and create archives missing the \0. Last we
  297. * check the checksum. If ok we have to assume it is a valid header.
  298. */
  299. if (hd->name[0] == '\0')
  300. return (-1);
  301. if (strncmp (hd->magic, TMAGIC, TMAGLEN - 1) != 0)
  302. return (-1);
  303. if (asc_ul (hd->chksum, sizeof (hd->chksum), OCT) !=
  304. tar_chksm (blk, BLKMULT))
  305. return (-1);
  306. return (0);
  307. }
  308. /*
  309. * Routines for reading tar files
  310. // Source: http://www.mkssoftware.com/docs/man4/pax.4.asp
  311. struct file_header { // PAX header is similar as file_header and can be completely ignored
  312. unsigned char[100] name;
  313. unsigned char[8] mode;
  314. unsigned char[8] uid; // unused
  315. unsigned char[8] gid; // unused
  316. unsigned char[12] size; // 0 if file is a link
  317. unsigned char[12] mtime;
  318. unsigned char[8] chksum;
  319. unsigned char[1] typeflag;
  320. unsigned char[100] linkname;
  321. unsigned char[6] magic; //ustar
  322. unsigned char[2] version; // 00
  323. unsigned char[32] uname; // unused
  324. unsigned char[32] gname; // unused
  325. unsigned char[8] devmajor; // unused ?
  326. unsigned char[8] devminor; // unused ?
  327. unsigned char[155] prefix; // only used for files > 100 characters. could be unused ?
  328. };
  329. enum {
  330. TYPE_REGULAR, //0
  331. TYPE_ARCHIVE_LINK, //1
  332. TYPE_SYMLINK, //2
  333. TYPE_CHARACTER_DEVICE, //3
  334. TYPE_BLOCK_DEVICE, //4
  335. TYPE_DIRECTORY, //5
  336. TYPE_FIFO, //6
  337. // Other types:
  338. TYPE_EXTENDED_USAGE, //xxxxx
  339. // A-Z are available for custom usage
  340. };
  341. // Extended attribute:
  342. // length keyword=value
  343. // atime, charset, comment, gname, linkpath, mtime, path, size, uname
  344. */
  345. enum {
  346. NEED_NOTHING,
  347. NEED_SKIP,
  348. NEED_SKIP_FILE, // distinguish between skipped file and unwanted blocks (extended headers etc)
  349. NEED_READ,
  350. NEED_SYNC_TRAIL,
  351. INVALID_HEADER,
  352. MEMORY_ALLOC_FAILED,
  353. };
  354. /*
  355. * ustar_rd()
  356. * extract the values out of block already determined to be a ustar header.
  357. * store the values in the ARCHD parameter.
  358. * Return:
  359. * 0
  360. */
  361. int n_dirs = 0;
  362. char ** dirs_headers_sent = NULL;
  363. int
  364. ustar_rd (int fd, struct file_header * untrusted_hdr, char *buf, struct stat * sb, struct filters *filters)
  365. {
  366. register HD_USTAR *hd;
  367. register char *dest;
  368. register int cnt = 0;
  369. int ret;
  370. int i;
  371. int should_extract;
  372. /* DISABLED: unused
  373. dev_t devmajor;
  374. dev_t devminor;
  375. */
  376. /*
  377. * we only get proper sized buffers
  378. */
  379. #ifdef DEBUG
  380. fprintf(stderr,"Checking if valid header\n");
  381. #endif
  382. if (ustar_id (buf, BLKMULT) < 0) {
  383. #ifdef DEBUG
  384. fprintf (stderr, "Invalid header\n");
  385. #endif
  386. return INVALID_HEADER;
  387. }
  388. #ifdef DEBUG
  389. fprintf(stderr,"Valid header!\n");
  390. #endif
  391. /* DISABLED: Internal to PAX
  392. arcn->org_name = arcn->name;
  393. arcn->sb.st_nlink = 1;
  394. arcn->pat = NULL;
  395. arcn->nlen = 0;
  396. */
  397. untrusted_hdr->namelen = 0;
  398. hd = (HD_USTAR *) buf;
  399. /*
  400. * see if the filename is split into two parts. if, so joint the parts.
  401. * we copy the prefix first and add a / between the prefix and name.
  402. */
  403. dest = untrusted_namebuf;
  404. if (*(hd->prefix) != '\0')
  405. {
  406. cnt = strlen(strncpy (dest, hd->prefix,
  407. MIN(sizeof (untrusted_namebuf) - 1,TPFSZ+1)));
  408. dest += cnt;
  409. *dest++ = '/';
  410. cnt++;
  411. }
  412. if (gnu_hack_string)
  413. {
  414. untrusted_hdr->namelen = cnt + strlen(strncpy (dest, gnu_hack_string,
  415. MIN(TNMSZ+1, sizeof (untrusted_namebuf) - cnt)));
  416. free(gnu_hack_string);
  417. gnu_hack_string = NULL;
  418. } else
  419. untrusted_hdr->namelen = cnt + strlen(strncpy (dest, hd->name,
  420. MIN(TNMSZ+1, sizeof (untrusted_namebuf) - cnt)));
  421. // qfile count the \0 in the namelen
  422. untrusted_hdr->namelen += 1;
  423. #ifdef DEBUG
  424. fprintf(stderr,"Retrieved name len: %d\n",untrusted_hdr->namelen);
  425. fprintf(stderr,"Retrieved name: %s\n",untrusted_namebuf);
  426. #endif
  427. /*
  428. * follow the spec to the letter. we should only have mode bits, strip
  429. * off all other crud we may be passed.
  430. */
  431. sb->st_mode = (mode_t) (asc_ul (hd->mode, sizeof (hd->mode), OCT) &
  432. 0xfff);
  433. untrusted_hdr->mode = sb->st_mode;
  434. #if defined(_FILE_OFFSET_BITS) && _FILE_OFFSET_BITS == 64
  435. sb->st_size = (off_t) asc_uqd (hd->size, sizeof (hd->size), OCT);
  436. #else
  437. sb->st_size = (off_t) asc_ul (hd->size, sizeof (hd->size), OCT);
  438. #endif
  439. untrusted_hdr->filelen = sb->st_size;
  440. untrusted_hdr->atime = (time_t) asc_ul (hd->mtime, sizeof (hd->mtime), OCT);
  441. untrusted_hdr->mtime = untrusted_hdr->atime;
  442. untrusted_hdr->atime_nsec = untrusted_hdr->mtime_nsec = 0;
  443. sb->st_mtime = (time_t) asc_ul (hd->mtime, sizeof (hd->mtime), OCT);
  444. sb->st_ctime = sb->st_atime = sb->st_mtime;
  445. /*
  446. * If we can find the ascii names for gname and uname in the password
  447. * and group files we will use the uid's and gid they bind. Otherwise
  448. * we use the uid and gid values stored in the header. (This is what
  449. * the posix spec wants).
  450. */
  451. /* DISABLED: unused
  452. hd->gname[sizeof (hd->gname) - 1] = '\0';
  453. if (gid_name (hd->gname, &(arcn->sb.st_gid)) < 0)
  454. arcn->sb.st_gid = (gid_t) asc_ul (hd->gid, sizeof (hd->gid), OCT);
  455. hd->uname[sizeof (hd->uname) - 1] = '\0';
  456. if (uid_name (hd->uname, &(arcn->sb.st_uid)) < 0)
  457. arcn->sb.st_uid = (uid_t) asc_ul (hd->uid, sizeof (hd->uid), OCT);
  458. */
  459. /*
  460. * set the defaults, these may be changed depending on the file type
  461. */
  462. /* Disabled: pax specific
  463. arcn->ln_name[0] = '\0';
  464. arcn->ln_nlen = 0;
  465. arcn->pad = 0;
  466. arcn->skip = 0;
  467. arcn->sb.st_rdev = (dev_t) 0;
  468. */
  469. /*
  470. * set the mode and PAX type according to the typeflag in the header
  471. */
  472. switch (hd->typeflag)
  473. {
  474. case FIFOTYPE:
  475. #ifdef DEBUG
  476. fprintf(stderr,"File is FIFOTYPE\n");
  477. #endif
  478. /* DISABLED: unused
  479. arcn->type = PAX_FIF;
  480. arcn->sb.st_mode |= S_IFIFO;
  481. */
  482. break;
  483. case DIRTYPE:
  484. #ifdef DEBUG
  485. fprintf(stderr,"File is DIRTYPE\n");
  486. #endif
  487. /* DISABLED: unused
  488. arcn->type = PAX_DIR;
  489. arcn->sb.st_mode |= S_IFDIR;
  490. arcn->sb.st_nlink = 2;
  491. */
  492. /*
  493. * Some programs that create ustar archives append a '/'
  494. * to the pathname for directories. This clearly violates
  495. * ustar specs, but we will silently strip it off anyway.
  496. */
  497. /*
  498. if (arcn->name[arcn->nlen - 1] == '/')
  499. arcn->name[--arcn->nlen] = '\0';
  500. */
  501. break;
  502. case BLKTYPE:
  503. #ifdef DEBUG
  504. fprintf(stderr,"File is BLKTYPE\n");
  505. #endif
  506. break;
  507. case CHRTYPE:
  508. #ifdef DEBUG
  509. fprintf(stderr,"File is CHRTYPE\n");
  510. #endif
  511. /*
  512. * this type requires the rdev field to be set.
  513. */
  514. if (hd->typeflag == BLKTYPE)
  515. {
  516. /*
  517. arcn->type = PAX_BLK;
  518. arcn->sb.st_mode |= S_IFBLK;
  519. */
  520. }
  521. else
  522. {
  523. /*
  524. arcn->type = PAX_CHR;
  525. arcn->sb.st_mode |= S_IFCHR;
  526. */
  527. }
  528. /* DISABLED: unused
  529. devmajor = (dev_t) asc_ul (hd->devmajor, sizeof (hd->devmajor), OCT);
  530. devminor = (dev_t) asc_ul (hd->devminor, sizeof (hd->devminor), OCT);
  531. */
  532. // arcn->sb.st_rdev = TODEV (devmajor, devminor);
  533. break;
  534. case SYMTYPE:
  535. #ifdef DEBUG
  536. fprintf(stderr,"File is SYMTYPE\n");
  537. #endif
  538. break;
  539. case LNKTYPE:
  540. #ifdef DEBUG
  541. fprintf(stderr,"File is LNKTYPE\n");
  542. #endif
  543. if (hd->typeflag == SYMTYPE)
  544. {
  545. // arcn->type = PAX_SLK;
  546. // arcn->sb.st_mode |= S_IFLNK;
  547. }
  548. else
  549. {
  550. // arcn->type = PAX_HLK;
  551. /*
  552. * so printing looks better
  553. */
  554. // arcn->sb.st_mode |= S_IFREG;
  555. // arcn->sb.st_nlink = 2;
  556. }
  557. /*
  558. * copy the link name
  559. */
  560. // arcn->ln_nlen = strlcpy (arcn->ln_name, hd->linkname,
  561. // MIN(TNMSZ+1,sizeof (arcn->ln_name)));
  562. break;
  563. case LONGLINKTYPE:
  564. #ifdef DEBUG
  565. fprintf(stderr,"File is LONGLINKTYPE\n");
  566. #endif
  567. break;
  568. case LONGNAMETYPE:
  569. #ifdef DEBUG
  570. fprintf(stderr,"File is LONGNAMETYPE\n");
  571. #endif
  572. /*
  573. * GNU long link/file; we tag these here and let the
  574. * pax internals deal with it -- too ugly otherwise.
  575. */
  576. // arcn->type =
  577. // hd->typeflag == LONGLINKTYPE ? PAX_GLL : PAX_GLF;
  578. // arcn->pad = TAR_PAD(arcn->sb.st_size);
  579. // arcn->skip = arcn->sb.st_size;
  580. // arcn->ln_name[0] = '\0';
  581. // arcn->ln_nlen = 0;
  582. break;
  583. case CONTTYPE:
  584. #ifdef DEBUG
  585. fprintf(stderr,"File is CONTTYPE\n");
  586. #endif
  587. break;
  588. case AREGTYPE:
  589. #ifdef DEBUG
  590. fprintf(stderr,"File is AREGTYPE\n");
  591. #endif
  592. break;
  593. case REGTYPE:
  594. #ifdef DEBUG
  595. fprintf(stderr,"File is REGTYPE of size %ld\n",sb->st_size);
  596. #endif
  597. // Check if user want to extract this file
  598. should_extract = 1;
  599. for (i=0; i < filters->filters_count; i++) {
  600. should_extract = 0;
  601. #ifdef DEBUG
  602. fprintf(stderr, "Comparing with filter %s\n", filters->filters[i]);
  603. #endif
  604. if (strncmp(untrusted_namebuf, filters->filters[i], strlen(filters->filters[i])) == 0) {
  605. #ifdef DEBUG
  606. fprintf(stderr, "Match (%d)\n", filters->filters_matches[i]);
  607. #endif
  608. should_extract = 1;
  609. filters->filters_matches[i]++;
  610. if (filters->filters_matches[i] == 1) {
  611. // first match
  612. filters->matched_filters++;
  613. }
  614. break;
  615. }
  616. }
  617. if (should_extract != 1) {
  618. #ifdef DEBUG
  619. fprintf(stderr, "File should be filtered.. Skipping\n");
  620. #endif
  621. return NEED_SKIP_FILE;
  622. }
  623. // Create a copy of untrusted_namebuf to be used for strtok
  624. char * dirbuf;
  625. dirbuf = malloc(sizeof (char) * (untrusted_hdr->namelen));
  626. if (dirbuf == NULL)
  627. return MEMORY_ALLOC_FAILED;
  628. dirbuf = strncpy(dirbuf, untrusted_namebuf, untrusted_hdr->namelen);
  629. int i = 0;
  630. int dir_found = 0;
  631. size_t pathsize = 0;
  632. char * path = NULL;
  633. struct file_header dir_header;
  634. // Split the path in directories and recompose it incrementally
  635. char * last_token = strtok(dirbuf,"/");
  636. char * token = strtok(NULL, "/");
  637. size_t len_last_token = 0;
  638. while (token != NULL) {
  639. #ifdef DEBUG
  640. fprintf(stderr,"Found directory %s (last:%s)\n",token,last_token);
  641. #endif
  642. // Recompose the path based on last discovered directory
  643. len_last_token = strlen(last_token);
  644. if (path == NULL) {
  645. path = malloc(sizeof (char) * (len_last_token+1));
  646. if (path == NULL)
  647. return MEMORY_ALLOC_FAILED;
  648. path = memcpy(path, last_token, len_last_token);
  649. path[len_last_token] = '\0';
  650. } else {
  651. pathsize = strlen(path);
  652. path = realloc(path, sizeof (char) * (strlen(path)+1+len_last_token+1));
  653. if (path == NULL)
  654. return MEMORY_ALLOC_FAILED;
  655. path[pathsize] = '/';
  656. memcpy(path+pathsize+1, last_token, len_last_token);
  657. path[pathsize+len_last_token+1] = '\0';
  658. }
  659. #ifdef DEBUG
  660. fprintf(stderr,"Path is %s\n",path);
  661. #endif
  662. #ifdef DEBUG
  663. fprintf(stderr,"Checking from i=0 i<%d\n",n_dirs);
  664. #endif
  665. // Verify if qfile headers for the current path have already been sent based on the dirs_headers_sent table
  666. dir_found = 0;
  667. for (i = 0; i < n_dirs; ++i) {
  668. #ifdef DEBUG
  669. fprintf(stderr,"Comparing with %d %d %s %s\n",i,n_dirs,dirs_headers_sent[i],path);
  670. #endif
  671. if (strcmp(dirs_headers_sent[i],path)==0) {
  672. #ifdef DEBUG
  673. fprintf(stderr,"Directory headers already sent\n");
  674. #endif
  675. dir_found=1;
  676. }
  677. }
  678. if (dir_found == 0) {
  679. // Register the current path as being sent in the dirs_headers_sent table
  680. #ifdef DEBUG
  681. fprintf(stderr,"Inserting %s into register\n",path);
  682. #endif
  683. dirs_headers_sent = realloc(dirs_headers_sent, sizeof (char*) * (++n_dirs));
  684. if (dirs_headers_sent == NULL)
  685. return MEMORY_ALLOC_FAILED;
  686. dirs_headers_sent[n_dirs-1] = malloc(sizeof (char) * (strlen(path)+1));
  687. if (dirs_headers_sent[n_dirs-1] == NULL)
  688. return MEMORY_ALLOC_FAILED;
  689. memcpy(dirs_headers_sent[n_dirs-1], path, strlen(path)+1);
  690. // Initialize the qfile headers for the current directory path
  691. dir_header.namelen = strlen(path)+1;
  692. dir_header.atime = untrusted_hdr->atime;
  693. dir_header.atime_nsec = untrusted_hdr->atime_nsec;
  694. dir_header.mtime = untrusted_hdr->mtime;
  695. dir_header.mtime_nsec = untrusted_hdr->mtime_nsec;
  696. dir_header.mode = untrusted_hdr->mode | S_IFDIR;
  697. dir_header.filelen = 0;
  698. #ifdef DEBUG
  699. fprintf(stderr,"Sending directory headers for %s\n",path);
  700. #endif
  701. // Send the qfile headers for the current directory path
  702. write_headers(&dir_header, path);
  703. }
  704. last_token = token;
  705. token = strtok(NULL, "/");
  706. }
  707. free(path);
  708. free(dirbuf);
  709. #ifdef DEBUG
  710. fprintf(stderr,"End of directory checks\n");
  711. #endif
  712. // Restore POSIX stat file mode (because PAX format use its own file type)
  713. untrusted_hdr->mode |= S_IFREG;
  714. #ifdef DEBUG
  715. fprintf(stderr,"Writing file header\n");
  716. #endif
  717. // Send header and file content
  718. write_headers(untrusted_hdr, untrusted_namebuf);
  719. #ifdef DEBUG
  720. fprintf(stderr,"Writing file content\n");
  721. #endif
  722. ret = copy_file_with_crc(1, fd, untrusted_hdr->filelen);
  723. #ifdef DEBUG
  724. fprintf(stderr,"Copyfile returned with error %d\n",ret);
  725. #endif
  726. if (ret != COPY_FILE_OK) {
  727. if (ret != COPY_FILE_WRITE_ERROR)
  728. gui_fatal("Copying file %s: %s", untrusted_namebuf,
  729. copy_file_status_to_str(ret));
  730. else {
  731. fprintf(stderr,"UNKNOWN ERROR RETURN STATUS:%d\n.. Waiting...\n",ret);
  732. set_block(0);
  733. wait_for_result();
  734. exit(1);
  735. }
  736. }
  737. // Extract extra padding
  738. #ifdef DEBUG
  739. fprintf(stderr,"Need to remove pad:%lld %lld\n",untrusted_hdr->filelen,BLKMULT-(untrusted_hdr->filelen%BLKMULT));
  740. #endif
  741. if (untrusted_hdr->filelen%BLKMULT > 0) {
  742. if (!read_all(fd, buf, BLKMULT-(untrusted_hdr->filelen%BLKMULT))) {
  743. wait_for_result();
  744. exit(1);
  745. }
  746. }
  747. // Resync trailing headers in order to find next file chunck in the tar file
  748. return NEED_SYNC_TRAIL;
  749. break;
  750. case EXTHEADERTYPE:
  751. #ifdef DEBUG
  752. fprintf(stderr,"Extended HEADER encountered\n");
  753. #endif
  754. return NEED_SKIP;
  755. break;
  756. default:
  757. #ifdef DEBUG
  758. fprintf(stderr,"Default type detected:%c\n",hd->typeflag);
  759. #endif
  760. return NEED_SKIP;
  761. /*
  762. * these types have file data that follows. Set the skip and
  763. * pad fields.
  764. */
  765. // arcn->type = PAX_REG;
  766. // arcn->pad = TAR_PAD (arcn->sb.st_size);
  767. // arcn->skip = arcn->sb.st_size;
  768. // arcn->sb.st_mode |= S_IFREG;
  769. break;
  770. }
  771. return NEED_SKIP;
  772. }
  773. void tar_file_processor(int fd, struct filters *filters)
  774. {
  775. int ret;
  776. int i;
  777. int current;
  778. struct file_header hdr;
  779. struct stat sb; /* stat buffer see stat(2) */
  780. char buf[BLKMULT+1];
  781. i=0;
  782. current = NEED_READ;
  783. size_t to_skip = 0;
  784. int sync_count = 0;
  785. while (read_all(fd, buf, BLKMULT)) {
  786. ret = 0;
  787. if (current==NEED_SYNC_TRAIL) {
  788. ret = tar_trail (buf, 1, &sync_count);
  789. #ifdef DEBUG
  790. fprintf(stderr,"Synchronizing trail: %d %d\n", ret, sync_count);
  791. #endif
  792. if (ret != 1) {
  793. current = NEED_READ;
  794. sync_count = 0;
  795. }
  796. }
  797. if (current==NEED_READ) {
  798. current = ustar_rd(fd, &hdr, buf, &sb, filters);
  799. #ifdef DEBUG
  800. fprintf(stderr,"Return %d\n", current);
  801. #endif
  802. }
  803. if (current==NEED_SKIP || current==NEED_SKIP_FILE) {
  804. if (current==NEED_SKIP_FILE &&
  805. filters->filters_count > 0 &&
  806. filters->filters_count == filters->matched_filters) {
  807. // This assume that either:
  808. // a) files are sorted (using full path as sort key)
  809. // b) all the directory content is in
  810. // consecutive block and only directories
  811. // are given as filters
  812. // This is true for backups prepared by qvm-backup
  813. #ifdef DEBUG
  814. fprintf(stderr, "All filters matched at least once - assuming end of requested data\n");
  815. #endif
  816. return;
  817. }
  818. to_skip = hdr.filelen;
  819. #ifdef DEBUG
  820. fprintf(stderr,"Need to skip %lld bytes (matched filters %d < %d)\n",
  821. hdr.filelen, filters->matched_filters, filters->filters_count);
  822. fprintf(stderr,"Need to remove pad:%ld %lld %lld\n",to_skip,hdr.filelen,BLKMULT-(hdr.filelen%BLKMULT));
  823. #endif
  824. if (to_skip%BLKMULT > 0) {
  825. to_skip += BLKMULT-(to_skip%BLKMULT);
  826. }
  827. if (use_seek) {
  828. int tries = 3;
  829. while (lseek(fd, to_skip, SEEK_CUR) < 0) {
  830. if (errno == ESPIPE) {
  831. // fallback to read()
  832. use_seek = 0;
  833. break;
  834. } else if (errno == EAGAIN) {
  835. /* WTF?! lseek theoretically never returns this error, but
  836. * in practice it was seen... */
  837. if (tries--)
  838. continue;
  839. }
  840. perror("lseek");
  841. exit(1);
  842. }
  843. }
  844. // not using "else" because above can fallback to read() method
  845. if (!use_seek) {
  846. while (to_skip > 0) {
  847. ret = read_all(fd, &buf, MIN(to_skip,BLKMULT));
  848. if (ret <= 0) {
  849. exit(1);
  850. }
  851. to_skip -= MIN(to_skip,BLKMULT);
  852. }
  853. }
  854. current = NEED_SYNC_TRAIL;
  855. }
  856. i++;
  857. //if (i >= 10)
  858. // exit(0);
  859. }
  860. }
  861. int main(int argc, char **argv)
  862. {
  863. char *entry;
  864. int fd = -1;
  865. int use_stdin = 1;
  866. struct filters filters;
  867. qfile_pack_init();
  868. /* when extracting backup header, dom0 will terminate the transfer with
  869. * EDQUOT just after getting qubes.xml */
  870. set_ignore_quota_error(1);
  871. set_nonblock(0);
  872. if (argc > 1) {
  873. if (strcmp(argv[1], "-")==0) {
  874. use_stdin = 1;
  875. } else {
  876. use_stdin = 0;
  877. entry = argv[1];
  878. #ifdef DEBUG
  879. fprintf(stderr, "Parsing file %s\n",entry);
  880. #endif
  881. fd = open(entry, O_RDONLY);
  882. if (fd < 0) {
  883. fprintf(stderr,"Error opening file %s\n",entry);
  884. exit(2);
  885. }
  886. }
  887. }
  888. // Parse tar file
  889. if (argc > 2)
  890. filters.filters_count = argc-2;
  891. else
  892. filters.filters_count = 0;
  893. filters.filters = argv+2;
  894. filters.filters_matches = calloc(filters.filters_count, sizeof(int));
  895. if (filters.filters_matches == NULL) {
  896. perror("calloc");
  897. exit(1);
  898. }
  899. filters.matched_filters = 0;
  900. if (use_stdin == 1) {
  901. #ifdef DEBUG
  902. fprintf(stderr,"Using STDIN\n");
  903. #endif
  904. set_block(0);
  905. fd = 0;
  906. }
  907. if (fd < 0) {
  908. fprintf(stderr, "No input file provided\n");
  909. exit(1);
  910. }
  911. tar_file_processor(fd, &filters);
  912. notify_end_and_wait_for_result();
  913. return 0;
  914. }