mirror of
https://github.com/xcat2/confluent.git
synced 2024-11-23 10:02:04 +00:00
83 lines
3.4 KiB
Python
83 lines
3.4 KiB
Python
import argparse
|
|
import struct
|
|
import os
|
|
import sys
|
|
|
|
# This script will create a PXE or HTTP boot entry specific for a linux nic using
|
|
# convenience of the linux interface name. It can help produce precise network boot order
|
|
# configuration more easily if that is useful
|
|
|
|
# only tested with Lenovo system firmware on PCIe network adapters
|
|
|
|
parser = argparse.ArgumentParser(description='Generate an efivars boot entry file')
|
|
parser.add_argument('interface', help='Current linux interface name to boot from')
|
|
parser.add_argument('outfile', help='The file to write, e.g. /sys/firmware/efi/efivars/Boot0007-8be4df61-93ca-11d2-aa0d-00e098032b8c')
|
|
parser.add_argument('-u', help='The HTTP url to boot, "" for generic HTTP boot instead of PXE')
|
|
args = parser.parse_args()
|
|
with open('/sys/class/net/{0}/address'.format(sys.argv[1])) as macaddr:
|
|
macaddr = macaddr.read()
|
|
macaddr = macaddr.split(':')
|
|
macaddr = [ int(x, 16) for x in macaddr ]
|
|
macaddr = bytes(macaddr)
|
|
|
|
olddir = os.getcwd()
|
|
os.chdir('/sys/class/net/{0}'.format(sys.argv[1]))
|
|
devdir = os.readlink('/sys/class/net/{0}/device'.format(sys.argv[1]))
|
|
os.chdir(devdir)
|
|
pcipath = os.getcwd()
|
|
pcipath = pcipath.split('/')
|
|
os.chdir('/'.join(pcipath[:4]))
|
|
acpiid = os.readlink('firmware_node')
|
|
os.chdir(acpiid)
|
|
hid = '0A03' # do not adapt, always say pciroot, pcieroot doesn't work open('hid').read().replace('PNP', '')
|
|
hid = int(hid, 16) << 16 | 0x41d0
|
|
hid = struct.pack('<I', hid)
|
|
uid = open('uid').read()
|
|
uid = struct.pack('<I', int(uid, 16))
|
|
pcipath = pcipath[4:]
|
|
os.chdir(olddir)
|
|
url = args.u
|
|
if url is not None:
|
|
url = url.encode('utf8')
|
|
|
|
with open(sys.argv[2], 'wb') as output:
|
|
output.write(struct.pack('<I', 7)) # write out attributes, for now always 7
|
|
output.write(struct.pack('<I', 1)) # write out attributes, for now always 1
|
|
# 0x50 is smallest, acpi, mac, and ip devpaths, each pci path hop adds 6 bytes
|
|
baselen = 0x50
|
|
if url is not None: #well, we need more length for http boot
|
|
baselen += len(url) + 4
|
|
output.write(struct.pack('<H', baselen + (len(pcipath) * 6)))
|
|
desc = sys.argv[1]
|
|
if url is None:
|
|
desc += '-pxe'
|
|
else:
|
|
desc += '-http'
|
|
output.write(desc.encode('utf16')[2:])
|
|
output.write(b'\x00\x00') # terminate the description
|
|
# first we need pciroot, which is an acpi path, table 52, HID is PNP0A03
|
|
output.write(b'\x02\x01\x0c\x00') # acpi devpath header
|
|
output.write(hid) # ACPI hid, either PciRoot or PcieRoot
|
|
output.write(uid) # ACPI uid of the pci root complex
|
|
for pcihop in pcipath:
|
|
pcihop = pcihop.rsplit(':', 1)[-1]
|
|
output.write(b'\x01\x01\x06\x00')
|
|
devnum, funnum = pcihop.split('.')
|
|
devnum = int(devnum, 16)
|
|
funnum = int(funnum, 16)
|
|
output.write(struct.pack('BB', funnum, devnum))
|
|
output.write(b'\x03\x0b\x25\x00')
|
|
output.write(macaddr)
|
|
output.write(b'\x00' * (32 - len(macaddr))) # pad out to the rest of the 32 bytes
|
|
output.write(b'\x01') # type is ethernet, 1, per rfc 3232
|
|
output.write(b'\x03\x0c\x1b\x00') # ipv4 header, table 71 from section 10.3.4.12
|
|
output.write(b'\x00' * 23) # all zero for the data
|
|
# if args.u, we need a Messaging device path, subtypee 24, uri, table 83
|
|
if url is not None:
|
|
output.write(b'\x03\x18')
|
|
output.write(struct.pack('<H', 4 + len(url)))
|
|
if url:
|
|
output.write(url)
|
|
output.write(b'\x7f\xff\x04\x00') # table 45, device path end structure
|
|
|