blob: 93fc191bcafc49b8eb570559632060c965427b50 [file] [log] [blame]
/*
* Copyright (C) 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.
*/
#include "socket_spec.h"
#include <string>
#include <unistd.h>
#include <android-base/file.h>
#include <android-base/stringprintf.h>
#include <gtest/gtest.h>
TEST(socket_spec, parse_tcp_socket_spec_failure) {
std::string hostname, error, serial;
int port;
EXPECT_FALSE(parse_tcp_socket_spec("sneakernet:5037", &hostname, &port, &serial, &error));
EXPECT_TRUE(error.find("sneakernet") != std::string::npos);
}
TEST(socket_spec, parse_tcp_socket_spec_just_port) {
std::string hostname, error, serial;
int port;
EXPECT_TRUE(parse_tcp_socket_spec("tcp:5037", &hostname, &port, &serial, &error));
EXPECT_EQ("", hostname);
EXPECT_EQ(5037, port);
EXPECT_EQ("", serial);
}
TEST(socket_spec, parse_tcp_socket_spec_bad_ports) {
std::string hostname, error, serial;
int port;
EXPECT_FALSE(parse_tcp_socket_spec("tcp:", &hostname, &port, &serial, &error));
EXPECT_FALSE(parse_tcp_socket_spec("tcp:-1", &hostname, &port, &serial, &error));
EXPECT_FALSE(parse_tcp_socket_spec("tcp:65536", &hostname, &port, &serial, &error));
}
TEST(socket_spec, parse_tcp_socket_spec_host_and_port) {
std::string hostname, error, serial;
int port;
EXPECT_TRUE(parse_tcp_socket_spec("tcp:localhost:1234", &hostname, &port, &serial, &error));
EXPECT_EQ("localhost", hostname);
EXPECT_EQ(1234, port);
EXPECT_EQ("localhost:1234", serial);
}
TEST(socket_spec, parse_tcp_socket_spec_host_no_port) {
std::string hostname, error, serial;
int port;
EXPECT_TRUE(parse_tcp_socket_spec("tcp:localhost", &hostname, &port, &serial, &error));
EXPECT_EQ("localhost", hostname);
EXPECT_EQ(5555, port);
EXPECT_EQ("localhost:5555", serial);
}
TEST(socket_spec, parse_tcp_socket_spec_host_bad_ports) {
std::string hostname, error, serial;
int port;
EXPECT_FALSE(parse_tcp_socket_spec("tcp:localhost:", &hostname, &port, &serial, &error));
EXPECT_FALSE(parse_tcp_socket_spec("tcp:localhost:-1", &hostname, &port, &serial, &error));
EXPECT_FALSE(parse_tcp_socket_spec("tcp:localhost:65536", &hostname, &port, &serial, &error));
}
TEST(socket_spec, parse_tcp_socket_spec_ipv6_and_port) {
std::string hostname, error, serial;
int port;
EXPECT_TRUE(parse_tcp_socket_spec("tcp:[::1]:1234", &hostname, &port, &serial, &error));
EXPECT_EQ("::1", hostname);
EXPECT_EQ(1234, port);
EXPECT_EQ("[::1]:1234", serial);
}
TEST(socket_spec, parse_tcp_socket_spec_ipv6_no_port) {
std::string hostname, error, serial;
int port;
EXPECT_TRUE(parse_tcp_socket_spec("tcp:::1", &hostname, &port, &serial, &error));
EXPECT_EQ("::1", hostname);
EXPECT_EQ(5555, port);
EXPECT_EQ("[::1]:5555", serial);
}
TEST(socket_spec, parse_tcp_socket_spec_ipv6_bad_ports) {
std::string hostname, error, serial;
int port;
EXPECT_FALSE(parse_tcp_socket_spec("tcp:[::1]", &hostname, &port, &serial, &error));
EXPECT_FALSE(parse_tcp_socket_spec("tcp:[::1]:", &hostname, &port, &serial, &error));
EXPECT_FALSE(parse_tcp_socket_spec("tcp:[::1]:-1", &hostname, &port, &serial, &error));
}
TEST(socket_spec, get_host_socket_spec_port) {
std::string error;
EXPECT_EQ(5555, get_host_socket_spec_port("tcp:5555", &error));
EXPECT_EQ(5555, get_host_socket_spec_port("tcp:localhost:5555", &error));
EXPECT_EQ(5555, get_host_socket_spec_port("tcp:[::1]:5555", &error));
EXPECT_EQ(5555, get_host_socket_spec_port("vsock:5555", &error));
}
TEST(socket_spec, get_host_socket_spec_port_no_port) {
std::string error;
EXPECT_EQ(5555, get_host_socket_spec_port("tcp:localhost", &error));
EXPECT_EQ(-1, get_host_socket_spec_port("vsock:localhost", &error));
}
TEST(socket_spec, get_host_socket_spec_port_bad_ports) {
std::string error;
EXPECT_EQ(-1, get_host_socket_spec_port("tcp:65536", &error));
EXPECT_EQ(-1, get_host_socket_spec_port("tcp:-5", &error));
EXPECT_EQ(-1, get_host_socket_spec_port("vsock:-5", &error));
EXPECT_EQ(-1, get_host_socket_spec_port("vsock:5:5555", &error));
}
TEST(socket_spec, get_host_socket_spec_port_bad_string) {
std::string error;
EXPECT_EQ(-1, get_host_socket_spec_port("tcpz:5555", &error));
EXPECT_EQ(-1, get_host_socket_spec_port("vsockz:5555", &error));
EXPECT_EQ(-1, get_host_socket_spec_port("abcd:5555", &error));
EXPECT_EQ(-1, get_host_socket_spec_port("abcd", &error));
}
TEST(socket_spec, socket_spec_listen_connect_tcp) {
std::string error, serial;
int port;
unique_fd server_fd, client_fd;
EXPECT_FALSE(socket_spec_connect(&client_fd, "tcp:localhost:7777", &port, &serial, &error));
server_fd.reset(socket_spec_listen("tcp:7777", &error, &port));
EXPECT_NE(server_fd.get(), -1);
EXPECT_TRUE(socket_spec_connect(&client_fd, "tcp:localhost:7777", &port, &serial, &error));
EXPECT_NE(client_fd.get(), -1);
}
TEST(socket_spec, socket_spec_connect_failure) {
std::string error, serial;
int port;
unique_fd client_fd;
EXPECT_FALSE(socket_spec_connect(&client_fd, "tcp:", &port, &serial, &error));
EXPECT_FALSE(socket_spec_connect(&client_fd, "acceptfd:", &port, &serial, &error));
EXPECT_FALSE(socket_spec_connect(&client_fd, "vsock:", &port, &serial, &error));
EXPECT_FALSE(socket_spec_connect(&client_fd, "vsock:x", &port, &serial, &error));
EXPECT_FALSE(socket_spec_connect(&client_fd, "vsock:5", &port, &serial, &error));
EXPECT_FALSE(socket_spec_connect(&client_fd, "vsock:5:x", &port, &serial, &error));
EXPECT_FALSE(socket_spec_connect(&client_fd, "sneakernet:", &port, &serial, &error));
}
TEST(socket_spec, socket_spec_listen_connect_localfilesystem) {
std::string error, serial;
int port;
unique_fd server_fd, client_fd;
TemporaryDir sock_dir;
// Only run this test if the created directory is writable.
int result = access(sock_dir.path, W_OK);
if (result == 0) {
std::string sock_addr =
android::base::StringPrintf("localfilesystem:%s/af_unix_socket", sock_dir.path);
EXPECT_FALSE(socket_spec_connect(&client_fd, sock_addr, &port, &serial, &error));
server_fd.reset(socket_spec_listen(sock_addr, &error, &port));
EXPECT_NE(server_fd.get(), -1);
EXPECT_TRUE(socket_spec_connect(&client_fd, sock_addr, &port, &serial, &error));
EXPECT_NE(client_fd.get(), -1);
}
}
TEST(socket_spec, is_socket_spec) {
EXPECT_TRUE(is_socket_spec("tcp:blah"));
EXPECT_TRUE(is_socket_spec("acceptfd:blah"));
EXPECT_TRUE(is_socket_spec("local:blah"));
EXPECT_TRUE(is_socket_spec("localreserved:blah"));
EXPECT_TRUE(is_socket_spec("vsock:123:456"));
}
TEST(socket_spec, is_local_socket_spec) {
EXPECT_TRUE(is_local_socket_spec("local:blah"));
EXPECT_TRUE(is_local_socket_spec("tcp:localhost"));
EXPECT_FALSE(is_local_socket_spec("tcp:www.google.com"));
}