NetClientSmtp.cpp

//------------------------------------------------------------------------------
//This file is part of embKernel. 
//See license.txt for the full license governing this code.
//------------------------------------------------------------------------------

#include "NetClientSmtp.hpp"
#include "NetClientDns.hpp"
#include <stdarg.h>

const uint8_t NetClientSmtp::BASE64_INDEX_TABLE[] = {
        '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', '+', '/' };

bool NetClientSmtp::start(const char* server, bool authRequired, const char* user, const char* pass, const char* from, const char* to, const char* subject) {

    NetDefs::IP_ADDR serverAddr;

    if (!NetClientDns::resolve(server, &serverAddr)) {
        return false;
    }

    if (!mSocket.connect(serverAddr, 25)) {
        return false;
    }

    int replyCode = getReplyCode();
    if (replyCode < 200 || replyCode >= 300) {
        return false;
    }

    mSocket.printf ("HELO %s\r\n", user);
    replyCode = getReplyCode();
    if (replyCode < 200 || replyCode >= 300) {
        return false;
    }

    if (authRequired) {
        mSocket.printf ("AUTH LOGIN\r\n");
        replyCode = getReplyCode();
        if (replyCode < 300 || replyCode >= 400) {
            return false;
        }

        sendBase64(user);
        mSocket.printf ("\r\n");
        replyCode = getReplyCode();
        if (replyCode < 300 || replyCode >= 400) {
            return false;
        }

        sendBase64(pass);
        mSocket.printf ("\r\n");
        replyCode = getReplyCode();
        if (replyCode < 200 || replyCode >= 300) {
            return false;
        }
    }

    mSocket.printf ("MAIL FROM:<%s>\r\n", from);
    replyCode = getReplyCode();
    if (replyCode < 200 || replyCode >= 300) {
        return false;
    }

    mSocket.printf ("RCPT TO:<%s>\r\n", to);
    replyCode = getReplyCode();
    if (replyCode < 200 || replyCode >= 300) {
        return false;
    }

    mSocket.printf ("DATA\r\n");
    replyCode = getReplyCode();
    if (replyCode < 300 || replyCode >= 400) {
        return false;
    }

    mSocket.printf ("from:<%s>\r\nsubject:%s\r\n\r\n", from, subject);

    return true;
}

int NetClientSmtp::queueBody(const char* format, ...) {
    va_list vaList;
    int len;

    va_start(vaList, format);

    len = LibStringFormat::vprintf(mSocket,format, vaList);

    va_end(vaList);

    return len;
}

bool NetClientSmtp::end() {
    mSocket.printf("\r\n.\r\n");
    mSocket.flush();
    int replyCode = getReplyCode();
    if (replyCode < 200 || replyCode >= 300) {
        return false;
    }
    return true;
}

void NetClientSmtp::closeConnection() {
    mSocket.close();
}

void NetClientSmtp::sendBase64(const char* string) {
    int len = strlen(string);

    for (int i = 0; i < len;) {
        int sentChars = 1;
        uint32_t bitPattern = (uint32_t) string[i++] << 16;
        if (i < len) {
            bitPattern += (uint32_t) string[i++] << 8;
            sentChars++;
        }
        if (i < len) {
            bitPattern += (uint32_t) string[i++];
            sentChars++;
        }

        uint8_t encoded[4];

        encoded[0] = BASE64_INDEX_TABLE[(bitPattern >> 18) & 0x3F];
        encoded[1] = BASE64_INDEX_TABLE[(bitPattern >> 12) & 0x3F];
        encoded[2] = BASE64_INDEX_TABLE[(bitPattern >> 6) & 0x3F];
        encoded[3] = BASE64_INDEX_TABLE[(bitPattern >> 0) & 0x3F];

        if (sentChars == 1) {
            encoded[2] = '=';
            encoded[3] = '=';
        }
        else if (sentChars == 2) {
            encoded[3] = '=';
        }

        mSocket.write(encoded, 4);
    }
}

int NetClientSmtp::getReplyCode() {
    while (true) {

        uint8_t tmp[4];
        if (mSocket.read(tmp, 4, Rtos::convertMsToTick(2000)) != 4) {
            return -1;
        }

        int replyCode = ((int) (tmp[0] - '0') * 100) + ((int) (tmp[1] - '0') * 10) + ((int) (tmp[2] - '0'));

        bool isLastLine = true;
        if (tmp[3] == '-') {
            isLastLine = false;
        }

        while (true) { //Reach the end of the line
            if (mSocket.read(tmp, 1, Rtos::convertMsToTick(2000)) != 1) {
                return -1;
            }
            if (tmp[0] == '\n') {
                if (isLastLine) {
                    return replyCode;
                }
                break;
            }
        }
    }
}