aboutsummaryrefslogtreecommitdiff
path: root/src/stable/tcpsocket.h
blob: 92b3c1f751b6c2f00e8dd639f24538439dbb8b7a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
/*
 * Copyright (C) 2007-2012 Xagasoft, All rights reserved.
 *
 * This file is part of the libbu++ library and is released under the
 * terms of the license contained in the file LICENSE.
 */

#ifndef BU_TCP_SOCKET_H
#define BU_TCP_SOCKET_H

#include <stdint.h>

#include "bu/config.h"
#include "bu/stream.h"
#include "bu/string.h"
#include "bu/exceptionbase.h"

namespace Bu
{
	subExceptionDeclBegin( TcpSocketException );
		enum {
			cRead,
			cWrite,
			cBadRead,
			cClosed,
			cTimeout
		};
	subExceptionDeclEnd();

	/**
	 * Network socket stream class.  This class provides a mechanism for
	 * communicating over a network using TCP/IP.  It will provide other low
	 * level protocol and addressing support later on, but for now it's just
	 * standard STREAM TCP/IP sockets.
	 *
	 * Unlike system sockets, these sockets are opened by default in
	 * non-blocking mode, you can specify your own timeout for opening a socket,
	 * and a number of non-fatal error messages have been automatically handled
	 * and treated as standard no-data-available-yet situations on read.
	 *
	 * Please note that there is a condition that will occur eventually (at
	 * least on *nix systems) that will trigger a SIGPIPE condition.  This
	 * will terminate your program immediately unless handled properly.  Most
	 * people doing any connections with TcpSocket will want to put this in
	 * their program somewhere before they use it:
	 *@code
	 #include <signal.h>
	 ...
	 ...
	 ...
	 sigset( SIGPIPE, SIG_IGN ); // do this before you use a Bu::TcpSocket
	 @endcode
	 * When this is done, Bu::TcpSocket will simply throw a broken pipe
	 * exception just like every other error condition, allowing your program
	 * to handle it sanely.
	 *
	 *@ingroup Serving
	 *@ingroup Streams
	 */
	class TcpSocket : public Stream
	{
	public:
#ifdef WIN32
		typedef unsigned int handle;
#else
		typedef int handle;
#endif

		TcpSocket( handle nTcpSocket );
		TcpSocket( const String &sAddr, int nPort, int nTimeout=30,
				bool bBlocking=true );
		virtual ~TcpSocket();
		
		virtual void close();
		virtual size read( void *pBuf, size nBytes );
		virtual size read( void *pBuf, size nBytes,
				uint32_t nSec, uint32_t nUSec=0 );
		virtual size write( const void *pBuf, size nBytes );
		virtual size write( const void *pBuf, size nBytes,
				uint32_t nSec, uint32_t nUSec=0 );
		using Stream::write;

		virtual size tell();
		virtual void seek( size offset );
		virtual void setPos( size pos );
		virtual void setPosEnd( size pos );
		virtual bool isEos();
		virtual bool isOpen();

		virtual void flush();

		virtual bool canRead();
		virtual bool canWrite();

		virtual bool isReadable();
		virtual bool isWritable();
		virtual bool isSeekable();

		virtual bool isBlocking();
		virtual void setBlocking( bool bBlocking=true );

		virtual void setSize( size iSize );

		Bu::String getAddress() const;
		operator handle() const;

		handle getHandle() const;
		handle takeHandle();

		virtual size getSize() const;
		virtual size getBlockSize() const;
		virtual Bu::String getLocation() const;

	private:
		void setAddress();

		handle nTcpSocket;
		
		bool bActive;
		bool bBlocking;
		String sReadBuf;
		String sAddress;
	};
}

#endif