From d141cb156f5088c3e6b99eca0b86694e957100a8 Mon Sep 17 00:00:00 2001 From: Martin Peach Date: Mon, 11 Jan 2010 14:25:48 +0000 Subject: Versions of netsend~ for udp. svn path=/trunk/externals/mrpeach/; revision=12953 --- net/udpsend~.c | 648 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 648 insertions(+) create mode 100644 net/udpsend~.c (limited to 'net/udpsend~.c') diff --git a/net/udpsend~.c b/net/udpsend~.c new file mode 100644 index 0000000..ce9bcad --- /dev/null +++ b/net/udpsend~.c @@ -0,0 +1,648 @@ +/* udpsend~ started by Martin Peach on 20100110, based on netsend~ */ +/* udpsend~ sends audio via udp only.*/ +/* It is a PD external, all Max stuff has been removed from the source */ +/* ------------------------ netsend~ ------------------------------------------ */ +/* */ +/* Tilde object to send uncompressed audio data to netreceive~. */ +/* Written by Olaf Matthes . */ +/* Based on streamout~ by Guenter Geiger. */ +/* Get source at http://www.akustische-kunst.org/ */ +/* */ +/* This program is free software; you can redistribute it and/or */ +/* modify it under the terms of the GNU General Public License */ +/* as published by the Free Software Foundation; either version 2 */ +/* of the License, or (at your option) any later version. */ +/* */ +/* See file LICENSE for further informations on licensing terms. */ +/* */ +/* 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., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. */ +/* */ +/* Based on PureData by Miller Puckette and others. */ +/* */ +/* This project was commissioned by the Society for Arts and Technology [SAT], */ +/* Montreal, Quebec, Canada, http://www.sat.qc.ca/. */ +/* */ +/* ---------------------------------------------------------------------------- */ + +#include "m_pd.h" + +#include "udpsend~.h" +#include "float_cast.h" /* tools for fast conversion from float to int */ + +#include +#include +#include +#ifdef UNIX +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#define SOCKET_ERROR -1 +#endif +#ifdef _WIN32 +#include +#include "pthread.h" +#endif + +#ifdef MSG_NOSIGNAL +#define SEND_FLAGS /*MSG_DONTWAIT|*/MSG_NOSIGNAL +#else +#define SEND_FLAGS 0 +#endif + +#ifndef SOL_IP +#define SOL_IP IPPROTO_IP +#endif + + +/* ------------------------ udpsend~ ----------------------------- */ + +static t_class *udpsend_tilde_class; + +static t_symbol *ps_nothing, *ps_localhost; +static t_symbol *ps_format, *ps_channels, *ps_framesize, *ps_overflow, *ps_underflow; +static t_symbol *ps_queuesize, *ps_average, *ps_sf_float, *ps_sf_16bit, *ps_sf_8bit; +static t_symbol *ps_sf_mp3, *ps_sf_aac, *ps_sf_unknown, *ps_bitrate, *ps_hostname; + +typedef struct _udpsend_tilde +{ + t_object x_obj; + t_outlet *x_outlet; + t_outlet *x_outlet2; + t_clock *x_clock; + int x_fd; + t_tag x_tag; + t_symbol* x_hostname; + int x_portno; + int x_connectstate; + char *x_cbuf; + int x_cbufsize; + int x_blocksize; /* set to DEFAULT_AUDIO_BUFFER_SIZE in udpsend_tilde_new() */ + int x_blockspersend; /* set to x->x_blocksize / x->x_vecsize in udpsend_tilde_perform() */ + int x_blockssincesend; + + long x_samplerate; /* samplerate we're running at */ + int x_vecsize; /* current DSP signal vector size */ + int x_ninlets; /* number of inlets */ + int x_channels; /* number of channels we want to stream */ + int x_format; /* format of streamed audio data */ + int x_bitrate; /* specifies bitrate for compressed formats */ + int x_count; /* total number of audio frames */ + t_int **x_myvec; /* vector we pass on in the DSP routine */ + + pthread_mutex_t x_mutex; + pthread_cond_t x_requestcondition; + pthread_cond_t x_answercondition; + pthread_t x_childthread; +} t_udpsend_tilde; + +/* function prototypes */ +static int udpsend_tilde_sockerror(char *s); +static void udpsend_tilde_closesocket(int fd); +static void udpsend_tilde_notify(t_udpsend_tilde *x); +static void udpsend_tilde_disconnect(t_udpsend_tilde *x); +static void *udpsend_tilde_doconnect(void *zz); +static void udpsend_tilde_connect(t_udpsend_tilde *x, t_symbol *host, t_floatarg fportno); +static t_int *udpsend_tilde_perform(t_int *w); +static void udpsend_tilde_dsp(t_udpsend_tilde *x, t_signal **sp); +static void udpsend_tilde_channels(t_udpsend_tilde *x, t_floatarg channels); +static void udpsend_tilde_format(t_udpsend_tilde *x, t_symbol* form, t_floatarg bitrate); +static void udpsend_tilde_float(t_udpsend_tilde* x, t_floatarg arg); +static void udpsend_tilde_info(t_udpsend_tilde *x); +static void *udpsend_tilde_new(t_floatarg inlets, t_floatarg prot); +static void udpsend_tilde_free(t_udpsend_tilde* x); +void udpsend_tilde_setup(void); + +/* functions */ +static void udpsend_tilde_notify(t_udpsend_tilde *x) +{ + pthread_mutex_lock(&x->x_mutex); + x->x_childthread = 0; + outlet_float(x->x_outlet, x->x_connectstate); + pthread_mutex_unlock(&x->x_mutex); +} + +static void udpsend_tilde_disconnect(t_udpsend_tilde *x) +{ + pthread_mutex_lock(&x->x_mutex); + if (x->x_fd != -1) + { + udpsend_tilde_closesocket(x->x_fd); + x->x_fd = -1; + x->x_connectstate = 0; + outlet_float(x->x_outlet, 0); + } + pthread_mutex_unlock(&x->x_mutex); +} + +static void *udpsend_tilde_doconnect(void *zz) +{ + t_udpsend_tilde *x = (t_udpsend_tilde *)zz; + struct sockaddr_in server; + struct hostent *hp; + int intarg = 1; + int sockfd; + int portno; + t_symbol *hostname; + + pthread_mutex_lock(&x->x_mutex); + hostname = x->x_hostname; + portno = x->x_portno; + pthread_mutex_unlock(&x->x_mutex); + + /* create a socket */ + sockfd = socket(AF_INET, SOCK_DGRAM, 0); + if (sockfd < 0) + { + post("udpsend~: connection to %s on port %d failed", hostname->s_name,portno); + udpsend_tilde_sockerror("socket"); + x->x_childthread = 0; + return (0); + } + + /* connect socket using hostname provided in command line */ + server.sin_family = AF_INET; + hp = gethostbyname(x->x_hostname->s_name); + if (hp == 0) + { + post("udpsend~: bad host?"); + x->x_childthread = 0; + return (0); + } + +#ifdef SO_PRIORITY + /* set high priority, LINUX only */ + intarg = 6; /* select a priority between 0 and 7 */ + if (setsockopt(sockfd, SOL_SOCKET, SO_PRIORITY, (const char*)&intarg, sizeof(int)) < 0) + { + error("udpsend~: setsockopt(SO_PRIORITY) failed"); + } +#endif + + memcpy((char *)&server.sin_addr, (char *)hp->h_addr, hp->h_length); + + /* assign client port number */ + server.sin_port = htons((unsigned short)portno); + + /* try to connect */ + if (connect(sockfd, (struct sockaddr *) &server, sizeof (server)) < 0) + { + udpsend_tilde_sockerror("connecting stream socket"); + udpsend_tilde_closesocket(sockfd); + x->x_childthread = 0; + return (0); + } + + post("udpsend~: connected host %s on port %d", hostname->s_name, portno); + + pthread_mutex_lock(&x->x_mutex); + x->x_fd = sockfd; + x->x_connectstate = 1; + clock_delay(x->x_clock, 0); + pthread_mutex_unlock(&x->x_mutex); + return (0); +} + +static void udpsend_tilde_connect(t_udpsend_tilde *x, t_symbol *host, t_floatarg fportno) +{ + pthread_mutex_lock(&x->x_mutex); + if (x->x_childthread != 0) + { + pthread_mutex_unlock(&x->x_mutex); + post("udpsend~: already trying to connect"); + return; + } + if (x->x_fd != -1) + { + pthread_mutex_unlock(&x->x_mutex); + post("udpsend~: already connected"); + return; + } + + if (host != ps_nothing) + x->x_hostname = host; + else + x->x_hostname = ps_localhost; /* default host */ + + if (!fportno) + x->x_portno = DEFAULT_PORT; + else + x->x_portno = (int)fportno; + x->x_count = 0; + + /* start child thread to connect */ + pthread_create(&x->x_childthread, 0, udpsend_tilde_doconnect, x); + pthread_mutex_unlock(&x->x_mutex); +} + +static t_int *udpsend_tilde_perform(t_int *w) +{ + t_udpsend_tilde* x = (t_udpsend_tilde*) (w[1]); + int n = (int)(w[2]); + t_float *in[DEFAULT_AUDIO_CHANNELS]; + const int offset = 3; + char* bp = NULL; + int i, length = x->x_blocksize * SF_SIZEOF(x->x_tag.format) * x->x_tag.channels; + int sent = 0; + + pthread_mutex_lock(&x->x_mutex); + + for (i = 0; i < x->x_ninlets; i++) + in[i] = (t_float *)(w[offset + i]); + + if (n != x->x_vecsize) /* resize buffer */ + { + x->x_vecsize = n; + x->x_blockspersend = x->x_blocksize / x->x_vecsize; + x->x_blockssincesend = 0; + length = x->x_blocksize * SF_SIZEOF(x->x_tag.format) * x->x_tag.channels; + } + + /* format the buffer */ + switch (x->x_tag.format) + { + case SF_FLOAT: + { + t_float* fbuf = (t_float *)x->x_cbuf + (x->x_blockssincesend * x->x_vecsize * x->x_tag.channels); + while (n--) + for (i = 0; i < x->x_tag.channels; i++) + *fbuf++ = *(in[i]++); + break; + } + case SF_16BIT: + { + short* cibuf = (short *)x->x_cbuf + (x->x_blockssincesend * x->x_vecsize * x->x_tag.channels); + while (n--) + for (i = 0; i < x->x_tag.channels; i++) + *cibuf++ = (short)lrint(32767.0 * *(in[i]++)); + break; + } + case SF_8BIT: + { + unsigned char* cbuf = (unsigned char*)x->x_cbuf + (x->x_blockssincesend * x->x_vecsize * x->x_tag.channels); + while (n--) + for (i = 0; i < x->x_tag.channels; i++) + *cbuf++ = (unsigned char)(128. * (1.0 + *(in[i]++))); + break; + } + default: + break; + } + + if (!(x->x_blockssincesend < x->x_blockspersend - 1)) /* time to send the buffer */ + { + x->x_blockssincesend = 0; + x->x_count++; /* count data packet we're going to send */ + + if (x->x_fd != -1) + { + bp = (char *)x->x_cbuf; + /* fill in the header tag */ + x->x_tag.framesize = length; + x->x_tag.count = x->x_count; + /* send the format tag */ + if (send(x->x_fd, (char*)&x->x_tag, sizeof(t_tag), SEND_FLAGS) < 0) + { + udpsend_tilde_sockerror("send tag"); + pthread_mutex_unlock(&x->x_mutex); + udpsend_tilde_disconnect(x); + return (w + offset + x->x_ninlets); + } +/* UDP: max. packet size is 64k (incl. headers) so we have to split */ + { +#ifdef __APPLE__ + /* WARNING: due to a 'bug' (maybe Apple would call it a feature?) in OS X + send calls with data packets larger than 16k fail with error number 40! + Thus we have to split the data packets into several packets that are + 16k in size. The other side will reassemble them again. */ + int size = DEFAULT_UDP_PACKT_SIZE; + if (length < size) /* maybe data fits into one packet? */ + size = length; + /* send the buffer */ + for (sent = 0; sent < length;) + { + int ret = 0; + ret = send(x->x_fd, bp, size, SEND_FLAGS); + if (ret <= 0) + { + udpsend_tilde_sockerror("send data"); + pthread_mutex_unlock(&x->x_mutex); + udpsend_tilde_disconnect(x); + return (w + offset + x->x_ninlets); + } + else + { + bp += ret; + sent += ret; + if ((length - sent) < size) + size = length - sent; + } + } +#else + /* send the buffer, the OS might segment it into smaller packets */ + int ret = send(x->x_fd, bp, length, SEND_FLAGS); + if (ret <= 0) + { + udpsend_tilde_sockerror("send data"); + pthread_mutex_unlock(&x->x_mutex); + udpsend_tilde_disconnect(x); + return (w + offset + x->x_ninlets); + } +#endif + } + } + +/* check whether user has updated any parameters */ + if (x->x_tag.channels != x->x_channels) + { + x->x_tag.channels = x->x_channels; + } + if (x->x_tag.format != x->x_format) + { + x->x_tag.format = x->x_format; + } + } + else + { + x->x_blockssincesend++; + } + pthread_mutex_unlock(&x->x_mutex); + return (w + offset + x->x_ninlets); +} + +static void udpsend_tilde_dsp(t_udpsend_tilde *x, t_signal **sp) +{ + int i; + + pthread_mutex_lock(&x->x_mutex); + + x->x_myvec[0] = (t_int*)x; + x->x_myvec[1] = (t_int*)sp[0]->s_n; + + x->x_samplerate = sp[0]->s_sr; + + for (i = 0; i < x->x_ninlets; i++) + { + x->x_myvec[2 + i] = (t_int*)sp[i]->s_vec; + } + + pthread_mutex_unlock(&x->x_mutex); + + if (DEFAULT_AUDIO_BUFFER_SIZE % sp[0]->s_n) + { + error("udpsend~: signal vector size too large (needs to be even divisor of %d)", DEFAULT_AUDIO_BUFFER_SIZE); + } + else + { + dsp_addv(udpsend_tilde_perform, x->x_ninlets + 2, (t_int*)x->x_myvec); + } +} + +static void udpsend_tilde_channels(t_udpsend_tilde *x, t_floatarg channels) +{ + pthread_mutex_lock(&x->x_mutex); + if (channels >= 0 && channels <= DEFAULT_AUDIO_CHANNELS) + { + x->x_channels = (int)channels; + post("udpsend~: channels set to %d", (int)channels); + } + pthread_mutex_unlock(&x->x_mutex); +} + +static void udpsend_tilde_format(t_udpsend_tilde *x, t_symbol* form, t_floatarg bitrate) +{ + pthread_mutex_lock(&x->x_mutex); + if (!strncmp(form->s_name,"float", 5) && x->x_tag.format != SF_FLOAT) + { + x->x_format = (int)SF_FLOAT; + } + else if (!strncmp(form->s_name,"16bit", 5) && x->x_tag.format != SF_16BIT) + { + x->x_format = (int)SF_16BIT; + } + else if (!strncmp(form->s_name,"8bit", 4) && x->x_tag.format != SF_8BIT) + { + x->x_format = (int)SF_8BIT; + } + + post("udpsend~: format set to %s", form->s_name); + pthread_mutex_unlock(&x->x_mutex); +} + +static void udpsend_tilde_float(t_udpsend_tilde* x, t_floatarg arg) +{ + if (arg == 0.0) + udpsend_tilde_disconnect(x); + else + udpsend_tilde_connect(x,x->x_hostname,(float) x->x_portno); +} + +/* send stream info */ +static void udpsend_tilde_info(t_udpsend_tilde *x) +{ + t_atom list[2]; + t_symbol *sf_format; + t_float bitrate; + + bitrate = (t_float)((SF_SIZEOF(x->x_tag.format) * x->x_samplerate * 8 * x->x_tag.channels) / 1000.); + + switch (x->x_tag.format) + { + case SF_FLOAT: + { + sf_format = ps_sf_float; + break; + } + case SF_16BIT: + { + sf_format = ps_sf_16bit; + break; + } + case SF_8BIT: + { + sf_format = ps_sf_8bit; + break; + } + default: + { + sf_format = ps_sf_unknown; + break; + } + } + + /* --- stream information (t_tag) --- */ + + /* audio format */ + SETSYMBOL(list, (t_symbol *)sf_format); + outlet_anything(x->x_outlet2, ps_format, 1, list); + + /* channels */ + SETFLOAT(list, (t_float)x->x_tag.channels); + outlet_anything(x->x_outlet2, ps_channels, 1, list); + + /* framesize */ + SETFLOAT(list, (t_float)x->x_tag.framesize); + outlet_anything(x->x_outlet2, ps_framesize, 1, list); + + /* bitrate */ + SETFLOAT(list, (t_float)bitrate); + outlet_anything(x->x_outlet2, ps_bitrate, 1, list); + + /* IP address */ + SETSYMBOL(list, (t_symbol *)x->x_hostname); + outlet_anything(x->x_outlet2, ps_hostname, 1, list); +} + +static void *udpsend_tilde_new(t_floatarg inlets, t_floatarg prot) +{ + int i; + + t_udpsend_tilde *x = (t_udpsend_tilde *)pd_new(udpsend_tilde_class); + if (x) + { + for (i = sizeof(t_object); i < (int)sizeof(t_udpsend_tilde); i++) + ((char *)x)[i] = 0; + } + + x->x_ninlets = CLIP((int)inlets, 1, DEFAULT_AUDIO_CHANNELS); + for (i = 1; i < x->x_ninlets; i++) + inlet_new(&x->x_obj, &x->x_obj.ob_pd, &s_signal, &s_signal); + + x->x_outlet = outlet_new(&x->x_obj, &s_float); + x->x_outlet2 = outlet_new(&x->x_obj, &s_list); + x->x_clock = clock_new(x, (t_method)udpsend_tilde_notify); + + x->x_myvec = (t_int **)t_getbytes(sizeof(t_int *) * (x->x_ninlets + 3)); + if (!x->x_myvec) + { + error("udpsend~: out of memory"); + return NULL; + } + + pthread_mutex_init(&x->x_mutex, 0); + pthread_cond_init(&x->x_requestcondition, 0); + pthread_cond_init(&x->x_answercondition, 0); + + x->x_hostname = ps_localhost; + x->x_portno = DEFAULT_PORT; + x->x_connectstate = 0; + x->x_childthread = 0; + x->x_fd = -1; + + x->x_tag.format = x->x_format = SF_FLOAT; + x->x_tag.channels = x->x_channels = x->x_ninlets; + x->x_tag.version = SF_BYTE_NATIVE; /* native endianness */ + x->x_vecsize = 64; /* this is updated in the perform routine udpsend_tilde_perform */ + x->x_bitrate = 0; /* not specified, use default */ + x->x_cbuf = NULL; + x->x_blocksize = DEFAULT_AUDIO_BUFFER_SIZE; /* <-- the only place blocksize is set */ + x->x_blockspersend = x->x_blocksize / x->x_vecsize; /* 1024/64 = 16 blocks */ + x->x_blockssincesend = 0; + x->x_cbufsize = x->x_blocksize * sizeof(t_float) * x->x_ninlets; + x->x_cbuf = (char *)t_getbytes(x->x_cbufsize); + +#ifdef UNIX + /* we don't want to get signaled in case send() fails */ + signal(SIGPIPE, SIG_IGN); +#endif + + return (x); +} + +static void udpsend_tilde_free(t_udpsend_tilde* x) +{ + udpsend_tilde_disconnect(x); + + /* free the memory */ + if (x->x_cbuf)t_freebytes(x->x_cbuf, x->x_cbufsize); + if (x->x_myvec)t_freebytes(x->x_myvec, sizeof(t_int) * (x->x_ninlets + 3)); + + clock_free(x->x_clock); + + pthread_cond_destroy(&x->x_requestcondition); + pthread_cond_destroy(&x->x_answercondition); + pthread_mutex_destroy(&x->x_mutex); +} + +void udpsend_tilde_setup(void) +{ + udpsend_tilde_class = class_new(gensym("udpsend~"), (t_newmethod)udpsend_tilde_new, (t_method)udpsend_tilde_free, + sizeof(t_udpsend_tilde), 0, A_DEFFLOAT, A_DEFFLOAT, A_NULL); + class_addmethod(udpsend_tilde_class, nullfn, gensym("signal"), 0); + class_addmethod(udpsend_tilde_class, (t_method)udpsend_tilde_dsp, gensym("dsp"), 0); + class_addfloat(udpsend_tilde_class, udpsend_tilde_float); + class_addmethod(udpsend_tilde_class, (t_method)udpsend_tilde_info, gensym("info"), 0); + class_addmethod(udpsend_tilde_class, (t_method)udpsend_tilde_connect, gensym("connect"), A_DEFSYM, A_DEFFLOAT, 0); + class_addmethod(udpsend_tilde_class, (t_method)udpsend_tilde_disconnect, gensym("disconnect"), 0); + class_addmethod(udpsend_tilde_class, (t_method)udpsend_tilde_channels, gensym("channels"), A_FLOAT, 0); + class_addmethod(udpsend_tilde_class, (t_method)udpsend_tilde_format, gensym("format"), A_SYMBOL, A_DEFFLOAT, 0); + class_sethelpsymbol(udpsend_tilde_class, gensym("udpsend~")); + post("udpsend~ v%s, (c) 2004-2005 Olaf Matthes, 2010 Martin Peach", VERSION); + + ps_nothing = gensym(""); + ps_localhost = gensym("localhost"); + ps_hostname = gensym("ipaddr"); + ps_format = gensym("format"); + ps_channels = gensym("channels"); + ps_framesize = gensym("framesize"); + ps_bitrate = gensym("bitrate"); + ps_sf_float = gensym("_float_"); + ps_sf_16bit = gensym("_16bit_"); + ps_sf_8bit = gensym("_8bit_"); + ps_sf_unknown = gensym("_unknown_"); +} + +/* Utility functions */ + +static int udpsend_tilde_sockerror(char *s) +{ +#ifdef _WIN32 + int err = WSAGetLastError(); + if (err == 10054) return 1; + else if (err == 10053) post("udpsend~: %s: software caused connection abort (%d)", s, err); + else if (err == 10055) post("udpsend~: %s: no buffer space available (%d)", s, err); + else if (err == 10060) post("udpsend~: %s: connection timed out (%d)", s, err); + else if (err == 10061) post("udpsend~: %s: connection refused (%d)", s, err); + else post("udpsend~: %s: %s (%d)", s, strerror(err), err); +#else + int err = errno; + post("udpsend~: %s: %s (%d)", s, strerror(err), err); +#endif +#ifdef _WIN32 + if (err == WSAEWOULDBLOCK) +#endif +#ifdef UNIX + if (err == EAGAIN) +#endif + { + return 1; /* recoverable error */ + } + return 0; /* indicate non-recoverable error */ +} + +static void udpsend_tilde_closesocket(int fd) +{ +#ifdef UNIX + close(fd); +#endif +#ifdef _WIN32 + closesocket(fd); +#endif +} + +/* fin udpsend~.c */ -- cgit v1.2.1