Initial commit

Signed-off-by: Mavridis Philippe <mavridisf@gmail.com>

1 files changed, 626 insertions, 0 deletions

diff --git a/src/klammail/clamdmail.c b/src/klammail/clamdmail.c
new file mode 100755
index 0000000..19f985e
--- /dev/null
+++ b/src/klammail/clamdmail.c
@@ -0,0 +1,626 @@
+/*
+*
+*  Clamdmail.c is based loosely on clamdscan.c by Tomasz Kojm.
+*
+*  This program takes a mail message as input from stdin, uses rfc822.c to extract
+*  attachments to a temp directory, gets clamd to scan them, and
+*  handles the message appropriately.
+*
+*  Copyright (C) 2003 Robert Hogan <robert@roberthogan.net>
+*
+*  This program is free software; you can redistribute it and/or modify
+*  it under the terms of the GNU General Public License as published by
+*  the Free Software Foundation; either version 2 of the License, or
+*  (at your option) any later version.
+*
+*  This program is distributed in the hope that it will be useful,
+*  but WITHOUT ANY WARRANTY; without even the implied warranty of
+*  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+*  GNU General Public License for more details.
+*
+*  You should have received a copy of the GNU General Public License
+*  along with this program; if not, write to the Free Software
+*  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+*/
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <sys/time.h>
+#include <time.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+#include <pwd.h>
+#include <clamav.h>
+
+#include "options.h"
+#include "defaults.h"
+#include "memory.h"
+#include "../version.h"
+#include "../../config.h"
+
+#define BUFFSIZE 1024
+
+
+
+/* this local macro takes care about freeing memory at exit */
+/*
+#define mexit(i)    if(opt) free_opt(opt);			    \
+                    mprintf("*Memory freed. Exit code: %d\n", i);   \
+                    exit(i);
+*/
+#define mexit(i)    exit(i)
+
+struct s_info {
+  int signs; /* number of signatures loaded */
+  int dirs; /* number of scanned directories */
+  int files; /* number of scanned files */
+  int ifiles; /* number of infected files */
+  int notremoved; /* number of not removed files (if --remove) */
+  int notmoved; /* number of not moved files (if --move) */
+  int errors; /*  ... of errors */
+  long int blocks; /* number of read 16kb blocks */
+};
+
+
+void help(void);
+void printtag(void);
+void startclamd(struct optstruct *opt);
+
+struct s_info claminfo;
+short printinfected = 0;
+short int mprintf_stdout;
+
+int clamdscan(struct optstruct *opt)
+{
+  int ds, dms;
+
+  struct timeval t1, t2;
+  struct timezone tz;
+  const char *bndrystore;
+  const char *tmpdir;
+  char *tmpfil;
+  char *pfx;
+  char *tmper;
+  char *dir;
+  struct passwd *user = NULL;
+  FILE *tmp;
+  FILE *fs;
+  int bytes;
+  char buff[BUFFSIZE];
+  struct cl_node *trie = NULL;
+  int threads = 0;
+  int fd, fdtmp, ret, no = 0;
+  unsigned long int size = 0;
+  long double mb;
+  const char *virname;
+  struct cl_engine *engine = NULL;
+#ifndef SUPPORT_CLAMAV_V095
+  struct cl_limits limits;
+#endif
+  struct stat sb;
+
+  if(optc(opt, 'V')) {
+    mprintf("clamdmail "KLAMAV_VERSION" \n");
+    mexit(0);
+  }
+
+  if(optc(opt, 'h')) {
+  free_opt(opt);
+  help();
+  }
+
+  if(optc(opt, 'i')) printinfected = 1;
+  else printinfected = 0;
+
+
+
+  memset(&claminfo, 0, sizeof(struct s_info));
+
+  gettimeofday(&t1, &tz);
+
+  //if(user)
+
+
+  char name[19];
+  char *tmpnm;
+  char *mdir;
+
+  if((mdir = getenv("TMPDIR")) == NULL)
+  #ifdef P_tmpdir
+      mdir = P_tmpdir;
+  #else
+      mdir = "/tmp";
+  #endif
+
+  tmpnm = (char*) mcalloc(strlen(mdir) + 1 + 16 + 1 + 7, sizeof(char));
+  if(tmpnm == NULL) {
+      exit(2);
+  }
+
+  sprintf(tmpnm, "%s/", mdir);
+  snprintf(name, sizeof(name), "klammailXXXXXX");
+  strncat(tmpnm, name, 19);
+  mkstemp(tmpnm);
+
+
+
+  fd = open(tmpnm,O_RDWR|O_CREAT, S_IRWXU);
+
+  while((bytes = read(0, buff, BUFFSIZE)) > 0) {
+
+          if(write(fd, buff, bytes) != bytes) {
+                  close(fd);
+                  return CL_EMEM;
+          }
+  }
+
+  if(fsync(fd) == -1) {
+          close(fd);
+#ifdef SUPPORT_CLAMAV_V095
+          return CL_ETMPFILE;
+#else
+          return CL_EIO;
+#endif
+  }
+
+  close(fd);
+
+  if((fd = open(tmpnm, O_RDONLY)) == -1) {
+    printf("Can't open file %s\n", tmpnm);
+    exit(2);
+  }
+
+  ret = 0;
+  ret = client(tmpnm, opt, &virname);
+
+
+
+  /* Clamd isn't running, scan the file ourselves */
+  if((ret == 2)) {
+      /* Clamd isn't running, start it so it is available next time. */
+      startclamd(opt);
+#ifdef SUPPORT_CLAMAV_V095
+      if((engine = cl_engine_new()) == NULL) {
+          printf("Database initialization error: %s\n", cl_strerror(ret));;
+          cl_engine_free(engine);
+          close(fd);
+          exit(2);
+      }
+#endif
+      if(optc(opt, 'd')) {
+#ifdef SUPPORT_CLAMAV_V095
+          if((ret = cl_load(getargc(opt, 'd'), engine, &no, CL_DB_STDOPT))) {
+#else
+          if((ret = cl_load(getargc(opt, 'd'), &engine, &no, CL_DB_STDOPT))) {
+#endif
+              printf("cl_load: %s\n", cl_strerror(ret));
+              close(fd);
+              return 50;
+          }
+      }else{
+#ifdef SUPPORT_CLAMAV_V095
+          if((ret = cl_load(cl_retdbdir(), engine, &no, CL_DB_STDOPT))) {
+#else
+          if((ret = cl_loaddbdir(cl_retdbdir(), &engine, &no, CL_DB_STDOPT))) {
+#endif
+          printf("cl_loaddbdir: %s\n", cl_strerror(ret));
+          close(fd);
+          exit(2);
+          }
+      }
+
+      /* build engine */
+#ifdef SUPPORT_CLAMAV_V095
+      if((ret = cl_engine_compile(engine))) {
+#else
+      if((ret = cl_build(engine))) {
+#endif
+          printf("Database initialization error: %s\n", cl_strerror(ret));;
+#ifdef SUPPORT_CLAMAV_V095
+          cl_engine_free(engine);
+#else
+          cl_free(engine);
+#endif
+          close(fd);
+          exit(2);
+      }
+
+#ifndef SUPPORT_CLAMAV_V095
+      /* set up archive limits */
+      memset(&limits, 0, sizeof(struct cl_limits));
+      limits.maxfiles = 1000; /* max files */
+      limits.maxfilesize = 10 * 1048576; /* maximum size of archived/compressed
+                                          * file (files exceeding this limit
+                                          * will be ignored)
+                                          */
+      /*limits.maxreclevel = 5;*/ /* maximum recursion level for archives */
+      /*limits.maxmailrec = 64;*/ /* maximum recursion level for mail files */
+      /*limits.maxratio = 200;*/ /* maximum compression ratio */
+      limits.archivememlim = 1;
+#endif
+#ifdef SUPPORT_CLAMAV_V095
+      ret = cl_scandesc(fd, &virname, &size, engine,
+             CL_SCAN_STDOPT | CL_SCAN_ARCHIVE | CL_SCAN_MAIL | CL_SCAN_OLE2 | CL_SCAN_HTML);
+      printf("scandesc returned: %i\n", cl_strerror(ret));;
+#else
+      ret = cl_scandesc(fd, &virname, &size, engine, &limits,
+                        CL_SCAN_STDOPT | CL_ARCHIVE | CL_MAIL | CL_OLE2 | CL_SCAN_HTML);
+#endif
+  }
+
+  /* scan descriptor (with archive and mail scanning enabled) */
+  close(fd);
+  fd = open(tmpnm,O_RDWR, S_IRWXU);
+
+  spoolstdin(tmpnm, fd, ret, &virname, &bndrystore, opt);
+
+  if((ret == 2))
+#ifdef SUPPORT_CLAMAV_V095
+      cl_engine_free(engine);
+#else
+      cl_free(engine);
+#endif
+
+  unlink(tmpnm);
+
+  mexit(0);
+}
+
+int spoolstdin(char *tmpnm, int fd, int ret, char **virname, const char **bndrystore, struct optstruct *opt)
+{
+  int bytes;
+  int i, j;
+  long int size = 0;
+  char buff[BUFFSIZE];
+  char string[1000];
+  char To[1000];
+  char ReplyTo[1000];
+  char DelivTo[1000];
+  char From[1000];
+  char ToR[1000];
+  char FromR[1000];
+  char boundary[1000];
+  char host[256];  /* could be HOST_NAME_MAX+1 on POSIX 1003.1-2001 */
+  struct timeval tv;
+  struct timezone tz;
+  char *s;
+  char *storage;
+  FILE *tmp;
+  int pcount = 0;
+  time_t starttime;
+  struct tm *tm;
+  char *out = NULL;
+  const char *format;
+  size_t out_length = 0;
+  time_t when;
+  int childpid;
+
+
+  strcpy (To, "To:");
+  strcpy (From, "From:");
+  strcpy (ReplyTo, "Reply-To:");
+  strcpy (DelivTo, "Delivered");
+
+  gethostname(host, sizeof(host));
+  gettimeofday(&tv, &tz);
+  time(&when);
+  tm = localtime(&when);
+  format = "%a, %_d %b %Y %H:%M:%S %z";
+  do {
+          out_length += 200;
+          out = (char *) realloc (out, out_length);
+          out[0] = '\1';
+  } while (strftime (out, out_length, format, tm) == 0 && out[0] != '\0');
+
+  if(ret == CL_VIRUS) {
+      if(!(optc(opt, 'f'))){
+
+          lseek(fd, 0, SEEK_SET);
+          if (!(tmp = fdopen(fd,"r"))){
+              mprintf("@Can't open file %s\n", tmpnm);
+              return 54;
+          }
+
+          mprintf_stdout = 1;
+
+          while (fgets(string, sizeof(string), tmp)) {
+              if (strstr(string, To) && !(strstr(string, ReplyTo)) && !(strstr(string,DelivTo))){
+                      strcpy(To, string);
+                      strcpy(ToR, string);
+              }
+              if (strstr(string, From)){
+                      strcpy(FromR, string);
+              }
+              if (strncmp(string, "\n", 1) == 0 && pcount != 0){
+                      break;
+              }
+              ++pcount;
+          }
+
+          if (strcmp(To,"To:") == 0){
+                  strcat(To, " ");
+                  strcat(To, getargl(opt, "admin"));
+                  strcat(To, "\n");
+          }
+          strcat(From, " ");
+          strcat(From, "KlamAV");
+          strcat(From, "\n");
+          strcat(ReplyTo, " ");
+          strcat(ReplyTo, "KlamAV");
+          strcat(ReplyTo, "\n");
+          mprintf("%s", From);
+          mprintf("%s", ReplyTo);
+          mprintf("%s", To);
+          mprintf("Subject: Virus %s found in attached mail by KlamAV.\n", *virname);
+          mprintf("Date: %s\n", out);
+          if(!(optl(opt, "quar"))){
+              mprintf("MIME-Version: 1.0\n");
+              mprintf("Content-Type: multipart/mixed;\n");
+              mprintf("  boundary=\"----------=_%d\"\n",tv.tv_sec);
+              mprintf("X-Virus-Status: Yes\n");
+              mprintf("X-Virus-Checker: Scanned by KlamAV %s on %s (virus-found %s);\n\t%s\n",
+                      KLAMAV_VERSION, host, *virname, out);
+              mprintf("\n");
+              mprintf("\n");
+              mprintf("------------=_%d\n",tv.tv_sec);
+          }
+          mprintf("Content-Type: text/plain;\n");
+          mprintf("  charset=\"us-ascii\"\n");
+          mprintf("Content-Transfer-Encoding: quoted-printable\n");
+          if((optl(opt, "quar"))){
+              mprintf("X-Virus-Status: Yes\n");
+              mprintf("X-Virus-Checker: Scanned by KlamAV %s on %s (virus-found %s);\n\t%s\n",
+                      KLAMAV_VERSION, host, *virname, out);
+              mprintf("\n");
+              mprintf("KlamAV anti-virus scanner has intercepted and quarantined a message addressed to you.\n");
+          }
+          else
+          {
+              mprintf("\n");
+              mprintf("KlamAV anti-virus scanner has detected a virus in the attached message.\n");
+          }
+          mprintf("\n");
+          mprintf("The following is a summary of the infected message:\n");
+          mprintf("\n");
+          mprintf("Virus name: %s\n", *virname);
+          mprintf("%s", FromR);
+          mprintf("%s", ToR);
+          mprintf("\n");
+          mprintf("Please be aware that a virus spread by email normally forges the \n");
+          mprintf("address of the sender. There is a good chance that the infected message\n");
+          mprintf("was not received from the sender listed above. \n");
+          mprintf("\n");
+          if(!(optl(opt, "quar"))){
+              mprintf("------------=_%d\n",tv.tv_sec);
+              mprintf("Content-Type: message/rfc822; x-virus-type=original\n");
+              mprintf("Content-Description: Infected Message - Open At Your Own Risk\n");
+              mprintf("Content-Disposition: inline\n");
+              mprintf("Content-Transfer-Encoding: 8bit\n");
+              mprintf("\n");
+              fflush(stdout);
+
+              lseek(fd, 0, SEEK_SET);
+              while((bytes = read(fd, buff, BUFFSIZE)) > 0) {
+
+                      if(write(1, buff, bytes) != bytes) {
+                              close(fd);
+                              return CL_EMEM;
+                      }
+              }
+              mprintf("------------=_%d--\n",tv.tv_sec);
+          }
+          fflush(stdout);
+
+          fclose(tmp);
+      }else{
+          int bcnt = 0;
+          lseek(fd, 0, SEEK_SET);
+          if (!(tmp = fdopen(fd,"r"))){
+                  mprintf("@Can't open file %s\n", tmpnm);
+                  return 54;
+          }
+
+          mprintf_stdout = 1;
+
+          while (fgets(string, sizeof(string), tmp)) {
+              if (*string == '\n') {
+              mprintf("X-Virus-Status: Yes\n");
+              mprintf("X-Virus-Checker: Scanned by KlamAV %s on %s (virus-found %s);\n\t%s\n",
+                      KLAMAV_VERSION, host, *virname, out);
+                      break;
+              }
+              fputs(string, stdout);
+          }
+          while (fgets(string, sizeof(string), tmp)) {
+              fputs(string, stdout);
+          }
+
+          if(optl(opt, "tag")) {
+              if (!(*bndrystore))
+                      printtag();
+          }
+          fclose(tmp);
+      }
+
+      if ( ( childpid=fork() ) == -1 ) {
+          perror("Failed to fork; quitting\n");
+          exit(2);
+      }
+
+      if ( childpid == 0 ) {
+          char *dialogmessage;
+          dialogmessage = malloc(77+sizeof(virname)+sizeof(FromR)+sizeof(ToR)+4+84+54);
+          sprintf(dialogmessage,"KlamAV has found an infected mail with the following details:\nVirus name: %s\n%s\n%s\nThe mail has been handled according to the filter rules set up in your mail client (probably put in your trash/deleted items directory).",*virname,FromR,ToR);
+
+          if (setenv("PATH","/usr/local/sbin:/usr/sbin:/sbin:/usr/bin:/bin:/usr/local/bin:/opt/kde/bin",1) == 0)
+                  execlp("kdialog", "kdialog", "--msgbox", dialogmessage, "--title", "Virus found by KlamAV:",  NULL);
+          free(dialogmessage);
+      }
+  } else if(ret == CL_CLEAN) {
+      int bcnt = 0;
+      lseek(fd, 0, SEEK_SET);
+      if (!(tmp = fdopen(fd,"r"))){
+          mprintf("@Can't open file %s\n", tmpnm);
+          return 54;
+      }
+
+      mprintf_stdout = 1;
+
+      while (fgets(string, sizeof(string), tmp)) {
+          if (*string == '\n') {
+                  mprintf("X-Virus-Status: No\n");
+                  mprintf("X-Virus-Checker: Scanned by KlamAV %s on %s (no viruses);\n\t%s\n\n",
+                          KLAMAV_VERSION, host, out);
+                  break;
+          }
+          fputs(string, stdout);
+      }
+      while (fgets(string, sizeof(string), tmp)) {
+          fputs(string, stdout);
+      }
+
+      if(optl(opt, "tag"))
+          if (!(*bndrystore))
+                  printtag();
+
+      fclose(tmp);
+  } else {
+      if(!printinfected)
+          mprintf("stdin: %s\n", cl_strerror(ret));
+  }
+  return ret;
+}
+
+void printtag(void)
+{
+
+    mprintf_stdout = 1;
+    mprintf("\n");
+    mprintf("----------------------------------------------------------------------------\n");
+    mprintf("			This message was scanned by\n");
+    mprintf("		ClamAV Open Source Anti-Virus Technology\n");
+    mprintf("				using KlamAV\n");
+    mprintf("\n");
+    mprintf("			http://www.clamav.net\n");
+    mprintf("			http://klamav.sf.net\n");
+    mprintf("----------------------------------------------------------------------------\n");
+    mprintf("\n");
+    fflush(stdout);
+
+}
+
+void help(void)
+{
+
+    mprintf_stdout = 1;
+
+    mprintf("\n");
+    mprintf("		   KlamAV Mail Processing Client "KLAMAV_VERSION"\n");
+    mprintf("		   (c) 2004 Robert Hogan <robert@roberthogan.net>\n");
+    mprintf("		   	Uses a lot of code written by:\n");
+    mprintf("		       Tomasz Kojm <zolw@konarski.edu.pl>\n");
+    mprintf("		       Nigel Horne <njh@bandsman.co.uk>\n");
+    mprintf("	  \n");
+    mprintf("    --help		-h	Show help\n");
+    mprintf("    --version		-V	Print version number and exit\n");
+    mprintf("    -f			Header Flag Only.\n");
+    mprintf("    --tag		Tag messages as scanned.\n");
+    mprintf("    -d			Location of virus definition database.\n");
+
+    exit(0);
+}
+
+void startclamd(struct optstruct *opt)
+{
+
+    int pfds[2];
+    int childpid;
+    int fd;
+    char tmpnm[19];
+    char conffile[31];
+    struct stat sb;
+    FILE *tmp;
+    char *fullpath;
+    char cwd[200];
+    char *scancmd;
+
+    snprintf(tmpnm, sizeof(tmpnm), "klammailXXXXXX");
+    mkstemp(tmpnm);
+
+    fd = open(tmpnm,O_RDWR|O_CREAT, S_IRWXU);
+
+
+    lseek(fd, 0, SEEK_SET);
+    if (!(tmp = fdopen(fd,"w"))){
+            mprintf("@Can't open file \n");
+    }
+
+    fprintf(tmp,"LocalSocket /tmp/KlamMailSock\n");
+    fprintf(tmp,"MaxDirectoryRecursion 15\n");
+    fprintf(tmp,"SelfCheck 900\n");
+    //fprintf(tmp,"ScanArchive\n");
+    if(optc(opt, 'd')) {
+            stat(getargc(opt, 'd'), &sb);
+            switch(sb.st_mode & S_IFMT) {
+                    case S_IFREG:
+                    fprintf(tmp,"DatabaseDirectory %s\n",getargc(opt, 'd'));
+                    break;
+                    case S_IFDIR:
+                    fprintf(tmp,"DatabaseDirectory %s\n",getargc(opt, 'd'));
+                    break;
+                    default:
+                    mprintf("@%s: Not supported database file type\n", getargc(opt, 'd'));
+                    break;
+                    }
+    }else{
+            fprintf(tmp,"DatabaseDirectory /usr/local/share/clamav\n");
+    }
+    //fprintf(tmp,"ScanMail\n");
+    if ((strstr(cl_retver(), "0.8")) || (strstr(cl_retver(), "0.7")))
+        fprintf(tmp,"FixStaleSocket\n");
+    else
+#ifdef SUPPORT_CLAMAV_V095
+        fprintf(tmp,"FixStaleSocket yes\n");
+#else
+        fprintf(tmp,"FixStaleSocket TRUE\n");
+#endif
+
+    fflush(tmp);
+
+    fclose(tmp);
+    close(fd);
+
+    fullpath = (char*) mcalloc(200 + strlen(tmpnm) + 10, sizeof(char));
+
+    if(!getcwd(cwd, 200)) {
+            mprintf("@Can't get absolute pathname of current working directory.\n");
+            return;
+    }
+    sprintf(fullpath, "%s/%s", cwd, tmpnm);
+
+
+    if ( pipe(pfds) == -1 ) {
+      perror("Failed to create pipe; quitting\n");
+      exit(1);
+    }
+
+    if ( ( childpid=fork() ) == -1 ) {
+      perror("Failed to fork; quitting\n");
+      exit(2);
+    }
+
+    /* instead of STDIN -> clamdmail -> STDOUT, do
+    STDIN -> clamdmail -> spamc -> STDOUT
+    (previously incorrect as STDIN -> spamc -> clamdmail -> STDOUT)
+    */
+    if ( childpid == 0 ) {
+            if (setenv("PATH","/usr/local/sbin:/usr/sbin:/sbin:/usr/bin:/bin:/usr/local/bin:/opt/kde/bin",1) == 0)
+                    execlp("clamd", "clamd", "-c", fullpath,  NULL);
+    }
+
+    sleep(1);
+    unlink(fullpath);
+}