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
|
/*
* 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
#include "typedefs.h"
#include "Socket.h"
#include "Singleton.h"
#include "SSL.h"
#ifndef SSL_SUCCESS
#define SSL_SUCCESS 1
#endif
namespace dcpp {
using std::unique_ptr;
using std::string;
class SSLSocketException : public SocketException
{
public:
#ifdef _DEBUG
SSLSocketException(const string& aError) noexcept : SocketException("SSLSocketException: " + aError) { }
#else //_DEBUG
SSLSocketException(const string& aError) noexcept : SocketException(aError) { }
#endif // _DEBUG
SSLSocketException(int aError) noexcept : SocketException(aError) { }
virtual ~SSLSocketException() throw() { }
};
class CryptoManager;
class SSLSocket : public Socket
{
public:
virtual ~SSLSocket() { }
virtual void accept(const Socket& listeningSocket);
virtual void connect(const string& aIp, const string &aPort);
virtual int read(void* aBuffer, int aBufLen);
virtual int write(const void* aBuffer, int aLen);
virtual int wait(uint32_t millis, int waitFor);
virtual void shutdown() noexcept;
virtual void close() noexcept;
virtual bool isSecure() const noexcept { return true; }
virtual bool isTrusted() const noexcept;
virtual string getCipherName() const noexcept;
virtual ByteVector getKeyprint() const noexcept;
virtual bool waitConnected(uint32_t millis);
virtual bool waitAccepted(uint32_t millis);
private:
friend class CryptoManager;
SSLSocket(SSL_CTX* context, Socket::Protocol proto);
SSLSocket(const SSLSocket&);
SSLSocket& operator=(const SSLSocket&);
SSL_CTX* ctx;
ssl::SSL ssl;
Socket::Protocol nextProto;
int checkSSL(int ret);
bool waitWant(int ret, uint32_t millis);
};
} // namespace dcpp
|
