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
|
/* Copyright (c) 2005 MySQL AB
This library is free software; you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public
License as published by the Free Software Foundation; either
version 2 of the License, or (at your option) any later version.
This library is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
Lesser General Public License for more details.
You should have received a copy of the GNU Lesser General Public
License along with this library; if not, write to the
Free Software Foundation, Inc., 59 Temple Place - Suite 330,
Boston, MA 02111-1307, USA.
*/
#ifndef _MNSOCKET_H_
#define _MNSOCKET_H_
#if defined(__WIN__) || defined(_WIN32) || defined(_WIN64)
#define MYX_SOCKET SOCKET
#include <winsock2.h>
#else
# define MYX_SOCKET int
#endif
#include <string>
#include <glib.h>
typedef enum {
MNSDisconnected,
MNSResolveError,
MNSConnectError,
MNSReadError,
MNSSendError,
MNSConnected
} MNSocketState;
class MNSocket {
protected:
MYX_SOCKET _socket;
std::string _host;
int _port;
char *_buffer;
size_t _buffer_used;
size_t _buffer_size;
size_t _next_offset;
MNSocketState _state;
void fixup_offset();
public:
MNSocket();
virtual ~MNSocket();
static void initialize();
void wrap_socket(MYX_SOCKET socket, const std::string &host, int port);
bool connect(const std::string &host, int port);
void disconnect();
void send_line(const std::string &line);
std::string get_line();
std::string get_zstring();
std::string get_block(int sep);
void send_data(const char *data, size_t size);
bool get_data(const char *&data, size_t &size);
bool is_connected() const;
inline MNSocketState get_state() const { return _state; };
std::string get_host() { return _host; };
};
#endif /* _MNSOCKET_H_ */
|