Michael Spencer / Mbed 2 deprecated LaOS

Important changes to repositories hosted on mbed.com

Mbed hosted mercurial repositories are deprecated and are due to be permanently deleted in July 2026.

To keep a copy of this software download the repository Zip archive or clone locally using Mercurial.

It is also possible to export all your personal repositories from the account settings page.

Dependencies:   mbed

NetServices/services/email/smtp/SMTPClient.cpp@1:f5ac63519541, 2014-03-05 (annotated)

Committer:
Michael J. Spencer
Date:
Wed Mar 05 06:14:02 2014 -0800
Revision:
1:f5ac63519541
Initial commit.

Who changed what in which revision?

UserRevisionLine numberNew contents of line
Michael J. Spencer 1:f5ac63519541 1
Michael J. Spencer 1:f5ac63519541 2 /*
Michael J. Spencer 1:f5ac63519541 3 Copyright (c) 2010 Donatien Garnier (donatiengar [at] gmail [dot] com) y Segundo Equipo
Michael J. Spencer 1:f5ac63519541 4
Michael J. Spencer 1:f5ac63519541 5 Permission is hereby granted, free of charge, to any person obtaining a copy
Michael J. Spencer 1:f5ac63519541 6 of this software and associated documentation files (the "Software"), to deal
Michael J. Spencer 1:f5ac63519541 7 in the Software without restriction, including without limitation the rights
Michael J. Spencer 1:f5ac63519541 8 to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
Michael J. Spencer 1:f5ac63519541 9 copies of the Software, and to permit persons to whom the Software is
Michael J. Spencer 1:f5ac63519541 10 furnished to do so, subject to the following conditions:
Michael J. Spencer 1:f5ac63519541 11
Michael J. Spencer 1:f5ac63519541 12 The above copyright notice and this permission notice shall be included in
Michael J. Spencer 1:f5ac63519541 13 all copies or substantial portions of the Software.
Michael J. Spencer 1:f5ac63519541 14
Michael J. Spencer 1:f5ac63519541 15 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
Michael J. Spencer 1:f5ac63519541 16 IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
Michael J. Spencer 1:f5ac63519541 17 FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
Michael J. Spencer 1:f5ac63519541 18 AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
Michael J. Spencer 1:f5ac63519541 19 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
Michael J. Spencer 1:f5ac63519541 20 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
Michael J. Spencer 1:f5ac63519541 21 THE SOFTWARE.
Michael J. Spencer 1:f5ac63519541 22 */
Michael J. Spencer 1:f5ac63519541 23 #include "core/netservice.h"
Michael J. Spencer 1:f5ac63519541 24 #include "SMTPClient.h"
Michael J. Spencer 1:f5ac63519541 25 #include "../util/base64.h"
Michael J. Spencer 1:f5ac63519541 26
Michael J. Spencer 1:f5ac63519541 27 #define __DEBUG
Michael J. Spencer 1:f5ac63519541 28 #include "dbg/dbg.h"
Michael J. Spencer 1:f5ac63519541 29
Michael J. Spencer 1:f5ac63519541 30 #define CHUNK_SIZE 256 //512
Michael J. Spencer 1:f5ac63519541 31 #define BUF_SIZE (CHUNK_SIZE + 1)
Michael J. Spencer 1:f5ac63519541 32
Michael J. Spencer 1:f5ac63519541 33 #define SMTP_REQUEST_TIMEOUT 15000
Michael J. Spencer 1:f5ac63519541 34 #define SMTP_PORT 25
Michael J. Spencer 1:f5ac63519541 35
Michael J. Spencer 1:f5ac63519541 36 SMTPClient::SMTPClient() : NetService(false) /*Not owned by the pool*/,
Michael J. Spencer 1:f5ac63519541 37 m_pCbItem(NULL), m_pCbMeth(NULL), m_pCb(NULL),
Michael J. Spencer 1:f5ac63519541 38 m_watchdog(), m_timeout(SMTP_REQUEST_TIMEOUT), m_pDnsReq(NULL), m_server(),
Michael J. Spencer 1:f5ac63519541 39 m_closed(true), m_state(SMTP_CLOSED), m_pMessage(NULL),
Michael J. Spencer 1:f5ac63519541 40 m_posInMsg(0), m_blockingResult(SMTP_PROCESSING),
Michael J. Spencer 1:f5ac63519541 41 m_username(""), m_password(""), m_auth(SMTP_AUTH_NONE),
Michael J. Spencer 1:f5ac63519541 42 m_heloDomain("localhost") {
Michael J. Spencer 1:f5ac63519541 43 DBG("New SMTPClient %p\n", this);
Michael J. Spencer 1:f5ac63519541 44 }
Michael J. Spencer 1:f5ac63519541 45
Michael J. Spencer 1:f5ac63519541 46 SMTPClient::SMTPClient(const Host& host, const char* heloDomain, const char* user, const char* password, SMTPAuth auth) : NetService(false),
Michael J. Spencer 1:f5ac63519541 47 m_pCbItem(NULL), m_pCbMeth(NULL), m_pCb(NULL),
Michael J. Spencer 1:f5ac63519541 48 m_watchdog(), m_timeout(SMTP_REQUEST_TIMEOUT), m_pDnsReq(NULL), m_server(host),
Michael J. Spencer 1:f5ac63519541 49 m_closed(true), m_state(SMTP_CLOSED), m_pMessage(NULL),
Michael J. Spencer 1:f5ac63519541 50 m_posInMsg(0), m_blockingResult(SMTP_PROCESSING),
Michael J. Spencer 1:f5ac63519541 51 m_username(string(user)), m_password(string(password)), m_auth(auth),
Michael J. Spencer 1:f5ac63519541 52 m_heloDomain(heloDomain) {
Michael J. Spencer 1:f5ac63519541 53 DBG("New SMTPClient %p\n", this);
Michael J. Spencer 1:f5ac63519541 54 }
Michael J. Spencer 1:f5ac63519541 55
Michael J. Spencer 1:f5ac63519541 56 SMTPClient::~SMTPClient() {
Michael J. Spencer 1:f5ac63519541 57 close();
Michael J. Spencer 1:f5ac63519541 58 }
Michael J. Spencer 1:f5ac63519541 59
Michael J. Spencer 1:f5ac63519541 60 void SMTPClient::setAuth(const char* user, const char* password) { // Plain authentication
Michael J. Spencer 1:f5ac63519541 61 m_username = string(user);
Michael J. Spencer 1:f5ac63519541 62 m_password = string(password);
Michael J. Spencer 1:f5ac63519541 63 m_auth = SMTP_AUTH_PLAIN;
Michael J. Spencer 1:f5ac63519541 64 }
Michael J. Spencer 1:f5ac63519541 65
Michael J. Spencer 1:f5ac63519541 66 void SMTPClient::clearAuth() { // Clear authentication
Michael J. Spencer 1:f5ac63519541 67 m_username = "";
Michael J. Spencer 1:f5ac63519541 68 m_password = "";
Michael J. Spencer 1:f5ac63519541 69 m_auth = SMTP_AUTH_NONE;
Michael J. Spencer 1:f5ac63519541 70 }
Michael J. Spencer 1:f5ac63519541 71
Michael J. Spencer 1:f5ac63519541 72 string SMTPClient::encodePlainAuth() {
Michael J. Spencer 1:f5ac63519541 73 string decStr = m_username;
Michael J. Spencer 1:f5ac63519541 74 decStr += '\0';
Michael J. Spencer 1:f5ac63519541 75 decStr += m_username;
Michael J. Spencer 1:f5ac63519541 76 decStr += '\0';
Michael J. Spencer 1:f5ac63519541 77 decStr += m_password;
Michael J. Spencer 1:f5ac63519541 78
Michael J. Spencer 1:f5ac63519541 79 string auth = "AUTH PLAIN ";
Michael J. Spencer 1:f5ac63519541 80 return auth.append(Base64::encode(decStr));
Michael J. Spencer 1:f5ac63519541 81 }
Michael J. Spencer 1:f5ac63519541 82
Michael J. Spencer 1:f5ac63519541 83 void SMTPClient::setHeloDomain(const char* heloDomain) {
Michael J. Spencer 1:f5ac63519541 84 m_heloDomain = string(heloDomain);
Michael J. Spencer 1:f5ac63519541 85 }
Michael J. Spencer 1:f5ac63519541 86
Michael J. Spencer 1:f5ac63519541 87 SMTPResult SMTPClient::send(EmailMessage* pMessage) { //Blocking
Michael J. Spencer 1:f5ac63519541 88 doSend(pMessage);
Michael J. Spencer 1:f5ac63519541 89 return blockingProcess();
Michael J. Spencer 1:f5ac63519541 90 }
Michael J. Spencer 1:f5ac63519541 91
Michael J. Spencer 1:f5ac63519541 92 SMTPResult SMTPClient::send(EmailMessage* pMessage, void (*pMethod)(SMTPResult)) { //Non blocking
Michael J. Spencer 1:f5ac63519541 93 setOnResult(pMethod);
Michael J. Spencer 1:f5ac63519541 94 doSend(pMessage);
Michael J. Spencer 1:f5ac63519541 95 return SMTP_PROCESSING;
Michael J. Spencer 1:f5ac63519541 96 }
Michael J. Spencer 1:f5ac63519541 97
Michael J. Spencer 1:f5ac63519541 98 void SMTPClient::doSend(EmailMessage* pMessage) {
Michael J. Spencer 1:f5ac63519541 99 setup(pMessage);
Michael J. Spencer 1:f5ac63519541 100 }
Michael J. Spencer 1:f5ac63519541 101
Michael J. Spencer 1:f5ac63519541 102 void SMTPClient::setOnResult( void (*pMethod)(SMTPResult) ) {
Michael J. Spencer 1:f5ac63519541 103 m_pCb = pMethod;
Michael J. Spencer 1:f5ac63519541 104 m_pCbItem = NULL;
Michael J. Spencer 1:f5ac63519541 105 m_pCbMeth = NULL;
Michael J. Spencer 1:f5ac63519541 106 }
Michael J. Spencer 1:f5ac63519541 107
Michael J. Spencer 1:f5ac63519541 108 void SMTPClient::setTimeout(int ms) {
Michael J. Spencer 1:f5ac63519541 109 m_timeout = ms;
Michael J. Spencer 1:f5ac63519541 110 }
Michael J. Spencer 1:f5ac63519541 111
Michael J. Spencer 1:f5ac63519541 112 void SMTPClient::poll() { //Called by NetServices
Michael J. Spencer 1:f5ac63519541 113 if ( (!m_closed) && (m_watchdog.read_ms() >= m_timeout) ) {
Michael J. Spencer 1:f5ac63519541 114 onTimeout();
Michael J. Spencer 1:f5ac63519541 115 }
Michael J. Spencer 1:f5ac63519541 116 }
Michael J. Spencer 1:f5ac63519541 117
Michael J. Spencer 1:f5ac63519541 118 void SMTPClient::resetTimeout() {
Michael J. Spencer 1:f5ac63519541 119 m_watchdog.reset();
Michael J. Spencer 1:f5ac63519541 120 m_watchdog.start();
Michael J. Spencer 1:f5ac63519541 121 }
Michael J. Spencer 1:f5ac63519541 122
Michael J. Spencer 1:f5ac63519541 123 void SMTPClient::init() { //Create and setup socket if needed
Michael J. Spencer 1:f5ac63519541 124 close(); //Remove previous elements
Michael J. Spencer 1:f5ac63519541 125 if (!m_closed) //Already opened
Michael J. Spencer 1:f5ac63519541 126 return;
Michael J. Spencer 1:f5ac63519541 127 m_state = SMTP_HELLO;
Michael J. Spencer 1:f5ac63519541 128 m_pTCPSocket = new TCPSocket;
Michael J. Spencer 1:f5ac63519541 129 m_pTCPSocket->setOnEvent(this, &SMTPClient::onTCPSocketEvent);
Michael J. Spencer 1:f5ac63519541 130 m_closed = false;
Michael J. Spencer 1:f5ac63519541 131 m_posInMsg = 0;
Michael J. Spencer 1:f5ac63519541 132 m_posInCRLF = 2;
Michael J. Spencer 1:f5ac63519541 133 m_response = "";
Michael J. Spencer 1:f5ac63519541 134 }
Michael J. Spencer 1:f5ac63519541 135
Michael J. Spencer 1:f5ac63519541 136 void SMTPClient::close() {
Michael J. Spencer 1:f5ac63519541 137 if (m_closed)
Michael J. Spencer 1:f5ac63519541 138 return;
Michael J. Spencer 1:f5ac63519541 139 m_state = SMTP_CLOSED;
Michael J. Spencer 1:f5ac63519541 140 m_closed = true; //Prevent recursive calling or calling on an object being destructed by someone else
Michael J. Spencer 1:f5ac63519541 141 m_watchdog.stop(); //Stop timeout
Michael J. Spencer 1:f5ac63519541 142 m_watchdog.reset();
Michael J. Spencer 1:f5ac63519541 143 m_pTCPSocket->resetOnEvent();
Michael J. Spencer 1:f5ac63519541 144 m_pTCPSocket->close();
Michael J. Spencer 1:f5ac63519541 145 delete m_pTCPSocket;
Michael J. Spencer 1:f5ac63519541 146 m_pTCPSocket = NULL;
Michael J. Spencer 1:f5ac63519541 147 if ( m_pDnsReq ) {
Michael J. Spencer 1:f5ac63519541 148 m_pDnsReq->close();
Michael J. Spencer 1:f5ac63519541 149 delete m_pDnsReq;
Michael J. Spencer 1:f5ac63519541 150 m_pDnsReq = NULL;
Michael J. Spencer 1:f5ac63519541 151 }
Michael J. Spencer 1:f5ac63519541 152 }
Michael J. Spencer 1:f5ac63519541 153
Michael J. Spencer 1:f5ac63519541 154 void SMTPClient::setServer(const Host& host) { //Setup request, make DNS Req if necessary
Michael J. Spencer 1:f5ac63519541 155 m_server = host;
Michael J. Spencer 1:f5ac63519541 156 }
Michael J. Spencer 1:f5ac63519541 157
Michael J. Spencer 1:f5ac63519541 158 void SMTPClient::setup(EmailMessage* pMessage) { //Setup request, make DNS Req if necessary
Michael J. Spencer 1:f5ac63519541 159
Michael J. Spencer 1:f5ac63519541 160 init(); //Initialize client in known state, create socket
Michael J. Spencer 1:f5ac63519541 161 m_pMessage = pMessage;
Michael J. Spencer 1:f5ac63519541 162 resetTimeout();
Michael J. Spencer 1:f5ac63519541 163
Michael J. Spencer 1:f5ac63519541 164 m_To = m_pMessage->m_lTo.begin(); // Point to first to recipient in TO list
Michael J. Spencer 1:f5ac63519541 165
Michael J. Spencer 1:f5ac63519541 166 //If port set to zero then use default port
Michael J. Spencer 1:f5ac63519541 167 if (!m_server.getPort()) {
Michael J. Spencer 1:f5ac63519541 168 m_server.setPort(SMTP_PORT);
Michael J. Spencer 1:f5ac63519541 169 DBG("Using default port %d\n", SMTP_PORT);
Michael J. Spencer 1:f5ac63519541 170 }
Michael J. Spencer 1:f5ac63519541 171
Michael J. Spencer 1:f5ac63519541 172 if (m_server.getIp().isNull()) {
Michael J. Spencer 1:f5ac63519541 173 //DNS query required
Michael J. Spencer 1:f5ac63519541 174 m_pDnsReq = new DNSRequest();
Michael J. Spencer 1:f5ac63519541 175 DBG("DNSRequest %p\r\n", m_pDnsReq);
Michael J. Spencer 1:f5ac63519541 176 m_pDnsReq->setOnReply(this, &SMTPClient::onDNSReply);
Michael J. Spencer 1:f5ac63519541 177 m_pDnsReq->resolve(&m_server);
Michael J. Spencer 1:f5ac63519541 178 return;
Michael J. Spencer 1:f5ac63519541 179 } else
Michael J. Spencer 1:f5ac63519541 180 connect();
Michael J. Spencer 1:f5ac63519541 181
Michael J. Spencer 1:f5ac63519541 182 }
Michael J. Spencer 1:f5ac63519541 183
Michael J. Spencer 1:f5ac63519541 184 void SMTPClient::connect() { //Start Connection
Michael J. Spencer 1:f5ac63519541 185 resetTimeout();
Michael J. Spencer 1:f5ac63519541 186 DBG("Connecting...\n");
Michael J. Spencer 1:f5ac63519541 187 m_pTCPSocket->connect(m_server);
Michael J. Spencer 1:f5ac63519541 188 }
Michael J. Spencer 1:f5ac63519541 189
Michael J. Spencer 1:f5ac63519541 190 void SMTPClient::onTCPSocketEvent(TCPSocketEvent e) {
Michael J. Spencer 1:f5ac63519541 191
Michael J. Spencer 1:f5ac63519541 192 DBG("Event %d in SMTPClient::onTCPSocketEvent()\n", e);
Michael J. Spencer 1:f5ac63519541 193
Michael J. Spencer 1:f5ac63519541 194 if (m_closed) {
Michael J. Spencer 1:f5ac63519541 195 DBG("WARN: Discarded\n");
Michael J. Spencer 1:f5ac63519541 196 return;
Michael J. Spencer 1:f5ac63519541 197 }
Michael J. Spencer 1:f5ac63519541 198
Michael J. Spencer 1:f5ac63519541 199 switch (e) {
Michael J. Spencer 1:f5ac63519541 200 case TCPSOCKET_READABLE:
Michael J. Spencer 1:f5ac63519541 201 resetTimeout();
Michael J. Spencer 1:f5ac63519541 202 process(false);
Michael J. Spencer 1:f5ac63519541 203 break;
Michael J. Spencer 1:f5ac63519541 204 case TCPSOCKET_WRITEABLE:
Michael J. Spencer 1:f5ac63519541 205 resetTimeout();
Michael J. Spencer 1:f5ac63519541 206 process(true);
Michael J. Spencer 1:f5ac63519541 207 break;
Michael J. Spencer 1:f5ac63519541 208 case TCPSOCKET_CONTIMEOUT:
Michael J. Spencer 1:f5ac63519541 209 case TCPSOCKET_CONRST:
Michael J. Spencer 1:f5ac63519541 210 case TCPSOCKET_CONABRT:
Michael J. Spencer 1:f5ac63519541 211 case TCPSOCKET_ERROR:
Michael J. Spencer 1:f5ac63519541 212 DBG("Connection error in SMTP Client.\n");
Michael J. Spencer 1:f5ac63519541 213 close();
Michael J. Spencer 1:f5ac63519541 214 onResult(SMTP_DISC);
Michael J. Spencer 1:f5ac63519541 215 break;
Michael J. Spencer 1:f5ac63519541 216 case TCPSOCKET_DISCONNECTED:
Michael J. Spencer 1:f5ac63519541 217 if (m_state != SMTP_BYE) {
Michael J. Spencer 1:f5ac63519541 218 DBG("Connection error in SMTP Client.\n");
Michael J. Spencer 1:f5ac63519541 219 close();
Michael J. Spencer 1:f5ac63519541 220 onResult(SMTP_DISC);
Michael J. Spencer 1:f5ac63519541 221 }
Michael J. Spencer 1:f5ac63519541 222 break;
Michael J. Spencer 1:f5ac63519541 223 }
Michael J. Spencer 1:f5ac63519541 224 }
Michael J. Spencer 1:f5ac63519541 225
Michael J. Spencer 1:f5ac63519541 226 void SMTPClient::onDNSReply(DNSReply r) {
Michael J. Spencer 1:f5ac63519541 227 if (m_closed) {
Michael J. Spencer 1:f5ac63519541 228 DBG("WARN: Discarded\n");
Michael J. Spencer 1:f5ac63519541 229 return;
Michael J. Spencer 1:f5ac63519541 230 }
Michael J. Spencer 1:f5ac63519541 231
Michael J. Spencer 1:f5ac63519541 232 if ( r != DNS_FOUND ) {
Michael J. Spencer 1:f5ac63519541 233 DBG("Could not resolve hostname.\n");
Michael J. Spencer 1:f5ac63519541 234 close();
Michael J. Spencer 1:f5ac63519541 235 onResult(SMTP_DNS);
Michael J. Spencer 1:f5ac63519541 236 return;
Michael J. Spencer 1:f5ac63519541 237 }
Michael J. Spencer 1:f5ac63519541 238
Michael J. Spencer 1:f5ac63519541 239 DBG("DNS Resolved to %d.%d.%d.%d\n", m_server.getIp()[0], m_server.getIp()[1], m_server.getIp()[2], m_server.getIp()[3]);
Michael J. Spencer 1:f5ac63519541 240 //If no error, m_server has been updated by m_pDnsReq so we're set to go !
Michael J. Spencer 1:f5ac63519541 241 m_pDnsReq->close();
Michael J. Spencer 1:f5ac63519541 242 delete m_pDnsReq;
Michael J. Spencer 1:f5ac63519541 243 m_pDnsReq = NULL;
Michael J. Spencer 1:f5ac63519541 244 connect();
Michael J. Spencer 1:f5ac63519541 245 }
Michael J. Spencer 1:f5ac63519541 246
Michael J. Spencer 1:f5ac63519541 247 void SMTPClient::onResult(SMTPResult r) { //Called when exchange completed or on failure
Michael J. Spencer 1:f5ac63519541 248 if (m_pCbItem && m_pCbMeth)
Michael J. Spencer 1:f5ac63519541 249 (m_pCbItem->*m_pCbMeth)(r);
Michael J. Spencer 1:f5ac63519541 250 else if (m_pCb)
Michael J. Spencer 1:f5ac63519541 251 m_pCb(r);
Michael J. Spencer 1:f5ac63519541 252 m_blockingResult = r; //Blocking mode
Michael J. Spencer 1:f5ac63519541 253 }
Michael J. Spencer 1:f5ac63519541 254
Michael J. Spencer 1:f5ac63519541 255 void SMTPClient::onTimeout() { //Connection has timed out
Michael J. Spencer 1:f5ac63519541 256 DBG("Timed out.\n");
Michael J. Spencer 1:f5ac63519541 257 close();
Michael J. Spencer 1:f5ac63519541 258 onResult(SMTP_TIMEOUT);
Michael J. Spencer 1:f5ac63519541 259 }
Michael J. Spencer 1:f5ac63519541 260
Michael J. Spencer 1:f5ac63519541 261 SMTPResult SMTPClient::blockingProcess() { //Called in blocking mode, calls Net::poll() until return code is available
Michael J. Spencer 1:f5ac63519541 262 //Disable callbacks
Michael J. Spencer 1:f5ac63519541 263 m_pCb = NULL;
Michael J. Spencer 1:f5ac63519541 264 m_pCbItem = NULL;
Michael J. Spencer 1:f5ac63519541 265 m_pCbMeth = NULL;
Michael J. Spencer 1:f5ac63519541 266 m_blockingResult = SMTP_PROCESSING;
Michael J. Spencer 1:f5ac63519541 267 do {
Michael J. Spencer 1:f5ac63519541 268 Net::poll();
Michael J. Spencer 1:f5ac63519541 269 } while (m_blockingResult == SMTP_PROCESSING);
Michael J. Spencer 1:f5ac63519541 270 Net::poll(); //Necessary for cleanup
Michael J. Spencer 1:f5ac63519541 271 return m_blockingResult;
Michael J. Spencer 1:f5ac63519541 272 }
Michael J. Spencer 1:f5ac63519541 273
Michael J. Spencer 1:f5ac63519541 274 int SMTPClient::rc(char* buf) { //Parse return code
Michael J. Spencer 1:f5ac63519541 275 int rc;
Michael J. Spencer 1:f5ac63519541 276 int len = sscanf(buf, "%d %*[^\r\n]\r\n", &rc);
Michael J. Spencer 1:f5ac63519541 277 if (len != 1)
Michael J. Spencer 1:f5ac63519541 278 return -1;
Michael J. Spencer 1:f5ac63519541 279 return rc;
Michael J. Spencer 1:f5ac63519541 280 }
Michael J. Spencer 1:f5ac63519541 281
Michael J. Spencer 1:f5ac63519541 282 bool SMTPClient::okPostAuthentication(int code) {
Michael J. Spencer 1:f5ac63519541 283 return (code == 250) || (code == 235);
Michael J. Spencer 1:f5ac63519541 284 }
Michael J. Spencer 1:f5ac63519541 285
Michael J. Spencer 1:f5ac63519541 286 void SMTPClient::process(bool writeable) { //Main state-machine
Michael J. Spencer 1:f5ac63519541 287
Michael J. Spencer 1:f5ac63519541 288 DBG("In state %d, writeable %d\n", m_state, writeable);
Michael J. Spencer 1:f5ac63519541 289
Michael J. Spencer 1:f5ac63519541 290 // If writeable but nothing to write then return
Michael J. Spencer 1:f5ac63519541 291 if (writeable && (m_state != SMTP_BODYMORE))
Michael J. Spencer 1:f5ac63519541 292 return;
Michael J. Spencer 1:f5ac63519541 293
Michael J. Spencer 1:f5ac63519541 294 // If not writeable then read
Michael J. Spencer 1:f5ac63519541 295 char buf[BUF_SIZE] = {0};
Michael J. Spencer 1:f5ac63519541 296 int responseCode = -1;
Michael J. Spencer 1:f5ac63519541 297 if (!writeable) {
Michael J. Spencer 1:f5ac63519541 298 bool firstBuf = true;
Michael J. Spencer 1:f5ac63519541 299 int read;
Michael J. Spencer 1:f5ac63519541 300 do { // read until nothing left but only process the response from first buffer
Michael J. Spencer 1:f5ac63519541 301 read = m_pTCPSocket->recv(buf, BUF_SIZE - 1);
Michael J. Spencer 1:f5ac63519541 302 if (firstBuf) {
Michael J. Spencer 1:f5ac63519541 303 m_response = string(buf);
Michael J. Spencer 1:f5ac63519541 304 responseCode = rc(buf);
Michael J. Spencer 1:f5ac63519541 305 firstBuf = false;
Michael J. Spencer 1:f5ac63519541 306 DBG("Response %d %d | %s", read, responseCode, buf);
Michael J. Spencer 1:f5ac63519541 307 }
Michael J. Spencer 1:f5ac63519541 308 } while (read > 0);
Michael J. Spencer 1:f5ac63519541 309 }
Michael J. Spencer 1:f5ac63519541 310
Michael J. Spencer 1:f5ac63519541 311 switch (m_state) {
Michael J. Spencer 1:f5ac63519541 312 case SMTP_HELLO:
Michael J. Spencer 1:f5ac63519541 313 if ( responseCode != 220 ) {
Michael J. Spencer 1:f5ac63519541 314 close();
Michael J. Spencer 1:f5ac63519541 315 onResult(SMTP_PRTCL);
Michael J. Spencer 1:f5ac63519541 316 return;
Michael J. Spencer 1:f5ac63519541 317 }
Michael J. Spencer 1:f5ac63519541 318 char* helloCommand;
Michael J. Spencer 1:f5ac63519541 319 if (m_auth == SMTP_AUTH_NONE) {
Michael J. Spencer 1:f5ac63519541 320 helloCommand = "HELO";
Michael J. Spencer 1:f5ac63519541 321 m_state = SMTP_FROM;
Michael J. Spencer 1:f5ac63519541 322 } else {
Michael J. Spencer 1:f5ac63519541 323 helloCommand = "EHLO";
Michael J. Spencer 1:f5ac63519541 324 m_state = SMTP_AUTH;
Michael J. Spencer 1:f5ac63519541 325 }
Michael J. Spencer 1:f5ac63519541 326 snprintf(buf, BUF_SIZE, "%s %s\r\n", helloCommand, m_heloDomain.c_str());
Michael J. Spencer 1:f5ac63519541 327 break;
Michael J. Spencer 1:f5ac63519541 328 case SMTP_AUTH:
Michael J. Spencer 1:f5ac63519541 329 if ( responseCode != 250 ) {
Michael J. Spencer 1:f5ac63519541 330 close();
Michael J. Spencer 1:f5ac63519541 331 onResult(SMTP_PRTCL);
Michael J. Spencer 1:f5ac63519541 332 return;
Michael J. Spencer 1:f5ac63519541 333 }
Michael J. Spencer 1:f5ac63519541 334 snprintf(buf, BUF_SIZE, "%s\r\n", encodePlainAuth().c_str());
Michael J. Spencer 1:f5ac63519541 335 m_state = SMTP_FROM;
Michael J. Spencer 1:f5ac63519541 336 break;
Michael J. Spencer 1:f5ac63519541 337 case SMTP_FROM:
Michael J. Spencer 1:f5ac63519541 338 if (!okPostAuthentication(responseCode)) {
Michael J. Spencer 1:f5ac63519541 339 close();
Michael J. Spencer 1:f5ac63519541 340 onResult(SMTP_PRTCL);
Michael J. Spencer 1:f5ac63519541 341 return;
Michael J. Spencer 1:f5ac63519541 342 }
Michael J. Spencer 1:f5ac63519541 343 snprintf(buf, BUF_SIZE, "MAIL FROM:<%s>\r\n", m_pMessage->m_from.c_str());
Michael J. Spencer 1:f5ac63519541 344 m_state = SMTP_TO;
Michael J. Spencer 1:f5ac63519541 345 break;
Michael J. Spencer 1:f5ac63519541 346 case SMTP_TO:
Michael J. Spencer 1:f5ac63519541 347 if ( responseCode != 250 ) {
Michael J. Spencer 1:f5ac63519541 348 close();
Michael J. Spencer 1:f5ac63519541 349 onResult(SMTP_PRTCL);
Michael J. Spencer 1:f5ac63519541 350 return;
Michael J. Spencer 1:f5ac63519541 351 }
Michael J. Spencer 1:f5ac63519541 352 snprintf(buf, BUF_SIZE, "RCPT TO:<%s>\r\n", (m_To++)->c_str());
Michael J. Spencer 1:f5ac63519541 353 if (m_To == m_pMessage->m_lTo.end())
Michael J. Spencer 1:f5ac63519541 354 m_state = SMTP_DATA;
Michael J. Spencer 1:f5ac63519541 355 break;
Michael J. Spencer 1:f5ac63519541 356 case SMTP_DATA:
Michael J. Spencer 1:f5ac63519541 357 if ( responseCode != 250 ) {
Michael J. Spencer 1:f5ac63519541 358 close();
Michael J. Spencer 1:f5ac63519541 359 onResult(SMTP_PRTCL);
Michael J. Spencer 1:f5ac63519541 360 return;
Michael J. Spencer 1:f5ac63519541 361 }
Michael J. Spencer 1:f5ac63519541 362 snprintf(buf, BUF_SIZE, "DATA\r\n");
Michael J. Spencer 1:f5ac63519541 363 m_state = SMTP_BODY;
Michael J. Spencer 1:f5ac63519541 364 break;
Michael J. Spencer 1:f5ac63519541 365 case SMTP_BODY:
Michael J. Spencer 1:f5ac63519541 366 if ( responseCode != 354 ) {
Michael J. Spencer 1:f5ac63519541 367 close();
Michael J. Spencer 1:f5ac63519541 368 onResult(SMTP_PRTCL);
Michael J. Spencer 1:f5ac63519541 369 return;
Michael J. Spencer 1:f5ac63519541 370 }
Michael J. Spencer 1:f5ac63519541 371 m_state = SMTP_BODYMORE;
Michael J. Spencer 1:f5ac63519541 372 buf[0] = '\0'; // clear buffer before carrying on into next state
Michael J. Spencer 1:f5ac63519541 373 case SMTP_BODYMORE:
Michael J. Spencer 1:f5ac63519541 374 if (strlen(buf) > 0) { // sending interrupted by a server response
Michael J. Spencer 1:f5ac63519541 375 close();
Michael J. Spencer 1:f5ac63519541 376 onResult(SMTP_PRTCL);
Michael J. Spencer 1:f5ac63519541 377 return;
Michael J. Spencer 1:f5ac63519541 378 }
Michael J. Spencer 1:f5ac63519541 379
Michael J. Spencer 1:f5ac63519541 380 if ( m_posInMsg < m_pMessage->m_content.length() ) { // if still something to send
Michael J. Spencer 1:f5ac63519541 381 int sendLen = 0;
Michael J. Spencer 1:f5ac63519541 382 while (sendLen < BUF_SIZE - 1) { // - 1 to allow room for extra dot or CR or LF
Michael J. Spencer 1:f5ac63519541 383 char c = m_pMessage->m_content.at(m_posInMsg++);
Michael J. Spencer 1:f5ac63519541 384 switch (c) { // thanks ExtraDotOutputStream.java (with extra check for naked CR)
Michael J. Spencer 1:f5ac63519541 385 case '.':
Michael J. Spencer 1:f5ac63519541 386 if (m_posInCRLF == 2) // add extra dot
Michael J. Spencer 1:f5ac63519541 387 buf[sendLen++] = '.';
Michael J. Spencer 1:f5ac63519541 388 m_posInCRLF = 0;
Michael J. Spencer 1:f5ac63519541 389 break;
Michael J. Spencer 1:f5ac63519541 390 case '\r':
Michael J. Spencer 1:f5ac63519541 391 if (m_posInCRLF == 1) // two CR in a row, so insert an LF first
Michael J. Spencer 1:f5ac63519541 392 buf[sendLen++] = '\n';
Michael J. Spencer 1:f5ac63519541 393 m_posInCRLF = 1;
Michael J. Spencer 1:f5ac63519541 394 break;
Michael J. Spencer 1:f5ac63519541 395 case '\n':
Michael J. Spencer 1:f5ac63519541 396 if (m_posInCRLF != 1) // convert naked LF to CRLF
Michael J. Spencer 1:f5ac63519541 397 buf[sendLen++] = '\r';
Michael J. Spencer 1:f5ac63519541 398 m_posInCRLF = 2;
Michael J. Spencer 1:f5ac63519541 399 break;
Michael J. Spencer 1:f5ac63519541 400 default:
Michael J. Spencer 1:f5ac63519541 401 if (m_posInCRLF == 1) { // convert naked CR to CRLF
Michael J. Spencer 1:f5ac63519541 402 buf[sendLen++] = '\n';
Michael J. Spencer 1:f5ac63519541 403 m_posInCRLF = 2;
Michael J. Spencer 1:f5ac63519541 404 } else
Michael J. Spencer 1:f5ac63519541 405 m_posInCRLF = 0; // we're no longer at the start of a line
Michael J. Spencer 1:f5ac63519541 406 break;
Michael J. Spencer 1:f5ac63519541 407 }
Michael J. Spencer 1:f5ac63519541 408 buf[sendLen++] = c;
Michael J. Spencer 1:f5ac63519541 409 if ( m_posInMsg == m_pMessage->m_content.length() )
Michael J. Spencer 1:f5ac63519541 410 break;
Michael J. Spencer 1:f5ac63519541 411 }
Michael J. Spencer 1:f5ac63519541 412 m_pTCPSocket->send( buf, sendLen );
Michael J. Spencer 1:f5ac63519541 413 DBG("Sending %d bytes of processed message content\n", sendLen);
Michael J. Spencer 1:f5ac63519541 414 } else {
Michael J. Spencer 1:f5ac63519541 415 if (m_posInCRLF == 0)
Michael J. Spencer 1:f5ac63519541 416 snprintf(buf, BUF_SIZE, "\r\n.\r\n");
Michael J. Spencer 1:f5ac63519541 417 else if (m_posInCRLF == 1)
Michael J. Spencer 1:f5ac63519541 418 snprintf(buf, BUF_SIZE, "\n.\r\n");
Michael J. Spencer 1:f5ac63519541 419 else
Michael J. Spencer 1:f5ac63519541 420 snprintf(buf, BUF_SIZE, ".\r\n");
Michael J. Spencer 1:f5ac63519541 421 m_state = SMTP_EOF;
Michael J. Spencer 1:f5ac63519541 422 }
Michael J. Spencer 1:f5ac63519541 423 break;
Michael J. Spencer 1:f5ac63519541 424 case SMTP_EOF:
Michael J. Spencer 1:f5ac63519541 425 if ( responseCode != 250 ) {
Michael J. Spencer 1:f5ac63519541 426 close();
Michael J. Spencer 1:f5ac63519541 427 onResult(SMTP_PRTCL);
Michael J. Spencer 1:f5ac63519541 428 return;
Michael J. Spencer 1:f5ac63519541 429 }
Michael J. Spencer 1:f5ac63519541 430 snprintf(buf, BUF_SIZE, "QUIT\r\n");
Michael J. Spencer 1:f5ac63519541 431 m_state = SMTP_BYE;
Michael J. Spencer 1:f5ac63519541 432 break;
Michael J. Spencer 1:f5ac63519541 433 case SMTP_BYE:
Michael J. Spencer 1:f5ac63519541 434 if ( responseCode != 221 ) {
Michael J. Spencer 1:f5ac63519541 435 close();
Michael J. Spencer 1:f5ac63519541 436 onResult(SMTP_PRTCL);
Michael J. Spencer 1:f5ac63519541 437 return;
Michael J. Spencer 1:f5ac63519541 438 }
Michael J. Spencer 1:f5ac63519541 439 close();
Michael J. Spencer 1:f5ac63519541 440 onResult(SMTP_OK);
Michael J. Spencer 1:f5ac63519541 441 return;
Michael J. Spencer 1:f5ac63519541 442 }
Michael J. Spencer 1:f5ac63519541 443
Michael J. Spencer 1:f5ac63519541 444 if ( m_state != SMTP_BODYMORE ) {
Michael J. Spencer 1:f5ac63519541 445 DBG("Sending | %s", buf);
Michael J. Spencer 1:f5ac63519541 446 m_pTCPSocket->send( buf, strlen(buf) );
Michael J. Spencer 1:f5ac63519541 447 }
Michael J. Spencer 1:f5ac63519541 448
Michael J. Spencer 1:f5ac63519541 449 }
Michael J. Spencer 1:f5ac63519541 450
Michael J. Spencer 1:f5ac63519541 451 string& SMTPClient::getLastResponse() { // Return last response set on result
Michael J. Spencer 1:f5ac63519541 452 return m_response;
Michael J. Spencer 1:f5ac63519541 453 }