// NeL - MMORPG Framework // Copyright (C) 2010 Winch Gate Property Limited // // This program is free software: you can redistribute it and/or modify // it under the terms of the GNU Affero General Public License as // published by the Free Software Foundation, either version 3 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 Affero General Public License for more details. // // You should have received a copy of the GNU Affero General Public License // along with this program. If not, see . #include "stdnet.h" #include "nel/misc/report.h" #include "nel/misc/path.h" #include "nel/net/tcp_sock.h" #include "nel/net/email.h" using namespace std; using namespace NLMISC; namespace NLNET { static string DefaultSMTPServer, DefaultFrom, DefaultTo; /* Conversion table. for base 64 */ static char tbl[65] = { 'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z', 'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '+', '/', '=' /* termination character */ }; /* * Encode the string S of length LENGTH to base64 format and place it * to STORE. STORE will be 0-terminated, and must point to a writable * buffer of at least 1+BASE64_LENGTH(length) bytes. * where BASE64_LENGTH(len) = (4 * ((LENGTH + 2) / 3)) */ static void uuencode (const char *s, const char *store, const int length) { int i; unsigned char *p = (unsigned char *)store; unsigned char *us = (unsigned char *)s; /* Transform the 3x8 bits to 4x6 bits, as required by base64. */ for (i = 0; i < length; i += 3) { *p++ = tbl[us[0] >> 2]; *p++ = tbl[((us[0] & 3) << 4) + (us[1] >> 4)]; *p++ = tbl[((us[1] & 0xf) << 2) + (us[2] >> 6)]; *p++ = tbl[us[2] & 0x3f]; us += 3; } /* Pad the result if necessary... */ if (i == length + 1) { *(p - 1) = tbl[64]; } else if (i == length + 2) { *(p - 1) = *(p - 2) = tbl[64]; } /* ...and zero-terminate it. */ *p = '\0'; } bool sendEMailCommand (CTcpSock &sock, const std::string &command, uint32 code = 250) { string buffer = command + "\r\n"; uint32 size = (uint32)buffer.size(); if(!command.empty()) { if (sock.send ((uint8 *)buffer.c_str(), size) != CSock::Ok) { nlwarning ("EMAIL: Can't send data to the server"); return false; } } string res; char c; for(;;) { size = 1; if (sock.receive((uint8*)&c, size, false) == CSock::Ok) { res += c; if (c == '\n') { uint32 c; fromString(res, c); if (c != code) { nlwarning ("EMAIL: EMail command '%s' returned '%s' instead of code %d on sock %s", command.substr(0, 20).c_str(), res.substr(0, res.size()-2).c_str(), code, sock.remoteAddr().asString().c_str()); return false; } return true; } } else { nlwarning ("EMAIL: EMail connection closed before end of line, command '%s' returned '%s' on sock %s (code %d)", command.substr(0, 20).c_str(), res.c_str(), sock.remoteAddr().asString().c_str(), code); return false; } } } bool sendEmail (const string &smtpServer, const string &from, const string &to, const string &subject, const string &body, const string &attachedFile, bool onlyCheck) { bool ok = false; CTcpSock sock; uint i; string formatedBody; string formatedFrom; string formatedTo; string formatedSMTPServer; try { if (smtpServer.empty()) { if(DefaultSMTPServer.empty()) { nlwarning ("EMAIL: Can't send email because no SMTPServer was provided"); goto end; } else { formatedSMTPServer = DefaultSMTPServer; } } else { formatedSMTPServer = smtpServer; } sock.connect(CInetAddress(formatedSMTPServer, 25)); if (!sock.connected()) { nlwarning ("EMAIL: Can't connect to email server %s", formatedSMTPServer.c_str()); goto end; } if (to.empty()) { if(DefaultTo.empty()) { nlwarning ("EMAIL: Can't send email because no To was provided"); goto end; } else { formatedTo = DefaultTo; } } else { formatedTo = to; } if(from.empty()) { if (DefaultFrom.empty()) { formatedFrom = CInetAddress::localHost().hostName(); formatedFrom += "@gnu.org"; } else { formatedFrom = DefaultFrom; } } else { formatedFrom = from; } // we must skip the first line formatedBody = "\r\n"; // replace \n with \r\n for (i = 0; i < body.size(); i++) { if (body[i] == '\n' && i > 0 && body[i-1] != '\r') { formatedBody += '\r'; } formatedBody += body[i]; } // add attachment if any if (!attachedFile.empty()) { string ext = CFile::getExtension(attachedFile); string mimepart; // mime header and main mail text mimepart += "Mime-Version: 1.0\r\n"; mimepart += "Content-Type: multipart/mixed;\r\n"; mimepart += " boundary=\"Multipart_nel\"\r\n"; mimepart += "\r\n"; mimepart += "This is a multi-part message in MIME format.\r\n"; mimepart += "\r\n"; mimepart += "--Multipart_nel\r\n"; mimepart += "Content-Type: text/plain; charset=us-ascii\r\n"; mimepart += "Content-Transfer-Encoding: 7bit\r\n"; formatedBody = mimepart + formatedBody; // mime attachment formatedBody += "--Multipart_nel\r\n"; formatedBody += "Content-Disposition: attachment;\r\n"; string lext = toLower(ext); if(lext == "tga") { formatedBody += "Content-Type: image/x-targa;\r\n"; } else if(lext == "bmp") { formatedBody += "Content-Type: image/bmp;\r\n"; } else if(lext == "png") { formatedBody += "Content-Type: image/png;\r\n"; } else if(lext == "jpg" || lext == "jpeg") { formatedBody += "Content-Type: image/jpeg;\r\n"; } else if(lext == "dmp") { formatedBody += "Content-Type: application/octet-stream;\r\n"; } else { formatedBody += "Content-Type: text/plain; charset=us-ascii\r\n"; } formatedBody += " name=\""+CFile::getFilename(attachedFile)+"\"\r\n"; formatedBody += "Content-Transfer-Encoding: base64\r\n"; formatedBody += " filename=\""+CFile::getFilename(attachedFile)+"\"\r\n"; // empty line to say that it s the end of the header formatedBody += "\r\n"; static const size_t src_buf_size = 45;// This *MUST* be a multiple of 3 static const size_t dst_buf_size = 4 * ((src_buf_size + 2) / 3); size_t write_size = dst_buf_size; char src_buf[src_buf_size + 1]; char dst_buf[dst_buf_size + 1]; size_t size; FILE *src_stream = fopen (attachedFile.c_str(), "rb"); if (src_stream == NULL) { nlwarning ("EMAIL: Can't attach file '%s' to the email because the file can't be open", attachedFile.c_str()); } else { while ((size = fread(src_buf, 1, src_buf_size, src_stream)) > 0) { if (size != src_buf_size) { /* write_size is always 60 until the last line */ write_size=(4 * ((size + 2) / 3)); /* pad with 0s so we can just encode extra bits */ memset(&src_buf[size], 0, src_buf_size - size); } /* Encode the buffer we just read in */ uuencode(src_buf, dst_buf, (int)size); formatedBody += dst_buf; formatedBody += "\r\n"; } fclose (src_stream); } formatedBody += "--Multipart_nel--"; } // debug, display what we send into a file // { FILE *fp = fopen (CFile::findNewFile(getLogDirectory() + "mail.txt").c_str(), "wb"); // fwrite (formatedBody.c_str(), 1, formatedBody.size(), fp); // fclose (fp); } if(!sendEMailCommand (sock, "", 220)) goto end; if(onlyCheck) { if(!sendEMailCommand (sock, "HELO localhost")) goto end; if(!sendEMailCommand (sock, "MAIL FROM: " + formatedFrom)) goto end; if(!sendEMailCommand (sock, "RCPT TO: " + formatedTo)) goto end; if(!sendEMailCommand (sock, "QUIT", 221)) goto end; ok = true; } else { if(!sendEMailCommand (sock, "HELO localhost")) goto end; if(!sendEMailCommand (sock, "MAIL FROM: " + formatedFrom)) goto end; if(!sendEMailCommand (sock, "RCPT TO: " + formatedTo)) goto end; if(!sendEMailCommand (sock, "DATA", 354)) goto end; string buffer = "From: " + formatedFrom + "\r\n" "To: " + formatedTo + "\r\n" "Subject: " + subject + "\r\n" + formatedBody + "\r\n."; if(!sendEMailCommand (sock, buffer)) goto end; if(!sendEMailCommand (sock, "QUIT", 221)) goto end; ok = true; } } catch (const Exception &e) { nlwarning ("EMAIL: Can't send email: %s", e.what()); goto end; } end: if (sock.connected()) sock.close (); return ok; } void setDefaultEmailParams (const std::string &smtpServer, const std::string &from, const std::string &to) { DefaultSMTPServer = smtpServer; DefaultFrom = from; DefaultTo = to; } } // NLNET