| index : archinstall32 | |
| Archlinux32 installer | gitolite user |
| summaryrefslogtreecommitdiff |
| -rw-r--r-- | archinstall.py | 56 |
diff --git a/archinstall.py b/archinstall.py index a9721e17..1d3672a0 100644 --- a/archinstall.py +++ b/archinstall.py @@ -1,7 +1,9 @@ #!/usr/bin/python3 -import psutil, os, re, struct, sys +import psutil, os, re, struct, sys, json +import urllib.request, urllib.parse from glob import glob -from socket import inet_ntoa, AF_INET, AF_INET6 +#from select import epoll, EPOLLIN, EPOLLHUP +from socket import socket, inet_ntoa, AF_INET, AF_INET6, AF_PACKET from collections import OrderedDict as oDict from subprocess import Popen, STDOUT, PIPE @@ -30,11 +32,14 @@ def get_default_gateway_linux(): return inet_ntoa(struct.pack("<L", int(fields[2], 16))) - #for nic, opts in psutil.net_if_addrs().items(): - # for addr in opts: - # if addr.family in (AF_INET, AF_INET6) and addr.address: - # if addr.address in ('127.0.0.1', '::1'): continue - # print(addr) +def get_local_MACs(): + macs = {} + for nic, opts in psutil.net_if_addrs().items(): + for addr in opts: + #if addr.family in (AF_INET, AF_INET6) and addr.address: + if addr.family == AF_PACKET: # MAC + macs[addr.address] = nic + return macs def run(cmd): #print('[!] {}'.format(cmd)) @@ -108,6 +113,25 @@ def update_drive_list(): if device_state(name): harddrives['/dev/{}'.format(name)] = psutil.disk_usage('/dev/{}'.format(name)) +def multisplit(s, splitters): + s = [s,] + for key in splitters: + ns = [] + for obj in s: + x = obj.split(key) + for index, part in enumerate(x): + if len(part): + ns.append(part) + if index < len(x)-1: + ns.append(key) + s = ns + return s + +def grab_url_data(path): + safe_path = path[:path.find(':')+1]+''.join([item if item in ('/', '?', '=', '&') else urllib.parse.quote(item) for item in multisplit(path[path.find(':')+1:], ('/', '?', '=', '&'))]) + response = urllib.request.urlopen(safe_path) + return response.read() + if __name__ == '__main__': update_git() # Breaks and restarts the script if an update was found. update_drive_list() @@ -188,6 +212,7 @@ if __name__ == '__main__': o = run('arch-chroot /mnt chmod 700 /root') #o = run('arch-chroot /mnt usermod --password {} root'.format(PIN)) #TODO: This doesn't work either: (why the hell not?) + # echo "newpass" | passwd --stdin root ? o = run("arch-chroot /mnt echo 'root:{pin}' | chpasswd".format(**args, pin=PIN)) if 'user' in args: o = run('arch-chroot /mnt useradd -m -G wheel {user}'.format(**args)) @@ -216,6 +241,23 @@ if __name__ == '__main__': entry.write('initrd /initramfs-linux.img\n') entry.write('options cryptdevice=UUID={UUID}:luksdev root=/dev/mapper/luksdev rw intel_pstate=no_hwp\n'.format(UUID=UUID)) + locmac = get_local_MACs() + for mac in locmac: + try: + instructions = grab_url_data('https://raw.githubusercontent.com/Torxed/archinstall/net-deploy/deployments/{}.json'.format(mac)) + except urllib.error.HTTPError: + print('[N] No instructions for this box on this mac: {}'.format(mac)) + continue + + instructions = json.loads(instructions.decode('UTF-8')) + + for title in instructions: + print('[N] {}'.format(title)) + for command in instructions[title]: + o = run(command) # arch-chroot /mnt ... + if instructions[title][command]: + print(o) + o = run('umount -R /mnt') if args['post'] == 'reboot': o = run('reboot now') |