mirror of
https://github.com/asterinas/asterinas.git
synced 2025-06-20 13:06:33 +00:00
Implement vsock socket layer
This commit is contained in:
committed by
Tate, Hongliang Tian
parent
83a7937334
commit
ad140cec3c
@ -31,6 +31,7 @@ TEST_APPS := \
|
||||
pthread \
|
||||
pty \
|
||||
signal_c \
|
||||
vsock \
|
||||
|
||||
# The C head and source files of all the apps, excluding the downloaded mongoose files
|
||||
C_SOURCES := $(shell find . -type f \( -name "*.c" -or -name "*.h" \) ! -name "mongoose.c" ! -name "mongoose.h")
|
||||
|
13
regression/apps/scripts/run_vsock_test.sh
Normal file
13
regression/apps/scripts/run_vsock_test.sh
Normal file
@ -0,0 +1,13 @@
|
||||
#!/bin/sh
|
||||
|
||||
# SPDX-License-Identifier: MPL-2.0
|
||||
|
||||
set -e
|
||||
|
||||
VSOCK_DIR=/regression/vsock
|
||||
cd ${VSOCK_DIR}
|
||||
|
||||
echo "Start vsock test......"
|
||||
# ./vsock_server
|
||||
./vsock_client
|
||||
echo "Vsock test passed."
|
5
regression/apps/vsock/Makefile
Normal file
5
regression/apps/vsock/Makefile
Normal file
@ -0,0 +1,5 @@
|
||||
# SPDX-License-Identifier: MPL-2.0
|
||||
|
||||
include ../test_common.mk
|
||||
|
||||
EXTRA_C_FLAGS :=
|
47
regression/apps/vsock/vsock_client.c
Normal file
47
regression/apps/vsock/vsock_client.c
Normal file
@ -0,0 +1,47 @@
|
||||
// SPDX-License-Identifier: MPL-2.0
|
||||
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <sys/socket.h>
|
||||
#include <linux/vm_sockets.h>
|
||||
#include <unistd.h>
|
||||
#include <string.h>
|
||||
|
||||
#define PORT 1234
|
||||
|
||||
int main()
|
||||
{
|
||||
int sock;
|
||||
char *hello = "Hello from client";
|
||||
char buffer[1024] = { 0 };
|
||||
struct sockaddr_vm serv_addr;
|
||||
|
||||
if ((sock = socket(AF_VSOCK, SOCK_STREAM, 0)) < 0) {
|
||||
printf("\n Socket creation error\n");
|
||||
return -1;
|
||||
}
|
||||
printf("\n Create socket successfully!\n");
|
||||
serv_addr.svm_family = AF_VSOCK;
|
||||
serv_addr.svm_cid = VMADDR_CID_HOST;
|
||||
serv_addr.svm_port = PORT;
|
||||
|
||||
if (connect(sock, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) <
|
||||
0) {
|
||||
printf("\nConnection Failed \n");
|
||||
return -1;
|
||||
}
|
||||
printf("\n Socket connect successfully!\n");
|
||||
|
||||
// Send message to the server and receive the reply
|
||||
if (send(sock, hello, strlen(hello), 0) < 0) {
|
||||
printf("\nSend Failed\n");
|
||||
return -1;
|
||||
}
|
||||
printf("Hello message sent\n");
|
||||
if (read(sock, buffer, 1024) < 0) {
|
||||
printf("\nRead Failed\n");
|
||||
return -1;
|
||||
}
|
||||
printf("Server: %s\n", buffer);
|
||||
return 0;
|
||||
}
|
61
regression/apps/vsock/vsock_server.c
Normal file
61
regression/apps/vsock/vsock_server.c
Normal file
@ -0,0 +1,61 @@
|
||||
// SPDX-License-Identifier: MPL-2.0
|
||||
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <sys/socket.h>
|
||||
#include <linux/vm_sockets.h>
|
||||
#include <unistd.h>
|
||||
#include <string.h>
|
||||
|
||||
#define CID 3
|
||||
#define PORT 4321
|
||||
|
||||
int main()
|
||||
{
|
||||
int sock, new_sock;
|
||||
char *hello = "Hello from client";
|
||||
char buffer[1024] = { 0 };
|
||||
struct sockaddr_vm serv_addr, client_addr;
|
||||
int addrlen = sizeof(client_addr);
|
||||
|
||||
if ((sock = socket(AF_VSOCK, SOCK_STREAM, 0)) < 0) {
|
||||
printf("\n Socket creation error\n");
|
||||
return -1;
|
||||
}
|
||||
printf("\nCreate socket successfully\n");
|
||||
serv_addr.svm_family = AF_VSOCK;
|
||||
serv_addr.svm_cid = CID;
|
||||
serv_addr.svm_port = PORT;
|
||||
|
||||
if (bind(sock, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) < 0) {
|
||||
printf("\nBind Failed \n");
|
||||
return -1;
|
||||
}
|
||||
printf("\nBind socket successfully\n");
|
||||
|
||||
if (listen(sock, 3) < 0) {
|
||||
printf("\nListen Failed\n");
|
||||
return -1;
|
||||
}
|
||||
printf("\nListen socket successfully\n");
|
||||
|
||||
if ((new_sock = accept(sock, (struct sockaddr *)&client_addr,
|
||||
(socklen_t *)&addrlen)) < 0) {
|
||||
printf("\nAccept Failed\n");
|
||||
return -1;
|
||||
}
|
||||
printf("\nAccept socket successfully\n");
|
||||
|
||||
// Send message to the server and receive the reply
|
||||
if (read(new_sock, buffer, 1024) < 0) {
|
||||
printf("\nRead Failed\n");
|
||||
return -1;
|
||||
}
|
||||
printf("Client: %s\n", buffer);
|
||||
if (send(new_sock, hello, strlen(hello), 0) < 0) {
|
||||
printf("\nSend Failed\n");
|
||||
return -1;
|
||||
}
|
||||
printf("Hello message sent\n");
|
||||
return 0;
|
||||
}
|
Reference in New Issue
Block a user