76 lines
2.3 KiB
Python
76 lines
2.3 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') or tarinfo.name.endswith('.sys'):
|
|
return None
|
|
|
|
return tarinfo
|
|
|
|
def symlink(file,target):
|
|
ti = tarfile.TarInfo(file)
|
|
ti.type = tarfile.SYMTYPE
|
|
ti.linkname = target
|
|
return ti
|
|
|
|
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)
|
|
ramdisk.addfile(symlink('bin/sh','esh'))
|
|
ramdisk.addfile(symlink('bin/mandelbrot','julia'))
|
|
|
|
|