X-Git-Url: https://code.citadel.org/?a=blobdiff_plain;f=citadel%2Fevent_client.h;h=dd334f9633311fd50c70435f7eb0eee1630843f4;hb=b5b30adb3ece1aa3de706047de6ba0245eea74fc;hp=ecaf7f2e5ec0509183b7fa13a9bcd26bf4286d59;hpb=035a1bee3715f9063afeaaa239b385532b479aee;p=citadel.git

diff --git a/citadel/event_client.h b/citadel/event_client.h
index ecaf7f2e5..dd334f963 100644
--- a/citadel/event_client.h
+++ b/citadel/event_client.h
@@ -1,38 +1,194 @@
-#include <event.h>
+#ifndef __EVENT_CLIENT_H__
+#define __EVENT_CLIENT_H__
+#define EV_COMPAT3 0
+#include <ev.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <netdb.h>
+#include <arpa/nameser.h>
+#include <ares.h>
+#include <curl/curl.h>
 
 typedef struct AsyncIO AsyncIO;
 
 typedef enum _eNextState {
+	eSendDNSQuery,
+	eReadDNSReply,
+
+	eDBQuery,
+
+	eConnect,
 	eSendReply, 
 	eSendMore,
+	eSendFile,
+
 	eReadMessage, 
+	eReadMore,
+	eReadPayload,
+	eReadFile,
+
 	eTerminateConnection,
 	eAbort
 }eNextState;
 
-typedef int (*EventContextAttach)(void *Data);
-typedef eNextState (*IO_CallBack)(void *Data);
+typedef eNextState (*IO_CallBack)(AsyncIO *IO);
 typedef eReadState (*IO_LineReaderCallback)(AsyncIO *IO);
+typedef void (*ParseDNSAnswerCb)(AsyncIO*, unsigned char*, int);
+typedef void (*FreeDNSReply)(void *DNSData);
 
-struct AsyncIO {
-	int sock;
-	struct event recv_event, send_event;
-	IOBuffer SendBuf, RecvBuf;
-	IO_LineReaderCallback LineReader;
-	IO_CallBack ReadDone, SendDone;
-	StrBuf *IOBuf;
+
+typedef struct __ReadAsyncMsg {
+	StrBuf *MsgBuf;
+	size_t maxlen;		/* maximum message length */
+
+	const char *terminator;	/* token signalling EOT */
+	long tlen;
+	int dodot;
+
+	int flushing;		/* if we read maxlen, read until nothing more arives and ignore this. */
+
+	int crlf;		/* CRLF newlines instead of LF */
+} ReadAsyncMsg;
+
+
+typedef struct _DNSQueryParts {
+	ParseDNSAnswerCb DNS_CB;
+	IO_CallBack PostDNS;
+
+	int DNSStatus;
+	void *VParsedDNSReply;
+	FreeDNSReply DNSReplyFree;
 	void *Data;
-	DeleteHashDataFunc DeleteData; /* data is expected to contain AsyncIO... */
+} DNSQueryParts;
+
+typedef struct _evcurl_request_data 
+{
+	CURL   		  *chnd;
+	struct curl_slist *headers;
+	char   		   errdesc[CURL_ERROR_SIZE];
+
+	int    		   attached;
+
+	char   		  *PlainPostData;
+	long   		   PlainPostDataLen;
+	StrBuf 		  *PostData;
+
+	StrBuf 		  *ReplyData;
+	long   		   httpcode;
+} evcurl_request_data;
+
+struct AsyncIO {
+	long ID;
        	eNextState NextState;
+
+	/* connection related */
+	ParsedURL *ConnectMe;
+	
+	/* read/send related... */
+	StrBuf *IOBuf;
+	IOBuffer SendBuf, 
+		RecvBuf;
+
+	FDIOBuffer IOB; /* when sending from / reading into files, this is used. */
+
+	/* our events... */
+	ev_cleanup abort_by_shutdown, /* server wants to go down... */
+		db_abort_by_shutdown; /* server wants to go down... */
+	ev_timer conn_fail,           /* connection establishing timed out */
+		rw_timeout;           /* timeout while sending data */
+	ev_idle unwind_stack,         /* get c-ares out of the stack */
+		db_unwind_stack,      /* wait for next db operation... */
+		conn_fail_immediate;  /* unwind stack, but fail immediately. */
+	ev_io recv_event,             /* receive data from the client */
+		send_event,           /* send more data to the client */
+		conn_event;           /* Connection successfully established */
+
+	StrBuf *ErrMsg; /* if we fail to connect, or lookup, error goes here. */
+
+	/* Citadel application callbacks... */
+	IO_CallBack ReadDone, /* Theres new data to read... */
+		SendDone,     /* we may send more data */
+		Terminate,    /* shutting down... */
+		Timeout,      /* Timeout handler; may also be connection timeout */
+		ConnFail,     /* What to do when one connection failed? */
+		DNSFail,      /* the dns lookup didn't work out. */
+		ShutdownAbort,/* we're going down. make your piece. */ 
+		NextDBOperation; /* Perform Database IO */
+
+	IO_LineReaderCallback LineReader; /* if we have linereaders, maybe we want to read more lines before the real application logic is called? */
+
+	/* DNS Related */
+	ev_io dns_recv_event, 
+		dns_send_event;
+	struct ares_options DNSOptions;
+	ares_channel DNSChannel;
+	DNSQueryParts *DNSQuery;
+	
+	evcurl_request_data HttpReq;
+
+	/* Saving / loading a message async from / to disk */
+	ReadAsyncMsg *ReadMsg;
+	struct CtdlMessage *AsyncMsg;
+	struct recptypes *AsyncRcp;
+	/* Custom data; its expected to contain  AsyncIO so we can save malloc()s... */
+	void *Data;        /* application specific data */
+	void *CitContext;  /* Citadel Session context... */
 };
 
+typedef struct _IOAddHandler {
+	AsyncIO *IO;
+	IO_CallBack EvAttch;
+}IOAddHandler; 
+
+#define EV_syslog(LEVEL, FORMAT, ...) syslog(LEVEL, "IO[%ld]" FORMAT, IO->ID, __VA_ARGS__)
+#define EVM_syslog(LEVEL, FORMAT) syslog(LEVEL, "IO[%ld]" FORMAT, IO->ID)
+
 void FreeAsyncIOContents(AsyncIO *IO);
 
-int QueueEventContext(void *Ctx, AsyncIO *IO, EventContextAttach CB);
+eNextState NextDBOperation(AsyncIO *IO, IO_CallBack CB);
+eNextState QueueDBOperation(AsyncIO *IO, IO_CallBack CB);
+eNextState QueueEventContext(AsyncIO *IO, IO_CallBack CB);
+eNextState QueueCurlContext(AsyncIO *IO);
+int ShutDownEventQueue(void);
+
+eNextState InitEventIO(AsyncIO *IO, 
+		       void *pData, 
+		       double conn_timeout, 
+		       double first_rw_timeout,
+		       int ReadFirst);
+void IO_postdns_callback(struct ev_loop *loop, ev_idle *watcher, int revents);
+
+int QueueQuery(ns_type Type, const char *name, AsyncIO *IO, DNSQueryParts *QueryParts, IO_CallBack PostDNS);
+void QueueGetHostByName(AsyncIO *IO, const char *Hostname, DNSQueryParts *QueryParts, IO_CallBack PostDNS);
+
+void QueryCbDone(AsyncIO *IO);
+
+void StopClient(AsyncIO *IO);
+
+void StopClientWatchers(AsyncIO *IO);
+
+void SetNextTimeout(AsyncIO *IO, double timeout);
+
+void InitC_ares_dns(AsyncIO *IO);
+
+#include <curl/curl.h>
+
+#define OPT(s, v) \
+	do { \
+		sta = curl_easy_setopt(chnd, (CURLOPT_##s), (v)); \
+		if (sta)  {						\
+			syslog(LOG_ERR, "error setting option " #s " on curl handle: %s", curl_easy_strerror(sta)); \
+	} } while (0)
+
+int evcurl_init(AsyncIO *IO,
+                void *CustomData,
+                const char* Desc,
+                IO_CallBack CallBack,
+                IO_CallBack Terminate, 
+		IO_CallBack ShutdownAbort);
+
+eNextState ReAttachIO(AsyncIO *IO, 
+		      void *pData, 
+		      int ReadFirst);
 
-void InitEventIO(AsyncIO *IO, 
-		 void *pData, 
-		 IO_CallBack ReadDone, 
-		 IO_CallBack SendDone, 
-		 IO_LineReaderCallback LineReader,
-		 int ReadFirst);
+#endif /* __EVENT_CLIENT_H__ */