001:
002:
003:
004:
005:
006:
007:
008:
009:
010:
011:
012:
013:
014:
015:
016:
017:
018:
019:
020:
021:
022:
023:
024:
025:
026:
027:
028:
029:
030:
031:
032:
033:
034:
035:
036:
037:
038:
039:
040:
041:
042:
043:
044:
045:
046:
047:
048:
049:
050:
051:
052:
053:
054:
055:
056:
057:
058:
059:
060:
061:
062:
063:
064:
065:
066:
067:
068:
069:
070:
071:
072:
073:
074:
075:
076:
077:
078:
079:
080:
081:
082:
083:
084:
085:
086:
087:
088:
089:
090:
091:
092:
093:
094:
095:
096:
097:
098:
099:
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:
127:
128:
129:
130:
131:
132:
133:
134:
135:
136:
137:
138:
139:
140:
141:
142:
143:
144:
145:
146:
/*
* Copyright (c), Zeriph Enterprises
* All rights reserved.
*
* Contributor(s):
* Zechariah Perez, omni (at) zeriph (dot) com
*
* THIS SOFTWARE IS PROVIDED BY ZERIPH AND 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 ZERIPH AND 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.
*/
#if !defined(OMNI_SOCKET_HPP)
#define OMNI_SOCKET_HPP 1
#include <omni/types/net_t.hpp>
#include <omni/delegate/0.hpp>
#include <omni/delegate/1.hpp>
#include <omni/exception.hpp>
#include <string>
#include <csignal>
namespace omni {
namespace net {
// DEV_NOTE: While we could make socket a stream object, we would be overloading just about everything
// because of how we handle sockets vs. normal streams
class socket
{
socket(const omni::net::socket &cp); // = delete
public:
socket();
socket(omni::net::address_family family,
omni::net::socket_type type,
omni::net::protocol_type protocol);
virtual ~socket();
bool blocking;
bool dontFragment;
bool enableBroadcast;
bool exclusiveAddressUse;
omni::net::linger_option linger_state;
bool multicastLoopback;
bool nodelay;
int receiveBufferSize;
int receiveTimeout;
int sendBufferSize;
int sendTimeout;
short ttl;
bool useOnlyOverlappedIO;
omni::net::address_family address_family() const;
int available() const;
omni::net::socket& accept();
omni::net::socket& bind(std::string ip);
bool connected() const;
bool close();
bool close(int timeout);
bool connect(std::string host, int port);
bool disconnect(bool reuse);
unsigned int native_handle() const;
bool is_bound() const;
bool is_shut() const;
omni::net::socket& listen(int backlog);
bool poll(int microSeconds, omni::net::select_mode::enum_t mode);
omni::net::protocol_type::enum_t protocol() const;
omni::net::socket& set_socket_option(omni::net::socket_option_level::enum_t optionLevel, omni::net::socket_option_name::enum_t optionName, int optionValue);
bool shutdown(omni::net::socket_shutdown::enum_t how);
omni::net::socket_type::enum_t socket_type() const;;
const std::string to_string() const;
omni::net::socket *duplicateAndClose();
char *getSocketOption(omni::net::socket_option_level::enum_t optionLevel, omni::net::socket_option_name::enum_t optionName, int optionLength) const;
int iocontrol(int ioControlCode, char *optionInValue, char *optionOutValue);
const char *localEndPoint() const;
const char *remoteEndPoint() const;
int receive(char *buffer, int size);
int receive(char *buffer, int size, omni::net::socket_flags::enum_t socketFlags);
int receive(char *buffer, int offset, int size, omni::net::socket_flags::enum_t socketFlags);
int receive(char *buffer, int offset, int size, omni::net::socket_flags::enum_t socketFlags, omni::net::socket_error::enum_t &errorCode);
int receiveFrom(char *buffer, int size, omni::net::socket_flags::enum_t socketFlags, const char *remoteIP);
int receiveFrom(char *buffer, int offset, int size, omni::net::socket_flags::enum_t socketFlags, const char *remoteIP);
int send(const char *buffer, int size, omni::net::socket_flags::enum_t socketFlags);
int send(const char *buffer, int offset, int size, omni::net::socket_flags::enum_t socketFlags);
int send(const char *buffer, int offset, int size, omni::net::socket_flags::enum_t socketFlags, omni::net::socket_error::enum_t &errorCode);
int sendto(const char *buffer, int size, omni::net::socket_flags::enum_t socketFlags, const char *remoteIP);
int sendto(const char *buffer, int offset, int size, omni::net::socket_flags::enum_t socketFlags, const char *remoteIP);
// Events
omni::action received;
// Operators
omni::net::socket& operator= (const omni::net::socket &other);
omni::net::socket& operator<< (const char* data); // writes data to socket
omni::net::socket& operator>> (const char* read); // reads data from socket
/* examples for operator overload:
omni::net::socket s("192.168.1.1", 80);
...
s.connect();
if (s.connected()) {
if (isServer) {
const char *data;
s >> data; // wait until data read from socket, then put to 'data' (don't forget about mem leeks, etc)
printf("data rcvd: %s\n", data);
s << "hiya! how's it going?"; // showing operator<<(const char*)
} else { // client
// write to socket
std::string data = "hello there";
const char *rcvd;
s << data; // showing operator<<(std::string)
s >> rcvd;
printf("data rcvd: %s\n", rcvd);
}
}
*/
private:
// Methods
void p_Initialize(omni::net::address_family::enum_t addressFamily, omni::net::socket_type::enum_t socketType, omni::net::protocol_type::enum_t protocolType);
bool p_CloseSocket(int timeout, bool shutdown);
// Members
omni::net::address_family::enum_t m_AddressFamily;
int m_Available;
bool m_IsConnected;
bool m_IsBound;
bool m_IsShutdown;
int m_Handle;
const char *m_LocalEndPoint;
omni::net::protocol_type::enum_t m_ProtocolType;
const char *m_RemoteEndPoint;
omni::net::socket_type::enum_t m_SocketType;
};
}
}
#endif // OMNI_SOCKET_HPP