forked from microsoft/mu_plus
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Fix Line Endings in Repo (LF -> CRLF) (microsoft#67)
Converts line endings in the following packages: * AdvLoggerPkg: Fix line endings (LF to CRLF) * DfciPkg: Fix line endings (LF to CRLF) * HidPkg: Fix line endings (LF to CRLF) * MfciPkg: Fix line endings (LF to CRLF) * MsCorePkg: Fix line endings (LF to CRLF) * MsGraphicsPkg: Fix line endings (LF to CRLF) * MsWheaPkg: Fix line endings (LF to CRLF) * PcBdsPkg: Fix line endings (LF to CRLF) * UefiTestingPkg: Fix line endings (LF to CRLF) * XmlSupportPkg: Fix line endings (LF to CRLF) * ZeroTouchPkg: Fix line endings (LF to CRLF) (cherry picked from commit a6195ca in release/202202) Signed-off-by: Michael Kubacki <michael.kubacki@microsoft.com>
- Loading branch information
Showing
310 changed files
with
22,064 additions
and
22,064 deletions.
There are no files selected for viewing
228 changes: 114 additions & 114 deletions
228
AdvLoggerPkg/Application/DecodeUefiLog/UefiVariablesSupportLib.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,114 +1,114 @@ | ||
| # @file | ||
| # | ||
| # Python lib to support Reading and writing UEFI variables from windows | ||
| # | ||
| # Modified from original source to not produce error messages for Variable NOT FOUND | ||
| # | ||
| # Copyright (c), Microsoft Corporation | ||
| # SPDX-License-Identifier: BSD-2-Clause-Patent | ||
|
|
||
| import os, sys | ||
| from ctypes import * | ||
| import logging | ||
| import pywintypes | ||
| import win32api, win32process, win32security, win32file | ||
| import winerror | ||
|
|
||
| kernel32 = windll.kernel32 | ||
| EFI_VAR_MAX_BUFFER_SIZE = 1024*1024 | ||
|
|
||
| class UefiVariable(object): | ||
|
|
||
| def __init__(self): | ||
| # enable required SeSystemEnvironmentPrivilege privilege | ||
| privilege = win32security.LookupPrivilegeValue( None, 'SeSystemEnvironmentPrivilege' ) | ||
| token = win32security.OpenProcessToken( win32process.GetCurrentProcess(), win32security.TOKEN_READ|win32security.TOKEN_ADJUST_PRIVILEGES ) | ||
| win32security.AdjustTokenPrivileges( token, False, [(privilege, win32security.SE_PRIVILEGE_ENABLED)] ) | ||
| win32api.CloseHandle( token ) | ||
|
|
||
| # import firmware variable API | ||
| try: | ||
| self._GetFirmwareEnvironmentVariable = kernel32.GetFirmwareEnvironmentVariableW | ||
| self._GetFirmwareEnvironmentVariable.restype = c_int | ||
| self._GetFirmwareEnvironmentVariable.argtypes = [c_wchar_p, c_wchar_p, c_void_p, c_int] | ||
| self._SetFirmwareEnvironmentVariable = kernel32.SetFirmwareEnvironmentVariableW | ||
| self._SetFirmwareEnvironmentVariable.restype = c_int | ||
| self._SetFirmwareEnvironmentVariable.argtypes = [c_wchar_p, c_wchar_p, c_void_p, c_int] | ||
| self._SetFirmwareEnvironmentVariableEx = kernel32.SetFirmwareEnvironmentVariableExW | ||
| self._SetFirmwareEnvironmentVariableEx.restype = c_int | ||
| self._SetFirmwareEnvironmentVariableEx.argtypes = [c_wchar_p, c_wchar_p, c_void_p, c_int, c_int] | ||
| except AttributeError as msg: | ||
| logging.warn( "G[S]etFirmwareEnvironmentVariableW function doesn't seem to exist" ) | ||
| pass | ||
|
|
||
| # | ||
| # Helper function to create buffer for var read/write | ||
| # | ||
| def CreateBuffer(self, init, size=None): | ||
| """CreateBuffer(aString) -> character array | ||
| CreateBuffer(anInteger) -> character array | ||
| CreateBuffer(aString, anInteger) -> character array | ||
| """ | ||
| if isinstance(init, str): | ||
| if size is None: | ||
| size = len(init)+1 | ||
| buftype = c_char * size | ||
| buf = buftype() | ||
| buf.value = init | ||
| return buf | ||
| elif isinstance(init, int): | ||
| buftype = c_char * init | ||
| buf = buftype() | ||
| return buf | ||
| raise TypeError(init) | ||
|
|
||
| # | ||
| #Function to get variable | ||
| # return a tuple of error code and variable data as string | ||
| # | ||
| def GetUefiVar(self, name, guid ): | ||
| err = 0 #success | ||
| efi_var = create_string_buffer( EFI_VAR_MAX_BUFFER_SIZE ) | ||
| if self._GetFirmwareEnvironmentVariable is not None: | ||
| logging.info("calling GetFirmwareEnvironmentVariable( name='%s', GUID='%s' ).." % (name, "{%s}" % guid) ) | ||
| length = self._GetFirmwareEnvironmentVariable( name, "{%s}" % guid, efi_var, EFI_VAR_MAX_BUFFER_SIZE ) | ||
| if (0 == length) or (efi_var is None): | ||
| err = kernel32.GetLastError() | ||
| # | ||
| # Don't produce an error message for NOT_FOUND | ||
| # | ||
| if err != 203: # 203 is NOT FOUND | ||
| logging.error( 'GetFirmwareEnvironmentVariable[Ex] failed (GetLastError = 0x%x)' % err) | ||
| logging.error(WinError()) | ||
| return (err, None, WinError(err)) | ||
| return (err, efi_var[:length], None) | ||
| # | ||
| #Function to set variable | ||
| # return a tuple of boolean status, errorcode, errorstring (None if not error) | ||
| # | ||
| def SetUefiVar(self, name, guid, var=None, attrs=None): | ||
| var_len = 0 | ||
| err = 0 | ||
| errorstring = None | ||
| if var is None: | ||
| var = bytes(0) | ||
| else: | ||
| var_len = len(var) | ||
| success = 0 # Fail | ||
| if(attrs == None): | ||
| if self._SetFirmwareEnvironmentVariable is not None: | ||
| logging.info("Calling SetFirmwareEnvironmentVariable (name='%s', Guid='%s')..." % (name, "{%s}" % guid, )) | ||
| success = self._SetFirmwareEnvironmentVariable(name, "{%s}" % guid, var, var_len) | ||
| else: | ||
| attrs = int(attrs) | ||
| if self._SetFirmwareEnvironmentVariableEx is not None: | ||
| logging.info(" calling SetFirmwareEnvironmentVariableEx( name='%s', GUID='%s', length=0x%X, attributes=0x%X ).." % (name, "{%s}" % guid, var_len, attrs) ) | ||
| success = self._SetFirmwareEnvironmentVariableEx( name, "{%s}" % guid, var, var_len, attrs ) | ||
|
|
||
| if 0 == success: | ||
| err = kernel32.GetLastError() | ||
| logging.error('SetFirmwareEnvironmentVariable failed (GetLastError = 0x%x)' % err ) | ||
| logging.error(WinError()) | ||
| errorstring = WinError(err) | ||
| return (success,err, errorstring) | ||
|
|
||
| # @file | ||
| # | ||
| # Python lib to support Reading and writing UEFI variables from windows | ||
| # | ||
| # Modified from original source to not produce error messages for Variable NOT FOUND | ||
| # | ||
| # Copyright (c), Microsoft Corporation | ||
| # SPDX-License-Identifier: BSD-2-Clause-Patent | ||
|
|
||
| import os, sys | ||
| from ctypes import * | ||
| import logging | ||
| import pywintypes | ||
| import win32api, win32process, win32security, win32file | ||
| import winerror | ||
|
|
||
| kernel32 = windll.kernel32 | ||
| EFI_VAR_MAX_BUFFER_SIZE = 1024*1024 | ||
|
|
||
| class UefiVariable(object): | ||
|
|
||
| def __init__(self): | ||
| # enable required SeSystemEnvironmentPrivilege privilege | ||
| privilege = win32security.LookupPrivilegeValue( None, 'SeSystemEnvironmentPrivilege' ) | ||
| token = win32security.OpenProcessToken( win32process.GetCurrentProcess(), win32security.TOKEN_READ|win32security.TOKEN_ADJUST_PRIVILEGES ) | ||
| win32security.AdjustTokenPrivileges( token, False, [(privilege, win32security.SE_PRIVILEGE_ENABLED)] ) | ||
| win32api.CloseHandle( token ) | ||
|
|
||
| # import firmware variable API | ||
| try: | ||
| self._GetFirmwareEnvironmentVariable = kernel32.GetFirmwareEnvironmentVariableW | ||
| self._GetFirmwareEnvironmentVariable.restype = c_int | ||
| self._GetFirmwareEnvironmentVariable.argtypes = [c_wchar_p, c_wchar_p, c_void_p, c_int] | ||
| self._SetFirmwareEnvironmentVariable = kernel32.SetFirmwareEnvironmentVariableW | ||
| self._SetFirmwareEnvironmentVariable.restype = c_int | ||
| self._SetFirmwareEnvironmentVariable.argtypes = [c_wchar_p, c_wchar_p, c_void_p, c_int] | ||
| self._SetFirmwareEnvironmentVariableEx = kernel32.SetFirmwareEnvironmentVariableExW | ||
| self._SetFirmwareEnvironmentVariableEx.restype = c_int | ||
| self._SetFirmwareEnvironmentVariableEx.argtypes = [c_wchar_p, c_wchar_p, c_void_p, c_int, c_int] | ||
| except AttributeError as msg: | ||
| logging.warn( "G[S]etFirmwareEnvironmentVariableW function doesn't seem to exist" ) | ||
| pass | ||
|
|
||
| # | ||
| # Helper function to create buffer for var read/write | ||
| # | ||
| def CreateBuffer(self, init, size=None): | ||
| """CreateBuffer(aString) -> character array | ||
| CreateBuffer(anInteger) -> character array | ||
| CreateBuffer(aString, anInteger) -> character array | ||
| """ | ||
| if isinstance(init, str): | ||
| if size is None: | ||
| size = len(init)+1 | ||
| buftype = c_char * size | ||
| buf = buftype() | ||
| buf.value = init | ||
| return buf | ||
| elif isinstance(init, int): | ||
| buftype = c_char * init | ||
| buf = buftype() | ||
| return buf | ||
| raise TypeError(init) | ||
|
|
||
| # | ||
| #Function to get variable | ||
| # return a tuple of error code and variable data as string | ||
| # | ||
| def GetUefiVar(self, name, guid ): | ||
| err = 0 #success | ||
| efi_var = create_string_buffer( EFI_VAR_MAX_BUFFER_SIZE ) | ||
| if self._GetFirmwareEnvironmentVariable is not None: | ||
| logging.info("calling GetFirmwareEnvironmentVariable( name='%s', GUID='%s' ).." % (name, "{%s}" % guid) ) | ||
| length = self._GetFirmwareEnvironmentVariable( name, "{%s}" % guid, efi_var, EFI_VAR_MAX_BUFFER_SIZE ) | ||
| if (0 == length) or (efi_var is None): | ||
| err = kernel32.GetLastError() | ||
| # | ||
| # Don't produce an error message for NOT_FOUND | ||
| # | ||
| if err != 203: # 203 is NOT FOUND | ||
| logging.error( 'GetFirmwareEnvironmentVariable[Ex] failed (GetLastError = 0x%x)' % err) | ||
| logging.error(WinError()) | ||
| return (err, None, WinError(err)) | ||
| return (err, efi_var[:length], None) | ||
| # | ||
| #Function to set variable | ||
| # return a tuple of boolean status, errorcode, errorstring (None if not error) | ||
| # | ||
| def SetUefiVar(self, name, guid, var=None, attrs=None): | ||
| var_len = 0 | ||
| err = 0 | ||
| errorstring = None | ||
| if var is None: | ||
| var = bytes(0) | ||
| else: | ||
| var_len = len(var) | ||
| success = 0 # Fail | ||
| if(attrs == None): | ||
| if self._SetFirmwareEnvironmentVariable is not None: | ||
| logging.info("Calling SetFirmwareEnvironmentVariable (name='%s', Guid='%s')..." % (name, "{%s}" % guid, )) | ||
| success = self._SetFirmwareEnvironmentVariable(name, "{%s}" % guid, var, var_len) | ||
| else: | ||
| attrs = int(attrs) | ||
| if self._SetFirmwareEnvironmentVariableEx is not None: | ||
| logging.info(" calling SetFirmwareEnvironmentVariableEx( name='%s', GUID='%s', length=0x%X, attributes=0x%X ).." % (name, "{%s}" % guid, var_len, attrs) ) | ||
| success = self._SetFirmwareEnvironmentVariableEx( name, "{%s}" % guid, var, var_len, attrs ) | ||
|
|
||
| if 0 == success: | ||
| err = kernel32.GetLastError() | ||
| logging.error('SetFirmwareEnvironmentVariable failed (GetLastError = 0x%x)' % err ) | ||
| logging.error(WinError()) | ||
| errorstring = WinError(err) | ||
| return (success,err, errorstring) | ||
Oops, something went wrong.