based on reworked servercore.

2020-02-29 19:25:13 -08:00
1 parent 50eb707f10
commit db3e82a18e
@@ -0,0 +1,89 @@
+#include "__SMTP_DATA.h"
+#include "SMTPServer.h"
+#include "Log.h"
+#include <chrono>
+#include <iomanip>
+
+namespace mail {
+      
+   int __SMTP_DATA::processCommand(coreutils::PString request, SMTPSession &session, SMTPServer &server, std::stringstream &data) {
+      
+      switch(session.mode) {
+	 
+       case WAIT_FOR_DATA:
+	 
+	 switch(session.state) {
+	    
+	  case RCPT:
+	    data << "354 Enter the mail message terminated by <CRLF>.<CRLF>" << CRLF;	 
+	    session.mailData.str("");
+	    session.mode = RECEIVE_DATA;
+	    session.server.commands.grabInput(&session, *this);		  
+	    //	       setTimer(120.0f);
+	    break;
+	    
+	  case MAIL:
+	    data << "503 Please use RCPT first" << CRLF;	 
+	    break;
+	       
+	  default:
+	    data << "503 Please use MAIL first" << CRLF;	 	    
+	    break;
+	    
+	 }
+	 break;
+	 
+       case RECEIVE_DATA:
+	 if(request.str() != ".") 
+	   session.mailData << request.str() << CRLF;	     	    	    
+	 else {		 
+	    session.mode = WAIT_FOR_DATA;  	       
+	    session.state = READY;
+	    session.server.commands.clearGrab(&session);
+	    //	       if(filterMessage(session.mailData)) {	    		    
+	    if(session.recipientList.size() > 0) {		      
+	       if(session.mailData.str().size() > 0) { 
+		  std::string ID = queueMail(server, session.sender, session.recipientList, session.mailData);
+		  if(ID != "") {     
+		     coreutils::Log(coreutils::LOG_INFO) << "Queued message " << ID << ".";
+		     data << "250 OK Queued message " << ID << CRLF;		  
+		  }
+		  else		     
+		    data << "550 Mail message too big" << CRLF; 
+	       } else 
+		 data << "550 Mail message was empty" << CRLF;		   
+	    } else 	       
+	      data << "250 OK Queued message " << CRLF; 
+	    //	       } else 	    
+	    //		 data << "550 Message is probably spam" << CRLF;
+	 }
+	 break;	 
+      }
+   }
+   
+   std::string __SMTP_DATA::generateMailFileName() {
+      std::stringstream temp;
+      struct timespec timex;
+      clock_gettime(CLOCK_REALTIME, &timex);
+      temp << timex.tv_sec << "." << std::setfill('0') << std::setw(9) << timex.tv_nsec; 
+      return temp.str();	
+   }      
+   
+   std::string __SMTP_DATA::queueMail(SMTPServer &server, std::string sender, std::vector<std::string> recipientList, std::stringstream &mailData) { 
+      std::string fileName = server.mailFileSystem.getMailPath() + "/.queue/" + generateMailFileName();
+      coreutils::File mailFile(fileName, O_CREAT | O_WRONLY, 0660);
+      mailFile.write(mailData.str());      
+
+      for(std::string recipient: recipientList) {
+	 std::string newName = fileName + recipient;
+	 link(fileName.c_str(), newName.c_str());
+      }
+      
+      unlink(fileName.c_str());
+      
+      return fileName;
+   }
+
+}
+
+