Nino Der Kinderen / Mbed 2 deprecated InetTest

Embedded C project:18/12/2014

Dependencies:   DS1307 TextLCD mbed

NetServices/services/email/smtp/SMTPClient.cpp@0:8d87bc453349, 2014-12-18 (annotated)

Committer:
ninoderkinderen
Date:
Thu Dec 18 09:35:49 2014 +0000
Revision:
0:8d87bc453349
Programma embedded C

Who changed what in which revision?

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