github/acrn-hypervisor
1
0
Fork 0
mirror of https://github.com/projectacrn/acrn-hypervisor.git synced 2025-06-02 20:35:32 +00:00
Code Issues Projects Releases Wiki Activity

misc:life_mngr: support S5 triggered by RTVM

Browse Source 
Currently the life_mngr daemon has supported:
listen from the SOS shutdown command,
this patch add support listen shutdown command from RTVM.

and this life_mngr daemon can also run on SOS for listening
shutdown commond from acrn-dm(triggered by RTVM).

Tracked-On: #4446
Signed-off-by: Mingqiang Chi <mingqiang.chi@intel.com>
Reviewed-by: Minggui Cao <minggui.cao@intel.com>
Reviewed-by: Yuhong, Tao <yuhong.tao@intel.com>
Acked-by: Yan, Like <like.yan@intel.com>
This commit is contained in:
Mingqiang Chi 2020-02-26 22:35:36 +08:00 committed by wenlingz
parent e641202c26
commit 7e74ed557b
1 changed files with 352 additions and 53 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

397
misc/life_mngr/life_mngr.c
View File

@ -7,28 +7,89 @@
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <stdbool.h>
#include <string.h>
#include <termios.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <pthread.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#define SOS_REQ "shutdown"
#define UOS_ACK "acked"
#define SHUTDOWN_CMD "shutdown"
#define ACK_CMD "acked"
#define FAIL_CMD "fail"
#define BUFF_SIZE 16U
#define MSG_SIZE 8U
#define NODE_SIZE 3U
#define TRY_SEND_CNT 3U
#define SOS_SOCKET_PORT (0x2000U)
#define UOS_SOCKET_PORT (SOS_SOCKET_PORT + 1U)
enum nodetype {
NODE_UNKNOWN = 0,
NODE_UOS_SERVER,
NODE_SOS_CLIENT,
/* life_mngr process run in SOS or UOS */
enum process_env {
PROCESS_UNKNOWN = 0,
PROCESS_RUN_IN_SOS,
PROCESS_RUN_IN_UOS,
};
int set_serial_interface_attributes(int fd, int speed)
/* Enumerated shutdown state machine only for UOS thread */
enum shutdown_state {
SHUTDOWN_REQ_WAITING = 0, /* Can receive shutdown cmd in this state */
SHUTDOWN_ACK_WAITING, /* Wait acked message from SOS */
SHUTDOWN_REQ_FROM_SOS, /* Trigger shutdown by SOS */
SHUTDOWN_REQ_FROM_UOS, /* Trigger shutdown by UOS */
};
/* the file description for dev/ttyS1 */
int tty_dev_fd = 0;
enum shutdown_state shutdown_state = SHUTDOWN_REQ_WAITING;
static FILE *log_fd;
#define LOG_PRINTF(format, args...) \
do { fprintf(log_fd, format, args); \
fflush(log_fd); } while (0)
#define LOG_WRITE(args) \
do { fwrite(args, 1, sizeof(args), log_fd); \
fflush(log_fd); } while (0)
/* it read from vuart, and if end is '\0' or '\n' or len = buff-len it will return */
int receive_message(int fd, uint8_t *buffer, int buf_len)
{
int rc = 0, count = 0;
do {
rc = read(fd, buffer + count, buf_len - count);
if (rc > 0) {
count += rc;
if ((buffer[count - 1] == '\0') || (buffer[count - 1] == '\n')
|| (count == buf_len)) {
break;
}
}
} while (rc > 0);
return count;
}
int send_message(int fd, char *buf, int len)
{
int ret = 0;
ret = write(fd, buf, len);
return (ret == len) ? (0) : (-1);
}
int set_tty_attr(int fd, int speed)
{
struct termios tty;
if (tcgetattr(fd, &tty) < 0) {
printf("Error from tcgetattr: %s\n", strerror(errno));
LOG_PRINTF("error from tcgetattr: %s\n", strerror(errno));
return errno;
}
@ -36,18 +97,15 @@ int set_serial_interface_attributes(int fd, int speed)
cfsetispeed(&tty, (speed_t)speed);
/* set input-mode */
tty.c_iflag &= ~(IGNBRK | BRKINT | PARMRK | ISTRIP | INLCR | IGNCR | ICRNL | IXON);
tty.c_iflag &= ~(IGNBRK | BRKINT | PARMRK |
ISTRIP | INLCR | IGNCR | ICRNL | IXON);
/* set output-mode */
tty.c_oflag &= ~OPOST;
/* set control-mode */
tty.c_cflag |= (CLOCAL | CREAD);
tty.c_cflag &= ~CSIZE;
tty.c_cflag |= CS8;
tty.c_cflag &= ~PARENB;
tty.c_cflag &= ~CSTOPB;
tty.c_cflag &= ~CRTSCTS;
tty.c_cflag |= (CLOCAL | CREAD | CS8);
tty.c_cflag &= ~(CSIZE | PARENB | CSTOPB | CRTSCTS);
/* set local-mode */
tty.c_lflag &= ~(ECHO | ECHONL | ICANON | ISIG | IEXTEN);
@ -56,62 +114,303 @@ int set_serial_interface_attributes(int fd, int speed)
tty.c_cc[VMIN] = 1;
tty.c_cc[VTIME] = 1;
tcflush(fd, TCIOFLUSH);
if (tcsetattr(fd, TCSANOW, &tty) != 0) {
printf("Error from tcsetattr: %s\n", strerror(errno));
LOG_PRINTF("error from tcsetattr: %s\n", strerror(errno));
return errno;
}
return 0;
}
int setup_socket_listen(unsigned short port)
{
int listen_fd;
struct sockaddr_in socket_addr;
int opt = SO_REUSEADDR;
listen_fd = socket(AF_INET, SOCK_STREAM, 0);
if (listen_fd == -1) {
LOG_WRITE("create an socket endpoint error!\n");
exit(1);
}
setsockopt(listen_fd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
memset(&socket_addr, 0, sizeof(socket_addr));
socket_addr.sin_family = AF_INET;
socket_addr.sin_port = htons(port);
socket_addr.sin_addr.s_addr = htonl(INADDR_ANY);
if (bind(listen_fd, (struct sockaddr *)&socket_addr, sizeof(socket_addr)) == -1) {
LOG_PRINTF("Bind to a socket(fd = 0x%x) error\n", listen_fd);
close(listen_fd);
exit(1);
}
if (listen(listen_fd, 5) == -1) {
LOG_PRINTF("listen for connection error on a socket(fd=0x%x)\n", listen_fd);
close(listen_fd);
exit(1);
}
return listen_fd;
}
/* this thread runs on Service VM:
* communiate between lifecycle-mngr and acrn-dm process in Service VM side
*/
void *sos_socket_thread(void *arg)
{
int listen_fd, connect_fd;
struct sockaddr_in client;
socklen_t len = sizeof(struct sockaddr_in);
int num, ret;
char buf[BUFF_SIZE];
listen_fd = setup_socket_listen(SOS_SOCKET_PORT);
LOG_PRINTF("life_mngr:listen_fd=0x%x socket port is 0x%x\r\n", listen_fd, SOS_SOCKET_PORT);
while (1) {
memset(buf, 0, BUFF_SIZE);
connect_fd = accept(listen_fd, (struct sockaddr *)&client, &len);
if (connect_fd == -1) {
LOG_WRITE("accept a connection error on a socket\n");
close(listen_fd);
exit(1);
}
/* Here assume the socket communication is reliable, no need to try */
num = read(connect_fd, buf, sizeof(SHUTDOWN_CMD));
if (num == -1) {
LOG_PRINTF("read error on a socket(fd = 0x%x)\n", connect_fd);
close(connect_fd);
close(listen_fd);
exit(1);
}
LOG_PRINTF("received msg [%s]\r\n", buf);
if (strncmp(SHUTDOWN_CMD, (const char *)buf, strlen(SHUTDOWN_CMD)) == 0) {
ret = send_message(connect_fd, ACK_CMD, sizeof(ACK_CMD));
if (ret != 0) {
LOG_WRITE("Send acked message to acrn-dm VM fail\n");
}
LOG_WRITE("Receive shutdown command from User VM\r\n");
ret = system("~/s5_trigger.sh sos");
LOG_PRINTF("call s5_trigger.sh ret=0x%x\r\n", ret);
break;
}
}
close(connect_fd);
close(listen_fd);
return NULL;
}
/* this thread runs on User VM:
* User VM wait for message from Service VM
*/
void *listener_fn_to_sos(void *arg)
{
int ret;
int retry = TRY_SEND_CNT;
bool shutdown_self = false;
unsigned char buf[BUFF_SIZE];
/* UOS-server wait for message from SOS */
do {
memset(buf, 0, BUFF_SIZE);
ret = receive_message(tty_dev_fd, buf, BUFF_SIZE);
if (ret > 0) {
LOG_PRINTF("receive buf :%s ret=0x%x shutdown_state=0x%x\r\n", buf, ret, shutdown_state);
}
switch (shutdown_state) {
/* it can receive shutdown command from SOS */
case SHUTDOWN_REQ_WAITING:
case SHUTDOWN_REQ_FROM_SOS:
if ((ret > 0) && (strncmp(SHUTDOWN_CMD, (const char *)buf, strlen(SHUTDOWN_CMD)) == 0)) {
shutdown_state = SHUTDOWN_REQ_FROM_SOS;
ret = send_message(tty_dev_fd, ACK_CMD, sizeof(ACK_CMD));
if (ret != 0) {
LOG_WRITE("UOS send acked message failed!\n");
} else {
shutdown_self = true;
}
LOG_WRITE("UOS start shutdown\n");
}
break;
/* it will try to resend shutdown cmd to sos if there is no acked message */
case SHUTDOWN_ACK_WAITING:
if ((ret > 0) && (strncmp(ACK_CMD, (const char *)buf, strlen(ACK_CMD)) == 0)) {
LOG_WRITE("received acked message from Service VM\n");
shutdown_self = true;
} else {
if (retry > 0) {
if (send_message(tty_dev_fd, SHUTDOWN_CMD, sizeof(SHUTDOWN_CMD) != 0)) {
LOG_PRINTF("Try resend shutdown cmd failed cnt = %d\n", retry);
}
retry--;
} else {
LOG_PRINTF("Cann't not receive acked message from SOS, have try %d times\r\n",
TRY_SEND_CNT);
shutdown_state = SHUTDOWN_REQ_WAITING;
retry = TRY_SEND_CNT;
}
}
break;
default:
LOG_PRINTF("Invalid shutdown_state=0x%x\r\n", shutdown_state);
break;
}
/* will poweroff self*/
if (shutdown_self) {
break;
}
sleep(1);
} while (1);
ret = system("poweroff");
return NULL;
}
/* this thread runs on User VM:
* User VM wait for shutdown from other process (e.g. s5_trigger.sh)
*/
void *listener_fn_to_operator(void *arg)
{
int listen_fd, connect_fd;
struct sockaddr_in client;
socklen_t len = sizeof(struct sockaddr_in);
int num, ret;
char buf[BUFF_SIZE];
listen_fd = setup_socket_listen(UOS_SOCKET_PORT);
LOG_PRINTF("listen_fd=0x%x socket port is 0x%x\r\n",
listen_fd, UOS_SOCKET_PORT);
while (1) {
connect_fd = accept(listen_fd, (struct sockaddr *)&client, &len);
if (connect_fd == -1) {
LOG_WRITE("accept a connection error on a socket\n");
close(listen_fd);
exit(1);
}
/* Here assume the socket communication is reliable, no need to try */
num = read(connect_fd, buf, sizeof(SHUTDOWN_CMD));
if (num == -1) {
LOG_PRINTF("read error on a socket(fd=0x%x)\n", connect_fd);
close(connect_fd);
close(listen_fd);
exit(1);
}
LOG_PRINTF("Receive msg is %s by the socket\r\n", buf);
if (strncmp(SHUTDOWN_CMD, (const char *)buf, strlen(SHUTDOWN_CMD)) == 0) {
if (shutdown_state != SHUTDOWN_REQ_WAITING) {
LOG_PRINTF("Cann't handle shutdown cmd during shutdowning...\
shutdown_state=0x%x\n", shutdown_state);
if (send_message(connect_fd, FAIL_CMD, sizeof(FAIL_CMD)) != 0) {
LOG_WRITE("Send fail message to the initiator failed\n");
}
continue;
}
shutdown_state = SHUTDOWN_REQ_FROM_UOS;
/* send acked message to the caller */
LOG_WRITE("Send acked message to the caller\r\n");
ret = send_message(connect_fd, ACK_CMD, sizeof(ACK_CMD));
if (ret != 0) {
LOG_WRITE("Send acked message fail\n");
}
LOG_WRITE("send shutdown message to sos\r\n");
/* send shutdown command to the Servcie VM */
ret = send_message(tty_dev_fd, SHUTDOWN_CMD, sizeof(SHUTDOWN_CMD));
if (ret != 0) {
LOG_WRITE("Send shutdown command to Service VM fail\n");
shutdown_state = SHUTDOWN_REQ_WAITING;
} else {
shutdown_state = SHUTDOWN_ACK_WAITING;
}
LOG_PRINTF("Write shutdown msg ret=0x%x\r\n", ret);
}
}
close(connect_fd);
close(listen_fd);
return NULL;
}
int main(int argc, char *argv[])
{
char *devname_uos = "";
int fd_uos = 0;
unsigned char recvbuf[BUFF_SIZE];
enum nodetype node = NODE_UNKNOWN;
int ret = 0;
char *devname_uos = "";
enum process_env env = PROCESS_UNKNOWN;
pthread_t sos_socket_pid;
/* User VM wait for shutdown from Service VM */
pthread_t uos_thread_pid_1;
/* User VM wait for shutdown from other process */
pthread_t uos_thread_pid_2;
log_fd = fopen("/var/log/life_mngr.log", "w+");
if (log_fd == NULL) {
printf("open log file failed\r\n");
return -EINVAL;
}
if (argc <= 2) {
printf("Too few options. Example: [./life_mngr uos /dev/ttyS1].\n");
LOG_WRITE("Too few options. Example: [./life_mngr uos /dev/ttyS1] or ./life_mngr sos /dev/ttyS1]\n");
fclose(log_fd);
return -EINVAL;
}
if (strncmp("uos", argv[1], NODE_SIZE) == 0) {
node = NODE_UOS_SERVER;
} else if (strncmp("sos", argv[1], NODE_SIZE) == 0) {
node = NODE_SOS_CLIENT;
} else {
printf("Invalid param. Example: [./life_mngr uos /dev/ttyS1].\n");
env = PROCESS_RUN_IN_UOS;
devname_uos = argv[2];
tty_dev_fd = open(devname_uos, O_RDWR | O_NOCTTY | O_SYNC | O_NONBLOCK);
if (tty_dev_fd < 0) {
LOG_PRINTF("Error opening %s: %s\n", devname_uos, strerror(errno));
fclose(log_fd);
return errno;
}
if (set_tty_attr(tty_dev_fd, B115200) != 0) {
return -EINVAL;
}
if (node == NODE_UOS_SERVER) {
devname_uos = argv[2];
fd_uos = open(devname_uos, O_RDWR | O_NOCTTY | O_SYNC);
if (fd_uos < 0) {
printf("Error opening %s: %s\n", devname_uos, strerror(errno));
return errno;
}
set_serial_interface_attributes(fd_uos, B115200);
ret = pthread_create(&uos_thread_pid_1, NULL, listener_fn_to_sos, NULL);
ret = pthread_create(&uos_thread_pid_2, NULL, listener_fn_to_operator, NULL);
} else if (strncmp("sos", argv[1], NODE_SIZE) == 0) {
env = PROCESS_RUN_IN_SOS;
ret = pthread_create(&sos_socket_pid, NULL, sos_socket_thread, NULL);
} else {
LOG_WRITE("Invalid param. Example: [./life_mngr uos /dev/ttyS1] or ./life_mngr sos /dev/ttyS1]\n");
fclose(log_fd);
return -EINVAL;
}
/* UOS-server wait for shutdown from SOS */
do {
if (node == NODE_UOS_SERVER) {
memset(recvbuf, 0, sizeof(recvbuf));
ret = read(fd_uos, recvbuf, sizeof(recvbuf));
if (env == PROCESS_RUN_IN_SOS) {
pthread_join(sos_socket_pid, NULL);
} else if (env == PROCESS_RUN_IN_UOS) {
pthread_join(uos_thread_pid_1, NULL);
pthread_join(uos_thread_pid_2, NULL);
close(tty_dev_fd);
} else {
if (strncmp(SOS_REQ, (const char *)recvbuf, MSG_SIZE) == 0) {
ret = write(fd_uos, UOS_ACK, sizeof(UOS_ACK));
if (ret != sizeof(UOS_ACK)) {
printf("UOS acked fail\n");
}
printf("SOS start shutdown\n");
ret = system("poweroff");
break;
}
}
} while (1);
fclose(log_fd);
return ret;
}