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
|
/* A little hacked up program that allows a script to play the part of a remote
SMTP/LMTP server on stdin/stdout for testing purposes. Hacked from the more
complicated version that does it over a socket. */
/* ANSI C standard includes */
#include <ctype.h>
#include <signal.h>
#include <stdarg.h>
#include <stddef.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
/* Unix includes */
#include <errno.h>
#include <unistd.h>
#ifndef CS
# define CS (char *)
#endif
static FILE *log;
/*************************************************
* SIGALRM handler - crash out *
*************************************************/
static void
sigalrm_handler(int sig)
{
sig = sig; /* Keep picky compilers happy */
fprintf(log, "Server timed out\n");
exit(99);
}
/*************************************************
* Main Program *
*************************************************/
int main(int argc, char **argv)
{
char *logfile;
char *logmode = "w";
FILE *script;
unsigned char sbuffer[1024];
unsigned char ibuffer[1024];
if (argc < 3)
{
fprintf(stdout, "500 Script and log file required\n");
exit(1);
}
/* Get the script and log open */
script = fopen(argv[1], "r");
if (script == NULL)
{
fprintf(stdout, "500 Failed to open script %s: %s\r\n", argv[1],
strerror(errno));
exit(1);
}
logfile = argv[2];
if (logfile[0] == '+')
{
logfile++;
logmode = "a";
}
log = fopen(logfile, logmode);
if (log == NULL)
{
fprintf(stdout, "500 Failed to open log %s: %s\r\n", logfile,
strerror(errno));
exit(1);
}
/* SIGALRM handler crashes out */
signal(SIGALRM, sigalrm_handler);
/* Read the script, and do what it says. */
while (fgets(CS sbuffer, sizeof(sbuffer), script) != NULL)
{
int n = (int)strlen(CS sbuffer);
while (n > 0 && isspace(sbuffer[n-1])) n--;
sbuffer[n] = 0;
/* If the script line starts with a digit, it is a response line which
we are to send. */
if (isdigit(sbuffer[0]))
{
fprintf(log, "%s\n", sbuffer);
fflush(log);
fprintf(stdout, "%s\r\n", sbuffer);
fflush(stdout);
}
/* If the script line starts with "*sleep" we just sleep for a while
before continuing. Do not write this to the log, as it may not get
written at the right place in a log that's being shared. */
else if (strncmp(CS sbuffer, "*sleep ", 7) == 0)
{
sleep(atoi(CS sbuffer+7));
}
/* Otherwise the script line is the start of an input line we are expecting
from the client, or "*eof" indicating we expect the client to close the
connection. Read command line or data lines; the latter are indicated
by the expected line being just ".". */
else
{
int data = strcmp(CS sbuffer, ".") == 0;
fprintf(log, "%s\n", sbuffer);
fflush(log);
/* Loop for multiple data lines */
for (;;)
{
int n;
alarm(5);
if (fgets(CS ibuffer, sizeof(ibuffer), stdin) == NULL)
{
fprintf(log, "%sxpected EOF read from client\n",
(strncmp(CS sbuffer, "*eof", 4) == 0)? "E" : "Une");
goto END_OFF;
}
alarm(0);
n = (int)strlen(CS ibuffer);
while (n > 0 && isspace(ibuffer[n-1])) n--;
ibuffer[n] = 0;
fprintf(log, "<<< %s\n", ibuffer);
if (!data || strcmp(CS ibuffer, ".") == 0) break;
}
/* Check received what was expected */
if (strncmp(CS sbuffer, CS ibuffer, (int)strlen(CS sbuffer)) != 0)
{
fprintf(log, "Comparison failed - bailing out\n");
goto END_OFF;
}
}
}
/* This could appear in the wrong place in a shared log, so forgo it. */
/* fprintf(log, "End of script\n"); */
END_OFF:
fclose(script);
fclose(log);
exit(0);
}
/* End of mtpscript.c */
|
