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