def RunPostflight(runtype):
"""Run the full postflight script."""
# support enterprise/802.1x user-context wifi auth.
# don't bother to perform postflight and exit OK immediately since there's no
# network connection.
if runtype == 'logoutinstall':
sys.exit(0)
# Ensure MSU is installed to /Applications
if not IsAppInPlace():
flight_common.RepairClient()
url = flight_common.GetServerURL()
client = mac_client.SimianAuthClient(
flight_common.GetClientIdentifier('auto')['uuid'], hostname=url)
client.SetAuthToken(flight_common.GetAuth1Token())
# read SecureConfig.plist.
plist = munkicommon.SecureManagedInstallsPreferences()
# Post client_id to server.
client_id = flight_common.GetClientIdentifier(runtype)
pkgs_to_install, apple_updates_to_install = (
flight_common.GetRemainingPackagesToInstall())
params = {
'client_id': flight_common.DictToStr(client_id),
'pkgs_to_install': pkgs_to_install,
'apple_updates_to_install': apple_updates_to_install,
}
client.PostReport('postflight', params)
# Report installs/etc to server.
flight_common.UploadAllManagedInstallReports(
client, client_id.get('on_corp', 'None'))
if not client.LogoutAuthToken():
logging.error('Logout failed')
# expire auth token and remove cookie from plist.
RemoveAuthTokenHeaderFromPlist(plist)
# Delete the temp dir that munkicommon creates on import.
munkicommon.cleanUpTmpDir()
# Mark successful run by writing to last success file.
NoteLastSuccess()
logging.debug('Postflight completed successfully.')
def LoginToServer(secure_config, client_id, user_settings, client_exit=None):
"""Sets an auth token cookie header to a plist object.
Args:
secure_config: secure Preferences object.
client_id: dict client identifier.
user_settings: dict of user settings.
client_exit: optional, default None, str explaining why the client is
requesting to exit its execution.
Returns:
Tuple of a SimianAuthClient, a dict containing feedback from the server.
"""
headers = []
# Preserve all non-Cookie and non-ClientID headers that may exist.
if munkicommon.ADDITIONAL_HTTP_HEADERS_KEY in secure_config:
for header in secure_config[munkicommon.ADDITIONAL_HTTP_HEADERS_KEY]:
if (not header.startswith('Cookie:') and
not header.startswith(MUNKI_CLIENT_ID_HEADER_KEY)):
headers.append(header)
client_id_str = flight_common.DictToStr(client_id)
if user_settings:
try:
user_settings_str = urllib.quote(
json.dumps(flight_common.Flatten(user_settings)))
except TypeError:
logging.error(
'preflight cannot flatten user_settings: %s', str(user_settings))
user_settings_str = ''
else:
user_settings_str = ''
client_params = {
'_report_type': 'preflight',
'client_id': client_id_str,
'user_settings': user_settings_str,
'json': '1',
}
if client_exit:
client_params['client_exit'] = client_exit
client_params = urllib.urlencode(client_params)
url = flight_common.GetServerURL()
client = mac_client.SimianAuthClient(hostname=url)
token = client.GetAuthToken()
response = client.PostReportBody(client_params)
feedback = {}
try:
feedback = json.loads(response[len(JSON_PREFIX):])
except ValueError:
logging.exception('Error parsing JSON')
if not isinstance(feedback, dict):
logging.error(
'preflight failure getting feedback dict (%r)', feedback)
# Add the Cookie and client id to the headers.
headers.append('User-Agent: gzip') # enforce GFE compression
headers.append('Cookie: %s' % token)
headers.append('%s: %s' % (MUNKI_CLIENT_ID_HEADER_KEY, client_id_str))
# Replace AdditionalHttpHeaders with the new headers list.
secure_config[munkicommon.ADDITIONAL_HTTP_HEADERS_KEY] = headers
return client, feedback