Changeset 109 for trunk/src/sh_portcheck.c

Timestamp:

Jun 18, 2007, 9:29:51 PM (17 years ago)

Author:

rainer

Message:

More graceful connection teardown in portcheck.

File:

• trunk/src/sh_portcheck.c (modified) (4 diffs)

trunk/src/sh_portcheck.c

@@ -39 +39 @@
 #include <errno.h>
 #include <unistd.h>
+#include <fcntl.h>
 
 #define PORTCHK_VERSION "1.0"
@@ -110 +111 @@
 #include "sh_error.h"
 #include "sh_mem.h"
+#include "sh_calls.h"
 #include "sh_utils.h"
 #include "sh_modules.h"
@@ -676 +678 @@
   /* struct in_addr     haddr; */
   int                retval;
+  int                flags;
   char             * p;
 #ifndef TEST_ONLY
@@ -728 +731 @@
         fprintf(stderr, _("check port: %5d on %15s open %s\n"), 
                 port, inet_ntoa(haddr), p);
+
+#if !defined(O_NONBLOCK)
+#if defined(O_NDELAY)
+#define O_NONBLOCK  O_NDELAY
+#else
+#define O_NONBLOCK  0
+#endif
+#endif
+
+      /* prepare to close connection gracefully
+       */
+      if      (port == 22)  /* ssh */
+        {
+          flags = retry_fcntl(FIL__, __LINE__, fd, F_GETFL, 0);
+          retry_fcntl(FIL__, __LINE__, fd, F_SETFL, flags | O_NONBLOCK);
+          write (fd, _("SSH-2.0-Foobar"), 14);
+          write (fd, "\r\n", 2);
+        }
+      else if (port == 25)  /* smtp */
+        {
+          flags = retry_fcntl(FIL__, __LINE__, fd, F_GETFL, 0);
+          retry_fcntl(FIL__, __LINE__, fd, F_SETFL, flags | O_NONBLOCK);
+          write (fd, _("QUIT"), 4);
+          write (fd, "\r\n", 2);
+        }
+      else if (port == 79)  /* finger */
+        {
+          flags = retry_fcntl(FIL__, __LINE__, fd, F_GETFL, 0);
+          retry_fcntl(FIL__, __LINE__, fd, F_SETFL, flags | O_NONBLOCK);
+          write (fd, "\r\n", 2);
+        }
+      else if (port == 110) /* pop3 */
+        {
+          flags = retry_fcntl(FIL__, __LINE__, fd, F_GETFL, 0);
+          retry_fcntl(FIL__, __LINE__, fd, F_SETFL, flags | O_NONBLOCK);
+          write (fd, _("QUIT"), 4);
+          write (fd, "\r\n", 2);
+        }
+      else if (port == 143) /* imap */
+        {
+          flags = retry_fcntl(FIL__, __LINE__, fd, F_GETFL, 0);
+          retry_fcntl(FIL__, __LINE__, fd, F_SETFL, flags | O_NONBLOCK);
+          write (fd, _("A01 LOGOUT"), 10);
+          write (fd, "\r\n", 2);
+        }
      }
   close (fd);