linux下基于SMTP协议的C++邮件客户端

完整代码下载：

https://github.com/WaPonX/SMTPMail

在网络中使用SMTP登陆的时候，需要将代码转换成base64编码。

下面这个函数是从网上抄的：

#include <string>

std::string Base64Encode(const std::string& src) {
	using std::string;
    int i, j, srcLen = src.length();
    string dst(srcLen / 3 * 4 + 4, 0);
    for(i = 0, j= 0; i <=srcLen - 3; i+=3, j+=4) {
	        dst[j] = (src[i] & 0xFC) >> 2;
	        dst[j+1] = ((src[i] & 0x03) << 4) + ((src[i+1] & 0xF0) >> 4);
	        dst[j+2] = ((src[i+1] & 0x0F) << 2) + ((src[i+2] & 0xC0) >> 6);
	        dst[j+3] = src[i+2] & 0x3F;
	    }
    if( srcLen % 3 == 1 ) {
	        dst[j] = (src[i] & 0xFC) >> 2;
	        dst[j+1] = ((src[i] & 0x03) << 4);
	        dst[j+2] = 64;
	        dst[j+3] = 64;
	        j += 4;
	    }
    else if( srcLen % 3 == 2 ) {
	        dst[j] = (src[i] & 0xFC) >> 2;
	        dst[j+1] = ((src[i] & 0x03) << 4) + ((src[i+1] & 0xF0) >> 4);
	        dst[j+2] = ((src[i+1] & 0x0F) << 2);
	        dst[j+3] = 64;
	        j+=4;
	    }

    static unsigned char *base64 = 
		(unsigned char*)("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/=");
    for(i = 0; i < j; ++i) {    //map 6 bit value to base64 ASCII character
	        dst[i] = base64[(int)dst[i]];
	    }
    
    return dst;
}

使用这个函数就可以完成base64转码

在linux下你可以使用如下命令：

echo string | base64

将string替换成你想要转换的字符串，输出的就是对应的base64编码。

#include <fcntl.h>
#include <unistd.h>
#include <stdio.h>
#include <string>
#include <time.h>



namespace Mail {
	bool OpenLogFile(int &log) {
		if ( (log = open("./mail.log", O_WRONLY | O_APPEND)) == -1) {
			printf("open log file error!\n");
			printf("please, make sure have a file called mail.log in the path\n");

			return false;
		}
		return true;
	}

	std::string GetTime() {
		time_t tt = time(NULL);

		return std::string(ctime(&tt));
	}
	void log(const char *str, size_t) {
		static int log = -1;
		if (log < 0) {
			if (!OpenLogFile(log)) {
				return ;
			}
		}
		std::string res(GetTime());
		res =  res + str + "\n";
		write (log, res.c_str(), res.length());
	}

}	// namespace Mail

对上面的三个函数进行说明：

1.

bool OpenLogFile(int &log);

这个函数包装了打开日志文件这个功能，使用了O_WRONLY | O_APPEND，因为只需要写，不需要读而且每次都要写到文件的最后。

2.

std::string GetTime();

这个函数为获取系统当前的时间，并按string类型返回。

3.

void log(const char *str, size_t);

这个函数就是将错误信息写入日志文件的函数，为什么第二个参数没有命名？一开始我是有命名的，后来发现这个参数没用，所以就不命名了。这样的做法可以避免编译器报错。是一个小技巧。

#include "base64encode.h"
#include "log.h"
#include "smtpmail.h"

#include <errno.h>
#include <netdb.h>
#include <netinet/in.h>
#include <stdio.h>
#include <string.h>
#include <sys/socket.h>


namespace {
	const std::string SMTPPORT("25");
}


namespace Mail {
	SMTPMail::SMTPMail (const String &username,
		const String &password,
		const String &host) :
	_islogin(false),
	_reuseaddr(true),
	_fd(-1),
	_paddrfilter(InitAddrInfoFilter()),
	_username(username),
	_password(Base64Encode(password)),
	_host(host){
		//_paddrfilter = InitAddrInfoFilter();
	}


	SMTPMail::~SMTPMail() {
		if (_fd > 0) {
			close(_fd);
		}
		if (_paddrfilter != NULL) {
			freeaddrinfo(_paddrfilter);
		}
	}

	SMTPMail::AddrType *SMTPMail::InitAddrInfoFilter() {
		AddrType *p = new AddrType;
		if (p == NULL) {
			return NULL;
		}
		memset(p, 0, sizeof(AddrType));
		//p->ai_flags = AI_CANONNAME;
		p->ai_family = AF_UNSPEC;
		p->ai_socktype = SOCK_STREAM;	
		struct addrinfo *res = NULL;
		int n = getaddrinfo(_host.c_str(), SMTPPORT.c_str(), p, &res);
		if (n != 0) {
			const char *str = gai_strerror(n);
			log(str, strlen(str));
			return NULL;
		}
		if (!Socket(res)) {
			freeaddrinfo(res);
			const char *str = strerror(errno);
			log(str, strlen(str));
			return NULL;
		}
		_islogin = true;

		return p;
	}

	bool SMTPMail::Socket(AddrType *res) {
		while (res != NULL) {
			_fd = socket(res->ai_family, res->ai_socktype, res->ai_protocol);
			if (_fd > 0) {
				if (setsockopt(_fd, SOL_SOCKET, SO_REUSEADDR, &_reuseaddr, sizeof(_reuseaddr))  == 0) {
					if (connect(_fd, res->ai_addr, sizeof(struct sockaddr)) == 0){
						return true;
					}
				}
				close(_fd);
				_fd = -1;
			}
			res = res->ai_next;
		}

		return false;	
	}
	
	bool SMTPMail::Login() {
		auto Check = [] (const String &res,
				const String &mode) -> void {
			if (res.substr(0, 3) != mode) {
				log(res.c_str(), res.length());
			}
		};

		Send(String("HELO "+ _username +"\r\n"));
		Check(Recv(), "250");
		
		Send(String("auth login\r\n"));
		Check(Recv(), "334");
		
		Send(Base64Encode(_username) + "\r\n");
		Check(Recv(), "334");

		Send(String(_password + "\r\n"));
		Check(Recv(), "334");

		return true;
	}

	void SMTPMail::SendEmail(const String &to, const String &text) {
		if(false == _islogin) {
			return;
		}
		if (!Login()) {
			return ;
		}
		Send(String("MAIL FROM: <" + _username + ">\r\n"));
		Recv();
	
		Send(String("RCPT TO: <" + to + ">\r\n"));
		Recv();
		
		Send(String("DATA\r\n"));
		Recv();

		Send(String("to:" + to + "\r\n" + text + "\r\n.\r\n"));
		Recv();

		Send(String("QUIT\r\n"));
		Recv();
	}

	void SMTPMail::Send(const String &msg) {
		if(send(_fd, msg.c_str(), msg.length(), MSG_CONFIRM) < 0) {
			const char *str = "send error\n";
			log(str, strlen(str));
		}
	}

	SMTPMail::String SMTPMail::Recv() {
		const size_t buflen = 512;
		static char buf[buflen];
		if (recv(_fd, buf, buflen, 0) < 0) {
			const char *str = "recv error\n";
			log(str, strlen(str));

			return String("");
		}

		return String(buf);
	}
}

在这段代码中我只挑一部分函数进行说明，毕竟有的函数很简单，没必要说明：

1.

SMTPMail::AddrType *SMTPMail::InitAddrInfoFilter();

这个函数是为了初始化AddrInfo类型的过滤器。因为是动态分配内存，而且是初始化成员，为了能在内存分配失败的时候更加容易的添加异常处理，我讲这部分代码独立出来。

需要注意的是AddrType是我自己typedef出来的一个类型:

typedef struct addrinfo AddrType;

struct addrinfo类型在删除的时候需要调用freeaddrinfo函数进行删除。

其实，这个做法还有一个更加明智的代替方法：

bool SMTPMail::Socket(AddrType *res)

使用智能指针。并且为这个智能智能指针定义自己的删除器。

具体做法请自行google

2.

bool SMTPMail::Socket(AddrType *res);

这个函数将尝试连接getaddrinfo返回的所有可用的地址，使用setsockopt函数设置端口可以复用，并连接。
如果成功，则返回true。