main.c revision 125820
11553Srgrimes/* 274532Sru * Copyright (c) 1999-2003 Sendmail, Inc. and its suppliers. 31553Srgrimes * All rights reserved. 480029Sobrien * 580029Sobrien * By using this file, you agree to the terms and conditions set 61553Srgrimes * forth in the LICENSE file which can be found at the top level of 780029Sobrien * the sendmail distribution. 81553Srgrimes * 983391Sru */ 1080029Sobrien 111553Srgrimes#include <sm/gen.h> 12189090SedSM_RCSID("@(#)$Id: main.c,v 8.64.2.13 2003/10/20 22:27:13 ca Exp $") 13189090Sed 141553Srgrimes#define _DEFINE 1 151553Srgrimes#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