main.c revision 125820 
1/*
2 *  Copyright (c) 1999-2003 Sendmail, 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>
12SM_RCSID("@(#)$Id: main.c,v 8.64.2.13 2003/10/20 22:27:13 ca Exp $")
13
14#define _DEFINE	1
15#include "libmilter.h"
16#include <fcntl.h>
17#include <sys/stat.h>
18
19
20static 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
32int
33smfi_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 (smfi->xxfi_version != SMFI_VERSION)
56	{
57		/* hard failure for now! */
58		smi_log(SMI_LOG_ERR,
59			"%s: smfi_register: version mismatch application: %d != milter: %d",
60			smfi->xxfi_name, smfi->xxfi_version,
61			(int) SMFI_VERSION);
62
63		/* XXX how about smfi? */
64		free(smfi->xxfi_name);
65		return MI_FAILURE;
66	}
67
68	return MI_SUCCESS;
69}
70
71/*
72**  SMFI_STOP -- stop milter
73**
74**	Parameters:
75**		none.
76**
77**	Returns:
78**		success.
79*/
80
81int
82smfi_stop()
83{
84	mi_stop_milters(MILTER_STOP);
85	return MI_SUCCESS;
86}
87
88/*
89**  Default values for some variables.
90**	Most of these can be changed with the functions below.
91*/
92
93static int dbg = 0;
94static char *conn = NULL;
95static int timeout = MI_TIMEOUT;
96static int backlog = MI_SOMAXCONN;
97
98#if _FFR_SMFI_OPENSOCKET
99/*
100**  SMFI_OPENSOCKET -- try the socket setup to make sure we'll be
101**		able to start up
102**
103**	Parameters:
104**		rmsocket -- if true, instructs libmilter to attempt
105**			to remove the socket before creating it;
106**			only applies for "local:" or "unix:" sockets
107**
108**	Return:
109**		MI_SUCCESS/MI_FAILURE
110*/
111
112int
113smfi_opensocket(rmsocket)
114	bool rmsocket;
115{
116	if (smfi == NULL || conn == NULL)
117		return MI_FAILURE;
118
119	return mi_opensocket(conn, backlog, dbg, rmsocket, smfi);
120}
121#endif /* _FFR_SMFI_OPENSOCKET */
122
123/*
124**  SMFI_SETDBG -- set debug level.
125**
126**	Parameters:
127**		odbg -- new debug level.
128**
129**	Returns:
130**		MI_SUCCESS
131*/
132
133int
134smfi_setdbg(odbg)
135	int odbg;
136{
137	dbg = odbg;
138	return MI_SUCCESS;
139}
140
141/*
142**  SMFI_SETTIMEOUT -- set timeout (for read/write).
143**
144**	Parameters:
145**		otimeout -- new timeout.
146**
147**	Returns:
148**		MI_SUCCESS
149*/
150
151int
152smfi_settimeout(otimeout)
153	int otimeout;
154{
155	timeout = otimeout;
156	return MI_SUCCESS;
157}
158
159/*
160**  SMFI_SETCONN -- set connection information (socket description)
161**
162**	Parameters:
163**		oconn -- new connection information.
164**
165**	Returns:
166**		MI_SUCCESS/MI_FAILURE
167*/
168
169int
170smfi_setconn(oconn)
171	char *oconn;
172{
173	size_t l;
174
175	if (oconn == NULL || *oconn == '\0')
176		return MI_FAILURE;
177	l = strlen(oconn) + 1;
178	if ((conn = (char *) malloc(l)) == NULL)
179		return MI_FAILURE;
180	if (sm_strlcpy(conn, oconn, l) >= l)
181		return MI_FAILURE;
182	return MI_SUCCESS;
183}
184
185/*
186**  SMFI_SETBACKLOG -- set backlog
187**
188**	Parameters:
189**		obacklog -- new backlog.
190**
191**	Returns:
192**		MI_SUCCESS/MI_FAILURE
193*/
194
195int
196smfi_setbacklog(obacklog)
197	int obacklog;
198{
199	if (obacklog <= 0)
200		return MI_FAILURE;
201	backlog = obacklog;
202	return MI_SUCCESS;
203}
204
205
206/*
207**  SMFI_MAIN -- setup milter connnection and start listener.
208**
209**	Parameters:
210**		none.
211**
212**	Returns:
213**		MI_SUCCESS/MI_FAILURE
214*/
215
216int
217smfi_main()
218{
219	int r;
220
221	(void) signal(SIGPIPE, SIG_IGN);
222	if (conn == NULL)
223	{
224		smi_log(SMI_LOG_FATAL, "%s: missing connection information",
225			smfi->xxfi_name);
226		return MI_FAILURE;
227	}
228
229	(void) atexit(mi_clean_signals);
230	if (mi_control_startup(smfi->xxfi_name) != MI_SUCCESS)
231	{
232		smi_log(SMI_LOG_FATAL,
233			"%s: Couldn't start signal thread",
234			smfi->xxfi_name);
235		return MI_FAILURE;
236	}
237	r = MI_SUCCESS;
238
239	/* Startup the listener */
240	if (mi_listener(conn, dbg, smfi, timeout, backlog) != MI_SUCCESS)
241		r = MI_FAILURE;
242	return r;
243}
244
245