ermyth/include/connection.h

/*
 * Copyright © 2005 Atheme Development Group
 * Rights to this code are as documented in doc/pod/license.pod.
 *
 * This contains the connection_t structure.
 *
 * $Id: connection.h,v 1.3 2007-08-28 17:08:06 pippijn Exp $
 */

#ifndef CONNECTION_H
#define CONNECTION_H

#include <deque>

#include <common/util.h>
#include <ermyth/callback.h>

#define SENDQSIZE (4096 - 40)

/* packetqueue struct */
struct packetqueue : zero_initialised
{
  node_t node;
  int firstused;                /* offset of first used byte */
  int firstfree;                /* 1 + offset of last used byte */
  char buf[SENDQSIZE];

  packetqueue ()
  : firstused (0), firstfree (0)
  {
  }
};

class connection_t : public zero_initialised
{
  void _destroy ()
  {
    delete this;
  }

  struct callbacks
  {
    functor::callback1<connection_t *> connected;
  };

public:
  enum flag
  {
    CF_UPLINK     = 1 << 0,
    CF_DCCOUT     = 1 << 1,
    CF_DCCIN      = 1 << 2,
    
    CF_CONNECTING = 1 << 3,
    CF_LISTENING  = 1 << 4,
    CF_CONNECTED  = 1 << 5,
    CF_DEAD       = 1 << 6,
    
    CF_NONEWLINE  = 1 << 7,
    CF_SEND_EOF   = 1 << 8,     /* shutdown(2) write end if sendque empty */
    CF_SEND_DEAD  = 1 << 9      /* write end shut down */
  };

  bool is_uplink ()      { return flags & CF_UPLINK; }
  bool is_dcc ()         { return flags & (CF_DCCOUT | CF_DCCIN); }
  bool is_dccout ()      { return flags & CF_DCCOUT; }
  bool is_dccin ()       { return flags & CF_DCCIN; }
  bool is_dead ()        { return flags & CF_DEAD; }
  bool is_connecting ()  { return flags & CF_CONNECTING; }
  bool is_listening ()   { return flags & CF_LISTENING; }

public:
  int index;

  typedef std::deque<packetqueue *> queue;

  char name[HOSTLEN];
  char hbuf[BUFSIZE + 1];

  queue recvq;
  queue sendq;

  int fd;
  int pollslot;

  time_t first_recv;
  time_t last_recv;

  struct sockaddr_in *sa;
  struct sockaddr saddr;
  unsigned int saddr_size;

  void (*read_handler) (connection_t *);
  void (*write_handler) (connection_t *);

  unsigned flags;

  void (*recvq_handler) (connection_t *);
  void (*close_handler) (connection_t *);
  connection_t *listener;
  void *userdata;

  static callbacks callback;

public:
  static connection_t *create (char const * const name, int fd, unsigned int flags,
                               void (*read_handler) (connection_t *), void (*write_handler) (connection_t *));
  static connection_t *open_tcp (char const * const host, char const * const vhost, unsigned int port,
                                 void (*read_handler) (connection_t *), void (*write_handler) (connection_t *));
  static connection_t *open_listener_tcp (char const * const host, unsigned int port, void (*read_handler) (connection_t *));
  static connection_t *accept_tcp (connection_t *, void (*)(connection_t *), void (*)(connection_t *));

  void setselect (connection_t *cptr, void (*)(connection_t *), void (*)(connection_t *));
  void close ();
  void close_soon ();
  void close_soon_children ();
  static void close_all (void);
  static void stats (void (*)(char const *, void *), void *);
  void write (char *format, ...);
  void write_raw (char *data);
  static connection_t *find (int);
  static void select (time_t delay);
  static int count (void);
};

typedef indexing_vector<connection_t> connection_vector;
E connection_vector connlist;

#endif
Revision:	1.4
Committed:	Thu Aug 30 19:56:19 2007 UTC (16 years, 9 months ago) by pippijn
Content type:	text/plain
Branch:	MAIN
Changes since 1.3:	+47 -41 lines
Log Message:	- put faultcodes into their own namespace - removed old files - limited header garbage in atheme.h - macros to inline bools for connection_t::is_* - put some connection_t functions into the connection_t class
#	User	Rev	Content
1	pippijn	1.1	/*
2			* Copyright © 2005 Atheme Development Group
3	pippijn	1.2	* Rights to this code are as documented in doc/pod/license.pod.
4	pippijn	1.1	*
5			* This contains the connection_t structure.
6			*
7	pippijn	1.4	* $Id: connection.h,v 1.3 2007-08-28 17:08:06 pippijn Exp $
8	pippijn	1.1	*/
9
10			#ifndef CONNECTION_H
11			#define CONNECTION_H
12
13			#include <deque>
14
15			#include <common/util.h>
16	pippijn	1.3	#include <ermyth/callback.h>
17	pippijn	1.1
18			#define SENDQSIZE (4096 - 40)
19
20			/* packetqueue struct */
21	pippijn	1.3	struct packetqueue : zero_initialised
22	pippijn	1.1	{
23			node_t node;
24			int firstused; /* offset of first used byte */
25			int firstfree; /* 1 + offset of last used byte */
26			char buf[SENDQSIZE];
27
28			packetqueue ()
29			: firstused (0), firstfree (0)
30			{
31			}
32			};
33
34	pippijn	1.3	class connection_t : public zero_initialised
35	pippijn	1.1	{
36	pippijn	1.3	void _destroy ()
37			{
38			delete this;
39			}
40
41			struct callbacks
42			{
43			functor::callback1<connection_t *> connected;
44			};
45
46			public:
47	pippijn	1.4	enum flag
48			{
49			CF_UPLINK = 1 << 0,
50			CF_DCCOUT = 1 << 1,
51			CF_DCCIN = 1 << 2,
52
53			CF_CONNECTING = 1 << 3,
54			CF_LISTENING = 1 << 4,
55			CF_CONNECTED = 1 << 5,
56			CF_DEAD = 1 << 6,
57
58			CF_NONEWLINE = 1 << 7,
59			CF_SEND_EOF = 1 << 8, /* shutdown(2) write end if sendque empty */
60			CF_SEND_DEAD = 1 << 9 /* write end shut down */
61			};
62
63			bool is_uplink () { return flags & CF_UPLINK; }
64			bool is_dcc () { return flags & (CF_DCCOUT \| CF_DCCIN); }
65			bool is_dccout () { return flags & CF_DCCOUT; }
66			bool is_dccin () { return flags & CF_DCCIN; }
67			bool is_dead () { return flags & CF_DEAD; }
68			bool is_connecting () { return flags & CF_CONNECTING; }
69			bool is_listening () { return flags & CF_LISTENING; }
70
71			public:
72	pippijn	1.3	int index;
73
74	pippijn	1.1	typedef std::deque<packetqueue *> queue;
75
76			char name[HOSTLEN];
77			char hbuf[BUFSIZE + 1];
78
79			queue recvq;
80			queue sendq;
81
82			int fd;
83			int pollslot;
84
85			time_t first_recv;
86			time_t last_recv;
87
88			struct sockaddr_in *sa;
89			struct sockaddr saddr;
90			unsigned int saddr_size;
91
92			void (read_handler) (connection_t );
93			void (write_handler) (connection_t );
94
95	pippijn	1.4	unsigned flags;
96	pippijn	1.1
97			void (recvq_handler) (connection_t );
98			void (close_handler) (connection_t );
99			connection_t *listener;
100			void *userdata;
101
102			static callbacks callback;
103	pippijn	1.4
104			public:
105			static connection_t create (char const const name, int fd, unsigned int flags,
106			void (read_handler) (connection_t ), void (write_handler) (connection_t ));
107			static connection_t open_tcp (char const const host, char const * const vhost, unsigned int port,
108			void (read_handler) (connection_t ), void (write_handler) (connection_t ));
109			static connection_t open_listener_tcp (char const const host, unsigned int port, void (read_handler) (connection_t ));
110			static connection_t accept_tcp (connection_t , void ()(connection_t ), void ()(connection_t ));
111
112			void setselect (connection_t cptr, void ()(connection_t ), void ()(connection_t *));
113			void close ();
114			void close_soon ();
115			void close_soon_children ();
116			static void close_all (void);
117			static void stats (void ()(char const , void ), void );
118			void write (char *format, ...);
119			void write_raw (char *data);
120			static connection_t *find (int);
121			static void select (time_t delay);
122			static int count (void);
123	pippijn	1.3	};
124	pippijn	1.1
125	pippijn	1.3	typedef indexing_vector<connection_t> connection_vector;
126			E connection_vector connlist;
127	pippijn	1.1
128			#endif