* include/mailutils/server.h (mu_m_server_prefork_fp): New

function type. (mu_m_server_destroy, mu_m_server_set_prefork) (mu_m_server_configured_count, mu_m_server_end): New functions. * mailbox/msrv.c (struct _mu_m_server.prefork): New member. (mu_m_server_set_prefork): New function. (m_srv_conn): Use prefork to decide whether to proceed with a connection. * lib/tcpwrap.c, lib/tcpwrap.h (mu_tcp_wrapper_prefork): New function. * imap4d/imap4d.c, maidag/lmtp.c, maidag/maidag.c, pop3d/pop3d.c: Use m-server prefork to handle TCP wrappers. * po/POTFILES.in: Add more files.

* include/mailutils/server.h (mu_m_server_prefork_fp): New
function type. (mu_m_server_destroy, mu_m_server_set_prefork) (mu_m_server_configured_count, mu_m_server_end): New functions. * mailbox/msrv.c (struct _mu_m_server.prefork): New member. (mu_m_server_set_prefork): New function. (m_srv_conn): Use prefork to decide whether to proceed with a connection. * lib/tcpwrap.c, lib/tcpwrap.h (mu_tcp_wrapper_prefork): New function. * imap4d/imap4d.c, maidag/lmtp.c, maidag/maidag.c, pop3d/pop3d.c: Use m-server prefork to handle TCP wrappers. * po/POTFILES.in: Add more files.
Sergey Poznyakoff
Commit fef074ea ... fef074ea553712804485bf71f9d65b33d37e00b5 authored 2008-01-05 10:22:40 +0000 by Sergey Poznyakoff
Showing 9 changed files with 46 additions and 39 deletions
imap4d/imap4d.c
include/mailutils/server.h
lib/tcpwrap.c
lib/tcpwrap.h
maidag/lmtp.c
maidag/maidag.c
mailbox/msrv.c
po/POTFILES.in
pop3d/pop3d.c
--- a/imap4d/imap4d.c
View file @fef074e
+++ b/imap4d/imap4d.c
View file @fef074e
@@ -374,24 +374,6 @@ imap4d_mainloop (int fd, FILE *infile, FILE *outfile)
  struct sockaddr_in cs;
  int debug_mode = isatty (fd);
-  mu_diag_output (MU_DIAG_INFO, _("Incoming connection opened"));
-  if (!debug_mode)
-    {
-      if (get_client_address (fd, &cs) == 0) 
-	{
-	  if (!mu_tcpwrapper_access (fd))
-	    {
-	      mu_error (_("Access from %s blocked."), inet_ntoa (cs.sin_addr));
-	      return 1;
-	    }
-	}
-      else if (mu_tcp_wrapper_enable)
-	{
-	  mu_error (_("Rejecting connection from unknown address"));
-	  return 1;
-	}
-    }
  /* Reset hup to exit. */
  signal (SIGHUP, imap4d_signal);
  /* Timeout alarm. */
@@ -502,6 +484,7 @@ main (int argc, char **argv)
  mu_m_server_create (&server, "GNU imap4d");
  mu_m_server_set_conn (server, imap4d_connection);
+  mu_m_server_set_prefork (server, mu_tcp_wrapper_prefork);
  mu_m_server_set_mode (server, MODE_INTERACTIVE);
  mu_m_server_set_max_children (server, 20);
  /* FIXME mu_m_server_set_pidfile (); */
--- a/include/mailutils/server.h
View file @fef074e
+++ b/include/mailutils/server.h
View file @fef074e
@@ -75,10 +75,12 @@ int mu_tcp_server_get_sockaddr (mu_tcp_server_t srv, struct sockaddr *s,
 /* m-server */
 typedef int (*mu_m_server_conn_fp) (int, void *, time_t, int);
+typedef int (*mu_m_server_prefork_fp) (int, struct sockaddr *s, int size);
 void mu_m_server_create (mu_m_server_t *psrv, const char *ident);
+void mu_m_server_destroy (mu_m_server_t *pmsrv);
 void mu_m_server_set_mode (mu_m_server_t srv, int mode);
 void mu_m_server_set_conn (mu_m_server_t srv, mu_m_server_conn_fp f);
+void mu_m_server_set_prefork (mu_m_server_t srv, mu_m_server_prefork_fp fun);
 void mu_m_server_set_data (mu_m_server_t srv, void *data);
 void mu_m_server_set_max_children (mu_m_server_t srv, size_t num);
 int mu_m_server_set_pidfile (mu_m_server_t srv, const char *pidfile);
@@ -91,8 +93,11 @@ int mu_m_server_mode (mu_m_server_t srv);
 time_t mu_m_server_timeout (mu_m_server_t srv);
 void mu_m_server_get_sigset (mu_m_server_t srv, sigset_t *sigset);
+void mu_m_server_configured_count (mu_m_server_t msrv, size_t count);
 void mu_m_server_begin (mu_m_server_t msrv);
 int mu_m_server_run (mu_m_server_t msrv);
+void mu_m_server_end (mu_m_server_t msrv);
 void mu_m_server_cfg_init (void);
--- a/lib/tcpwrap.c
View file @fef074e
+++ b/lib/tcpwrap.c
View file @fef074e
 /* GNU Mailutils -- a suite of utilities for electronic mail
   Copyright (C) 1999, 2001, 2002, 2003, 2004, 
-   2005, 2006, 2007 Free Software Foundation, Inc.
+   2005, 2006, 2007, 2008 Free Software Foundation, Inc.
   GNU Mailutils is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
@@ -20,6 +20,7 @@
 #ifdef HAVE_CONFIG_H
 # include <config.h>
 #endif
+#include <stdlib.h>
 #include <syslog.h>
 #include <string.h>
 #include <mailutils/debug.h>
@@ -27,6 +28,7 @@
 #include <mailutils/syslog.h>
 #include <mailutils/cfg.h>
 #include <mailutils/diag.h>
+#include <mailutils/error.h>
 int mu_tcp_wrapper_enable = 1;
 char *mu_tcp_wrapper_daemon;
@@ -138,3 +140,17 @@ mu_tcpwrapper_access (int fd)
 }
 #endif
+int
+mu_tcp_wrapper_prefork (int fd, struct sockaddr *sa, int salen)
+{
+  if (mu_tcp_wrapper_enable && !mu_tcpwrapper_access (fd))
+    {
+      char *p = mu_sockaddr_to_astr (sa, salen);
+      mu_error (_("Access from %s blocked by TCP wrappers."), p);
+      free (p);
+      return 1;
+    }
+  return 0;
+}
--- a/lib/tcpwrap.h
View file @fef074e
+++ b/lib/tcpwrap.h
View file @fef074e
 /* GNU Mailutils -- a suite of utilities for electronic mail
   Copyright (C) 1999, 2001, 2002, 2003, 2004, 
-   2005, 2006, 2007 Free Software Foundation, Inc.
+   2005, 2006, 2007, 2008 Free Software Foundation, Inc.
   GNU Mailutils is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
@@ -30,8 +30,8 @@ extern int mu_tcp_wrapper_cb_hosts_allow_syslog (mu_debug_t debug, void *data,
 extern int mu_tcp_wrapper_cb_hosts_deny_syslog (mu_debug_t debug, void *data,
 						char *arg);
 extern int mu_tcpwrapper_access (int fd);
 extern void mu_tcpwrapper_cfg_init (void);
+extern int mu_tcp_wrapper_prefork (int fd, struct sockaddr *sa, int salen);
 #ifdef WITH_LIBWRAP
 # define TCP_WRAPPERS_CONFIG { "tcp-wrappers", mu_cfg_section },
--- a/maidag/lmtp.c
View file @fef074e
+++ b/maidag/lmtp.c
View file @fef074e
@@ -556,10 +556,16 @@ lmtp_loop (FILE *in, FILE *out, unsigned int timeout)
  return 0;
 }
+typedef union
+{
+  struct sockaddr sa;
+  struct sockaddr_in s_in;
+  struct sockaddr_un s_un;
+} all_addr_t;
 int
 lmtp_connection (int fd, void *data, time_t timeout, int transcript)
 {
-  /* FIXME: TCP wrappers */
  lmtp_transcript = transcript;
  lmtp_loop (fdopen (fd, "r"), fdopen (fd, "w"), timeout);
  return 0;
--- a/maidag/maidag.c
View file @fef074e
+++ b/maidag/maidag.c
View file @fef074e
@@ -478,6 +478,7 @@ main (int argc, char *argv[])
  mu_m_server_create (&server, "GNU maidag");
  mu_m_server_set_conn (server, lmtp_connection);
+  mu_m_server_set_prefork (server, mu_tcp_wrapper_prefork);
  mu_m_server_set_mode (server, MODE_INTERACTIVE);
  mu_m_server_set_max_children (server, 20);
  mu_m_server_set_timeout (server, 600);
--- a/mailbox/msrv.c
View file @fef074e
+++ b/mailbox/msrv.c
View file @fef074e
@@ -66,6 +66,7 @@ struct _mu_m_server
  mu_server_t server;
  mu_list_t srvlist;
  mu_m_server_conn_fp conn;
+  mu_m_server_prefork_fp prefork;
  void *data;
  int mode;
  int foreground;
@@ -192,6 +193,12 @@ mu_m_server_set_conn (mu_m_server_t srv, mu_m_server_conn_fp conn)
 }
 void
+mu_m_server_set_prefork (mu_m_server_t srv, mu_m_server_prefork_fp fun)
+{
+  srv->prefork = fun;
+}
+void
 mu_m_server_set_data (mu_m_server_t srv, void *data)
 {
  srv->data = data;
@@ -502,7 +509,9 @@ m_srv_conn (int fd, struct sockaddr *sa, int salen,
          pause ();
          return 0;
        }
+      if (pconf->msrv->prefork && pconf->msrv->prefork (fd, sa, salen))
+	return 0;
      pid = fork ();
      if (pid == -1)
 	mu_diag_output (MU_DIAG_ERROR, "fork: %s", strerror (errno));
--- a/po/POTFILES.in
View file @fef074e
+++ b/po/POTFILES.in
View file @fef074e
@@ -31,6 +31,7 @@ lib/argp-help.c
 lib/argp-parse.c
 lib/getopt.c
 lib/mailcap.c
+lib/tcpwrap.c
 lib/xmalloc.c
 libargp/auth.c
 libargp/cmdline.c
--- a/pop3d/pop3d.c
View file @fef074e
+++ b/pop3d/pop3d.c
View file @fef074e
@@ -291,21 +291,6 @@ pop3d_mainloop (int fd, FILE *infile, FILE *outfile)
 {
  int status = OK;
  char buffer[512];
-  struct sockaddr_in cs;
-  if (pop3d_get_client_address (fd, &cs) == 0)
-    {
-      if (!mu_tcpwrapper_access (fd))
-	{
-	  mu_error (_("Access from %s blocked."), inet_ntoa (cs.sin_addr));
-	  return 1;
-	}
-    }
-  else if (!debug_mode && mu_tcp_wrapper_enable)
-    {
-      mu_error (_("Rejecting connection from unknown address"));
-      return 1;
-    }
  /* Reset hup to exit.  */
  signal (SIGHUP, pop3d_signal);
@@ -494,6 +479,7 @@ main (int argc, char **argv)
  mu_m_server_create (&server, "GNU pop3d");
  mu_m_server_set_conn (server, pop3d_connection);
+  mu_m_server_set_prefork (server, mu_tcp_wrapper_prefork);
  mu_m_server_set_mode (server, MODE_INTERACTIVE);
  mu_m_server_set_max_children (server, 20);
  /* FIXME mu_m_server_set_pidfile (); */