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
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
|
/*
* Copyright (C) 2001-2012 Jacek Sieka, arnetheduck on gmail point com
*
* 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.
*
* 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 <https://www.gnu.org/licenses/>.
*/
#pragma once
#ifdef _WIN32
#include "w.h"
typedef int socklen_t;
typedef SOCKET socket_t;
#else
#include <sys/ioctl.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>
#include <fcntl.h>
#include <errno.h>
typedef int socket_t;
const int INVALID_SOCKET = -1;
#define SOCKET_ERROR -1
#endif
#include "GetSet.h"
#include "Util.h"
#include "Exception.h"
namespace dcpp {
class SocketException : public Exception {
public:
#ifdef _DEBUG
SocketException(const string& aError) noexcept : Exception("SocketException: " + aError) { }
#else //_DEBUG
SocketException(const string& aError) noexcept : Exception(aError) { }
#endif // _DEBUG
SocketException(int aError) noexcept;
virtual ~SocketException() throw() { }
private:
static string errorToString(int aError) noexcept;
};
class Socket
{
public:
enum {
WAIT_NONE = 0x00,
WAIT_CONNECT = 0x01,
WAIT_READ = 0x02,
WAIT_WRITE = 0x04
};
enum {
TYPE_TCP,
TYPE_UDP
};
enum Protocol {
PROTO_DEFAULT = 0,
PROTO_NMDC = 1,
PROTO_ADC = 2
};
Socket() : sock(INVALID_SOCKET), type(TYPE_TCP), connected(false), proto(PROTO_DEFAULT) { }
Socket(const string& aIp, const string& aPort) : sock(INVALID_SOCKET), type(TYPE_TCP), connected(false), proto(PROTO_DEFAULT) { connect(aIp, aPort); }
virtual ~Socket() { disconnect(); }
/**
* Connects a socket to an address/ip, closing any other connections made with
* this instance.
* @param aAddr Server address, in dns or xxx.xxx.xxx.xxx format.
* @param aPort Server port.
* @throw SocketException If any connection error occurs.
*/
virtual void connect(const string& aIp, const string &aPort, const string &localPort = Util::emptyString);
/**
* Same as connect(), but through the SOCKS5 server
*/
void socksConnect(const string& aIp, const string &aPort, uint32_t timeout = 0);
/**
* Sends data, will block until all data has been sent or an exception occurs
* @param aBuffer Buffer with data
* @param aLen Data length
* @throw SocketExcpetion Send failed.
*/
void writeAll(const void* aBuffer, int aLen, uint32_t timeout = 0);
virtual int write(const void* aBuffer, int aLen);
int write(const string& aData) { return write(aData.data(), (int)aData.length()); }
virtual void writeTo(const string& aIp, const std::string &aPort, const void* aBuffer, int aLen, bool proxy = true);
void writeTo(const string& aIp, const string& aPort, const string& aData) { writeTo(aIp, aPort, aData.data(), (int)aData.length()); }
virtual void shutdown() noexcept;
virtual void close() noexcept;
void disconnect() noexcept;
virtual bool waitConnected(uint32_t millis);
virtual bool waitAccepted(uint32_t millis);
/**
* Reads zero to aBufLen characters from this socket,
* @param aBuffer A buffer to store the data in.
* @param aBufLen Size of the buffer.
* @return Number of bytes read, 0 if disconnected and -1 if the call would block.
* @throw SocketException On any failure.
*/
virtual int read(void* aBuffer, int aBufLen);
/**
* Reads zero to aBufLen characters from this socket,
* @param aBuffer A buffer to store the data in.
* @param aBufLen Size of the buffer.
* @param aIP Remote IP address
* @return Number of bytes read, 0 if disconnected and -1 if the call would block.
* @throw SocketException On any failure.
*/
virtual int read(void* aBuffer, int aBufLen, sockaddr_in& remote);
/**
* Reads data until aBufLen bytes have been read or an error occurs.
* If the socket is closed, or the timeout is reached, the number of bytes read
* actually read is returned.
* On exception, an unspecified amount of bytes might have already been read.
*/
int readAll(void* aBuffer, int aBufLen, uint32_t timeout = 0);
virtual int wait(uint32_t millis, int waitFor);
bool isConnected() { return connected; }
static string resolve(const string& aDns);
static uint64_t getTotalDown() { return stats.totalDown; }
static uint64_t getTotalUp() { return stats.totalUp; }
void setBlocking(bool block) noexcept;
string getLocalIp() noexcept;
string getLocalPort() noexcept;
Protocol getNextProtocol() noexcept;
// Low level interface
virtual void create(int aType = TYPE_TCP);
/** Binds a socket to a certain local port and possibly IP. */
virtual const string bind(const string &aPort = Util::emptyString, const string& aIp = "0.0.0.0");
virtual void listen();
virtual void accept(const Socket& listeningSocket);
int getSocketOptInt(int option);
void setSocketOpt(int option, int value);
virtual bool isSecure() const noexcept { return false; }
virtual bool isTrusted() const noexcept { return false; }
virtual string getCipherName() const noexcept { return Util::emptyString; }
virtual ByteVector getKeyprint() const noexcept { return ByteVector(); }
/** When socks settings are updated, this has to be called... */
static void socksUpdated();
string getIfaceI4 (const string &iface);
GETSET(string, ip, Ip);
socket_t sock;
protected:
int type;
bool connected;
Protocol proto;
class Stats {
public:
uint64_t totalDown;
uint64_t totalUp;
};
static Stats stats;
static string udpServer;
static string udpPort;
private:
Socket(const Socket&);
Socket& operator=(const Socket&);
void socksAuth(uint32_t timeout);
static int getLastError();
static int checksocket(int ret);
static int check(int ret, bool blockOk = false);
};
} // namespace dcpp
|
