/* --------------------------  netclient  ------------------------------------- */
/*                                                                              */
/* Extended 'netsend', connects to 'netserver'.                                 */
/* Uses child thread to connect to server. Thus needs pd0.35-test17 or later.   */
/* Written by Olaf Matthes (olaf.matthes@gmx.de)                                */
/* Get source at http://www.akustische-kunst.org/puredata/maxlib/               */
/*                                                                              */
/* 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.                       */
/*                                                                              */
/* 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.                             */
/*                                                                              */
/* ---------------------------------------------------------------------------- */


#include "m_pd.h"
#include "s_stuff.h"  /* for sys_addpollfn() sys_rmpollfm() */

#include <sys/types.h>
#include <string.h>
#include <pthread.h>
#ifdef UNIX
#include <sys/socket.h>
#include <sys/errno.h>
#include <netinet/in.h>
#include <netdb.h>
#include <stdio.h>
#include <unistd.h>
#define SOCKET_ERROR -1
#else
#include <winsock.h>
#endif

#define STRBUF_SIZE 32	/* maximum numbers of characters to read */
/* 
 * No longer needed since clock_delay() has been swapped for sys_addpollfn() 
 */
/* #define DEFPOLLTIME 20 */  /* check for input every 20 ms */

static char *version = "netclient v0.1, written by Olaf Matthes <olaf.matthes@gmx.de>\n\
      (converted to pollfn)";

static t_class *netclient_class;

typedef struct _netclient
{
    t_object x_obj;
	t_clock *x_clock;
/* 	t_clock *x_poll; */
	t_outlet *x_outdata;
	t_outlet *x_outconnect;
    int x_fd;
	char *x_hostname;
	int x_connectstate;
	int x_port;
    int x_protocol;
		/* multithread stuff */
	pthread_t x_threadid;            /* id of child thread */
	pthread_attr_t x_threadattr;     /* attributes of child thread */
} t_netclient;

/*
 * these are in s_stuff.h, which is now included because of sys_*pollfn()
 */
/* static void sys_sockerror(char *s) */
/* { */
/* #ifdef NT */
/*     int err = WSAGetLastError(); */
/*     if (err == 10054) return; */
/* #endif */
/* #ifdef UNIX */
/*     int err = errno; */
/* #endif */
/*     post("%s: %s (%d)\n", s, strerror(err), err); */
/* } */

/* static void sys_closesocket(int fd) { */

/* #ifdef UNIX */
/*     close(fd); */
/* #endif */
/* #ifdef NT */
/*     closesocket(fd); */
/* #endif */
/* } */


static void netclient_tick(t_netclient *x)
{
    outlet_float(x->x_outconnect, 1);
}

static void netclient_rcv(t_netclient *x)
{
	int sockfd = x->x_fd;
	int ret;
    char resp[STRBUF_SIZE];
	fd_set readset;
	fd_set exceptset;
    struct timeval  ztout;
		/* output data */
	t_binbuf *binbuf;
    t_atom messbuf[1024];
    int msg, natom;
    t_atom *at;
	int i;
	
	if(x->x_connectstate)
	{
			/* check if we can read/write from/to the socket */
		FD_ZERO(&readset);
		FD_ZERO(&exceptset);
		FD_SET(x->x_fd, &readset );
		FD_SET(x->x_fd, &exceptset );

		ztout.tv_sec = 0;
		ztout.tv_usec = 0;

		ret = select(sockfd+1, &readset, NULL, &exceptset, &ztout);
		if(ret < 0)
		{
			error("netclient: can not read from socket");
			sys_closesocket(sockfd);
			return;
		}
		if(FD_ISSET(sockfd, &readset) || FD_ISSET(sockfd, &exceptset))
		{
				/* read from server */
			ret = recv(sockfd, resp, STRBUF_SIZE, 0);
			if(ret > 0) 
			{
				// post("netclient: received: %s, %d bytes, %d stringlen", resp, ret, strlen(resp));
					/* convert string into atoms using a binbuf */
				binbuf = binbuf_new();
				binbuf_text(binbuf, resp, ret); //strlen(resp));
				natom = binbuf_getnatom(binbuf);	/* get number of atoms */
				at = binbuf_getvec(binbuf);	/* get the atoms */
					/* now split it into several parts at every A_SEMI because
					   we probably received more than one message at a time     */
				for (msg = 0; msg < natom;)
				{
    				int emsg;
					for (emsg = msg; emsg < natom && at[emsg].a_type != A_COMMA
						&& at[emsg].a_type != A_SEMI; emsg++);

					if (emsg > msg)
					{
						int ii;
						for (ii = msg; ii < emsg; ii++)
	    					if (at[ii].a_type == A_DOLLAR || at[ii].a_type == A_DOLLSYM)
							{
	    						pd_error(x, "netserver: -- got dollar sign in message");
								goto nodice;
							}
						if (at[msg].a_type == A_FLOAT)
						{
	    					if (emsg > msg + 1)
								outlet_list(x->x_outdata, 0, emsg-msg, at + msg);
							else outlet_float(x->x_outdata, at[msg].a_w.w_float);
						}
						else if (at[msg].a_type == A_SYMBOL)
	    					outlet_anything(x->x_outdata, at[msg].a_w.w_symbol,
							emsg-msg-1, at + msg + 1);
					}
					nodice:
    					msg = emsg + 1;
				}
				binbuf_free(binbuf);
			}
			else post("netclient: read() did not return any data");
		}
	}
	else post("netclient: not connected");
}

static int netclient_poll(t_netclient *x, int fd)
{
    if(x->x_connectstate)
		netclient_rcv(x);	/* if we're connected, try to read */
/* 	clock_delay(x->x_poll, DEFPOLLTIME); */	/* see you later */

	 return 1;
}

static void *netclient_child_connect(void *w)
{
	t_netclient *x = (t_netclient*) w;
    struct sockaddr_in server;
    struct hostent *hp;
    int sockfd;
    int portno = x->x_port;
    if (x->x_fd >= 0)
    {
    	error("netclient_connect: already connected");
    	return (x);
    }

    	/* create a socket */
    sockfd = socket(AF_INET, x->x_protocol, 0);
#if 0
    fprintf(stderr, "send socket %d\n", sockfd);
#endif
    if (sockfd < 0)
    {
    	sys_sockerror("socket");
    	return (x);
    }
		/* connect socket using hostname provided in command line */
    server.sin_family = AF_INET;
    hp = gethostbyname(x->x_hostname);
    if (hp == 0)
    {
		post("bad host?\n");
		return (x);
    }
    memcpy((char *)&server.sin_addr, (char *)hp->h_addr, hp->h_length);

		/* assign client port number */
    server.sin_port = htons((u_short)portno);

    post("connecting to port %d", portno);
		/* try to connect */
    if (connect(sockfd, (struct sockaddr *) &server, sizeof (server)) < 0)
    {
    	sys_sockerror("connecting stream socket");
    	sys_closesocket(sockfd);
    	return (x);
    }
    x->x_fd = sockfd;
	 sys_addpollfn(x->x_fd, (t_fdpollfn)netclient_poll, x);
		/* outlet_float is not threadsafe ! */
    // outlet_float(x->x_obj.ob_outlet, 1);
	x->x_connectstate = 1;
		/* use callback instead to set outlet */
 	clock_delay(x->x_clock, 0);
	return (x);
}

static void netclient_connect(t_netclient *x, t_symbol *hostname,
    t_floatarg fportno)
{
		/* we get hostname and port and pass them on
		to the child thread that establishes the connection */
	x->x_hostname = hostname->s_name;
	x->x_port = fportno;
	x->x_connectstate = 0;
		/* start child thread */
	if(pthread_create( &x->x_threadid, &x->x_threadattr, netclient_child_connect, x) < 0)
		post("netclient: could not create new thread");
}

static void netclient_disconnect(t_netclient *x)
{
	if (x->x_fd >= 0)
	{
		sys_rmpollfn(x->x_fd);
		sys_closesocket(x->x_fd);
		x->x_fd = -1;
		x->x_connectstate = 0;
		outlet_float(x->x_outconnect, 0);
    }
}

static void netclient_send(t_netclient *x, t_symbol *s, int argc, t_atom *argv)
{
    if (x->x_fd >= 0)
    {
	t_binbuf *b = binbuf_new();
	char *buf, *bp;
	int length, sent;
	t_atom at;
	binbuf_add(b, argc, argv);
	SETSEMI(&at);
	binbuf_add(b, 1, &at);
	binbuf_gettext(b, &buf, &length);
	for (bp = buf, sent = 0; sent < length;)
	{
	    static double lastwarntime;
	    static double pleasewarn;
	    double timebefore = sys_getrealtime();
    	    int res = send(x->x_fd, buf, length-sent, 0);
    	    double timeafter = sys_getrealtime();
    	    int late = (timeafter - timebefore > 0.005);
    	    if (late || pleasewarn)
    	    {
    	    	if (timeafter > lastwarntime + 2)
    	    	{
    	    	     post("netclient blocked %d msec",
    	    	     	(int)(1000 * ((timeafter - timebefore) + pleasewarn)));
    	    	     pleasewarn = 0;
    	    	     lastwarntime = timeafter;
    	    	}
    	    	else if (late) pleasewarn += timeafter - timebefore;
    	    }
    	    if (res <= 0)
    	    {
    		sys_sockerror("netclient");
    		netclient_disconnect(x);
    		break;
    	    }
    	    else
    	    {
    		sent += res;
    		bp += res;
    	    }
	}
	t_freebytes(buf, length);
	binbuf_free(b);
    }
    else error("netclient: not connected");
}

static void *netclient_new(t_floatarg udpflag)
{
    t_netclient *x = (t_netclient *)pd_new(netclient_class);
    x->x_outdata = outlet_new(&x->x_obj, &s_anything);	/* received data */
    x->x_outconnect = outlet_new(&x->x_obj, &s_float);	/* connection state */
    x->x_clock = clock_new(x, (t_method)netclient_tick);
/*     x->x_poll = clock_new(x, (t_method)netclient_poll); */
    x->x_fd = -1;
    x->x_protocol = (udpflag != 0 ? SOCK_DGRAM : SOCK_STREAM);
		/* prepare child thread */
    if(pthread_attr_init(&x->x_threadattr) < 0)
       post("netclient: warning: could not prepare child thread" );
    if(pthread_attr_setdetachstate(&x->x_threadattr, PTHREAD_CREATE_DETACHED) < 0)
       post("netclient: warning: could not prepare child thread" );
	/* start polling the input */
/* 	 clock_delay(x->x_poll, 0); */

	 return (x);
}

static void netclient_free(t_netclient *x)
{
    netclient_disconnect(x);
/*     clock_free(x->x_poll); */
	 clock_free(x->x_clock);
}

#ifndef MAXLIB
void netclient_setup(void)
{
    netclient_class = class_new(gensym("netclient"), (t_newmethod)netclient_new,
    	(t_method)netclient_free,
    	sizeof(t_netclient), 0, A_DEFFLOAT, 0);
    class_addmethod(netclient_class, (t_method)netclient_connect, gensym("connect"), A_SYMBOL, A_FLOAT, 0);
    class_addmethod(netclient_class, (t_method)netclient_disconnect, gensym("disconnect"), 0);
    class_addmethod(netclient_class, (t_method)netclient_send, gensym("send"), A_GIMME, 0);
	class_addmethod(netclient_class, (t_method)netclient_rcv, gensym("receive"), 0);
	class_addmethod(netclient_class, (t_method)netclient_rcv, gensym("rcv"), 0);
    	/* removed to be compatible with 0.37 */ 
 	/* class_sethelpsymbol(netclient_class, gensym("help-netclient.pd"));*/
	post(version);
}
#else
void maxlib_netclient_setup(void)
{
    netclient_class = class_new(gensym("maxlib_netclient"), (t_newmethod)netclient_new,
    	(t_method)netclient_free,
    	sizeof(t_netclient), 0, A_DEFFLOAT, 0);
	class_addcreator((t_newmethod)netclient_new, gensym("netclient"), A_DEFFLOAT, 0);
    class_addmethod(netclient_class, (t_method)netclient_connect, gensym("connect"), A_SYMBOL, A_FLOAT, 0);
    class_addmethod(netclient_class, (t_method)netclient_disconnect, gensym("disconnect"), 0);
    class_addmethod(netclient_class, (t_method)netclient_send, gensym("send"), A_GIMME, 0);
	class_addmethod(netclient_class, (t_method)netclient_rcv, gensym("receive"), 0);
	class_addmethod(netclient_class, (t_method)netclient_rcv, gensym("rcv"), 0);
    class_sethelpsymbol(netclient_class, gensym("maxlib/help-netclient.pd"));
}
#endif