1 /*
2 * Copyright (c) 1999-2003, 2006, 2007 Proofpoint, Inc. and its suppliers.
3 * All rights reserved.
4 *
5 * By using this file, you agree to the terms and conditions set
6 * forth in the LICENSE file which can be found at the top level of
7 * the sendmail distribution.
8 *
9 */
10
11 #include <sm/gen.h>
12 SM_RCSID("@(#)$Id: main.c,v 8.85 2013-11-22 20:51:36 ca Exp $")
13
14 #define _DEFINE 1
15 #include "libmilter.h"
16 #include <fcntl.h>
17 #include <sys/stat.h>
18
19
20 static smfiDesc_ptr smfi = NULL;
21
22 /*
23 ** SMFI_REGISTER -- register a filter description
24 **
25 ** Parameters:
26 ** smfilter -- description of filter to register
27 **
28 ** Returns:
29 ** MI_SUCCESS/MI_FAILURE
30 */
31
32 int
smfi_register(smfilter)33 smfi_register(smfilter)
34 smfiDesc_str smfilter;
35 {
36 size_t len;
37
38 if (smfi == NULL)
39 {
40 smfi = (smfiDesc_ptr) malloc(sizeof *smfi);
41 if (smfi == NULL)
42 return MI_FAILURE;
43 }
44 (void) memcpy(smfi, &smfilter, sizeof *smfi);
45 if (smfilter.xxfi_name == NULL)
46 smfilter.xxfi_name = "Unknown";
47
48 len = strlen(smfilter.xxfi_name) + 1;
49 smfi->xxfi_name = (char *) malloc(len);
50 if (smfi->xxfi_name == NULL)
51 return MI_FAILURE;
52 (void) sm_strlcpy(smfi->xxfi_name, smfilter.xxfi_name, len);
53
54 /* compare milter version with hard coded version */
55 if ((SM_LM_VRS_MAJOR(smfi->xxfi_version) != SM_LM_VRS_MAJOR(SMFI_VERSION) ||
56 SM_LM_VRS_MINOR(smfi->xxfi_version) != SM_LM_VRS_MINOR(SMFI_VERSION)) &&
57 smfi->xxfi_version != 2 &&
58 smfi->xxfi_version != 3 &&
59 smfi->xxfi_version != 4)
60 {
61 /* hard failure for now! */
62 smi_log(SMI_LOG_ERR,
63 "%s: smfi_register: version mismatch application: %d != milter: %d",
64 smfi->xxfi_name, smfi->xxfi_version,
65 (int) SMFI_VERSION);
66
67 /* XXX how about smfi? */
68 free(smfi->xxfi_name);
69 return MI_FAILURE;
70 }
71
72 return MI_SUCCESS;
73 }
74
75 /*
76 ** SMFI_STOP -- stop milter
77 **
78 ** Parameters:
79 ** none.
80 **
81 ** Returns:
82 ** success.
83 */
84
85 int
smfi_stop()86 smfi_stop()
87 {
88 mi_stop_milters(MILTER_STOP);
89 return MI_SUCCESS;
90 }
91
92 /*
93 ** Default values for some variables.
94 ** Most of these can be changed with the functions below.
95 */
96
97 static int dbg = 0;
98 static char *conn = NULL;
99 static int timeout = MI_TIMEOUT;
100 static int backlog = MI_SOMAXCONN;
101
102 /*
103 ** SMFI_OPENSOCKET -- try the socket setup to make sure we'll be
104 ** able to start up
105 **
106 ** Parameters:
107 ** rmsocket -- if true, instructs libmilter to attempt
108 ** to remove the socket before creating it;
109 ** only applies for "local:" or "unix:" sockets
110 **
111 ** Return:
112 ** MI_SUCCESS/MI_FAILURE
113 */
114
115 int
smfi_opensocket(rmsocket)116 smfi_opensocket(rmsocket)
117 bool rmsocket;
118 {
119 if (smfi == NULL || conn == NULL)
120 return MI_FAILURE;
121
122 return mi_opensocket(conn, backlog, dbg, rmsocket, smfi);
123 }
124
125 /*
126 ** SMFI_SETDBG -- set debug level.
127 **
128 ** Parameters:
129 ** odbg -- new debug level.
130 **
131 ** Returns:
132 ** MI_SUCCESS
133 */
134
135 int
smfi_setdbg(odbg)136 smfi_setdbg(odbg)
137 int odbg;
138 {
139 dbg = odbg;
140 return MI_SUCCESS;
141 }
142
143 /*
144 ** SMFI_SETTIMEOUT -- set timeout (for read/write).
145 **
146 ** Parameters:
147 ** otimeout -- new timeout.
148 **
149 ** Returns:
150 ** MI_SUCCESS
151 */
152
153 int
smfi_settimeout(otimeout)154 smfi_settimeout(otimeout)
155 int otimeout;
156 {
157 timeout = otimeout;
158 return MI_SUCCESS;
159 }
160
161 /*
162 ** SMFI_SETCONN -- set connection information (socket description)
163 **
164 ** Parameters:
165 ** oconn -- new connection information.
166 **
167 ** Returns:
168 ** MI_SUCCESS/MI_FAILURE
169 */
170
171 int
smfi_setconn(oconn)172 smfi_setconn(oconn)
173 char *oconn;
174 {
175 size_t l;
176
177 if (oconn == NULL || *oconn == '\0')
178 return MI_FAILURE;
179 l = strlen(oconn) + 1;
180 if ((conn = (char *) malloc(l)) == NULL)
181 return MI_FAILURE;
182 if (sm_strlcpy(conn, oconn, l) >= l)
183 return MI_FAILURE;
184 return MI_SUCCESS;
185 }
186
187 /*
188 ** SMFI_SETBACKLOG -- set backlog
189 **
190 ** Parameters:
191 ** obacklog -- new backlog.
192 **
193 ** Returns:
194 ** MI_SUCCESS/MI_FAILURE
195 */
196
197 int
smfi_setbacklog(obacklog)198 smfi_setbacklog(obacklog)
199 int obacklog;
200 {
201 if (obacklog <= 0)
202 return MI_FAILURE;
203 backlog = obacklog;
204 return MI_SUCCESS;
205 }
206
207
208 /*
209 ** SMFI_MAIN -- setup milter connection and start listener.
210 **
211 ** Parameters:
212 ** none.
213 **
214 ** Returns:
215 ** MI_SUCCESS/MI_FAILURE
216 */
217
218 int
smfi_main()219 smfi_main()
220 {
221 int r;
222
223 (void) signal(SIGPIPE, SIG_IGN);
224 if (conn == NULL)
225 {
226 smi_log(SMI_LOG_FATAL, "%s: missing connection information",
227 smfi->xxfi_name);
228 return MI_FAILURE;
229 }
230
231 (void) atexit(mi_clean_signals);
232 if (mi_control_startup(smfi->xxfi_name) != MI_SUCCESS)
233 {
234 smi_log(SMI_LOG_FATAL,
235 "%s: Couldn't start signal thread",
236 smfi->xxfi_name);
237 return MI_FAILURE;
238 }
239 r = MI_MONITOR_INIT();
240
241 /* Startup the listener */
242 if (mi_listener(conn, dbg, smfi, timeout, backlog) != MI_SUCCESS)
243 r = MI_FAILURE;
244
245 return r;
246 }
247
248