HTTPClient using static IP

Dependencies:   mbed

Committer:
mr_q
Date:
Mon May 30 11:53:37 2011 +0000
Revision:
0:d8f2f7d5f31b
v0.01 Draft

Who changed what in which revision?

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