code.delx.au - pulseaudio/blob - polyp/socket-server.c

   1 /* $Id$ */
   2
   3 /***
   4   This file is part of polypaudio.
   5
   6   polypaudio is free software; you can redistribute it and/or modify
   7   it under the terms of the GNU General Public License as published
   8   by the Free Software Foundation; either version 2 of the License,
   9   or (at your option) any later version.
  10
  11   polypaudio is distributed in the hope that it will be useful, but
  12   WITHOUT ANY WARRANTY; without even the implied warranty of
  13   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  14   General Public License for more details.
  15
  16   You should have received a copy of the GNU General Public License
  17   along with polypaudio; if not, write to the Free Software
  18   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  19   USA.
  20 ***/
  21
  22 #ifdef HAVE_CONFIG_H
  23 #include <config.h>
  24 #endif
  25
  26 #include <stdlib.h>
  27 #include <assert.h>
  28 #include <errno.h>
  29 #include <string.h>
  30 #include <sys/types.h>
  31 #include <sys/socket.h>
  32 #include <stdio.h>
  33 #include <unistd.h>
  34 #include <sys/un.h>
  35 #include <netinet/in.h>
  36 #include <arpa/inet.h>
  37
  38 #include "socket-server.h"
  39 #include "socket-util.h"
  40
  41 struct pa_socket_server {
  42     int fd;
  43     char *filename;
  44
  45     void (*on_connection)(struct pa_socket_server*s, struct pa_iochannel *io, void *userdata);
  46     void *userdata;
  47
  48     void *mainloop_source;
  49     struct pa_mainloop_api *mainloop;
  50     enum { SOCKET_SERVER_GENERIC, SOCKET_SERVER_IPV4, SOCKET_SERVER_UNIX } type;
  51 };
  52
  53 static void callback(struct pa_mainloop_api *mainloop, void *id, int fd, enum pa_mainloop_api_io_events events, void *userdata) {
  54     struct pa_socket_server *s = userdata;
  55     struct pa_iochannel *io;
  56     int nfd;
  57     assert(s && s->mainloop == mainloop && s->mainloop_source == id && id && fd >= 0 && fd == s->fd && events == PA_MAINLOOP_API_IO_EVENT_INPUT);
  58
  59     if ((nfd = accept(fd, NULL, NULL)) < 0) {
  60         fprintf(stderr, "accept(): %s\n", strerror(errno));
  61         return;
  62     }
  63
  64     if (!s->on_connection) {
  65         close(nfd);
  66         return;
  67     }
  68
  69     /* There should be a check for socket type here */
  70     if (s->type == SOCKET_SERVER_IPV4)
  71         pa_socket_tcp_low_delay(fd);
  72     else
  73         pa_socket_low_delay(fd);
  74
  75     io = pa_iochannel_new(s->mainloop, nfd, nfd);
  76     assert(io);
  77     s->on_connection(s, io, s->userdata);
  78 }
  79
  80 struct pa_socket_server* pa_socket_server_new(struct pa_mainloop_api *m, int fd) {
  81     struct pa_socket_server *s;
  82     assert(m && fd >= 0);
  83
  84     s = malloc(sizeof(struct pa_socket_server));
  85     assert(s);
  86     s->fd = fd;
  87     s->filename = NULL;
  88     s->on_connection = NULL;
  89     s->userdata = NULL;
  90
  91     s->mainloop = m;
  92     s->mainloop_source = m->source_io(m, fd, PA_MAINLOOP_API_IO_EVENT_INPUT, callback, s);
  93     assert(s->mainloop_source);
  94
  95     s->type = SOCKET_SERVER_GENERIC;
  96
  97     return s;
  98 }
  99
 100 struct pa_socket_server* pa_socket_server_new_unix(struct pa_mainloop_api *m, const char *filename) {
 101     int fd = -1;
 102     struct sockaddr_un sa;
 103     struct pa_socket_server *s;
 104
 105     assert(m && filename);
 106
 107     if ((fd = socket(PF_LOCAL, SOCK_STREAM, 0)) < 0) {
 108         fprintf(stderr, "socket(): %s\n", strerror(errno));
 109         goto fail;
 110     }
 111
 112     sa.sun_family = AF_LOCAL;
 113     strncpy(sa.sun_path, filename, sizeof(sa.sun_path)-1);
 114     sa.sun_path[sizeof(sa.sun_path) - 1] = 0;
 115
 116     pa_socket_low_delay(fd);
 117
 118     if (bind(fd, (struct sockaddr*) &sa, SUN_LEN(&sa)) < 0) {
 119         fprintf(stderr, "bind(): %s\n", strerror(errno));
 120         goto fail;
 121     }
 122
 123     if (listen(fd, 5) < 0) {
 124         fprintf(stderr, "listen(): %s\n", strerror(errno));
 125         goto fail;
 126     }
 127
 128     s = pa_socket_server_new(m, fd);
 129     assert(s);
 130
 131     s->filename = strdup(filename);
 132     assert(s->filename);
 133
 134     s->type = SOCKET_SERVER_UNIX;
 135
 136     return s;
 137
 138 fail:
 139     if (fd >= 0)
 140         close(fd);
 141
 142     return NULL;
 143 }
 144
 145 struct pa_socket_server* pa_socket_server_new_ipv4(struct pa_mainloop_api *m, uint32_t address, uint16_t port) {
 146     struct pa_socket_server *ss;
 147     int fd = -1;
 148     struct sockaddr_in sa;
 149     int on = 1;
 150
 151     assert(m && port);
 152
 153     if ((fd = socket(PF_INET, SOCK_STREAM, 0)) < 0) {
 154         fprintf(stderr, "socket(): %s\n", strerror(errno));
 155         goto fail;
 156     }
 157
 158     if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on)) < 0)
 159         fprintf(stderr, "setsockopt(): %s\n", strerror(errno));
 160
 161     pa_socket_tcp_low_delay(fd);
 162
 163     sa.sin_family = AF_INET;
 164     sa.sin_port = htons(port);
 165     sa.sin_addr.s_addr = htonl(address);
 166
 167     if (bind(fd, (struct sockaddr *) &sa, sizeof(sa)) < 0) {
 168         fprintf(stderr, "bind(): %s\n", strerror(errno));
 169         goto fail;
 170     }
 171
 172     if (listen(fd, 5) < 0) {
 173         fprintf(stderr, "listen(): %s\n", strerror(errno));
 174         goto fail;
 175     }
 176
 177     if ((ss = pa_socket_server_new(m, fd)))
 178         ss->type = SOCKET_SERVER_IPV4;
 179
 180     return ss;
 181
 182 fail:
 183     if (fd >= 0)
 184         close(fd);
 185
 186     return NULL;
 187 }
 188
 189 void pa_socket_server_free(struct pa_socket_server*s) {
 190     assert(s);
 191     close(s->fd);
 192
 193     if (s->filename) {
 194         unlink(s->filename);
 195         free(s->filename);
 196     }
 197
 198
 199     s->mainloop->cancel_io(s->mainloop, s->mainloop_source);
 200
 201     free(s);
 202 }
 203
 204 void pa_socket_server_set_callback(struct pa_socket_server*s, void (*on_connection)(struct pa_socket_server*s, struct pa_iochannel *io, void *userdata), void *userdata) {
 205     assert(s);
 206
 207     s->on_connection = on_connection;
 208     s->userdata = userdata;
 209 }