You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
202 lines
11 KiB
Python
202 lines
11 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
# Name: distro.py
|
|
# Purpose: Module to detect if distro types supported by multibootusb (by extracting specific files)
|
|
# Authors: Sundar
|
|
# Licence: This file is a part of multibootusb package. You can redistribute it or modify
|
|
# under the terms of GNU General Public License, v.2 or above
|
|
|
|
import os
|
|
import platform
|
|
import re
|
|
from .iso import *
|
|
from .isodump3 import ISO9660
|
|
from .gen import *
|
|
from . import _7zip
|
|
|
|
|
|
def distro(iso_cfg_ext_dir, iso_link, expose_exception=False):
|
|
"""
|
|
Detect if distro is supported by multibootusb.
|
|
:param iso_cfg_ext_dir: Directory where *.cfg files are extracted.
|
|
:return: Detected distro name as string.
|
|
"""
|
|
# iso9660fs = ISO9660(iso_link)
|
|
# iso_file_list = iso9660fs.readDir("/")
|
|
iso_file_list = _7zip.list_iso(iso_link, expose_exception=expose_exception)
|
|
if platform.system() == "Linux" or platform.system() == "Windows":
|
|
for path, subdirs, files in os.walk(iso_cfg_ext_dir):
|
|
for name in files:
|
|
if name.endswith(('.cfg', '.CFG', '.txt', '.TXT', '.lst')):
|
|
try:
|
|
# errors='ignore' is required as some files also contain non utf character
|
|
string = open(os.path.join(path, name), errors='ignore').read()
|
|
except IOError:
|
|
return "Read Error."
|
|
else:
|
|
if any("f4ubcd" in s.lower() for s in iso_file_list):
|
|
return "f4ubcd"
|
|
if re.search(r'ubcd', string, re.I):
|
|
return "ubcd"
|
|
elif re.search(r'Super Grub Disk', string, re.I):
|
|
return "sgrubd2"
|
|
elif re.search(r'hbcd', string, re.I):
|
|
return "hbcd"
|
|
elif re.search(r'systemrescuecd', string, re.I):
|
|
return "systemrescuecd"
|
|
elif re.search(r'pmagic|partedmagic', string, re.I) and isolinux_bin_exist(iso_link):
|
|
return "parted-magic"
|
|
elif re.search(r'mgalive', string, re.I): # mounting fat filesystem hard coded in to initrd.
|
|
# Can be modified only under linux.
|
|
return "mageialive"
|
|
elif re.search(r'archisolabel|misolabel|parabolaisolabel', string, re.I):
|
|
return "arch"
|
|
elif re.search(r'chakraisolabel', string, re.I):
|
|
return "chakra"
|
|
elif re.search(r'kdeosisolabel', string, re.I):
|
|
return "kaos"
|
|
elif re.search(r'boot=live', string, re.I) and isolinux_bin_exist(iso_link):
|
|
return "debian"
|
|
elif re.search(r'grml', string, re.I) and re.search(r'live-media-path=', string, re.I):
|
|
return "grml"
|
|
elif re.search(r'debian-installer', string, re.I) and not any("casper" in s.lower() for s in iso_file_list):
|
|
return "debian-install"
|
|
elif re.search(r'solydx', string, re.I):
|
|
return "solydx"
|
|
elif re.search(r'knoppix', string, re.I):
|
|
return "knoppix"
|
|
elif re.search(r'root=live:CDLABEL=', string, re.I) or re.search(r'root=live:LABEL=', string, re.I):
|
|
return "fedora"
|
|
elif re.search(r'redcore', string, re.I):
|
|
return "fedora"
|
|
elif re.search(r'redhat', string, re.I):
|
|
return "redhat"
|
|
elif re.search(
|
|
r'slitaz|dban |ophcrack|tinycore|rescue.cpi|xpud|untangle|4mlinux|partition wizard|android-x86.png|'
|
|
r'riplinux|lebel dummy|http://pogostick.net/~pnh/ntpasswd/|AVG Rescue CD|AntivirusLiveCD|'
|
|
r'lkrn|Nanolinux|OSForensics|acentos', string, re.I): # acentos is a text found in slitaz distro
|
|
return "slitaz"
|
|
elif re.search(r'minimal Slackware|Slackware-HOWTO', string, re.I):
|
|
# for minimal slackware detection
|
|
return "slitaz"
|
|
# elif re.search(r'suse', string, re.I):
|
|
# return "suse"
|
|
elif re.search(r'class opensuse', string, re.I):
|
|
return "opensuse-install"
|
|
elif re.search(r'boot=casper', string, re.I):
|
|
return "ubuntu"
|
|
elif re.search(r'wifislax', string, re.I):
|
|
return "wifislax"
|
|
elif re.search(r'slax', string, re.I):
|
|
return "slax"
|
|
elif re.search(r'sms.jpg|vector |autoexec', string, re.I) and isolinux_bin_exist(iso_link):
|
|
return "sms"
|
|
elif re.search(r'antix', string, re.I):
|
|
return "antix"
|
|
elif re.search(r'porteus', string, re.I):
|
|
return "porteus"
|
|
elif re.search(r'livecd=livecd|PCLinuxOS', string, re.I):
|
|
return "pclinuxos"
|
|
elif re.search(r'looptype=squashfs|http://dee.su/liberte', string, re.I):
|
|
return "gentoo"
|
|
elif re.search(r'finnix', string, re.I):
|
|
return "finnix"
|
|
elif re.search(r'wifiway', string, re.I):
|
|
return "wifiway"
|
|
elif re.search(r'puppy|quirky|fatdog|slacko|xenialpup', string, re.I):
|
|
return "puppy"
|
|
elif re.search(r'ipcop', string, re.I):
|
|
return "ipcop"
|
|
elif re.search(r'ipfire', string, re.I):
|
|
return "ipfire"
|
|
elif re.search(r'zenwalk|slack|salix', string, re.I) and re.search(r'live', string, re.I):
|
|
return "salix-live"
|
|
elif re.search(r'zenwalk|slack|salix', string, re.I) and not re.search(r'slacko', string, re.I):
|
|
return "zenwalk"
|
|
elif re.search(r'ubuntu server', string, re.I):
|
|
return "ubuntu-server"
|
|
elif re.search(r'root=live:CDLABEL=CentOS',
|
|
string, re.I):
|
|
return 'centos' # centos-live
|
|
elif re.search(r'Install CentOS', string, re.I):
|
|
return "centos-install"
|
|
elif re.search(r'CentOS', string, re.I):
|
|
return "centos"
|
|
elif re.search(r'Trinity Rescue Kit', string, re.I):
|
|
return "trinity-rescue"
|
|
elif re.search(r'alpine', string, re.I):
|
|
return "alpine"
|
|
elif re.search(r'http://support.kaspersky.com', string, re.I):
|
|
return "kaspersky"
|
|
elif re.search(r'ALT Linux', string, re.I):
|
|
return "alt-linux"
|
|
elif re.search(r'Sergei Strelec', string, re.I):
|
|
return "Windows"
|
|
elif re.search(r'ReactOS', string, re.I):
|
|
return "ReactOS"
|
|
elif re.search(r'fsecure', string, re.I):
|
|
return "fsecure"
|
|
elif re.search(r'default rwp', string, re.I):
|
|
return "pc-unlocker"
|
|
elif re.search(r'/system/stage1', string, re.I):
|
|
return 'pc-tool'
|
|
elif re.search(r'vba32rescue', string, re.I):
|
|
return 'grub2only'
|
|
elif re.search(r'BOOT_IMAGE=rising', string, re.I):
|
|
return 'rising-av'
|
|
elif re.search(r'Avira Rescue System', string, re.I):
|
|
return 'Avira-RS'
|
|
elif any("alpine-release" in s.lower() for s in iso_file_list):
|
|
return 'alpine'
|
|
elif re.search(r'BOOT_IMAGE=insert', string, re.I):
|
|
return 'insert'
|
|
|
|
distro = detect_iso_from_file_list(iso_link, iso_file_list)
|
|
|
|
if distro:
|
|
return distro
|
|
# FIXME: See the below comments.
|
|
# else:
|
|
# # FIXME: The idea of detecting as generic is to work like a unetbootin if other methods fails.
|
|
# # This simply extracts distro to root of the USB and install syslinux on isolinux.bin directory.
|
|
# # All works fine but unable to boot the distro successfully. Also, see the generic section from
|
|
# # syslinux, update_cfg and install_distro modules.
|
|
# if self.isolinux_bin_exist():
|
|
# return "generic"
|
|
elif str(iso_link).lower().endswith('.iso'):
|
|
return 'memdisk_iso'
|
|
elif str(iso_link).lower().endswith('.img'):
|
|
return 'memdisk_img'
|
|
else:
|
|
return None
|
|
|
|
|
|
def detect_iso_from_file_list(iso_link, iso_file_list):
|
|
"""
|
|
Fallback detection script from the content of an ISO.
|
|
:return: supported distro as string
|
|
"""
|
|
if os.path.exists(iso_link):
|
|
if any("sources" in s.lower() for s in iso_file_list) and any("boot.wim" in s.lower() for s in iso_file_list):
|
|
return "Windows"
|
|
elif any("config.isoclient" in s.lower() for s in iso_file_list):
|
|
return "opensuse"
|
|
elif any("dban" in s.lower() for s in iso_file_list):
|
|
return "slitaz"
|
|
elif any("memtest.img" in s.lower() for s in iso_file_list):
|
|
return "memtest"
|
|
elif any("mt86.png" in s.lower() for s in iso_file_list) and any("isolinux" in s.lower() for s in iso_file_list):
|
|
return 'raw_iso'
|
|
elif any("menu.lst" in s.lower() for s in iso_file_list):
|
|
return "grub4dos"
|
|
elif any("bootwiz.cfg" in s.lower() for s in iso_file_list) and any("bootmenu_logo.png" in s.lower() for s in iso_file_list):
|
|
return "grub4dos_iso"
|
|
else:
|
|
log(iso_file_list)
|
|
|
|
if __name__ == '__main__':
|
|
iso_cfg_ext_dir = os.path.join(multibootusb_host_dir(), "iso_cfg_ext_dir")
|
|
iso_link = 'Downloads/clonezilla-live-2.4.2-32-amd64.iso'
|
|
iso_extract_file(iso_link, iso_cfg_ext_dir, 'cfg')
|
|
log(distro(iso_cfg_ext_dir))
|