diff options
Diffstat (limited to 'tools/obex-server-tool.c')
| -rw-r--r-- | tools/obex-server-tool.c | 458 |
1 files changed, 458 insertions, 0 deletions
diff --git a/tools/obex-server-tool.c b/tools/obex-server-tool.c new file mode 100644 index 00000000..e37c56fd --- /dev/null +++ b/tools/obex-server-tool.c @@ -0,0 +1,458 @@ +/* + * + * OBEX library with GLib integration + * + * Copyright (C) 2011 Intel Corporation. All rights reserved. + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License version 2 as + * published by the Free Software Foundation. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA + * + */ + +#ifdef HAVE_CONFIG_H +# include "config.h" +#endif + +#include <sys/types.h> +#include <sys/socket.h> +#include <fcntl.h> +#include <sys/un.h> +#include <unistd.h> +#include <stdlib.h> +#include <stdio.h> +#include <string.h> +#include <errno.h> + +#include <gobex/gobex.h> +#include <btio/btio.h> + +static GMainLoop *main_loop = NULL; + +static GSList *clients = NULL; + +static gboolean option_packet = FALSE; +static gboolean option_bluetooth = FALSE; +static int option_channel = -1; +static int option_imtu = -1; +static int option_omtu = -1; +static char *option_root = NULL; + +static void sig_term(int sig) +{ + g_print("Terminating due to signal %d\n", sig); + g_main_loop_quit(main_loop); +} + +static GOptionEntry options[] = { + { "unix", 'u', G_OPTION_FLAG_REVERSE, G_OPTION_ARG_NONE, + &option_bluetooth, "Use a UNIX socket" }, + { "bluetooth", 'b', 0, G_OPTION_ARG_NONE, + &option_bluetooth, "Use Bluetooth" }, + { "channel", 'c', 0, G_OPTION_ARG_INT, + &option_channel, "Transport channel", "CHANNEL" }, + { "packet", 'p', 0, G_OPTION_ARG_NONE, + &option_packet, "Packet based transport" }, + { "stream", 's', G_OPTION_FLAG_REVERSE, G_OPTION_ARG_NONE, + &option_packet, "Stream based transport" }, + { "root", 'r', 0, G_OPTION_ARG_STRING, + &option_root, "Root dir", "/..." }, + { "input-mtu", 'i', 0, G_OPTION_ARG_INT, + &option_imtu, "Transport input MTU", "MTU" }, + { "output-mtu", 'o', 0, G_OPTION_ARG_INT, + &option_omtu, "Transport output MTU", "MTU" }, + { NULL }, +}; + +static void disconn_func(GObex *obex, GError *err, gpointer user_data) +{ + g_print("Client disconnected: %s\n", err ? err->message : "<no err>"); + clients = g_slist_remove(clients, obex); + g_obex_unref(obex); +} + +struct transfer_data { + int fd; +}; + +static void transfer_complete(GObex *obex, GError *err, gpointer user_data) +{ + struct transfer_data *data = user_data; + + if (err != NULL) + g_printerr("transfer failed: %s\n", err->message); + else + g_print("transfer succeeded\n"); + + close(data->fd); + g_free(data); +} + +static gboolean recv_data(const void *buf, gsize len, gpointer user_data) +{ + struct transfer_data *data = user_data; + + g_print("received %zu bytes of data\n", len); + + if (write(data->fd, buf, len) < 0) { + g_printerr("write: %s\n", strerror(errno)); + return FALSE; + } + + return TRUE; +} + +static void handle_put(GObex *obex, GObexPacket *req, gpointer user_data) +{ + GError *err = NULL; + GObexHeader *hdr; + const char *type, *name; + struct transfer_data *data; + gsize type_len; + + hdr = g_obex_packet_get_header(req, G_OBEX_HDR_TYPE); + if (hdr != NULL) { + g_obex_header_get_bytes(hdr, (const guint8 **) &type, + &type_len); + if (type[type_len - 1] != '\0') { + g_printerr("non-nul terminated type header\n"); + type = NULL; + } + } else + type = NULL; + + hdr = g_obex_packet_get_header(req, G_OBEX_HDR_NAME); + if (hdr != NULL) + g_obex_header_get_unicode(hdr, &name); + else + name = NULL; + + g_print("put type \"%s\" name \"%s\"\n", type ? type : "", + name ? name : ""); + + data = g_new0(struct transfer_data, 1); + + data->fd = open(name, O_WRONLY | O_CREAT | O_NOCTTY, 0600); + if (data->fd < 0) { + g_printerr("open(%s): %s\n", name, strerror(errno)); + g_free(data); + g_obex_send_rsp(obex, G_OBEX_RSP_FORBIDDEN, NULL, + G_OBEX_HDR_INVALID); + return; + } + + g_obex_put_rsp(obex, req, recv_data, transfer_complete, data, &err, + G_OBEX_HDR_INVALID); + if (err != NULL) { + g_printerr("Unable to send response: %s\n", err->message); + g_error_free(err); + g_free(data); + } +} + +static gssize send_data(void *buf, gsize len, gpointer user_data) +{ + struct transfer_data *data = user_data; + gssize ret; + + ret = read(data->fd, buf, len); + g_print("sending %zu bytes of data\n", ret); + + return ret; +} + +static void handle_get(GObex *obex, GObexPacket *req, gpointer user_data) +{ + GError *err = NULL; + struct transfer_data *data; + const char *type, *name; + GObexHeader *hdr; + gsize type_len; + + hdr = g_obex_packet_get_header(req, G_OBEX_HDR_TYPE); + if (hdr != NULL) { + g_obex_header_get_bytes(hdr, (const guint8 **) &type, + &type_len); + if (type[type_len - 1] != '\0') { + g_printerr("non-nul terminated type header\n"); + type = NULL; + } + } else + type = NULL; + + hdr = g_obex_packet_get_header(req, G_OBEX_HDR_NAME); + if (hdr != NULL) + g_obex_header_get_unicode(hdr, &name); + else + name = NULL; + + g_print("get type \"%s\" name \"%s\"\n", type ? type : "", + name ? name : ""); + + data = g_new0(struct transfer_data, 1); + + data->fd = open(name, O_RDONLY | O_NOCTTY, 0); + if (data->fd < 0) { + g_printerr("open(%s): %s\n", name, strerror(errno)); + g_free(data); + g_obex_send_rsp(obex, G_OBEX_RSP_FORBIDDEN, NULL, + G_OBEX_HDR_INVALID); + return; + } + + g_obex_get_rsp(obex, send_data, transfer_complete, data, &err, + G_OBEX_HDR_INVALID); + if (err != NULL) { + g_printerr("Unable to send response: %s\n", err->message); + g_error_free(err); + g_free(data); + } +} + +static void handle_connect(GObex *obex, GObexPacket *req, gpointer user_data) +{ + GObexPacket *rsp; + + g_print("connect\n"); + + rsp = g_obex_packet_new(G_OBEX_RSP_SUCCESS, TRUE, G_OBEX_HDR_INVALID); + g_obex_send(obex, rsp, NULL); +} + +static void transport_accept(GIOChannel *io) +{ + GObex *obex; + GObexTransportType transport; + + g_io_channel_set_flags(io, G_IO_FLAG_NONBLOCK, NULL); + g_io_channel_set_close_on_unref(io, TRUE); + + if (option_packet) + transport = G_OBEX_TRANSPORT_PACKET; + else + transport = G_OBEX_TRANSPORT_STREAM; + + obex = g_obex_new(io, transport, option_imtu, option_omtu); + g_obex_set_disconnect_function(obex, disconn_func, NULL); + g_obex_add_request_function(obex, G_OBEX_OP_PUT, handle_put, NULL); + g_obex_add_request_function(obex, G_OBEX_OP_GET, handle_get, NULL); + g_obex_add_request_function(obex, G_OBEX_OP_CONNECT, handle_connect, + NULL); + clients = g_slist_append(clients, obex); +} + +static gboolean unix_accept(GIOChannel *chan, GIOCondition cond, gpointer data) +{ + struct sockaddr_un addr; + socklen_t addrlen; + int sk, cli_sk; + GIOChannel *io; + + if (cond & G_IO_NVAL) + return FALSE; + + if (cond & (G_IO_HUP | G_IO_ERR)) { + g_io_channel_shutdown(chan, TRUE, NULL); + return FALSE; + } + + sk = g_io_channel_unix_get_fd(chan); + + memset(&addr, 0, sizeof(addr)); + addrlen = sizeof(addr); + + cli_sk = accept(sk, (struct sockaddr *) &addr, &addrlen); + if (cli_sk < 0) { + g_printerr("accept: %s (%d)\n", strerror(errno), errno); + return TRUE; + } + + g_print("Accepted new client connection on unix socket (fd=%d)\n", + cli_sk); + + io = g_io_channel_unix_new(cli_sk); + + transport_accept(io); + g_io_channel_unref(io); + + return TRUE; +} + +static void bluetooth_accept(GIOChannel *io, GError *err, gpointer data) +{ + if (err) { + g_printerr("accept: %s\n", err->message); + return; + } + + g_print("Accepted new client connection on bluetooth socket\n"); + + transport_accept(io); +} + +static gboolean bluetooth_watch(GIOChannel *chan, GIOCondition cond, gpointer data) +{ + if (cond & G_IO_NVAL) + return FALSE; + + g_io_channel_shutdown(chan, TRUE, NULL); + return FALSE; +} + +static GIOChannel *l2cap_listen(GError **err) +{ + return bt_io_listen(bluetooth_accept, NULL, NULL, + NULL, err, + BT_IO_OPT_PSM, option_channel, + BT_IO_OPT_MODE, BT_IO_MODE_ERTM, + BT_IO_OPT_OMTU, option_omtu, + BT_IO_OPT_IMTU, option_imtu, + BT_IO_OPT_INVALID); +} + +static GIOChannel *rfcomm_listen(GError **err) +{ + return bt_io_listen(bluetooth_accept, NULL, NULL, + NULL, err, + BT_IO_OPT_CHANNEL, option_channel, + BT_IO_OPT_INVALID); +} + +static guint bluetooth_listen(void) +{ + GIOChannel *io; + guint id; + GError *err = NULL; + + if (option_channel == -1) { + g_printerr("Bluetooth channel not set\n"); + return 0; + } + + if (option_packet || option_channel > 31) + io = l2cap_listen(&err); + else + io = rfcomm_listen(&err); + + if (io == NULL) { + g_printerr("%s\n", err->message); + g_error_free(err); + return 0; + } + + g_print("Bluetooth socket created\n"); + + id = g_io_add_watch(io, G_IO_HUP | G_IO_ERR | G_IO_NVAL, + bluetooth_watch, NULL); + + g_io_channel_set_flags(io, G_IO_FLAG_NONBLOCK, NULL); + g_io_channel_set_close_on_unref(io, TRUE); + g_io_channel_unref(io); + + return id; +} + +static guint unix_listen(void) +{ + GIOChannel *io; + struct sockaddr_un addr = { + AF_UNIX, "\0/gobex/server" + }; + int sk, err, sock_type; + guint id; + + if (option_packet) + sock_type = SOCK_SEQPACKET; + else + sock_type = SOCK_STREAM; + + sk = socket(PF_LOCAL, sock_type, 0); + if (sk < 0) { + err = errno; + g_printerr("Can't create unix socket: %s (%d)\n", + strerror(err), err); + return 0; + } + + if (bind(sk, (struct sockaddr *) &addr, sizeof(addr)) < 0) { + g_printerr("Can't bind unix socket: %s (%d)\n", + strerror(errno), errno); + close(sk); + return 0; + } + + if (listen(sk, 1) < 0) { + g_printerr("Can't listen on unix socket: %s (%d)\n", + strerror(errno), errno); + close(sk); + return 0; + } + + g_print("Unix socket created: %d\n", sk); + + io = g_io_channel_unix_new(sk); + id = g_io_add_watch(io, G_IO_IN | G_IO_HUP | G_IO_ERR | G_IO_NVAL, + unix_accept, NULL); + + g_io_channel_set_flags(io, G_IO_FLAG_NONBLOCK, NULL); + g_io_channel_set_close_on_unref(io, TRUE); + g_io_channel_unref(io); + + return id; +} + +int main(int argc, char *argv[]) +{ + GOptionContext *context; + GError *err = NULL; + struct sigaction sa; + guint server_id; + + context = g_option_context_new(NULL); + g_option_context_add_main_entries(context, options, NULL); + + g_option_context_parse(context, &argc, &argv, &err); + if (err != NULL) { + g_printerr("%s\n", err->message); + g_error_free(err); + exit(EXIT_FAILURE); + } + + if (option_root && chdir(option_root) > 0) { + perror("chdir:"); + exit(EXIT_FAILURE); + } + + if (option_bluetooth) + server_id = bluetooth_listen(); + else + server_id = unix_listen(); + + if (server_id == 0) + exit(EXIT_FAILURE); + + memset(&sa, 0, sizeof(sa)); + sa.sa_handler = sig_term; + sigaction(SIGINT, &sa, NULL); + sigaction(SIGTERM, &sa, NULL); + + main_loop = g_main_loop_new(NULL, FALSE); + + g_main_loop_run(main_loop); + + g_source_remove(server_id); + g_slist_free_full(clients, (GDestroyNotify) g_obex_unref); + g_option_context_free(context); + g_main_loop_unref(main_loop); + + exit(EXIT_SUCCESS); +} |