postal/tcp.h at main · simta/postal · GitHub

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