diff options
Diffstat (limited to 'hosts/vidhar/borg/copy.py')
-rwxr-xr-x | hosts/vidhar/borg/copy.py | 32 |
1 files changed, 7 insertions, 25 deletions
diff --git a/hosts/vidhar/borg/copy.py b/hosts/vidhar/borg/copy.py index 96426682..6adaa817 100755 --- a/hosts/vidhar/borg/copy.py +++ b/hosts/vidhar/borg/copy.py | |||
@@ -21,7 +21,6 @@ from xdg import xdg_runtime_dir | |||
21 | import pathlib | 21 | import pathlib |
22 | 22 | ||
23 | import unshare | 23 | import unshare |
24 | import pyprctl | ||
25 | 24 | ||
26 | import signal | 25 | import signal |
27 | from time import sleep | 26 | from time import sleep |
@@ -94,38 +93,21 @@ def copy_archive(src_repo_path, dst_repo_path, entry): | |||
94 | child = os.fork() | 93 | child = os.fork() |
95 | if child == 0: | 94 | if child == 0: |
96 | # print('unshare/chroot', file=stderr) | 95 | # print('unshare/chroot', file=stderr) |
97 | uid, gid = os.geteuid(), os.getegid() | 96 | unshare.unshare(unshare.CLONE_NEWNS) |
98 | unshare.unshare(unshare.CLONE_NEWNS | unshare.CLONE_NEWUSER) | ||
99 | ps_effective = set() # {pyprctl.Cap.SETUID, pyprctl.Cap.SETGID} | ||
100 | ps_ambient = {pyprctl.Cap.SYS_ADMIN} | ||
101 | pyprctl.cap_permitted.add(*(ps_effective | ps_ambient)) | ||
102 | pyprctl.cap_effective.add(*(ps_effective | ps_ambient)) | ||
103 | pyprctl.cap_inheritable.add(*ps_ambient) | ||
104 | pyprctl.cap_ambient.add(*ps_ambient) | ||
105 | with open('/proc/self/setgroups', 'w') as setgroups: | ||
106 | setgroups.write('deny') | ||
107 | with open('/proc/self/uid_map', 'w') as uid_map: | ||
108 | uid_map.write(f'0 {uid} 1') | ||
109 | with open('/proc/self/gid_map', 'w') as gid_map: | ||
110 | gid_map.write(f'0 {gid} 1') | ||
111 | subprocess.run(['mount', '--make-rprivate', '/'], check=True) | 97 | subprocess.run(['mount', '--make-rprivate', '/'], check=True) |
112 | chroot = pathlib.Path(tmpdir) / 'chroot' | 98 | chroot = pathlib.Path(tmpdir) / 'chroot' |
113 | chroot.mkdir() | 99 | upper = pathlib.Path(tmpdir) / 'upper' |
114 | # upper = pathlib.Path(tmpdir) / 'upper' | 100 | work = pathlib.Path(tmpdir) / 'work' |
115 | # work = pathlib.Path(tmpdir) / 'work' | 101 | for path in [chroot,upper,work]: |
116 | # for path in [chroot,upper,work]: | 102 | path.mkdir() |
117 | # path.mkdir() | 103 | subprocess.run(['mount', '-t', 'overlay', 'overlay', '-o', f'lowerdir=/,upperdir={upper},workdir={work}', chroot], check=True) |
118 | # print(f'euid={os.getuid()}', file=stderr) | 104 | bindMounts = ['nix', 'run', 'proc', 'dev', 'sys', pathlib.Path(os.path.expanduser('~')).relative_to('/')] |
119 | # subprocess.run(['stat', '/', upper, work, chroot], check=True) | ||
120 | # subprocess.run(['mount', '-t', 'overlay', 'overlay', '-o', f'lowerdir=/,upperdir={upper},workdir={work}', chroot], check=True) | ||
121 | bindMounts = ['etc', 'nix', 'run', 'proc', 'dev', 'sys', pathlib.Path(os.path.expanduser('~')).relative_to('/')] | ||
122 | if not ":" in src_repo_path: | 105 | if not ":" in src_repo_path: |
123 | bindMounts.append(pathlib.Path(src_repo_path).relative_to('/')) | 106 | bindMounts.append(pathlib.Path(src_repo_path).relative_to('/')) |
124 | if 'SSH_AUTH_SOCK' in os.environ: | 107 | if 'SSH_AUTH_SOCK' in os.environ: |
125 | bindMounts.append(pathlib.Path(os.environ['SSH_AUTH_SOCK']).parent.relative_to('/')) | 108 | bindMounts.append(pathlib.Path(os.environ['SSH_AUTH_SOCK']).parent.relative_to('/')) |
126 | for bindMount in bindMounts: | 109 | for bindMount in bindMounts: |
127 | (chroot / bindMount).mkdir(parents=True,exist_ok=True) | 110 | (chroot / bindMount).mkdir(parents=True,exist_ok=True) |
128 | print(*['mount', '--bind', pathlib.Path('/') / bindMount, chroot / bindMount], file=stderr) | ||
129 | subprocess.run(['mount', '--bind', pathlib.Path('/') / bindMount, chroot / bindMount], check=True) | 111 | subprocess.run(['mount', '--bind', pathlib.Path('/') / bindMount, chroot / bindMount], check=True) |
130 | os.chroot(chroot) | 112 | os.chroot(chroot) |
131 | os.chdir('/') | 113 | os.chdir('/') |