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 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 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 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 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 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 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 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 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