Commit Graph

504 Commits (0ca16bc46cd68b17c2d377258aa877652d561149)
 

Author SHA1 Message Date
Alejandro Sirgo Rica adbf02d170 utils: add set_windows_hostname
Add function to redefine the hostname of a Windows install.
Windows hostnames can't be larger than 15 characters due to
legacy heritage.

Hostname modification is done by modifying 3 registry values in
the ControlSetXXX entry of the SYSTEM hive. ControlSet001 is
generally the entry to be edited but one must query the value
of the 'Current' key in the 'Select' entry of the SYSTEM hive
to retrieve the active ControlSet.

The hostname has to be introduced in the following entries:
path = 'ControlSetXX/Control/ComputerName/ComputerName'
key = 'ComputerName'

path = 'ControlSetXXX/Services/Tcpip/Parameters'
key = 'HostName'
key = 'NV Hostname'

The value to store in those keys is of tipe SZ and has to be
encoded in UCS-2 Little Endian (utf-16-le).
2024-07-29 15:07:22 +02:00
Alejandro Sirgo Rica 567fea276e utils: add postinstall.py
Replace ConfigureOs script by native calls to:

- update BCD via hivex using bcd.py and winreg.py infrastructure.
- restore efi bootloader restore_windows_efi_bootloader().

Call legacy scripts for remaining postinstall procedures to
replace them incrementally.

Define variable CONFIGUREOS_LEGACY_ENABLED as False by default.
Run legacy configureOs when CONFIGUREOS_LEGACY_ENABLED = True.
This serves as a auxiliar method to keep the restoration functional
in case of problems with the new configure_os logic.
2024-07-29 15:07:22 +02:00
Alejandro Sirgo Rica 9d5291f47a utils: add BCD native support
Implement update_bcd() as replacement of ogWindowsBootParameters.
The legacy function modified the BCD of a freshly restored system
invoking the privative binary "spartlnx".

The script edits a set of BCD entries needed for a proper system
boot.

Each main BCD entry is identified by an UUID and contain a set of
subnodes, these hold the configuration and entry information.
Each node contains data in the form of key-value.

Common BCD structure:

'Objects'
...

The Boot Manager entry is always identified by the UUID
9dea862c-5cdd-4e70-acc1-f32b344d4795.
Some entries always have the same UUID as identifier such the
Boot Manager while other have different UUID depending on the system.
To identify these entries with a not known UUID we query the value
'Type' of the node 'Description'. This contains a 32 bit value whose
bytes codify the nature of the entry.

We obtain 3 different values as a tuple, each value is the result of
applying a bitmask to the Type value. These masks are 0xf0000000,
0x00f00000 and 0x000fffff. The resulting tuple of 3 values is then
used to obtain the corresponding entry from a map.

The data we modify from the BCD are disk and partition references
to point to the new disk and partition in the system where the
images are restored. Partition and disk information is stored as
UUID in specific offsets inside binary data in the BCD.
To update these we need to obtain the disk and partition UUIDs,
then convert it to bytes as follows:

original UUID: C4C61C51-3456-4733-96AD-AE760A41C392
UUID as bytes: 51 1C C6 C4 56 34 33 47 96 AD AE 76 0A 41 C3 92

The modified entries are: Resume from Hibernation, Windows Boot
Loader OS, Windows Boot Loader Recovery, Windows Recovery,
Boot Loader Settings, Windows Boot Manager and Windows Memory
Diagnostic.
Some of these options could be omited as the system restoration
does not include a recovery partition so in this case all the
recovery related entries just point to the main system partition.

Most entries are edited modifying the value corresponding to the
key 'Element' in 2 subnodes of the 'Elements' node.
These subnodes are '11000001' and '21000001'.

'Objects'

The 2 values stored in these 2 entries is binary data where we
store the partition and disk bytes. We simply replace the byte
representation of out UUIDS in the binary data. Partition is
stored in offset 32 and the disk in offset 56.
The exception is the Bootloader Recovery entry, in which the
partition offset is 84 and the disk offset 108.

Note that the legacy function only does a proper BCD edit in UEFI
systems. The new implementation follows the same behavior with
the possibility of implementing BCD modification under MBR
partitions in the future.

Set the field 16000009 (RECOVERY_ENABLED) to the value x00 to
disable recovery in both Resume from Hibernation and Windows
Boot Loader OS entries in the BCD.

The system install does not include any recovery partition so
it makes no sense to have it enabled.

This commit is preparatory work for the new native postinstall code.
2024-07-29 15:07:22 +02:00
Alejandro Sirgo Rica bb03e92840 utils: add win edit registry utilities
Add winreg.py to the utils folder. Implement hive enum types,
hive handler validation and validated get_* functions for
nodes and registry values.

Implement the utility hive operations through the Hivex library.

This serves as preparatory work for BCD manipulation but it also
has potential to improve registry usage in previous code.

UCS-2 Little Endian is the prefered windows registry text encoding
for binary content. Define a WINDOWS_HIVE_ENCODING global variable
to use when encoding string to write in the win registry.

This commit is preparatory work for the new native postinstall code.
2024-07-29 15:07:22 +02:00
Alejandro Sirgo Rica 7ab965c0b5 utils: add restore_windows_efi_bootloader
Add restore_windows_efi_bootloader to copy the EFI loader from
the filesystem in the restored system into the EFI partition.

This commit is preparatory work for the new native postinstall code.
2024-07-29 15:07:22 +02:00
Alejandro Sirgo Rica 4f31bde549 utils: add functions to obtain disk and partition UUIDs
Add get_partition_id and get_disk_id to obtain the UUID of a disk
or partition as a string. These values are obtained invoking the
program blkid as a subprocess.

This commit is preparatory work for the new native postinstall code.
2024-07-29 15:07:22 +02:00
Alejandro Sirgo Rica 41b5f830c6 utils: consolidate code to find efi loader
Add find_windows_efi_loader and find_linux_efi_loader to reduce
code duplication and to centralize efi loader path modifications.
2024-07-29 15:07:22 +02:00
Alejandro Sirgo Rica e20cda122b utils: rename ogExtendFs to extend_filesystem
Use a more consistent name format for the ogExtendFs function.
2024-07-29 15:07:22 +02:00
OpenGnSys Support Team 2c10b4c92a utils: fs: error out if mkfs fails
If formatting fails, log shows:

	mkfs.ntfs reports return code 1 for /dev/sda2

but ogclient reports success to the ogserver.

Raise an exception so front-end gets an indication that formatting has failed.
2024-07-29 14:22:48 +02:00
OpenGnSys Support Team 59fa3bb120 live: improve logging for file removal from cache
Provide a bit more logging to make it easier to debug issues.
2024-07-21 22:14:07 +02:00
OpenGnSys Support Team fcfa5f9fbc live: add logging to indicate image file copy from samba
For consistency with tiptorrent download.
2024-07-21 21:59:27 +02:00
OpenGnSys Support Team 59ab9204ce utils: consolidate logging to write checksum file to cache
Move log message to function that is called both by unicast and tiptorrent.
2024-07-21 21:56:09 +02:00
OpenGnSys Support Team 8ac8dc306e utils: missing f-string with filesystem resize error
... failed to resize {partdev} with ext4
2024-07-15 09:50:38 +02:00
OpenGnSys Support Team 8453a8d9ca utils: clean up error reporting related to checksum
When checksum is not available, it displays:

(2024-07-15 09:04:14) ogClient: [ERROR] - URL error when fetching checksum: Not Found

"URL error" is leaking an internal implementation details, reword this report.
2024-07-15 09:50:37 +02:00
OpenGnSys Support Team 9d8a95cc74 live: add checksum field to image/create response
Report image checksum to ogserver through HTTP response.
2024-07-15 09:50:30 +02:00
OpenGnSys Support Team 43039749c5 live: check permissions when trying to fetch file via unicast
check sufficient permissions and bail out in case of issues.
2024-06-27 12:16:14 +02:00
OpenGnSys Support Team a9d8cdd210 live: revisit error log when failing to validate checksum
Add explicit check for .full.sum after downloading it.

Rewrite errors log, one of them is misleading when checksum validation
fails, it refers to missing .full.sum, but it could be a different
reason.
2024-06-27 11:46:30 +02:00
Alejandro Sirgo Rica 6282cb41a8 live: add restricted execution mode to shell/run
Try to find the script to run for a shell/run request in
/opt/opengnsys/shell/, restricted mode is enabled if the
script is found.

Excute the script without shell=True and executable=OG_SHELL in
restricted mode.

Restricted mode is a safer execution method as it only executes
code manually defined by the administrator.

Each script needs to define a shebang, this way more than just bash
is supported.
2024-06-25 13:41:41 +02:00
OpenGnSys Support Team 1c9a13cd96 rest: add cmd field to POST /shell/run
echo command that has been run for storage in ogserver, until GET /shell/output
is invoked.
2024-06-21 14:59:48 +02:00
OpenGnSys Support Team 19cd1b9a78 rest: add retcode field to POST /shell/run
provide return code as result to ogserver.

Update virtual mode driver to return dummy value, although this command
is unimplemented, this seems to be broken due to possible TypeError when
accessing result from caller.
2024-06-21 14:47:30 +02:00
Alejandro Sirgo Rica 399a5dceb8 live: handle no cache in tiptorrent and unicast restore
Properly report the lack of cache partition when restoring an
image using TIPTORRENT and UNICAST as methods.
Abort any restore in case of no cache partition.
2024-06-07 10:09:50 +02:00
Alejandro Sirgo Rica ec132cfb95 utils: remove bogus cache not found error after refresh command
Remove log message "Cannot find device path to cache" spamming
logs during operations such as refresh in clients without CACHE
partition.
2024-06-07 09:12:51 +02:00
Alejandro Sirgo Rica 5698aa66d2 utils: fix get_image_info regression
Fix image size, permissions and creation time.
Improve error report related to these parameters now showing the
exact cause of the problem if any occurred during the definition
of image size, file permissions or image creation time values.
2024-06-03 11:29:57 +02:00
OpenGnSys Support Team a1774c795b live: add winrecov partition type
DE94BBA4-06D1-4D40-A16A-BFD50179D6AC is already set in in GUID_MAP as partition
code, but not in GPT_PARTTYPES.
2024-06-02 00:48:31 +02:00
OpenGnSys Support Team d732a214bf live: revisit logging for checksum file creation
Add logging before calculating checksum to provide a hint to user on what is
going on while ogClient is still busy.
2024-05-31 17:26:20 +02:00
Alejandro Sirgo Rica 914bd993fc src: cleanup ogGetImageInfo
Rename ogGetImageInfo to get_image_info.
Move code from ogOperations.py to obtain image data into
get_image_info.
2024-05-30 17:22:23 +02:00
Alejandro Sirgo Rica 34007857f6 src: rename legacy.py into image.py
legacy.py contais mostly functions related to system images.
Rename the file to better represent the contents in it.
2024-05-30 17:22:23 +02:00
Alejandro Sirgo Rica 9a5e83ea1a src: stop using hardcoded paths to cache image directory
Use the constant OG_CACHE_IMAGE_PATH from cache.py to obtain the
location of the directory where images are stored.
This way the path can be changed from one single point.
2024-05-30 17:22:23 +02:00
Alejandro Sirgo Rica dd003e688f tiptorrent: check cache availability in tip_write_csum
Remove old TODO message.
Check if the cache is available before trying to generate an
image's checksum.
2024-05-30 17:22:23 +02:00
Alejandro Sirgo Rica 60803fe0ed src: add cache info to the image/restore response
Add a 'cache' field into the json payload the client sends to
the server after a restore operation so the server can update
the new cache contents.

Resquest response structure:
{
    ...
    'cache': [
        {'name': 'windows.img', 'size': 2432370213, checksum: '5d4dcc677bc19f40a647d0002f4ade90'},
        {'name': 'linux.img', 'size': 243234534213, checksum: '3eb22f888f88a55ad954f55644e1192e'}
    ]
    ...
}
2024-05-30 17:22:23 +02:00
Alejandro Sirgo Rica 72d8943576 virtual: handle copy error in image restore
Add a proper error report for the shutil.copy operation in
image_restore() instead of silently returning.
2024-05-30 17:22:23 +02:00
Alejandro Sirgo Rica 8de2b785a9 src: add POST cache/delete method
Add API REST method to delete cache contents.

Resquest payload structure:
{
    'images': ['windows.img', 'linux.img']
}

The client will try to delete as many images in cache as available
with names matching the list of filenames in the 'images' field.

Resquest response structure:
{
    'cache': [
        {'name': 'windows.img', 'size': 2432370213, checksum: '5d4dcc677bc19f40a647d0002f4ade90'},
        {'name': 'linux.img', 'size': 243234534213, checksum: '3eb22f888f88a55ad954f55644e1192e'}
    ]
}
2024-05-30 17:22:23 +02:00
Alejandro Sirgo Rica e2d48ba3a0 live: add cache contents to the /refresh payload
Add the list of images in the client's cache partition in the
payload sent to the server.
The information sent is a list of {image_name, img_size, checksum}
elements where img_size is the size of the respective image in bytes.

Resquest response structure:
{
    ...
    'cache': [
        {'name': 'windows.img', 'size': 2432370213, checksum: '5d4dcc677bc19f40a647d0002f4ade90'},
        {'name': 'linux.img', 'size': 243234534213, checksum: '3eb22f888f88a55ad954f55644e1192e'}
    ]
    ...
}
2024-05-30 17:22:18 +02:00
Alejandro Sirgo Rica 03d33d4fa9 utils: cache: redefine cache path constants
Rename OGIMG as OG_IMAGE_PATH.
Rename OGCACHE_MOUNTPOINT as OG_CACHE_PATH.
Define OG_CACHE_IMAGE_PATH as OG_CACHE_PATH + OG_IMAGE_PATH.

This will serve to have a unique point to obtain cache related
paths.
2024-05-27 17:09:10 +02:00
OpenGnSys Support Team 4626383cc4 live: remove unused return value in image_restore()
Never used what configureOs() returns, remove it.
2024-05-27 11:31:19 +02:00
OpenGnSys Support Team 9952c3cc85 live: incorrect reference to image checksum file in logs
checksum file name end by .img.full.sum, not .full.sum
2024-05-25 09:57:17 +02:00
OpenGnSys Support Team fa5b37b2a6 live: rename variable that stores json body in refresh()
Just a simple cleanup.
2024-05-21 11:51:29 +02:00
OpenGnSys Support Team b9ca7278a9 live: parttypes: add EFI partition type in MBR partition scheme
Add 0xef partition type for EFI in MBR.
2024-05-14 11:49:54 +02:00
OpenGnSys Support Team 3eb4332b38 live: partcodes: remove reference to website
Remove reference to external website in code.
2024-05-14 11:35:16 +02:00
OpenGnSys Support Team d109e99dbe utils: rename cache_probe() to get_cache_dev_path()
This method reports the /dev path to cache partition, rename it.

Add explicit check if blkid is successful.

And add logging to report that device path to cache is not found.
2024-05-09 11:47:06 +02:00
Alejandro Sirgo Rica 6bfbf6cc7b utils: add error checks to checksum file creation in tip_write_csum
Add a check for potential permission or IO errors during the
creation of the image checksum.
2024-05-07 20:47:05 +02:00
Alejandro Sirgo Rica abea6583d3 utils: add mkdir error report in mount_mkdir
Add exception checks to the os.mkdir operation and log the error
found. The previous implementation was too optimistic and only
handled mount related errors.
2024-05-07 17:24:41 +02:00
Alejandro Sirgo Rica 493d998a4e utils: remove unused json import
The json library was a dependency during the development of the
boot OS functions and it is no longer needed in that file.
2024-05-07 16:08:38 +02:00
Alejandro Sirgo Rica b0d03ce588 utils:fs: fix logging statement in get_filesystem_type
Call the error() logging function from the logging object instead
of the non existent log variable previously referenced.
2024-05-07 12:20:26 +02:00
Alejandro Sirgo Rica de17bb6812 utils:fs: add mkfs logs when return code is not 0
Report mkfs failure for every partition. This does not raise an
exception as that would skip partprobe operations and the mkfs
operations in the next potentially well formated partitions.
2024-05-07 12:20:26 +02:00
OpenGnSys Support Team 9ffe1c81bf live: report LINUX-SWAP instead of SWAP
ogCP expects LINUX-SWAP to specify a swap filesystem.

Add a similar workaround to the one that is done for VFAT for symmetry between
inputs and outputs that circulate over the API.
2024-05-07 11:54:39 +02:00
OpenGnSys Support Team 1ca3639389 live: rewrite log in case tiptorrent client fails
Specify that image file cannot be found in cache because tiptorrent has failed,
otherwise it is confusing.
2024-05-06 19:10:48 +02:00
OpenGnSys Support Team 8ecd57552a live: restore partprobe before building filesystem
Otherwise mkfs silently fails because OS reports out-of-sync partition table.
2024-05-06 19:08:12 +02:00
OpenGnSys Support Team 2dd5105995 live: force flush to disk after partition table is written 2024-05-06 18:51:57 +02:00
OpenGnSys Support Team f42e2ba201 live: partprobe breaks with mounted partitions
partprobe requires that all disk partitions are unmounted.

partprobe needs to be called to report the OS that the partition table
has changed, otherwise ogclient reports incorrect partition information.

iterate over the partition list and mount cache after partprobe is
called.
2024-05-06 18:33:41 +02:00