Annotation of embedaddon/strongswan/src/libfast/fast_smtp.c, revision 1.1

1.1     ! misho       1: /*
        !             2:  * Copyright (C) 2010 Martin Willi
        !             3:  * Copyright (C) 2010 revosec AG
        !             4:  *
        !             5:  * This program is free software; you can redistribute it and/or modify it
        !             6:  * under the terms of the GNU General Public License as published by the
        !             7:  * Free Software Foundation; either version 2 of the License, or (at your
        !             8:  * option) any later version.  See <http://www.fsf.org/copyleft/gpl.txt>.
        !             9:  *
        !            10:  * This program is distributed in the hope that it will be useful, but
        !            11:  * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
        !            12:  * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
        !            13:  * for more details.
        !            14:  */
        !            15: 
        !            16: #include "fast_smtp.h"
        !            17: 
        !            18: #include <unistd.h>
        !            19: #include <errno.h>
        !            20: 
        !            21: #include <utils/debug.h>
        !            22: 
        !            23: typedef struct private_fast_smtp_t private_fast_smtp_t;
        !            24: 
        !            25: /**
        !            26:  * Private data of an fast_smtp_t object.
        !            27:  */
        !            28: struct private_fast_smtp_t {
        !            29: 
        !            30:        /**
        !            31:         * Public fast_smtp_t interface.
        !            32:         */
        !            33:        fast_smtp_t public;
        !            34: 
        !            35:        /**
        !            36:         * file stream to SMTP server
        !            37:         */
        !            38:        FILE *f;
        !            39: };
        !            40: 
        !            41: /**
        !            42:  * Read the response code from an SMTP server
        !            43:  */
        !            44: static int read_response(private_fast_smtp_t *this)
        !            45: {
        !            46:        char buf[256], *end;
        !            47:        int res = 0;
        !            48: 
        !            49:        while (TRUE)
        !            50:        {
        !            51:                if (!fgets(buf, sizeof(buf), this->f))
        !            52:                {
        !            53:                        return 0;
        !            54:                }
        !            55:                res = strtol(buf, &end, 10);
        !            56:                switch (*end)
        !            57:                {
        !            58:                        case '-':
        !            59:                                continue;
        !            60:                        case ' ':
        !            61:                        case '\0':
        !            62:                        case '\n':
        !            63:                                break;
        !            64:                        default:
        !            65:                                return 0;
        !            66:                }
        !            67:                break;
        !            68:        }
        !            69:        return res;
        !            70: }
        !            71: 
        !            72: /**
        !            73:  * write a SMTP command to the server, read response code
        !            74:  */
        !            75: static int write_cmd(private_fast_smtp_t *this, char *fmt, ...)
        !            76: {
        !            77:        char buf[256];
        !            78:        va_list args;
        !            79: 
        !            80:        va_start(args, fmt);
        !            81:        vsnprintf(buf, sizeof(buf), fmt, args);
        !            82:        va_end(args);
        !            83: 
        !            84:        if (fprintf(this->f, "%s\n", buf) < 1)
        !            85:        {
        !            86:                DBG1(DBG_LIB, "sending SMTP command failed");
        !            87:                return 0;
        !            88:        }
        !            89:        return read_response(this);
        !            90: }
        !            91: 
        !            92: METHOD(fast_smtp_t, send_mail, bool,
        !            93:        private_fast_smtp_t *this, char *from, char *to, char *subject, char *fmt, ...)
        !            94: {
        !            95:        va_list args;
        !            96: 
        !            97:        if (write_cmd(this, "MAIL FROM:<%s>", from) != 250)
        !            98:        {
        !            99:                DBG1(DBG_LIB, "SMTP MAIL FROM failed");
        !           100:                return FALSE;
        !           101:        }
        !           102:        if (write_cmd(this, "RCPT TO:<%s>", to) != 250)
        !           103:        {
        !           104:                DBG1(DBG_LIB, "SMTP RCPT TO failed");
        !           105:                return FALSE;
        !           106:        }
        !           107:        if (write_cmd(this, "DATA") != 354)
        !           108:        {
        !           109:                DBG1(DBG_LIB, "SMTP DATA failed");
        !           110:                return FALSE;
        !           111:        }
        !           112: 
        !           113:        fprintf(this->f, "From: %s\n", from);
        !           114:        fprintf(this->f, "To: %s\n", to);
        !           115:        fprintf(this->f, "Subject: %s\n", subject);
        !           116:        fprintf(this->f, "\n");
        !           117:        va_start(args, fmt);
        !           118:        vfprintf(this->f, fmt, args);
        !           119:        va_end(args);
        !           120:        fprintf(this->f, "\n.\n");
        !           121:        return read_response(this) == 250;
        !           122: }
        !           123: 
        !           124: 
        !           125: METHOD(fast_smtp_t, destroy, void,
        !           126:        private_fast_smtp_t *this)
        !           127: {
        !           128:        write_cmd(this, "QUIT");
        !           129:        fclose(this->f);
        !           130:        free(this);
        !           131: }
        !           132: 
        !           133: /**
        !           134:  * See header
        !           135:  */
        !           136: fast_smtp_t *fast_smtp_create()
        !           137: {
        !           138:        private_fast_smtp_t *this;
        !           139:        struct sockaddr_in addr = {
        !           140:                .sin_family = AF_INET,
        !           141:                .sin_port = htons(25),
        !           142:                .sin_addr = {
        !           143:                        .s_addr = htonl(INADDR_LOOPBACK),
        !           144:                },
        !           145:        };
        !           146:        int s;
        !           147: 
        !           148:        INIT(this,
        !           149:                .public = {
        !           150:                        .send_mail = _send_mail,
        !           151:                        .destroy = _destroy,
        !           152:                },
        !           153:        );
        !           154: 
        !           155:        s = socket(AF_INET, SOCK_STREAM, 0);
        !           156:        if (s < 0)
        !           157:        {
        !           158:                DBG1(DBG_LIB, "opening SMTP socket failed: %s", strerror(errno));
        !           159:                free(this);
        !           160:                return NULL;
        !           161:        }
        !           162:        if (connect(s, (struct sockaddr*)&addr, sizeof(addr)) < 0)
        !           163:        {
        !           164:                DBG1(DBG_LIB, "connecting to SMTP server failed: %s", strerror(errno));
        !           165:                close(s);
        !           166:                free(this);
        !           167:                return NULL;
        !           168:        }
        !           169:        this->f = fdopen(s, "a+");
        !           170:        if (!this->f)
        !           171:        {
        !           172:                DBG1(DBG_LIB, "opening stream to SMTP server failed: %s",
        !           173:                         strerror(errno));
        !           174:                close(s);
        !           175:                free(this);
        !           176:                return NULL;
        !           177:        }
        !           178:        if (read_response(this) != 220 ||
        !           179:                write_cmd(this, "EHLO localhost") != 250)
        !           180:        {
        !           181:                DBG1(DBG_LIB, "SMTP EHLO failed");
        !           182:                fclose(this->f);
        !           183:                free(this);
        !           184:                return NULL;
        !           185:        }
        !           186:        return &this->public;
        !           187: }

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>