From fffd213c8cba2135afda493d797c41c10354770e Mon Sep 17 00:00:00 2001 From: Othmar Gsenger Date: Sat, 12 Apr 2008 11:38:42 +0000 Subject: big svn cleanup --- src/Sockets/SmtpdSocket.h | 143 ++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 143 insertions(+) create mode 100644 src/Sockets/SmtpdSocket.h (limited to 'src/Sockets/SmtpdSocket.h') diff --git a/src/Sockets/SmtpdSocket.h b/src/Sockets/SmtpdSocket.h new file mode 100644 index 0000000..d884d05 --- /dev/null +++ b/src/Sockets/SmtpdSocket.h @@ -0,0 +1,143 @@ +/** + ** \file SmtpdSocket.h + ** \date 2007-05-10 + ** \author grymse@alhem.net +**/ +/* +Copyright (C) 2007 Anders Hedstrom + +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., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. +*/ +#ifndef _SOCKETS_SmtpdSocket_H +#define _SOCKETS_SmtpdSocket_H + +#include "sockets-config.h" +#include +#include "TcpSocket.h" + +#ifdef SOCKETS_NAMESPACE +namespace SOCKETS_NAMESPACE { +#endif + + +/** Smtp server base class. */ +class SmtpdSocket : public TcpSocket +{ +protected: + typedef enum { + SMTP_NO_HELLO, + SMTP_NAME_TOO_LONG, + SMTP_DOMAIN_TOO_LONG, + SMTP_QUIT + } reason_t; + +public: + class EmailAddress { + public: + EmailAddress(const std::string& str_in) + { + std::string str = str_in; + size_t i = str.find("<"); + if (i != std::string::npos) + str = str.substr(i + 1); + i = str.find("@"); + if (i != std::string::npos) + { + m_name = str.substr(0, i); + str = str.substr(i + 1); + i = str.find(">"); + if (i != std::string::npos) + str = str.substr(0, i); + m_domain = str; + } + while (m_name.size() && m_name[m_name.size() - 1] == ' ') + m_name.resize(m_name.size() - 1); + while (m_domain.size() && m_domain[m_domain.size() - 1] == ' ') + m_domain.resize(m_domain.size() - 1); + while (m_name.size() && m_name[0] == ' ') + m_name = m_name.substr(1); + while (m_domain.size() && m_domain[0] == ' ') + m_domain = m_domain.substr(1); + m_top = m_domain; + { + for (size_t i = 0; i < m_domain.size(); i++) + { + if (m_domain[i] == '.') + { + m_sub = m_top; + m_top = m_domain.substr(i + 1); + } + } + } + } + + const std::string& GetName() const { return m_name; } + const std::string& GetDomain() const { return m_domain; } + const std::string& GetTopDomain() const { return m_top; } + const std::string& GetSubDomain() const { return m_sub; } + + std::string ToString() const { return m_name + "@" + m_domain; } + + private: + std::string m_name; + std::string m_domain; + std::string m_top; + std::string m_sub; + }; + +public: + SmtpdSocket(ISocketHandler&); + + void OnAccept(); + void OnLine(const std::string&); + + /** \return 'false' to abort */ + virtual bool OnHello(const std::string& domain) = 0; + + /** \return 'false' to abort */ + virtual bool OnMailFrom(const EmailAddress& addr) = 0; + + /** \return 'false' to abort */ + virtual bool OnRcptTo(const EmailAddress& addr) = 0; + + virtual void OnHeader(const std::string& key, const std::string& value) = 0; + + virtual void OnHeaderComplete() = 0; + + virtual void OnData(const std::string& line) = 0; + + /** \return 'false' if message write failed (message will probably be resent) */ + virtual bool OnDataComplete() = 0; + + virtual void OnRset() = 0; + + virtual void OnAbort(reason_t) = 0; + + virtual void OnNotSupported(const std::string& cmd, const std::string& arg) = 0; + +private: + bool m_hello; // we need HELO or EHLO first of all + bool m_data; + bool m_header; + std::string m_header_line; +}; + + +#ifdef SOCKETS_NAMESPACE +} // namespace SOCKETS_NAMESPACE { +#endif + +#endif // _SOCKETS_SmtpdSocket_H + -- cgit v1.2.3