Docs: Remove GNUmake idioms from Makefile, give power to "mv"
[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 #ifndef CS
23 # define CS (char *)
24 #endif
25
26
27 static FILE *log;
28
29
30 /*************************************************
31 *            SIGALRM handler - crash out         *
32 *************************************************/
33
34 static void
35 sigalrm_handler(int sig)
36 {
37 sig = sig;    /* Keep picky compilers happy */
38 fprintf(log, "Server timed out\n");
39 exit(99);
40 }
41
42
43
44 /*************************************************
45 *                 Main Program                   *
46 *************************************************/
47
48 int main(int argc, char **argv)
49 {
50 char *logfile;
51 char *logmode = "w";
52 FILE *script;
53 unsigned char sbuffer[1024];
54 unsigned char ibuffer[1024];
55
56 if (argc < 3)
57   {
58   fprintf(stdout, "500 Script and log file required\n");
59   exit(1);
60   }
61
62 /* Get the script and log open */
63
64 script = fopen(argv[1], "r");
65 if (script == NULL)
66   {
67   fprintf(stdout, "500 Failed to open script %s: %s\r\n", argv[1],
68     strerror(errno));
69   exit(1);
70   }
71
72 logfile = argv[2];
73 if (logfile[0] == '+')
74   {
75   logfile++;
76   logmode = "a";
77   }
78
79 log = fopen(logfile, logmode);
80 if (log == NULL)
81   {
82   fprintf(stdout, "500 Failed to open log %s: %s\r\n", logfile,
83     strerror(errno));
84   exit(1);
85   }
86
87 /* SIGALRM handler crashes out */
88
89 signal(SIGALRM, sigalrm_handler);
90
91 /* Read the script, and do what it says. */
92
93 while (fgets(CS sbuffer, sizeof(sbuffer), script) != NULL)
94   {
95   int n = (int)strlen(CS sbuffer);
96   while (n > 0 && isspace(sbuffer[n-1])) n--;
97   sbuffer[n] = 0;
98
99   /* If the script line starts with a digit, it is a response line which
100   we are to send. */
101
102   if (isdigit(sbuffer[0]))
103     {
104     fprintf(log, "%s\n", sbuffer);
105     fflush(log);
106     fprintf(stdout, "%s\r\n", sbuffer);
107     fflush(stdout);
108     }
109
110   /* If the script line starts with "*sleep" we just sleep for a while
111   before continuing. Do not write this to the log, as it may not get
112   written at the right place in a log that's being shared. */
113
114   else if (strncmp(CS sbuffer, "*sleep ", 7) == 0)
115     {
116     sleep(atoi(CS sbuffer+7));
117     }
118
119   /* Otherwise the script line is the start of an input line we are expecting
120   from the client, or "*eof" indicating we expect the client to close the
121   connection. Read command line or data lines; the latter are indicated
122   by the expected line being just ".". */
123
124   else
125     {
126     int data = strcmp(CS sbuffer, ".") == 0;
127
128     fprintf(log, "%s\n", sbuffer);
129     fflush(log);
130
131     /* Loop for multiple data lines */
132
133     for (;;)
134       {
135       int n;
136       alarm(5);
137       if (fgets(CS ibuffer, sizeof(ibuffer), stdin) == NULL)
138         {
139         fprintf(log, "%sxpected EOF read from client\n",
140           (strncmp(CS sbuffer, "*eof", 4) == 0)? "E" : "Une");
141         goto END_OFF;
142         }
143       alarm(0);
144       n = (int)strlen(CS ibuffer);
145       while (n > 0 && isspace(ibuffer[n-1])) n--;
146       ibuffer[n] = 0;
147       fprintf(log, "<<< %s\n", ibuffer);
148       if (!data || strcmp(CS ibuffer, ".") == 0) break;
149       }
150
151     /* Check received what was expected */
152
153     if (strncmp(CS sbuffer, CS ibuffer, (int)strlen(CS sbuffer)) != 0)
154       {
155       fprintf(log, "Comparison failed - bailing out\n");
156       goto END_OFF;
157       }
158     }
159   }
160
161 /* This could appear in the wrong place in a shared log, so forgo it. */
162 /* fprintf(log, "End of script\n"); */
163
164 END_OFF:
165 fclose(script);
166 fclose(log);
167
168 exit(0);
169 }
170
171 /* End of mtpscript.c */