/*
 * Copyright (c) 2004 Teodor Sigaev <teodor@sigaev.ru>
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *        notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *        notice, this list of conditions and the following disclaimer in the
 *        documentation and/or other materials provided with the distribution.
 * 3. Neither the name of the author nor the names of any co-contributors
 *        may be used to endorse or promote products derived from this software
 *        without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY CONTRIBUTORS ``AS IS'' AND ANY EXPRESS
 * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL CONTRIBUTORS BE LIABLE FOR ANY
 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
 * GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
 * IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
 * IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#ifndef __CONNECTION_H__
#define __CONNECTION_H__

#include <sys/types.h>
#include <sys/time.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>



#define CS_OK   	0
#define CS_INPROCESS    1
#define CS_CONNECTED    2
#define CS_READ         3
#define CS_SEND         4
#define CS_WAIT         5
#define CS_ERROR        6
#define CS_FINISHSEND   7
#define CS_FINISHREAD   8
#define CS_TIMEOUT   	9
#define CS_CLOSED   	10	
#define CS_NOTINITED   	11
#define CS_AGAIN	12
#define CS_FINISH	13


#define READ_INCRIMENT_BUFSIZ 1024

typedef struct {
	/* I/O buffer */
        u_int32_t  len;
        char    *buf;
        char    *ptr;

	/* internal fields */
        int     fd;
        u_int32_t
		readyio:1,  
		state:29;
        struct sockaddr_in serv_addr;

	/* external link */
	void*   	data;
} TC_Connection;

#define TCCONNHDRSZ   ( sizeof(TC_Connection) - sizeof(void*) )

TC_Connection *TC_fillConnection( TC_Connection *cs, char *name, u_int32_t port );
TC_Connection* TC_AcceptTcp(TC_Connection *cs);
u_int32_t TC_ClientInitConnection(TC_Connection *cs, char *name, u_int32_t port);
u_int32_t TC_ServerInitConnect( TC_Connection *cs );
u_int32_t TC_ServerConnect( TC_Connection *cs );
u_int32_t TC_Send( TC_Connection *cs );
u_int32_t TC_Read( TC_Connection *cs );
u_int32_t TC_Talk( TC_Connection *cs );
void TC_FreeConnection( TC_Connection *cs );
int TC_ReadyIO( TC_Connection **cs, int number, int timeout );

typedef struct {
        u_int32_t       len;
        u_int32_t       type;
        char            data[1];
} TCMsg;

#define TCMSGHDRSZ      (2*sizeof(u_int32_t))


/* udp */
typedef struct {
	char	*host;
	u_int32_t	port;
	TCMsg	*msg;

	/* private members */
	int		sockfd;
	struct sockaddr_in host_addr;
} Msg; 

/* 
 * Udp server loop:
 * TC_Connection  conn,*connptr;
 * conn.fd = TC_AcceptUdp("127.0.0.1", 5432);
 * conn.state = CS_READ;
 * connptr = &conn;
 * while(1) {
 * 	if ( TC_ReadyIO( &connptr, 1, 100 ) ) {
 * 		Msg m; 
 * 		if ( TC_getMsg(conn.fd, &m) == CS_OK ) {
 * 			//do something
 * 		} 
 * 	}
 * }
 * close(conn.fd);
 *
 * Udp client send:
 * Msg msg;
 * msg.host        = "127.0.0.1";
 * msg.port        = 5432;
 * msg.sockfd =-1;
 * msg.msg = GOTFILLEDPMSG();
 * if ( TC_sendMsg(&msg)!=CS_OK ) {
 *        //Very bad
 * }
 * msg.msg = GOTFILLEDPMSG();
 * if ( TC_sendMsg(&msg)!=CS_OK ) {
 *        //Very bad
 * }
 * TC_closefd(&msg);
 */ 

int TC_AcceptUdp(char *host, int port);
u_int32_t TC_getMsg( int sockfd, Msg *msg );
u_int32_t TC_sendMsg( Msg *msg );
void	  TC_closefd( Msg *msg );

typedef struct {
        u_int32_t       len;
        u_int32_t       number;
        TC_Connection   **conn;
} PoolConnection;

void TC_addConnection(PoolConnection *pool, TC_Connection *c);
void TC_deleteConnectionByN(PoolConnection *pool, int n);
void TC_deleteConnectionByC(PoolConnection *pool, TC_Connection *c);

#endif