116 lines
3.4 KiB
C++
116 lines
3.4 KiB
C++
/*
|
|
* Copyright 2016 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.
|
|
*/
|
|
#define LOG_TAG "VtsAgentTcpServer"
|
|
|
|
#include "TcpServerForRunner.h"
|
|
|
|
#include <netdb.h>
|
|
#include <netinet/in.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/types.h>
|
|
|
|
#include <android-base/logging.h>
|
|
|
|
#include "AgentRequestHandler.h"
|
|
#include "BinderClientToDriver.h"
|
|
#include "test/vts/proto/AndroidSystemControlMessage.pb.h"
|
|
|
|
namespace android {
|
|
namespace vts {
|
|
|
|
// Starts to run a TCP server (foreground).
|
|
int StartTcpServerForRunner(const char* spec_dir_path,
|
|
const char* hal_driver_path32,
|
|
const char* hal_driver_path64,
|
|
const char* shell_driver_path32,
|
|
const char* shell_driver_path64) {
|
|
int sockfd;
|
|
socklen_t clilen;
|
|
struct sockaddr_in serv_addr;
|
|
struct sockaddr_in cli_addr;
|
|
|
|
sockfd = socket(AF_INET, SOCK_STREAM, 0);
|
|
if (sockfd < 0) {
|
|
LOG(ERROR) << "Can't open the socket.";
|
|
return -1;
|
|
}
|
|
|
|
bzero((char*)&serv_addr, sizeof(serv_addr));
|
|
serv_addr.sin_family = AF_INET;
|
|
serv_addr.sin_addr.s_addr = INADDR_ANY;
|
|
serv_addr.sin_port = htons(0);
|
|
|
|
if (::bind(sockfd, (struct sockaddr*)&serv_addr, sizeof(serv_addr)) == -1) {
|
|
PLOG(ERROR) << "bind failed";
|
|
return -1;
|
|
}
|
|
|
|
socklen_t sa_len = sizeof(serv_addr);
|
|
if (getsockname(sockfd, (struct sockaddr*) &serv_addr, &sa_len) == -1) {
|
|
PLOG(ERROR) << "getsockname failed";
|
|
return -1;
|
|
}
|
|
|
|
LOG(DEBUG) << "TCP server port is " << (int)ntohs(serv_addr.sin_port);
|
|
FILE* fp = fopen("/data/local/tmp/vts_tcp_server_port", "wt");
|
|
if (!fp) {
|
|
LOG(ERROR) << "Can't write to "
|
|
<< "/data/local/tmp/vts_tcp_server_port";
|
|
return -1;
|
|
}
|
|
fprintf(fp, "%d", (int) ntohs(serv_addr.sin_port));
|
|
fclose(fp);
|
|
|
|
LOG(DEBUG) << "Listening";
|
|
if (listen(sockfd, 5) == -1) {
|
|
LOG(ERROR) << " listen failed.";
|
|
return -1;
|
|
}
|
|
clilen = sizeof(cli_addr);
|
|
while (true) {
|
|
LOG(DEBUG) << "Accepting";
|
|
int newsockfd = ::accept(sockfd, (struct sockaddr*)&cli_addr, &clilen);
|
|
if (newsockfd < 0) {
|
|
LOG(ERROR) << "accept failed";
|
|
return -1;
|
|
}
|
|
|
|
LOG(DEBUG) << "[runner->agent] NEW SESSION";
|
|
LOG(DEBUG) << "[runner->agent] ===========";
|
|
pid_t pid = fork();
|
|
if (pid == 0) { // child
|
|
close(sockfd);
|
|
LOG(DEBUG) << "Process for a runner - pid = " << getpid();
|
|
AgentRequestHandler handler(spec_dir_path, hal_driver_path32,
|
|
hal_driver_path64, shell_driver_path32,
|
|
shell_driver_path64);
|
|
handler.SetSockfd(newsockfd);
|
|
while (handler.ProcessOneCommand())
|
|
;
|
|
exit(-1);
|
|
} else if (pid < 0) {
|
|
LOG(ERROR) << "Can't fork a child process to handle a session.";
|
|
return -1;
|
|
} else {
|
|
close(newsockfd);
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
} // namespace vts
|
|
} // namespace android
|