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
|
#ifndef message_h
#define message_h
#include "rfc2045/rfc2045.h"
//////////////////////////////////////////////////////////////////////////////
//
// A Message object represents the message being filtered. It is expected
// that the message is going to be scanned, repetitively, from start to
// finish, in one direction. Therefore, access to the message contents is
// sequential.
//
// A Message object can be initiated in two ways. It can be initiated from
// a file. The Message object will check if the file is seekable. If so,
// the Message object will simply use the Mio class to access the file.
// If the file is not seekable, the message will be saved in a temporary
// file, and the Mio class will be used to access the temporary file.
//
// However, if the message size turns out to be smaller the SMALLMSG bytes,
// the message is saved in a memory buffer, and accessed from there, for
// speed. We presume that most messages are less than SMALLMSG bytes long.
//
// The second way is for the messages contents to be manually specified.
// For this initialization method, first the message is saved into the
// memory buffer. If it proves to be too small, the message gets written
// out into a temporary file.
//
//////////////////////////////////////////////////////////////////////////////
#include "config.h"
#include "mio.h"
#include "tempfile.h"
#include <sys/types.h>
#include <errno.h>
#include <string.h>
#if HAVE_UNISTD_H
#include <unistd.h>
#else
#ifndef SEEK_SET
#define SEEK_SET 0
#endif
#endif
class Buffer;
class Message {
Mio mio;
TempFile tempfile;
char *buffer;
char *bufptr;
char *extra_headers;
char *extra_headersptr;
off_t msgsize;
off_t msglines;
static void readerr();
static void seekerr();
public:
Message();
~Message();
void Init(int, const Buffer &extra_headers);
// Initialize from file descriptor
void Init(); // Begin initializing externally
void Init(const void *, unsigned); // From existing contents.
void ExtraHeaders(const Buffer &);
void Rewind(); // Start reading the message
void RewindIgnore(); // Rewind, ignore msginfo
int appendline(Buffer &, int=1); // Read newline-terminated line.
void seek(off_t);
off_t tell();
int get_c();
int peek();
off_t MessageSize();
off_t MessageLines() { return (msglines); }
void setmsgsize();
// API translator for rfc2045 functions
struct rfc2045src rfc2045src_parser;
struct rfc2045 *rfc2045p;
} ;
#include "funcs.h"
#include "maildrop.h"
inline int Message::peek() // Current character.
{
if (extra_headersptr)
return ( (unsigned char) *extra_headersptr );
if (mio.fd() >= 0)
{
errno=0;
int c=mio.peek();
if (c < 0 && errno) readerr();
return (c);
}
if (bufptr >= buffer + msgsize) return (-1);
return ( (int)(unsigned char)*bufptr );
}
inline int Message::get_c() // Get character.
{
int c;
if (extra_headersptr)
{
c= (unsigned char) *extra_headersptr++;
if (!*extra_headersptr) extra_headersptr=0;
return (c);
}
if (mio.fd() >= 0)
{
errno=0;
c=mio.get();
if (c < 0 && errno) readerr();
return (c);
}
if (bufptr >= buffer + msgsize) return (-1);
return ( (int)(unsigned char)*bufptr++ );
}
inline off_t Message::tell()
{
off_t pos;
if ( mio.fd() < 0)
pos=bufptr - buffer;
else
{
pos=mio.tell();
if (pos == -1) seekerr();
}
if (extra_headersptr)
pos += extra_headersptr-extra_headers;
else
{
if (extra_headers) pos += strlen(extra_headers);
}
return (pos);
}
inline void Message::seek(off_t n)
{
int l=0;
if (extra_headers && n < (l=strlen(extra_headers)))
{
extra_headersptr= extra_headers + n;
n=0;
}
else
{
extra_headersptr=0;
n -= l;
}
if (mio.fd() >= 0)
{
if (mio.seek(n, SEEK_SET) < 0) seekerr();
}
else
{
if (n > msgsize) n=msgsize;
bufptr=buffer+n;
}
}
inline off_t Message::MessageSize()
{
off_t s=msgsize;
if (extra_headers)
s += strlen(extra_headers);
return (s);
}
#endif
|
