void svr_session(int sock, int childpipe) {
char *host, *port;
size_t len;
common_session_init(sock, sock);
svr_ses.connect_time = monotonic_now();;
/* Initialise server specific parts of the session */
svr_ses.childpipe = childpipe;
#ifdef USE_VFORK
svr_ses.server_pid = getpid();
#endif
svr_authinitialise();
chaninitialise(svr_chantypes);
svr_chansessinitialise();
/* for logging the remote address */
get_socket_address(ses.sock_in, NULL, NULL, &host, &port, 0);
len = strlen(host) + strlen(port) + 2;
svr_ses.addrstring = m_malloc(len);
snprintf(svr_ses.addrstring, len, "%s:%s", host, port);
m_free(host);
m_free(port);
get_socket_address(ses.sock_in, NULL, NULL,
&svr_ses.remotehost, NULL, 1);
/* set up messages etc */
ses.remoteclosed = svr_remoteclosed;
ses.extra_session_cleanup = svr_session_cleanup;
/* packet handlers */
ses.packettypes = svr_packettypes;
ses.isserver = 1;
/* We're ready to go now */
sessinitdone = 1;
/* exchange identification, version etc */
send_session_identification();
kexfirstinitialise(); /* initialise the kex state */
/* start off with key exchange */
send_msg_kexinit();
/* Run the main for loop. NULL is for the dispatcher - only the client
* code makes use of it */
session_loop(svr_sessionloop);
/* Not reached */
}
void
f_socket_write (void)
{
int i, fd, port;
svalue_t *arg;
char addr[ADDR_BUF_SIZE];
int num_arg = st_num_arg;
arg = sp - num_arg + 1;
if ((num_arg == 3) && (arg[2].type != T_STRING)) {
bad_arg(3, F_SOCKET_WRITE);
}
fd = arg[0].u.number;
get_socket_address(fd, addr, &port, 0);
if (VALID_SOCKET("write")) {
i = socket_write(fd, &arg[1],
(num_arg == 3) ? arg[2].u.string : (char *) NULL);
pop_n_elems(num_arg - 1);
sp->u.number = i;
} else {
pop_n_elems(num_arg - 1);
sp->u.number = EESECURITY;
}
}
char *
get_peer_ipaddr(int socket)
{
char *p;
if ((p = get_socket_address(socket, 1, NI_NUMERICHOST)) != NULL)
return p;
return xstrdup("UNKNOWN");
}
char *
get_local_ipaddr(int sock)
{
char *p;
if ((p = get_socket_address(sock, 0, NI_NUMERICHOST)) != NULL)
return p;
return xstrdup("UNKNOWN");
}
void
f_socket_close (void)
{
int fd, port;
char addr[ADDR_BUF_SIZE];
fd = sp->u.number;
get_socket_address(fd, addr, &port, 0);
sp->u.number = VALID_SOCKET("close") ? socket_close(fd, 0) : EESECURITY;
}
void
f_socket_accept (void)
{
int port, fd;
char addr[ADDR_BUF_SIZE];
if (!(sp->type & (T_STRING | T_FUNCTION))) {
bad_arg(3, F_SOCKET_ACCEPT);
}
get_socket_address(fd = (sp-2)->u.number, addr, &port, 0);
(sp-2)->u.number = VALID_SOCKET("accept") ?
socket_accept(fd, (sp - 1), sp) :
EESECURITY;
pop_2_elems();
}
void
f_socket_connect (void)
{
int i, fd, port;
char addr[ADDR_BUF_SIZE];
if (!((sp - 1)->type & (T_FUNCTION | T_STRING))) {
bad_arg(3, F_SOCKET_CONNECT);
}
if (!(sp->type & (T_FUNCTION | T_STRING))) {
bad_arg(4, F_SOCKET_CONNECT);
}
fd = (sp - 3)->u.number;
get_socket_address(fd, addr, &port, 0);
if (!strcmp(addr, "0.0.0.0") && port == 0) {
/*
* socket descriptor is not bound yet
*/
char *s;
int start = 0;
addr[0] = '\0';
if ((s = strchr((sp - 2)->u.string, ' '))) {
/*
* use specified address and port
*/
i = s - (sp - 2)->u.string;
if (i > ADDR_BUF_SIZE - 1) {
start = i - ADDR_BUF_SIZE - 1;
i = ADDR_BUF_SIZE - 1;
}
strncat(addr, (sp - 2)->u.string + start, i);
port = atoi(s + 1);
}
#ifdef DEBUG
} else {
fprintf(stderr, "socket_connect: socket already bound to address/port: %s/%d\n",
addr, port);
fprintf(stderr, "socket_connect: requested on: %s\n", (sp - 2)->u.string);
#endif
}
(sp-3)->u.number = VALID_SOCKET("connect") ?
socket_connect(fd, (sp - 2)->u.string, sp - 1, sp) : EESECURITY;
pop_3_elems();
}
char *
get_local_name(int fd)
{
char *host, myname[NI_MAXHOST];
/* Assume we were passed a socket */
if ((host = get_socket_address(fd, 0, NI_NAMEREQD)) != NULL)
return host;
/* Handle the case where we were passed a pipe */
if (gethostname(myname, sizeof(myname)) == -1) {
verbose("get_local_name: gethostname: %s", strerror(errno));
} else {
host = xstrdup(myname);
}
return host;
}
void
f_socket_listen (void)
{
int i, fd, port;
char addr[ADDR_BUF_SIZE];
fd = (sp - 1)->u.number;
get_socket_address(fd, addr, &port, 0);
if (VALID_SOCKET("listen")) {
i = socket_listen(fd, sp);
pop_stack();
sp->u.number = i;
} else {
pop_stack();
sp->u.number = EESECURITY;
}
}
int socket_connect_client(const char* host, int port) {
int rc = 0;
int socketfd = 0;
struct addrinfo hints;
struct addrinfo* server_info = 0;
memset(&hints, 0, sizeof(hints));
hints.ai_family = AF_UNSPEC;
hints.ai_socktype = SOCK_DGRAM;
char port_string[10];
sprintf(port_string, "%d", port);
rc = getaddrinfo(host, port_string, &hints, &server_info);
check(rc == 0, "Failed to get address info.");
socketfd = socket(server_info->ai_family, server_info->ai_socktype, IPPROTO_UDP);
check(socketfd > 0, "Could not open socket.");
// set socket option
//int sndBuf = 1024;
//rc = setsockopt(socketfd, SOL_SOCKET, SO_SNDBUF, (char*)&sndBuf, sizeof(sndBuf));
//check(rc != -1, "Failed to set socket option SNDBUF.");
rc = connect(socketfd, server_info->ai_addr, server_info->ai_addrlen);
check(rc != -1, "Failed to connect to server.");
char address[INET6_ADDRSTRLEN];
inet_ntop(server_info->ai_family,
get_socket_address(server_info->ai_addr),
address,
sizeof(address));
freeaddrinfo(server_info);
return socketfd;
error:
if (socketfd != 0) {
close(socketfd);
}
freeaddrinfo(server_info);
return -1;
}
void
f_socket_acquire (void)
{
int fd, port;
char addr[ADDR_BUF_SIZE];
if (!((sp - 1)->type & (T_FUNCTION | T_STRING))) {
bad_arg(3, F_SOCKET_ACQUIRE);
}
if (!(sp->type & (T_FUNCTION | T_STRING))) {
bad_arg(4, F_SOCKET_ACQUIRE);
}
fd = (sp - 3)->u.number;
get_socket_address(fd, addr, &port, 0);
(sp-3)->u.number = VALID_SOCKET("acquire") ?
socket_acquire((sp - 3)->u.number, (sp - 2),
(sp - 1), sp) : EESECURITY;
pop_3_elems();
}
void
f_socket_release (void)
{
int fd, port;
char addr[ADDR_BUF_SIZE];
if (!(sp->type & (T_STRING | T_FUNCTION))) {
bad_arg(3, F_SOCKET_RELEASE);
}
fd = (sp - 2)->u.number;
get_socket_address(fd, addr, &port, 0);
(sp-2)->u.number = VALID_SOCKET("release") ?
socket_release((sp - 2)->u.number, (sp - 1)->u.ob, sp) :
EESECURITY;
pop_stack();
/* the object might have been dested an removed from the stack */
if (sp->type == T_OBJECT)
free_object(sp->u.ob, "socket_release()");
sp--;
}
void
f_socket_address (void)
{
char *str;
int local, port;
char addr[ADDR_BUF_SIZE];
char buf[2 * ADDR_BUF_SIZE]; /* a bit of overkill to be safe */
/*
* Ok, we will add in a cute little check thing here to see if it is
* an object or not...
*/
local = (sp--)->u.number;
if (sp->type & T_OBJECT) {
char *tmp;
/* This is so we can get the address of interactives as well. */
if (!sp->u.ob->interactive) {
free_object(sp->u.ob, "f_socket_address:1");
*sp = const0u;
return;
}
tmp = inet_ntoa(sp->u.ob->interactive->addr.sin_addr);
sprintf(buf, "%s %d", tmp,
ntohs(sp->u.ob->interactive->addr.sin_port));
str = string_copy(buf, "f_socket_address");
free_object(sp->u.ob, "f_socket_address:2");
put_malloced_string(str);
return;
}
get_socket_address(sp->u.number, addr, &port, local);
sprintf(buf, "%s %d", addr, port);
str = string_copy(buf, "f_socket_address");
put_malloced_string(str);
} /* f_socket_address() */
void
f_socket_bind (void)
{
int i, fd, port, num_arg = st_num_arg;
svalue_t *arg;
char addr[ADDR_BUF_SIZE];
arg = sp - num_arg + 1;
if ((num_arg == 3) && (arg[2].type != T_STRING)) {
bad_arg(3, F_SOCKET_BIND);
}
fd = arg[0].u.number;
get_socket_address(fd, addr, &port, 0);
if (VALID_SOCKET("bind")) {
i = socket_bind(fd, arg[1].u.number, (num_arg == 3 ? arg[2].u.string : 0));
pop_n_elems(num_arg - 1);
sp->u.number = i;
} else {
pop_n_elems(num_arg - 1);
sp->u.number = EESECURITY;
}
}
char *
get_local_name(int sock)
{
return get_socket_address(sock, 0, NI_NAMEREQD);
}
static struct sockaddr *simple_socket_address(struct socket *gs,
struct error *err)
{
struct simple_socket *s = (struct simple_socket *)gs;
return get_socket_address(s->fd, err);
}
/*
* dbus_new_server
* Set up a D-BUS server, integrate with the event loop
* for handling file descriptor and timed events
*/
int sbus_new_server(TALLOC_CTX *mem_ctx,
struct tevent_context *ev,
const char *address,
uid_t uid, gid_t gid,
bool use_symlink,
struct sbus_connection **_server,
sbus_server_conn_init_fn init_fn,
void *init_pvt_data)
{
struct sbus_connection *server;
DBusServer *dbus_server;
DBusError dbus_error;
dbus_bool_t dbret;
char *tmp;
int ret, tmp_ret;
char *filename;
char *symlink_filename = NULL;
const char *socket_address;
struct stat stat_buf;
TALLOC_CTX *tmp_ctx;
*_server = NULL;
tmp_ctx = talloc_new(NULL);
if (!tmp_ctx) return ENOMEM;
socket_address = get_socket_address(tmp_ctx, address, use_symlink);
if (!socket_address) {
ret = ENOMEM;
goto done;
}
/* Set up D-BUS server */
dbus_error_init(&dbus_error);
dbus_server = dbus_server_listen(socket_address, &dbus_error);
if (!dbus_server) {
DEBUG(SSSDBG_CRIT_FAILURE,
"dbus_server_listen failed! (name=%s, message=%s)\n",
dbus_error.name, dbus_error.message);
if (dbus_error_is_set(&dbus_error)) dbus_error_free(&dbus_error);
ret = EIO;
goto done;
}
filename = strchr(socket_address, '/');
if (filename == NULL) {
DEBUG(SSSDBG_CRIT_FAILURE,
"Unexpected dbus address [%s].\n", socket_address);
ret = EIO;
goto done;
}
if (use_symlink) {
symlink_filename = strchr(address, '/');
if (symlink_filename == NULL) {
DEBUG(SSSDBG_CRIT_FAILURE,
"Unexpected dbus address [%s].\n", address);
ret = EIO;
goto done;
}
ret = create_socket_symlink(filename, symlink_filename);
if (ret != EOK) {
DEBUG(SSSDBG_CRIT_FAILURE, "Could not create symlink [%d]: %s\n",
ret, strerror(ret));
ret = EIO;
goto done;
}
}
/* Both check_file and chmod can handle both the symlink and
* the socket */
ret = check_file(filename,
getuid(), getgid(), S_IFSOCK, S_IFMT, &stat_buf, true);
if (ret != EOK) {
DEBUG(SSSDBG_CRIT_FAILURE, "check_file failed for [%s].\n", filename);
ret = EIO;
goto done;
}
if ((stat_buf.st_mode & ~S_IFMT) != (S_IRUSR|S_IWUSR)) {
ret = chmod(filename, (S_IRUSR|S_IWUSR));
if (ret != EOK) {
ret = errno;
DEBUG(SSSDBG_CRIT_FAILURE,
"chmod failed for [%s]: [%d][%s].\n", filename, ret,
sss_strerror(ret));
ret = EIO;
goto done;
}
}
if (stat_buf.st_uid != uid || stat_buf.st_gid != gid) {
ret = chown(filename, uid, gid);
if (ret != EOK) {
ret = errno;
DEBUG(SSSDBG_CRIT_FAILURE,
"chown failed for [%s]: [%d][%s].\n", filename, ret,
sss_strerror(ret));
ret = EIO;
goto done;
}
}
tmp = dbus_server_get_address(dbus_server);
DEBUG(SSSDBG_TRACE_FUNC, "D-BUS Server listening on %s\n", tmp);
free(tmp);
server = talloc_zero(tmp_ctx, struct sbus_connection);
if (!server) {
ret = ENOMEM;
goto done;
}
server->ev = ev;
server->type = SBUS_SERVER;
server->dbus.server = dbus_server;
server->srv_init_fn = init_fn;
server->srv_init_data = init_pvt_data;
talloc_set_destructor((TALLOC_CTX *)server, sbus_server_destructor);
if (use_symlink) {
server->symlink = talloc_strdup(server, symlink_filename);
if (!server->symlink) {
ret = ENOMEM;
goto done;
}
}
/* Set up D-BUS new connection handler */
dbus_server_set_new_connection_function(server->dbus.server,
sbus_server_init_new_connection,
server, NULL);
/* Set up DBusWatch functions */
dbret = dbus_server_set_watch_functions(server->dbus.server,
sbus_add_watch,
sbus_remove_watch,
sbus_toggle_watch,
server, NULL);
if (!dbret) {
DEBUG(SSSDBG_CONF_SETTINGS,
"Error setting up D-BUS server watch functions\n");
ret = EIO;
goto done;
}
/* Set up DBusTimeout functions */
dbret = dbus_server_set_timeout_functions(server->dbus.server,
sbus_add_timeout,
sbus_remove_timeout,
sbus_toggle_timeout,
server, NULL);
if (!dbret) {
DEBUG(SSSDBG_CONF_SETTINGS,
"Error setting up D-BUS server timeout functions\n");
dbus_server_set_watch_functions(server->dbus.server,
NULL, NULL, NULL, NULL, NULL);
ret = EIO;
goto done;
}
*_server = talloc_steal(mem_ctx, server);
ret = EOK;
done:
if (ret != EOK && symlink_filename) {
tmp_ret = unlink(symlink_filename);
/* non-fatal failure */
if (tmp_ret != EOK) {
tmp_ret = errno;
DEBUG(SSSDBG_MINOR_FAILURE,
"Failed to remove symbolic link '%s': %d [%s]!\n",
symlink_filename, tmp_ret, sss_strerror(tmp_ret));
}
}
talloc_free(tmp_ctx);
return ret;
}
