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 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146
|
//////////////////////////////////////////////////////////////////////////////
//////////////////////////////////////////////////////////////////////////////
//
// copyright : (C) 2014 The CodeLite Team
// file name : clSocketBase.h
//
// -------------------------------------------------------------------------
// A
// _____ _ _ _ _
// / __ \ | | | | (_) |
// | / \/ ___ __| | ___| | _| |_ ___
// | | / _ \ / _ |/ _ \ | | | __/ _ )
// | \__/\ (_) | (_| | __/ |___| | || __/
// \____/\___/ \__,_|\___\_____/_|\__\___|
//
// F i l e
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or
// (at your option) any later version.
//
//////////////////////////////////////////////////////////////////////////////
//////////////////////////////////////////////////////////////////////////////
#ifndef CLSOCKETBASE_H
#define CLSOCKETBASE_H
#include <string>
#include <wx/sharedptr.h>
#include <wx/string.h>
#include "codelite_exports.h"
#ifdef _WIN32
#include <winsock2.h>
typedef SOCKET socket_t;
typedef int socklen_t;
#else
typedef int socket_t;
#define INVALID_SOCKET -1
#endif
class WXDLLIMPEXP_CL clSocketException
{
std::string m_what;
public:
clSocketException(const std::string & what ) : m_what(what) {}
~clSocketException() {}
const std::string& what() const {
return m_what;
}
};
class WXDLLIMPEXP_CL clSocketBase
{
protected:
socket_t m_socket;
bool m_closeOnExit;
public:
typedef wxSharedPtr<clSocketBase> Ptr_t;
enum {
kSuccess = 1,
kTimeout = 2,
kError = 3,
};
std::string error() const;
public:
/**
* @brief set the socket into blocking/non-blocking mode
* @param blocking
*/
void MakeSocketBlocking(bool blocking);
clSocketBase(int sockfd = INVALID_SOCKET);
virtual ~clSocketBase();
void SetCloseOnExit(bool closeOnExit) {
this->m_closeOnExit = closeOnExit;
}
bool IsCloseOnExit() const {
return m_closeOnExit;
}
/**
* @brief return the descriptor and clear this socket.
*/
socket_t Release() ;
/**
* @brief initialize the socket library
*/
static void Initialize();
socket_t GetSocket() const {
return m_socket;
}
/**
* @brief
* @param msg
*/
void Send(const std::string& msg) throw (clSocketException);
/**
* @brief
* @return
*/
int Read(char* buffer, size_t bufferSize, size_t& bytesRead, long timeout = -1) throw (clSocketException);
/**
* @brief
* @param seconds
* @return
*/
int SelectRead(long seconds = -1) throw (clSocketException);
/**
* @brief select for write
* @return
*/
int SelectWrite(long seconds = -1) throw (clSocketException);
/**
* @brief read a full message that was sent with 'SendMessage'
* @param message
* @param timeout seconds to wait
* @return kSuccess, kTimeout or kError
*/
int ReadMessage(wxString &message, int timeout) throw (clSocketException);
/**
* @brief write a full message
* @param message
*/
void WriteMessage(const wxString &message) throw (clSocketException);
protected:
/**
* @brief
*/
void DestroySocket();
};
#endif // CLSOCKETBASE_H
|