^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 1) // SPDX-License-Identifier: GPL-2.0
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 2) #include "util/copyfile.h"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 3) #include "util/namespaces.h"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 4) #include <internal/lib.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 5) #include <sys/mman.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 6) #include <sys/stat.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 7) #include <errno.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 8) #include <fcntl.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 9) #include <stdio.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 10) #include <stdlib.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 11) #include <string.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 12) #include <unistd.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 13)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 14) static int slow_copyfile(const char *from, const char *to, struct nsinfo *nsi)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 15) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 16) int err = -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 17) char *line = NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 18) size_t n;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 19) FILE *from_fp, *to_fp;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 20) struct nscookie nsc;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 21)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 22) nsinfo__mountns_enter(nsi, &nsc);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 23) from_fp = fopen(from, "r");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 24) nsinfo__mountns_exit(&nsc);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 25) if (from_fp == NULL)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 26) goto out;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 27)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 28) to_fp = fopen(to, "w");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 29) if (to_fp == NULL)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 30) goto out_fclose_from;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 31)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 32) while (getline(&line, &n, from_fp) > 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 33) if (fputs(line, to_fp) == EOF)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 34) goto out_fclose_to;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 35) err = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 36) out_fclose_to:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 37) fclose(to_fp);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 38) free(line);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 39) out_fclose_from:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 40) fclose(from_fp);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 41) out:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 42) return err;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 43) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 44)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 45) int copyfile_offset(int ifd, loff_t off_in, int ofd, loff_t off_out, u64 size)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 46) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 47) void *ptr;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 48) loff_t pgoff;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 49)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 50) pgoff = off_in & ~(page_size - 1);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 51) off_in -= pgoff;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 52)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 53) ptr = mmap(NULL, off_in + size, PROT_READ, MAP_PRIVATE, ifd, pgoff);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 54) if (ptr == MAP_FAILED)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 55) return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 56)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 57) while (size) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 58) ssize_t ret = pwrite(ofd, ptr + off_in, size, off_out);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 59) if (ret < 0 && errno == EINTR)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 60) continue;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 61) if (ret <= 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 62) break;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 63)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 64) size -= ret;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 65) off_in += ret;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 66) off_out += ret;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 67) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 68) munmap(ptr, off_in + size);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 69)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 70) return size ? -1 : 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 71) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 72)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 73) static int copyfile_mode_ns(const char *from, const char *to, mode_t mode,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 74) struct nsinfo *nsi)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 75) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 76) int fromfd, tofd;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 77) struct stat st;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 78) int err;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 79) char *tmp = NULL, *ptr = NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 80) struct nscookie nsc;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 81)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 82) nsinfo__mountns_enter(nsi, &nsc);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 83) err = stat(from, &st);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 84) nsinfo__mountns_exit(&nsc);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 85) if (err)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 86) goto out;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 87) err = -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 88)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 89) /* extra 'x' at the end is to reserve space for '.' */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 90) if (asprintf(&tmp, "%s.XXXXXXx", to) < 0) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 91) tmp = NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 92) goto out;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 93) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 94) ptr = strrchr(tmp, '/');
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 95) if (!ptr)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 96) goto out;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 97) ptr = memmove(ptr + 1, ptr, strlen(ptr) - 1);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 98) *ptr = '.';
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 99)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 100) tofd = mkstemp(tmp);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 101) if (tofd < 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 102) goto out;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 103)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 104) if (st.st_size == 0) { /* /proc? do it slowly... */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 105) err = slow_copyfile(from, tmp, nsi);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 106) if (!err && fchmod(tofd, mode))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 107) err = -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 108) goto out_close_to;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 109) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 110)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 111) if (fchmod(tofd, mode))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 112) goto out_close_to;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 113)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 114) nsinfo__mountns_enter(nsi, &nsc);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 115) fromfd = open(from, O_RDONLY);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 116) nsinfo__mountns_exit(&nsc);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 117) if (fromfd < 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 118) goto out_close_to;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 119)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 120) err = copyfile_offset(fromfd, 0, tofd, 0, st.st_size);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 121)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 122) close(fromfd);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 123) out_close_to:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 124) close(tofd);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 125) if (!err)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 126) err = link(tmp, to);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 127) unlink(tmp);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 128) out:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 129) free(tmp);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 130) return err;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 131) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 132)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 133) int copyfile_ns(const char *from, const char *to, struct nsinfo *nsi)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 134) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 135) return copyfile_mode_ns(from, to, 0755, nsi);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 136) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 137)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 138) int copyfile_mode(const char *from, const char *to, mode_t mode)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 139) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 140) return copyfile_mode_ns(from, to, mode, NULL);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 141) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 142)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 143) int copyfile(const char *from, const char *to)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 144) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 145) return copyfile_mode(from, to, 0755);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 146) }
Orange Pi5 kernel
Deprecated Linux kernel 5.10.110 for OrangePi 5/5B/5+ boards
3 Commits
0 Branches
0 Tags