mirror of
https://github.com/projectacrn/acrn-hypervisor.git
synced 2025-05-30 19:05:33 +00:00
DM: add socket module for command monitor
The socket module is to provide interfaces below, the unix domain socket can be created by command monitor for the communication between DM instance and libvirt daemon or kata container: init_socket: allocate a new socket instance according to socket path deinit_socket: free a socket instance open_socket: open one unix domain socket server, initialize a socket, create one thread to listen to client, another thread to poll message from client. close_socket: close one unix domain socket server find_socket_client: find socket client instance according to fd write_socket_char: send message through unix domain socket server v1--v2: Update some log message format and copyright format. v2-->v3: Update SOCKET_MAX_CLIENT to 1H since the socket instance of command monitor only have one client (libvirt or kata container). Tracked-On: #5921 Signed-off-by: Xiangyang Wu <xiangyang.wu@intel.com> Acked-by: Wang, Yu1 <yu1.wang@intel.com>
This commit is contained in:
Xiangyang Wu
acrnsi-robot
parent
3594eb66e9
commit
26fdfc6a30
300
devicemodel/core/cmd_monitor/socket.c
Normal file
300
devicemodel/core/cmd_monitor/socket.c
Normal file
@ -0,0 +1,300 @@
|
||||
/*
|
||||
* Copyright (C) 2022 Intel Corporation
|
||||
* SPDX-License-Identifier: BSD-3-Clause
|
||||
*/
|
||||
|
||||
#include <errno.h>
|
||||
#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/queue.h>
|
||||
#include <sys/socket.h>
|
||||
#include <netinet/in.h>
|
||||
#include <arpa/inet.h>
|
||||
#include "socket.h"
|
||||
#include "log.h"
|
||||
#include "mevent.h"
|
||||
|
||||
static int setup_and_listen_unix_socket(const char *sock_path, int num)
|
||||
{
|
||||
struct sockaddr_un s_un;
|
||||
int sock_fd, ret;
|
||||
|
||||
sock_fd = socket(AF_UNIX, SOCK_STREAM, 0);
|
||||
if (sock_fd < 0)
|
||||
goto err;
|
||||
s_un.sun_family = AF_UNIX;
|
||||
ret = snprintf(s_un.sun_path, sizeof(s_un.sun_path), "%s",
|
||||
sock_path);
|
||||
if (ret < 0 || ret >= sizeof(s_un.sun_path))
|
||||
goto err_close_socket;
|
||||
|
||||
if (bind(sock_fd, (struct sockaddr *)&s_un, sizeof(s_un)) < 0)
|
||||
goto err_close_socket;
|
||||
if (listen(sock_fd, num) < 0)
|
||||
goto err_close_socket;
|
||||
pr_info("Listening on: %s, socket fd = %d.\r\n", sock_path, sock_fd);
|
||||
return sock_fd;
|
||||
err_close_socket:
|
||||
close(sock_fd);
|
||||
err:
|
||||
return -1;
|
||||
}
|
||||
static void free_socket_client(struct socket_dev *sock, struct socket_client *client)
|
||||
{
|
||||
pthread_mutex_lock(&sock->client_mtx);
|
||||
LIST_REMOVE(client, list);
|
||||
pthread_mutex_unlock(&sock->client_mtx);
|
||||
|
||||
close(client->fd);
|
||||
client->fd = -1;
|
||||
free(client);
|
||||
}
|
||||
|
||||
int write_socket_char(struct socket_client *client)
|
||||
{
|
||||
struct msghdr msg;
|
||||
struct iovec iov[1];
|
||||
int ret;
|
||||
char control[CMSG_SPACE(sizeof(int))];
|
||||
struct cmsghdr *cmsg;
|
||||
|
||||
memset(&msg, 0, sizeof(msg));
|
||||
memset(control, 0, sizeof(control));
|
||||
|
||||
iov[0].iov_base = (void *)client->buf;
|
||||
iov[0].iov_len = client->len;
|
||||
|
||||
msg.msg_iov = iov;
|
||||
msg.msg_iovlen = 1;
|
||||
|
||||
msg.msg_control = control;
|
||||
msg.msg_controllen = sizeof(control);
|
||||
|
||||
cmsg = CMSG_FIRSTHDR(&msg);
|
||||
cmsg->cmsg_len = CMSG_LEN(sizeof(int));
|
||||
cmsg->cmsg_level = SOL_SOCKET;
|
||||
cmsg->cmsg_type = SCM_RIGHTS;
|
||||
memcpy(CMSG_DATA(cmsg), &client->fd, sizeof(int));
|
||||
|
||||
do {
|
||||
ret = sendmsg(client->fd, &msg, 0);
|
||||
} while (ret < 0 && errno == EINTR);
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
int read_socket_char(struct socket_dev *sock, struct socket_client *client)
|
||||
{
|
||||
struct iovec iov;
|
||||
struct msghdr msg;
|
||||
struct cmsghdr *cmsg;
|
||||
char buf[CMSG_SPACE(sizeof(int))];
|
||||
int fdflags_recvmsg = MSG_CMSG_CLOEXEC;
|
||||
|
||||
memset(&msg, 0, sizeof(msg));
|
||||
memset(client->buf, '\0', CLIENT_BUF_LEN);
|
||||
iov.iov_base = client->buf;
|
||||
iov.iov_len = CLIENT_BUF_LEN;
|
||||
msg.msg_iov = &iov;
|
||||
msg.msg_iovlen = 1;
|
||||
msg.msg_name = NULL;
|
||||
msg.msg_namelen = 0;
|
||||
|
||||
msg.msg_control = buf;
|
||||
msg.msg_controllen = sizeof(buf);
|
||||
cmsg = CMSG_FIRSTHDR(&msg);
|
||||
cmsg->cmsg_level = SOL_SOCKET;
|
||||
cmsg->cmsg_type = SCM_RIGHTS;
|
||||
cmsg->cmsg_len = CMSG_LEN(sizeof(int));
|
||||
|
||||
client->len = recvmsg(client->fd, &msg, fdflags_recvmsg);
|
||||
if (client->len <= 0) {
|
||||
pr_err("Socket Disconnect: %d.\r\n", client->fd);
|
||||
free_socket_client(sock, client);
|
||||
sock->num_client--;
|
||||
return -1;
|
||||
}
|
||||
if (client->len == CLIENT_BUF_LEN) {
|
||||
pr_err("Socket buffer overflow!\r\n");
|
||||
return -1;
|
||||
}
|
||||
for (unsigned int i = 0U; i < CLIENT_BUF_LEN; i++)
|
||||
if (client->buf[i] == 0xa)
|
||||
client->buf[i] = '\0';
|
||||
pr_info("Receive data:(%s)\n", client->buf);
|
||||
if (sock->data_handler != NULL)
|
||||
sock->data_handler(client->buf, client->fd);
|
||||
return 0;
|
||||
}
|
||||
static struct socket_client *new_socket_client(struct socket_dev *sock)
|
||||
{
|
||||
struct socket_client *client;
|
||||
|
||||
client = calloc(1, sizeof(*client));
|
||||
if (!client) {
|
||||
pr_err("%s: failed to allocate memory for client\n",
|
||||
__func__);
|
||||
goto alloc_client;
|
||||
}
|
||||
|
||||
client->addr_len = sizeof(client->addr);
|
||||
client->fd =
|
||||
accept(sock->sock_fd, (struct sockaddr *)&client->addr,
|
||||
&client->addr_len);
|
||||
if (client->fd < 0) {
|
||||
if (sock->listening)
|
||||
pr_err("%s: Failed to accept from fd %d, err: %s\n",
|
||||
__func__, sock->sock_fd, strerror(errno));
|
||||
goto accept_con;
|
||||
}
|
||||
|
||||
pthread_mutex_lock(&sock->client_mtx);
|
||||
LIST_INSERT_HEAD(&sock->client_head, client, list);
|
||||
pthread_mutex_unlock(&sock->client_mtx);
|
||||
|
||||
return client;
|
||||
|
||||
accept_con:
|
||||
free(client);
|
||||
alloc_client:
|
||||
return NULL;
|
||||
}
|
||||
static void *listen_socket_client(void *arg)
|
||||
{
|
||||
struct socket_dev *sock = (struct socket_dev *)arg;
|
||||
struct socket_client *client;
|
||||
|
||||
pr_info("Socket Listening %d...\n", sock->sock_fd);
|
||||
while (sock->listening) {
|
||||
/* wait connection */
|
||||
if (sock->num_client >= SOCKET_MAX_CLIENT) {
|
||||
usleep(500000);
|
||||
continue;
|
||||
}
|
||||
|
||||
client = new_socket_client(sock);
|
||||
if (!client) {
|
||||
usleep(500000);
|
||||
continue;
|
||||
}
|
||||
pr_info("Socket Connected:%d\n", client->fd);
|
||||
sock->num_client++;
|
||||
}
|
||||
pr_info("Stop listening %d.\n", sock->sock_fd);
|
||||
return NULL;
|
||||
}
|
||||
static void *socket_poll_events(void *arg)
|
||||
{
|
||||
struct socket_dev *sock = (struct socket_dev *)arg;
|
||||
struct socket_client *client;
|
||||
fd_set rfd;
|
||||
int max_fd = 0;
|
||||
struct timeval timeout;
|
||||
struct socket_client *poll_client[SOCKET_MAX_CLIENT];
|
||||
int nfd, i;
|
||||
|
||||
pr_info("Polling socket %d.\n", sock->sock_fd);
|
||||
while (sock->polling) {
|
||||
max_fd = 0;
|
||||
nfd = 0;
|
||||
pthread_mutex_lock(&sock->client_mtx);
|
||||
FD_ZERO(&rfd);
|
||||
LIST_FOREACH(client, &sock->client_head, list) {
|
||||
FD_SET(client->fd, &rfd);
|
||||
poll_client[nfd] = client;
|
||||
nfd++;
|
||||
if (client->fd > max_fd)
|
||||
max_fd = client->fd;
|
||||
}
|
||||
pthread_mutex_unlock(&sock->client_mtx);
|
||||
|
||||
timeout.tv_sec = 0;
|
||||
timeout.tv_usec = 10000;
|
||||
select(max_fd + 1, &rfd, NULL, NULL, &timeout);
|
||||
|
||||
for (i = 0; i < nfd; i++) {
|
||||
client = poll_client[i];
|
||||
if (!FD_ISSET(client->fd, &rfd))
|
||||
continue;
|
||||
|
||||
if (read_socket_char(sock, client) < 0)
|
||||
continue;
|
||||
}
|
||||
}
|
||||
pr_info("Stop polling on socket %d.\n", sock->sock_fd);
|
||||
|
||||
return NULL;
|
||||
}
|
||||
struct socket_client *find_socket_client(struct socket_dev *sock, int fd)
|
||||
{
|
||||
struct socket_client *client = NULL;
|
||||
pthread_mutex_lock(&sock->client_mtx);
|
||||
LIST_FOREACH(client, &sock->client_head, list) {
|
||||
if (client->fd == fd)
|
||||
break;
|
||||
}
|
||||
pthread_mutex_unlock(&sock->client_mtx);
|
||||
return client;
|
||||
}
|
||||
int open_socket(struct socket_dev *sock, data_handler_f *fn)
|
||||
{
|
||||
sock->listening = true;
|
||||
sock->polling = true;
|
||||
sock->data_handler = fn;
|
||||
pthread_mutex_init(&sock->client_mtx, NULL);
|
||||
unlink(sock->unix_sock_path);
|
||||
sock->sock_fd = setup_and_listen_unix_socket(sock->unix_sock_path, SOCKET_MAX_CLIENT);
|
||||
if (sock->sock_fd < 0)
|
||||
return -1;
|
||||
pthread_create(&sock->listen_thread, NULL, listen_socket_client, sock);
|
||||
pthread_create(&sock->connect_thread, NULL, socket_poll_events, sock);
|
||||
return 0;
|
||||
}
|
||||
|
||||
void close_socket(struct socket_dev *sock)
|
||||
{
|
||||
struct socket_client *client, *tclient;
|
||||
|
||||
sock->listening = false;
|
||||
sock->polling = false;
|
||||
shutdown(sock->sock_fd, SHUT_RDWR);
|
||||
|
||||
pthread_join(sock->listen_thread, NULL);
|
||||
pthread_join(sock->connect_thread, NULL);
|
||||
|
||||
pthread_mutex_lock(&sock->client_mtx);
|
||||
list_foreach_safe(client, &sock->client_head, list, tclient) {
|
||||
LIST_REMOVE(client, list);
|
||||
close(client->fd);
|
||||
client->fd = -1;
|
||||
free(client);
|
||||
}
|
||||
pthread_mutex_unlock(&sock->client_mtx);
|
||||
|
||||
close(sock->sock_fd);
|
||||
unlink(sock->unix_sock_path);
|
||||
}
|
||||
struct socket_dev *init_socket(char *path)
|
||||
{
|
||||
struct socket_dev *sock;
|
||||
|
||||
sock = calloc(1, sizeof(*sock));
|
||||
if (!sock) {
|
||||
pr_err("%s: Failed to allocate memory for socket\n", __func__);
|
||||
return NULL;
|
||||
}
|
||||
memset(sock, 0x0, sizeof(struct socket_dev));
|
||||
strncpy(sock->unix_sock_path, path, UNIX_SOCKET_PATH_MAX - 1);
|
||||
return sock;
|
||||
}
|
||||
void deinit_socket(struct socket_dev *sock)
|
||||
{
|
||||
if (sock != NULL)
|
||||
free(sock);
|
||||
}
|
||||
76
devicemodel/core/cmd_monitor/socket.h
Normal file
76
devicemodel/core/cmd_monitor/socket.h
Normal file
@ -0,0 +1,76 @@
|
||||
/*
|
||||
* Copyright (C) 2022 Intel Corporation
|
||||
* SPDX-License-Identifier: BSD-3-Clause
|
||||
*/
|
||||
#ifndef _SOCKET_H_
|
||||
#define _SOCKET_H_
|
||||
#include <sys/queue.h>
|
||||
#include <pthread.h>
|
||||
#include <sys/un.h>
|
||||
|
||||
#define BUFFER_SIZE 16U
|
||||
#define UNIX_SOCKET_PATH_MAX 1024U
|
||||
#define CLIENT_BUF_LEN 4096U
|
||||
#define SOCKET_MAX_CLIENT 1
|
||||
|
||||
typedef void data_handler_f(char *cmd_name, int fd);
|
||||
|
||||
struct socket_client {
|
||||
struct sockaddr_un addr;
|
||||
int fd;
|
||||
socklen_t addr_len;
|
||||
char buf[CLIENT_BUF_LEN];
|
||||
int len; /* buf len */
|
||||
|
||||
LIST_ENTRY(socket_client) list;
|
||||
};
|
||||
|
||||
struct socket_dev {
|
||||
char unix_sock_path[UNIX_SOCKET_PATH_MAX];
|
||||
int sock_fd;
|
||||
int logfd;
|
||||
|
||||
data_handler_f *data_handler;
|
||||
|
||||
bool listening;
|
||||
bool polling;
|
||||
pthread_t listen_thread;
|
||||
pthread_t connect_thread;
|
||||
|
||||
LIST_HEAD(client_list, socket_client) client_head; /* clients for this server */
|
||||
pthread_mutex_t client_mtx;
|
||||
int num_client;
|
||||
};
|
||||
|
||||
/**
|
||||
* @brief Send message through unix domain socket server
|
||||
*/
|
||||
int write_socket_char(struct socket_client *client);
|
||||
/**
|
||||
* @brief Find socket client instance according to fd
|
||||
*/
|
||||
struct socket_client *find_socket_client(struct socket_dev *sock, int fd);
|
||||
/**
|
||||
* @brief Open one unix domain socket server, initialize a socket,
|
||||
* create one thread to listen to client, another thread to poll message from client.
|
||||
*/
|
||||
int open_socket(struct socket_dev *sock, data_handler_f *fn);
|
||||
/**
|
||||
* @brief Close one unix domain socket server
|
||||
*/
|
||||
void close_socket(struct socket_dev *sock);
|
||||
/**
|
||||
* @brief Initialize a socket
|
||||
*
|
||||
* @param path the socket path
|
||||
* @return struct socket_dev* the socket instance
|
||||
*/
|
||||
struct socket_dev *init_socket(char *path);
|
||||
/**
|
||||
* @brief Deinit a socket
|
||||
*
|
||||
* @param sock The pointer of socket instance
|
||||
*/
|
||||
void deinit_socket(struct socket_dev *sock);
|
||||
|
||||
#endif
|
||||
Loading…
Reference in New Issue
Block a user