mirror of
https://github.com/projectacrn/acrn-hypervisor.git
synced 2025-07-01 09:42:10 +00:00
misc: life_mngr: refine script to trigger system shutdown request
Provide a script to user to trigger system shutdown request in service VM or user VM. System shutdown logic have been integrated in the lifecycle manager, only need to send system shutdown request command through unix domain socket in this script. v3-->v4: Rewirte system shutdown trigger script using python since phython script have better portability. v4-->v5: Update command name and copyright header. Tracked-On: #6652 Signed-off-by: Xiangyang Wu <xiangyang.wu@intel.com> Reviewed-by: fei1.li@intel.com
This commit is contained in:
Xiangyang Wu
wenlingz
parent
0d85463abe
commit
7a5146c2ed
@ -1,216 +0,0 @@
|
||||
#! /bin/bash
|
||||
# Copyright (C) 2020 Intel Corporation.
|
||||
# SPDX-License-Identifier: BSD-3-Clause
|
||||
|
||||
MAX_ARRAY=6
|
||||
|
||||
started="started"
|
||||
stopped="stopped"
|
||||
SOS="sos"
|
||||
UOS="uos"
|
||||
SHUTDOWN="shutdown"
|
||||
ACKED="acked"
|
||||
|
||||
#config
|
||||
IP_ADDR="127.0.0.1"
|
||||
SOCKET_PORT="8193"
|
||||
|
||||
#send message by socket
|
||||
send_message()
|
||||
{
|
||||
message="$1"
|
||||
echo "Sending: $message"
|
||||
echo -ne "$message" >&6 &
|
||||
}
|
||||
|
||||
#read message by socket
|
||||
read_message()
|
||||
{
|
||||
read -r -d $'\0' ret_msg <&6
|
||||
}
|
||||
|
||||
#send message by tty
|
||||
send_pre_message()
|
||||
{
|
||||
message="$1"
|
||||
echo "Sending: $message"
|
||||
echo -ne "$message" > /dev/ttyS1
|
||||
}
|
||||
|
||||
#read message by tty
|
||||
read_pre_message()
|
||||
{
|
||||
read -r -d $'\0' ret_msg < /dev/ttyS1
|
||||
}
|
||||
|
||||
power_off_post_vms() {
|
||||
vm_list=$(acrnctl list)
|
||||
echo $vm_list
|
||||
|
||||
array=($vm_list)
|
||||
num=${#array[@]}
|
||||
echo "Number of VMs: " $num
|
||||
|
||||
if [ $num -gt $MAX_ARRAY ]; then
|
||||
echo "There are no VMs running or acrnctl encountered an internal error."
|
||||
return 0
|
||||
fi
|
||||
|
||||
#shut down post-launched VMs
|
||||
for ((i=0; i<$num; i+=2))
|
||||
do
|
||||
if [ ${array[$i+1]} == $started ]; then
|
||||
echo "Shutting down: " ${array[$i]}
|
||||
acrnctl stop ${array[$i]}
|
||||
sleep 5s
|
||||
fi
|
||||
done
|
||||
|
||||
return 1
|
||||
}
|
||||
|
||||
check_post_vms_status() {
|
||||
#check post-launched VM status for some time
|
||||
check_times=5
|
||||
while [[ $check_times > 0 ]]
|
||||
do
|
||||
vm_list=$(acrnctl list)
|
||||
array=($vm_list)
|
||||
num=${#array[@]}
|
||||
echo "VM status: " $vm_list
|
||||
|
||||
if [ $num -gt $MAX_ARRAY ]; then
|
||||
sleep 5s
|
||||
let check_times--
|
||||
echo "Check #" $check_times
|
||||
continue;
|
||||
fi
|
||||
|
||||
flag=0
|
||||
|
||||
for ((i=0; i<$num; i+=2))
|
||||
do
|
||||
if [ ${array[$i+1]} != $stopped ]; then
|
||||
flag=1
|
||||
break;
|
||||
fi
|
||||
done
|
||||
|
||||
if [ $flag -eq 1 ]; then
|
||||
sleep 5s
|
||||
let check_times--
|
||||
echo "Check #" $check_times
|
||||
else
|
||||
echo "VM status: " $vm_list
|
||||
break;
|
||||
fi
|
||||
|
||||
if [ $check_times -eq 0 ]; then
|
||||
echo "Timed out waiting for VMs..."
|
||||
break;
|
||||
fi
|
||||
done
|
||||
|
||||
if [ $check_times -gt 0 ]; then
|
||||
return 1;
|
||||
else
|
||||
return 0;
|
||||
fi
|
||||
}
|
||||
|
||||
check_post_vms_alive() {
|
||||
#check if there is any post-launched VM alive, and return if there is
|
||||
vm_list=$(acrnctl list)
|
||||
array=($vm_list)
|
||||
num=${#array[@]}
|
||||
|
||||
for ((i=0; i<$num; i+=2))
|
||||
do
|
||||
if [ ${array[$i+1]} == $started ]; then
|
||||
echo $vm_list " VM alive!"
|
||||
return 1
|
||||
fi
|
||||
done
|
||||
|
||||
echo "No VM alive: " $vm_list
|
||||
return 0
|
||||
}
|
||||
|
||||
if [ "$1" = "$SOS" ]; then
|
||||
try_times=2
|
||||
|
||||
#shut down post-launched VMs
|
||||
while [[ $try_times -gt 0 ]]
|
||||
do
|
||||
echo "Checking whether post-launched VMs are alive..."
|
||||
check_post_vms_alive
|
||||
if [ $? -eq 0 ]; then
|
||||
try_times=1
|
||||
break
|
||||
fi
|
||||
|
||||
echo "Powering off VMs..."
|
||||
power_off_post_vms
|
||||
if [ $? -eq 0 ]; then
|
||||
break
|
||||
fi
|
||||
|
||||
echo "Checking the status of post-launched VMs..."
|
||||
check_post_vms_status
|
||||
if [ $? -eq 1 ]; then
|
||||
break
|
||||
fi
|
||||
|
||||
let try_times--
|
||||
done
|
||||
|
||||
if [ $try_times -eq 0 ]; then
|
||||
echo "S5 failed!"
|
||||
exit
|
||||
fi
|
||||
|
||||
echo $(acrnctl list)
|
||||
|
||||
#send shutdown message to the pre-launched VM
|
||||
send_pre_message $SHUTDOWN
|
||||
#read ack message from the pre-launched VM
|
||||
read_pre_message
|
||||
echo "ret_msg: $ret_msg"
|
||||
|
||||
#check the ack message
|
||||
if [ "$ret_msg" = "$ACKED" ]; then
|
||||
echo "Received ACK message"
|
||||
else
|
||||
echo "Could not receive ACK message from the pre-launched User VM"
|
||||
exit
|
||||
fi
|
||||
|
||||
#all pre-launched and post-launched VMs have shut down
|
||||
echo "Shutting down the Service VM itself..."
|
||||
sleep 3s
|
||||
poweroff
|
||||
|
||||
elif [ "$1" = "$UOS" ]; then
|
||||
echo "Trying to open socket..."
|
||||
if ! exec 6<>/dev/tcp/$IP_ADDR/$SOCKET_PORT
|
||||
then
|
||||
echo "Failed to open socket"
|
||||
exit 1
|
||||
fi
|
||||
echo "Socket opened"
|
||||
|
||||
#send shutdown message
|
||||
send_message $SHUTDOWN
|
||||
#read ack message
|
||||
read_message
|
||||
echo "ret_msg: $ret_msg"
|
||||
|
||||
#check the ack message
|
||||
if [ "$ret_msg" = "$ACKED" ]; then
|
||||
echo "Received ACK message"
|
||||
else
|
||||
echo "Could not receive ACK message"
|
||||
fi
|
||||
else
|
||||
echo "Error! Please use: ./s5_trigger.sh sos OR ./s5_trigger.sh uos"
|
||||
fi
|
||||
31
misc/services/life_mngr/s5_trigger_linux.py
Normal file
31
misc/services/life_mngr/s5_trigger_linux.py
Normal file
@ -0,0 +1,31 @@
|
||||
#!/usr/bin/env python3
|
||||
#
|
||||
# Copyright (C) 2021 Intel Corporation.
|
||||
#
|
||||
# SPDX-License-Identifier: BSD-3-Clause
|
||||
#
|
||||
import socket
|
||||
|
||||
class SocketClient:
|
||||
def __init__(self):
|
||||
pass
|
||||
def connect_to_server(self):
|
||||
SOKET_ADDR = '/var/lib/life_mngr/monitor.sock'
|
||||
SHUTDOWN_REQ = 'req_sys_shutdown'
|
||||
BUF_LEN = 1024
|
||||
|
||||
# unix domain sockets
|
||||
server_address = SOKET_ADDR
|
||||
socket_family = socket.AF_UNIX
|
||||
socket_type = socket.SOCK_STREAM
|
||||
|
||||
sock = socket.socket(socket_family, socket_type)
|
||||
sock.connect(server_address)
|
||||
sock.sendall(SHUTDOWN_REQ.encode())
|
||||
data = sock.recv(BUF_LEN)
|
||||
print(f"Waiting for ACK message...: {data.decode()}")
|
||||
sock.close()
|
||||
|
||||
if __name__ == "__main__":
|
||||
socket_client_obj = SocketClient()
|
||||
socket_client_obj.connect_to_server()
|
||||
Loading…
Reference in New Issue
Block a user