You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
88 lines
2.9 KiB
88 lines
2.9 KiB
/*
|
|
* Copyright (C) 2020 The Android Open Source Project
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#include <sys/socket.h>
|
|
#include <unistd.h>
|
|
|
|
// Needs to be included after sys/socket.h
|
|
#include <linux/vm_sockets.h>
|
|
|
|
#include <iostream>
|
|
#include <optional>
|
|
|
|
#include "android-base/file.h"
|
|
#include "android-base/logging.h"
|
|
#include "android-base/parseint.h"
|
|
#include "android-base/unique_fd.h"
|
|
#include "virt/VirtualizationTest.h"
|
|
|
|
using namespace android::base;
|
|
using namespace android::os;
|
|
|
|
namespace virt {
|
|
|
|
static constexpr int kGuestPort = 45678;
|
|
static constexpr const char kVmConfigPath[] = "/data/local/tmp/virt-test/vsock_config.json";
|
|
static constexpr const char kTestMessage[] = "HelloWorld";
|
|
|
|
TEST_F(VirtualizationTest, TestVsock) {
|
|
binder::Status status;
|
|
|
|
unique_fd server_fd(TEMP_FAILURE_RETRY(socket(AF_VSOCK, SOCK_STREAM, 0)));
|
|
ASSERT_GE(server_fd, 0) << strerror(errno);
|
|
|
|
struct sockaddr_vm server_sa = (struct sockaddr_vm){
|
|
.svm_family = AF_VSOCK,
|
|
.svm_port = kGuestPort,
|
|
.svm_cid = VMADDR_CID_ANY,
|
|
};
|
|
|
|
int ret = TEMP_FAILURE_RETRY(bind(server_fd, (struct sockaddr *)&server_sa, sizeof(server_sa)));
|
|
ASSERT_EQ(ret, 0) << strerror(errno);
|
|
|
|
LOG(INFO) << "Listening on port " << kGuestPort << "...";
|
|
ret = TEMP_FAILURE_RETRY(listen(server_fd, 1));
|
|
ASSERT_EQ(ret, 0) << strerror(errno);
|
|
|
|
sp<IVirtualMachine> vm;
|
|
unique_fd vm_config_fd(open(kVmConfigPath, O_RDONLY | O_CLOEXEC));
|
|
status =
|
|
mVirtManager->startVm(ParcelFileDescriptor(std::move(vm_config_fd)), std::nullopt, &vm);
|
|
ASSERT_TRUE(status.isOk()) << "Error starting VM: " << status;
|
|
|
|
int32_t cid;
|
|
status = vm->getCid(&cid);
|
|
ASSERT_TRUE(status.isOk()) << "Error getting CID: " << status;
|
|
LOG(INFO) << "VM starting with CID " << cid;
|
|
|
|
LOG(INFO) << "Accepting connection...";
|
|
struct sockaddr_vm client_sa;
|
|
socklen_t client_sa_len = sizeof(client_sa);
|
|
unique_fd client_fd(
|
|
TEMP_FAILURE_RETRY(accept(server_fd, (struct sockaddr *)&client_sa, &client_sa_len)));
|
|
ASSERT_GE(client_fd, 0) << strerror(errno);
|
|
LOG(INFO) << "Connection from CID " << client_sa.svm_cid << " on port " << client_sa.svm_port;
|
|
|
|
LOG(INFO) << "Reading message from the client...";
|
|
std::string msg;
|
|
ASSERT_TRUE(ReadFdToString(client_fd, &msg));
|
|
|
|
LOG(INFO) << "Received message: " << msg;
|
|
ASSERT_EQ(msg, kTestMessage);
|
|
}
|
|
|
|
} // namespace virt
|