mirror of
https://github.com/yse/easy_profiler.git
synced 2024-12-31 02:49:13 +08:00
98 lines
2.2 KiB
C++
98 lines
2.2 KiB
C++
/**
|
|
Lightweight profiler library for c++
|
|
Copyright(C) 2016 Sergey Yagovtsev
|
|
|
|
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 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program 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 General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program.If not, see <http://www.gnu.org/licenses/>.
|
|
**/
|
|
#ifndef EASY________SOCKET_________H
|
|
#define EASY________SOCKET_________H
|
|
|
|
#include <stdint.h>
|
|
#include "profiler/profiler.h"
|
|
#ifndef _WIN32
|
|
#include <sys/types.h>
|
|
#include <sys/socket.h>
|
|
#include <netdb.h>
|
|
#include <stdio.h>
|
|
#include <unistd.h>
|
|
#include <fcntl.h>
|
|
#else
|
|
|
|
#define WIN32_LEAN_AND_MEAN
|
|
|
|
#include <winsock2.h>
|
|
#include <windows.h>
|
|
#include <ws2tcpip.h>
|
|
#include <stdlib.h>
|
|
#endif
|
|
|
|
class PROFILER_API EasySocket
|
|
{
|
|
public:
|
|
|
|
#ifdef _WIN32
|
|
typedef SOCKET socket_t;
|
|
#else
|
|
typedef int socket_t;
|
|
#endif
|
|
|
|
enum ConnectionState
|
|
{
|
|
CONNECTION_STATE_UNKNOWN,
|
|
CONNECTION_STATE_SUCCESS,
|
|
|
|
CONNECTION_STATE_DISCONNECTED
|
|
};
|
|
|
|
private:
|
|
|
|
void checkResult(int result);
|
|
bool checkSocket(socket_t s) const;
|
|
static int _close(socket_t s);
|
|
void setBlocking(socket_t s, bool blocking);
|
|
|
|
socket_t m_socket = 0;
|
|
socket_t m_replySocket = 0;
|
|
uint16_t m_port = 0;
|
|
|
|
int wsaret = -1;
|
|
|
|
struct hostent * server;
|
|
struct sockaddr_in serv_addr;
|
|
|
|
ConnectionState m_state = CONNECTION_STATE_UNKNOWN;
|
|
|
|
public:
|
|
|
|
EasySocket();
|
|
~EasySocket();
|
|
|
|
int send(const void *buf, size_t nbyte);
|
|
int receive(void *buf, size_t nbyte);
|
|
int listen(int count=5);
|
|
int accept();
|
|
int bind(uint16_t portno);
|
|
|
|
bool setAddress(const char* serv, uint16_t port);
|
|
int connect();
|
|
|
|
void flush();
|
|
void init();
|
|
|
|
void setState(ConnectionState state){m_state=state;}
|
|
ConnectionState state() const{return m_state;}
|
|
};
|
|
|
|
#endif // EASY________SOCKET_________H
|