1 /*
2 * Copyright (c) 2007, 2010, Oracle and/or its affiliates. All rights reserved.
3 * Copyright 2013 Nexenta Systems, Inc. All rights reserved.
4 */
5
6 /*
7 * BSD 3 Clause License
8 *
9 * Copyright (c) 2007, The Storage Networking Industry Association.
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 * - Redistributions of source code must retain the above copyright
15 * notice, this list of conditions and the following disclaimer.
16 *
17 * - Redistributions in binary form must reproduce the above copyright
18 * notice, this list of conditions and the following disclaimer in
19 * the documentation and/or other materials provided with the
20 * distribution.
21 *
22 * - Neither the name of The Storage Networking Industry Association (SNIA)
23 * nor the names of its contributors may be used to endorse or promote
24 * products derived from this software without specific prior written
25 * permission.
26 *
27 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
28 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
29 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
30 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
31 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
32 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
33 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
34 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
35 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
36 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
37 * POSSIBILITY OF SUCH DAMAGE.
38 */
39 #include <stdlib.h>
40 #include <ctype.h>
41 #include <stdio.h>
42 #include <limits.h>
43 #include <string.h>
44 #include <time.h>
45 #include <sys/types.h>
46 #include <sys/acl.h>
47 #include <sys/mkdev.h>
48 #include <utime.h>
49 #include <unistd.h>
50 #include <pthread.h>
51 #include <archives.h>
52 #include <priv.h>
53 #include <tlm.h>
54 #include <libzfs.h>
55 #include <pwd.h>
56 #include <grp.h>
57 #include <ndmpd_prop.h>
58 #include "tlm_proto.h"
59
60
61 #define PM_EXACT_OR_CHILD(m) ((m) == PM_EXACT || (m) == PM_CHILD)
62 #define ERROR_IS_FATAL(err) ((err) == ENOSPC || (err) == EDQUOT)
63
64 typedef boolean_t name_match_fp_t(char *s, char *t);
65
66 static int set_acl(char *name, tlm_acls_t *acls);
67 static int restore_file(int *fp,
68 char *real_name,
69 long size,
70 longlong_t huge_size,
71 tlm_acls_t *,
72 boolean_t want_this_file,
73 tlm_cmd_t *,
74 tlm_job_stats_t *,
75 long *);
76 static long restore_xattr_hdr(int *fp,
77 char *name,
78 char *fname,
79 long size,
80 tlm_acls_t *acls,
81 tlm_cmd_t *local_commands,
82 tlm_job_stats_t *job_stats);
83 static int get_long_name(int lib,
84 int drv,
85 long recsize,
86 char *name,
87 long *buf_spot,
88 tlm_cmd_t *local_commands);
89 static int get_humongus_file_header(int lib,
90 int drv,
91 long recsize,
92 longlong_t *size,
93 char *name,
94 tlm_cmd_t *);
95 static int create_directory(char *dir,
96 tlm_job_stats_t *);
97 static int create_hard_link(char *name,
98 char *link,
99 tlm_acls_t *,
100 tlm_job_stats_t *);
101 static int create_sym_link(char *dst,
102 char *target,
103 tlm_acls_t *,
104 tlm_job_stats_t *);
105 static int create_special(char,
106 char *name,
107 tlm_acls_t *,
108 int,
109 int,
110 tlm_job_stats_t *);
111 static long load_acl_info(int lib,
112 int drv,
113 long size,
114 tlm_acls_t *,
115 long *acl_spot,
116 tlm_cmd_t *);
117 static char *get_read_buffer(int want,
118 int *error,
119 int *actual_size,
120 tlm_cmd_t *);
121 static boolean_t wildcard_enabled(void);
122 static boolean_t is_file_wanted(char *name,
123 char **sels,
124 char **exls,
125 int flags,
126 int *mchtype,
127 int *pos);
128 static char *catnames(struct rs_name_maker *rnp,
129 char *buf,
130 int pos,
131 char *path);
132
133 static char *rs_new_name(struct rs_name_maker *rnp,
134 char *real_name,
135 int pos,
136 char *path);
137
138 static void rs_create_new_bkpath(char *bk_path,
139 char *path,
140 char *pbuf);
141
142 typedef struct stack_ent {
143 char *se_name;
144 tlm_acls_t se_acls;
145 } stack_ent_t;
146
147
148 /*
149 * dtree_push
150 */
151 int
dtree_push(cstack_t * stp,char * nmp,tlm_acls_t * acls)152 dtree_push(cstack_t *stp, char *nmp, tlm_acls_t *acls)
153 {
154 int len;
155 stack_ent_t *sp;
156
157 sp = ndmp_malloc(sizeof (stack_ent_t));
158 if (!sp || !nmp || !acls) {
159 free(sp);
160 return (-1);
161 }
162
163 len = strlen(nmp) + 1;
164 sp->se_name = ndmp_malloc(len);
165 if (!sp->se_name) {
166 free(sp);
167 return (-1);
168 }
169
170 (void) strlcpy(sp->se_name, nmp, len);
171 (void) memcpy(&sp->se_acls, acls, sizeof (*acls));
172 (void) memset(acls, 0, sizeof (tlm_acls_t));
173
174 return (cstack_push(stp, (void *)sp, sizeof (*sp)));
175 }
176
177 /*
178 * dtree_pop
179 */
180 int
dtree_pop(cstack_t * stp)181 dtree_pop(cstack_t *stp)
182 {
183 int err;
184 stack_ent_t *sp;
185
186 err = cstack_pop(stp, (void **)&sp, (void *)NULL);
187 if (err)
188 return (-1);
189
190 err = set_acl(sp->se_name, &sp->se_acls);
191
192 free(sp->se_name);
193 free(sp);
194 return (err);
195 }
196
197
198 /*
199 * dtree_peek
200 */
201 char *
dtree_peek(cstack_t * stp)202 dtree_peek(cstack_t *stp)
203 {
204 int err;
205 stack_ent_t *sp;
206
207 err = cstack_top(stp, (void **)&sp, (void *)NULL);
208 if (err)
209 return (NULL);
210
211 return (sp->se_name);
212 }
213
214 /*
215 * NBU and EBS may not send us the correct file list containing hardlinks
216 * during a DAR restore, e.g. they appear always send the first name
217 * associated with an inode, even if other link names were
218 * selected for the restore. As a workaround, we use the file name entry
219 * in sels[] (ignore the name in the tar header) as restore target.
220 */
221 static char *
rs_darhl_new_name(struct rs_name_maker * rnp,char * name,char ** sels,int * pos,char * longname)222 rs_darhl_new_name(struct rs_name_maker *rnp, char *name, char **sels, int *pos,
223 char *longname)
224 {
225 int x;
226
227 for (x = 0; sels[x] != NULL; x++) {
228 if (strcmp(sels[x], " ")) {
229 *pos = x;
230 (void) strlcpy(longname, sels[x], TLM_MAX_PATH_NAME);
231 NDMP_LOG(LOG_DEBUG,
232 "to replace hardlink name [%s], pos [%d]",
233 longname, *pos);
234
235 return (rs_new_name(rnp, name, *pos, longname));
236 }
237 }
238
239 return (NULL);
240 }
241
242
243 /*
244 * Main dir restore function for tar
245 *
246 * If this function returns non-zero return value it means that fatal error
247 * was encountered.
248 */
249 int
tar_getdir(tlm_commands_t * commands,tlm_cmd_t * local_commands,tlm_job_stats_t * job_stats,struct rs_name_maker * rnp,int lib,int drv,char ** sels,char ** exls,int flags,int DAR,char * bk_path,struct hardlink_q * hardlink_q)250 tar_getdir(tlm_commands_t *commands,
251 tlm_cmd_t *local_commands,
252 tlm_job_stats_t *job_stats,
253 struct rs_name_maker *rnp,
254 int lib,
255 int drv,
256 char **sels, /* what to get off the tape */
257 char **exls, /* what to leave behind */
258 int flags,
259 int DAR,
260 char *bk_path,
261 struct hardlink_q *hardlink_q)
262 {
263 int fp = 0; /* file being restored ... */
264 /* ...need to preserve across volume changes */
265 tlm_acls_t *acls; /* file access info */
266 char *longname;
267 boolean_t is_long_name = FALSE;
268 char *longlink;
269 char *hugename;
270 longlong_t huge_size = 0; /* size of a HUGE file */
271 long acl_spot; /* any ACL info on the next volume */
272 long file_size; /* size of file to restore */
273 long size_left = 0; /* need this after volume change */
274 int last_action = 0; /* what we are doing at EOT */
275 boolean_t multi_volume = FALSE; /* is this a multi-volume switch ? */
276 int chk_rv; /* scratch area */
277
278 int mchtype, pos;
279 /*
280 * if an exact match is found for
281 * restore and its position in the
282 * selections list
283 */
284 int nzerohdr; /* the number of empty tar headers */
285 int rv;
286 long nm_end, lnk_end;
287 char *name, *nmp;
288 cstack_t *stp;
289 char *bkpath;
290 char *parentlnk;
291 int dir_dar = 0;
292
293 /*
294 * The directory where temporary files may be created during a partial
295 * non-DAR restore of hardlinks. It is intended to be initialized by
296 * an environment variable that can be set by user.
297 *
298 * It is not initialized for now. We keep it here for future use.
299 */
300 char *tmplink_dir = NULL;
301 int dar_recovered = 0;
302 char *thname_buf;
303
304 /*
305 * startup
306 */
307
308 longname = ndmp_malloc(TLM_MAX_PATH_NAME);
309 longlink = ndmp_malloc(TLM_MAX_PATH_NAME);
310 hugename = ndmp_malloc(TLM_MAX_PATH_NAME);
311 parentlnk = ndmp_malloc(TLM_MAX_PATH_NAME);
312 thname_buf = ndmp_malloc(TLM_MAX_PATH_NAME);
313 name = ndmp_malloc(TLM_MAX_PATH_NAME);
314 acls = ndmp_malloc(sizeof (tlm_acls_t));
315 stp = cstack_new();
316 if (longname == NULL || longlink == NULL || hugename == NULL ||
317 name == NULL || acls == NULL || stp == NULL || parentlnk == NULL ||
318 thname_buf == NULL) {
319 cstack_delete(stp);
320 free(longname);
321 free(longlink);
322 free(hugename);
323 free(parentlnk);
324 free(name);
325 free(acls);
326 free(thname_buf);
327 return (-TLM_NO_SCRATCH_SPACE);
328 }
329
330 acl_spot = 0;
331 *hugename = '\0';
332 *parentlnk = '\0';
333 nm_end = 0;
334 *longname = '\0';
335 lnk_end = 0;
336 *longlink = '\0';
337 (void) memset(acls, 0, sizeof (tlm_acls_t));
338 if (IS_SET(flags, RSFLG_OVR_ALWAYS)) {
339 acls->acl_overwrite = TRUE;
340 NDMP_LOG(LOG_DEBUG, "RSFLG_OVR_ALWAYS");
341 } else if (IS_SET(flags, RSFLG_OVR_UPDATE)) {
342 acls->acl_update = TRUE;
343 NDMP_LOG(LOG_DEBUG, "RSFLG_OVR_UPDATE");
344 }
345
346 /*
347 * work
348 */
349 rv = 0;
350 nzerohdr = 0;
351 while (commands->tcs_writer != TLM_ABORT &&
352 local_commands->tc_writer != TLM_STOP && rv == 0) {
353 tlm_tar_hdr_t fake_tar_hdr;
354 char *file_name;
355 char *link_name;
356 int erc;
357 int actual_size;
358 boolean_t want_this_file;
359 int want = sizeof (tlm_tar_hdr_t);
360 tlm_tar_hdr_t *tar_hdr;
361
362 /* The inode of an LF_LINK type. */
363 unsigned long hardlink_inode = 0;
364
365 /*
366 * Indicate whether a file with the same inode has been
367 * restored.
368 */
369 int hardlink_done = 0;
370
371 /* The path of the restored hardlink file */
372 char *hardlink_target = NULL;
373 int is_hardlink = 0;
374
375 /*
376 * Whether a temporary file should be created for restoring
377 * hardlink.
378 */
379 int hardlink_tmp_file = 0;
380 char *hardlink_tmp_name = ".tmphlrsnondar";
381
382 /* used to make up hardlink_tmp_name */
383 static int hardlink_tmp_idx = 0;
384
385 if (multi_volume) {
386 NDMP_LOG(LOG_DEBUG, "multi_volume %c %d",
387 last_action, size_left);
388
389 /*
390 * the previous volume is out of data
391 * and is back in the rack, a new tape
392 * is loaded and ready to read.
393 *
394 * We need to pick up where we left off.
395 */
396 (void) memset(&fake_tar_hdr, 0, sizeof (fake_tar_hdr));
397 file_size = size_left;
398 tar_hdr = &fake_tar_hdr;
399 tar_hdr->th_linkflag = last_action;
400
401 multi_volume = FALSE;
402 last_action = 0;
403 } else {
404 tar_hdr = (tlm_tar_hdr_t *)get_read_buffer(want,
405 &erc, &actual_size, local_commands);
406
407 if (tar_hdr == NULL) {
408 rv = -1;
409 continue;
410 }
411
412 /*
413 * we can ignore read errors here because
414 * 1) they are logged by Restore Reader
415 * 2) we are not doing anything important here
416 * just looking for the next work record.
417 */
418 if (actual_size < want) {
419 /*
420 * EOF hits here
421 *
422 * wait for another buffer to come along
423 * or until the Reader thread tells us
424 * that no more tapes will be loaded ...
425 * time to stop.
426 */
427 continue;
428 }
429
430 /*
431 * check for "we are lost"
432 */
433 chk_rv = tlm_vfy_tar_checksum(tar_hdr);
434 if (chk_rv == 0) {
435 /* one of the end of tar file marks */
436 if (++nzerohdr >= 2) {
437 NDMP_LOG(LOG_DEBUG,
438 "nzerohdr %d, breaking",
439 nzerohdr);
440 /* end of tar file */
441 break;
442 }
443 NDMP_LOG(LOG_DEBUG, "nzerohdr %d, continuing",
444 nzerohdr);
445 continue;
446 } else if (chk_rv < 0) {
447 nzerohdr = 0;
448 /* skip this record */
449 continue;
450 }
451 nzerohdr = 0;
452
453 /*
454 * When files are spanned to the next tape, the
455 * information of the acls must not be over-written
456 * by the information of the LF_MULTIVOL and LF_VOLHDR
457 * header, whose information is irrelevant to the file.
458 * The information of the original header must be
459 * kept in the 'acl'.
460 */
461 if (tar_hdr->th_linkflag != LF_MULTIVOL &&
462 tar_hdr->th_linkflag != LF_VOLHDR) {
463 if (tar_hdr->th_linkflag != LF_HUMONGUS) {
464 acls->acl_attr.st_mode =
465 oct_atoi(tar_hdr->th_mode);
466 acls->acl_attr.st_size =
467 oct_atoi(tar_hdr->th_size);
468 acls->acl_attr.st_uid =
469 oct_atoi(tar_hdr->th_uid);
470 acls->acl_attr.st_gid =
471 oct_atoi(tar_hdr->th_gid);
472 acls->acl_attr.st_mtime =
473 oct_atoi(tar_hdr->th_mtime);
474 (void) strlcpy(acls->uname,
475 tar_hdr->th_uname,
476 sizeof (acls->uname));
477 (void) strlcpy(acls->gname,
478 tar_hdr->th_gname,
479 sizeof (acls->gname));
480 }
481 file_size = oct_atoi(tar_hdr->th_size);
482 acl_spot = 0;
483 last_action = tar_hdr->th_linkflag;
484 }
485 }
486
487 NDMP_LOG(LOG_DEBUG, "n [%s] f [%c] s %lld m %o u %d g %d t %d",
488 tar_hdr->th_name, tar_hdr->th_linkflag,
489 acls->acl_attr.st_size, acls->acl_attr.st_mode,
490 acls->acl_attr.st_uid, acls->acl_attr.st_gid,
491 acls->acl_attr.st_mtime);
492
493 /*
494 * If the restore is running using DAR we should check for
495 * extended attribute entries
496 */
497 if (dar_recovered &&
498 tar_hdr->th_linkflag != LF_XATTR)
499 break;
500
501 rs_create_new_bkpath(bk_path, tar_hdr->th_name, thname_buf);
502
503 switch (tar_hdr->th_linkflag) {
504 case LF_MULTIVOL:
505 multi_volume = TRUE;
506 break;
507 case LF_LINK:
508 is_hardlink = 1;
509 hardlink_inode =
510 oct_atoi(tar_hdr->th_shared.th_hlink_ino);
511
512 /*
513 * Check if we have restored a link with the same inode
514 * If the inode is 0, we have to restore it as a
515 * regular file.
516 */
517 if (hardlink_inode) {
518 hardlink_done = !hardlink_q_get(hardlink_q,
519 hardlink_inode, 0, &hardlink_target);
520 }
521
522 if (hardlink_done) {
523 NDMP_LOG(LOG_DEBUG,
524 "found hardlink, inode = %u, target = [%s]",
525 hardlink_inode,
526 hardlink_target? hardlink_target : "--");
527
528 /* create a hardlink to hardlink_target */
529 file_name = (*longname == 0) ?
530 thname_buf : longname;
531
532 if (!is_file_wanted(file_name, sels, exls,
533 flags, &mchtype, &pos)) {
534 nmp = NULL;
535 /*
536 * This means that DMA did not send us
537 * the correct fh_info for the file
538 * in restore list. We use the file
539 * name entry in sels[] (ignore the
540 * name in the tar header) as restore
541 * target.
542 */
543 if (DAR) {
544 nmp = rs_darhl_new_name(rnp,
545 name, sels, &pos,
546 file_name);
547 }
548 } else {
549 nmp = rs_new_name(rnp, name, pos,
550 file_name);
551 if (!nmp) {
552 NDMP_LOG(LOG_DEBUG,
553 "can't make name for %s",
554 longname);
555 }
556 }
557
558 if (nmp) {
559 if (hardlink_target) {
560 erc = create_hard_link(
561 hardlink_target, nmp,
562 acls, job_stats);
563 if (ERROR_IS_FATAL(erc)) {
564 rv = erc;
565 continue;
566 }
567 if (erc == 0) {
568 (void)
569 tlm_entry_restored(
570 job_stats,
571 file_name, pos);
572 NDMP_LOG(LOG_DEBUG,
573 "restored %s -> %s",
574 nmp,
575 hardlink_target);
576 }
577 } else {
578 NDMP_LOG(LOG_DEBUG,
579 "no target for hardlink %s",
580 nmp);
581 }
582
583 name[0] = 0;
584 is_long_name = FALSE;
585 }
586
587 nm_end = 0;
588 longname[0] = 0;
589 lnk_end = 0;
590 longlink[0] = 0;
591
592 break;
593 }
594 /* otherwise fall through, restore like a normal file */
595 /*FALLTHROUGH*/
596 case LF_OLDNORMAL:
597 /*
598 * check for TAR's end-of-tape method
599 * of zero filled records.
600 */
601 if (tar_hdr->th_name[0] == 0) {
602 break;
603 }
604 /*
605 * otherwise fall through,
606 * this is an old style normal file header
607 */
608 /*FALLTHROUGH*/
609 case LF_NORMAL:
610 case LF_CONTIG:
611 job_stats->js_files_so_far++;
612 if (*hugename != 0) {
613 (void) strlcpy(longname, hugename,
614 TLM_MAX_PATH_NAME);
615 } else if (*longname == 0) {
616 if (tar_hdr->th_name[0] != '/') {
617 /*
618 * check for old tar format, it
619 * does not have a leading "/"
620 */
621 longname[0] = '/';
622 longname[1] = 0;
623 (void) strlcat(longname,
624 tar_hdr->th_name,
625 TLM_MAX_PATH_NAME);
626 } else {
627 (void) strlcpy(longname,
628 thname_buf,
629 TLM_MAX_PATH_NAME);
630 }
631 }
632
633 want_this_file = is_file_wanted(longname, sels, exls,
634 flags, &mchtype, &pos);
635 if (!want_this_file) {
636 nmp = NULL;
637 /*
638 * This means that DMA did not send us valid
639 * fh_info for the file in restore list. We
640 * use the file name entry in sels[] (ignore
641 * the name in the tar header) as restore
642 * target.
643 */
644 if (DAR && (tar_hdr->th_linkflag == LF_LINK)) {
645 nmp = rs_darhl_new_name(rnp, name,
646 sels, &pos, longname);
647 if (nmp == NULL) {
648 rv = ENOMEM;
649 continue;
650 }
651
652 want_this_file = TRUE;
653 mchtype = PM_EXACT;
654 }
655 } else {
656 nmp = rs_new_name(rnp, name, pos, longname);
657 if (!nmp)
658 want_this_file = FALSE;
659 }
660
661 if (nmp)
662 (void) strlcpy(parentlnk, nmp, strlen(nmp) + 1);
663
664 /*
665 * For a hardlink, even if it's not asked to be
666 * restored, we restore it to a temporary location,
667 * in case other links to the same file need to be
668 * restored later.
669 *
670 * The temp files are created in tmplink_dir, with
671 * names like ".tmphlrsnondar*". They are cleaned up
672 * at the completion of a restore. However, if a
673 * restore were interrupted, e.g. by a system reboot,
674 * they would have to be cleaned up manually in order
675 * for the disk space to be freed.
676 *
677 * If tmplink_dir is NULL, no temperorary files are
678 * created during a restore. This may result in some
679 * hardlinks not being restored during a partial
680 * restore.
681 */
682 if (is_hardlink && !DAR && !want_this_file && !nmp) {
683 if (tmplink_dir) {
684 (void) snprintf(name, TLM_MAX_PATH_NAME,
685 "%s/%s_%d", tmplink_dir,
686 hardlink_tmp_name,
687 hardlink_tmp_idx);
688 nmp = name;
689
690 hardlink_tmp_idx++;
691 hardlink_tmp_file = 1;
692 want_this_file = TRUE;
693 NDMP_LOG(LOG_DEBUG,
694 "To restore temp hardlink file %s.",
695 nmp);
696 } else {
697 NDMP_LOG(LOG_DEBUG,
698 "No tmplink_dir specified.");
699 }
700 }
701
702 rv = restore_file(&fp, nmp, file_size,
703 huge_size, acls, want_this_file, local_commands,
704 job_stats, &size_left);
705 if (rv != 0)
706 continue;
707
708 /*
709 * In the case of non-DAR, we have to record the first
710 * link for an inode that has multiple links. That's
711 * the only link with data records actually backed up.
712 * In this way, when we run into the other links, they
713 * will be treated as links, and we won't go to look
714 * for the data records to restore. This is not a
715 * problem for DAR, where DMA tells the tape where
716 * to locate the data records.
717 */
718 if (is_hardlink && !DAR) {
719 if (hardlink_q_add(hardlink_q, hardlink_inode,
720 0, nmp, hardlink_tmp_file))
721 NDMP_LOG(LOG_DEBUG,
722 "failed to add (%u, %s) to HL q",
723 hardlink_inode, nmp);
724 }
725
726 /* remove / reverse the temporary stuff */
727 if (hardlink_tmp_file) {
728 nmp = NULL;
729 want_this_file = FALSE;
730 hardlink_tmp_file = 0;
731 }
732
733 /*
734 * Check if it is time to set the attribute
735 * of the restored directory
736 */
737 while (nmp && ((bkpath = dtree_peek(stp)) != NULL)) {
738 int erc;
739
740 if (strstr(nmp, bkpath))
741 break;
742
743 erc = dtree_pop(stp);
744 if (ERROR_IS_FATAL(erc)) {
745 rv = erc;
746 break;
747 }
748 }
749 if (rv != 0)
750 continue;
751
752 NDMP_LOG(LOG_DEBUG, "sizeleft %s %d, %lld", longname,
753 size_left, huge_size);
754
755 if (want_this_file) {
756 job_stats->js_bytes_total += file_size;
757 job_stats->js_files_total++;
758 }
759
760 huge_size -= file_size;
761 if (huge_size < 0) {
762 huge_size = 0;
763 }
764 if (size_left == 0 && huge_size == 0) {
765 if (PM_EXACT_OR_CHILD(mchtype)) {
766 (void) tlm_entry_restored(job_stats,
767 longname, pos);
768
769 /*
770 * Add an entry to hardlink_q to record
771 * this hardlink.
772 */
773 if (is_hardlink) {
774 NDMP_LOG(LOG_DEBUG,
775 "Restored hardlink file %s",
776 nmp);
777
778 if (DAR) {
779 (void) hardlink_q_add(
780 hardlink_q,
781 hardlink_inode, 0,
782 nmp, 0);
783 }
784 }
785 }
786
787 nm_end = 0;
788 longname[0] = 0;
789 lnk_end = 0;
790 longlink[0] = 0;
791 hugename[0] = 0;
792 name[0] = 0;
793 is_long_name = FALSE;
794 }
795 break;
796 case LF_XATTR:
797 file_name = (*longname == 0) ? thname_buf :
798 longname;
799
800 size_left = restore_xattr_hdr(&fp, parentlnk,
801 file_name, file_size, acls, local_commands,
802 job_stats);
803
804 break;
805 case LF_SYMLINK:
806 file_name = (*longname == 0) ? thname_buf :
807 longname;
808 link_name = (*longlink == 0) ?
809 tar_hdr->th_linkname : longlink;
810 NDMP_LOG(LOG_DEBUG, "file_name[%s]", file_name);
811 NDMP_LOG(LOG_DEBUG, "link_name[%s]", link_name);
812 if (is_file_wanted(file_name, sels, exls, flags,
813 &mchtype, &pos)) {
814 nmp = rs_new_name(rnp, name, pos, file_name);
815 if (nmp) {
816 erc = create_sym_link(nmp, link_name,
817 acls, job_stats);
818 if (ERROR_IS_FATAL(erc)) {
819 rv = erc;
820 continue;
821 }
822 if (erc == 0 &&
823 PM_EXACT_OR_CHILD(mchtype))
824 (void) tlm_entry_restored(
825 job_stats, file_name, pos);
826 name[0] = 0;
827 }
828 }
829 nm_end = 0;
830 longname[0] = 0;
831 lnk_end = 0;
832 longlink[0] = 0;
833 break;
834 case LF_DIR:
835 file_name = *longname == 0 ? thname_buf :
836 longname;
837 if (is_file_wanted(file_name, sels, exls, flags,
838 &mchtype, &pos)) {
839 dir_dar = DAR;
840 nmp = rs_new_name(rnp, name, pos, file_name);
841 if (nmp && mchtype != PM_PARENT) {
842 (void) strlcpy(parentlnk, nmp,
843 strlen(nmp));
844 erc = create_directory(nmp, job_stats);
845 if (ERROR_IS_FATAL(erc)) {
846 rv = erc;
847 continue;
848 }
849 if (erc == 0 &&
850 PM_EXACT_OR_CHILD(mchtype))
851 (void) tlm_entry_restored(
852 job_stats, file_name, pos);
853 /*
854 * Check if it is time to set
855 * the attribute of the restored
856 * directory
857 */
858 while ((bkpath = dtree_peek(stp))
859 != NULL) {
860 int rc;
861
862 if (strstr(nmp, bkpath))
863 break;
864 rc = dtree_pop(stp);
865 if (ERROR_IS_FATAL(rc)) {
866 rv = rc;
867 break;
868 }
869 }
870 if (rv != 0)
871 continue;
872
873 (void) dtree_push(stp, nmp, acls);
874 name[0] = 0;
875 }
876 } else {
877 dir_dar = 0;
878 }
879 nm_end = 0;
880 longname[0] = 0;
881 lnk_end = 0;
882 longlink[0] = 0;
883 break;
884 case LF_FIFO:
885 case LF_BLK:
886 case LF_CHR:
887 file_name = *longname == 0 ? thname_buf :
888 longname;
889 if (is_file_wanted(file_name, sels, exls, flags,
890 &mchtype, &pos)) {
891 nmp = rs_new_name(rnp, name, pos, file_name);
892 if (nmp) {
893 erc = create_special(
894 tar_hdr->th_linkflag, nmp, acls,
895 oct_atoi(tar_hdr->th_shared.
896 th_dev.th_devmajor),
897 oct_atoi(tar_hdr->th_shared.
898 th_dev.th_devminor), job_stats);
899 if (ERROR_IS_FATAL(erc)) {
900 rv = erc;
901 continue;
902 }
903 if (erc == 0 &&
904 PM_EXACT_OR_CHILD(mchtype))
905 (void) tlm_entry_restored(
906 job_stats, file_name, pos);
907 name[0] = 0;
908 }
909 }
910 nm_end = 0;
911 longname[0] = 0;
912 lnk_end = 0;
913 longlink[0] = 0;
914 break;
915 case LF_LONGLINK:
916 file_size = min(file_size,
917 TLM_MAX_PATH_NAME - lnk_end - 1);
918 file_size = max(0, file_size);
919 size_left = get_long_name(lib, drv, file_size, longlink,
920 &lnk_end, local_commands);
921
922 if (size_left != 0)
923 NDMP_LOG(LOG_DEBUG,
924 "fsize %d sleft %d lnkend %d",
925 file_size, size_left, lnk_end);
926 break;
927 case LF_LONGNAME:
928 file_size = min(file_size,
929 TLM_MAX_PATH_NAME - nm_end - 1);
930 file_size = max(0, file_size);
931 size_left = get_long_name(lib, drv, file_size, longname,
932 &nm_end, local_commands);
933
934 if (size_left != 0)
935 NDMP_LOG(LOG_DEBUG,
936 "fsize %d sleft %d nmend %d",
937 file_size, size_left, nm_end);
938 is_long_name = TRUE;
939 break;
940 case LF_ACL:
941 size_left = load_acl_info(lib, drv, file_size, acls,
942 &acl_spot, local_commands);
943 break;
944 case LF_VOLHDR:
945 break;
946 case LF_HUMONGUS:
947 (void) memset(hugename, 0, TLM_MAX_PATH_NAME);
948 (void) get_humongus_file_header(lib, drv, file_size,
949 &huge_size, hugename, local_commands);
950 break;
951 default:
952 break;
953
954 }
955
956 /*
957 * If the restore is running using DAR we should check for
958 * long file names and HUGE file sizes.
959 */
960 if (DAR && tar_hdr->th_linkflag != LF_ACL &&
961 tar_hdr->th_linkflag != LF_XATTR &&
962 !huge_size && !is_long_name && !dir_dar)
963 dar_recovered = 1;
964 }
965
966 /*
967 * tear down
968 */
969 if (rv != 0)
970 commands->tcs_reader = TLM_ABORT;
971 if (fp != 0) {
972 (void) close(fp);
973 }
974 while (dtree_pop(stp) != -1)
975 ;
976 cstack_delete(stp);
977 free(acls);
978 free(longname);
979 free(parentlnk);
980 free(longlink);
981 free(hugename);
982 free(name);
983 free(thname_buf);
984 return (rv);
985 }
986
987 /*
988 * Main file restore function for tar (should run as a thread)
989 */
990 int
tar_getfile(tlm_backup_restore_arg_t * argp)991 tar_getfile(tlm_backup_restore_arg_t *argp)
992 {
993 tlm_job_stats_t *job_stats;
994 char **sels; /* list of files desired */
995 char **exls; /* list of files not wanted */
996 char *dir; /* where to restore the files */
997 char job[TLM_MAX_BACKUP_JOB_NAME+1];
998 /* the restore job name */
999 int erc; /* error return codes */
1000 int flags;
1001 struct rs_name_maker rn;
1002 tlm_commands_t *commands;
1003 tlm_cmd_t *local_commands;
1004 char *list = NULL;
1005
1006 commands = argp->ba_commands;
1007 local_commands = argp->ba_cmd;
1008
1009 flags = 0;
1010
1011 dir = ndmp_malloc(TLM_MAX_PATH_NAME);
1012 if (dir == NULL) {
1013 local_commands->tc_reader = TLM_STOP;
1014 (void) pthread_barrier_wait(&argp->ba_barrier);
1015 return (-1);
1016 }
1017
1018 (void) strlcpy(job, argp->ba_job, TLM_MAX_BACKUP_JOB_NAME+1);
1019 (void) strlcpy(dir, argp->ba_dir, TLM_MAX_PATH_NAME);
1020
1021 flags |= RSFLG_OVR_ALWAYS;
1022 flags |= RSFLG_IGNORE_CASE;
1023
1024 /*
1025 * do not test for "dir" having no string, since that
1026 * is a legal condition. Restore to origional location
1027 * will not have a restore directory.
1028 */
1029 if (*job == '\0') {
1030 NDMP_LOG(LOG_DEBUG, "No job defined");
1031 local_commands->tc_reader = TLM_STOP;
1032 free(dir);
1033 (void) pthread_barrier_wait(&argp->ba_barrier);
1034 return (-1);
1035 }
1036
1037 sels = argp->ba_sels;
1038 if (sels == NULL) {
1039 local_commands->tc_reader = TLM_STOP;
1040 free(dir);
1041 (void) pthread_barrier_wait(&argp->ba_barrier);
1042 return (-1);
1043 }
1044 exls = &list;
1045
1046 tlm_log_list("selections", sels);
1047 tlm_log_list("exclusions", exls);
1048
1049 if (wildcard_enabled())
1050 flags |= RSFLG_MATCH_WCARD;
1051
1052 local_commands->tc_ref++;
1053 commands->tcs_writer_count++;
1054
1055 /*
1056 * let the launcher continue
1057 */
1058 (void) pthread_barrier_wait(&argp->ba_barrier);
1059
1060 job_stats = tlm_ref_job_stats(job);
1061
1062 rn.rn_fp = catnames;
1063 rn.rn_nlp = dir;
1064
1065 /*
1066 * work
1067 */
1068 NDMP_LOG(LOG_DEBUG, "start restore job %s", job);
1069 erc = tar_getdir(commands, local_commands, job_stats, &rn, 1, 1,
1070 sels, exls, flags, 0, NULL, NULL);
1071
1072 /*
1073 * teardown
1074 */
1075 NDMP_LOG(LOG_DEBUG, "end restore job %s", job);
1076 tlm_un_ref_job_stats(job);
1077 tlm_release_list(sels);
1078 tlm_release_list(exls);
1079
1080 commands->tcs_writer_count--;
1081 local_commands->tc_reader = TLM_STOP;
1082 tlm_release_reader_writer_ipc(local_commands);
1083 free(dir);
1084 return (erc);
1085 }
1086
1087 /*
1088 * Creates the directories all the way down to the
1089 * end if they dont exist
1090 */
1091 int
make_dirs(char * dir)1092 make_dirs(char *dir)
1093 {
1094 char c;
1095 char *cp, *end;
1096 struct stat64 st;
1097
1098 cp = dir;
1099 cp += strspn(cp, "/");
1100 end = dir + strlen(dir);
1101 do {
1102 if (*cp == '\0' || *cp == '/') {
1103 c = *cp;
1104 *cp = '\0';
1105 if (lstat64(dir, &st) < 0)
1106 if (mkdir(dir, 0777) < 0) {
1107 NDMP_LOG(LOG_DEBUG, "Error %d"
1108 " creating directory %s",
1109 errno, dir);
1110 *cp = c;
1111 return (errno);
1112 }
1113
1114 *cp = c;
1115 }
1116 } while (++cp <= end);
1117
1118 return (0);
1119 }
1120
1121 /*
1122 * Creates the directories leading to the given path
1123 */
1124 int
mkbasedir(char * path)1125 mkbasedir(char *path)
1126 {
1127 int rv;
1128 char *cp;
1129 struct stat64 st;
1130
1131 if (!path || !*path) {
1132 NDMP_LOG(LOG_DEBUG, "Invalid argument");
1133 return (-1);
1134 }
1135
1136 cp = strrchr(path, '/');
1137 if (cp)
1138 *cp = '\0';
1139 rv = lstat64(path, &st);
1140 if (rv < 0) /* need new directories */
1141 rv = make_dirs(path);
1142 if (cp)
1143 *cp = '/';
1144
1145 return (rv);
1146 }
1147
1148
1149 /*
1150 * read the file off the tape back onto disk
1151 *
1152 * If the function returns a non-zero return code, it means that fatal error
1153 * was encountered and restore should terminate immediately.
1154 */
1155 static int
restore_file(int * fp,char * real_name,long size,longlong_t huge_size,tlm_acls_t * acls,boolean_t want_this_file,tlm_cmd_t * local_commands,tlm_job_stats_t * job_stats,long * size_left)1156 restore_file(int *fp,
1157 char *real_name,
1158 long size,
1159 longlong_t huge_size,
1160 tlm_acls_t *acls,
1161 boolean_t want_this_file,
1162 tlm_cmd_t *local_commands,
1163 tlm_job_stats_t *job_stats,
1164 long *size_left)
1165 {
1166 struct stat64 attr;
1167 int ret, rv;
1168
1169 *size_left = 0;
1170 if (!real_name) {
1171 if (want_this_file) {
1172 NDMP_LOG(LOG_DEBUG, "No file name but wanted!");
1173 want_this_file = FALSE;
1174 }
1175 } else
1176 NDMP_LOG(LOG_DEBUG, "new file[%s]", real_name);
1177
1178 /*
1179 * OK, some FM is creeping in here ...
1180 * int *fp is used to keep the
1181 * backup file channel open through
1182 * the interruption of EOT and
1183 * processing the headers of the
1184 * next tape. So, if *fp is zero
1185 * then no file is open yet and all
1186 * is normal. If *fp has a number
1187 * then we are returning after an
1188 * EOT break.
1189 *
1190 * *fp is now also open for HUGE files
1191 * that are put back in sections.
1192 */
1193
1194 if (*fp == 0 && want_this_file) {
1195
1196 ret = mkbasedir(real_name);
1197 if (ret != 0) {
1198 job_stats->js_errors++;
1199 if (ERROR_IS_FATAL(ret))
1200 return (ret);
1201 }
1202
1203 ret = stat64(real_name, (struct stat64 *)&attr);
1204 if (ret < 0) {
1205 /*EMPTY*/
1206 /* new file */
1207 } else if (acls->acl_overwrite) {
1208 /*EMPTY*/
1209 /* take this file no matter what */
1210 } else if (acls->acl_update) {
1211 if (attr.st_mtime < acls->acl_attr.st_mtime) {
1212 /*EMPTY*/
1213 /* tape is newer */
1214 } else {
1215 /* disk file is newer */
1216 want_this_file = FALSE;
1217 }
1218 } else {
1219 /*
1220 * no overwrite, no update,
1221 * do not ever replace old files.
1222 */
1223 want_this_file = TRUE;
1224 }
1225 if (want_this_file) {
1226
1227 *fp = open(real_name, O_CREAT | O_TRUNC | O_WRONLY,
1228 S_IRUSR | S_IWUSR);
1229 if (*fp == -1) {
1230 NDMP_LOG(LOG_ERR,
1231 "Could not open %s for restore: %d",
1232 real_name, errno);
1233 job_stats->js_errors++;
1234 want_this_file = FALSE;
1235 /*
1236 * In case of non-fatal error we cannot return
1237 * here, because the file is still on the tape
1238 * and must be skipped over.
1239 */
1240 if (ERROR_IS_FATAL(errno))
1241 return (errno);
1242 }
1243 }
1244 (void) strlcpy(local_commands->tc_file_name, real_name,
1245 TLM_MAX_PATH_NAME);
1246 }
1247
1248 /*
1249 * this is the size left in the next segment
1250 */
1251 huge_size -= size;
1252
1253 /*
1254 * work
1255 */
1256 rv = 0;
1257 while (size > 0 && local_commands->tc_writer == TLM_RESTORE_RUN) {
1258 int actual_size;
1259 int error;
1260 char *rec;
1261 int write_size;
1262
1263 /*
1264 * Use bytes_in_file field to tell reader the amount
1265 * of data still need to be read for this file.
1266 */
1267 job_stats->js_bytes_in_file = size;
1268
1269 error = 0;
1270 rec = get_read_buffer(size, &error, &actual_size,
1271 local_commands);
1272 if (actual_size <= 0) {
1273 NDMP_LOG(LOG_DEBUG,
1274 "RESTORE WRITER> error %d, actual_size %d",
1275 error, actual_size);
1276
1277 /* no more data for this file for now */
1278 job_stats->js_bytes_in_file = 0;
1279 *size_left = size;
1280 return (0);
1281 } else if (error) {
1282 NDMP_LOG(LOG_DEBUG, "Error %d in file [%s]",
1283 error, local_commands->tc_file_name);
1284 break;
1285 }
1286
1287 write_size = min(size, actual_size);
1288 if (want_this_file) {
1289 ret = write(*fp, rec, write_size);
1290 if (ret < 0) {
1291 NDMP_LOG(LOG_ERR,
1292 "Write error %d for file [%s]", errno,
1293 local_commands->tc_file_name);
1294 job_stats->js_errors++;
1295 if (ERROR_IS_FATAL(errno)) {
1296 rv = errno;
1297 break;
1298 }
1299 } else {
1300 NS_ADD(wdisk, ret);
1301 NS_INC(wfile);
1302 if (ret < write_size) {
1303 NDMP_LOG(LOG_ERR,
1304 "Partial write for file [%s]",
1305 local_commands->tc_file_name);
1306 }
1307 }
1308 }
1309 size -= write_size;
1310 }
1311
1312 /* no more data for this file for now */
1313 job_stats->js_bytes_in_file = 0;
1314
1315 /*
1316 * teardown
1317 */
1318 if (*fp != 0 && huge_size <= 0) {
1319 (void) close(*fp);
1320 *fp = 0;
1321 if (rv == 0) {
1322 ret = set_acl(real_name, acls);
1323 if (ERROR_IS_FATAL(ret))
1324 return (ret);
1325 }
1326 }
1327 return (rv);
1328 }
1329
1330 /*
1331 * Set the extended attributes file attribute
1332 */
1333 static void
set_xattr(int fd,struct stat64 st)1334 set_xattr(int fd, struct stat64 st)
1335 {
1336 struct timeval times[2];
1337
1338 times[0].tv_sec = st.st_atime;
1339 times[1].tv_sec = st.st_mtime;
1340
1341 (void) fchmod(fd, st.st_mode);
1342 (void) fchown(fd, st.st_uid, st.st_gid);
1343 (void) futimesat(fd, ".", times);
1344 }
1345
1346 /*
1347 * Read the system attribute file in a single buffer to write
1348 * it as a single write. A partial write to system attribute would
1349 * cause an EINVAL on write.
1350 */
1351 static char *
get_read_one_buf(char * rec,int actual_size,int size,int * error,tlm_cmd_t * lc)1352 get_read_one_buf(char *rec, int actual_size, int size, int *error,
1353 tlm_cmd_t *lc)
1354 {
1355 char *buf, *p;
1356 int read_size;
1357 int len;
1358
1359 if (actual_size > size)
1360 return (rec);
1361
1362 buf = ndmp_malloc(size);
1363 if (buf == NULL) {
1364 *error = ENOMEM;
1365 return (NULL);
1366 }
1367 (void) memcpy(buf, rec, actual_size);
1368 rec = buf;
1369 buf += actual_size;
1370 while (actual_size < size) {
1371 p = get_read_buffer(size - actual_size, error, &read_size, lc);
1372 len = min(size - actual_size, read_size);
1373 (void) memcpy(buf, p, len);
1374 actual_size += len;
1375 buf += len;
1376 }
1377 return (rec);
1378 }
1379
1380
1381 /*
1382 * read the extended attribute header and write
1383 * it to the file
1384 */
1385 static long
restore_xattr_hdr(int * fp,char * name,char * fname,long size,tlm_acls_t * acls,tlm_cmd_t * local_commands,tlm_job_stats_t * job_stats)1386 restore_xattr_hdr(int *fp,
1387 char *name,
1388 char *fname,
1389 long size,
1390 tlm_acls_t *acls,
1391 tlm_cmd_t *local_commands,
1392 tlm_job_stats_t *job_stats)
1393 {
1394 tlm_tar_hdr_t *tar_hdr;
1395 struct xattr_hdr *xhdr;
1396 struct xattr_buf *xbuf;
1397 int namelen;
1398 char *xattrname;
1399 int actual_size;
1400 int error;
1401
1402 if (!fname) {
1403 NDMP_LOG(LOG_DEBUG, "No file name but wanted!");
1404 } else {
1405 NDMP_LOG(LOG_DEBUG, "new xattr[%s]", fname);
1406 }
1407
1408 error = 0;
1409 xhdr = (struct xattr_hdr *)get_read_buffer(size, &error,
1410 &actual_size, local_commands);
1411 if (xhdr == NULL || error != 0) {
1412 NDMP_LOG(LOG_DEBUG,
1413 "Could not read xattr [%s:%s] for restore. ",
1414 name, fname);
1415 job_stats->js_errors++;
1416 return (0);
1417 }
1418
1419 /* Check extended attribute header */
1420 if (strcmp(xhdr->h_version, XATTR_ARCH_VERS) != 0) {
1421 NDMP_LOG(LOG_DEBUG,
1422 "Unrecognized header format [%s]", xhdr->h_version);
1423 return (0);
1424 }
1425 xbuf = (struct xattr_buf *)(((char *)xhdr) + sizeof (struct xattr_hdr));
1426
1427 (void) sscanf(xbuf->h_namesz, "%7d", &namelen);
1428 xattrname = xbuf->h_names + strlen(xbuf->h_names) + 1;
1429
1430 if (*fp == 0) {
1431 int fd;
1432
1433 fd = attropen(name, xattrname, O_CREAT | O_RDWR, 0755);
1434 if (fd == -1) {
1435 NDMP_LOG(LOG_DEBUG,
1436 "Could not open xattr [%s:%s] for restore err=%d.",
1437 name, xattrname, errno);
1438 job_stats->js_errors++;
1439 return (0);
1440 }
1441 (void) strlcpy(local_commands->tc_file_name, xattrname,
1442 TLM_MAX_PATH_NAME);
1443 *fp = fd;
1444 }
1445
1446 /* Get the actual extended attribute file */
1447 tar_hdr = (tlm_tar_hdr_t *)get_read_buffer(sizeof (*tar_hdr),
1448 &error, &actual_size, local_commands);
1449 if (tar_hdr == NULL || error != 0) {
1450 NDMP_LOG(LOG_DEBUG,
1451 "Could not read xattr data [%s:%s] for restore. ",
1452 fname, xattrname);
1453 job_stats->js_errors++;
1454 return (0);
1455 }
1456 acls->acl_attr.st_mode = oct_atoi(tar_hdr->th_mode);
1457 acls->acl_attr.st_size = oct_atoi(tar_hdr->th_size);
1458 acls->acl_attr.st_uid = oct_atoi(tar_hdr->th_uid);
1459 acls->acl_attr.st_gid = oct_atoi(tar_hdr->th_gid);
1460 acls->acl_attr.st_mtime = oct_atoi(tar_hdr->th_mtime);
1461
1462 NDMP_LOG(LOG_DEBUG, "xattr_hdr: %s size %d mode %06o uid %d gid %d",
1463 xattrname, acls->acl_attr.st_size, acls->acl_attr.st_mode,
1464 acls->acl_attr.st_uid, acls->acl_attr.st_gid);
1465
1466 size = acls->acl_attr.st_size;
1467 while (size > 0 && local_commands->tc_writer == TLM_RESTORE_RUN) {
1468 char *rec;
1469 int write_size;
1470 int sysattr_write = 0;
1471
1472 error = 0;
1473 rec = get_read_buffer(size, &error, &actual_size,
1474 local_commands);
1475
1476 if ((actual_size < size) && sysattr_rw(xattrname)) {
1477 rec = get_read_one_buf(rec, actual_size, size, &error,
1478 local_commands);
1479 if (rec == NULL) {
1480 NDMP_LOG(LOG_DEBUG, "Error %d in file [%s]",
1481 error, xattrname);
1482 return (size);
1483 }
1484 actual_size = size;
1485 sysattr_write = 1;
1486 }
1487 if (actual_size <= 0) {
1488 NDMP_LOG(LOG_DEBUG,
1489 "RESTORE WRITER> error %d, actual_size %d",
1490 error, actual_size);
1491
1492 return (size);
1493 } else if (error) {
1494 NDMP_LOG(LOG_DEBUG, "Error %d in file [%s]",
1495 error, local_commands->tc_file_name);
1496 break;
1497 } else {
1498 write_size = min(size, actual_size);
1499 if ((write_size = write(*fp, rec, write_size)) < 0) {
1500 if (sysattr_write)
1501 free(rec);
1502
1503 break;
1504 }
1505
1506 NS_ADD(wdisk, write_size);
1507 NS_INC(wfile);
1508 size -= write_size;
1509 }
1510 if (sysattr_write)
1511 free(rec);
1512 }
1513
1514 if (*fp != 0) {
1515 set_xattr(*fp, acls->acl_attr);
1516 (void) close(*fp);
1517 *fp = 0;
1518 }
1519 return (0);
1520 }
1521
1522 /*
1523 * Match the name with the list
1524 */
1525 static int
exact_find(char * name,char ** list)1526 exact_find(char *name, char **list)
1527 {
1528 boolean_t found;
1529 int i;
1530 char *cp;
1531
1532 found = FALSE;
1533 for (i = 0; *list != NULL; list++, i++) {
1534 cp = *list + strspn(*list, "/");
1535 if (match(cp, name)) {
1536 found = TRUE;
1537 NDMP_LOG(LOG_DEBUG, "exact_find> found[%s]", cp);
1538 break;
1539 }
1540 }
1541
1542 return (found);
1543 }
1544
1545 /*
1546 * On error, return FALSE and prevent restoring(probably) unwanted data.
1547 */
1548 static int
is_parent(char * parent,char * child,int flags)1549 is_parent(char *parent, char *child, int flags)
1550 {
1551 char tmp[TLM_MAX_PATH_NAME];
1552 boolean_t rv;
1553
1554 if (IS_SET(flags, RSFLG_MATCH_WCARD)) {
1555 if (!tlm_cat_path(tmp, parent, "*")) {
1556 NDMP_LOG(LOG_DEBUG,
1557 "is_parent> path too long [%s]", parent);
1558 rv = FALSE;
1559 } else
1560 rv = (match(tmp, child) != 0) ? TRUE : FALSE;
1561 } else {
1562 if (!tlm_cat_path(tmp, parent, "/")) {
1563 NDMP_LOG(LOG_DEBUG,
1564 "is_parent> path too long [%s]", parent);
1565 rv = FALSE;
1566 } else
1567 rv = (strncmp(tmp, child, strlen(tmp)) == 0) ?
1568 TRUE : FALSE;
1569 }
1570
1571 return (rv);
1572 }
1573
1574 /*
1575 * Used to match the filename inside the list
1576 */
1577 static boolean_t
strexactcmp(char * s,char * t)1578 strexactcmp(char *s, char *t)
1579 {
1580 return ((strcmp(s, t) == 0) ? TRUE : FALSE);
1581 }
1582
1583 /*
1584 * Check if the file is needed to be restored
1585 */
1586 static boolean_t
is_file_wanted(char * name,char ** sels,char ** exls,int flags,int * mchtype,int * pos)1587 is_file_wanted(char *name,
1588 char **sels,
1589 char **exls,
1590 int flags,
1591 int *mchtype,
1592 int *pos)
1593 {
1594 char *p_sel;
1595 char *uc_name, *retry, *namep;
1596 boolean_t found;
1597 int i;
1598 name_match_fp_t *cmp_fp;
1599
1600 if (name == NULL || sels == NULL || exls == NULL)
1601 return (FALSE);
1602
1603 found = FALSE;
1604 if (mchtype != NULL)
1605 *mchtype = PM_NONE;
1606 if (pos != NULL)
1607 *pos = 0;
1608
1609 /*
1610 * For empty selection, restore everything
1611 */
1612 if (*sels == NULL || **sels == '\0') {
1613 NDMP_LOG(LOG_DEBUG, "is_file_wanted: Restore all");
1614 return (TRUE);
1615 }
1616
1617 retry = ndmp_malloc(TLM_MAX_PATH_NAME);
1618 if (retry == NULL)
1619 return (FALSE);
1620
1621 if (IS_SET(flags, RSFLG_MATCH_WCARD))
1622 cmp_fp = match;
1623 else
1624 cmp_fp = strexactcmp;
1625
1626 namep = name + strspn(name, "/");
1627
1628 if (IS_SET(flags, RSFLG_IGNORE_CASE)) {
1629 uc_name = ndmp_malloc(TLM_MAX_PATH_NAME);
1630 if (uc_name == NULL) {
1631 free(retry);
1632 return (FALSE);
1633 }
1634 (void) strlcpy(uc_name, namep, TLM_MAX_PATH_NAME);
1635 (void) strupr(uc_name);
1636 namep = uc_name;
1637 }
1638 NDMP_LOG(LOG_DEBUG, "is_file_wanted> flg: 0x%x name: [%s]",
1639 flags, name);
1640
1641 for (i = 0; *sels != NULL; sels++, i++) {
1642 p_sel = *sels + strspn(*sels, "/");
1643
1644 /*
1645 * Try exact match.
1646 */
1647 if ((*cmp_fp)(p_sel, namep)) {
1648 NDMP_LOG(LOG_DEBUG, "match1> pos: %d [%s][%s]",
1649 i, p_sel, name);
1650 found = TRUE;
1651 if (mchtype != NULL)
1652 *mchtype = PM_EXACT;
1653 break;
1654 }
1655 /*
1656 * Try "entry/" and the current selection. The
1657 * current selection may be something like "<something>/".
1658 */
1659 (void) tlm_cat_path(retry, namep, "/");
1660 if ((*cmp_fp)(p_sel, retry)) {
1661 NDMP_LOG(LOG_DEBUG, "match2> pos %d [%s][%s]",
1662 i, p_sel, name);
1663 found = TRUE;
1664 if (mchtype != NULL)
1665 *mchtype = PM_EXACT;
1666 break;
1667 }
1668 /*
1669 * If the following check returns true it means that the
1670 * 'name' is an entry below the 'p_sel' hierarchy.
1671 */
1672 if (is_parent(p_sel, namep, flags)) {
1673 NDMP_LOG(LOG_DEBUG, "parent1> pos %d [%s][%s]",
1674 i, p_sel, name);
1675 found = TRUE;
1676 if (mchtype != NULL)
1677 *mchtype = PM_CHILD;
1678 break;
1679 }
1680 /*
1681 * There is a special case for parent directories of a
1682 * selection. If 'p_sel' is something like "*d1", the
1683 * middle directories of the final entry can't be determined
1684 * until the final entry matches with 'p_sel'. At that
1685 * time the middle directories of the entry have been passed
1686 * and they can't be restored.
1687 */
1688 if (is_parent(namep, p_sel, flags)) {
1689 NDMP_LOG(LOG_DEBUG, "parent2> pos %d [%s][%s]",
1690 i, p_sel, name);
1691 found = TRUE;
1692 if (mchtype != NULL)
1693 *mchtype = PM_PARENT;
1694 break;
1695 }
1696 }
1697
1698 /* Check for exclusions. */
1699 if (found && exact_find(namep, exls)) {
1700 if (mchtype != NULL)
1701 *mchtype = PM_NONE;
1702 found = FALSE;
1703 }
1704 if (found && pos != NULL)
1705 *pos = i;
1706
1707 if (IS_SET(flags, RSFLG_IGNORE_CASE))
1708 free(uc_name);
1709 free(retry);
1710 return (found);
1711 }
1712
1713 /*
1714 * Read the specified amount data into the buffer. Detects EOT or EOF
1715 * during read.
1716 *
1717 * Returns the number of bytes actually read. On error returns -1.
1718 */
1719 static int
input_mem(int l,int d,tlm_cmd_t * lcmds,char * mem,int len)1720 input_mem(int l,
1721 int d,
1722 tlm_cmd_t *lcmds,
1723 char *mem,
1724 int len)
1725 {
1726 int err;
1727 int toread, actual_size, rec_size;
1728 char *rec;
1729
1730 if (l <= 0 || d <= 0 || !lcmds || !mem) {
1731 NDMP_LOG(LOG_DEBUG, "Invalid argument");
1732 return (-1);
1733 }
1734
1735 toread = len;
1736 while (toread > 0) {
1737 rec = get_read_buffer(toread, &err, &actual_size, lcmds);
1738 if (actual_size <= 0) {
1739 NDMP_LOG(LOG_DEBUG, "err %d act_size %d detected",
1740 err, actual_size);
1741 break;
1742 } else if (err) {
1743 NDMP_LOG(LOG_DEBUG, "error %d reading data", err);
1744 return (-1);
1745 }
1746 rec_size = min(actual_size, toread);
1747 (void) memcpy(mem, rec, rec_size);
1748 mem += rec_size;
1749 toread -= rec_size;
1750 }
1751
1752 return (len - toread);
1753 }
1754
1755 /*
1756 * pick up the name and size of a HUGE file
1757 */
1758 static int
get_humongus_file_header(int lib,int drv,long recsize,longlong_t * size,char * name,tlm_cmd_t * local_commands)1759 get_humongus_file_header(int lib,
1760 int drv,
1761 long recsize,
1762 longlong_t *size,
1763 char *name,
1764 tlm_cmd_t *local_commands)
1765 {
1766 char *p_record, *value;
1767 int rv;
1768
1769 NDMP_LOG(LOG_DEBUG, "HUGE Record found: %d", recsize);
1770
1771 rv = 0;
1772 if (recsize == 0) {
1773 /*
1774 * The humongus_file_header was written in a
1775 * RECORDSIZE block and the header.size field of this
1776 * record was 0 before this fix. For backward compatiblity
1777 * read only one RECORDSIZE-size block if the header.size
1778 * field is 0. Otherwise the header.size field should show
1779 * the length of the data of this header.
1780 */
1781 NDMP_LOG(LOG_DEBUG, "Old HUGE record found");
1782 recsize = RECORDSIZE;
1783 }
1784
1785 if (input_mem(lib, drv, local_commands, name, recsize) != recsize) {
1786 rv = -1;
1787 *size = 0;
1788 *name = '\0';
1789 NDMP_LOG(LOG_DEBUG, "Error reading a HUGE file name");
1790 } else {
1791 NDMP_LOG(LOG_DEBUG, "HUGE [%s]", name);
1792
1793 p_record = name;
1794 value = parse(&p_record, " ");
1795 *size = atoll(value);
1796 /*
1797 * Note: Since the backed up names are not longer than
1798 * NAME_MAX and the buffer passed to us is
1799 * TLM_MAX_PATH_NAME, it should be safe to use strlcpy
1800 * without check on the buffer size.
1801 */
1802 (void) strlcpy(name, p_record, TLM_MAX_PATH_NAME);
1803 }
1804
1805 NDMP_LOG(LOG_DEBUG, "HUGE Record %lld [%s]", *size, name);
1806
1807 return (rv);
1808 }
1809
1810 /*
1811 * pick up the long name from the special tape file
1812 */
1813 static int
get_long_name(int lib,int drv,long recsize,char * name,long * buf_spot,tlm_cmd_t * local_commands)1814 get_long_name(int lib,
1815 int drv,
1816 long recsize,
1817 char *name,
1818 long *buf_spot,
1819 tlm_cmd_t *local_commands)
1820 {
1821 int nread;
1822
1823 NDMP_LOG(LOG_DEBUG, "LONGNAME Record found rs %d bs %d", recsize,
1824 *buf_spot);
1825
1826 if (*buf_spot < 0)
1827 *buf_spot = 0;
1828
1829 nread = input_mem(lib, drv, local_commands, name + *buf_spot,
1830 recsize);
1831 if (nread < 0) {
1832 nread = recsize; /* return 0 as size left */
1833 name[*buf_spot] = '\0';
1834 NDMP_LOG(LOG_ERR, "Error %d reading a long file name %s.",
1835 nread, name);
1836 } else {
1837 *buf_spot += nread;
1838 name[*buf_spot] = '\0';
1839 NDMP_LOG(LOG_DEBUG, "LONGNAME [%s]", name);
1840 }
1841
1842 return (recsize - nread);
1843 }
1844
1845 /*
1846 * create a new directory
1847 */
1848 static int
create_directory(char * dir,tlm_job_stats_t * job_stats)1849 create_directory(char *dir, tlm_job_stats_t *job_stats)
1850 {
1851 struct stat64 attr;
1852 char *p;
1853 char temp;
1854 int erc;
1855
1856 /*
1857 * Make sure all directories in this path exist, create them if
1858 * needed.
1859 */
1860 NDMP_LOG(LOG_DEBUG, "new dir[%s]", dir);
1861
1862 erc = 0;
1863 p = &dir[1];
1864 do {
1865 temp = *p;
1866 if (temp == '/' || temp == 0) {
1867 *p = 0;
1868 if (stat64(dir, &attr) < 0) {
1869 if (mkdir(dir, 0777) != 0 && errno != EEXIST) {
1870 erc = errno;
1871 job_stats->js_errors++;
1872 NDMP_LOG(LOG_DEBUG,
1873 "Could not create directory %s: %d",
1874 dir, errno);
1875 break;
1876 }
1877 }
1878 *p = temp;
1879 }
1880 p++;
1881 } while (temp != 0);
1882
1883 return (erc);
1884 }
1885
1886 /*
1887 * create a new hardlink
1888 */
1889 static int
create_hard_link(char * name_old,char * name_new,tlm_acls_t * acls,tlm_job_stats_t * job_stats)1890 create_hard_link(char *name_old, char *name_new,
1891 tlm_acls_t *acls, tlm_job_stats_t *job_stats)
1892 {
1893 int erc;
1894
1895 erc = mkbasedir(name_new);
1896 if (erc != 0)
1897 return (erc);
1898
1899 if (link(name_old, name_new) != 0)
1900 erc = errno;
1901
1902 if (erc) {
1903 /* Nothing to do if the destination already exists */
1904 if (erc == EEXIST)
1905 return (0);
1906 job_stats->js_errors++;
1907 NDMP_LOG(LOG_DEBUG, "error %d (errno %d) hardlink [%s] to [%s]",
1908 erc, errno, name_new, name_old);
1909 return (erc);
1910 }
1911 return (set_acl(name_new, acls));
1912 }
1913
1914 /*
1915 * create a new symlink
1916 */
1917 /*ARGSUSED*/
1918 static int
create_sym_link(char * dst,char * target,tlm_acls_t * acls,tlm_job_stats_t * job_stats)1919 create_sym_link(char *dst, char *target, tlm_acls_t *acls,
1920 tlm_job_stats_t *job_stats)
1921 {
1922 int erc;
1923 struct stat64 *st;
1924
1925 erc = mkbasedir(dst);
1926 if (erc != 0)
1927 return (erc);
1928
1929 st = &acls->acl_attr;
1930 if (symlink(target, dst) != 0) {
1931 erc = errno;
1932 job_stats->js_errors++;
1933 NDMP_LOG(LOG_DEBUG, "error %d softlink [%s] to [%s]",
1934 errno, dst, target);
1935 } else {
1936 st->st_mode |= S_IFLNK;
1937 erc = set_acl(dst, acls);
1938 }
1939
1940 return (erc);
1941 }
1942
1943 /*
1944 * create a new FIFO, char/block device special files
1945 */
1946 static int
create_special(char flag,char * name,tlm_acls_t * acls,int major,int minor,tlm_job_stats_t * job_stats)1947 create_special(char flag, char *name, tlm_acls_t *acls, int major, int minor,
1948 tlm_job_stats_t *job_stats)
1949 {
1950 dev_t dev;
1951 mode_t mode;
1952
1953 switch (flag) {
1954 case LF_CHR:
1955 mode = S_IFCHR;
1956 dev = makedev(major, minor);
1957 break;
1958 case LF_BLK:
1959 mode = S_IFBLK;
1960 dev = makedev(major, minor);
1961 break;
1962 case LF_FIFO:
1963 mode = S_IFIFO;
1964 dev = 0;
1965 break;
1966 default:
1967 NDMP_LOG(LOG_ERR, "unsupported flag %d", flag);
1968 return (-1);
1969 }
1970
1971 /* Remove the old entry first */
1972 if (rmdir(name) < 0) {
1973 if (errno == ENOTDIR)
1974 (void) unlink(name);
1975 }
1976 if (mknod(name, 0777 | mode, dev) != 0) {
1977 job_stats->js_errors++;
1978 NDMP_LOG(LOG_DEBUG, "error %d mknod [%s] major"
1979 " %d minor %d", errno, name, major, minor);
1980 return (errno);
1981 }
1982 return (set_acl(name, acls));
1983 }
1984
1985 /*
1986 * read in the ACLs for the next file
1987 */
1988 static long
load_acl_info(int lib,int drv,long file_size,tlm_acls_t * acls,long * acl_spot,tlm_cmd_t * local_commands)1989 load_acl_info(int lib,
1990 int drv,
1991 long file_size,
1992 tlm_acls_t *acls,
1993 long *acl_spot,
1994 tlm_cmd_t *local_commands)
1995 {
1996 char *bp;
1997 int nread;
1998
1999 /*
2000 * If the ACL is spanned on tapes, then the acl_spot should NOT be
2001 * 0 on next calls to this function to read the rest of the ACL
2002 * on next tapes.
2003 */
2004 if (*acl_spot == 0) {
2005 (void) memset(acls, 0, sizeof (tlm_acls_t));
2006 }
2007
2008 bp = ((char *)&acls->acl_info) + *acl_spot;
2009 nread = input_mem(lib, drv, local_commands, (void *)bp, file_size);
2010 if (nread < 0) {
2011 *acl_spot = 0;
2012 (void) memset(acls, 0, sizeof (tlm_acls_t));
2013 NDMP_LOG(LOG_DEBUG, "Error reading ACL data");
2014 return (0);
2015 }
2016 *acl_spot += nread;
2017 acls->acl_non_trivial = TRUE;
2018
2019 return (file_size - nread);
2020 }
2021
2022 static int
ndmp_set_eprivs_least(void)2023 ndmp_set_eprivs_least(void)
2024 {
2025 priv_set_t *priv_set;
2026
2027 if ((priv_set = priv_allocset()) == NULL) {
2028 NDMP_LOG(LOG_ERR, "Out of memory.");
2029 return (-1);
2030 }
2031
2032 priv_basicset(priv_set);
2033
2034 (void) priv_addset(priv_set, PRIV_PROC_AUDIT);
2035 (void) priv_addset(priv_set, PRIV_PROC_SETID);
2036 (void) priv_addset(priv_set, PRIV_PROC_OWNER);
2037 (void) priv_addset(priv_set, PRIV_FILE_CHOWN);
2038 (void) priv_addset(priv_set, PRIV_FILE_CHOWN_SELF);
2039 (void) priv_addset(priv_set, PRIV_FILE_DAC_READ);
2040 (void) priv_addset(priv_set, PRIV_FILE_DAC_SEARCH);
2041 (void) priv_addset(priv_set, PRIV_FILE_DAC_WRITE);
2042 (void) priv_addset(priv_set, PRIV_FILE_OWNER);
2043 (void) priv_addset(priv_set, PRIV_FILE_SETID);
2044 (void) priv_addset(priv_set, PRIV_SYS_LINKDIR);
2045 (void) priv_addset(priv_set, PRIV_SYS_DEVICES);
2046 (void) priv_addset(priv_set, PRIV_SYS_MOUNT);
2047 (void) priv_addset(priv_set, PRIV_SYS_CONFIG);
2048
2049 if (setppriv(PRIV_SET, PRIV_EFFECTIVE, priv_set) == -1) {
2050 NDMP_LOG(LOG_ERR, "Additional privileges required.");
2051 priv_freeset(priv_set);
2052 return (-1);
2053 }
2054 priv_freeset(priv_set);
2055 return (0);
2056 }
2057
2058 static int
ndmp_set_eprivs_all(void)2059 ndmp_set_eprivs_all(void)
2060 {
2061 priv_set_t *priv_set;
2062
2063 if ((priv_set = priv_allocset()) == NULL) {
2064 NDMP_LOG(LOG_ERR, "Out of memory.");
2065 return (-1);
2066 }
2067
2068 priv_fillset(priv_set);
2069
2070 if (setppriv(PRIV_SET, PRIV_EFFECTIVE, priv_set) != 0) {
2071 NDMP_LOG(LOG_ERR, "Additional privileges required.");
2072 return (-1);
2073 }
2074 priv_freeset(priv_set);
2075 return (0);
2076 }
2077
2078 /*
2079 * Set the standard attributes of the file
2080 */
2081 static int
set_attr(char * name,tlm_acls_t * acls)2082 set_attr(char *name, tlm_acls_t *acls)
2083 {
2084 struct utimbuf tbuf;
2085 boolean_t priv_all = FALSE;
2086 struct stat64 *st;
2087 uid_t uid;
2088 gid_t gid;
2089 struct passwd *pwd;
2090 struct group *grp;
2091 int erc = 0;
2092
2093
2094 if (!name || !acls)
2095 return (0);
2096
2097 st = &acls->acl_attr;
2098 NDMP_LOG(LOG_DEBUG, "set_attr: %s uid %d gid %d uname %s gname %s "
2099 "mode %o", name, st->st_uid, st->st_gid, acls->uname, acls->gname,
2100 st->st_mode);
2101
2102 uid = st->st_uid;
2103 if ((pwd = getpwnam(acls->uname)) != NULL) {
2104 NDMP_LOG(LOG_DEBUG, "set_attr: new uid %d old %d",
2105 pwd->pw_uid, uid);
2106 uid = pwd->pw_uid;
2107 }
2108
2109 gid = st->st_gid;
2110 if ((grp = getgrnam(acls->gname)) != NULL) {
2111 NDMP_LOG(LOG_DEBUG, "set_attr: new gid %d old %d",
2112 grp->gr_gid, gid);
2113 gid = grp->gr_gid;
2114 }
2115
2116 erc = lchown(name, uid, gid);
2117 if (erc != 0) {
2118 erc = errno;
2119 NDMP_LOG(LOG_ERR,
2120 "Could not set uid or/and gid for file %s.", name);
2121 }
2122
2123 if ((st->st_mode & (S_ISUID | S_ISGID)) != 0) {
2124 /*
2125 * Change effective privileges to 'all' which is required to
2126 * change setuid bit for 'root' owned files. If fails, just
2127 * send error to log file and proceed.
2128 */
2129 if (ndmp_set_eprivs_all()) {
2130 NDMP_LOG(LOG_ERR,
2131 "Could not set effective privileges to 'all'.");
2132 } else {
2133 priv_all = TRUE;
2134 }
2135 }
2136
2137 if (!S_ISLNK(st->st_mode)) {
2138 erc = chmod(name, st->st_mode);
2139 if (erc != 0) {
2140 erc = errno;
2141 NDMP_LOG(LOG_ERR, "Could not set correct file"
2142 " permission for file %s: %d", name, errno);
2143 }
2144
2145 tbuf.modtime = st->st_mtime;
2146 tbuf.actime = st->st_atime;
2147 (void) utime(name, &tbuf);
2148 }
2149
2150 if (priv_all == TRUE) {
2151 /*
2152 * Give up the 'all' privileges for effective sets and go back
2153 * to least required privileges. If fails, just send error to
2154 * log file and proceed.
2155 */
2156 if (ndmp_set_eprivs_least())
2157 NDMP_LOG(LOG_ERR,
2158 "Could not set least required privileges.");
2159 }
2160
2161 return (erc);
2162 }
2163
2164 /*
2165 * Set the ACL info for the file
2166 */
2167 static int
set_acl(char * name,tlm_acls_t * acls)2168 set_acl(char *name, tlm_acls_t *acls)
2169 {
2170 int erc;
2171 acl_t *aclp = NULL;
2172
2173 if (name)
2174 NDMP_LOG(LOG_DEBUG, "set_acl: %s", name);
2175 if (acls == NULL)
2176 return (0);
2177
2178 /* Need a place to save real modification time */
2179
2180 erc = set_attr(name, acls);
2181 if (ERROR_IS_FATAL(erc))
2182 return (erc);
2183
2184 if (!acls->acl_non_trivial) {
2185 (void) memset(acls, 0, sizeof (tlm_acls_t));
2186 NDMP_LOG(LOG_DEBUG, "set_acl: skipping trivial");
2187 return (erc);
2188 }
2189
2190 erc = acl_fromtext(acls->acl_info.attr_info, &aclp);
2191 if (erc != 0) {
2192 NDMP_LOG(LOG_DEBUG,
2193 "TAPE RESTORE> acl_fromtext errno %d", erc);
2194 }
2195 if (aclp) {
2196 erc = acl_set(name, aclp);
2197 if (erc < 0) {
2198 erc = errno;
2199 NDMP_LOG(LOG_DEBUG,
2200 "TAPE RESTORE> acl_set errno %d", errno);
2201 }
2202 acl_free(aclp);
2203 }
2204 (void) memset(acls, 0, sizeof (tlm_acls_t));
2205 return (erc);
2206 }
2207
2208 /*
2209 * a wrapper to tlm_get_read_buffer so that
2210 * we can cleanly detect ABORT commands
2211 * without involving the TLM library with
2212 * our problems.
2213 */
2214 static char *
get_read_buffer(int want,int * error,int * actual_size,tlm_cmd_t * local_commands)2215 get_read_buffer(int want,
2216 int *error,
2217 int *actual_size,
2218 tlm_cmd_t *local_commands)
2219 {
2220 while (local_commands->tc_writer == TLM_RESTORE_RUN) {
2221 char *rec;
2222 rec = tlm_get_read_buffer(want, error,
2223 local_commands->tc_buffers, actual_size);
2224 if (rec != 0) {
2225 return (rec);
2226 }
2227 }
2228
2229 /*
2230 * the job is ending, give Writer a buffer that will never be read ...
2231 * it does not matter anyhow, we are aborting.
2232 */
2233 *actual_size = RECORDSIZE;
2234 return (NULL);
2235 }
2236
2237 /*
2238 * Enable wildcard for restore options
2239 */
2240 static boolean_t
wildcard_enabled(void)2241 wildcard_enabled(void)
2242 {
2243 char *cp;
2244
2245 cp = ndmpd_get_prop_default(NDMP_RESTORE_WILDCARD_ENABLE, "n");
2246 return ((toupper(*cp) == 'Y') ? TRUE : FALSE);
2247 }
2248
2249
2250 /*
2251 * Concatenate two names
2252 */
2253 /*ARGSUSED*/
2254 static char *
catnames(struct rs_name_maker * rnp,char * buf,int pos,char * path)2255 catnames(struct rs_name_maker *rnp, char *buf, int pos, char *path)
2256 {
2257 char *rv;
2258
2259 rv = NULL;
2260 if (!buf) {
2261 NDMP_LOG(LOG_DEBUG, "buf is NULL");
2262 } else if (!path) {
2263 NDMP_LOG(LOG_DEBUG, "path is NULL");
2264 } else if (!rnp->rn_nlp) {
2265 NDMP_LOG(LOG_DEBUG, "rn_nlp is NULL [%s]", path);
2266 } else if (!tlm_cat_path(buf, rnp->rn_nlp, path)) {
2267 NDMP_LOG(LOG_DEBUG, "Path too long [%s][%s]",
2268 rnp->rn_nlp, path);
2269 } else
2270 rv = buf;
2271
2272 return (rv);
2273 }
2274
2275
2276 /*
2277 * Create a new name path for restore
2278 */
2279 static char *
rs_new_name(struct rs_name_maker * rnp,char * buf,int pos,char * path)2280 rs_new_name(struct rs_name_maker *rnp, char *buf, int pos, char *path)
2281 {
2282 if (!rnp || !rnp->rn_fp)
2283 return (NULL);
2284
2285 return (*rnp->rn_fp)(rnp, buf, pos, path);
2286 }
2287
2288 /*
2289 * Clear the extra "/" in the tar header if exists
2290 */
2291 static void
rs_create_new_bkpath(char * bk_path,char * path,char * pbuf)2292 rs_create_new_bkpath(char *bk_path, char *path, char *pbuf)
2293 {
2294 char *p, *slashp;
2295
2296 if ((p = strstr(path, bk_path)) == NULL) {
2297 (void) strlcpy(pbuf, path, TLM_MAX_PATH_NAME);
2298 return;
2299 }
2300 if (*(p += strlen(bk_path)) == '/')
2301 p++;
2302
2303 slashp = bk_path + strlen(bk_path) - 1;
2304 if (*slashp == '/')
2305 (void) snprintf(pbuf, TLM_MAX_PATH_NAME, "%s%s", bk_path, p);
2306 else
2307 (void) snprintf(pbuf, TLM_MAX_PATH_NAME, "%s/%s", bk_path, p);
2308
2309 NDMP_LOG(LOG_DEBUG, "old path [%s] new path [%s]", path, pbuf);
2310 }
2311
2312
2313 /*
2314 * Iterate over ZFS metadata stored in the backup stream and use the callback
2315 * to restore it.
2316 */
2317 int
ndmp_iter_zfs(ndmp_context_t * nctx,int (* np_restore_property)(nvlist_t *,void *),void * ptr)2318 ndmp_iter_zfs(ndmp_context_t *nctx, int (*np_restore_property)(nvlist_t *,
2319 void *), void *ptr)
2320 {
2321 tlm_commands_t *cmds;
2322 ndmp_metadata_header_t *mhp;
2323 ndmp_metadata_header_ext_t *mhpx;
2324 ndmp_metadata_property_t *mpp;
2325 ndmp_metadata_property_ext_t *mppx;
2326 tlm_cmd_t *lcmd;
2327 int actual_size;
2328 nvlist_t *nvl;
2329 nvlist_t *valp;
2330 nvpair_t *nvp = NULL;
2331 char plname[100];
2332 char *mhbuf, *pp, *tp;
2333 int rv, i;
2334 int size, lsize, sz;
2335 int align = RECORDSIZE - 1;
2336
2337 if (nctx == NULL || (cmds = (tlm_commands_t *)nctx->nc_cmds) == NULL)
2338 return (-1);
2339
2340 nctx->nc_plname = plname;
2341 if ((lcmd = cmds->tcs_command) == NULL ||
2342 lcmd->tc_buffers == NULL)
2343 return (-1);
2344
2345 /* Default minimum bytes needed */
2346 size = sizeof (ndmp_metadata_header_t) +
2347 ZFS_MAX_PROPS * sizeof (ndmp_metadata_property_t);
2348 size += align;
2349 size &= ~align;
2350
2351 if ((mhbuf = malloc(size)) == NULL)
2352 return (-1);
2353
2354 /* LINTED improper alignment */
2355 while ((mhp = (ndmp_metadata_header_t *)get_read_buffer(size, &rv,
2356 &actual_size, lcmd)) != NULL) {
2357 pp = mhbuf;
2358
2359 if (strncmp(mhp->nh_magic, ZFS_META_MAGIC,
2360 sizeof (mhp->nh_magic)) != 0 &&
2361 strncmp(mhp->nh_magic, ZFS_META_MAGIC_EXT,
2362 sizeof (mhp->nh_magic)) != 0) {
2363 /* No more metadata */
2364 tlm_unget_read_buffer(lcmd->tc_buffers, actual_size);
2365 free(mhbuf);
2366 return (0);
2367 }
2368
2369 if (strncmp(mhp->nh_magic, ZFS_META_MAGIC_EXT,
2370 sizeof (mhp->nh_magic)) == 0) {
2371 mhpx = (ndmp_metadata_header_ext_t *)mhp;
2372 if (mhpx->nh_total_bytes > size) {
2373 if ((pp = realloc(mhbuf, mhpx->nh_total_bytes))
2374 == NULL) {
2375 free(mhbuf);
2376 return (-1);
2377 }
2378 mhbuf = pp;
2379 }
2380 size = mhpx->nh_total_bytes;
2381 }
2382
2383 (void) memcpy(pp, (char *)mhp, (actual_size < size) ?
2384 actual_size : size);
2385 pp += (actual_size < size) ? actual_size : size;
2386
2387 sz = actual_size;
2388 while (sz < size &&
2389 ((tp = get_read_buffer(size - sz, &rv, &lsize,
2390 lcmd))) != NULL) {
2391 (void) memcpy(pp, tp, lsize);
2392 sz += lsize;
2393 pp += lsize;
2394 }
2395 if (sz > size) {
2396 tlm_unget_read_buffer(lcmd->tc_buffers, sz - size);
2397 }
2398
2399 /* LINTED improper alignment */
2400 mhp = (ndmp_metadata_header_t *)mhbuf;
2401
2402 nvl = NULL;
2403 if (strncmp(mhp->nh_magic, ZFS_META_MAGIC_EXT,
2404 sizeof (mhp->nh_magic)) == 0) {
2405 /* New metadata format */
2406 /* LINTED improper alignment */
2407 mhpx = (ndmp_metadata_header_ext_t *)mhbuf;
2408
2409 if (mhpx->nh_major > META_HDR_MAJOR_VERSION) {
2410 /* Major header mismatch */
2411 NDMP_LOG(LOG_ERR, "metadata header mismatch",
2412 "M%d != M%d", mhpx->nh_major,
2413 META_HDR_MAJOR_VERSION);
2414 free(mhbuf);
2415 return (-1);
2416 }
2417 if (mhpx->nh_major == META_HDR_MAJOR_VERSION &&
2418 mhpx->nh_minor > META_HDR_MINOR_VERSION) {
2419 /* Minor header mismatch */
2420 NDMP_LOG(LOG_ERR, "Warning:"
2421 "metadata header mismatch m%d != m%d",
2422 mhpx->nh_minor,
2423 META_HDR_MINOR_VERSION);
2424 continue;
2425 }
2426
2427 nctx->nc_plversion = mhpx->nh_plversion;
2428 (void) strlcpy(plname, mhpx->nh_plname,
2429 sizeof (plname));
2430
2431 if (nvlist_alloc(&nvl, NV_UNIQUE_NAME, 0) != 0)
2432 goto nvlist_err;
2433
2434 mppx = &mhpx->nh_property[0];
2435 for (i = 0; i < mhpx->nh_count && mppx; i++, mppx++) {
2436 if (!*mppx->mp_name)
2437 continue;
2438 valp = NULL;
2439 if (nvlist_alloc(&valp,
2440 NV_UNIQUE_NAME, 0) != 0 ||
2441 nvlist_add_string(valp, "value",
2442 mppx->mp_value) != 0 ||
2443 nvlist_add_string(valp, "source",
2444 mppx->mp_source) != 0 ||
2445 nvlist_add_nvlist(nvl, mppx->mp_name,
2446 valp) != 0) {
2447 nvlist_free(valp);
2448 goto nvlist_err;
2449 }
2450 nvlist_free(valp);
2451 }
2452 } else {
2453 nctx->nc_plversion = mhp->nh_plversion;
2454 (void) strlcpy(plname, mhp->nh_plname,
2455 sizeof (plname));
2456
2457 if (nvlist_alloc(&nvl, NV_UNIQUE_NAME, 0) != 0)
2458 goto nvlist_err;
2459
2460 mpp = &mhp->nh_property[0];
2461 for (i = 0; i < mhp->nh_count && mpp; i++, mpp++) {
2462 if (!*mpp->mp_name)
2463 continue;
2464 valp = NULL;
2465 if (nvlist_alloc(&valp,
2466 NV_UNIQUE_NAME, 0) != 0 ||
2467 nvlist_add_string(valp, "value",
2468 mpp->mp_value) != 0 ||
2469 nvlist_add_string(valp, "source",
2470 mpp->mp_source) != 0 ||
2471 nvlist_add_nvlist(nvl, mpp->mp_name,
2472 valp) != 0) {
2473 nvlist_free(valp);
2474 goto nvlist_err;
2475 }
2476 nvlist_free(valp);
2477 }
2478 }
2479
2480 if (np_restore_property(nvl, ptr) != 0)
2481 goto nvlist_err;
2482
2483 while ((nvp = nvlist_next_nvpair(nvl, nvp)) != NULL &&
2484 nvpair_value_nvlist(nvp, &valp) == 0) {
2485 nvlist_free(valp);
2486 }
2487 nvlist_free(nvl);
2488 }
2489
2490 free(mhbuf);
2491 return (0);
2492
2493 nvlist_err:
2494 free(mhbuf);
2495
2496 while ((nvp = nvlist_next_nvpair(nvl, nvp)) != NULL &&
2497 nvpair_value_nvlist(nvp, &valp) == 0) {
2498 nvlist_free(valp);
2499 }
2500 nvlist_free(nvl);
2501 return (-1);
2502 }
2503
2504 /*
2505 * Returns the version number of the plugin which created the metadata
2506 */
2507 uint_t
ndmp_context_get_version(ndmp_context_t * nctx)2508 ndmp_context_get_version(ndmp_context_t *nctx)
2509 {
2510 tlm_commands_t *cmds;
2511 ndmp_metadata_header_t *mhp;
2512 tlm_cmd_t *lcmd;
2513 int actual_size;
2514 int rv;
2515 int size;
2516 int align = RECORDSIZE - 1;
2517
2518 if (nctx == NULL || (cmds = (tlm_commands_t *)nctx->nc_cmds) == NULL)
2519 return (0);
2520
2521 if ((lcmd = cmds->tcs_command) == NULL ||
2522 lcmd->tc_buffers == NULL)
2523 return (0);
2524
2525 size = sizeof (ndmp_metadata_header_t);
2526 size += align;
2527 size &= ~align;
2528
2529 /* LINTED improper alignment */
2530 if ((mhp = (ndmp_metadata_header_t *)get_read_buffer(size, &rv,
2531 &actual_size, lcmd)) != NULL) {
2532 if (strncmp(mhp->nh_magic, ZFS_META_MAGIC,
2533 sizeof (mhp->nh_magic)) != 0) {
2534 /* No more metadata */
2535 tlm_unget_read_buffer(lcmd->tc_buffers, actual_size);
2536 return (0);
2537 }
2538
2539 nctx->nc_plversion = mhp->nh_plversion;
2540 tlm_unget_read_buffer(lcmd->tc_buffers, actual_size);
2541 }
2542
2543 return (nctx->nc_plversion);
2544 }
2545