Documentation: Added changebars, fixed date
[users/heiko/exim.git] / test / src / mtpscript.c
1 /* $Cambridge: exim/test/src/mtpscript.c,v 1.1 2006/02/06 16:24:05 ph10 Exp $ */
2
3 /* A little hacked up program that allows a script to play the part of a remote
4 SMTP/LMTP server on stdin/stdout for testing purposes. Hacked from the more
5 complicated version that does it over a socket. */
6
7
8 /* ANSI C standard includes */
9
10 #include <ctype.h>
11 #include <signal.h>
12 #include <stdarg.h>
13 #include <stddef.h>
14 #include <stdio.h>
15 #include <stdlib.h>
16 #include <string.h>
17 #include <time.h>
18
19 /* Unix includes */
20
21 #include <errno.h>
22 #include <unistd.h>
23
24
25 static FILE *log;
26
27
28 /*************************************************
29 *            SIGALRM handler - crash out         *
30 *************************************************/
31
32 static void
33 sigalrm_handler(int sig)
34 {
35 sig = sig;    /* Keep picky compilers happy */
36 fprintf(log, "Server timed out\n");
37 exit(99);
38 }
39
40
41
42 /*************************************************
43 *                 Main Program                   *
44 *************************************************/
45
46 int main(int argc, char **argv)
47 {
48 char *logfile;
49 char *logmode = "w";
50 FILE *script;
51 unsigned char sbuffer[1024];
52 unsigned char ibuffer[1024];
53
54 if (argc < 3)
55   {
56   fprintf(stdout, "500 Script and log file required\n");
57   exit(1);
58   }
59
60 /* Get the script and log open */
61
62 script = fopen(argv[1], "r");
63 if (script == NULL)
64   {
65   fprintf(stdout, "500 Failed to open script %s: %s\r\n", argv[1],
66     strerror(errno));
67   exit(1);
68   }
69
70 logfile = argv[2];
71 if (logfile[0] == '+')
72   {
73   logfile++;
74   logmode = "a";
75   }
76
77 log = fopen(logfile, logmode);
78 if (log == NULL)
79   {
80   fprintf(stdout, "500 Failed to open log %s: %s\r\n", logfile,
81     strerror(errno));
82   exit(1);
83   }
84
85 /* SIGALRM handler crashes out */
86
87 signal(SIGALRM, sigalrm_handler);
88
89 /* Read the script, and do what it says. */
90
91 while (fgets(sbuffer, sizeof(sbuffer), script) != NULL)
92   {
93   int n = (int)strlen(sbuffer);
94   while (n > 0 && isspace(sbuffer[n-1])) n--;
95   sbuffer[n] = 0;
96
97   /* If the script line starts with a digit, it is a response line which
98   we are to send. */
99
100   if (isdigit(sbuffer[0]))
101     {
102     fprintf(log, "%s\n", sbuffer);
103     fflush(log);
104     fprintf(stdout, "%s\r\n", sbuffer);
105     fflush(stdout);
106     }
107
108   /* If the script line starts with "*sleep" we just sleep for a while
109   before continuing. Do not write this to the log, as it may not get
110   written at the right place in a log that's being shared. */
111
112   else if (strncmp(sbuffer, "*sleep ", 7) == 0)
113     {
114     sleep(atoi(sbuffer+7));
115     }
116
117   /* Otherwise the script line is the start of an input line we are expecting
118   from the client, or "*eof" indicating we expect the client to close the
119   connection. Read command line or data lines; the latter are indicated
120   by the expected line being just ".". */
121
122   else
123     {
124     int data = strcmp(sbuffer, ".") == 0;
125
126     fprintf(log, "%s\n", sbuffer);
127     fflush(log);
128
129     /* Loop for multiple data lines */
130
131     for (;;)
132       {
133       int n;
134       alarm(5);
135       if (fgets(ibuffer, sizeof(ibuffer), stdin) == NULL)
136         {
137         fprintf(log, "%sxpected EOF read from client\n",
138           (strncmp(sbuffer, "*eof", 4) == 0)? "E" : "Une");
139         goto END_OFF;
140         }
141       alarm(0);
142       n = (int)strlen(ibuffer);
143       while (n > 0 && isspace(ibuffer[n-1])) n--;
144       ibuffer[n] = 0;
145       fprintf(log, "<<< %s\n", ibuffer);
146       if (!data || strcmp(ibuffer, ".") == 0) break;
147       }
148
149     /* Check received what was expected */
150
151     if (strncmp(sbuffer, ibuffer, (int)strlen(sbuffer)) != 0)
152       {
153       fprintf(log, "Comparison failed - bailing out\n");
154       goto END_OFF;
155       }
156     }
157   }
158
159 /* This could appear in the wrong place in a shared log, so forgo it. */
160 /* fprintf(log, "End of script\n"); */
161
162 END_OFF:
163 fclose(script);
164 fclose(log);
165
166 exit(0);
167 }
168
169 /* End of mtpscript.c */