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
|
#ifndef SMTP_H
#define SMTP_H
using namespace std;
#include <string>
#include <cstring>
#include <time.h>
#include "conf.h"
#ifdef HAVE_EXT_HASH_MAP
using namespace __gnu_cxx;
#include <ext/hash_map>
#else
#include <hash_map.h>
#endif
#include "tcp.h"
#include "mutex.h"
class results;
class UserList;
#define MAP_SIZE 8 * 1024
// Comparison operator for hash map of names to unsigned longs
struct eqlng
{
bool operator()(unsigned long l1, unsigned long l2) const
{
return (l1 == l2);
}
};
typedef hash_map<unsigned long, string *, hash<unsigned long>, eqlng> NAME_MAP;
class smtpData
{
public:
smtpData();
~smtpData();
const string &quit() const { return m_quit; }
// create a random string of length len that includes a trailing "\r\n"
// string is not zero terminated
void randomString(char *buf, int len) const;
// fill a buffer with random lines of text
void randomBuf(char *buf, int len) const;
void date(char *buf) const;
const string msgId(const char *sender, const unsigned threadNum) const;
// return the X-Postal lines
const string &postalMsg() const { return m_postalMsg; }
// frequency determines how long it should have been since the last
// operation for this function to actually do anything.
// It will not sleep for that many seconds or block on IO in any way.
void setRand(int frequency);
// get the mail name for an IP address
const string *getMailName(struct sockaddr_in &in);
private:
// Some const strings that we only want one copy of
const string m_quit;
const char *m_randomLetters;
const int m_randomLen;
const string m_postalMsg;
Mutex m_dnsLock;
// time of the last randomise
time_t m_timeLastAction;
char m_randBuf[MAP_SIZE];
// Map of IP addresses to names
NAME_MAP m_names;
smtpData(const smtpData&);
smtpData & operator=(const smtpData&);
};
class smtp : public tcp
{
public:
smtp(int *exitCount, const char *addr, const char *ourAddr
, UserList &ul, UserList *senderList, int minMsgSize, int maxMsgSize
, int numMsgsPerConnection, int processes, Logit *log, TRISTATE netscape
, bool useLMTP
#ifdef USE_SSL
, int ssl
#endif
, unsigned short port, Logit *debug);
virtual ~smtp();
// Connect returns 0 for connect, 1 for can't connect, and 2 for serious
// errors.
int Connect();
int sendMsg();
virtual int disconnect();
int msgsPerConnection() const { return m_msgsPerConnection; }
private:
virtual int action(PVOID);
smtp(int threadNum, const smtp *parent);
virtual Thread *newThread(int threadNum);
int pollRead();
virtual int WriteWork(PVOID buf, int size, int timeout);
virtual ERROR_TYPE readCommandResp(bool) { return readCommandResp(); }
ERROR_TYPE readCommandResp();
void error();
virtual void sentData(int bytes);
virtual void receivedData(int);
UserList &m_ul, *m_senderList;
const int m_minMsgSize, m_maxMsgSize;
smtpData *m_data;
int m_msgsPerConnection;
results *m_res;
TRISTATE m_netscape;
string m_helo;
time_t m_nextPrint;
bool m_useLMTP;
smtp(const smtp&);
smtp & operator=(const smtp&);
};
#endif
|
