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
|
#ifndef TCP_H
#define TCP_H
// Misnamed really. This class does all the common functionality between SMTP
// and POP clients.
using namespace std;
#include "postal.h"
#include <sys/poll.h>
#include <string>
#include "cmd5.h"
#include "thread.h"
#include <sys/types.h>
#include <netinet/in.h>
class Logit;
class address;
#ifdef USE_SSL
#ifdef USE_OPENSSL
#ifndef TCP_BODY
struct SSL_METHOD;
struct SSL_CTX;
struct SSL;
struct X509;
#else // TCP_BODY
#include <openssl/crypto.h>
#include <openssl/x509.h>
#include <openssl/pem.h>
#include <openssl/ssl.h>
#include <openssl/err.h>
#endif // TCP_BODY
#else // USE_OPENSSL
#include <gnutls/gnutls.h>
#endif // USE_OPENSSL
#endif // USE_SSL
class tcp : public Thread
{
public:
tcp(int *exitCount, const char *addr, unsigned short default_port, Logit *log
#ifdef USE_SSL
, int ssl
#endif
, const char *sourceAddr, Logit *debug);
tcp(int threadNum, const tcp *parent);
virtual ~tcp();
// connect returns 0 for connect, 1 for can't connect, and 2 for serious
// errors.
int Connect(short port = 0);
#ifdef USE_SSL
// after calling Connect() and getting high-level protocol ready call
// ConnectTLS() to start TLS.
int ConnectTLS();
#endif
int sendMsg();
virtual int disconnect();
int doAllWork(int rate);
protected:
virtual int pollRead() = 0;
virtual int WriteWork(PVOID buf, int size, int timeout) = 0;
int readLine(char *buf, int bufSize);
ERROR_TYPE sendData(CPCCHAR buf, int size);
ERROR_TYPE sendString(const string &s) { return sendData(s.c_str(), s.size()); }
void endIt();
virtual void error() = 0;
virtual void sentData(int bytes) = 0;
virtual void receivedData(int bytes) = 0;
Cmd5 m_md5;
virtual ERROR_TYPE readCommandResp(bool important = true) = 0;
virtual ERROR_TYPE sendCommandData(const char *buf, int size, bool important = true);
virtual ERROR_TYPE sendCommandString(const string &str, bool important = true);
int m_destAffinity;
Logit *m_log;
struct sockaddr_in m_connectionLocalAddr;
#ifdef USE_SSL
bool m_canTLS;
int m_useTLS;
#endif
int *m_exitCount;
private:
int m_fd;
pollfd m_poll;
char m_buf[4096];
int m_start;
int m_end;
bool m_open;
address *m_addr; // destination
address *m_sourceAddr;
// debug file management object (NULL if no debugging)
// if used a new Logit object is created for each instance unlike the
// m_log instance
Logit *m_debug;
#ifdef USE_SSL
#ifdef USE_OPENSSL
SSL_METHOD *m_sslMeth;
SSL_CTX* m_sslCtx;
SSL *m_ssl;
#else
gnutls_session_t *m_gnutls_session;
gnutls_anon_client_credentials_t m_gnutls_anoncred;
#endif
bool m_isTLS;
#endif
tcp(const tcp&);
tcp & operator=(const tcp&);
};
#endif
|
