sh/cutoff/inform_shopfloor.sh

#!/bin/sh
#
# Copyright 2015 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
#
# This script makes shopfloor call to inform shopfloor server. It can be used
# after factory wiping (in-place wiping tmpfs) or after factory reset
# (in factory reset shim) to inform shopfloor the operation is completed.

SCRIPT_DIR="$(dirname "$(readlink -f "$0")")"
DISPLAY_MESSAGE="${SCRIPT_DIR}/display_wipe_message.sh"
. "${SCRIPT_DIR}/options.sh"

GENERATE_FINALIZE_REQUEST="${SCRIPT_DIR}/generate_finalize_request.sh"

POST_FILE="$(mktemp --tmpdir)"

on_exit() {
  rm -f "${POST_FILE}"
}

die_with_error_message() {
  "${DISPLAY_MESSAGE}" "inform_shopfloor_failed"

  # In factory install shim, we need to reset tty to show message correctly.
  stty -F "${TTY}" sane
  echo "$@" > "${TTY}"
  # Stop here to show error message on the screen.
  read
}

usage_help() {
  echo "Usage: $0 SHOPFLOOR_URL <POST_FILE | factory_reset | factory_wipe>"
}

post_to_shopfloor() {
  local shopfloor_url="$1"
  local post_file="$2"
  local response=""
  local rc=""

  while true; do
    echo "Sending data to shopfloor service ${shopfloor_url}..." >"${TTY}"
    rc=0
    response="$(curl --header 'Content-Type: text/xml' --data "@${post_file}" \
      --connect-timeout 10 --retry 1 -s "${shopfloor_url}")" || rc="$?"
    if [ "${rc}" != 0 ]; then
      echo "Cannot connect to server: ${shopfloor_url}" >"${TTY}"
    elif ! echo "${response}" | grep -qw "methodResponse"; then
      echo "Unknown response from server: ${response}" >"${TTY}"
    elif echo "${response}" | grep -qw "fault"; then
      echo "Shopfloor Service failed, response: ${response}" >"${TTY}"
    else
      echo "SUCCESS: Invoked Shopfloor Service." >"${TTY}"
      return
    fi

    echo "Retry in 10s..." >"${TTY}"
    sleep 10
    # Clear screen.
    printf "\033c" >"${TTY}"
  done
}

main() {
  options_find_tty

  if [ $# -ne 2 ]; then
    usage_help
    exit 1
  fi

  local shopfloor_url="$1"
  if [ -z "${shopfloor_url}" ]; then
    shopfloor_url="${SHOPFLOOR_URL}"
  fi

  if [ -z "${shopfloor_url}" ]; then
    echo "No shopfloor URL specified, ignore inform request."
    exit
  fi

  "${DISPLAY_MESSAGE}" "inform_shopfloor"

  case "$2" in
    factory_reset|factory_wipe )
      if ! "${GENERATE_FINALIZE_REQUEST}" "$2" >"${POST_FILE}"; then
        local err=""
        err="$(cat "${POST_FILE}")"
        die_with_error_message "Failed to generate request: ${err}"
      fi
      post_to_shopfloor "${shopfloor_url}" "${POST_FILE}"
      ;;
    * )
      if [ -f "$2" ]; then
        wait_for_ethernet
        post_to_shopfloor "${shopfloor_url}" "$2"
      else
        usage_help
        exit 1
      fi
      ;;
  esac
}

trap on_exit EXIT
main "$@"