354 lines
9.2 KiB
C++
354 lines
9.2 KiB
C++
|
// NeL - MMORPG Framework <http://dev.ryzom.com/projects/nel/>
|
||
|
|
// 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 <http://www.gnu.org/licenses/>.
|
||
|
|
|
||
|
|
#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 = 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, 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("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 (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
|