blob: f9841b8655e22a6d5c011fffa98a9bb8c183efe8 [file] [log] [blame] [edit]
// Copyright (c) 2021, the Dart project authors. Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.
// This is a utility program for testing that a Dart program can connect to an
// abstract UNIX socket created by a non-Dart program. It creates such a socket
// accepts one connection, echoes back the first message it receives, and then
// closes the connection and UNIX socket.
#include "platform/globals.h"
#if defined(DART_HOST_OS_LINUX) || defined(DART_HOST_OS_ANDROID)
#include <stdio.h>
#include <stdlib.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <unistd.h>
const intptr_t kOffsetOfPtr = 32;
#define OFFSET_OF(type, field) \
(reinterpret_cast<intptr_t>( \
&(reinterpret_cast<type*>(kOffsetOfPtr)->field)) - \
kOffsetOfPtr) // NOLINT
int main(int argc, char* argv[]) {
struct sockaddr_un addr;
char* socket_path;
int server_socket;
char buf[1024];
if (argc < 2) {
fprintf(
stderr,
"Usage: abstract_socket_test <address>\n\n"
"<address> should be an abstract UNIX socket address like @hidden\n");
exit(-1);
}
socket_path = argv[1];
if (socket_path[0] != '@') {
fprintf(stderr,
"The first argument should be an abstract socket "
"address and start with '@'\n");
exit(-1);
}
if ((server_socket = socket(AF_UNIX, SOCK_STREAM, 0)) == -1) {
perror("socket error");
exit(-1);
}
memset(&addr, 0, sizeof(addr));
addr.sun_family = AF_UNIX;
addr.sun_path[0] = '\0';
strncpy(addr.sun_path + 1, socket_path + 1, sizeof(addr.sun_path) - 2);
int address_length =
OFFSET_OF(struct sockaddr_un, sun_path) + strlen(socket_path);
if (bind(server_socket, (struct sockaddr*)&addr, address_length) == -1) {
perror("bind error");
exit(-1);
}
if (listen(server_socket, 5) == -1) {
perror("listen error");
exit(-1);
}
int client_socket;
if ((client_socket = accept(server_socket, nullptr, nullptr)) == -1) {
perror("accept error");
exit(-1);
}
int read_count;
while ((read_count = read(client_socket, buf, sizeof(buf))) > 0) {
int write_count = 0;
while (write_count < read_count) {
int w;
if ((w = write(client_socket, buf, read_count)) < 0) {
perror("write");
exit(-1);
}
write_count += w;
}
}
if (read_count == -1) {
perror("read");
exit(-1);
}
close(client_socket);
close(server_socket);
return 0;
}
#else
int main() {
return -1;
}
#endif // defined(DART_HOST_OS_LINUX) || defined(DART_HOST_OS_ANDROID)