|
Packit Service |
8264ee |
/*
|
|
Packit Service |
8264ee |
*
|
|
Packit Service |
8264ee |
* BlueZ - Bluetooth protocol stack for Linux
|
|
Packit Service |
8264ee |
*
|
|
Packit Service |
8264ee |
* Copyright (C) 2012-2014 Intel Corporation. All rights reserved.
|
|
Packit Service |
8264ee |
*
|
|
Packit Service |
8264ee |
*
|
|
Packit Service |
8264ee |
* This library is free software; you can redistribute it and/or
|
|
Packit Service |
8264ee |
* modify it under the terms of the GNU Lesser General Public
|
|
Packit Service |
8264ee |
* License as published by the Free Software Foundation; either
|
|
Packit Service |
8264ee |
* version 2.1 of the License, or (at your option) any later version.
|
|
Packit Service |
8264ee |
*
|
|
Packit Service |
8264ee |
* This library is distributed in the hope that it will be useful,
|
|
Packit Service |
8264ee |
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
Packit Service |
8264ee |
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
Packit Service |
8264ee |
* Lesser General Public License for more details.
|
|
Packit Service |
8264ee |
*
|
|
Packit Service |
8264ee |
* You should have received a copy of the GNU Lesser General Public
|
|
Packit Service |
8264ee |
* License along with this library; if not, write to the Free Software
|
|
Packit Service |
8264ee |
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
|
|
Packit Service |
8264ee |
*
|
|
Packit Service |
8264ee |
*/
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
#ifdef HAVE_CONFIG_H
|
|
Packit Service |
8264ee |
#include <config.h>
|
|
Packit Service |
8264ee |
#endif
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
#include <unistd.h>
|
|
Packit Service |
8264ee |
#include <errno.h>
|
|
Packit Service |
8264ee |
#include <sys/socket.h>
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
#include "src/shared/mainloop.h"
|
|
Packit Service |
8264ee |
#include "src/shared/util.h"
|
|
Packit Service |
8264ee |
#include "src/shared/io.h"
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
struct io {
|
|
Packit Service |
8264ee |
int ref_count;
|
|
Packit Service |
8264ee |
int fd;
|
|
Packit Service |
8264ee |
uint32_t events;
|
|
Packit Service |
8264ee |
bool close_on_destroy;
|
|
Packit Service |
8264ee |
io_callback_func_t read_callback;
|
|
Packit Service |
8264ee |
io_destroy_func_t read_destroy;
|
|
Packit Service |
8264ee |
void *read_data;
|
|
Packit Service |
8264ee |
io_callback_func_t write_callback;
|
|
Packit Service |
8264ee |
io_destroy_func_t write_destroy;
|
|
Packit Service |
8264ee |
void *write_data;
|
|
Packit Service |
8264ee |
io_callback_func_t disconnect_callback;
|
|
Packit Service |
8264ee |
io_destroy_func_t disconnect_destroy;
|
|
Packit Service |
8264ee |
void *disconnect_data;
|
|
Packit Service |
8264ee |
};
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
static struct io *io_ref(struct io *io)
|
|
Packit Service |
8264ee |
{
|
|
Packit Service |
8264ee |
if (!io)
|
|
Packit Service |
8264ee |
return NULL;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
__sync_fetch_and_add(&io->ref_count, 1);
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
return io;
|
|
Packit Service |
8264ee |
}
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
static void io_unref(struct io *io)
|
|
Packit Service |
8264ee |
{
|
|
Packit Service |
8264ee |
if (!io)
|
|
Packit Service |
8264ee |
return;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
if (__sync_sub_and_fetch(&io->ref_count, 1))
|
|
Packit Service |
8264ee |
return;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
free(io);
|
|
Packit Service |
8264ee |
}
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
static void io_cleanup(void *user_data)
|
|
Packit Service |
8264ee |
{
|
|
Packit Service |
8264ee |
struct io *io = user_data;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
if (io->write_destroy)
|
|
Packit Service |
8264ee |
io->write_destroy(io->write_data);
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
if (io->read_destroy)
|
|
Packit Service |
8264ee |
io->read_destroy(io->read_data);
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
if (io->disconnect_destroy)
|
|
Packit Service |
8264ee |
io->disconnect_destroy(io->disconnect_data);
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
if (io->close_on_destroy)
|
|
Packit Service |
8264ee |
close(io->fd);
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
io->fd = -1;
|
|
Packit Service |
8264ee |
}
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
static void io_callback(int fd, uint32_t events, void *user_data)
|
|
Packit Service |
8264ee |
{
|
|
Packit Service |
8264ee |
struct io *io = user_data;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
io_ref(io);
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
if ((events & (EPOLLRDHUP | EPOLLHUP | EPOLLERR))) {
|
|
Packit Service |
8264ee |
io->read_callback = NULL;
|
|
Packit Service |
8264ee |
io->write_callback = NULL;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
if (!io->disconnect_callback) {
|
|
Packit Service |
8264ee |
mainloop_remove_fd(io->fd);
|
|
Packit Service |
8264ee |
io_unref(io);
|
|
Packit Service |
8264ee |
return;
|
|
Packit Service |
8264ee |
}
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
if (!io->disconnect_callback(io, io->disconnect_data)) {
|
|
Packit Service |
8264ee |
if (io->disconnect_destroy)
|
|
Packit Service |
8264ee |
io->disconnect_destroy(io->disconnect_data);
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
io->disconnect_callback = NULL;
|
|
Packit Service |
8264ee |
io->disconnect_destroy = NULL;
|
|
Packit Service |
8264ee |
io->disconnect_data = NULL;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
io->events &= ~EPOLLRDHUP;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
mainloop_modify_fd(io->fd, io->events);
|
|
Packit Service |
8264ee |
}
|
|
Packit Service |
8264ee |
}
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
if ((events & EPOLLIN) && io->read_callback) {
|
|
Packit Service |
8264ee |
if (!io->read_callback(io, io->read_data)) {
|
|
Packit Service |
8264ee |
if (io->read_destroy)
|
|
Packit Service |
8264ee |
io->read_destroy(io->read_data);
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
io->read_callback = NULL;
|
|
Packit Service |
8264ee |
io->read_destroy = NULL;
|
|
Packit Service |
8264ee |
io->read_data = NULL;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
io->events &= ~EPOLLIN;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
mainloop_modify_fd(io->fd, io->events);
|
|
Packit Service |
8264ee |
}
|
|
Packit Service |
8264ee |
}
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
if ((events & EPOLLOUT) && io->write_callback) {
|
|
Packit Service |
8264ee |
if (!io->write_callback(io, io->write_data)) {
|
|
Packit Service |
8264ee |
if (io->write_destroy)
|
|
Packit Service |
8264ee |
io->write_destroy(io->write_data);
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
io->write_callback = NULL;
|
|
Packit Service |
8264ee |
io->write_destroy = NULL;
|
|
Packit Service |
8264ee |
io->write_data = NULL;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
io->events &= ~EPOLLOUT;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
mainloop_modify_fd(io->fd, io->events);
|
|
Packit Service |
8264ee |
}
|
|
Packit Service |
8264ee |
}
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
io_unref(io);
|
|
Packit Service |
8264ee |
}
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
struct io *io_new(int fd)
|
|
Packit Service |
8264ee |
{
|
|
Packit Service |
8264ee |
struct io *io;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
if (fd < 0)
|
|
Packit Service |
8264ee |
return NULL;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
io = new0(struct io, 1);
|
|
Packit Service |
8264ee |
io->fd = fd;
|
|
Packit Service |
8264ee |
io->events = 0;
|
|
Packit Service |
8264ee |
io->close_on_destroy = false;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
if (mainloop_add_fd(io->fd, io->events, io_callback,
|
|
Packit Service |
8264ee |
io, io_cleanup) < 0) {
|
|
Packit Service |
8264ee |
free(io);
|
|
Packit Service |
8264ee |
return NULL;
|
|
Packit Service |
8264ee |
}
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
return io_ref(io);
|
|
Packit Service |
8264ee |
}
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
void io_destroy(struct io *io)
|
|
Packit Service |
8264ee |
{
|
|
Packit Service |
8264ee |
if (!io)
|
|
Packit Service |
8264ee |
return;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
io->read_callback = NULL;
|
|
Packit Service |
8264ee |
io->write_callback = NULL;
|
|
Packit Service |
8264ee |
io->disconnect_callback = NULL;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
mainloop_remove_fd(io->fd);
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
io_unref(io);
|
|
Packit Service |
8264ee |
}
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
int io_get_fd(struct io *io)
|
|
Packit Service |
8264ee |
{
|
|
Packit Service |
8264ee |
if (!io)
|
|
Packit Service |
8264ee |
return -ENOTCONN;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
return io->fd;
|
|
Packit Service |
8264ee |
}
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
bool io_set_close_on_destroy(struct io *io, bool do_close)
|
|
Packit Service |
8264ee |
{
|
|
Packit Service |
8264ee |
if (!io)
|
|
Packit Service |
8264ee |
return false;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
io->close_on_destroy = do_close;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
return true;
|
|
Packit Service |
8264ee |
}
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
bool io_set_read_handler(struct io *io, io_callback_func_t callback,
|
|
Packit Service |
8264ee |
void *user_data, io_destroy_func_t destroy)
|
|
Packit Service |
8264ee |
{
|
|
Packit Service |
8264ee |
uint32_t events;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
if (!io || io->fd < 0)
|
|
Packit Service |
8264ee |
return false;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
if (io->read_destroy)
|
|
Packit Service |
8264ee |
io->read_destroy(io->read_data);
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
if (callback)
|
|
Packit Service |
8264ee |
events = io->events | EPOLLIN;
|
|
Packit Service |
8264ee |
else
|
|
Packit Service |
8264ee |
events = io->events & ~EPOLLIN;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
io->read_callback = callback;
|
|
Packit Service |
8264ee |
io->read_destroy = destroy;
|
|
Packit Service |
8264ee |
io->read_data = user_data;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
if (events == io->events)
|
|
Packit Service |
8264ee |
return true;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
if (mainloop_modify_fd(io->fd, events) < 0)
|
|
Packit Service |
8264ee |
return false;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
io->events = events;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
return true;
|
|
Packit Service |
8264ee |
}
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
bool io_set_write_handler(struct io *io, io_callback_func_t callback,
|
|
Packit Service |
8264ee |
void *user_data, io_destroy_func_t destroy)
|
|
Packit Service |
8264ee |
{
|
|
Packit Service |
8264ee |
uint32_t events;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
if (!io || io->fd < 0)
|
|
Packit Service |
8264ee |
return false;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
if (io->write_destroy)
|
|
Packit Service |
8264ee |
io->write_destroy(io->write_data);
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
if (callback)
|
|
Packit Service |
8264ee |
events = io->events | EPOLLOUT;
|
|
Packit Service |
8264ee |
else
|
|
Packit Service |
8264ee |
events = io->events & ~EPOLLOUT;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
io->write_callback = callback;
|
|
Packit Service |
8264ee |
io->write_destroy = destroy;
|
|
Packit Service |
8264ee |
io->write_data = user_data;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
if (events == io->events)
|
|
Packit Service |
8264ee |
return true;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
if (mainloop_modify_fd(io->fd, events) < 0)
|
|
Packit Service |
8264ee |
return false;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
io->events = events;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
return true;
|
|
Packit Service |
8264ee |
}
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
bool io_set_disconnect_handler(struct io *io, io_callback_func_t callback,
|
|
Packit Service |
8264ee |
void *user_data, io_destroy_func_t destroy)
|
|
Packit Service |
8264ee |
{
|
|
Packit Service |
8264ee |
uint32_t events;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
if (!io || io->fd < 0)
|
|
Packit Service |
8264ee |
return false;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
if (io->disconnect_destroy)
|
|
Packit Service |
8264ee |
io->disconnect_destroy(io->disconnect_data);
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
if (callback)
|
|
Packit Service |
8264ee |
events = io->events | EPOLLRDHUP;
|
|
Packit Service |
8264ee |
else
|
|
Packit Service |
8264ee |
events = io->events & ~EPOLLRDHUP;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
io->disconnect_callback = callback;
|
|
Packit Service |
8264ee |
io->disconnect_destroy = destroy;
|
|
Packit Service |
8264ee |
io->disconnect_data = user_data;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
if (events == io->events)
|
|
Packit Service |
8264ee |
return true;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
if (mainloop_modify_fd(io->fd, events) < 0)
|
|
Packit Service |
8264ee |
return false;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
io->events = events;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
return true;
|
|
Packit Service |
8264ee |
}
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
ssize_t io_send(struct io *io, const struct iovec *iov, int iovcnt)
|
|
Packit Service |
8264ee |
{
|
|
Packit Service |
8264ee |
ssize_t ret;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
if (!io || io->fd < 0)
|
|
Packit Service |
8264ee |
return -ENOTCONN;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
do {
|
|
Packit Service |
8264ee |
ret = writev(io->fd, iov, iovcnt);
|
|
Packit Service |
8264ee |
} while (ret < 0 && errno == EINTR);
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
if (ret < 0)
|
|
Packit Service |
8264ee |
return -errno;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
return ret;
|
|
Packit Service |
8264ee |
}
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
bool io_shutdown(struct io *io)
|
|
Packit Service |
8264ee |
{
|
|
Packit Service |
8264ee |
if (!io || io->fd < 0)
|
|
Packit Service |
8264ee |
return false;
|
|
Packit Service |
8264ee |
|
|
Packit Service |
8264ee |
return shutdown(io->fd, SHUT_RDWR) == 0;
|
|
Packit Service |
8264ee |
}
|