toaruos/util/createramdisk.py

68 lines
2.1 KiB
Python

#!/usr/bin/python3
"""
Generates, from this source repository, a "tarramdisk" - a ustar archive
suitable for booting ToaruOS.
"""
import os
import tarfile
users = {
'root': 0,
'local': 1000,
'guest': 1001,
}
restricted_files = {
'etc/master.passwd': 0o600,
'etc/sudoers': 0o600,
'tmp': 0o777,
'var': 0o755,
'bin/sudo': 0o4555,
'bin/gsudo': 0o4555,
}
def file_filter(tarinfo):
# Root owns files by default.
tarinfo.uid = 0
tarinfo.gid = 0
if tarinfo.name.startswith('home/'):
# Home directory contents are owned by their users.
user = tarinfo.name.split('/')[1]
tarinfo.uid = users.get(user,0)
tarinfo.gid = tarinfo.uid
elif tarinfo.name in restricted_files:
tarinfo.mode = restricted_files[tarinfo.name]
if tarinfo.name.startswith('usr/include/kuroko') and tarinfo.type == tarfile.SYMTYPE:
return None
if tarinfo.name.startswith('src'):
# Let local own the files here
tarinfo.uid = users.get('local')
tarinfo.gid = tarinfo.uid
# Skip object files
if tarinfo.name.endswith('.so') or tarinfo.name.endswith('.o'):
return None
return tarinfo
with tarfile.open('ramdisk.igz','w:gz') as ramdisk:
ramdisk.add('base',arcname='/',filter=file_filter)
ramdisk.add('.',arcname='/src',filter=file_filter,recursive=False) # Add a src directory
ramdisk.add('apps',arcname='/src/apps',filter=file_filter)
ramdisk.add('kernel',arcname='/src/kernel',filter=file_filter)
ramdisk.add('linker',arcname='/src/linker',filter=file_filter)
ramdisk.add('lib',arcname='/src/lib',filter=file_filter)
ramdisk.add('libc',arcname='/src/libc',filter=file_filter)
#ramdisk.add('boot',arcname='/src/boot',filter=file_filter)
#ramdisk.add('modules',arcname='/src/modules',filter=file_filter)
if os.path.exists('tags'):
ramdisk.add('tags',arcname='/src/tags',filter=file_filter)
ramdisk.add('util/auto-dep.krk',arcname='/bin/auto-dep.krk',filter=file_filter)
ramdisk.add('kuroko/src/kuroko',arcname='/usr/include/kuroko',filter=file_filter)