src/Sockets/SmtpdSocket.cpp

   1 /**
   2  **     \file SmtpdSocket.cpp
   3  **     \date  2007-05-10
   4  **     \author grymse@alhem.net
   5 **/
   6 /*
   7 Copyright (C) 2007  Anders Hedstrom
   8
   9 This program is free software; you can redistribute it and/or
  10 modify it under the terms of the GNU General Public License
  11 as published by the Free Software Foundation; either version 2
  12 of the License, or (at your option) any later version.
  13
  14 This program is distributed in the hope that it will be useful,
  15 but WITHOUT ANY WARRANTY; without even the implied warranty of
  16 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17 GNU General Public License for more details.
  18
  19 You should have received a copy of the GNU General Public License
  20 along with this program; if not, write to the Free Software
  21 Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  22 */
  23 #include "SmtpdSocket.h"
  24 #include "Parse.h"
  25 #include "Utility.h"
  26
  27
  28 SmtpdSocket::SmtpdSocket(ISocketHandler& h)
  29 :TcpSocket(h)
  30 ,m_hello(false)
  31 ,m_data(false)
  32 ,m_header(false)
  33 {
  34         SetLineProtocol();
  35 }
  36
  37
  38 void SmtpdSocket::OnAccept()
  39 {
  40         Send("220 ESMTP; \r\n");
  41 }
  42
  43
  44 void SmtpdSocket::OnLine(const std::string& line)
  45 {
  46         if (m_data)
  47         {
  48                 if (m_header)
  49                 {
  50                         if (!line.size())
  51                         {
  52                                 if (m_header_line.size())
  53                                 {
  54                                         Parse pa(m_header_line, ":");
  55                                         std::string key = pa.getword();
  56                                         OnHeader(key, pa.getrest());
  57                                 }
  58                                 m_header = false;
  59                                 OnHeaderComplete();
  60                         }
  61                         else
  62                         if (line[0] == ' ' || line[0] == '\t')
  63                         {
  64                                 m_header_line += line;
  65                         }
  66                         else
  67                         {
  68                                 if (m_header_line.size())
  69                                 {
  70                                         Parse pa(m_header_line, ":");
  71                                         std::string key = pa.getword();
  72                                         OnHeader(key, pa.getrest());
  73                                 }
  74                                 m_header_line = line;
  75                         }
  76                 }
  77                 else
  78                 if (line == ".")
  79                 {
  80                         m_data = false;
  81                         if (OnDataComplete())
  82                                 Send("250 OK\r\n");
  83                         else
  84                                 Send("550 Failed\r\n");
  85                 }
  86                 else
  87                 if (line.size() && line[0] == '.')
  88                 {
  89                         OnData(line.substr(1));
  90                 }
  91                 else
  92                 {
  93                         OnData(line);
  94                 }
  95                 return;
  96         }
  97         Parse pa(line);
  98         std::string cmd = Utility::ToUpper(pa.getword());
  99
 100         if (cmd == "EHLO")
 101         {
 102                 if (!OnHello(pa.getrest()))
 103                 {
 104                         Send("550 Failed\r\n");
 105                 }
 106                 else
 107                 {
 108                         m_hello = true;
 109                         Send("250 mail.alhem.net\r\n");
 110                 }
 111         }
 112         else
 113         if (cmd == "HELO")
 114         {
 115                 if (!OnHello(pa.getrest()))
 116                 {
 117                         Send("550 Failed\r\n");
 118                 }
 119                 else
 120                 {
 121                         m_hello = true;
 122                         Send("250 mail.alhem.net\r\n");
 123                 }
 124         }
 125         else
 126         if (!m_hello)
 127         {
 128                 OnAbort(SMTP_NO_HELLO);
 129                 SetCloseAndDelete();
 130         }
 131         else
 132         if (cmd == "MAIL") // mail from:
 133         {
 134                 Parse pa(line, ":");
 135                 pa.getword(); // 'mail'
 136                 pa.getword(); // 'from'
 137                 std::string email = Utility::ToLower(pa.getrest());
 138
 139                 EmailAddress addr( email );
 140                 if (addr.GetName().size() > 64)
 141                 {
 142                         OnAbort(SMTP_NAME_TOO_LONG);
 143                         Send("500 Name too long.\r\n");
 144                         return;
 145                 }
 146                 if (addr.GetDomain().size() > 64)
 147                 {
 148                         OnAbort(SMTP_DOMAIN_TOO_LONG);
 149                         Send("500 Domain too long.\r\n");
 150                         return;
 151                 }
 152
 153                 if (!OnMailFrom( addr ))
 154                 {
 155                         Send("550 Failed\r\n");
 156                 }
 157                 else
 158                 {
 159                         Send("250 OK\r\n");
 160                 }
 161         }
 162         else
 163         if (cmd == "RCPT") // rcpt to:
 164         {
 165                 Parse pa(line, ":");
 166                 pa.getword(); // 'rcpt'
 167                 pa.getword(); // 'to'
 168                 std::string email = Utility::ToLower(pa.getrest());
 169                 // %! reject based on user / domain?
 170                 EmailAddress addr( email );
 171
 172                 if (addr.GetName().size() > 64)
 173                 {
 174                         OnAbort(SMTP_NAME_TOO_LONG);
 175                         Send("500 Name too long.\r\n");
 176                         return;
 177                 }
 178                 if (addr.GetDomain().size() > 64)
 179                 {
 180                         OnAbort(SMTP_DOMAIN_TOO_LONG);
 181                         Send("500 Domain too long.\r\n");
 182                         return;
 183                 }
 184
 185                 if (!OnRcptTo( addr ))
 186                 {
 187                         Send("553 Failed\r\n");
 188                 }
 189                 else
 190                 {
 191                         Send("250 OK\r\n");
 192                 }
 193         }
 194         else
 195         if (cmd == "DATA")
 196         {
 197                 Send("354 Enter mail, end with \".\" on a line by itself\r\n");
 198                 m_data = true;
 199                 m_header = false;
 200         }
 201         else
 202         if (cmd == "RSET")
 203         {
 204                 m_data = false;
 205                 m_header = false;
 206                 OnRset();
 207                 Send("250 OK\r\n"); // %! ???
 208         }
 209         else
 210         if (cmd == "QUIT")
 211         {
 212                 OnAbort(SMTP_QUIT);
 213                 Send("221 Bye Bye\r\n");
 214                 SetCloseAndDelete();
 215         }
 216         else
 217         if (cmd == "NOOP")
 218         {
 219                 Send("250 OK\r\n");
 220         }
 221         else
 222         {
 223                 OnNotSupported(cmd, pa.getrest());
 224         }
 225 }
 226
 227