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> 12 SM_RCSID("@(#)$Id: main.c,v 8.79 2003/10/20 22:25:09 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 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 (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 81 int 82 smfi_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 93 static int dbg = 0; 94 static char *conn = NULL; 95 static int timeout = MI_TIMEOUT; 96 static int backlog = MI_SOMAXCONN; 97 98 /* 99 ** SMFI_OPENSOCKET -- try the socket setup to make sure we'll be 100 ** able to start up 101 ** 102 ** Parameters: 103 ** rmsocket -- if true, instructs libmilter to attempt 104 ** to remove the socket before creating it; 105 ** only applies for "local:" or "unix:" sockets 106 ** 107 ** Return: 108 ** MI_SUCCESS/MI_FAILURE 109 */ 110 111 int 112 smfi_opensocket(rmsocket) 113 bool rmsocket; 114 { 115 if (smfi == NULL || conn == NULL) 116 return MI_FAILURE; 117 118 return mi_opensocket(conn, backlog, dbg, rmsocket, smfi); 119 } 120 121 /* 122 ** SMFI_SETDBG -- set debug level. 123 ** 124 ** Parameters: 125 ** odbg -- new debug level. 126 ** 127 ** Returns: 128 ** MI_SUCCESS 129 */ 130 131 int 132 smfi_setdbg(odbg) 133 int odbg; 134 { 135 dbg = odbg; 136 return MI_SUCCESS; 137 } 138 139 /* 140 ** SMFI_SETTIMEOUT -- set timeout (for read/write). 141 ** 142 ** Parameters: 143 ** otimeout -- new timeout. 144 ** 145 ** Returns: 146 ** MI_SUCCESS 147 */ 148 149 int 150 smfi_settimeout(otimeout) 151 int otimeout; 152 { 153 timeout = otimeout; 154 return MI_SUCCESS; 155 } 156 157 /* 158 ** SMFI_SETCONN -- set connection information (socket description) 159 ** 160 ** Parameters: 161 ** oconn -- new connection information. 162 ** 163 ** Returns: 164 ** MI_SUCCESS/MI_FAILURE 165 */ 166 167 int 168 smfi_setconn(oconn) 169 char *oconn; 170 { 171 size_t l; 172 173 if (oconn == NULL || *oconn == '\0') 174 return MI_FAILURE; 175 l = strlen(oconn) + 1; 176 if ((conn = (char *) malloc(l)) == NULL) 177 return MI_FAILURE; 178 if (sm_strlcpy(conn, oconn, l) >= l) 179 return MI_FAILURE; 180 return MI_SUCCESS; 181 } 182 183 /* 184 ** SMFI_SETBACKLOG -- set backlog 185 ** 186 ** Parameters: 187 ** obacklog -- new backlog. 188 ** 189 ** Returns: 190 ** MI_SUCCESS/MI_FAILURE 191 */ 192 193 int 194 smfi_setbacklog(obacklog) 195 int obacklog; 196 { 197 if (obacklog <= 0) 198 return MI_FAILURE; 199 backlog = obacklog; 200 return MI_SUCCESS; 201 } 202 203 204 /* 205 ** SMFI_MAIN -- setup milter connnection and start listener. 206 ** 207 ** Parameters: 208 ** none. 209 ** 210 ** Returns: 211 ** MI_SUCCESS/MI_FAILURE 212 */ 213 214 int 215 smfi_main() 216 { 217 int r; 218 219 (void) signal(SIGPIPE, SIG_IGN); 220 if (conn == NULL) 221 { 222 smi_log(SMI_LOG_FATAL, "%s: missing connection information", 223 smfi->xxfi_name); 224 return MI_FAILURE; 225 } 226 227 (void) atexit(mi_clean_signals); 228 if (mi_control_startup(smfi->xxfi_name) != MI_SUCCESS) 229 { 230 smi_log(SMI_LOG_FATAL, 231 "%s: Couldn't start signal thread", 232 smfi->xxfi_name); 233 return MI_FAILURE; 234 } 235 r = MI_SUCCESS; 236 237 /* Startup the listener */ 238 if (mi_listener(conn, dbg, smfi, timeout, backlog) != MI_SUCCESS) 239 r = MI_FAILURE; 240 241 return r; 242 } 243 244